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

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

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

{aient-1.0.77.dist-info → aient-1.0.78.dist-info}/RECORD

@@ -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.77.dist-info/licenses/LICENSE,sha256=XNdbcWldt0yaNXXWB_Bakoqnxb3OVhUft4MgMA_71ds,1051
-aient-1.0.77.dist-info/METADATA,sha256=C5HByv10I5sF4pNpCeJcxYlVefOZBiqD9HBvSmQI4Kg,5000
-aient-1.0.77.dist-info/WHEEL,sha256=SmOxYU7pzNKBqASvQJ7DjX3XGUF92lrGhMb3R6_iiqI,91
-aient-1.0.77.dist-info/top_level.txt,sha256=3oXzrP5sAVvyyqabpeq8A2_vfMtY554r4bVE-OHBrZk,6
-aient-1.0.77.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,,