spaik-sdk 0.6.4__py3-none-any.whl → 0.6.6__py3-none-any.whl

spaik_sdk/agent/base_agent.py

@@ -17,7 +17,6 @@ from spaik_sdk.llm.cost.cost_provider import CostProvider
 from spaik_sdk.llm.langchain_service import LangChainService
 from spaik_sdk.models.llm_config import LLMConfig
 from spaik_sdk.models.llm_model import LLMModel
-from spaik_sdk.models.providers.provider_type import ProviderType
 from spaik_sdk.prompt.get_prompt_loader import get_prompt_loader
 from spaik_sdk.prompt.prompt_loader import PromptLoader
 from spaik_sdk.prompt.prompt_loader_mode import PromptLoaderMode
@@ -147,11 +146,9 @@ class BaseAgent(ABC):
         if llm_model is None:
             llm_model = self.get_llm_model()
 
-        provider_type = ProviderType.from_family(llm_model.family)
-
         return LLMConfig(
             model=llm_model,
-            provider_type=provider_type,
+            provider_type=env_config.get_provider_type(),
             reasoning=reasoning if reasoning is not None else llm_model.reasoning,
             tool_usage=len(self.tools) > 0,
         )

spaik_sdk/config/env.py

@@ -1,5 +1,4 @@
 import os
-from typing import Dict
 from typing import Optional as OptionalType
 
 from spaik_sdk.models.llm_model import LLMModel
@@ -16,16 +15,6 @@ class EnvConfig:
             raise ValueError(f"Environment variable {key} is required but not set")
         return value
 
-    def get_azure_keys(self) -> Dict[str, str]:
-        return {
-            "api_key": self.get_key("AZURE_API_KEY"),
-            "api_version": self.get_key("AZURE_API_VERSION"),
-            "endpoint": self.get_key("AZURE_ENDPOINT"),
-            "o3-mini_deployment": self.get_key("AZURE_O3_MINI_DEPLOYMENT", required=False),
-            "gpt-4_1_deployment": self.get_key("AZURE_GPT_4_1_DEPLOYMENT", required=False),
-            "gpt-4o_deployment": self.get_key("AZURE_GPT_4O_DEPLOYMENT", required=False),
-        }
-
     def get_default_model(self) -> LLMModel:
         return ModelRegistry.from_name(self.get_key("DEFAULT_MODEL"))
 

spaik_sdk/models/factories/anthropic_factory.py

@@ -21,7 +21,7 @@ class AnthropicModelFactory(BaseModelFactory):
         model_config: Dict[str, Any] = {
             "model_name": config.model.name,
             "streaming": config.streaming,
-            "max_tokens": config.max_output_tokens,
+            "max_tokens": config.max_output_tokens if config.max_output_tokens is not None else 8192,
         }
 
         # Handle thinking mode via model_kwargs for LangChain compatibility

spaik_sdk/models/factories/base_model_factory.py

@@ -54,15 +54,27 @@ class BaseModelFactory(ABC):
         """Factory method to create appropriate factory instance."""
 
         from spaik_sdk.models.factories.anthropic_factory import AnthropicModelFactory
+        from spaik_sdk.models.factories.cohere_factory import CohereModelFactory
+        from spaik_sdk.models.factories.deepseek_factory import DeepSeekModelFactory
         from spaik_sdk.models.factories.google_factory import GoogleModelFactory
+        from spaik_sdk.models.factories.meta_factory import MetaModelFactory
+        from spaik_sdk.models.factories.mistral_factory import MistralModelFactory
+        from spaik_sdk.models.factories.moonshot_factory import MoonshotModelFactory
         from spaik_sdk.models.factories.ollama_factory import OllamaModelFactory
         from spaik_sdk.models.factories.openai_factory import OpenAIModelFactory
+        from spaik_sdk.models.factories.xai_factory import XAIModelFactory
 
         factories = [
             AnthropicModelFactory(),
             OpenAIModelFactory(),
             GoogleModelFactory(),
             OllamaModelFactory(),
+            DeepSeekModelFactory(),
+            XAIModelFactory(),
+            CohereModelFactory(),
+            MistralModelFactory(),
+            MetaModelFactory(),
+            MoonshotModelFactory(),
         ]
         for factory in factories:
             if factory.supports_model_config(config):

spaik_sdk/models/factories/cohere_factory.py

@@ -0,0 +1,24 @@
+from typing import Any, Dict, Optional
+
+from spaik_sdk.models.factories.base_model_factory import BaseModelFactory
+from spaik_sdk.models.llm_config import LLMConfig
+from spaik_sdk.models.llm_families import LLMFamilies
+from spaik_sdk.models.llm_model import LLMModel
+from spaik_sdk.models.model_registry import ModelRegistry
+
+
+class CohereModelFactory(BaseModelFactory):
+    MODELS = ModelRegistry.get_by_family(LLMFamilies.COHERE)
+
+    def supports_model(self, model: LLMModel) -> bool:
+        return model in CohereModelFactory.MODELS
+
+    def get_cache_control(self, config: LLMConfig) -> Optional[Dict[str, Any]]:
+        return None
+
+    def get_model_specific_config(self, config: LLMConfig) -> Dict[str, Any]:
+        model_config: Dict[str, Any] = {
+            "model": config.model.name,
+            "temperature": config.temperature,
+        }
+        return model_config

spaik_sdk/models/factories/deepseek_factory.py

@@ -0,0 +1,26 @@
+from typing import Any, Dict, Optional
+
+from spaik_sdk.models.factories.base_model_factory import BaseModelFactory
+from spaik_sdk.models.llm_config import LLMConfig
+from spaik_sdk.models.llm_families import LLMFamilies
+from spaik_sdk.models.llm_model import LLMModel
+from spaik_sdk.models.model_registry import ModelRegistry
+
+
+class DeepSeekModelFactory(BaseModelFactory):
+    MODELS = ModelRegistry.get_by_family(LLMFamilies.DEEPSEEK)
+
+    def supports_model(self, model: LLMModel) -> bool:
+        return model in DeepSeekModelFactory.MODELS
+
+    def get_cache_control(self, config: LLMConfig) -> Optional[Dict[str, Any]]:
+        return None
+
+    def get_model_specific_config(self, config: LLMConfig) -> Dict[str, Any]:
+        model_config: Dict[str, Any] = {
+            "model": config.model.name,
+            "temperature": config.temperature,
+        }
+        if config.max_output_tokens is not None:
+            model_config["max_tokens"] = config.max_output_tokens
+        return model_config

spaik_sdk/models/factories/meta_factory.py

