init

AlgoriAgent/examples/conversation_self_organizing/tools.py

@@ -0,0 +1,81 @@
+# -*- coding: utf-8 -*-
+"""some utils of angent_builder example"""
+import re
+
+
+def load_txt(instruction_file: str) -> str:
+    """
+    load .txt file
+    Arguments:
+        instruction_file: str type, which is the .txt file pth
+
+    Returns:
+        instruction: str type, which is the str in the instruction_file
+
+    """
+    with open(instruction_file, "r", encoding="utf-8") as f:
+        instruction = f.read()
+    return instruction
+
+
+# extract scenario和participants
+def extract_scenario_and_participants(content: str) -> dict:
+    """
+    extract the scenario and participants from agent builder's response
+
+    Arguments:
+        content: the agent builders response
+
+    Returns:
+        result: dict
+
+    Examples:
+        content: #scenario#: Astronomy club meeting
+    #participants#:
+    * Club Leader: Act as the club leader who is knowledgeable about\
+     astronomy and optics. You are leading a discussion about the \
+     capabilities of telescopes versus the human eye. Please provide \
+     accurate information and guide the discussion.
+    * Curious Member: Act as a curious club member who is interested \
+    in astronomy but may not know all the technical details. You are \
+    eager to learn and ask questions.
+    * Experienced Astronomer: Act as an experienced astronomer who has \
+    practical experience using telescopes for stargazing. You can \
+    provide real-world examples and insights into the topic.
+
+    Return:
+        {'Scenario': 'Astronomy club meeting',
+        'Participants':
+        {'Club_Leader': 'Act as the club leader who is knowledgeable \
+        about astronomy and optics. You are leading a discussion about the  \
+        capabilities of telescopes versus the human eye. Please provide\
+        accurate information and guide the discussion.',
+        'Curious_Member': 'Act as a curious club member who is interested \
+        in astronomy but may not know all the technical details. You are \
+        eager to learn and ask questions.',
+        'Experienced_Astronomer': 'Act as an experienced astronomer who has\
+        practical experience using telescopes for stargazing. You can\
+        provide real-world examples and insights into the topic.'}}
+
+    """
+    result = {}
+    # define regular expression
+    scenario_pattern = r"#scenario#:\s*(.*)"
+    participants_pattern = r"\*\s*([^:\n]+):\s*([^\n]+)"
+
+    # search and extract scenario
+    scenario_match = re.search(scenario_pattern, content)
+    if scenario_match:
+        result["Scenario"] = scenario_match.group(1).strip()
+
+    # search and extract participants
+    participants_matches = re.finditer(participants_pattern, content)
+    participants_dict = {}
+    for match in participants_matches:
+        participant_type, characteristic = match.groups()
+        participants_dict[
+            participant_type.strip().replace(" ", "_")
+        ] = characteristic.strip()
+    result["Participants"] = participants_dict
+
+    return result