[Roadmap] Multi-Agent System Based on the Role-Playing Module

.github/workflows/documentation.yml

@@ -14,7 +14,7 @@ jobs:
  - name: Set up Python environment and install dependencies
  uses: ./.github/actions/camel_install
  with:
- python-version: "3.8"
+ python-version: "3.9"
  - name: Sphinx build
  run: |
  cd docs

.github/workflows/pytest_apps.yml

@@ -22,7 +22,7 @@ jobs:
  - name: Set up Python environment and install dependencies
  uses: ./.github/actions/camel_install
  with:
- python-version: "3.8"
+ python-version: "3.9"
  - name: Run pytest
  env:
  OPENAI_API_KEY: "${{ secrets.OPENAI_API_KEY }}"
@@ -40,7 +40,7 @@ jobs:
  - name: Set up Python environment and install dependencies
  uses: ./.github/actions/camel_install
  with:
- python-version: "3.8"
+ python-version: "3.9"
  - name: Run pytest
  env:
  OPENAI_API_KEY: "${{ secrets.OPENAI_API_KEY }}"

.github/workflows/pytest_package.yml

@@ -20,7 +20,7 @@ jobs:
  - name: Set up Python environment and install dependencies
  uses: ./.github/actions/camel_install
  with:
- python-version: "3.8"
+ python-version: "3.9"
  - name: Run pytest
  env:
  OPENAI_API_KEY: "${{ secrets.OPENAI_API_KEY }}"
@@ -38,7 +38,7 @@ jobs:
  - name: Set up Python environment and install dependencies
  uses: ./.github/actions/camel_install
  with:
- python-version: "3.8"
+ python-version: "3.9"
  - name: Run pytest
  env:
  OPENAI_API_KEY: "${{ secrets.OPENAI_API_KEY }}"
@@ -56,7 +56,7 @@ jobs:
  - name: Set up Python environment and install dependencies
  uses: ./.github/actions/camel_install
  with:
- python-version: "3.8"
+ python-version: "3.9"
  - name: Run pytest
  env:
  OPENAI_API_KEY: "${{ secrets.OPENAI_API_KEY }}"

camel/agents/__init__.py

@@ -15,7 +15,8 @@
 from .chat_agent import ChatAgent
 from .critic_agent import CriticAgent
 from .embodied_agent import EmbodiedAgent
