aient 1.1.10__py3-none-any.whl → 1.1.12__py3-none-any.whl

aient/core/request.py

@@ -306,7 +306,7 @@ async def get_vertex_gemini_payload(request, engine, provider, api_key=None):
     search_tool = None
 
     # https://cloud.google.com/vertex-ai/generative-ai/docs/models/gemini/2-0-flash?hl=zh-cn
-    pro_models = ["gemini-2.5"]
+    pro_models = ["gemini-2.5", "gemini-2.0"]
     if any(pro_model in original_model for pro_model in pro_models):
         location = gemini2
         search_tool = {"googleSearch": {}}

aient/models/chatgpt.py

@@ -558,12 +558,12 @@ class chatgpt(BaseLLM):
                             tool_response = chunk.replace("function_response:", "")
                         else:
                             yield chunk
-                if tool_name == "read_file" and "<read_file error>" not in tool_response:
+                if tool_name == "read_file" and "<tool_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":
+                elif tool_name == "write_to_file" and "<tool_error>" not in tool_response:
                     all_responses.append(f"[{tool_name} Result]:\n\n{tool_response}")
-                elif tool_name == "read_image":
+                elif tool_name == "read_image" and "<tool_error>" not in tool_response:
                     tool_info["base64_image"] = tool_response
                     all_responses.append(f"[{tool_name}({tool_args}) Result]:\n\nRead image successfully!")
                 else:

aient/plugins/read_file.py

@@ -51,11 +51,11 @@ Examples:
     try:
         # 检查文件是否存在
         if not os.path.exists(file_path):
-            return f"<read_file error>文件 '{file_path}' 不存在</read_file error>"
+            return f"<tool_error>文件 '{file_path}' 不存在</tool_error>"
 
         # 检查是否为文件
         if not os.path.isfile(file_path):
-            return f"<read_file error>'{file_path}' 不是一个文件</read_file error>"
+            return f"<tool_error>'{file_path}' 不是一个文件</tool_error>"
 
         # 检查文件扩展名
         if file_path.lower().endswith('.pdf'):
@@ -64,7 +64,7 @@ Examples:
 
             # 如果提取结果为空
             if not text_content:
-                return f"<read_file error>无法从 '{file_path}' 提取文本内容</read_file error>"
+                return f"<tool_error>无法从 '{file_path}' 提取文本内容</tool_error>"
         elif file_path.lower().endswith('.ipynb'):
             try:
                 with open(file_path, 'r', encoding='utf-8') as file:
@@ -96,9 +96,9 @@ Examples:
 
                 text_content = json.dumps(notebook_content, indent=2, ensure_ascii=False)
             except json.JSONDecodeError:
-                return f"<read_file error>文件 '{file_path}' 不是有效的JSON格式 (IPython Notebook)。</read_file error>"
+                return f"<tool_error>文件 '{file_path}' 不是有效的JSON格式 (IPython Notebook)。</tool_error>"
             except Exception as e:
-                return f"<read_file error>处理IPython Notebook文件 '{file_path}' 时发生错误: {e}</read_file error>"
+                return f"<tool_error>处理IPython Notebook文件 '{file_path}' 时发生错误: {e}</tool_error>"
         else:
             # 更新：修改通用文件读取逻辑以支持多种编码
             # 这部分替换了原有的 else 块内容
@@ -152,25 +152,25 @@ Examples:
                         elif primary_encoding_to_try: # 如果有检测结果但置信度低
                             detected_str_part = f"低置信度检测编码 '{primary_encoding_to_try}' (置信度 {confidence:.2f}), "
 
-                        return f"<read_file error>文件 '{file_path}' 无法解码。已尝试: {detected_str_part}UTF-8, UTF-16。</read_file error>"
+                        return f"<tool_error>文件 '{file_path}' 无法解码。已尝试: {detected_str_part}UTF-8, UTF-16。</tool_error>"
 
             except FileNotFoundError:
                 # 此处不太可能触发 FileNotFoundError，因为函数开头已有 os.path.exists 检查
-                return f"<read_file error>文件 '{file_path}' 在读取过程中未找到。</read_file error>"
+                return f"<tool_error>文件 '{file_path}' 在读取过程中未找到。</tool_error>"
             except Exception as e:
                 # 捕获在此块中可能发生的其他错误，例如未被早期检查捕获的文件读取问题
-                return f"<read_file error>处理通用文件 '{file_path}' 时发生错误: {e}</read_file error>"
+                return f"<tool_error>处理通用文件 '{file_path}' 时发生错误: {e}</tool_error>"
 
         # 返回文件内容
         return text_content
 
     except PermissionError:
