beswarm 0.1.18__py3-none-any.whl → 0.1.20__py3-none-any.whl

beswarm/aient/setup.py

@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="aient",
-    version="1.0.77",
+    version="1.0.79",
     description="Aient: The Awakening of Agent.",
     long_description=Path.open(Path("README.md"), encoding="utf-8").read(),
     long_description_content_type="text/markdown",

beswarm/aient/src/aient/models/chatgpt.py

@@ -412,9 +412,11 @@ class chatgpt(BaseLLM):
             for tool_dict in function_parameter:
                 if tool_dict.get("function_name", "") not in self.plugins.keys():
                     function_parameter.remove(tool_dict)
+                    full_response = full_response + f"\n\nFunction: {tool_dict.get('function_name', '')} does not exist! I must use existing functions. I need to try again."
             if function_parameter:
                 need_function_call = True
             else:
+                need_function_call = False
                 if self.print_log:
                     print("Failed to parse function_parameter", function_parameter)
                     print("full_response", full_response)

beswarm/aient/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))

beswarm/tools/worker.py

@@ -92,6 +92,8 @@ async def worker(goal, tools, work_dir, cache_messages=None):
             print("\n🤖 指令智能体生成的下一步指令:", next_instruction)
             if "fetch_gpt_response_stream HTTP Error', 'status_code': 404" in next_instruction:
                 raise Exception(f"Model: {instruction_agent_config['engine']} not found!")
+            if "'status_code': 413" in next_instruction:
+                raise Exception(f"The request body is too long, please try again.")
             next_instruction = extract_xml_content(next_instruction, "instructions")
             if not next_instruction:
                 print("\n❌ 指令智能体生成的指令不符合要求，请重新生成。")