@@ -0,0 +1,26 @@
+from typing import Any, Dict, Optional
+
+from spaik_sdk.models.factories.base_model_factory import BaseModelFactory
+from spaik_sdk.models.llm_config import LLMConfig
+from spaik_sdk.models.llm_families import LLMFamilies
+from spaik_sdk.models.llm_model import LLMModel
+from spaik_sdk.models.model_registry import ModelRegistry
+
+
+class MetaModelFactory(BaseModelFactory):
+    MODELS = ModelRegistry.get_by_family(LLMFamilies.META)
+
+    def supports_model(self, model: LLMModel) -> bool:
+        return model in MetaModelFactory.MODELS
+
+    def get_cache_control(self, config: LLMConfig) -> Optional[Dict[str, Any]]:
+        return None
+
+    def get_model_specific_config(self, config: LLMConfig) -> Dict[str, Any]:
+        model_config: Dict[str, Any] = {
+            "model": config.model.name,
+            "temperature": config.temperature,
+        }
+        if config.max_output_tokens is not None:
+            model_config["max_tokens"] = config.max_output_tokens
+        return model_config

spaik_sdk/models/factories/mistral_factory.py

@@ -0,0 +1,26 @@
+from typing import Any, Dict, Optional
+
+from spaik_sdk.models.factories.base_model_factory import BaseModelFactory
+from spaik_sdk.models.llm_config import LLMConfig
+from spaik_sdk.models.llm_families import LLMFamilies
+from spaik_sdk.models.llm_model import LLMModel
+from spaik_sdk.models.model_registry import ModelRegistry
+
+
+class MistralModelFactory(BaseModelFactory):
+    MODELS = ModelRegistry.get_by_family(LLMFamilies.MISTRAL)
+
+    def supports_model(self, model: LLMModel) -> bool:
+        return model in MistralModelFactory.MODELS
+
+    def get_cache_control(self, config: LLMConfig) -> Optional[Dict[str, Any]]:
+        return None
+
+    def get_model_specific_config(self, config: LLMConfig) -> Dict[str, Any]:
+        model_config: Dict[str, Any] = {
+            "model": config.model.name,
+            "temperature": config.temperature,
+        }
+        if config.max_output_tokens is not None:
+            model_config["max_tokens"] = config.max_output_tokens
+        return model_config

spaik_sdk/models/factories/moonshot_factory.py

@@ -0,0 +1,26 @@
+from typing import Any, Dict, Optional
+
+from spaik_sdk.models.factories.base_model_factory import BaseModelFactory
+from spaik_sdk.models.llm_config import LLMConfig
+from spaik_sdk.models.llm_families import LLMFamilies
+from spaik_sdk.models.llm_model import LLMModel
+from spaik_sdk.models.model_registry import ModelRegistry
+
+
+class MoonshotModelFactory(BaseModelFactory):
+    MODELS = ModelRegistry.get_by_family(LLMFamilies.MOONSHOT)
+
+    def supports_model(self, model: LLMModel) -> bool:
+        return model in MoonshotModelFactory.MODELS
+
+    def get_cache_control(self, config: LLMConfig) -> Optional[Dict[str, Any]]:
+        return None
+
+    def get_model_specific_config(self, config: LLMConfig) -> Dict[str, Any]:
+        model_config: Dict[str, Any] = {
+            "model": config.model.name,
+            "temperature": config.temperature,
+        }
+        if config.max_output_tokens is not None:
+            model_config["max_tokens"] = config.max_output_tokens
+        return model_config

spaik_sdk/models/factories/xai_factory.py

@@ -0,0 +1,26 @@
+from typing import Any, Dict, Optional
+
+from spaik_sdk.models.factories.base_model_factory import BaseModelFactory
+from spaik_sdk.models.llm_config import LLMConfig
+from spaik_sdk.models.llm_families import LLMFamilies
+from spaik_sdk.models.llm_model import LLMModel
+from spaik_sdk.models.model_registry import ModelRegistry
+
+
+class XAIModelFactory(BaseModelFactory):
+    MODELS = ModelRegistry.get_by_family(LLMFamilies.XAI)
+
+    def supports_model(self, model: LLMModel) -> bool:
+        return model in XAIModelFactory.MODELS
+
+    def get_cache_control(self, config: LLMConfig) -> Optional[Dict[str, Any]]:
+        return None
+
+    def get_model_specific_config(self, config: LLMConfig) -> Dict[str, Any]:
+        model_config: Dict[str, Any] = {
+            "model": config.model.name,
+            "temperature": config.temperature,
+        }
+        if config.max_output_tokens is not None:
+            model_config["max_tokens"] = config.max_output_tokens
+        return model_config

spaik_sdk/models/llm_config.py

@@ -16,7 +16,7 @@ class LLMConfig:
     streaming: bool = True
     reasoning_summary: str = "detailed"  # Options: "auto", "concise", "detailed", None
     reasoning_effort: str = "medium"  # Options: "low", "medium", "high"
-    max_output_tokens: int = 8192
+    max_output_tokens: Optional[int] = None
     reasoning_budget_tokens: int = 4096
     temperature: float = 0.1
     structured_response: bool = False

spaik_sdk/models/llm_families.py

@@ -5,3 +5,9 @@ class LLMFamilies:
     OPENAI = "openai"
     GOOGLE = "google"
     OLLAMA = "ollama"
+    DEEPSEEK = "deepseek"
+    MISTRAL = "mistral"
+    META = "meta"
+    COHERE = "cohere"
+    XAI = "xai"
+    MOONSHOT = "moonshot"

spaik_sdk/models/model_registry.py

@@ -22,17 +22,32 @@ class ModelRegistry:
 
     # OpenAI models
     GPT_4_1 = LLMModel(family=LLMFamilies.OPENAI, name="gpt-4.1", reasoning=False, prompt_caching=True)
+    GPT_4_1_MINI = LLMModel(family=LLMFamilies.OPENAI, name="gpt-4.1-mini", reasoning=False, prompt_caching=True)
+    GPT_4_1_NANO = LLMModel(family=LLMFamilies.OPENAI, name="gpt-4.1-nano", reasoning=False, prompt_caching=True)
     GPT_4O = LLMModel(family=LLMFamilies.OPENAI, name="gpt-4o", reasoning=False, prompt_caching=True)
+    GPT_4O_MINI = LLMModel(family=LLMFamilies.OPENAI, name="gpt-4o-mini", reasoning=False, prompt_caching=True)
+    O1 = LLMModel(family=LLMFamilies.OPENAI, name="o1")
+    O1_MINI = LLMModel(family=LLMFamilies.OPENAI, name="o1-mini")
+    O3 = LLMModel(family=LLMFamilies.OPENAI, name="o3")
+    O3_MINI = LLMModel(family=LLMFamilies.OPENAI, name="o3-mini")
+    O3_PRO = LLMModel(family=LLMFamilies.OPENAI, name="o3-pro")
     O4_MINI = LLMModel(family=LLMFamilies.OPENAI, name="o4-mini")
     O4_MINI_APRIL_2025 = LLMModel(family=LLMFamilies.OPENAI, name="o4-mini-2025-04-16")
+    CODEX_MINI = LLMModel(family=LLMFamilies.OPENAI, name="codex-mini")
     GPT_5 = LLMModel(family=LLMFamilies.OPENAI, name="gpt-5", reasoning=True, reasoning_min_effort="minimal", prompt_caching=True)
     GPT_5_MINI = LLMModel(family=LLMFamilies.OPENAI, name="gpt-5-mini", reasoning=True, reasoning_min_effort="minimal", prompt_caching=True)
     GPT_5_NANO = LLMModel(family=LLMFamilies.OPENAI, name="gpt-5-nano", reasoning=True, reasoning_min_effort="minimal", prompt_caching=True)
