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

synth_ai/zyk/lms/core/all.py

@@ -9,6 +9,7 @@ from synth_ai.zyk.lms.vendors.supported.together import TogetherAPI
 from synth_ai.zyk.lms.vendors.supported.groq import GroqAPI
 from synth_ai.zyk.lms.vendors.core.mistral_api import MistralAPI
 from synth_ai.zyk.lms.vendors.supported.custom_endpoint import CustomEndpointAPI
+from synth_ai.zyk.lms.vendors.supported.openrouter import OpenRouterAPI
 
 
 class OpenAIClient(OpenAIPrivate):
@@ -51,3 +52,8 @@ class MistralClient(MistralAPI):
 class CustomEndpointClient(CustomEndpointAPI):
     def __init__(self, endpoint_url: str):
         super().__init__(endpoint_url=endpoint_url)
+
+
+class OpenRouterClient(OpenRouterAPI):
+    def __init__(self):
+        super().__init__()

synth_ai/zyk/lms/core/vendor_clients.py

@@ -11,6 +11,7 @@ from synth_ai.zyk.lms.core.all import (
     OpenAIStructuredOutputClient,
     TogetherClient,
     CustomEndpointClient,
+    OpenRouterClient,
 )
 
 openai_naming_regexes: List[Pattern] = [
@@ -48,12 +49,17 @@ groq_naming_regexes: List[Pattern] = [
     re.compile(r"^meta-llama/llama-4-scout-17b-16e-instruct$"),
     re.compile(r"^meta-llama/llama-4-maverick-17b-128e-instruct$"),
     re.compile(r"^qwen/qwen3-32b$"),
+    re.compile(r"^moonshotai/kimi-k2-instruct$"),
 ]
 
 mistral_naming_regexes: List[Pattern] = [
     re.compile(r"^mistral-.*$"),
 ]
 
+openrouter_naming_regexes: List[Pattern] = [
+    re.compile(r"^openrouter/.*$"),  # openrouter/model-name pattern
+]
+
 # Custom endpoint patterns - check these before generic patterns
 custom_endpoint_naming_regexes: List[Pattern] = [
     # Modal endpoints: org--app.modal.run
@@ -92,6 +98,8 @@ def get_client(
         return GroqClient()
     elif any(regex.match(model_name) for regex in mistral_naming_regexes):
         return MistralClient()
+    elif any(regex.match(model_name) for regex in openrouter_naming_regexes):
+        return OpenRouterClient()
     elif any(regex.match(model_name) for regex in custom_endpoint_naming_regexes):
         # Custom endpoints are passed as the endpoint URL
         return CustomEndpointClient(endpoint_url=model_name)

synth_ai/zyk/lms/vendors/supported/openrouter.py

@@ -0,0 +1,70 @@
+import os
+from typing import Any, Dict, List, Optional
+from openai import AsyncOpenAI, OpenAI
+from synth_ai.zyk.lms.vendors.openai_standard import OpenAIStandard
+from synth_ai.zyk.lms.vendors.base import BaseLMResponse
+from synth_ai.zyk.lms.tools.base import BaseTool
+
+
+class OpenRouterAPI(OpenAIStandard):
+    """OpenRouter API client for accessing various models through OpenRouter's unified API."""
+    
+    def __init__(self):
+        api_key = os.getenv("OPENROUTER_API_KEY")
+        if not api_key:
+            raise ValueError("OPENROUTER_API_KEY environment variable is not set")
+        
+        # OpenRouter requires specific headers
+        default_headers = {
+            "HTTP-Referer": os.getenv("OPENROUTER_APP_URL", "https://github.com/synth-laboratories/synth-ai"),
+            "X-Title": os.getenv("OPENROUTER_APP_TITLE", "synth-ai")
+        }
+        
+        super().__init__(
+            sync_client=OpenAI(
+                api_key=api_key,
+                base_url="https://openrouter.ai/api/v1",
+                default_headers=default_headers
+            ),
+            async_client=AsyncOpenAI(
+                api_key=api_key,
+                base_url="https://openrouter.ai/api/v1",
+                default_headers=default_headers
+            )
+        )
+    
+    def _strip_prefix(self, model: str) -> str:
+        """Remove the 'openrouter/' prefix from model names."""
+        if model.startswith("openrouter/"):
+            return model[len("openrouter/"):]
+        return model
+    
+    async def _hit_api_async(
+        self,
+        model: str,
+        messages: List[Dict[str, Any]],
+        lm_config: Dict[str, Any],
+        use_ephemeral_cache_only: bool = False,
+        reasoning_effort: str = "high",
+        tools: Optional[List[BaseTool]] = None,
+    ) -> BaseLMResponse:
+        # Strip the 'openrouter/' prefix before calling the API
+        model = self._strip_prefix(model)
+        return await super()._hit_api_async(
+            model, messages, lm_config, use_ephemeral_cache_only, reasoning_effort, tools
+        )
+    
+    def _hit_api_sync(
+        self,
+        model: str,
+        messages: List[Dict[str, Any]],
+        lm_config: Dict[str, Any],
+        use_ephemeral_cache_only: bool = False,
+        reasoning_effort: str = "high",
+        tools: Optional[List[BaseTool]] = None,
+    ) -> BaseLMResponse:
+        # Strip the 'openrouter/' prefix before calling the API
+        model = self._strip_prefix(model)
+        return super()._hit_api_sync(
+            model, messages, lm_config, use_ephemeral_cache_only, reasoning_effort, tools
+        )

{synth_ai-0.1.6.dist-info → synth_ai-0.1.8.dist-info}/METADATA

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

{synth_ai-0.1.6.dist-info → synth_ai-0.1.8.dist-info}/RECORD

@@ -11,10 +11,10 @@ synth_ai/zyk/lms/caching/handler.py,sha256=HEo_e8q3kqDSVW0hN1AA6Rx5cBvlEeizi-kiO
 synth_ai/zyk/lms/caching/initialize.py,sha256=zZls6RKAax6Z-8oJInGaSg_RPN_fEZ6e_RCX64lMLJw,416
 synth_ai/zyk/lms/caching/persistent.py,sha256=ZaY1A9qhvfNKzcAI9FnwbIrgMKvVeIfb_yCyl3M8dxE,2860
 synth_ai/zyk/lms/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-synth_ai/zyk/lms/core/all.py,sha256=MAlhGk5LBK7En-oUzFBoemGl_GZbenmvqhlkSQ7cfGc,1410
+synth_ai/zyk/lms/core/all.py,sha256=hy6voRBO76geqG9IWWOyb_65Gay_iaEuRNblTz-gEi4,1574
 synth_ai/zyk/lms/core/exceptions.py,sha256=K0BVdAzxVIchsvYZAaHEH1GAWBZvpxhFi-SPcJOjyPQ,205
 synth_ai/zyk/lms/core/main.py,sha256=pLz7COTdvDWQivYaA1iYYF2onUOosD_sFaPJG48bdKM,10598
-synth_ai/zyk/lms/core/vendor_clients.py,sha256=7D30Ol-eVaJ8pZnbqeKADJ6q_Nz5buOd2qQJrrn8Rlk,3647
+synth_ai/zyk/lms/core/vendor_clients.py,sha256=aOON3T3OJ69U9mLF3C6-LQ9kcx6-nKZIbF2YzO_5Cl8,3948
 synth_ai/zyk/lms/cost/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 synth_ai/zyk/lms/cost/monitor.py,sha256=cSKIvw6WdPZIRubADWxQoh1MdB40T8-jjgfNUeUHIn0,5
 synth_ai/zyk/lms/cost/statefulness.py,sha256=TOsuXL8IjtKOYJ2aJQF8TwJVqn_wQ7AIwJJmdhMye7U,36
@@ -40,9 +40,10 @@ synth_ai/zyk/lms/vendors/supported/custom_endpoint.py,sha256=awluDoUgcRROsZdH_o_
 synth_ai/zyk/lms/vendors/supported/deepseek.py,sha256=BElW0NGpkSA62wOqzzMtDw8XR36rSNXK5LldeHJkQrc,2430
 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/openrouter.py,sha256=QdpqZFVCap-0oPsKiJkWr7o-dGaX3IMPSGehS2jw1IM,2607
 synth_ai/zyk/lms/vendors/supported/together.py,sha256=Ni_jBqqGPN0PkkY-Ew64s3gNKk51k3FCpLSwlNhKbf0,342
-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,,
+synth_ai-0.1.8.dist-info/licenses/LICENSE,sha256=ynhjRQUfqA_RdGRATApfFA_fBAy9cno04sLtLUqxVFM,1069
+synth_ai-0.1.8.dist-info/METADATA,sha256=mXoPocDv-vqgeXFJkrve0JBF-neWEU9CNEPSMjSfv9s,1084
+synth_ai-0.1.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+synth_ai-0.1.8.dist-info/top_level.txt,sha256=fBmtZyVHuKaGa29oHBaaUkrUIWTqSpoVMPiVdCDP3k8,9
+synth_ai-0.1.8.dist-info/RECORD,,