mirror of
https://github.com/OpenBMB/ChatDev.git
synced 2024-11-07 18:40:13 +03:00
130 lines
4.9 KiB
Python
130 lines
4.9 KiB
Python
# =========== Copyright 2023 @ CAMEL-AI.org. All Rights Reserved. ===========
|
|
# Licensed under the Apache License, Version 2.0 (the “License”);
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an “AS IS” BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
# =========== Copyright 2023 @ CAMEL-AI.org. All Rights Reserved. ===========
|
|
from typing import Any, Dict, Sequence
|
|
|
|
from colorama import Fore
|
|
|
|
from camel.messages import ChatMessage
|
|
from camel.utils import print_text_animated
|
|
|
|
|
|
class Human:
|
|
r"""A class representing a human user.
|
|
|
|
Args:
|
|
name (str): The name of the human user.
|
|
(default: :obj:`"Kill Switch Engineer"`).
|
|
logger_color (Any): The color of the menu options displayed to the
|
|
user. (default: :obj:`Fore.MAGENTA`)
|
|
|
|
Attributes:
|
|
name (str): The name of the human user.
|
|
logger_color (Any): The color of the menu options displayed to the
|
|
user.
|
|
input_button (str): The text displayed for the input button.
|
|
kill_button (str): The text displayed for the kill button.
|
|
options_dict (Dict[str, str]): A dictionary containing the options
|
|
displayed to the user.
|
|
"""
|
|
|
|
def __init__(self, name: str = "Kill Switch Engineer",
|
|
logger_color: Any = Fore.MAGENTA) -> None:
|
|
self.name = name
|
|
self.logger_color = logger_color
|
|
self.input_button = f"Input by {self.name}."
|
|
self.kill_button = "Stop!!!"
|
|
self.options_dict: Dict[str, str] = dict()
|
|
|
|
def display_options(self, messages: Sequence[ChatMessage]) -> None:
|
|
r"""Displays the options to the user.
|
|
|
|
Args:
|
|
messages (Sequence[ChatMessage]): A list of `ChatMessage` objects.
|
|
|
|
Returns:
|
|
None
|
|
"""
|
|
options = [message.content for message in messages]
|
|
options.append(self.input_button)
|
|
options.append(self.kill_button)
|
|
print_text_animated(
|
|
self.logger_color + "\n> Proposals from "
|
|
f"{messages[0].role_name} ({messages[0].role_type}). "
|
|
"Please choose an option:\n")
|
|
for index, option in enumerate(options):
|
|
print_text_animated(
|
|
self.logger_color +
|
|
f"\x1b[3mOption {index + 1}:\n{option}\x1b[0m\n")
|
|
self.options_dict[str(index + 1)] = option
|
|
|
|
def get_input(self) -> str:
|
|
r"""Gets the input from the user.
|
|
|
|
Returns:
|
|
str: The user's input.
|
|
"""
|
|
while True:
|
|
human_input = input(
|
|
self.logger_color +
|
|
f"Please enter your choice ([1-{len(self.options_dict)}]): ")
|
|
print("\n")
|
|
if human_input in self.options_dict:
|
|
break
|
|
print_text_animated(self.logger_color +
|
|
"\n> Invalid choice. Please try again.\n")
|
|
|
|
return human_input
|
|
|
|
def parse_input(self, human_input: str,
|
|
meta_chat_message: ChatMessage) -> ChatMessage:
|
|
r"""Parses the user's input and returns a `ChatMessage` object.
|
|
|
|
Args:
|
|
human_input (str): The user's input.
|
|
meta_chat_message (ChatMessage): A `ChatMessage` object.
|
|
|
|
Returns:
|
|
ChatMessage: A `ChatMessage` object.
|
|
"""
|
|
if self.options_dict[human_input] == self.input_button:
|
|
meta_chat_message.content = input(self.logger_color +
|
|
"Please enter your message: ")
|
|
return meta_chat_message
|
|
elif self.options_dict[human_input] == self.kill_button:
|
|
exit(self.logger_color + f"Killed by {self.name}.")
|
|
else:
|
|
meta_chat_message.content = self.options_dict[human_input]
|
|
return meta_chat_message
|
|
|
|
def step(self, messages: Sequence[ChatMessage]) -> ChatMessage:
|
|
r"""Performs one step of the conversation by displaying options to the
|
|
user, getting their input, and parsing their choice.
|
|
|
|
Args:
|
|
messages (Sequence[ChatMessage]): A list of ChatMessage objects.
|
|
|
|
Returns:
|
|
ChatMessage: A `ChatMessage` object representing the user's choice.
|
|
"""
|
|
meta_chat_message = ChatMessage(
|
|
role_name=messages[0].role_name,
|
|
role_type=messages[0].role_type,
|
|
meta_dict=messages[0].meta_dict,
|
|
role=messages[0].role,
|
|
content="",
|
|
)
|
|
self.display_options(messages)
|
|
human_input = self.get_input()
|
|
return self.parse_input(human_input, meta_chat_message)
|