Removed inheritance of TaskSpecifier, TaskPlanner and CriticAgent from ChatAgent

camel/agents/critic_agent.py

@@ -23,7 +23,7 @@
 from camel.utils import get_first_int, print_text_animated
 
 
-class CriticAgent(ChatAgent):
+class CriticAgent:
     r"""A class for the critic agent that assists in selecting an option.
 
     Args:
@@ -53,8 +53,8 @@ def __init__(
         verbose: bool = False,
         logger_color: Any = Fore.MAGENTA,
     ) -> None:
-        super().__init__(system_message, model, model_config,
-                         message_window_size)
+        self.chat_agent = ChatAgent(system_message, model, model_config,
+                                    message_window_size)
         self.options_dict: Dict[str, str] = dict()
         self.retry_attempts = retry_attempts
         self.verbose = verbose
@@ -96,15 +96,15 @@ def get_option(self, input_message: BaseMessage) -> str:
         msg_content = input_message.content
         i = 0
         while i < self.retry_attempts:
-            critic_response = super().step(input_message)
+            critic_response = self.chat_agent.step(input_message)
 
             if critic_response.msgs is None or len(critic_response.msgs) == 0:
                 raise RuntimeError("Got None critic messages.")
             if critic_response.terminated:
                 raise RuntimeError("Critic step failed.")
 
             critic_msg = critic_response.msg
-            self.update_messages('assistant', critic_msg)
+            self.chat_agent.update_messages('assistant', critic_msg)
             if self.verbose:
                 print_text_animated(self.logger_color + "\n> Critic response: "
                                     f"\x1b[3m{critic_msg.content}\x1b[0m\n")

camel/agents/task_agent.py

@@ -20,7 +20,7 @@
 from camel.typing import ModelType, RoleType, TaskType
 
 
-class TaskSpecifyAgent(ChatAgent):
+class TaskSpecifyAgent:
     r"""An agent that specifies a given task prompt by prompting the user to
     provide more details.
 
@@ -54,6 +54,7 @@ def __init__(
         output_language: Optional[str] = None,
     ) -> None:
 
+        self.task_specify_prompt: Union[str, TextPrompt]
         if task_specify_prompt is None:
             task_specify_prompt_template = PromptTemplateGenerator(
             ).get_task_specify_prompt(task_type)
@@ -72,10 +73,10 @@ def __init__(
             content="You can make a task more specific.",
         )
 
-        super().__init__(system_message, model, model_config,
-                         output_language=output_language)
+        self.chat_agent = ChatAgent(system_message, model, model_config,
+                                    output_language=output_language)
 
