aient 1.2.21__py3-none-any.whl → 1.2.23__py3-none-any.whl

aient/models/chatgpt.py

@@ -188,9 +188,9 @@ class chatgpt(BaseLLM):
                             ])
                     self.conversation[convo_id].append(AssistantMessage(convert_functions_to_xml(function_arguments)))
                     if image_message_list:
-                        self.conversation[convo_id].append(UserMessage(message + image_message_list))
+                        self.conversation[convo_id].append(UserMessage(message + image_message_list, Texts("\n\nYour message **must** end with [done] to signify the end of your output.", name="done", visible=self.check_done)))
                     else:
-                        self.conversation[convo_id].append(UserMessage(message))
+                        self.conversation[convo_id].append(UserMessage(message, Texts("\n\nYour message **must** end with [done] to signify the end of your output.", name="done", visible=self.check_done)))
                 else:
                     self.conversation[convo_id].append(AssistantMessage("我已经执行过这个工具了，接下来我需要做什么？"))
         else:
@@ -600,9 +600,6 @@ class chatgpt(BaseLLM):
                 else:
                     all_responses.append(f"[{tool_name}({tool_args}) Result]:\n\n{tool_response}")
 
-            if self.check_done:
-                all_responses.append("Your message **must** end with [done] to signify the end of your output.")
-
             # 合并所有工具响应
             function_response = "\n\n".join(all_responses).strip()
             if missing_required_params:

aient/plugins/read_file.py

@@ -7,7 +7,7 @@ from .registry import register_tool
 
 # 读取文件内容
 @register_tool()