-        return f"<read_file error>没有权限访问文件 '{file_path}'</read_file error>"
+        return f"<tool_error>没有权限访问文件 '{file_path}'</tool_error>"
     except UnicodeDecodeError:
         # 更新：修改全局 UnicodeDecodeError 错误信息使其更通用
-        return f"<read_file error>文件 '{file_path}' 包含无法解码的字符 (UnicodeDecodeError)。</read_file error>"
+        return f"<tool_error>文件 '{file_path}' 包含无法解码的字符 (UnicodeDecodeError)。</tool_error>"
     except Exception as e:
-        return f"<read_file error>读取文件时发生错误: {e}</read_file error>"
+        return f"<tool_error>读取文件时发生错误: {e}</tool_error>"
 
 if __name__ == "__main__":
     # python -m beswarm.aient.src.aient.plugins.read_file

aient/plugins/read_image.py

@@ -19,17 +19,17 @@ def read_image(image_path: str):
     try:
         # 检查路径是否存在
         if not os.path.exists(image_path):
-            return f"错误: 图片路径 '{image_path}' 不存在。"
+            return f"<tool_error>图片路径 '{image_path}' 不存在。</tool_error>"
         # 检查是否为文件
         if not os.path.isfile(image_path):
-            return f"错误: 路径 '{image_path}' 不是一个有效的文件 (可能是一个目录)。"
+            return f"<tool_error>路径 '{image_path}' 不是一个有效的文件 (可能是一个目录)。</tool_error>"
 
         # 尝试猜测MIME类型
         mime_type, _ = mimetypes.guess_type(image_path) # encoding 变量通常不需要
 
         if not mime_type or not mime_type.startswith('image/'):
             # 如果mimetypes无法识别，或者不是图片类型
-            return f"错误: 文件 '{image_path}' 的MIME类型无法识别为图片 (检测到: {mime_type})。请确保文件是常见的图片格式 (e.g., PNG, JPG, GIF, WEBP)。"
+            return f"<tool_error>文件 '{image_path}' 的MIME类型无法识别为图片 (检测到: {mime_type})。请确保文件是常见的图片格式 (e.g., PNG, JPG, GIF, WEBP)。</tool_error>"
 
         with open(image_path, "rb") as image_file:
             image_data = image_file.read()
@@ -45,10 +45,10 @@ def read_image(image_path: str):
     except FileNotFoundError:
         # 这个异常通常由 open() 抛出，如果 os.path.exists 通过但文件在读取前被删除
         # 或者路径检查逻辑未能完全覆盖所有情况 (理论上不应发生)
-        return f"错误: 图片路径 '{image_path}' 未找到 (可能在检查后被删除或移动)。"
+        return f"<tool_error>图片路径 '{image_path}' 未找到 (可能在检查后被删除或移动)。</tool_error>"
     except PermissionError:
-        return f"错误: 没有权限访问图片路径 '{image_path}'。"
+        return f"<tool_error>没有权限访问图片路径 '{image_path}'。</tool_error>"
     except IOError as e: # 例如文件损坏无法读取，或磁盘问题
-        return f"错误: 读取图片 '{image_path}' 时发生 I/O 错误: {e}"
+        return f"<tool_error>读取图片 '{image_path}' 时发生 I/O 错误: {e}</tool_error>"
     except Exception as e:
-        return f"读取图片 '{image_path}' 时发生未知错误: {e}"
+        return f"<tool_error>读取图片 '{image_path}' 时发生未知错误: {e}</tool_error>"

aient/utils/scripts.py

