pycoze 0.1.316__tar.gz → 0.1.318__tar.gz

{pycoze-0.1.316 → pycoze-0.1.318}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pycoze
-Version: 0.1.316
+Version: 0.1.318
 Summary: Package for pycoze only!
 Author: Yuan Jie Xiong
 Author-email: aiqqqqqqq@qq.com

{pycoze-0.1.316 → pycoze-0.1.318}/pycoze/bot/tools.py

@@ -8,6 +8,25 @@ from .lib import get_formatted_filelist_str, read_local_file, resolve_relative_p
 from pycoze import ai
 from pycoze.api import window, web
 import traceback
+import inspect
+import asyncio
+
+async def handle_async_func(func, params):
+    return await func(**params)
+
+def handle_generator_func(func, params):
+    return list(func(**params))
+
+def handle_normal_func(func, params):
+    return func(**params)
+
+def call_func(func, params):
+    if inspect.isgeneratorfunction(func):
+        return handle_generator_func(func, params)
+    elif inspect.iscoroutinefunction(func):
+        return asyncio.run(handle_async_func(func, params))
+    else:
+        return handle_normal_func(func, params)
 
 
 class InvalidToolError(Exception):
@@ -265,8 +284,9 @@ class ToolExecutor:
             elif tool_name in [a.__name__ for a in abilities]:
                 func = [a for a in abilities if a.__name__ == tool_name][0]
                 try:
-                    result = func(**params)
+                    result = call_func(func, params)
                 except Exception as e:
+                    print(traceback.format_exc())
                     return False, traceback.format_exc()
                 return True, str(result)
             else:

{pycoze-0.1.316 → pycoze-0.1.318}/pycoze.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pycoze
-Version: 0.1.316
+Version: 0.1.318
 Summary: Package for pycoze only!
 Author: Yuan Jie Xiong
 Author-email: aiqqqqqqq@qq.com

{pycoze-0.1.316 → pycoze-0.1.318}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="pycoze",
-    version="0.1.316",
+    version="0.1.318",
     packages=find_packages(),
     install_requires=[],
     author="Yuan Jie Xiong",