aient 1.0.76__tar.gz → 1.0.78__tar.gz

{aient-1.0.76/src/aient.egg-info → aient-1.0.78}/PKG-INFO

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

{aient-1.0.76 → aient-1.0.78}/setup.py

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

@@ -409,7 +409,19 @@ class chatgpt(BaseLLM):
 
         function_parameter = parse_function_xml(full_response)
         if function_parameter:
-            need_function_call = True
+            for tool_dict in function_parameter:
+                if tool_dict.get("function_name", "") not in self.plugins.keys():
+                    function_parameter.remove(tool_dict)
+            if function_parameter:
+                need_function_call = True
+            else:
+                if self.print_log:
+                    print("Failed to parse function_parameter", function_parameter)
+                    print("full_response", full_response)
+                full_response = (
+                    f"{full_response}"
+                    "\n\nFailed to parse function parameter, I need to try again."
+                )
 
         # 处理函数调用
         if need_function_call and self.use_plugins != False:

{aient-1.0.76 → aient-1.0.78}/src/aient/plugins/read_file.py

@@ -1,4 +1,5 @@
 import os
+import json
 from pdfminer.high_level import extract_text
 
 from .registry import register_tool
@@ -63,6 +64,40 @@ Examples:
             # 如果提取结果为空
             if not text_content:
                 return f"错误: 无法从 '{file_path}' 提取文本内容"
+        elif file_path.lower().endswith('.ipynb'):
+            try:
+                with open(file_path, 'r', encoding='utf-8') as file:
+                    notebook_content = json.load(file)
+
+                for cell in notebook_content.get('cells', []):
+                    if cell.get('cell_type') == 'code' and 'outputs' in cell:
+                        filtered_outputs = []
+                        for output in cell.get('outputs', []):
+                            new_output = output.copy()
+                            if 'data' in new_output:
+                                original_data = new_output['data']
+                                filtered_data = {}
+                                for key, value in original_data.items():
+                                    if key.startswith('image/'):
+                                        continue
+                                    if key == 'text/html':
+                                        html_content = "".join(value) if isinstance(value, list) else value
+                                        if isinstance(html_content, str) and '<table class="show_videos"' in html_content:
+                                            continue
+                                    filtered_data[key] = value
+                                if filtered_data:
+                                    new_output['data'] = filtered_data
+                                    filtered_outputs.append(new_output)
+                            elif 'output_type' in new_output and new_output['output_type'] in ['stream', 'error']:
+                                filtered_outputs.append(new_output)
+
+                        cell['outputs'] = filtered_outputs
+
+                text_content = json.dumps(notebook_content, indent=2, ensure_ascii=False)
+            except json.JSONDecodeError:
+                return f"错误: 文件 '{file_path}' 不是有效的JSON格式 (IPython Notebook)。"
+            except Exception as e:
+                return f"处理IPython Notebook文件 '{file_path}' 时发生错误: {e}"
         else:
             # 读取文件内容
             with open(file_path, 'r', encoding='utf-8') as file:
@@ -76,4 +111,10 @@ Examples:
     except UnicodeDecodeError:
         return f"错误: 文件 '{file_path}' 不是文本文件或编码不是UTF-8"
     except Exception as e:
-        return f"读取文件时发生错误: {e}"
+        return f"读取文件时发生错误: {e}"
+
+if __name__ == "__main__":
+    # python -m beswarm.aient.src.aient.plugins.read_file
+    result = read_file("./work/cax/Lenia Notebook.ipynb")
+    print(result)
+    print(len(result))

{aient-1.0.76 → aient-1.0.78/src/aient.egg-info}/PKG-INFO

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