stores 0.1.2__py3-none-any.whl → 0.1.4__py3-none-any.whl

stores/indexes/index.py

@@ -19,6 +19,7 @@ class Index(BaseIndex):
         env_var: dict[str, dict] | None = None,
         cache_dir: Optional[os.PathLike] = None,
         reset_cache=False,
+        sys_executable: str | None = None,
     ):
         self.env_var = env_var or {}
         tools = tools or []
@@ -44,6 +45,7 @@ class Index(BaseIndex):
                             env_var=self.env_var.get(index_name),
                             cache_dir=cache_dir,
                             reset_cache=reset_cache,
+                            sys_executable=sys_executable,
                         )
                     except Exception:
                         logger.warning(

stores/indexes/local_index.py

@@ -1,6 +1,7 @@
 import importlib
 import logging
 import os
+import subprocess
 import sys
 import venv
 from pathlib import Path
@@ -25,6 +26,7 @@ class LocalIndex(BaseIndex):
         index_folder: os.PathLike,
         create_venv: bool = False,
         env_var: dict | None = None,
+        sys_executable: str | None = None,
     ):
         self.index_folder = Path(index_folder)
         self.env_var = env_var or {}
@@ -38,7 +40,14 @@ class LocalIndex(BaseIndex):
             # Create venv and install deps
             self.venv = self.index_folder / VENV_NAME
             if not self.venv.exists():
-                venv.create(self.venv, symlinks=True, with_pip=True, upgrade_deps=True)
+                if sys_executable:
+                    subprocess.run(
+                        [sys_executable, "-m", "venv", str(self.venv)], check=True
+                    )
+                else:
+                    venv.create(
+                        self.venv, symlinks=True, with_pip=True, upgrade_deps=True
+                    )
             install_venv_deps(self.index_folder)
             # Initialize tools
             tools = init_venv_tools(self.index_folder, self.env_var)

stores/indexes/remote_index.py

@@ -1,6 +1,7 @@
 import json
 import logging
 import shutil
+import subprocess
 import venv
 from os import PathLike
 from pathlib import Path
@@ -52,6 +53,7 @@ class RemoteIndex(BaseIndex):
         env_var: dict | None = None,
         cache_dir: Optional[PathLike] = None,
         reset_cache=False,
+        sys_executable: str | None = None,
     ):
         self.index_id = index_id
         if cache_dir is None:
@@ -89,7 +91,12 @@ class RemoteIndex(BaseIndex):
         # Create venv and install deps
         self.venv = self.index_folder / VENV_NAME
         if not self.venv.exists():
-            venv.create(self.venv, symlinks=True, with_pip=True, upgrade_deps=True)
+            if sys_executable:
+                subprocess.run(
+                    [sys_executable, "-m", "venv", str(self.venv)], check=True
+                )
+            else:
+                venv.create(self.venv, symlinks=True, with_pip=True, upgrade_deps=True)
         install_venv_deps(self.index_folder)
         # Initialize tools
         tools = init_venv_tools(self.index_folder, self.env_var)

stores/indexes/venv_utils.py

@@ -93,6 +93,7 @@ def init_venv_tools(index_folder: os.PathLike, env_var: dict | None = None):
             tool_id=tool_id,
             index_folder=index_folder,
             venv=VENV_NAME,
