stores 0.1.7.dev4__py3-none-any.whl → 0.1.7.dev6__py3-none-any.whl

stores/indexes/venv_utils.py

@@ -5,6 +5,7 @@ import json
 import logging
 import os
 import pickle
+import queue
 import socket
 import subprocess
 import sys
@@ -340,16 +341,43 @@ def parse_tool_signature(
     elif signature_dict.get("isgeneratorfunction"):
 
         def func_handler(*args, **kwargs):
-            for value in run_remote_tool(
-                tool_id=signature_dict["tool_id"],
-                index_folder=index_folder,
-                args=args,
-                kwargs=kwargs,
-                venv=venv,
-                env_var=env_var,
-                stream=True,
-            ):
-                yield value
+            q = queue.Queue()
+            sentinel = object()
+
+            def run():
+                async def runner():
+                    try:
+                        async for item in run_remote_tool(
+                            tool_id=signature_dict["tool_id"],
+                            index_folder=index_folder,
+                            args=args,
+                            kwargs=kwargs,
+                            venv=venv,
+                            env_var=env_var,
+                            stream=True,
+                        ):
+                            q.put(item)
+                    except Exception as e:
+                        q.put(e)
+                    finally:
+                        q.put(sentinel)
+
+                asyncio.run(runner())
+
+            t = threading.Thread(target=run)
+            t.start()
+
+            while True:
+                item = q.get()
+                if item is sentinel:
+                    break
+                elif isinstance(item, Exception):
+                    raise item
+                else:
+                    yield item
+
+            t.join()
+
     elif signature_dict.get("iscoroutinefunction"):
 
         async def func_handler(*args, **kwargs):

{stores-0.1.7.dev4.dist-info → stores-0.1.7.dev6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stores
-Version: 0.1.7.dev4
+Version: 0.1.7.dev6
 Summary: Repository of Python functions and tools for LLMs
 License-File: LICENSE
 Requires-Python: >=3.10

{stores-0.1.7.dev4.dist-info → stores-0.1.7.dev6.dist-info}/RECORD

@@ -8,8 +8,8 @@ stores/indexes/base_index.py,sha256=YrEwETZ5eXj3rXK5qxOllRXqFifQoteYdzPAasbvEyg,
 stores/indexes/index.py,sha256=Cub5mtnYGipHfPR8BexJYRSKfuJmcGPp0B3ou2bGNqs,2901
 stores/indexes/local_index.py,sha256=Gg9LkEo1L0_NZZYPItsF_-1y6nFP369C3QlPvPyvPx8,3708
 stores/indexes/remote_index.py,sha256=-GV4l2c7GL6_bcVOQnSK5rzYru237bwC5L6eiO-QFzM,3438
-stores/indexes/venv_utils.py,sha256=-6cL9dcW9TVc1rTGHTCFIIrA38FaG8DKElNuwRL9-78,17764
-stores-0.1.7.dev4.dist-info/METADATA,sha256=kmOekRAa1E8TGBgejyKqDNuzt3A7XSCCOeytEfwp1_s,3081
-stores-0.1.7.dev4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-stores-0.1.7.dev4.dist-info/licenses/LICENSE,sha256=VTidYE7_Dam0Dwyq095EhhDIqi47g03oVpLAHQgKws0,1066
-stores-0.1.7.dev4.dist-info/RECORD,,
+stores/indexes/venv_utils.py,sha256=k664ptVhPfs4AHCpB6CiE-pWHGcqi982YjsZDDt64Ys,18577
+stores-0.1.7.dev6.dist-info/METADATA,sha256=Gg4CaLZDS0isd_R0Vhpn7IGCIQQ51Og5rBuBANnV7tk,3081
+stores-0.1.7.dev6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+stores-0.1.7.dev6.dist-info/licenses/LICENSE,sha256=VTidYE7_Dam0Dwyq095EhhDIqi47g03oVpLAHQgKws0,1066
+stores-0.1.7.dev6.dist-info/RECORD,,