-def read_file(file_path):
+def read_file(file_path, head: int = None):
     """
 Description: Request to read the contents of a file at the specified path. Use this when you need to examine the contents of an existing file you do not know the contents of, for example to analyze code, review text files, or extract information from configuration files. Automatically extracts raw text from PDF and DOCX files. May not be suitable for other types of binary files, as it returns the raw content as a string.
 
@@ -16,6 +16,7 @@ Description: Request to read the contents of a file at the specified path. Use t
 
 参数:
     file_path: 要读取的文件路径，(required) The path of the file to read (relative to the current working directory)
+    head: (可选) 读取文件的前N行，默认为None，读取整个文件
 
 返回:
     文件内容的字符串
@@ -161,13 +162,23 @@ Examples:
                 # 捕获在此块中可能发生的其他错误，例如未被早期检查捕获的文件读取问题
                 return f"<tool_error>处理通用文件 '{file_path}' 时发生错误: {e}</tool_error>"
 
-        if file_path.lower().endswith('.csv'):
-            lines = text_content.splitlines(True)
-            if len(lines) > 500:
-                top_lines = lines[:250]
-                bottom_lines = lines[-250:]
-                omitted_count = len(lines) - 500
-                text_content = "".join(top_lines) + f"\n... (中间省略了 {omitted_count} 行) ...\n" + "".join(bottom_lines)
+        if head is not None:
+            try:
+                num_lines = int(head)
+                if num_lines > 0:
+                    lines = text_content.splitlines(True)
+                    return "".join(lines[:num_lines])
+            except (ValueError, TypeError):
+                # Invalid head value, ignore and proceed with normal logic.
+                pass
+
+        # if file_path.lower().endswith('.csv'):
+        #     lines = text_content.splitlines(True)
+        #     if len(lines) > 500:
+        #         top_lines = lines[:250]
+        #         bottom_lines = lines[-250:]
+        #         omitted_count = len(lines) - 500
+        #         text_content = "".join(top_lines) + f"\n... (中间省略了 {omitted_count} 行) ...\n" + "".join(bottom_lines)
 
         # 返回文件内容
         return text_content
@@ -182,6 +193,6 @@ Examples:
 
 if __name__ == "__main__":
     # python -m beswarm.aient.aient.plugins.read_file
-    result = read_file("./work/cax/Lenia Notebook.ipynb")
+    result = read_file("./work/cax/Lenia Notebook.ipynb", head=10)
     print(result)
     print(len(result))

{aient-1.2.21.dist-info → aient-1.2.23.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aient
-Version: 1.2.21
+Version: 1.2.23
 Summary: Aient: The Awakening of Agent.
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{aient-1.2.21.dist-info → aient-1.2.23.dist-info}/RECORD

@@ -17,7 +17,7 @@ aient/core/test/test_payload.py,sha256=8jBiJY1uidm1jzL-EiK0s6UGmW9XkdsuuKFGrwFhF
 aient/models/__init__.py,sha256=ZTiZgbfBPTjIPSKURE7t6hlFBVLRS9lluGbmqc1WjxQ,43
 aient/models/audio.py,sha256=kRd-8-WXzv4vwvsTGwnstK-WR8--vr9CdfCZzu8y9LA,1934
 aient/models/base.py,sha256=-nnihYnx-vHZMqeVO9ljjt3k4FcD3n-iMk4tT-10nRQ,7232
-aient/models/chatgpt.py,sha256=96E6Ggsy49GS_KWECi5agE0zeaEjaHTRd336fykyIJI,42606
+aient/models/chatgpt.py,sha256=DtcbRJObXyCXlJdel9OItd1wgpOCVvenS2fXs4_xO34,42710
 aient/plugins/__init__.py,sha256=p3KO6Aa3Lupos4i2SjzLQw1hzQTigOAfEHngsldrsyk,986
 aient/plugins/arXiv.py,sha256=yHjb6PS3GUWazpOYRMKMzghKJlxnZ5TX8z9F6UtUVow,1461
 aient/plugins/config.py,sha256=TGgZ5SnNKZ8MmdznrZ-TEq7s2ulhAAwTSKH89bci3dA,7079
@@ -25,7 +25,7 @@ aient/plugins/excute_command.py,sha256=b-rxsyFN6_HnZJAhUi9Qsp8iJ6XTf-zU-CIUIxeQO
 aient/plugins/get_time.py,sha256=Ih5XIW5SDAIhrZ9W4Qe5Hs1k4ieKPUc_LAd6ySNyqZk,654
 aient/plugins/image.py,sha256=ZElCIaZznE06TN9xW3DrSukS7U3A5_cjk1Jge4NzPxw,2072
 aient/plugins/list_directory.py,sha256=V_uKkLx_fQDL5z__bSDC-PqAP-o32KmQW6Pdhx0Fx0s,1433
-aient/plugins/read_file.py,sha256=AOGGzSzkALG1R_ncpDmegX_zKFZBZr8ZhTBQCiwhZfU,8691
+aient/plugins/read_file.py,sha256=qHAhdesOr1VMOCDkeHNvI8UV2ZI98HmJl6GhN4Aq9dU,9183
 aient/plugins/read_image.py,sha256=4FbIiMNVFUQpNyiH5ApGSRvOD9ujcXGyuqlGTJMd7ac,4017
 aient/plugins/readonly.py,sha256=qK5-kBM3NDH1b-otFxFHpAjV5BXEY_e7cTWBcpP7G5k,710
 aient/plugins/registry.py,sha256=YknzhieU_8nQ3oKlUSSWDB4X7t2Jx0JnqT2Jd9Xsvfk,3574
@@ -35,8 +35,8 @@ aient/plugins/write_file.py,sha256=Jt8fOEwqhYiSWpCbwfAr1xoi_BmFnx3076GMhuL06uI,3
 aient/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aient/utils/prompt.py,sha256=UcSzKkFE4-h_1b6NofI6xgk3GoleqALRKY8VBaXLjmI,11311
 aient/utils/scripts.py,sha256=VqtK4RFEx7KxkmcqG3lFDS1DxoNlFFGErEjopVcc8IE,40974
-aient-1.2.21.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
-aient-1.2.21.dist-info/METADATA,sha256=Z08dMsZnseBncF3d5A4Xbtq1hv6sgmw4Vqe93nEAavc,4842
-aient-1.2.21.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aient-1.2.21.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
-aient-1.2.21.dist-info/RECORD,,
+aient-1.2.23.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
+aient-1.2.23.dist-info/METADATA,sha256=70F7ytGF6-OpyIrMWjt-TsupS_ZrAQGZIDixwMk_WI8,4842
+aient-1.2.23.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aient-1.2.23.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
+aient-1.2.23.dist-info/RECORD,,