+    GPT_5_CHAT = LLMModel(family=LLMFamilies.OPENAI, name="gpt-5-chat", reasoning=False, prompt_caching=True)
+    GPT_5_CODEX = LLMModel(family=LLMFamilies.OPENAI, name="gpt-5-codex", reasoning=True, prompt_caching=True)
+    GPT_5_PRO = LLMModel(family=LLMFamilies.OPENAI, name="gpt-5-pro", reasoning=True, prompt_caching=True)
     GPT_5_1 = LLMModel(family=LLMFamilies.OPENAI, name="gpt-5.1", reasoning=True, prompt_caching=True)
+    GPT_5_1_CHAT = LLMModel(family=LLMFamilies.OPENAI, name="gpt-5.1-chat", reasoning=True, prompt_caching=True)
     GPT_5_1_CODEX = LLMModel(family=LLMFamilies.OPENAI, name="gpt-5.1-codex", reasoning=True, prompt_caching=True)
     GPT_5_1_CODEX_MINI = LLMModel(family=LLMFamilies.OPENAI, name="gpt-5.1-codex-mini", reasoning=True, prompt_caching=True)
     GPT_5_1_CODEX_MAX = LLMModel(family=LLMFamilies.OPENAI, name="gpt-5.1-codex-max", reasoning=True, prompt_caching=True)
     GPT_5_2 = LLMModel(family=LLMFamilies.OPENAI, name="gpt-5.2", reasoning=True, prompt_caching=True)
+    GPT_5_2_CHAT = LLMModel(family=LLMFamilies.OPENAI, name="gpt-5.2-chat", reasoning=False, prompt_caching=True)
+    GPT_5_2_CODEX = LLMModel(family=LLMFamilies.OPENAI, name="gpt-5.2-codex", reasoning=True, prompt_caching=True)
     GPT_5_2_PRO = LLMModel(family=LLMFamilies.OPENAI, name="gpt-5.2-pro", reasoning=True, prompt_caching=True)
 
     # Google models
@@ -43,6 +58,35 @@ class ModelRegistry:
     GEMINI_3_FLASH = LLMModel(family=LLMFamilies.GOOGLE, name="gemini-3-flash-preview", prompt_caching=True)
     GEMINI_3_PRO = LLMModel(family=LLMFamilies.GOOGLE, name="gemini-3-pro-preview", prompt_caching=True)
 
+    # DeepSeek models
+    DEEPSEEK_V3 = LLMModel(family=LLMFamilies.DEEPSEEK, name="DeepSeek-V3-0324")
+    DEEPSEEK_V3_1 = LLMModel(family=LLMFamilies.DEEPSEEK, name="DeepSeek-V3.1")
+    DEEPSEEK_V3_2 = LLMModel(family=LLMFamilies.DEEPSEEK, name="DeepSeek-V3.2")
+    DEEPSEEK_V3_2_SPECIALE = LLMModel(family=LLMFamilies.DEEPSEEK, name="DeepSeek-V3.2-Speciale")
+    DEEPSEEK_R1 = LLMModel(family=LLMFamilies.DEEPSEEK, name="DeepSeek-R1")
+    DEEPSEEK_R1_0528 = LLMModel(family=LLMFamilies.DEEPSEEK, name="DeepSeek-R1-0528")
+
+    # Mistral models
+    MISTRAL_LARGE_3 = LLMModel(family=LLMFamilies.MISTRAL, name="Mistral-Large-3", reasoning=False)
+
+    # Meta Llama models
+    LLAMA_4_MAVERICK = LLMModel(family=LLMFamilies.META, name="Llama-4-Maverick-17B-128E-Instruct-FP8", reasoning=False)
+    LLAMA_3_3_70B = LLMModel(family=LLMFamilies.META, name="Llama-3.3-70B-Instruct", reasoning=False)
+
+    # Cohere models
+    COHERE_COMMAND_A = LLMModel(family=LLMFamilies.COHERE, name="Cohere-command-a", reasoning=False)
+
+    # xAI Grok models
+    GROK_3 = LLMModel(family=LLMFamilies.XAI, name="grok-3")
+    GROK_3_MINI = LLMModel(family=LLMFamilies.XAI, name="grok-3-mini")
+    GROK_4 = LLMModel(family=LLMFamilies.XAI, name="grok-4")
+    GROK_4_FAST_REASONING = LLMModel(family=LLMFamilies.XAI, name="grok-4-fast-reasoning")
+    GROK_4_FAST_NON_REASONING = LLMModel(family=LLMFamilies.XAI, name="grok-4-fast-non-reasoning", reasoning=False)
+    GROK_CODE_FAST_1 = LLMModel(family=LLMFamilies.XAI, name="grok-code-fast-1")
+
+    # Moonshot AI models
+    KIMI_K2_THINKING = LLMModel(family=LLMFamilies.MOONSHOT, name="Kimi-K2-Thinking")
+
     # Registry for custom models
     _custom_models: Set[LLMModel] = set()
 
