aient 1.0.82__py3-none-any.whl → 1.0.84__py3-none-any.whl

aient/core/utils.py

@@ -21,7 +21,7 @@ def get_model_dict(provider):
         if type(model) == str:
             model_dict[model] = model
         if isinstance(model, dict):
-            model_dict.update({new: old for old, new in model.items()})
+            model_dict.update({str(new): old for old, new in model.items()})
     return model_dict
 
 class BaseAPI:

aient/models/chatgpt.py

@@ -540,10 +540,10 @@ class chatgpt(BaseLLM):
                 if tool_name == "read_file" and "<read_file error>" not in tool_response:
                     self.latest_file_content[tool_info['parameter']["file_path"]] = tool_response
                     all_responses.append(f"[{tool_name}({tool_args}) Result]:\n\nRead file successfully! The file content has been updated in the tag <latest_file_content>.")
-                elif tool_name == "write_to_file":
-                    tool_info['parameter']["content"] = "...文件已写入，内容已省略以节省上下文..."
-                    tool_args = json.dumps(tool_info['parameter'], ensure_ascii=False) if not isinstance(tool_info['parameter'], str) else tool_info['parameter']
-                    all_responses.append(f"[{tool_name}({tool_args}) Result]:\n\n{tool_response}")
+                # elif tool_name == "write_to_file":
+                #     tool_info['parameter']["content"] = "...文件已写入，内容已省略以节省上下文..."
+                #     tool_args = json.dumps(tool_info['parameter'], ensure_ascii=False) if not isinstance(tool_info['parameter'], str) else tool_info['parameter']
+                #     all_responses.append(f"[{tool_name}({tool_args}) Result]:\n\n{tool_response}")
                 else:
                     all_responses.append(f"[{tool_name}({tool_args}) Result]:\n\n{tool_response}")
 

aient/plugins/excute_command.py

@@ -1,21 +1,26 @@
 import subprocess
 from .registry import register_tool
 
+import html
+
+def unescape_html(input_string: str) -> str:
+  """
+  将字符串中的 HTML 实体（例如 &）转换回其原始字符（例如 &）。
+
+  Args:
+    input_string: 包含 HTML 实体的输入字符串。
+
+  Returns:
+    转换后的字符串。
+  """
+  return html.unescape(input_string)
+
 # 执行命令
 @register_tool()
 def excute_command(command):
     """
 执行命令并返回输出结果
 禁止用于查看pdf，禁止使用 pdftotext 命令
-请确保生成的命令字符串可以直接在终端执行，特殊字符（例如 &&）必须保持原样，不要进行 HTML 编码或任何形式的转义，禁止使用 &&
-
-for example:
-
-correct:
-ls -l && echo 'Hello, World!'
-
-incorrect:
-ls -l && echo 'Hello, World!'
 
 参数:
     command: 要执行的命令，可以克隆仓库，安装依赖，运行代码等
@@ -25,6 +30,7 @@ ls -l && echo 'Hello, World!'
     """
     try:
         # 使用subprocess.run捕获命令输出
+        command = unescape_html(command)
         result = subprocess.run(command, shell=True, check=True, capture_output=True, text=True)
         # 返回命令的标准输出
         if "pip install" in command:
@@ -36,4 +42,8 @@ ls -l && echo 'Hello, World!'
         # 如果命令执行失败，返回错误信息和错误输出
         return f"执行命令失败 (退出码 {e.returncode}):\n错误: {e.stderr}\n输出: {e.stdout}"
     except Exception as e:
-        return f"执行命令时发生异常: {e}"
+        return f"执行命令时发生异常: {e}"
+
+if __name__ == "__main__":
+    print(excute_command("ls -l && echo 'Hello, World!'"))
+    print(excute_command("ls -l && echo 'Hello, World!'"))

{aient-1.0.82.dist-info → aient-1.0.84.dist-info}/METADATA

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

{aient-1.0.82.dist-info → aient-1.0.84.dist-info}/RECORD

@@ -6,14 +6,14 @@ aient/core/log_config.py,sha256=kz2_yJv1p-o3lUQOwA3qh-LSc3wMHv13iCQclw44W9c,274
 aient/core/models.py,sha256=_1wYZg_n9kb2A3C8xCboyqleH2iHc9scwOvtx9DPeok,7582
 aient/core/request.py,sha256=DzjOtWb7rm1frsyKRIsnhONCuttURfwtqERc_qJqHgg,60931
 aient/core/response.py,sha256=CSAOQmybBu3i2Yq2YUyi91dZWTN1tmtJEBTI8RFwj_s,30594
