spaik-sdk 0.6.4__tar.gz → 0.6.5__tar.gz

{spaik_sdk-0.6.4 → spaik_sdk-0.6.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: spaik-sdk
-Version: 0.6.4
+Version: 0.6.5
 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

{spaik_sdk-0.6.4 → spaik_sdk-0.6.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "spaik-sdk"
-version = "0.6.4"
+version = "0.6.5"
 description = "Python SDK for building AI agents with multi-LLM support, streaming, and production-ready infrastructure"
 readme = "README.md"
 license = "MIT"

{spaik_sdk-0.6.4 → spaik_sdk-0.6.5}/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-0.6.4 → spaik_sdk-0.6.5}/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-0.6.5/spaik_sdk/models/providers/azure_provider.py

@@ -0,0 +1,55 @@
+import os
+from typing import Any, Collection, Dict
+
+from langchain_core.language_models.chat_models import BaseChatModel
+from langchain_openai import AzureChatOpenAI
+
+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.providers.base_provider import BaseProvider
+
+# Model name -> Environment variable for Azure deployment name
+AZURE_DEPLOYMENT_ENV_VARS: Dict[str, str] = {
+    "gpt-4.1": "AZURE_GPT_4_1_DEPLOYMENT",
+    "gpt-4o": "AZURE_GPT_4O_DEPLOYMENT",
+    "o4-mini": "AZURE_O4_MINI_DEPLOYMENT",
+    "o4-mini-2025-04-16": "AZURE_O4_MINI_2025_04_16_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.1": "AZURE_GPT_5_1_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-pro": "AZURE_GPT_5_2_PRO_DEPLOYMENT",
+}
+
+
+class AzureProvider(BaseProvider):
+    def get_supported_models(self) -> Collection[LLMModel]:
+        return OpenAIModelFactory.MODELS
+
+    def get_model_config(self, config: LLMConfig) -> Dict[str, Any]:
+        return {
+            "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:
+        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-0.6.4 → spaik_sdk-0.6.5}/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-0.6.4 → spaik_sdk-0.6.5}/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-0.6.4 → spaik_sdk-0.6.5}/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)