{beswarm-0.1.18.dist-info → beswarm-0.1.20.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beswarm
-Version: 0.1.18
+Version: 0.1.20
 Summary: MAS
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown

{beswarm-0.1.18.dist-info → beswarm-0.1.20.dist-info}/RECORD

@@ -1,7 +1,7 @@
 beswarm/__init__.py,sha256=HZjUOJtZR5QhMuDbq-wukQQn1VrBusNWai_ysGo-VVI,20
 beswarm/utils.py,sha256=AdDCcqAIIKQEMl7PfryVgeT9G5sHe7QNsZnrvmTGA8E,283
 beswarm/aient/main.py,sha256=SiYAIgQlLJqYusnTVEJOx1WNkSJKMImhgn5aWjfroxg,3814
-beswarm/aient/setup.py,sha256=xH0Y3kjGbroYYcerSOt236rxmEN3pe3l5oWABAHbkSQ,487
+beswarm/aient/setup.py,sha256=EzoroFZIKVg3AI6yCw9_O8zbBKEUmPzLYs5ls_Gdh9E,487
 beswarm/aient/src/aient/__init__.py,sha256=SRfF7oDVlOOAi6nGKiJIUK6B_arqYLO9iSMp-2IZZps,21
 beswarm/aient/src/aient/core/__init__.py,sha256=NxjebTlku35S4Dzr16rdSqSTWUvvwEeACe8KvHJnjPg,34
 beswarm/aient/src/aient/core/log_config.py,sha256=kz2_yJv1p-o3lUQOwA3qh-LSc3wMHv13iCQclw44W9c,274
@@ -15,7 +15,7 @@ beswarm/aient/src/aient/core/test/test_payload.py,sha256=8jBiJY1uidm1jzL-EiK0s6U
 beswarm/aient/src/aient/models/__init__.py,sha256=ouNDNvoBBpIFrLsk09Q_sq23HR0GbLAKfGLIFmfEuXE,219
 beswarm/aient/src/aient/models/audio.py,sha256=kRd-8-WXzv4vwvsTGwnstK-WR8--vr9CdfCZzu8y9LA,1934
 beswarm/aient/src/aient/models/base.py,sha256=z-Z0pJfTN2x0cuwfvu0BdMRY9O-RmLwHEnBIJN1x4Fg,6719
-beswarm/aient/src/aient/models/chatgpt.py,sha256=711ynHjMLXNtpoh5c93JHAesD9GZ9v7MpBospTUpJZM,42888
+beswarm/aient/src/aient/models/chatgpt.py,sha256=fWow2ezxduRU3sQxjoK7ZTOAQeY0byP-lCwYf-JcJUM,43105
 beswarm/aient/src/aient/models/claude.py,sha256=thK9P8qkaaoUN3OOJ9Shw4KDs-pAGKPoX4FOPGFXva8,28597
 beswarm/aient/src/aient/models/duckduckgo.py,sha256=1l7vYCs9SG5SWPCbcl7q6pCcB5AUF_r-a4l9frz3Ogo,8115
 beswarm/aient/src/aient/models/gemini.py,sha256=chGLc-8G_DAOxr10HPoOhvVFW1RvMgHd6mt--VyAW98,14730
@@ -28,7 +28,7 @@ beswarm/aient/src/aient/plugins/excute_command.py,sha256=0sQZGLCaRfh0yMM63fXU8XZ
 beswarm/aient/src/aient/plugins/get_time.py,sha256=Ih5XIW5SDAIhrZ9W4Qe5Hs1k4ieKPUc_LAd6ySNyqZk,654
 beswarm/aient/src/aient/plugins/image.py,sha256=ZElCIaZznE06TN9xW3DrSukS7U3A5_cjk1Jge4NzPxw,2072
 beswarm/aient/src/aient/plugins/list_directory.py,sha256=5ubm-mfrj-tanGSDp4M_Tmb6vQb3dx2-XVfQ2yL2G8A,1394
-beswarm/aient/src/aient/plugins/read_file.py,sha256=kjHqtAGSy6AV8kFpv342x6yWGYfvZ9LGmxFZXJ3XzQI,2369
+beswarm/aient/src/aient/plugins/read_file.py,sha256=v07127SXOhS8YGJ6X82Gw2xlk-nlKvX7SdxzZVe1z0Y,4625
 beswarm/aient/src/aient/plugins/registry.py,sha256=YknzhieU_8nQ3oKlUSSWDB4X7t2Jx0JnqT2Jd9Xsvfk,3574
 beswarm/aient/src/aient/plugins/run_python.py,sha256=dgcUwBunMuDkaSKR5bToudVzSdrXVewktDDFUz_iIOQ,4589
 beswarm/aient/src/aient/plugins/websearch.py,sha256=yiBzqXK5X220ibR-zko3VDsn4QOnLu1k6E2YOygCeTQ,15185
@@ -124,8 +124,8 @@ beswarm/tools/planner.py,sha256=lguBCS6kpwNPoXQvqH-WySabVubT82iyWOkJnjt6dXw,1265
 beswarm/tools/repomap.py,sha256=CwvwoN5Swr42EzrORTTeV8MMb7mPviy4a4b0fxBu50k,40828
 beswarm/tools/search_arxiv.py,sha256=9slwBemXjEqrd7-YgVmyMijPXlkhZCybEDRVhWVQ9B0,7937
 beswarm/tools/think.py,sha256=WLw-7jNIsnS6n8MMSYUin_f-BGLENFmnKM2LISEp0co,1760
-beswarm/tools/worker.py,sha256=3YnSyXaboNJ9j052NMLoIEOtxWL-uJ6sllonm7jV02E,5104
-beswarm-0.1.18.dist-info/METADATA,sha256=Z2n5ijYcI9b_6Vt4ANp0C_GSAicldrWWUDdRFpyvJkM,2870
-beswarm-0.1.18.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
-beswarm-0.1.18.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
-beswarm-0.1.18.dist-info/RECORD,,
+beswarm/tools/worker.py,sha256=JZkMT4JAkrV_0-CBATLAPVZzVGEMMEM2XUTTMGmED0E,5245
+beswarm-0.1.20.dist-info/METADATA,sha256=k7N6PewO59xBaDZoNU2ZfhzIdx8E3RQu2K7iVCZs0ug,2870
+beswarm-0.1.20.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
+beswarm-0.1.20.dist-info/top_level.txt,sha256=pJw4O87wvt5882smuSO6DfByJz7FJ8SxxT8h9fHCmpo,8
+beswarm-0.1.20.dist-info/RECORD,,