toolchemy 0.2.185__tar.gz → 0.2.186__tar.gz

{toolchemy-0.2.185 → toolchemy-0.2.186}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: toolchemy
-Version: 0.2.185
+Version: 0.2.186
 Summary: a set of auxiliary programming tools
 License-File: LICENSE
 Author: Cyprian Nosek
@@ -16,7 +16,7 @@ Requires-Dist: mlflow (>=3.8.1,<4.0.0)
 Requires-Dist: neptune-scale (>=0.13.0,<0.14.0)
 Requires-Dist: numpy (==2.0.2)
 Requires-Dist: ollama (>=0.6.0,<0.7.0)
-Requires-Dist: openai (>=1.82.1,<2.0.0)
+Requires-Dist: openai (>=2.16.0,<3.0.0)
 Requires-Dist: pillow (>=11.2.1,<12.0.0)
 Requires-Dist: python-dateutil (>=2.9.0.post0,<3.0.0)
 Requires-Dist: tenacity (>=9.1.2,<10.0.0)

{toolchemy-0.2.185 → toolchemy-0.2.186}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "toolchemy"
-version = "0.2.185"
+version = "0.2.186"
 description = "a set of auxiliary programming tools"
 authors = ["Cyprian Nosek <cyprian.nosek@protonmail.com>"]
 
@@ -11,7 +11,7 @@ torch = "^2.7.0"
 numpy = "2.0.2"
 mlflow = "^3.8.1"
 neptune-scale = "^0.13.0"
-openai = "^1.82.1"
+openai = "^2.16.0"
 pillow = "^11.2.1"
 wyoming = "^1.7.1"
 python-dateutil = "^2.9.0.post0"

toolchemy-0.2.186/toolchemy/__main__.py

@@ -0,0 +1,18 @@
+import click
+
+from toolchemy.ai.prompter import run_studio
+
+
+@click.group(context_settings=dict(allow_interspersed_args=False))
+def cli():
+    pass
+
+
+@cli.command(name="prompt-studio")
+@click.option("--registry-dir", "-r", type=str, required=False, default=None, help="Optional path to the registry dir")
+def prompt_studio(registry_dir: str | None = None):
+    run_studio(registry_path=registry_dir)
+
+
+if __name__ == '__main__':
+    cli()

{toolchemy-0.2.185 → toolchemy-0.2.186}/toolchemy/ai/prompter.py

@@ -27,8 +27,8 @@ class IPrompter(ABC):
         pass
 
 
-def run_studio():
-    prompter = PrompterMLflow()
+def run_studio(registry_path: str | None = None):
+    prompter = PrompterMLflow(registry_store_dir=registry_path)
     prompter.run_studio()
 
 

toolchemy-0.2.185/toolchemy/__main__.py

@@ -1,9 +0,0 @@
-from toolchemy.utils.logger import get_logger
-
-
-def main():
-    logger = get_logger(log_dir="/tmp")
-    logger.info("test")
-
-if __name__ == '__main__':
-    main()