+            env_var=env_var,
         )
         tool = parse_tool_signature(
             signature_dict=tool_sig,
@@ -105,9 +106,15 @@ def init_venv_tools(index_folder: os.PathLike, env_var: dict | None = None):
 
 
 # TODO: Sanitize tool_id, args, and kwargs
-def get_tool_signature(tool_id: str, index_folder: os.PathLike, venv: str = VENV_NAME):
+def get_tool_signature(
+    tool_id: str,
+    index_folder: os.PathLike,
+    venv: str = VENV_NAME,
+    env_var: dict | None = None,
+):
     module_name = ".".join(tool_id.split(".")[:-1])
     tool_name = tool_id.split(".")[-1]
+    env_var = env_var or {}
 
     runner = f"""
 import pickle, sys, traceback, inspect, enum
@@ -197,6 +204,7 @@ except Exception as e:
         [f"{venv}/bin/python", "-c", runner],
         capture_output=True,
         cwd=index_folder,
+        env=env_var,
     )
     try:
         response = pickle.loads(result.stdout)

{stores-0.1.2.dist-info → stores-0.1.4.dist-info}/METADATA

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

stores-0.1.4.dist-info/RECORD

@@ -0,0 +1,15 @@
+stores/__init__.py,sha256=KYpKkNrMLx6ssVUbxHnn9wFBq5F5KnaFchcimIfDf9g,186
+stores/constants.py,sha256=7WqFmoGCtmUKHA5WHxOJvvK7g-yYu_KGoqnuVFADNao,57
+stores/format.py,sha256=LduYBVDiUDB1J1HDyu9jHrRG1V97pw6C5g76OirpJJY,7792
+stores/parse.py,sha256=HYPNPzQod2vpu1Cln7yQ8aVkZT1Mw2IN0sZ2A1DIaqE,4967
+stores/utils.py,sha256=GPWT6lCoGobwP3PlEOHyJfKyd0dobamjyErcR7lgm7M,242
+stores/indexes/__init__.py,sha256=s-RNqml8uGREQhxwSdDoxcbcxeD8soB9BcL5dBKsQfI,215
+stores/indexes/base_index.py,sha256=K4--_OAFyPNESaisD0JPNRIEffCnjw9nziahDgjdZHU,10708
+stores/indexes/index.py,sha256=C3i5JAwYoKEK7jIJRAgnLrQxjeewtInKcMinep6yvyA,2432
+stores/indexes/local_index.py,sha256=TVk7W8BXF5KEqjEK0LYqq3CtLMfHaF-Tk_5cZ6bjsBU,3189
+stores/indexes/remote_index.py,sha256=CHa9kQVDQlgI8qw3XQXchrE1cjXJk9fNY2j2HWL2gio,3226
+stores/indexes/venv_utils.py,sha256=U69UsHn9APTc615TchW2aZmrBORgbZhuZunv4zQv_hU,12514
+stores-0.1.4.dist-info/METADATA,sha256=Rkj7Ot2ClwUiCfkhejFU7Hs8PvXxTEdBXNQeWtlIrGA,3076
+stores-0.1.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+stores-0.1.4.dist-info/licenses/LICENSE,sha256=VTidYE7_Dam0Dwyq095EhhDIqi47g03oVpLAHQgKws0,1066
+stores-0.1.4.dist-info/RECORD,,

stores-0.1.2.dist-info/RECORD

@@ -1,15 +0,0 @@
-stores/__init__.py,sha256=KYpKkNrMLx6ssVUbxHnn9wFBq5F5KnaFchcimIfDf9g,186
-stores/constants.py,sha256=7WqFmoGCtmUKHA5WHxOJvvK7g-yYu_KGoqnuVFADNao,57
-stores/format.py,sha256=LduYBVDiUDB1J1HDyu9jHrRG1V97pw6C5g76OirpJJY,7792
-stores/parse.py,sha256=HYPNPzQod2vpu1Cln7yQ8aVkZT1Mw2IN0sZ2A1DIaqE,4967
-stores/utils.py,sha256=GPWT6lCoGobwP3PlEOHyJfKyd0dobamjyErcR7lgm7M,242
-stores/indexes/__init__.py,sha256=s-RNqml8uGREQhxwSdDoxcbcxeD8soB9BcL5dBKsQfI,215
-stores/indexes/base_index.py,sha256=K4--_OAFyPNESaisD0JPNRIEffCnjw9nziahDgjdZHU,10708
-stores/indexes/index.py,sha256=X8q2Pe6PCmzQ9j0KVvJmqZid1A1RB0jUe-JugcOrl7o,2330
-stores/indexes/local_index.py,sha256=NS4AWzeMVZ51YPo0eJ_udbF5aASuksG-88YXLDn41fQ,2880
-stores/indexes/remote_index.py,sha256=dp84keyEDyL-cQrmshwHd4QqDGMw0CTSJcQxqZQiULY,2983
-stores/indexes/venv_utils.py,sha256=Nakh2L7gMXpP_0_g7pnjQkB1g-P-jUCxpWSxg68JjbQ,12388
-stores-0.1.2.dist-info/METADATA,sha256=GDKpmD4EPbitZaIA2BSEu7r-CWBoo9Y7bO-90f79u6k,3076
-stores-0.1.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-stores-0.1.2.dist-info/licenses/LICENSE,sha256=VTidYE7_Dam0Dwyq095EhhDIqi47g03oVpLAHQgKws0,1066
-stores-0.1.2.dist-info/RECORD,,