stores 0.1.7.dev5__tar.gz → 0.1.7.dev6__tar.gz

{stores-0.1.7.dev5 → stores-0.1.7.dev6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stores
-Version: 0.1.7.dev5
+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.dev5 → stores-0.1.7.dev6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "stores"
-version = "0.1.7dev5"
+version = "0.1.7dev6"
 description = "Repository of Python functions and tools for LLMs"
 readme = "README.md"
 requires-python = ">=3.10"

stores-0.1.7.dev6/run_sandbox.py

@@ -0,0 +1,24 @@
+import stores  # noqa
+
+index = stores.Index(
+    ["silanthro/python-sandbox"],
+    # env_var={
+    #     "silanthro/python-sandbox": {
+    #         "DENO_PATH": "/drive3/Silanthro/tools/python-sandbox/deno"
+    #     }
+    # },
+)
+print(index.tools)
+
+# for value in index.stream_execute("sandbox.run_code", {"code": "1+1"}):
+#     print(value)
+
+print(
+    index.execute(
+        "sandbox.run_code",
+        kwargs={
+            "timeout": 5,
+            "code": "def is_prime(n):\n    if n <= 1:\n        return False\n    for i in range(2, int(n**0.5) + 1):\n        if n % i == 0:\n            return False\n    return True\n\nprimes = []\nnum = 2\nwhile len(primes) < 10:\n    if is_prime(num):\n        primes.append(num)\n    num += 1\n\nsum(primes)",
+        },
+    )
+)

{stores-0.1.7.dev5 → stores-0.1.7.dev6}/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,28 +341,43 @@ def parse_tool_signature(
     elif signature_dict.get("isgeneratorfunction"):
 
         def func_handler(*args, **kwargs):
-            async def collect():
-                async 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
-
-            loop = asyncio.new_event_loop()
-            asyncio.set_event_loop(loop)
-            agen = collect()
-            try:
-                while True:
-                    yield loop.run_until_complete(agen.__anext__())
-            except StopAsyncIteration:
-                pass
-            finally:
-                loop.close()
+            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.dev5/run_sandbox.py

@@ -1,16 +0,0 @@
-import stores  # noqa
-
-index = stores.Index(
-    ["silanthro/python-sandbox"],
-    # env_var={
-    #     "silanthro/python-sandbox": {
-    #         "DENO_PATH": "/drive3/Silanthro/tools/python-sandbox/deno"
-    #     }
-    # },
-)
-print(index.tools)
-
-# for value in index.stream_execute("sandbox.run_code", {"code": "1+1"}):
-#     print(value)
-
-print(index.execute("sandbox.run_code", {"code": "1+1"}))