@@ -82,6 +126,7 @@ class ModelRegistry:
     def _get_aliases(cls) -> Dict[str, LLMModel]:
         """Get aliases mapping."""
         return {
+            # Claude aliases
             "sonnet": cls.CLAUDE_4_SONNET,
             "sonnet 3.7": cls.CLAUDE_3_7_SONNET,
             "sonnet 4.5": cls.CLAUDE_4_5_SONNET,
@@ -100,25 +145,73 @@ class ModelRegistry:
             "claude 4.5 sonnet": cls.CLAUDE_4_5_SONNET,
             "claude 4.5 haiku": cls.CLAUDE_4_5_HAIKU,
             "claude 4 opus": cls.CLAUDE_4_OPUS,
+            # OpenAI aliases
+            "o1": cls.O1,
+            "o1 mini": cls.O1_MINI,
+            "o3": cls.O3,
+            "o3 mini": cls.O3_MINI,
+            "o3 pro": cls.O3_PRO,
             "o4 mini": cls.O4_MINI,
             "o4 mini 2025-04-16": cls.O4_MINI_APRIL_2025,
+            "codex mini": cls.CODEX_MINI,
             "gpt 4.1": cls.GPT_4_1,
+            "gpt 4.1 mini": cls.GPT_4_1_MINI,
+            "gpt 4.1 nano": cls.GPT_4_1_NANO,
             "gpt 4o": cls.GPT_4O,
+            "gpt 4o mini": cls.GPT_4O_MINI,
             "gpt 5": cls.GPT_5,
             "gpt 5 mini": cls.GPT_5_MINI,
             "gpt 5 nano": cls.GPT_5_NANO,
+            "gpt 5 chat": cls.GPT_5_CHAT,
+            "gpt 5 codex": cls.GPT_5_CODEX,
+            "gpt 5 pro": cls.GPT_5_PRO,
             "gpt 5.1": cls.GPT_5_1,
+            "gpt 5.1 chat": cls.GPT_5_1_CHAT,
             "gpt 5.1 codex": cls.GPT_5_1_CODEX,
             "gpt 5.1 codex mini": cls.GPT_5_1_CODEX_MINI,
             "gpt 5.1 codex max": cls.GPT_5_1_CODEX_MAX,
             "gpt 5.2": cls.GPT_5_2,
+            "gpt 5.2 chat": cls.GPT_5_2_CHAT,
+            "gpt 5.2 codex": cls.GPT_5_2_CODEX,
             "gpt 5.2 pro": cls.GPT_5_2_PRO,
+            # Gemini aliases
             "gemini 2.5 flash": cls.GEMINI_2_5_FLASH,
             "gemini 2.5 pro": cls.GEMINI_2_5_PRO,
             "gemini 3 flash": cls.GEMINI_3_FLASH,
             "gemini 3.0 flash": cls.GEMINI_3_FLASH,
             "gemini 3 pro": cls.GEMINI_3_PRO,
             "gemini 3.0 pro": cls.GEMINI_3_PRO,
+            # DeepSeek aliases
+            "deepseek": cls.DEEPSEEK_V3_2,
+            "deepseek v3": cls.DEEPSEEK_V3,
+            "deepseek v3.1": cls.DEEPSEEK_V3_1,
+            "deepseek v3.2": cls.DEEPSEEK_V3_2,
+            "deepseek v3.2 speciale": cls.DEEPSEEK_V3_2_SPECIALE,
+            "deepseek r1": cls.DEEPSEEK_R1,
+            # Mistral aliases
+            "mistral": cls.MISTRAL_LARGE_3,
+            "mistral large": cls.MISTRAL_LARGE_3,
+            "mistral large 3": cls.MISTRAL_LARGE_3,
+            # Meta Llama aliases
+            "llama": cls.LLAMA_3_3_70B,
+            "llama 3.3": cls.LLAMA_3_3_70B,
+            "llama 3.3 70b": cls.LLAMA_3_3_70B,
+            "llama 4": cls.LLAMA_4_MAVERICK,
+            "llama 4 maverick": cls.LLAMA_4_MAVERICK,
+            # Cohere aliases
+            "cohere": cls.COHERE_COMMAND_A,
+            "cohere command": cls.COHERE_COMMAND_A,
+            "command a": cls.COHERE_COMMAND_A,
+            # xAI Grok aliases
+            "grok": cls.GROK_4,
+            "grok 3": cls.GROK_3,
+            "grok 3 mini": cls.GROK_3_MINI,
+            "grok 4": cls.GROK_4,
+            "grok 4 fast": cls.GROK_4_FAST_REASONING,
+            "grok code": cls.GROK_CODE_FAST_1,
+            # Moonshot aliases
+            "kimi": cls.KIMI_K2_THINKING,
+            "kimi k2": cls.KIMI_K2_THINKING,
         }
 
 

spaik_sdk/models/providers/azure_provider.py

@@ -1,31 +1,101 @@
-from typing import Any, Collection, Dict
+import os
+from typing import Any, Collection, Dict, Set
 
 from langchain_core.language_models.chat_models import BaseChatModel
 from langchain_openai import AzureChatOpenAI
 
-from spaik_sdk.config.env import env_config
-from spaik_sdk.models.factories.openai_factory import OpenAIModelFactory
 from spaik_sdk.models.llm_config import LLMConfig
 from spaik_sdk.models.llm_model import LLMModel
+from spaik_sdk.models.model_registry import ModelRegistry
 from spaik_sdk.models.providers.base_provider import BaseProvider
 
+# Model name -> Environment variable for Azure deployment name
+AZURE_DEPLOYMENT_ENV_VARS: Dict[str, str] = {
+    # OpenAI models
+    "gpt-4.1": "AZURE_GPT_4_1_DEPLOYMENT",
+    "gpt-4.1-mini": "AZURE_GPT_4_1_MINI_DEPLOYMENT",
+    "gpt-4.1-nano": "AZURE_GPT_4_1_NANO_DEPLOYMENT",
+    "gpt-4o": "AZURE_GPT_4O_DEPLOYMENT",
+    "gpt-4o-mini": "AZURE_GPT_4O_MINI_DEPLOYMENT",
+    "o1": "AZURE_O1_DEPLOYMENT",
+    "o1-mini": "AZURE_O1_MINI_DEPLOYMENT",
+    "o3": "AZURE_O3_DEPLOYMENT",
+    "o3-mini": "AZURE_O3_MINI_DEPLOYMENT",
+    "o3-pro": "AZURE_O3_PRO_DEPLOYMENT",
+    "o4-mini": "AZURE_O4_MINI_DEPLOYMENT",
+    "o4-mini-2025-04-16": "AZURE_O4_MINI_2025_04_16_DEPLOYMENT",
+    "codex-mini": "AZURE_CODEX_MINI_DEPLOYMENT",
+    "gpt-5": "AZURE_GPT_5_DEPLOYMENT",
+    "gpt-5-mini": "AZURE_GPT_5_MINI_DEPLOYMENT",
+    "gpt-5-nano": "AZURE_GPT_5_NANO_DEPLOYMENT",
+    "gpt-5-chat": "AZURE_GPT_5_CHAT_DEPLOYMENT",
+    "gpt-5-codex": "AZURE_GPT_5_CODEX_DEPLOYMENT",
+    "gpt-5-pro": "AZURE_GPT_5_PRO_DEPLOYMENT",
+    "gpt-5.1": "AZURE_GPT_5_1_DEPLOYMENT",
+    "gpt-5.1-chat": "AZURE_GPT_5_1_CHAT_DEPLOYMENT",
+    "gpt-5.1-codex": "AZURE_GPT_5_1_CODEX_DEPLOYMENT",
+    "gpt-5.1-codex-mini": "AZURE_GPT_5_1_CODEX_MINI_DEPLOYMENT",
+    "gpt-5.1-codex-max": "AZURE_GPT_5_1_CODEX_MAX_DEPLOYMENT",
+    "gpt-5.2": "AZURE_GPT_5_2_DEPLOYMENT",
+    "gpt-5.2-chat": "AZURE_GPT_5_2_CHAT_DEPLOYMENT",
+    "gpt-5.2-codex": "AZURE_GPT_5_2_CODEX_DEPLOYMENT",
+    "gpt-5.2-pro": "AZURE_GPT_5_2_PRO_DEPLOYMENT",
+    # DeepSeek models (Azure AI Foundry)
+    "DeepSeek-V3-0324": "AZURE_DEEPSEEK_V3_DEPLOYMENT",
+    "DeepSeek-V3.1": "AZURE_DEEPSEEK_V3_1_DEPLOYMENT",
+    "DeepSeek-V3.2": "AZURE_DEEPSEEK_V3_2_DEPLOYMENT",
+    "DeepSeek-V3.2-Speciale": "AZURE_DEEPSEEK_V3_2_SPECIALE_DEPLOYMENT",
+    "DeepSeek-R1": "AZURE_DEEPSEEK_R1_DEPLOYMENT",
+    "DeepSeek-R1-0528": "AZURE_DEEPSEEK_R1_0528_DEPLOYMENT",
+    # Mistral models (Azure AI Foundry)
+    "Mistral-Large-3": "AZURE_MISTRAL_LARGE_3_DEPLOYMENT",
+    # Meta Llama models (Azure AI Foundry)
+    "Llama-4-Maverick-17B-128E-Instruct-FP8": "AZURE_LLAMA_4_MAVERICK_DEPLOYMENT",
+    "Llama-3.3-70B-Instruct": "AZURE_LLAMA_3_3_70B_DEPLOYMENT",
+    # Cohere models (Azure AI Foundry)
+    "Cohere-command-a": "AZURE_COHERE_COMMAND_A_DEPLOYMENT",
+    # xAI Grok models (Azure AI Foundry)
+    "grok-3": "AZURE_GROK_3_DEPLOYMENT",
+    "grok-3-mini": "AZURE_GROK_3_MINI_DEPLOYMENT",
+    "grok-4": "AZURE_GROK_4_DEPLOYMENT",
+    "grok-4-fast-reasoning": "AZURE_GROK_4_FAST_REASONING_DEPLOYMENT",
+    "grok-4-fast-non-reasoning": "AZURE_GROK_4_FAST_NON_REASONING_DEPLOYMENT",
+    "grok-code-fast-1": "AZURE_GROK_CODE_FAST_1_DEPLOYMENT",
+    # Moonshot AI models (Azure AI Foundry)
+    "Kimi-K2-Thinking": "AZURE_KIMI_K2_THINKING_DEPLOYMENT",
+}
+
 
 class AzureProvider(BaseProvider):
     def get_supported_models(self) -> Collection[LLMModel]:
