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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: spaik-sdk
-Version: 0.6.3
+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.3 → spaik_sdk-0.6.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "spaik-sdk"
-version = "0.6.3"
+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.3 → spaik_sdk-0.6.5}/spaik_sdk/agent/base_agent.py

@@ -1,4 +1,5 @@
 import asyncio
+import uuid
 from abc import ABC
 from typing import Any, AsyncGenerator, Dict, List, Optional, Type, TypeVar
 
@@ -16,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
@@ -57,9 +57,16 @@ class BaseAgent(ABC):
         cost_provider: Optional[CostProvider] = None,
     ):
         logger.debug("Initializing BaseAgent")
+        # Generate unique instance ID for trace correlation
+        self.agent_instance_id: str = str(uuid.uuid4())
         self.prompt_loader = prompt_loader or get_prompt_loader(prompt_loader_mode)
         self.system_prompt = system_prompt or self._get_system_prompt(system_prompt_args, system_prompt_version)
-        self.trace = trace or AgentTrace(self.system_prompt, self.__class__.__name__, trace_sink=trace_sink)
+        self.trace = trace or AgentTrace(
+            self.system_prompt,
+            self.__class__.__name__,
+            trace_sink=trace_sink,
+            agent_instance_id=self.agent_instance_id,
+        )
         self.thread_container = thread_container or ThreadContainer(self.system_prompt)
         self.tools = tools or self._create_tools(tool_providers)
         self.llm_config = llm_config or self.create_llm_config(llm_model, reasoning)
@@ -139,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.3 → spaik_sdk-0.6.5}/spaik_sdk/config/env.py

@@ -1,5 +1,5 @@
 import os
-from typing import Dict
+from typing import Optional as OptionalType
 
 from spaik_sdk.models.llm_model import LLMModel
 from spaik_sdk.models.model_registry import ModelRegistry
@@ -15,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"))
 
@@ -46,8 +36,16 @@ class EnvConfig:
     def get_prompt_loader_mode(self) -> PromptLoaderMode:
         return PromptLoaderMode.from_name(self.get_key("PROMPT_LOADER_MODE", "local"))
 
-    def get_trace_sink_mode(self) -> TraceSinkMode:
-        return TraceSinkMode.from_name(self.get_key("TRACE_SINK_MODE", "local"))
+    def get_trace_sink_mode(self) -> OptionalType[TraceSinkMode]:
+        """Get the trace sink mode from environment variable.
+
+        Returns:
+            TraceSinkMode.LOCAL if TRACE_SINK_MODE=local,
+            TraceSinkMode.NOOP if TRACE_SINK_MODE=noop,
+            None if TRACE_SINK_MODE is not set or empty (allows global/default behavior).
+        """
+        mode_str = self.get_key("TRACE_SINK_MODE", "", required=False)
+        return TraceSinkMode.from_name(mode_str)
 
     def get_credentials_provider_type(self) -> str:
         return self.get_key("CREDENTIALS_PROVIDER_TYPE", "env")

{spaik_sdk-0.6.3 → spaik_sdk-0.6.5}/spaik_sdk/models/factories/google_factory.py

@@ -22,6 +22,11 @@ class GoogleModelFactory(BaseModelFactory):
         if config.reasoning:
             model_config["thinking_budget"] = config.reasoning_budget_tokens
             model_config["include_thoughts"] = True
+        else:
+            # Gemini models have thinking enabled by default, so we must explicitly
+            # set thinking_budget=0 to disable it (omitting the parameter doesn't work)
+            model_config["thinking_budget"] = 0
+            model_config["include_thoughts"] = False
 
         # Handle streaming - Google models use disable_streaming instead of streaming
         if not config.streaming:

{spaik_sdk-0.6.3 → spaik_sdk-0.6.5}/spaik_sdk/models/factories/openai_factory.py

@@ -36,15 +36,21 @@ class OpenAIModelFactory(BaseModelFactory):
         if config.tool_usage:
             model_config["model_kwargs"] = {"parallel_tool_calls": True}
 
-        # Add model-specific configurations for reasoning models
+        # Handle reasoning configuration based on user preference (config.reasoning)
+        # and model capability (config.model.reasoning)
         if config.model.reasoning:
-            # Enable Responses API for reasoning models
+            # Model supports reasoning - check user preference
             model_config["use_responses_api"] = True
 