@@ -440,7 +440,18 @@ def parse_function_xml(xml_content: str, check_line_start: bool = True) -> List[
         # 如果 '<' 不在行首 (即 tag_start > 0 且其前一个字符不是换行符)，
         # 则将其视为普通文本的一部分，移动 position 并继续搜索
         if check_line_start:
-            if tag_start > 0 and xml_content[tag_start - 1] != '\n':
+            # 检查标签是否在行首，或者行首到标签之间只有空格
+            is_start_of_line_or_only_spaces_before = True
+            if tag_start > 0:
+                # 从 tag_start - 1 向前检查，直到行首或遇到非空格字符
+                check_pos = tag_start - 1
+                while check_pos >= 0 and xml_content[check_pos] != '\n':
+                    if not xml_content[check_pos].isspace():
+                        is_start_of_line_or_only_spaces_before = False
+                        break
+                    check_pos -= 1
+
+            if not is_start_of_line_or_only_spaces_before:
                 position = tag_start + 1  # 从 '<' 之后继续搜索
                 continue
 
@@ -666,15 +677,15 @@ if __name__ == "__main__":
 </read_file>
 </tool_call>好的，我现在读取 `README.md` 文件。
 """
-    test_xml = """首先使用read_file工具读取论文内容，然后使用excute_command工具克隆代码仓库到本地。\n```xml\n<read_file>\n<file_path>/Users/yanyuming/Downloads/GitHub/OceanSynthesis/papers/2412.06410v1.pdf</file_path>\n</read_file>\n\n<excute_command>\n<command>git clone https://github.com/bartbussmann/BatchTopK.git</command>\n</excute_command>\n```"""
+    # test_xml = """首先使用read_file工具读取论文内容，然后使用excute_command工具克隆代码仓库到本地。\n```xml\n<read_file>\n<file_path>/Users/yanyuming/Downloads/GitHub/OceanSynthesis/papers/2412.06410v1.pdf</file_path>\n</read_file>\n\n<excute_command>\n<command>git clone https://github.com/bartbussmann/BatchTopK.git</command>\n</excute_command>\n```"""
     test_xml = """
 ✅ 好的，我现在读取 `README.md` 文件。
-<read_file>
-<file_path>README.md</file_path>
-</read_file>
-<read_file>
-<file_path>README.md</file_path>
-</read_file>
+    <read_file>
+    <file_path>README.md</file_path>
+    </read_file>
+    <read_file>
+    <file_path>README.md</file_path>
+    </read_file>
 
 <tool_call>
 <read_file>
@@ -682,6 +693,53 @@ if __name__ == "__main__":
 </read_file>
 </tool_call>
 好的，我现在读取 `README.md` 文件。`<answer> </answer>`
+"""
+    test_xml = """
+好的，我来报告上一组指令的执行结果：
+
+1.  **创建日志目录 `/root/ctm-server/ctm_tinyimagenet_logs_v5_drastic_reduction`**:
+    <excute_command>
+    <command>mkdir -p /root/ctm-server/ctm_tinyimagenet_logs_v5_drastic_reduction</command>
+    </excute_command>
+    *(我需要等待这个命令的实际执行结果才能确认。假设它会成功，因为 `mkdir -p` 通常不会失败。)*
+
+2.  **执行第五次实验的训练命令**:
+    <excute_command>
+    <command>python /root/ctm-server/continuous-thought-machines/tasks/image_classification/train.py \
+        --model ctm \
+        --dataset tinyimagenet \
+        --data_root /root/ctm-server/tiny-imagenet-200 \
+        --batch_size 1024 \
+        --training_iterations 10000 \
+        --device 0 \
+        --log_dir /root/ctm-server/ctm_tinyimagenet_logs_v5_drastic_reduction \
+        --save_every 1000 \
+        --track_every 200 \
+        --num_workers_train 2 \
+        --d_model 128 \
+        --iterations 10 \
+        --d_input 64 \
+        --heads 1 \
+        --synapse_depth 1 \
+        --n_synch_out 128 \
+        --n_synch_action 128 \
+        --memory_length 8 \
+        --backbone_type resnet18-1 \
+        --positional_embedding_type none \
+        --dropout 0.5 \
+        --weight_decay 1e-3 \
+        --use_amp</command>
+    </excute_command>
+    *(我需要等待这个训练命令的实际执行结果来确认训练是否完成。训练通常需要一些时间。)*
+
+在上述两个 `excute_command` 的结果返回后，并且如果训练确认已完成，我将执行下一步：
+
+3.  **读取第五次实验的准确率图像**:
+    <read_image>
+    <image_path>/root/ctm-server/ctm_tinyimagenet_logs_v5_drastic_reduction/accuracies.png</image_path>
+    </read_image>
+
+请提供前两个 `excute_command` 的执行结果。
 """
 
     print(parse_function_xml(test_xml))

{aient-1.1.10.dist-info → aient-1.1.12.dist-info}/METADATA

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

{aient-1.1.10.dist-info → aient-1.1.12.dist-info}/RECORD

@@ -4,7 +4,7 @@ aient/core/.gitignore,sha256=5JRRlYYsqt_yt6iFvvzhbqh2FTUQMqwo6WwIuFzlGR8,13
 aient/core/__init__.py,sha256=NxjebTlku35S4Dzr16rdSqSTWUvvwEeACe8KvHJnjPg,34
 aient/core/log_config.py,sha256=kz2_yJv1p-o3lUQOwA3qh-LSc3wMHv13iCQclw44W9c,274
 aient/core/models.py,sha256=kF-HLi1I2k_G5r153ZHuiGH8_NmpTlFMfK0_myB28YQ,7366
-aient/core/request.py,sha256=Sa2muc4buCkSNPQITOvXwQt0CGA_A39TpIoeVqdqwFQ,61929
+aient/core/request.py,sha256=VItemXnWzqzS10W-RuLVrARki1w7MZMBZdyqyA5axw8,61943
 aient/core/response.py,sha256=BNHLazjfQT8mVg7LnPLzlX429aQM3S03pumPbOpczCI,31518
 aient/core/utils.py,sha256=n3dyaApN4rrSduI8cjZbeD0mv8_O5LPTTbwRkj1_v4w,26540
 aient/core/test/test_base_api.py,sha256=pWnycRJbuPSXKKU9AQjWrMAX1wiLC_014Qc9hh5C2Pw,524
@@ -14,7 +14,7 @@ aient/core/test/test_payload.py,sha256=8jBiJY1uidm1jzL-EiK0s6UGmW9XkdsuuKFGrwFhF
 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=6SuMfV8n0pBOaKN3WGqhPc53_LHBdyOECQgVsHscso8,46169
+aient/models/chatgpt.py,sha256=sfXqmYAiTTAa_dH2h49BZnVvjebEl99LJdQGepD2sCQ,46244
 aient/models/claude.py,sha256=JezghW7y0brl4Y5qiSHvnYR5prQCFywX4RViHt39pGI,26037
 aient/models/duckduckgo.py,sha256=1l7vYCs9SG5SWPCbcl7q6pCcB5AUF_r-a4l9frz3Ogo,8115
 aient/models/gemini.py,sha256=chGLc-8G_DAOxr10HPoOhvVFW1RvMgHd6mt--VyAW98,14730
@@ -27,8 +27,8 @@ aient/plugins/excute_command.py,sha256=A3WmfZboEikU1EHvtMWhBv-xHxCyMxbDddQ982I_8
 aient/plugins/get_time.py,sha256=Ih5XIW5SDAIhrZ9W4Qe5Hs1k4ieKPUc_LAd6ySNyqZk,654
 aient/plugins/image.py,sha256=ZElCIaZznE06TN9xW3DrSukS7U3A5_cjk1Jge4NzPxw,2072
 aient/plugins/list_directory.py,sha256=JZVuImecMSfEv6jLqii-0uQJ1UCsrpMNmYlwW3PEDg4,1374
-aient/plugins/read_file.py,sha256=-RRmaj-rSl8y--5VKnxCsZ1YQHe75OhnqvsDRLJyujM,8412
-aient/plugins/read_image.py,sha256=goBnpmnmu753pQBkEROTo1ZaGE23fx5WJVr8T8z4598,2577
+aient/plugins/read_file.py,sha256=Lv03AW-gWGzM2esos2vLTXHcceczdTqEO7_vqFT4yoY,8302
+aient/plugins/read_image.py,sha256=jFu8MGCPWg19YOG-STWH3pqacf94F17i_4JaJu1XWLs,2704
 aient/plugins/registry.py,sha256=YknzhieU_8nQ3oKlUSSWDB4X7t2Jx0JnqT2Jd9Xsvfk,3574
 aient/plugins/run_python.py,sha256=dgcUwBunMuDkaSKR5bToudVzSdrXVewktDDFUz_iIOQ,4589
 aient/plugins/websearch.py,sha256=llxy1U0vJiNMiKvamMr4p7IruLb3nnDR4YErz8TYimc,15215
@@ -37,9 +37,9 @@ aient/prompt/__init__.py,sha256=GBtn6-JDT8KHFCcuPpfSNE_aGddg5p4FEyMCy4BfwGs,20
 aient/prompt/agent.py,sha256=y2GETN6ScC5yQVs75VFfzm4YUWzblbqLYz0Sy6JnPRw,24950
 aient/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aient/utils/prompt.py,sha256=UcSzKkFE4-h_1b6NofI6xgk3GoleqALRKY8VBaXLjmI,11311
-aient/utils/scripts.py,sha256=NXmTxcZqHoRv3S13isLsv7kvqktXnA5ej7uMsxCJUe0,26656
-aient-1.1.10.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
-aient-1.1.10.dist-info/METADATA,sha256=59vJo7hMpLQ6Y9ZHXoLkZwiHuVpk1B5qLSepbTxXrXc,4968
-aient-1.1.10.dist-info/WHEEL,sha256=Nw36Djuh_5VDukK0H78QzOX-_FQEo6V37m3nkm96gtU,91
-aient-1.1.10.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
-aient-1.1.10.dist-info/RECORD,,
+aient/utils/scripts.py,sha256=wutPtgbs-WXo5AACLpnCJaRQBOSKXWNnsf2grbYDzyQ,29098
+aient-1.1.12.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
+aient-1.1.12.dist-info/METADATA,sha256=Qw9hW21BsAFZHELCET7zEMTFUvvCaJv0r2WBgJmZM-I,4968
+aient-1.1.12.dist-info/WHEEL,sha256=zaaOINJESkSfm_4HQVc5ssNzHCPXhJm0kEUakpsEHaU,91
+aient-1.1.12.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
+aient-1.1.12.dist-info/RECORD,,

{aient-1.1.10.dist-info → aient-1.1.12.dist-info}/WHEEL

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