-        """Get list of models supported by Azure provider."""
-        return OpenAIModelFactory.MODELS
+        supported: Set[LLMModel] = set()
+        for model_name in AZURE_DEPLOYMENT_ENV_VARS.keys():
+            try:
+                model = ModelRegistry.from_name(model_name)
+                supported.add(model)
+            except ValueError:
+                pass
+        return supported
 
     def get_model_config(self, config: LLMConfig) -> Dict[str, Any]:
-        """Get Azure AI Foundry provider configuration (provider-specific only)."""
         return {
-            "api_key": env_config.get_azure_keys()["api_key"],
-            "api_version": env_config.get_azure_keys()["api_version"],
-            "azure_endpoint": env_config.get_azure_keys()["endpoint"],
+            "api_key": self._get_required_env("AZURE_API_KEY"),
+            "api_version": self._get_required_env("AZURE_API_VERSION"),
+            "azure_endpoint": self._get_required_env("AZURE_ENDPOINT"),
         }
 
     def create_langchain_model(self, config: LLMConfig, full_config: Dict[str, Any]) -> BaseChatModel:
-        """Create Azure langchain model with complete configuration and model-specific deployments."""
-        # Add Azure provider-specific deployment configuration
-        full_config["deployment_name"] = env_config.get_azure_keys()[f"{config.model.name.lower()}_deployment"]
-
+        full_config["deployment_name"] = self._get_deployment_name(config.model.name)
         return AzureChatOpenAI(**full_config)
+
+    def _get_deployment_name(self, model_name: str) -> str:
+        env_var = AZURE_DEPLOYMENT_ENV_VARS.get(model_name)
+        if not env_var:
+            raise ValueError(f"Model '{model_name}' not supported on Azure. Add it to AZURE_DEPLOYMENT_ENV_VARS.")
+        return os.environ.get(env_var, model_name)
+
+    def _get_required_env(self, key: str) -> str:
+        value = os.environ.get(key)
+        if not value:
+            raise ValueError(f"Environment variable {key} is required but not set")
+        return value

spaik_sdk/models/providers/base_provider.py

@@ -58,5 +58,21 @@ class BaseProvider(ABC):
             from spaik_sdk.models.providers.ollama_provider import OllamaProvider
 
             return OllamaProvider()
+        elif provider_type == ProviderType.DEEPSEEK:
+            from spaik_sdk.models.providers.deepseek_provider import DeepSeekProvider
+
+            return DeepSeekProvider()
+        elif provider_type == ProviderType.XAI:
+            from spaik_sdk.models.providers.xai_provider import XAIProvider
+
+            return XAIProvider()
+        elif provider_type == ProviderType.COHERE:
+            from spaik_sdk.models.providers.cohere_provider import CohereProvider
+
+            return CohereProvider()
+        elif provider_type == ProviderType.MISTRAL:
+            from spaik_sdk.models.providers.mistral_provider import MistralProvider
+
+            return MistralProvider()
         else:
             raise ValueError(f"Unsupported provider type: {provider_type}")

spaik_sdk/models/providers/cohere_provider.py

@@ -0,0 +1,23 @@
+from typing import Any, Collection, Dict
+
+from langchain_cohere import ChatCohere
+from langchain_core.language_models.chat_models import BaseChatModel
+
+from spaik_sdk.config.get_credentials_provider import credentials_provider
+from spaik_sdk.models.factories.cohere_factory import CohereModelFactory
+from spaik_sdk.models.llm_config import LLMConfig
+from spaik_sdk.models.llm_model import LLMModel
+from spaik_sdk.models.providers.base_provider import BaseProvider
+
+
+class CohereProvider(BaseProvider):
+    def get_supported_models(self) -> Collection[LLMModel]:
+        return CohereModelFactory.MODELS
+
+    def get_model_config(self, config: LLMConfig) -> Dict[str, Any]:
+        return {
+            "cohere_api_key": credentials_provider.get_provider_key("cohere"),
+        }
+
+    def create_langchain_model(self, config: LLMConfig, full_config: Dict[str, Any]) -> BaseChatModel:
+        return ChatCohere(**full_config)

spaik_sdk/models/providers/deepseek_provider.py

@@ -0,0 +1,23 @@
+from typing import Any, Collection, Dict
+
+from langchain_core.language_models.chat_models import BaseChatModel
+from langchain_deepseek import ChatDeepSeek
+
+from spaik_sdk.config.get_credentials_provider import credentials_provider
+from spaik_sdk.models.factories.deepseek_factory import DeepSeekModelFactory
+from spaik_sdk.models.llm_config import LLMConfig
+from spaik_sdk.models.llm_model import LLMModel
+from spaik_sdk.models.providers.base_provider import BaseProvider
+
+
+class DeepSeekProvider(BaseProvider):
+    def get_supported_models(self) -> Collection[LLMModel]:
+        return DeepSeekModelFactory.MODELS
+
+    def get_model_config(self, config: LLMConfig) -> Dict[str, Any]:
+        return {
+            "api_key": credentials_provider.get_provider_key("deepseek"),
+        }
+
+    def create_langchain_model(self, config: LLMConfig, full_config: Dict[str, Any]) -> BaseChatModel:
+        return ChatDeepSeek(**full_config)

spaik_sdk/models/providers/mistral_provider.py