-            # Configure reasoning through model_kwargs as per LangChain docs
-            if config.reasoning_summary:
-                model_config["model_kwargs"] = {"reasoning": {"effort": config.reasoning_effort, "summary": config.reasoning_summary}}
+            if config.reasoning:
+                # User wants reasoning enabled - use configured effort
+                if config.reasoning_summary:
+                    model_config["model_kwargs"] = {"reasoning": {"effort": config.reasoning_effort, "summary": config.reasoning_summary}}
+            else:
+                # User wants reasoning disabled - use model's minimum effort level
+                model_config["model_kwargs"] = {"reasoning": {"effort": config.model.reasoning_min_effort}}
         else:
+            # Model doesn't support reasoning
             model_config["temperature"] = config.temperature
 
         return model_config

{spaik_sdk-0.6.3 → spaik_sdk-0.6.5}/spaik_sdk/models/llm_model.py

@@ -8,6 +8,7 @@ class LLMModel:
     family: str
     name: str
     reasoning: bool = True
+    reasoning_min_effort: str = "none"  # Minimum reasoning effort: 'none' or 'minimal'
     prompt_caching: bool = False
 
     def __str__(self) -> str:

{spaik_sdk-0.6.3 → spaik_sdk-0.6.5}/spaik_sdk/models/model_registry.py

@@ -25,9 +25,9 @@ class ModelRegistry:
     GPT_4O = LLMModel(family=LLMFamilies.OPENAI, name="gpt-4o", reasoning=False, prompt_caching=True)
     O4_MINI = LLMModel(family=LLMFamilies.OPENAI, name="o4-mini")
     O4_MINI_APRIL_2025 = LLMModel(family=LLMFamilies.OPENAI, name="o4-mini-2025-04-16")
-    GPT_5 = LLMModel(family=LLMFamilies.OPENAI, name="gpt-5", reasoning=True, prompt_caching=True)
-    GPT_5_MINI = LLMModel(family=LLMFamilies.OPENAI, name="gpt-5-mini", reasoning=True, prompt_caching=True)
-    GPT_5_NANO = LLMModel(family=LLMFamilies.OPENAI, name="gpt-5-nano", reasoning=True, prompt_caching=True)
+    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_1 = LLMModel(family=LLMFamilies.OPENAI, name="gpt-5.1", 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)

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.3 → 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.3 → 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.3 → spaik_sdk-0.6.5}/spaik_sdk/tracing/__init__.py

@@ -1,6 +1,7 @@
 from spaik_sdk.tracing.agent_trace import AgentTrace
-from spaik_sdk.tracing.get_trace_sink import get_trace_sink
+from spaik_sdk.tracing.get_trace_sink import configure_tracing, get_trace_sink
 from spaik_sdk.tracing.local_trace_sink import LocalTraceSink
+from spaik_sdk.tracing.noop_trace_sink import NoOpTraceSink
 from spaik_sdk.tracing.trace_sink import TraceSink
 from spaik_sdk.tracing.trace_sink_mode import TraceSinkMode
 
@@ -8,6 +9,8 @@ __all__ = [
     "AgentTrace",
     "TraceSink",
     "LocalTraceSink",
+    "NoOpTraceSink",
     "TraceSinkMode",
+    "configure_tracing",
     "get_trace_sink",
 ]

{spaik_sdk-0.6.3 → spaik_sdk-0.6.5}/spaik_sdk/tracing/agent_trace.py

@@ -1,5 +1,6 @@
 import json
 import time
+import uuid
 from typing import Optional, Type
 
 from pydantic import BaseModel
@@ -15,12 +16,15 @@ class AgentTrace:
         system_prompt: str,
         save_name: Optional[str] = None,
         trace_sink: Optional[TraceSink] = None,
+        agent_instance_id: Optional[str] = None,
     ):
         self.system_prompt: str = system_prompt
         self._start_time_monotonic: float = time.monotonic()
         self._steps: list[tuple[float, str]] = []
         self.save_name: Optional[str] = save_name
         self._trace_sink: TraceSink = trace_sink or get_trace_sink()
+        # Generate UUID if not provided (backward compatibility)
+        self.agent_instance_id: str = agent_instance_id or str(uuid.uuid4())
 
     def add_step(self, step_content: str) -> None:
         current_time_monotonic: float = time.monotonic()
@@ -69,4 +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._trace_sink.save_trace(name, trace_content, self.system_prompt, self.agent_instance_id)

spaik_sdk-0.6.5/spaik_sdk/tracing/get_trace_sink.py

