aient 1.1.31__tar.gz → 1.1.33__tar.gz

{aient-1.1.31/src/aient.egg-info → aient-1.1.33}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aient
-Version: 1.1.31
+Version: 1.1.33
 Summary: Aient: The Awakening of Agent.
 Description-Content-Type: text/markdown
 License-File: LICENSE

{aient-1.1.31 → aient-1.1.33}/setup.py

@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="aient",
-    version="1.1.31",
+    version="1.1.33",
     description="Aient: The Awakening of Agent.",
     long_description=Path.open(Path("README.md"), encoding="utf-8").read(),
     long_description_content_type="text/markdown",

{aient-1.1.31 → aient-1.1.33}/src/aient/core/models.py

@@ -1,3 +1,4 @@
+import json
 from io import IOBase
 from pydantic import BaseModel, Field, model_validator, ConfigDict
 from typing import List, Dict, Optional, Union, Tuple, Literal, Any

{aient-1.1.31 → aient-1.1.33}/src/aient/models/chatgpt.py

@@ -12,7 +12,7 @@ from pathlib import Path
 
 from .base import BaseLLM
 from ..plugins import PLUGINS, get_tools_result_async, function_call_list, update_tools_config
-from ..utils.scripts import safe_get, async_generator_to_sync, parse_function_xml, parse_continuous_json, convert_functions_to_xml
+from ..utils.scripts import safe_get, async_generator_to_sync, parse_function_xml, parse_continuous_json, convert_functions_to_xml, remove_xml_tags_and_content
 from ..core.request import prepare_request_payload
 from ..core.response import fetch_response_stream
 
@@ -465,7 +465,7 @@ class chatgpt(BaseLLM):
                     if isinstance(self.conversation[convo_id][-1]["content"], str) and \
                     "<tool_error>" in self.conversation[convo_id][-1]["content"]:
                         need_function_call = False
-                        full_response = "接下来我需要做什么？"
+                        full_response = remove_xml_tags_and_content(full_response) + "上面是我的分析，还没有实际行动。\n\n接下来我需要做什么？"
                 else:
                     need_function_call = False
                     if self.print_log:

{aient-1.1.31 → aient-1.1.33}/src/aient/plugins/websearch.py

@@ -341,6 +341,8 @@ async def get_search_results(query):
 
 if __name__ == "__main__":
     os.system("clear")
+    # python -m beswarm.aient.src.aient.plugins.websearch
+    print(get_url_content(""))
     # from aient.models import chatgpt
     # print(get_search_results("今天的微博热搜有哪些？", chatgpt.chatgpt_api_url.v1_url))
 

{aient-1.1.31 → aient-1.1.33}/src/aient/utils/scripts.py

@@ -1,4 +1,5 @@
 import os
+import re
 import json
 import requests
 import urllib.parse
@@ -148,6 +149,35 @@ def safe_get(data, *keys, default=None):
             return default
     return data
 
+def remove_xml_tags_and_content(text: str) -> str:
+    """
+    删除字符串中所有的XML标签及其包裹的内容。
+    这个函数通过迭代地移除最内层的标签来处理嵌套标签和自闭合标签。
+
+    Args:
+        text: 包含XML标签的输入字符串。
+
+    Returns:
+        清理掉XML标签和内容后的字符串。
+    """
+    # 正则表达式匹配成对的标签及其内容（非贪婪模式）以及自闭合标签
+    # 捕获组 \1 用于确保开始和结束标签名匹配
+    pair_pattern = r'<([a-zA-Z0-9_:]+)\b[^>]*>.*?</\1>'
+    self_closing_pattern = r'<[^>/]+/>'
+
+    cleaned_text = text
+    while True:
+        # 使用 re.sub 替换所有非重叠的匹配项
+        new_text = re.sub(pair_pattern, '', cleaned_text, flags=re.DOTALL)
+        new_text = re.sub(self_closing_pattern, '', new_text)
+
+        # 如果没有更多内容被移除，则退出循环
+        if new_text == cleaned_text:
+            break
+        cleaned_text = new_text
+
+    return cleaned_text.strip()
+
 import asyncio
 def async_generator_to_sync(async_gen):
     """
@@ -919,14 +949,15 @@ if __name__ == "__main__":
 
 请提供前两个 `excute_command` 的执行结果。
 """
-    test_xml = """
-好的，我现在执行第一步。
-<tools>
-<list_directory>
-<path>/Downloads/GitHub/beswarm/work/test</path>
-</list_directory>
-</tools>
-"""
-    print(parse_function_xml(test_xml))
+#     test_xml = """
+# 好的，我现在执行第一步。
+# <tools>
+# <list_directory>
+# <path>/Downloads/GitHub/beswarm/work/test</path>
+# </list_directory>
+# </tools>
+# """
+    # print(parse_function_xml(test_xml))
+    print(remove_xml_tags_and_content(test_xml))
 
 # 运行本文件：python -m beswarm.aient.src.aient.utils.scripts

{aient-1.1.31 → aient-1.1.33/src/aient.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aient
-Version: 1.1.31
+Version: 1.1.33
 Summary: Aient: The Awakening of Agent.
 Description-Content-Type: text/markdown
 License-File: LICENSE