aient 1.1.26__tar.gz → 1.1.28__tar.gz

{aient-1.1.26/src/aient.egg-info → aient-1.1.28}/PKG-INFO

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

{aient-1.1.26 → aient-1.1.28}/setup.py

@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="aient",
-    version="1.1.26",
+    version="1.1.28",
     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.26 → aient-1.1.28}/src/aient/models/chatgpt.py

@@ -462,6 +462,10 @@ class chatgpt(BaseLLM):
                     print("full_response", full_response)
                 if function_parameter:
                     need_function_call = True
+                    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 = "接下来我需要做什么？"
                 else:
                     need_function_call = False
                     if self.print_log:

{aient-1.1.26 → aient-1.1.28}/src/aient/plugins/excute_command.py

@@ -1,10 +1,9 @@
 import subprocess
 from .registry import register_tool
-from ..utils.scripts import sandbox
+from ..utils.scripts import sandbox, unescape_html
 
 import re
 import os
-import html
 import select
 
 # 检查是否在 Unix-like 系统上 (pty 模块主要用于 Unix)
@@ -39,18 +38,6 @@ def compare_line(line: str) -> bool:
     # print(f"similarity: {similarity}")
     return similarity > 0.89
 
-def unescape_html(input_string: str) -> str:
-  """
-  将字符串中的 HTML 实体（例如 &）转换回其原始字符（例如 &）。
-
-  Args:
-    input_string: 包含 HTML 实体的输入字符串。
-
-  Returns:
-    转换后的字符串。
-  """
-  return html.unescape(input_string)
-
 def get_python_executable(command: str) -> str:
     """
     获取 Python 可执行文件的路径。
@@ -191,18 +178,21 @@ def excute_command(command):
         # print(f"output_lines: {len(new_output_lines)}")
 
         if process.returncode == 0:
-            return f"执行命令成功:\n{final_output_log}"
+            if final_output_log.strip() == "":
+                return f"执行命令成功"
+            else:
+                return f"执行命令成功:\n{final_output_log.strip()}"
         else:
             # 如果是 PTY 模式，stderr 已经包含在 final_output_log 中
             if IS_UNIX:
-                return f"执行命令失败 (退出码 {process.returncode}):\n输出/错误:\n{final_output_log}"
+                return f"<tool_error>执行命令失败 (退出码 {process.returncode}):\n输出/错误:\n{final_output_log}</tool_error>"
             else:
-                return f"执行命令失败 (退出码 {process.returncode}):\n错误: {stderr_output}\n输出: {final_output_log}"
+                return f"<tool_error>执行命令失败 (退出码 {process.returncode}):\n错误: {stderr_output}\n输出: {final_output_log}</tool_error>"
 
     except FileNotFoundError:
-        return f"执行命令失败: 命令或程序未找到 ({command})"
+        return f"<tool_error>执行命令失败: 命令或程序未找到 ({command})</tool_error>"
     except Exception as e:
-        return f"执行命令时发生异常: {e}"
+        return f"<tool_error>执行命令时发生异常: {e}</tool_error>"
 
 if __name__ == "__main__":
     # print(excute_command("ls -l && echo 'Hello, World!'"))

{aient-1.1.26 → aient-1.1.28}/src/aient/plugins/list_directory.py

@@ -43,8 +43,8 @@ def list_directory(path="."):
         return result
 
     except FileNotFoundError:
-        return f"错误: 路径 '{path}' 不存在"
+        return f"<tool_error>路径 '{path}' 不存在</tool_error>"
     except PermissionError:
-        return f"错误: 没有权限访问路径 '{path}'"
+        return f"<tool_error>没有权限访问路径 '{path}'</tool_error>"
     except Exception as e:
-        return f"列出目录时发生错误: {e}"
+        return f"<tool_error>列出目录时发生错误: {e}</tool_error>"

{aient-1.1.26 → aient-1.1.28}/src/aient/plugins/run_python.py

@@ -129,7 +129,7 @@ if _result is not None:
 
     except Exception as e:
         logging.error(f"Error executing code: {str(e)}")
-        return f"Error: {str(e)}"
+        return f"<tool_error>Error: {str(e)}</tool_error>"
 
     finally:
         try:

{aient-1.1.26 → aient-1.1.28}/src/aient/plugins/write_file.py

@@ -1,19 +1,7 @@
 from .registry import register_tool
+from ..utils.scripts import unescape_html
 
 import os
-import html
-
-def unescape_html(input_string: str) -> str:
-  """
-  将字符串中的 HTML 实体（例如 &）转换回其原始字符（例如 &）。
-
-  Args:
-    input_string: 包含 HTML 实体的输入字符串。
-
-  Returns:
-    转换后的字符串。
-  """
-  return html.unescape(input_string)
 
 @register_tool()
 def write_to_file(path, content, mode='w'):
@@ -65,7 +53,7 @@ Example: Requesting to write to frontend-config.json
         with open(path, mode, encoding='utf-8') as file:
             file.write(unescape_html(content))
     except PermissionError as e:
-        return f"写入文件失败: {e}"
+        return f"<tool_error>写入文件失败: {e}</tool_error>"
 
     return f"已成功写入文件：{path}"
 

{aient-1.1.26 → aient-1.1.28}/src/aient/utils/scripts.py

@@ -202,6 +202,19 @@ def parse_tools_from_cursor_prompt(text):
             print(f"解析 JSON 时出错: {e}")
     return []
 
+import html
+def unescape_html(input_string: str) -> str:
+  """
+  将字符串中的 HTML 实体（例如 &）转换回其原始字符（例如 &）。
+
+  Args:
+    input_string: 包含 HTML 实体的输入字符串。
+
+  Returns:
+    转换后的字符串。
+  """
+  return html.unescape(input_string)
+
 import sys
 import shlex
 import builtins

{aient-1.1.26 → aient-1.1.28/src/aient.egg-info}/PKG-INFO

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