@@ -0,0 +1,77 @@
+from typing import Optional
+
+from spaik_sdk.tracing.local_trace_sink import LocalTraceSink
+from spaik_sdk.tracing.noop_trace_sink import NoOpTraceSink
+from spaik_sdk.tracing.trace_sink import TraceSink
+from spaik_sdk.tracing.trace_sink_mode import TraceSinkMode
+
+# Module-level storage for globally configured trace sink
+_global_trace_sink: Optional[TraceSink] = None
+
+
+def configure_tracing(sink: Optional[TraceSink]) -> None:
+    """Configure the global trace sink used by all agents.
+
+    Call this once at application startup to set a custom trace sink
+    that will be used by all subsequently created agents.
+
+    Resolution order:
+    1. TRACE_SINK_MODE=local env var -> LocalTraceSink (escape hatch)
+    2. TRACE_SINK_MODE=noop env var -> NoOpTraceSink
+    3. Global sink set via this function -> the configured sink
+    4. No configuration -> NoOpTraceSink (silent default)
+
+    Args:
+        sink: The TraceSink to use globally, or None to clear the global
+              configuration (reverts to default no-op behavior).
+
+    Example:
+        from spaik_sdk.tracing import configure_tracing, LocalTraceSink
+
+        # At application startup
+        configure_tracing(LocalTraceSink(traces_dir="my_traces"))
+
+        # Or with a custom sink for observability
+        configure_tracing(MyDatadogTraceSink())
+
+        # Clear global config
+        configure_tracing(None)
+    """
+    global _global_trace_sink
+    _global_trace_sink = sink
+
+
+def get_trace_sink(mode: Optional[TraceSinkMode] = None) -> TraceSink:
+    """Get the appropriate trace sink based on configuration.
+
+    Resolution order:
+    1. If mode parameter is provided, use that mode
+    2. Check TRACE_SINK_MODE env var (LOCAL or NOOP override everything)
+    3. Check for globally configured sink via configure_tracing()
+    4. Default to NoOpTraceSink (silent no-op)
+
+    Args:
+        mode: Optional explicit mode to use (overrides all other config).
+
+    Returns:
+        The appropriate TraceSink instance.
+    """
+    # Lazy import to avoid circular dependency with env.py
+    from spaik_sdk.config.env import env_config
+
+    # If explicit mode parameter provided, use it
+    if mode is None:
+        mode = env_config.get_trace_sink_mode()
+
+    # Step 1-2: Check env var mode (LOCAL or NOOP)
+    if mode == TraceSinkMode.LOCAL:
+        return LocalTraceSink()
+    if mode == TraceSinkMode.NOOP:
+        return NoOpTraceSink()
+
+    # Step 3: Check global sink
+    if _global_trace_sink is not None:
+        return _global_trace_sink
+
+    # Step 4: Default to no-op
+    return NoOpTraceSink()

{spaik_sdk-0.6.3 → spaik_sdk-0.6.5}/spaik_sdk/tracing/local_trace_sink.py

@@ -10,7 +10,14 @@ class LocalTraceSink(TraceSink):
     def __init__(self, traces_dir: Optional[str] = None):
         self.traces_dir = traces_dir or "traces"
 
-    def save_trace(self, name: str, trace_content: str, system_prompt: str) -> None:
+    def save_trace(
+        self,
+        name: str,
+        trace_content: str,
+        system_prompt: str,
+        agent_instance_id: Optional[str] = None,
+    ) -> None:
+        # agent_instance_id is accepted but intentionally ignored - file naming is unchanged
         os.makedirs(self.traces_dir, exist_ok=True)
 
         trace_path = os.path.join(self.traces_dir, f"{name}.txt")

spaik_sdk-0.6.5/spaik_sdk/tracing/noop_trace_sink.py

@@ -0,0 +1,21 @@
+from typing import Optional
+
+from spaik_sdk.tracing.trace_sink import TraceSink
+
+
+class NoOpTraceSink(TraceSink):
+    """TraceSink implementation that does nothing.
+
+    Used as the default when no tracing is configured, ensuring traces
+    are silently discarded without any side effects.
+    """
+
+    def save_trace(
+        self,
+        name: str,
+        trace_content: str,
+        system_prompt: str,
+        agent_instance_id: Optional[str] = None,
+    ) -> None:
+        """Do nothing - silently discard the trace."""
+        pass

{spaik_sdk-0.6.3 → spaik_sdk-0.6.5}/spaik_sdk/tracing/trace_sink.py

@@ -1,4 +1,5 @@
 from abc import ABC, abstractmethod
+from typing import Optional
 
 
 class TraceSink(ABC):