@@ -0,0 +1,23 @@
+from typing import Any, Collection, Dict
+
+from langchain_core.language_models.chat_models import BaseChatModel
+from langchain_mistralai import ChatMistralAI
+
+from spaik_sdk.config.get_credentials_provider import credentials_provider
+from spaik_sdk.models.factories.mistral_factory import MistralModelFactory
+from spaik_sdk.models.llm_config import LLMConfig
+from spaik_sdk.models.llm_model import LLMModel
+from spaik_sdk.models.providers.base_provider import BaseProvider
+
+
+class MistralProvider(BaseProvider):
+    def get_supported_models(self) -> Collection[LLMModel]:
+        return MistralModelFactory.MODELS
+
+    def get_model_config(self, config: LLMConfig) -> Dict[str, Any]:
+        return {
+            "api_key": credentials_provider.get_provider_key("mistral"),
+        }
+
+    def create_langchain_model(self, config: LLMConfig, full_config: Dict[str, Any]) -> BaseChatModel:
+        return ChatMistralAI(**full_config)

spaik_sdk/models/providers/provider_type.py

@@ -9,6 +9,10 @@ class ProviderType(Enum):
     OPENAI_DIRECT = "openai"
     GOOGLE = "google"
     OLLAMA = "ollama"
+    DEEPSEEK = "deepseek"
+    XAI = "xai"
+    COHERE = "cohere"
+    MISTRAL = "mistral"
 
     @classmethod
     def from_name(cls, name: str) -> "ProviderType":
@@ -39,6 +43,14 @@ class ProviderType(Enum):
             return cls.GOOGLE
         elif family_lower == "ollama":
             return cls.OLLAMA
+        elif family_lower == "deepseek":
+            return cls.DEEPSEEK
+        elif family_lower == "xai":
+            return cls.XAI
+        elif family_lower == "cohere":
+            return cls.COHERE
+        elif family_lower == "mistral":
+            return cls.MISTRAL
         else:
             raise ValueError(f"Unknown model family: {family}")
 
@@ -51,6 +63,11 @@ PROVIDER_ALIASES = {
     "openai": ProviderType.OPENAI_DIRECT,
     "google": ProviderType.GOOGLE,
     "gemini": ProviderType.GOOGLE,
+    "deepseek": ProviderType.DEEPSEEK,
+    "xai": ProviderType.XAI,
+    "grok": ProviderType.XAI,
+    "cohere": ProviderType.COHERE,
+    "mistral": ProviderType.MISTRAL,
 }
 
 

spaik_sdk/models/providers/xai_provider.py

@@ -0,0 +1,23 @@
+from typing import Any, Collection, Dict
+
+from langchain_core.language_models.chat_models import BaseChatModel
+from langchain_xai import ChatXAI
+
+from spaik_sdk.config.get_credentials_provider import credentials_provider
+from spaik_sdk.models.factories.xai_factory import XAIModelFactory
+from spaik_sdk.models.llm_config import LLMConfig
+from spaik_sdk.models.llm_model import LLMModel
+from spaik_sdk.models.providers.base_provider import BaseProvider
+
+
+class XAIProvider(BaseProvider):
+    def get_supported_models(self) -> Collection[LLMModel]:
+        return XAIModelFactory.MODELS
+
+    def get_model_config(self, config: LLMConfig) -> Dict[str, Any]:
+        return {
+            "xai_api_key": credentials_provider.get_provider_key("xai"),
+        }
+
+    def create_langchain_model(self, config: LLMConfig, full_config: Dict[str, Any]) -> BaseChatModel:
+        return ChatXAI(**full_config)

spaik_sdk/server/api/routers/audio_router_factory.py

@@ -178,7 +178,7 @@ class AudioRouterFactory:
             try:
                 audio_bytes = await file.read()
                 filename = file.filename or "audio.webm"
