pycoze 0.1.316__py3-none-any.whl → 0.1.318__py3-none-any.whl

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.dist-info → pycoze-0.1.318.dist-info}/METADATA

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

@@ -17,7 +17,7 @@ pycoze/bot/chat_base.py,sha256=czmOKUum3Rwo4lAOJtfslxaF8RKauWzj26s_WMmQa48,9180
 pycoze/bot/lib.py,sha256=IIihXPODCbM29TlxS0lhm2Fw3GWm2MzUeAn9LGa86lg,6889
 pycoze/bot/message.py,sha256=Zq-_k8HztBMOUIs3hbOvWvwHBNopn4UJJBliCROIGcc,718
 pycoze/bot/prompt.md,sha256=ehjdvZ-se9NaH0OSiDEvrFE9kvdj9MTZlWZUTxPY780,15743
-pycoze/bot/tools.py,sha256=987hUTy-kZfTwUUd3WStPhNUPkOIGCVm-JWL3Qh70pg,9648
+pycoze/bot/tools.py,sha256=utuQKp5S0XjE7qMhX6_jnpoXBzg4CsiP1rW733QVp0o,10237
 pycoze/reference/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pycoze/reference/bot.py,sha256=pxHVYo0G3P3YZ--vBYbMEiEyBoxxPwaO5dMTf9WFMSc,2014
 pycoze/reference/lib.py,sha256=6jvowP2LqjyV21ppRhmtxHxnQJHCbHmlpbj9btV2RQ4,2191
@@ -33,8 +33,8 @@ pycoze/utils/arg.py,sha256=jop1tBfe5hYkHW1NSpCeaZBEznkgguBscj_7M2dWfrs,503
 pycoze/utils/env.py,sha256=5pWlXfM1F5ZU9hhv1rHlDEanjEW5wf0nbyez9bNRqqA,559
 pycoze/utils/socket.py,sha256=bZbFFRH4mfThzRqt55BAAGQ6eICx_ja4x8UGGrUdAm8,2428
 pycoze/utils/text_or_file.py,sha256=gpxZVWt2DW6YiEg_MnMuwg36VNf3TX383QD_1oZNB0Y,551
-pycoze-0.1.316.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
-pycoze-0.1.316.dist-info/METADATA,sha256=-Shyz80jC19pUcPlpYDC8TkkDYCmsYeOnQpNYxnEp7U,854
-pycoze-0.1.316.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-pycoze-0.1.316.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
-pycoze-0.1.316.dist-info/RECORD,,
+pycoze-0.1.318.dist-info/LICENSE,sha256=QStd_Qsd0-kAam_-sOesCIp_uKrGWeoKwt9M49NVkNU,1090
+pycoze-0.1.318.dist-info/METADATA,sha256=agzpVdOKUSQyLvqFKmTlRuY0B-0Fp_iLNQv9w3gRo40,854
+pycoze-0.1.318.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+pycoze-0.1.318.dist-info/top_level.txt,sha256=76dPeDhKvOCleL3ZC5gl1-y4vdS1tT_U1hxWVAn7sFo,7
+pycoze-0.1.318.dist-info/RECORD,,