aient 1.0.76__py3-none-any.whl → 1.0.78__py3-none-any.whl

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/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.dist-info → aient-1.0.78.dist-info}/METADATA

@@ -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.dist-info → aient-1.0.78.dist-info}/RECORD

@@ -13,7 +13,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=t2AXtSfnuOquFcFDil2wYoB_xDYUGmqWvHManCm0gyU,42292
+aient/models/chatgpt.py,sha256=711ynHjMLXNtpoh5c93JHAesD9GZ9v7MpBospTUpJZM,42888
 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
@@ -26,7 +26,7 @@ aient/plugins/excute_command.py,sha256=0sQZGLCaRfh0yMM63fXU8XZV7jlo9FKeyEbM-rfgT
 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
-aient/plugins/read_file.py,sha256=kjHqtAGSy6AV8kFpv342x6yWGYfvZ9LGmxFZXJ3XzQI,2369
+aient/plugins/read_file.py,sha256=v07127SXOhS8YGJ6X82Gw2xlk-nlKvX7SdxzZVe1z0Y,4625
 aient/plugins/registry.py,sha256=YknzhieU_8nQ3oKlUSSWDB4X7t2Jx0JnqT2Jd9Xsvfk,3574
 aient/plugins/run_python.py,sha256=dgcUwBunMuDkaSKR5bToudVzSdrXVewktDDFUz_iIOQ,4589
 aient/plugins/websearch.py,sha256=yiBzqXK5X220ibR-zko3VDsn4QOnLu1k6E2YOygCeTQ,15185
@@ -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=n0jR5eXCBIK12W4bIx-xU1FVl1hZ4zDC7hq_BWQHYJU,27537
-aient-1.0.76.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
-aient-1.0.76.dist-info/METADATA,sha256=4NkFjXhTRJ2VaIcQG8PIHzTTaD2t8iZt85--GpxR3B4,5000
-aient-1.0.76.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
-aient-1.0.76.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
-aient-1.0.76.dist-info/RECORD,,
+aient-1.0.78.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
+aient-1.0.78.dist-info/METADATA,sha256=u_ftnmrNBcaKp2aPJrFYey2mWB67bYmGqER_iKT4Th8,5000
+aient-1.0.78.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+aient-1.0.78.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
+aient-1.0.78.dist-info/RECORD,,