beswarm 0.1.84__tar.gz → 0.1.85__tar.gz

{beswarm-0.1.84 → beswarm-0.1.85}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beswarm
-Version: 0.1.84
+Version: 0.1.85
 Summary: MAS
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{beswarm-0.1.84 → beswarm-0.1.85}/beswarm/tools/worker.py

@@ -3,12 +3,13 @@ import re
 import copy
 import json
 import platform
+from pathlib import Path
 from datetime import datetime
 
 from ..aient.src.aient.models import chatgpt
 from ..aient.src.aient.plugins import register_tool, get_function_call_list
 from ..prompt import worker_system_prompt, instruction_system_prompt
-from ..utils import extract_xml_content, get_current_screen_image_message
+from ..utils import extract_xml_content, get_current_screen_image_message, replace_xml_content
 
 @register_tool()
 async def worker(goal, tools, work_dir, cache_messages=None):
@@ -48,7 +49,13 @@ async def worker(goal, tools, work_dir, cache_messages=None):
         "function_call_max_loop": 100,
     }
     if cache_messages:
-        work_agent_config["cache_messages"] = cache_messages
+        if isinstance(cache_messages, bool) and cache_messages == True:
+            cache_file_path = Path(work_dir) / ".beswarm" / "work_agent_conversation_history.json"
+            if cache_file_path.exists():
+                with cache_file_path.open("r", encoding="utf-8") as f:
+                    cache_messages = json.load(f)
+        first_user_message = replace_xml_content(cache_messages[1]["content"], "goal", goal)
+        work_agent_config["cache_messages"] = cache_messages[0:1] + [{"role": "user", "content": first_user_message}] + cache_messages[2:]
 
     instruction_agent_config = {
         "api_key": os.getenv("API_KEY"),
@@ -131,7 +138,7 @@ async def worker(goal, tools, work_dir, cache_messages=None):
                 if conversation_history == []:
                     next_instruction = (
                         "任务描述：\n"
-                        f"{goal}\n\n"
+                        f"<goal>{goal}</goal>\n\n"
                         "现在开始执行第一步：\n"
                         f"{next_instruction}"
                     )
@@ -186,7 +193,13 @@ async def worker_gen(goal, tools, work_dir, cache_messages=None):
         "function_call_max_loop": 100,
     }
     if cache_messages:
-        work_agent_config["cache_messages"] = cache_messages
+        if isinstance(cache_messages, bool) and cache_messages == True:
+            cache_file_path = Path(work_dir) / ".beswarm" / "work_agent_conversation_history.json"
+            if cache_file_path.exists():
+                with cache_file_path.open("r", encoding="utf-8") as f:
+                    cache_messages = json.load(f)
+        first_user_message = replace_xml_content(cache_messages[1]["content"], "goal", goal)
+        work_agent_config["cache_messages"] = cache_messages[0:1] + [{"role": "user", "content": first_user_message}] + cache_messages[2:]
 
     instruction_agent_config = {
         "api_key": os.getenv("API_KEY"),
@@ -270,7 +283,7 @@ async def worker_gen(goal, tools, work_dir, cache_messages=None):
                 if conversation_history == []:
                     next_instruction = (
                         "任务描述：\n"
-                        f"{goal}\n\n"
+                        f"<goal>{goal}</goal>\n\n"
                         "现在开始执行第一步：\n"
                         f"{next_instruction}"
                     )

{beswarm-0.1.84 → beswarm-0.1.85}/beswarm/utils.py

@@ -10,6 +10,28 @@ def extract_xml_content(text, xml_tag):
         return ''
     return result
 
+def replace_xml_content(original_string: str, tag_name: str, replacement_content: str) -> str:
+    """
+    将指定XML标签内的内容替换为新内容。
+
+    此函数使用正则表达式查找所有匹配的XML标签对（例如 `<tag>...</tag>`），
+    并将其中的内容替换为 `replacement_content`。
+
+    Args:
+        original_string (str): 包含XML标记的原始字符串。
+        tag_name (str): 要定位的XML标签的名称（不带尖括号）。
+        replacement_content (str): 用于替换标签内部内容的新字符串。
+
+    Returns:
+        str: 返回内容已被替换的新字符串。如果未找到匹配的标签，则返回原始字符串。
+    """
+    pattern = f"<{tag_name}>.*?<\\/{tag_name}>"
+    replacement = f"<{tag_name}>{replacement_content}</{tag_name}>"
+
+    new_string = re.sub(pattern, replacement, original_string, flags=re.DOTALL)
+
+    return new_string
+
 import io
 import base64
 from .aient.src.aient.core.utils import get_image_message, get_text_message

{beswarm-0.1.84 → beswarm-0.1.85}/beswarm.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beswarm
-Version: 0.1.84
+Version: 0.1.85
 Summary: MAS
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{beswarm-0.1.84 → beswarm-0.1.85}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "beswarm"
-version = "0.1.84"
+version = "0.1.85"
 description = "MAS"
 readme = "README.md"
 requires-python = ">=3.11"