-from .role_assignment_agent import RoleAssignmentAgent
+from .insight_agent import InsightAgent
+from .multi_agent import MultiAgent
 from .task_agent import (
  TaskCreationAgent,
  TaskPlannerAgent,
@@ -28,6 +29,8 @@
 __all__ = [
  'BaseAgent',
  'ChatAgent',
+ 'DeductiveReasonerAgent',
+ 'InsightAgent',
  'TaskSpecifyAgent',
  'TaskPlannerAgent',
  'TaskCreationAgent',
@@ -36,5 +39,5 @@
  'BaseToolAgent',
  'HuggingFaceToolAgent',
  'EmbodiedAgent',
- 'RoleAssignmentAgent',
+ 'MultiAgent',
 ]

camel/agents/deductive_reasoner_agent.py

@@ -11,14 +11,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # =========== Copyright 2023 @ CAMEL-AI.org. All Rights Reserved. ===========
-import re
 from typing import Dict, List, Optional, Union
 
 from camel.agents.chat_agent import ChatAgent
 from camel.configs import BaseConfig
 from camel.messages import BaseMessage
 from camel.prompts import TextPrompt
 from camel.types import ModelType, RoleType
+from camel.utils.structure_output import extract_json_from_string
 
 
 class DeductiveReasonerAgent(ChatAgent):
@@ -144,16 +144,27 @@ def deduce_conditions_and_quality(
 {target_state}
 
 {role_with_description_prompt}
-===== ANSWER TEMPLATE =====
-- Characterization and comparison of $A$ and $B$:\n<BLANK>
-- Historical & Empirical Analysis:\n<BLANK>/None
-- Logical Deduction of Conditions ($C$) (multiple conditions can be deduced):
- condition <NUM>:
- <BLANK>.
-- Entity/Label Recognition of Conditions:\n[<BLANK>, <BLANK>, ...] (include square brackets)
-- Quality Assessment ($Q$) (do not use symbols):
- <BLANK>.
-- Iterative Evaluation:\n<BLANK>/None"""
+"""
+
+ answer_json_template = """
+===== ANSWER TEMPLATE (JSON) =====
+{
+ "Characterization and comparison of A and B": "<BLANK>",
+ "Historical & Empirical Analysis": "<BLANK>", // use "None" if not applicable
+ "Logical Deduction of Conditions (C)": {
+ "condition <NUM>": "<BLANK>",
+ // it is allowed to have multiple conditions
+ "condition <NUM_2>": "<BLANK>",
+ },
+ "Entity or Label Recognition of Conditions": [
+ "<BLANK>",
+ // it is allowed to have multiple labels
+ "<BLANK>",
+ ],
+ "Quality Assessment (Q)": "<BLANK>", // do not use symbols
+ "Iterative Evaluation": "<BLANK>", // use "None" if not applicable
+}
+"""
 
  if role_descriptions_dict is not None:
  role_names = role_descriptions_dict.keys()
@@ -169,11 +180,12 @@ def deduce_conditions_and_quality(
  role_with_description_prompt = ""
  deduce_prompt = TextPrompt(deduce_prompt)
 
- deduce = deduce_prompt.format(
+ deduce: str = deduce_prompt.format(
  starting_state=starting_state,
  target_state=target_state,
  role_with_description_prompt=role_with_description_prompt,
  )
+ deduce += answer_json_template
 
  conditions_and_quality_generation_msg = BaseMessage.make_user_message(
  role_name="Deductive Reasoner", content=deduce
@@ -188,48 +200,31 @@ def deduce_conditions_and_quality(
  "Deduction failed. Error:\n" + f"{response.info}"
  )
  msg: BaseMessage = response.msg
- print(f"Message content:\n{msg.content}")
-
- # Extract the conditions from the message
- condistions_dict = {
- f"condition {i}": cdt.replace("<", "")
- .replace(">", "")
- .strip()
- .strip('\n')
- for i, cdt in re.findall(
- r"condition (\d+):\s*(.+?)(?=condition \d+|- Entity)",
- msg.content,
- re.DOTALL,
- )
- }
-
- # Extract the labels from the message
- labels = [
- label.strip().strip('\n').strip("\"'")
- for label in re.findall(
- r"Entity/Label Recognition of Conditions:\n\[(.+?)\]",
- msg.content,
- re.DOTALL,
- )[0].split(",")
- ]
-
- # Extract the quality from the message
- quality = next(
- q.strip().strip('\n')
- for q in re.findall(
- r"Quality Assessment \(\$Q\$\) \(do not use symbols\):"
- r"\n(.+?)- Iterative",
- msg.content,
- re.DOTALL,
- )
+
+ # Convert them into the dictionary
+ json_dict = extract_json_from_string(msg.content)
+
+ conditions = (
+ json_dict["Logical Deduction of Conditions (C)"]
+ if "Logical Deduction of Conditions (C)" in json_dict
+ else []
+ )
+ labels = (
+ json_dict["Entity or Label Recognition of Conditions"]
+ if "Entity or Label Recognition of Conditions" in json_dict
+ else []
+ )
+ quality = (
+ json_dict["Quality Assessment (Q)"]
+ if "Quality Assessment (Q)" in json_dict
+ else ""
  )
 
- # Convert them into JSON format
- conditions_and_quality_json: Dict[
+ conditions_and_quality_dict: Dict[
  str, Union[List[str], Dict[str, str]]
  ] = {}
- conditions_and_quality_json["conditions"] = condistions_dict
- conditions_and_quality_json["labels"] = labels
- conditions_and_quality_json["evaluate_quality"] = quality
+ conditions_and_quality_dict["conditions"] = conditions
+ conditions_and_quality_dict["labels"] = labels
+ conditions_and_quality_dict["evaluate_quality"] = quality
 
- return conditions_and_quality_json
+ return conditions_and_quality_dict