synth-ai 0.1.4__tar.gz → 0.1.6__tar.gz

{synth_ai-0.1.4 → synth_ai-0.1.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: synth-ai
-Version: 0.1.4
+Version: 0.1.6
 Home-page: https://github.com/synth-laboratories/synth-ai
 Author: Josh Purtell
 Author-email: josh@usesynth.com

{synth_ai-0.1.4 → synth_ai-0.1.6}/setup.py

@@ -2,7 +2,7 @@ from setuptools import find_packages, setup
 
 setup(
     name="synth-ai",
-    version="0.1.4",
+    version="0.1.6",
     packages=find_packages(),
     install_requires=[
         "openai",

{synth_ai-0.1.4 → synth_ai-0.1.6}/synth_ai/__init__.py

@@ -4,5 +4,5 @@ Synth AI - Software for aiding the best and multiplying the will.
 
 from synth_ai.zyk import LM  # Assuming LM is in zyk.py in the same directory
 
-__version__ = "0.1.4"
+__version__ = "0.1.6"
 __all__ = ["LM"]  # Explicitly define public API

synth_ai-0.1.6/synth_ai/zyk/lms/tools/__init__.py

@@ -0,0 +1,3 @@
+from .base import BaseTool
+
+__all__ = ["BaseTool"]

synth_ai-0.1.6/synth_ai/zyk/lms/tools/base.py

@@ -0,0 +1,118 @@
+from typing import Type
+
+from pydantic import BaseModel
+
+
+class BaseTool(BaseModel):
+    name: str
+    arguments: Type[BaseModel]
+    description: str = ""
+    strict: bool = True
+
+    def to_openai_tool(self):
+        schema = self.arguments.model_json_schema()
+        schema["additionalProperties"] = False
+
+        if "properties" in schema:
+            for prop_name, prop_schema in schema["properties"].items():
+                if prop_schema.get("type") == "array":
+                    items_schema = prop_schema.get("items", {})
+                    if not isinstance(items_schema, dict) or not items_schema.get("type"):
+                        prop_schema["items"] = {"type": "string"}
+
+                elif "anyOf" in prop_schema:
+                    for sub_schema in prop_schema["anyOf"]:
+                        if isinstance(sub_schema, dict) and sub_schema.get("type") == "array":
+                            items_schema = sub_schema.get("items", {})
+                            if not isinstance(items_schema, dict) or not items_schema.get("type"):
+                                sub_schema["items"] = {"type": "string"}
+
+        return {
+            "type": "function",
+            "function": {
+                "name": self.name,
+                "description": self.description,
+                "parameters": schema,
+                "strict": self.strict,
+            },
+        }
+
+    def to_anthropic_tool(self):
+        schema = self.arguments.model_json_schema()
+        schema["additionalProperties"] = False
+
+        return {
+            "name": self.name,
+            "description": self.description,
+            "input_schema": {
+                "type": "object",
+                "properties": schema["properties"],
+                "required": schema.get("required", []),
+            },
+        }
+
+    def to_mistral_tool(self):
+        schema = self.arguments.model_json_schema()
+        properties = {}
+        for prop_name, prop in schema.get("properties", {}).items():
+            prop_type = prop["type"]
+            if prop_type == "array" and "items" in prop:
+                properties[prop_name] = {
+                    "type": "array",
+                    "items": prop["items"],
+                    "description": prop.get("description", ""),
+                }
+                continue
+
+            properties[prop_name] = {
+                "type": prop_type,
+                "description": prop.get("description", ""),
+            }
+            if "enum" in prop:
+                properties[prop_name]["enum"] = prop["enum"]
+
+        parameters = {
+            "type": "object",
+            "properties": properties,
+            "required": schema.get("required", []),
+            "additionalProperties": False,
+        }
+        return {
+            "type": "function",
+            "function": {
+                "name": self.name,
+                "description": self.description,
+                "parameters": parameters,
+            },
+        }
+
+    def to_gemini_tool(self):
+        schema = self.arguments.model_json_schema()
+        # Convert Pydantic schema types to Gemini schema types
+        properties = {}
+        for name, prop in schema["properties"].items():
+            prop_type = prop.get("type", "string")
+            if prop_type == "array" and "items" in prop:
+                properties[name] = {
+                    "type": "array",
+                    "items": prop["items"],
+                    "description": prop.get("description", ""),
+                }
+                continue
+
+            properties[name] = {
+                "type": prop_type,
+                "description": prop.get("description", ""),
+            }
+            if "enum" in prop:
+                properties[name]["enum"] = prop["enum"]
+
+        return {
+            "name": self.name,
+            "description": self.description,
+            "parameters": {
+                "type": "object",
+                "properties": properties,
+                "required": schema.get("required", []),
+            },
+        }

{synth_ai-0.1.4 → synth_ai-0.1.6}/synth_ai.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: synth-ai
-Version: 0.1.4
+Version: 0.1.6
 Home-page: https://github.com/synth-laboratories/synth-ai
 Author: Josh Purtell
 Author-email: josh@usesynth.com

{synth_ai-0.1.4 → synth_ai-0.1.6}/synth_ai.egg-info/SOURCES.txt

@@ -30,6 +30,8 @@ synth_ai/zyk/lms/structured_outputs/__init__.py
 synth_ai/zyk/lms/structured_outputs/handler.py
 synth_ai/zyk/lms/structured_outputs/inject.py
 synth_ai/zyk/lms/structured_outputs/rehabilitate.py
+synth_ai/zyk/lms/tools/__init__.py
+synth_ai/zyk/lms/tools/base.py
 synth_ai/zyk/lms/vendors/__init__.py
 synth_ai/zyk/lms/vendors/base.py
 synth_ai/zyk/lms/vendors/openai_standard.py