-                
+
                 logger.info(f"STT request: language={language}, filename={filename}, size={len(audio_bytes)}")
 
                 options = STTOptions(

spaik_sdk/server/authorization/base_authorizer.py

@@ -54,7 +54,7 @@ class BaseAuthorizer(ABC, Generic[TUser]):
 
     async def can_read_file(self, user: TUser, file_metadata: "FileMetadata") -> bool:
         """Check if user has permission to read a file.
-        
+
         By default, users can read files they own, or files owned by 'system' (agent-generated).
         """
         return file_metadata.owner_id == user.get_id() or file_metadata.owner_id == "system"

spaik_sdk/tracing/agent_trace.py

@@ -73,6 +73,4 @@ class AgentTrace:
 
     def save(self, name: str) -> None:
         trace_content = self.to_string(include_system_prompt=False)
-        self._trace_sink.save_trace(
-            name, trace_content, self.system_prompt, self.agent_instance_id
-        )
+        self._trace_sink.save_trace(name, trace_content, self.system_prompt, self.agent_instance_id)

{spaik_sdk-0.6.4.dist-info → spaik_sdk-0.6.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: spaik-sdk
-Version: 0.6.4
+Version: 0.6.6
 Summary: Python SDK for building AI agents with multi-LLM support, streaming, and production-ready infrastructure
 Project-URL: Homepage, https://github.com/siilisolutions/spaik-sdk
 Project-URL: Repository, https://github.com/siilisolutions/spaik-sdk
@@ -24,12 +24,16 @@ Requires-Dist: dotenv>=0.9.9
 Requires-Dist: fastapi>=0.115.12
 Requires-Dist: httpx>=0.25.0
 Requires-Dist: langchain-anthropic>=1.3.0
+Requires-Dist: langchain-cohere>=0.5.0
 Requires-Dist: langchain-core>=1.2.0
+Requires-Dist: langchain-deepseek>=1.0.0
 Requires-Dist: langchain-google-genai>=4.0.0
 Requires-Dist: langchain-mcp-adapters>=0.2.1
+Requires-Dist: langchain-mistralai>=0.2.0
 Requires-Dist: langchain-ollama>=0.3.0
 Requires-Dist: langchain-openai>=1.1.0
 Requires-Dist: langchain-tavily>=0.2.15
+Requires-Dist: langchain-xai>=0.2.0
 Requires-Dist: langchain>=1.2.0
 Requires-Dist: langgraph>=1.0.0
 Requires-Dist: mcp>=1.9.2

{spaik_sdk-0.6.4.dist-info → spaik_sdk-0.6.6.dist-info}/RECORD

@@ -1,7 +1,7 @@
 spaik_sdk/__init__.py,sha256=UhJdqPEBVFTlyWHPicbcpcvOuOqmObenwnJv_GkPbVA,576
 spaik_sdk/py.typed,sha256=Nqnn8clbgv-5l0PgxcTOldg8mkMKrFn4TvPL-rYUUGg,1
 spaik_sdk/agent/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-spaik_sdk/agent/base_agent.py,sha256=sk1Q3ediUHm6hOVIIAv90XwXAL7pSrlLcDIm7qtG6z0,11037
+spaik_sdk/agent/base_agent.py,sha256=KuIsiLtZlthYNNimA5569BxgUvoMFuYcQZuzIBvYoMU,10920
 spaik_sdk/attachments/__init__.py,sha256=ckqaKkl8rCVg-V8hNkE_RG57peYkt1zMWTszSRgWZeE,678
 spaik_sdk/attachments/builder.py,sha256=WgB14KcZ491KqjY6QMeIYXS18KElqsnvO-XAc7wuP0s,1758
 spaik_sdk/attachments/file_storage_provider.py,sha256=3EKDCCfhKi2iDpLR3BMsKt9KCR8iFvUZz2LV8cMFs3s,692
@@ -21,7 +21,7 @@ spaik_sdk/audio/providers/google_tts.py,sha256=-7gohXszE_A3sFQbiMp3kk1VZBhGFyLkx
 spaik_sdk/audio/providers/openai_stt.py,sha256=ZD44obwqOTmvnmvPcD6RQu4cL6B4rWkMmSmLwlEeoQY,2078
 spaik_sdk/audio/providers/openai_tts.py,sha256=SHfav2hgPnk84Dy784XcJHAGQ7PgUcUuwvq2eU4ceW0,3436
 spaik_sdk/config/credentials_provider.py,sha256=mfaAUb8yRr9VEHqwxxIGlbXv-_v7ZnmaBxqDmq9SRys,299
-spaik_sdk/config/env.py,sha256=H4xdFhRqoE64Rj6J0DM_J_zaq10seh2WpDRldNmxaZI,2754
+spaik_sdk/config/env.py,sha256=xBEBq7yvOjxy59KhNhBIHGm0DwVelHC1LQeo4Pky4eQ,2208
 spaik_sdk/config/env_credentials_provider.py,sha256=Y4Tti-T3IAAFQDV2rTsBaa3mngZcJ6RBY6Pk1TMmJRM,307
 spaik_sdk/config/get_credentials_provider.py,sha256=D2EF3ezConXlCmNvl9mQTSoR3BsIlDmxKM0a7jgGym8,516
 spaik_sdk/image_gen/__init__.py,sha256=AEujgMjNTj0y0x5EPsbX8IV9MB_UXDRjf3nc3Sc7Ou4,245
@@ -53,25 +53,35 @@ spaik_sdk/llm/streaming/streaming_content_handler.py,sha256=-RLUEczCUKtzHRFatVgX
 spaik_sdk/llm/streaming/streaming_event_handler.py,sha256=6YX1xQnaDEfExQbIIxb_A-x-ETdowm1DeGg5BGngcs0,10629
 spaik_sdk/llm/streaming/streaming_state_manager.py,sha256=Nsfcf9umUk2WAVdsank3a_a0ihi4g94HoYA0lY1wcXc,2249
 spaik_sdk/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-spaik_sdk/models/llm_config.py,sha256=ICIY_y_-loD7X0h-0_sjJUufxPpZgrUKfEQ5uVxA740,1641
-spaik_sdk/models/llm_families.py,sha256=mtsMfih5FbkX52eFMkMSfjiSIpkroZ_uqe8JlreJscQ,169
+spaik_sdk/models/llm_config.py,sha256=eCp3I7Dbqm5Xoudak0IosEWfzW5a05xxK28zxNq8uZU,1651
+spaik_sdk/models/llm_families.py,sha256=oyjAVMWV6EePMFIYlDCRSGtYTdxqlrPRlun_-4v0IJw,301
 spaik_sdk/models/llm_model.py,sha256=DO8wlN4Gj_AB_lxTpqzQGDABvDgyOI3JcQda_J-UiKU,440
 spaik_sdk/models/llm_wrapper.py,sha256=CB07qSPJUWScN3hj1SO_9qi8QQ7Zg5p53JLnXFZ4O6A,929
-spaik_sdk/models/model_registry.py,sha256=bzMi-_amifpuE-YNkK2lAPDYOiAPX0Unz2xNzhATEQM,7914
+spaik_sdk/models/model_registry.py,sha256=U9hUcdeAjNpPAbOixq5aAoai_W7NK8AV7U8fhjs1pn0,12955
 spaik_sdk/models/factories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-spaik_sdk/models/factories/anthropic_factory.py,sha256=1Dmn-RBTmDHnXVCt2D3xx1Xw9OVMaecTk1kBFur8prs,1330
-spaik_sdk/models/factories/base_model_factory.py,sha256=iLTUhVh7G4l9TiQdFlq9HU-cs-t5uaul4SeoIdSFNww,2845
+spaik_sdk/models/factories/anthropic_factory.py,sha256=0bnA02fr2iQcwCUXGw_ZXgoogFtkpREJ_vGSak7F_uw,1380
+spaik_sdk/models/factories/base_model_factory.py,sha256=iZh9DLxIVgXDA6u5Hv4mh_zU0i_ed5ce2DJMWlj1dH0,3535
+spaik_sdk/models/factories/cohere_factory.py,sha256=Qn4UHV0jKZ8j1SYrj3ziNYRzRXp6UPnwkje-iR9PDHk,884
+spaik_sdk/models/factories/deepseek_factory.py,sha256=tq6HxumlVI67XJ4h_SFE9lApkVljEfBLCCHVzDPi_G4,1005
 spaik_sdk/models/factories/google_factory.py,sha256=5Xc-I6h_SwPJFaSVu0m87O2xcRgWns0OwURK_hFUpbg,1486
+spaik_sdk/models/factories/meta_factory.py,sha256=J0sdmRV726OIwCMTXy_L7jgb60u-wYzU1TomdLpPHis,993
+spaik_sdk/models/factories/mistral_factory.py,sha256=84J12FWWohGRWRpa57kTpCXaiQ680cqCuPjaP_piXyg,1002
+spaik_sdk/models/factories/moonshot_factory.py,sha256=3S-SSPFAkB6zf3XxOHxcYUdRRuAto0hqnwJLefEJ-QA,1005
 spaik_sdk/models/factories/ollama_factory.py,sha256=7RXPlbF7b-wJA45FluSGkVIzMXdHToxxGQ99KatiLfs,1674
 spaik_sdk/models/factories/openai_factory.py,sha256=LvKx5ueuL3uROfI453BQSq6vuLFHJwMzLVqIbeg4G9s,2489
+spaik_sdk/models/factories/xai_factory.py,sha256=chuxuemTPDxbip7zZUHBg7QHpdGALL31tP222qKOCzI,990
 spaik_sdk/models/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 spaik_sdk/models/providers/anthropic_provider.py,sha256=NY_K8cbph5PKMRdKWOY4iw2C6wUZtnZ4id8DGwuKv1M,1329
-spaik_sdk/models/providers/azure_provider.py,sha256=Ddzhmi1TCWleAMJx1qoagxxMEfv8Tot46h15eocS-Bg,1453
-spaik_sdk/models/providers/base_provider.py,sha256=RoAyUAVMxHIVrYXgz9mHX2hfCfMj1qI-B9Y8g2YmRVk,2392
+spaik_sdk/models/providers/azure_provider.py,sha256=5i74zs97gU-kWjls2PgKmnietoTrKhyTTOl_NxVEA2Y,4608
+spaik_sdk/models/providers/base_provider.py,sha256=qcbigiCMLS4OtRVXnzd-uAhktWWethiCmHfAMo3YC0k,3072
+spaik_sdk/models/providers/cohere_provider.py,sha256=vsXFWVXNiyTIxsJDigE7ar5s6hwUoq38JbIukDXBv3E,924
+spaik_sdk/models/providers/deepseek_provider.py,sha256=-B_qynQNZyV56CfXDd4J-vj_vKyaO8kNS3UZdUFqe4s,933
 spaik_sdk/models/providers/google_provider.py,sha256=HVl9tQaAiUX9orM9v5zwTKMhYYGrPLSPRVZ5hso8-_g,1180
+spaik_sdk/models/providers/mistral_provider.py,sha256=Dc4E-hzjdcSqaDcbthzDOBmouIS0LXL4a7YtVAAEC7U,931
 spaik_sdk/models/providers/ollama_provider.py,sha256=7BnAb5rdeTyc8EKy39OquuOyMexULC53FdxtvhHdHvY,1099
 spaik_sdk/models/providers/openai_provider.py,sha256=mEA1C5wGY01ZiDJyihkhn5eZB2rx9P_ebaSbreHtMqA,1127
-spaik_sdk/models/providers/provider_type.py,sha256=tMu_JwR1MwFDgG9jagQLYlzLLDTraGo8aSOSDK_Ny6o,3141
+spaik_sdk/models/providers/provider_type.py,sha256=Hj4-oCFC9uhzCAjEn2_8OKa0eoA8Jz_OLdGdODv1a9c,3675
+spaik_sdk/models/providers/xai_provider.py,sha256=MEZmkax38coElrj7foATzUO5WioMJBdwVvvPBID5RDU,897
 spaik_sdk/orchestration/__init__.py,sha256=SOsO7m686FX81698irjigoDWW2xhMSYvtPLufWdGlwM,582
 spaik_sdk/orchestration/base_orchestrator.py,sha256=ksPkia-WvLYD6byBy5PgmohdgHJr0IZeASQ2dgGbQUc,8705
 spaik_sdk/orchestration/checkpoint.py,sha256=ToiCtD5vP4FTslNyM2MkKMVjIG9YShor2RNPzVFdir0,2416
@@ -92,7 +102,7 @@ spaik_sdk/recording/impl/local_recorder.py,sha256=6v7ARykjseSskenvsaBU5UcgOJ15e8
 spaik_sdk/server/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 spaik_sdk/server/api/routers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 spaik_sdk/server/api/routers/api_builder.py,sha256=svuqLYJJWam7I5Aywg9ncHtB6YS8MStv28-UoLbMYX0,6655
-spaik_sdk/server/api/routers/audio_router_factory.py,sha256=x0W7reFAVerdUqUkKn_rc4WvVyIaMHckbDuESGUmOhY,7369
+spaik_sdk/server/api/routers/audio_router_factory.py,sha256=5MyUzWSPpYrQr7eAh9DczIs_FyQw-EKaAMWXUKM1Evw,7353
 spaik_sdk/server/api/routers/file_router_factory.py,sha256=0nT-L61AcCr6Z7jsXJgGPPDucho1MzoA8jDYbx5OQHs,4020
 spaik_sdk/server/api/routers/thread_router_factory.py,sha256=h8nCA0-a26WnKHOR9QvaBnQ6ezXn8sIsG21Vf1qfBB0,14183
 spaik_sdk/server/api/streaming/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -100,7 +110,7 @@ spaik_sdk/server/api/streaming/format_sse_event.py,sha256=EEBxbXL_Dbne30Lv31JeNp
 spaik_sdk/server/api/streaming/negotiate_streaming_response.py,sha256=InamT7fTY99tUkcuT-68CmH1JB9Nult0Jp4qzpAeKUk,149
 spaik_sdk/server/api/streaming/streaming_negotiator.py,sha256=xfjgm_GRfh0VV9Fw_5fimz3eyb5VMKSlzlAoVn4euf8,369
 spaik_sdk/server/authorization/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-spaik_sdk/server/authorization/base_authorizer.py,sha256=ry_r_uBMoJoI179Fk_THiqjTQovJW-Hw1VjIuDeQ2xw,2628
+spaik_sdk/server/authorization/base_authorizer.py,sha256=5geuXut1h3OArGWjJRpaknHmV_LBPZQUz3WSfwJ_F0M,2620
 spaik_sdk/server/authorization/base_user.py,sha256=x3bthycXLy_ws3xKH3ZSb_84ho1qymdlOQTWrfqqmXo,274
 spaik_sdk/server/authorization/dummy_authorizer.py,sha256=uaMW916GISBFopzydhmUIU7ClZKb8rS7wiL5LwrD-kM,573
 spaik_sdk/server/job_processor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -149,7 +159,7 @@ spaik_sdk/tools/impl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 spaik_sdk/tools/impl/mcp_tool_provider.py,sha256=g2ULJkTW7J3nyXI43RdrTj-AzFtMiGCP5Aog79vKnMI,2750
 spaik_sdk/tools/impl/search_tool_provider.py,sha256=fi8SBmvC7--n0cCNaTz6PhEe6Bf4RkyrxYkqKzEDlY4,515
 spaik_sdk/tracing/__init__.py,sha256=kmLFmy1Lb7bS2sryIFoFaknxYXlyuswuP_4gHmwEtv0,526
-spaik_sdk/tracing/agent_trace.py,sha256=ORxHCiilgElebx7O5CyZtwgeWTiruB7sFky98xgsf_M,3039
+spaik_sdk/tracing/agent_trace.py,sha256=9G2WHQOiVL_HOnlD8Qz2DHtd1Trldqg4jqcFDPGPf_w,3017
 spaik_sdk/tracing/get_trace_sink.py,sha256=ZPg8pVLS1BbY0PwXJXC-O8qSvUvQSzmDb4SWRtsRSSc,2573
 spaik_sdk/tracing/local_trace_sink.py,sha256=QTqkzDv8S0cLtRvScwPmejnj6EpccYaHFjd7KkP9Xrk,984
 spaik_sdk/tracing/noop_trace_sink.py,sha256=AxImIYh8MPzISTp6qDp8ShtWyPLVqLRwsh7yyAVDSjs,540
@@ -157,6 +167,6 @@ spaik_sdk/tracing/trace_sink.py,sha256=LU6aF848Kz2hMZuz0q6l-4IaD0sC-ex0AKFk8mVTS
 spaik_sdk/tracing/trace_sink_mode.py,sha256=74qiL4P3sNVGM3_DUkWKlqlu9UvT928NLKTskD_vxgk,791
 spaik_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 spaik_sdk/utils/init_logger.py,sha256=htxNtHMxRXVNAXBbS9f6Wmd0aET7kl3ClJ062b3YHmQ,791
-spaik_sdk-0.6.4.dist-info/METADATA,sha256=Si8vJmRo18czibAqovNT87HsVqPl9nxqz-ztBYG90DM,10085
-spaik_sdk-0.6.4.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-spaik_sdk-0.6.4.dist-info/RECORD,,
+spaik_sdk-0.6.6.dist-info/METADATA,sha256=yOAYCAPgoLuONrEw2lCgVHJbm40DaQpAxl0hHtkPjNA,10243
+spaik_sdk-0.6.6.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+spaik_sdk-0.6.6.dist-info/RECORD,,