-    def step(
+    def specify_prompt(
         self,
         original_task_prompt: Union[str, TextPrompt],
         meta_dict: Optional[Dict[str, Any]] = None,
@@ -92,7 +93,8 @@ def step(
         Returns:
             TextPrompt: The specified task prompt.
         """
-        self.reset()
+        self.chat_agent.reset()
+
         self.task_specify_prompt = self.task_specify_prompt.format(
             task=original_task_prompt)
 
@@ -102,7 +104,7 @@ def step(
 
         task_msg = BaseMessage.make_user_message(
             role_name="Task Specifier", content=self.task_specify_prompt)
-        specifier_response = super().step(task_msg)
+        specifier_response = self.chat_agent.step(task_msg)
         if (specifier_response.msgs is None
                 or len(specifier_response.msgs) == 0):
             raise RuntimeError("Task specification failed.")
@@ -114,7 +116,7 @@ def step(
         return TextPrompt(specified_task_msg.content)
 
 
-class TaskPlannerAgent(ChatAgent):
+class TaskPlannerAgent:
     r"""An agent that helps divide a task into subtasks based on the input
     task prompt.
 
@@ -148,10 +150,10 @@ def __init__(
             content="You are a helpful task planner.",
         )
 
-        super().__init__(system_message, model, model_config,
-                         output_language=output_language)
+        self.chat_agent = ChatAgent(system_message, model, model_config,
+                                    output_language=output_language)
 
-    def step(
+    def create_plan(
         self,
         task_prompt: Union[str, TextPrompt],
     ) -> TextPrompt:
@@ -165,14 +167,14 @@ def step(
             TextPrompt: A prompt for the subtasks generated by the agent.
         """
         # TODO: Maybe include roles information.
-        self.reset()
+        self.chat_agent.reset()
+
         self.task_planner_prompt = self.task_planner_prompt.format(
             task=task_prompt)
 
         task_msg = BaseMessage.make_user_message(
             role_name="Task Planner", content=self.task_planner_prompt)
-        # sub_tasks_msgs, terminated, _
-        task_response = super().step(task_msg)
+        task_response = self.chat_agent.step(task_msg)
 
         if task_response.msgs is None:
             raise RuntimeError("Got None Subtasks messages.")

camel/messages/base.py

@@ -221,7 +221,7 @@ def to_openai_assistant_message(self) -> OpenAIAssistantMessage:
         """
         return {"role": "assistant", "content": self.content}
 
-    def to_dict(self) -> Dict:
+    def to_dict(self) -> Dict[str, Any]:
         r"""Converts the message to a dictionary.
 
         Returns:

camel/societies/role_playing.py

@@ -112,7 +112,7 @@ def __init__(
                 output_language=output_language,
                 **(task_specify_agent_kwargs or {}),
             )
-            self.specified_task_prompt = task_specify_agent.step(
+            self.specified_task_prompt = task_specify_agent.specify_prompt(
                 task_prompt,
                 meta_dict=task_specify_meta_dict,
             )
@@ -127,7 +127,8 @@ def __init__(
                 output_language=output_language,
                 **(task_planner_agent_kwargs or {}),
             )
-            self.planned_task_prompt = task_planner_agent.step(task_prompt)
+            self.planned_task_prompt = task_planner_agent.create_plan(
+                task_prompt)
             task_prompt = f"{task_prompt}\n{self.planned_task_prompt}"
         else:
             self.planned_task_prompt = None

test/agents/test_task_agent.py

@@ -31,7 +31,7 @@ def test_task_specify_ai_society_agent(model: Optional[ModelType]):
     print(f"Original task prompt:\n{original_task_prompt}\n")
     task_specify_agent = TaskSpecifyAgent(
         model_config=ChatGPTConfig(temperature=1.0), model=model)
-    specified_task_prompt = task_specify_agent.step(
+    specified_task_prompt = task_specify_agent.specify_prompt(
         original_task_prompt, meta_dict=dict(assistant_role="Musician",
                                              user_role="Student"))
     assert ("{" and "}" not in task_specify_agent.task_specify_prompt)
@@ -47,7 +47,7 @@ def test_task_specify_code_agent(model: Optional[ModelType]):
         model_config=ChatGPTConfig(temperature=1.0),
         model=model,
     )
-    specified_task_prompt = task_specify_agent.step(
+    specified_task_prompt = task_specify_agent.specify_prompt(
         original_task_prompt, meta_dict=dict(domain="Chemistry",
                                              language="Python"))
     assert ("{" and "}" not in task_specify_agent.task_specify_prompt)
@@ -63,11 +63,11 @@ def test_task_planner_agent(model: Optional[ModelType]):
         model_config=ChatGPTConfig(temperature=1.0),
         model=model,
     )
-    specified_task_prompt = task_specify_agent.step(
+    specified_task_prompt = task_specify_agent.specify_prompt(
         original_task_prompt, meta_dict=dict(domain="Chemistry",
                                              language="Python"))
     print(f"Specified task prompt:\n{specified_task_prompt}\n")
     task_planner_agent = TaskPlannerAgent(
         model_config=ChatGPTConfig(temperature=1.0), model=model)
-    planned_task_prompt = task_planner_agent.step(specified_task_prompt)
+    planned_task_prompt = task_planner_agent.create_plan(specified_task_prompt)
     print(f"Planned task prompt:\n{planned_task_prompt}\n")

test/messages/test_chat_message.py

@@ -11,6 +11,8 @@
 # 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
+
 import pytest
 
 from camel.messages import BaseMessage
@@ -59,12 +61,12 @@ def test_chat_message(chat_message: BaseMessage) -> None:
     assert chat_message.content == content
 
     dictionary = chat_message.to_dict()
-    assert dictionary == {
+    reference_dict: Dict[str, Any] = {
         "role_name": role_name,
         "role_type": role_type.name,
-        **(meta_dict or {}),
         "content": content,
     }
+    assert dictionary == reference_dict
 
 
 def test_assistant_chat_message(assistant_chat_message: BaseMessage) -> None:
@@ -79,12 +81,12 @@ def test_assistant_chat_message(assistant_chat_message: BaseMessage) -> None:
     assert assistant_chat_message.content == content
 
     dictionary = assistant_chat_message.to_dict()
-    assert dictionary == {
+    reference_dict: Dict[str, Any] = {
         "role_name": role_name,
         "role_type": role_type.name,
-        **(meta_dict or {}),
         "content": content,
     }
+    assert dictionary == reference_dict
 
 
 def test_user_chat_message(user_chat_message: BaseMessage) -> None:
@@ -99,9 +101,9 @@ def test_user_chat_message(user_chat_message: BaseMessage) -> None:
     assert user_chat_message.content == content
 
     dictionary = user_chat_message.to_dict()
-    assert dictionary == {
+    reference_dict: Dict[str, Any] = {
         "role_name": role_name,
         "role_type": role_type.name,
-        **(meta_dict or {}),
         "content": content,
     }
+    assert dictionary == reference_dict