spaik-sdk 0.6.4__py3-none-any.whl → 0.6.5__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/providers/azure_provider.py

@@ -1,31 +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.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.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]:
-        """Get list of models supported by Azure provider."""
         return OpenAIModelFactory.MODELS
 
     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/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.5.dist-info}/METADATA

@@ -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.dist-info → spaik_sdk-0.6.5.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
@@ -66,7 +66,7 @@ spaik_sdk/models/factories/ollama_factory.py,sha256=7RXPlbF7b-wJA45FluSGkVIzMXdH
 spaik_sdk/models/factories/openai_factory.py,sha256=LvKx5ueuL3uROfI453BQSq6vuLFHJwMzLVqIbeg4G9s,2489
 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/azure_provider.py,sha256=Myqt6el7uDg8yGtGkaNJJEM6Gl3-Sf6msS1fPPHit8U,2369
 spaik_sdk/models/providers/base_provider.py,sha256=RoAyUAVMxHIVrYXgz9mHX2hfCfMj1qI-B9Y8g2YmRVk,2392
 spaik_sdk/models/providers/google_provider.py,sha256=HVl9tQaAiUX9orM9v5zwTKMhYYGrPLSPRVZ5hso8-_g,1180
 spaik_sdk/models/providers/ollama_provider.py,sha256=7BnAb5rdeTyc8EKy39OquuOyMexULC53FdxtvhHdHvY,1099
@@ -92,7 +92,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 +100,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 +149,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 +157,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.5.dist-info/METADATA,sha256=xBhp5B6_3ia1AWf1sPVPztbCe_-9Ac7a7_DlPa3zeV4,10085
+spaik_sdk-0.6.5.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+spaik_sdk-0.6.5.dist-info/RECORD,,