synth-ai 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

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/zyk/lms/tools/__init__.py

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

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.dist-info → synth_ai-0.1.6.dist-info}/METADATA

@@ -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.dist-info → synth_ai-0.1.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-synth_ai/__init__.py,sha256=06VvZlwVtT7OXvu5iqFvWhLw46BmQsIAYQe8wHNUiB0,225
+synth_ai/__init__.py,sha256=2y79prX5jWHSuU6LWUXIQueT6y_o79P1ACvxDjNSKI0,225
 synth_ai/zyk/__init__.py,sha256=kGMD-drlBVdsyT-QFODMwaZUtxPCJ9mg58GKQUvFqo0,134
 synth_ai/zyk/lms/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 synth_ai/zyk/lms/config.py,sha256=UBMi0DIFQDBV_eGPK5vG8R7VwxXcV10BPGq1iV8vVjg,282
@@ -22,6 +22,8 @@ synth_ai/zyk/lms/structured_outputs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQ
 synth_ai/zyk/lms/structured_outputs/handler.py,sha256=4DboLNZyXpqNB5YNCSgGHBsNbWqFk-8uwV944nOYNo8,16919
 synth_ai/zyk/lms/structured_outputs/inject.py,sha256=Fy-zDeleRxOZ8ZRM6IuZ6CP2XZnMe4K2PEn4Q9c_KPY,11777
 synth_ai/zyk/lms/structured_outputs/rehabilitate.py,sha256=ecKGWrgWYUSplqHzK40KdohwaN8gBV0xl4LUReLN_vg,7910
+synth_ai/zyk/lms/tools/__init__.py,sha256=3JM5vqZqKloaeHaxuO49C8A_0qljys3pQ1yt70WKhho,51
+synth_ai/zyk/lms/tools/base.py,sha256=i-AIVRlitiQ4JMJ_BBFRSpUcWgxWIUYoHxAqfxHN_7E,4056
 synth_ai/zyk/lms/vendors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 synth_ai/zyk/lms/vendors/base.py,sha256=aK4PEtkMLt_o3qD22kW-x3HJUEKdIk06zlH4kX0VkAE,760
 synth_ai/zyk/lms/vendors/openai_standard.py,sha256=dgHC7RWrxwaWto6_frKdfEKazKvvAMYJM1YgJgbVpb8,12279
@@ -39,8 +41,8 @@ synth_ai/zyk/lms/vendors/supported/deepseek.py,sha256=BElW0NGpkSA62wOqzzMtDw8XR3
 synth_ai/zyk/lms/vendors/supported/groq.py,sha256=Fbi7QvhdLx0F-VHO5PY-uIQlPR0bo3C9h1MvIOx8nz0,388
 synth_ai/zyk/lms/vendors/supported/ollama.py,sha256=K30VBFRTd7NYyPmyBVRZS2sm0UB651AHp9i3wd55W64,469
 synth_ai/zyk/lms/vendors/supported/together.py,sha256=Ni_jBqqGPN0PkkY-Ew64s3gNKk51k3FCpLSwlNhKbf0,342
-synth_ai-0.1.4.dist-info/licenses/LICENSE,sha256=ynhjRQUfqA_RdGRATApfFA_fBAy9cno04sLtLUqxVFM,1069
-synth_ai-0.1.4.dist-info/METADATA,sha256=Bz_Gf7gZtnLIJdgjrD17XS_HQNj_oGlXdFufcrLP2Ko,1084
-synth_ai-0.1.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-synth_ai-0.1.4.dist-info/top_level.txt,sha256=fBmtZyVHuKaGa29oHBaaUkrUIWTqSpoVMPiVdCDP3k8,9
-synth_ai-0.1.4.dist-info/RECORD,,
+synth_ai-0.1.6.dist-info/licenses/LICENSE,sha256=ynhjRQUfqA_RdGRATApfFA_fBAy9cno04sLtLUqxVFM,1069
+synth_ai-0.1.6.dist-info/METADATA,sha256=0hsFxGTFlxSxCisga5bK7WZXYRgO-ycB7dueqW4VW1g,1084
+synth_ai-0.1.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+synth_ai-0.1.6.dist-info/top_level.txt,sha256=fBmtZyVHuKaGa29oHBaaUkrUIWTqSpoVMPiVdCDP3k8,9
+synth_ai-0.1.6.dist-info/RECORD,,