-aient/core/utils.py,sha256=Pmr_ghNsmhKTG65g8_M8BBc_pHGgnD88-SNsJNPJFw4,26158
+aient/core/utils.py,sha256=W-PDhwoJIbmlt4xfyrV9GXHu9TwRk4pivBOcDVXjgsc,26163
 aient/core/test/test_base_api.py,sha256=pWnycRJbuPSXKKU9AQjWrMAX1wiLC_014Qc9hh5C2Pw,524
 aient/core/test/test_image.py,sha256=_T4peNGdXKBHHxyQNx12u-NTyFE8TlYI6NvvagsG2LE,319
 aient/core/test/test_payload.py,sha256=8jBiJY1uidm1jzL-EiK0s6UGmW9XkdsuuKFGrwFhFkw,2755
 aient/models/__init__.py,sha256=ouNDNvoBBpIFrLsk09Q_sq23HR0GbLAKfGLIFmfEuXE,219
 aient/models/audio.py,sha256=kRd-8-WXzv4vwvsTGwnstK-WR8--vr9CdfCZzu8y9LA,1934
 aient/models/base.py,sha256=z-Z0pJfTN2x0cuwfvu0BdMRY9O-RmLwHEnBIJN1x4Fg,6719
-aient/models/chatgpt.py,sha256=ysw9jEXnGRAxqyvSjf4y7EzLGeyCZqElkYS3sW6G6KI,44775
+aient/models/chatgpt.py,sha256=0LbiDbBnAUQNnvWBPhQuBWNhKEPXWE171DDwEGhLdIY,44783
 aient/models/claude.py,sha256=thK9P8qkaaoUN3OOJ9Shw4KDs-pAGKPoX4FOPGFXva8,28597
 aient/models/duckduckgo.py,sha256=1l7vYCs9SG5SWPCbcl7q6pCcB5AUF_r-a4l9frz3Ogo,8115
 aient/models/gemini.py,sha256=chGLc-8G_DAOxr10HPoOhvVFW1RvMgHd6mt--VyAW98,14730
@@ -22,7 +22,7 @@ aient/models/vertex.py,sha256=qVD5l1Q538xXUPulxG4nmDjXE1VoV4yuAkTCpIeJVw0,16795
 aient/plugins/__init__.py,sha256=p3KO6Aa3Lupos4i2SjzLQw1hzQTigOAfEHngsldrsyk,986
 aient/plugins/arXiv.py,sha256=yHjb6PS3GUWazpOYRMKMzghKJlxnZ5TX8z9F6UtUVow,1461
 aient/plugins/config.py,sha256=KnZ5xtb5o41FI2_qvxTEQhssdd3WJc7lIAFNR85INQw,7817
-aient/plugins/excute_command.py,sha256=ixSN1q2d3JoJK8OipRB9M2XGMnf4R1b1o_E7F5cUSOU,1392
+aient/plugins/excute_command.py,sha256=A5jvmo8DGcluylIooczMHk8t8zs7vq-wQzbF1ETYY2A,1606
 aient/plugins/get_time.py,sha256=Ih5XIW5SDAIhrZ9W4Qe5Hs1k4ieKPUc_LAd6ySNyqZk,654
 aient/plugins/image.py,sha256=ZElCIaZznE06TN9xW3DrSukS7U3A5_cjk1Jge4NzPxw,2072
 aient/plugins/list_directory.py,sha256=5ubm-mfrj-tanGSDp4M_Tmb6vQb3dx2-XVfQ2yL2G8A,1394
@@ -36,8 +36,8 @@ aient/prompt/agent.py,sha256=WLs0KiNZs29FJ5w7N3kQZYLVEeS7K8vxIOUw06LcXVE,23825
 aient/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aient/utils/prompt.py,sha256=UcSzKkFE4-h_1b6NofI6xgk3GoleqALRKY8VBaXLjmI,11311
 aient/utils/scripts.py,sha256=PPwaJEigPkpciJHUXOag483iq1GjvaLReHDHjkinv6c,26780
-aient-1.0.82.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
-aient-1.0.82.dist-info/METADATA,sha256=3SFzfABMavQXgBZAn_f9KboguAB1FulPO27UwaQdo3w,5000
-aient-1.0.82.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
-aient-1.0.82.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
-aient-1.0.82.dist-info/RECORD,,
+aient-1.0.84.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
+aient-1.0.84.dist-info/METADATA,sha256=Kr00tLyGLjyc11A17QHNl3jDCByHvEnDtisXyGKt_H0,5000
+aient-1.0.84.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
+aient-1.0.84.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
+aient-1.0.84.dist-info/RECORD,,

{aient-1.0.82.dist-info → aient-1.0.84.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (79.0.1)
+Generator: setuptools (80.0.0)
 Root-Is-Purelib: true
 Tag: py3-none-any