@@ -8,12 +9,21 @@ class TraceSink(ABC):
     """
 
     @abstractmethod
-    def save_trace(self, name: str, trace_content: str, system_prompt: str) -> None:
+    def save_trace(
+        self,
+        name: str,
+        trace_content: str,
+        system_prompt: str,
+        agent_instance_id: Optional[str] = None,
+    ) -> None:
         """Save a trace with its system prompt.
 
         Args:
             name: Identifier for the trace (e.g., agent class name)
             trace_content: The formatted trace content (without system prompt)
             system_prompt: The system prompt used for the agent
+            agent_instance_id: Optional UUID identifying the agent instance for correlation.
+                              Custom sinks can use this to correlate traces in observability
+                              backends (e.g., Datadog, OpenTelemetry).
         """
         pass

spaik_sdk-0.6.5/spaik_sdk/tracing/trace_sink_mode.py

@@ -0,0 +1,28 @@
+from enum import Enum
+from typing import Optional
+
+
+class TraceSinkMode(Enum):
+    LOCAL = "local"
+    NOOP = "noop"
+
+    @classmethod
+    def from_name(cls, name: Optional[str]) -> Optional["TraceSinkMode"]:
+        """Convert a string name to a TraceSinkMode.
+
+        Args:
+            name: The mode name ("local", "noop") or None/empty string.
+
+        Returns:
+            The corresponding TraceSinkMode, or None if name is empty/None.
+            Returns None for unrecognized values (silent fallthrough to default behavior).
+        """
+        if not name:
+            return None
+
+        for mode in cls:
+            if mode.value == name.lower():
+                return mode
+
+        # Unknown values fall through to None (let get_trace_sink handle default)
+        return None

{spaik_sdk-0.6.3 → spaik_sdk-0.6.5}/tests/unit/spaik_sdk/agent/test_base_agent.py

@@ -1,4 +1,5 @@
 import time
+import uuid
 from typing import List, Optional
 
 import pytest
@@ -275,3 +276,62 @@ class TestBaseAgent:
         # Verify cost estimation works
         cost = BuiltinCostProvider().get_cost_estimate(ModelRegistry.CLAUDE_4_SONNET, total_consumption)
         assert cost.cost > 0
+
+
+@pytest.mark.unit
+class TestBaseAgentInstanceId:
+    """Tests for BaseAgent agent_instance_id."""
+
+    def test_base_agent_has_agent_instance_id(self):
+        """BaseAgent has an agent_instance_id attribute that is a UUID."""
+        agent = ConcreteTestAgent(
+            recording_name="test_instance_id_has_id",
+            llm_model=ModelRegistry.CLAUDE_4_SONNET,
+        )
+
+        assert hasattr(agent, "agent_instance_id")
+        assert agent.agent_instance_id is not None
+        assert isinstance(agent.agent_instance_id, str)
+
+        # Should be a valid UUID format
+        parsed_uuid = uuid.UUID(agent.agent_instance_id)
+        assert str(parsed_uuid) == agent.agent_instance_id
+
+    def test_two_agents_have_different_instance_ids(self):
+        """Two BaseAgent instances have different instance IDs."""
+        agent1 = ConcreteTestAgent(
+            recording_name="test_instance_id_unique_1",
+            llm_model=ModelRegistry.CLAUDE_4_SONNET,
+        )
+        agent2 = ConcreteTestAgent(
+            recording_name="test_instance_id_unique_2",
+            llm_model=ModelRegistry.CLAUDE_4_SONNET,
+        )
+
+        assert agent1.agent_instance_id != agent2.agent_instance_id
+
+    def test_same_agent_class_different_instance_ids(self):
+        """Two agents created from same class have different instance IDs."""
+        # Create two agents of the exact same type with same parameters
+        agent1 = ConcreteTestAgent(
+            recording_name="test_same_class_different_id_1",
+            system_prompt="Same prompt",
+            llm_model=ModelRegistry.CLAUDE_4_SONNET,
+        )
+        agent2 = ConcreteTestAgent(
+            recording_name="test_same_class_different_id_2",
+            system_prompt="Same prompt",
+            llm_model=ModelRegistry.CLAUDE_4_SONNET,
+        )
+
+        assert agent1.agent_instance_id != agent2.agent_instance_id
+
+    def test_agent_instance_id_passed_to_trace(self):
+        """The agent_instance_id is passed to AgentTrace during construction."""
+        agent = ConcreteTestAgent(
+            recording_name="test_instance_id_in_trace",
+            llm_model=ModelRegistry.CLAUDE_4_SONNET,
+        )
+
+        # The agent's instance ID should match the trace's instance ID
+        assert agent.agent_instance_id == agent.trace.agent_instance_id