fix some test

tests/agents/test_agent_hooks.py

@@ -13,8 +13,8 @@ from cai.sdk.agents.run import Runner
 from cai.sdk.agents.run_context import RunContextWrapper, TContext
 from cai.sdk.agents.tool import Tool
 
-from .fake_model import FakeModel
-from .test_responses import (
+from tests.fake_model import FakeModel
+from tests.core.test_responses import (
     get_final_output_message,
     get_function_tool,
     get_function_tool_call,

tests/agents/test_agent_one_tool.py

@@ -7,7 +7,7 @@ generic_linux_command tool.
 
 import pytest
 from tests.fake_model import FakeModel
-from tests.test_responses import (
+from tests.core.test_responses import (
     get_text_message,
     get_function_tool_call,
     get_function_tool,

tests/agents/test_agent_runner.py

@@ -27,7 +27,7 @@ from cai.sdk.agents.agent import ToolsToFinalOutputResult
 from cai.sdk.agents.tool import FunctionToolResult, function_tool
 
 from tests.fake_model import FakeModel
-from tests.test_responses import (
+from tests.core.test_responses import (
     get_final_output_message,
     get_function_tool,
     get_function_tool_call,

tests/agents/test_agent_runner_streamed.py

@@ -25,7 +25,7 @@ from cai.sdk.agents.run import RunConfig
 from cai.sdk.agents.stream_events import AgentUpdatedStreamEvent
 
 from tests.fake_model import FakeModel
-from tests.test_responses import (
+from tests.core.test_responses import (
     get_final_output_message,
     get_function_tool,
     get_function_tool_call,

tests/agents/test_global_hooks.py

@@ -9,8 +9,8 @@ from typing_extensions import TypedDict
 
 from cai.sdk.agents import Agent, RunContextWrapper, RunHooks, Runner, TContext, Tool
 
-from .fake_model import FakeModel
-from .test_responses import (
+from tests.fake_model import FakeModel
+from tests.core.test_responses import (
     get_final_output_message,
     get_function_tool,
     get_function_tool_call,

tests/agents/test_max_turns.py

@@ -7,8 +7,8 @@ from typing_extensions import TypedDict
 
 from cai.sdk.agents import Agent, MaxTurnsExceeded, Runner
 
-from .fake_model import FakeModel
-from .test_responses import get_function_tool, get_function_tool_call, get_text_message
+from tests.fake_model import FakeModel
+from tests.core.test_responses import get_function_tool, get_function_tool_call, get_text_message
 
 
 @pytest.mark.asyncio

tests/conftest.py

@@ -8,7 +8,7 @@ from cai.sdk.agents.models.openai_responses import OpenAIResponsesModel
 from cai.sdk.agents.tracing import set_trace_processors
 from cai.sdk.agents.tracing.setup import GLOBAL_TRACE_PROVIDER
 
-from .testing_processor import SPAN_PROCESSOR_TESTING
+from tests.testing_processor import SPAN_PROCESSOR_TESTING
 
 
 # This fixture will run once before any tests are executed

tests/mcp/test_caching.py

@@ -3,9 +3,9 @@ from unittest.mock import AsyncMock, patch
 import pytest
 from mcp.types import ListToolsResult, Tool as MCPTool
 
-from agents.mcp import MCPServerStdio
+from cai.sdk.agents.mcp import MCPServerStdio
 
-from .helpers import DummyStreamsContextManager, tee
+from tests.mcp.helpers import DummyStreamsContextManager, tee
 
 
 @pytest.mark.asyncio

tests/mcp/test_connect_disconnect.py

@@ -3,7 +3,7 @@ from unittest.mock import AsyncMock, patch
 import pytest
 from mcp.types import ListToolsResult, Tool as MCPTool
 
-from agents.mcp import MCPServerStdio
+from cai.sdk.agents.mcp import MCPServerStdio
 
 from .helpers import DummyStreamsContextManager, tee
 

tests/mcp/test_mcp_tracing.py

@@ -1,11 +1,11 @@
 import pytest
 from inline_snapshot import snapshot
 
-from agents import Agent, Runner
+from cai.sdk.agents import Agent, Runner
 
-from ..fake_model import FakeModel
-from ..test_responses import get_function_tool, get_function_tool_call, get_text_message
-from ..testing_processor import SPAN_PROCESSOR_TESTING, fetch_normalized_spans
+from tests.fake_model import FakeModel
+from tests.core.test_responses import get_function_tool, get_function_tool_call, get_text_message
+from tests.testing_processor import SPAN_PROCESSOR_TESTING, fetch_normalized_spans
 from .helpers import FakeMCPServer
 
 

tests/mcp/test_mcp_util.py

@@ -5,9 +5,9 @@ import pytest
 from mcp.types import Tool as MCPTool
 from pydantic import BaseModel
 
-from agents import FunctionTool, RunContextWrapper
-from agents.exceptions import AgentsException, ModelBehaviorError
-from agents.mcp import MCPServer, MCPUtil
+from cai.sdk.agents import FunctionTool, RunContextWrapper
+from cai.sdk.agents.exceptions import AgentsException, ModelBehaviorError
+from cai.sdk.agents.mcp import MCPServer, MCPUtil
 
 from .helpers import FakeMCPServer
 

tests/mcp/test_runner_calls_mcp.py

@@ -3,10 +3,10 @@ import json
 import pytest
 from pydantic import BaseModel
 
-from agents import Agent, ModelBehaviorError, Runner, UserError
+from cai.sdk.agents import Agent, ModelBehaviorError, Runner, UserError
 
-from ..fake_model import FakeModel
-from ..test_responses import get_function_tool_call, get_text_message
+from tests.fake_model import FakeModel
+from tests.core.test_responses import get_function_tool_call, get_text_message
 from .helpers import FakeMCPServer
 
 

tests/mcp/test_server_errors.py

@@ -1,7 +1,7 @@
 import pytest
 
-from agents.exceptions import UserError
-from agents.mcp.server import _MCPServerWithClientSession
+from cai.sdk.agents.exceptions import UserError
+from cai.sdk.agents.mcp.server import _MCPServerWithClientSession
 
 
 class CrashingClientSessionServer(_MCPServerWithClientSession):

tests/others/test_function_schema.py

@@ -6,9 +6,9 @@ import pytest
 from pydantic import BaseModel, ValidationError
 from typing_extensions import TypedDict
 
-from agents import RunContextWrapper
-from agents.exceptions import UserError
-from agents.function_schema import function_schema
+from cai.sdk.agents import RunContextWrapper
+from cai.sdk.agents.exceptions import UserError
+from cai.sdk.agents.function_schema import function_schema
 
 
 def no_args_function():

tests/others/test_pretty_print.py

@@ -9,7 +9,7 @@ from cai.sdk.agents.agent_output import _WRAPPER_DICT_KEY
 from cai.sdk.agents.util._pretty_print import pretty_print_result, pretty_print_run_result_streaming
 from tests.fake_model import FakeModel
 
-from .test_responses import get_final_output_message, get_text_message
+from tests.core.test_responses import get_final_output_message, get_text_message
 
 
 @pytest.mark.asyncio

tests/tools/test_tool_choice_reset.py

@@ -4,7 +4,7 @@ from cai.sdk.agents import Agent, ModelSettings, Runner
 from cai.sdk.agents._run_impl import AgentToolUseTracker, RunImpl
 
 from tests.fake_model import FakeModel
-from tests.test_responses import get_function_tool, get_function_tool_call, get_text_message
+from tests.core.test_responses import get_function_tool, get_function_tool_call, get_text_message
 
 
 class TestToolChoiceReset:

tests/tools/test_tool_use_behavior.py

@@ -18,7 +18,7 @@ from cai.sdk.agents import (
 )
 from cai.sdk.agents._run_impl import RunImpl
 
-from tests.test_responses import get_function_tool
+from tests.core.test_responses import get_function_tool
 
 
 def _make_function_tool_result(

tests/tracing/test_agent_tracing.py

@@ -8,7 +8,7 @@ from inline_snapshot import snapshot
 from cai.sdk.agents import Agent, RunConfig, Runner, trace
 
 from tests.fake_model import FakeModel
-from tests.test_responses import get_text_message
+from tests.core.test_responses import get_text_message
 from tests.testing_processor import assert_no_traces, fetch_normalized_spans
 
 

tests/tracing/test_responses_tracing.py

@@ -7,7 +7,7 @@ from cai.sdk.agents import ModelSettings, ModelTracing, OpenAIResponsesModel, tr
 from cai.sdk.agents.tracing.span_data import ResponseSpanData
 from tests import fake_model
 
-from .testing_processor import assert_no_spans, fetch_normalized_spans, fetch_ordered_spans
+from tests.testing_processor import assert_no_spans, fetch_normalized_spans, fetch_ordered_spans
 
 
 class DummyTracing:

tests/tracing/test_tracing_errors.py

@@ -20,7 +20,7 @@ from cai.sdk.agents import (
 )
 
 from tests.fake_model import FakeModel
-from tests.test_responses import (
+from tests.core.test_responses import (
     get_final_output_message,
     get_function_tool,
     get_function_tool_call,

tests/tracing/test_tracing_errors_streamed.py

@@ -23,7 +23,7 @@ from cai.sdk.agents import (
 )
 
 from tests.fake_model import FakeModel
-from tests.test_responses import (
+from tests.core.test_responses import (
     get_final_output_message,
     get_function_tool,
     get_function_tool_call,

tests/voice/test_input.py

@@ -5,9 +5,9 @@ import numpy as np
 import pytest
 
 try:
-    from agents import UserError
-    from agents.voice import AudioInput, StreamedAudioInput
-    from agents.voice.input import DEFAULT_SAMPLE_RATE, _buffer_to_audio_file
+    from cai.sdk.agents import UserError
+    from cai.sdk.agents.voice import AudioInput, StreamedAudioInput
+    from cai.sdk.agents.voice.input import DEFAULT_SAMPLE_RATE, _buffer_to_audio_file
 except ImportError:
     pass
 

tests/voice/test_openai_stt.py

@@ -9,11 +9,11 @@ import numpy as np
 import pytest
 
 try:
-    from agents.voice import OpenAISTTTranscriptionSession, StreamedAudioInput, STTModelSettings
-    from agents.voice.exceptions import STTWebsocketConnectionError
-    from agents.voice.models.openai_stt import EVENT_INACTIVITY_TIMEOUT
+    from cai.sdk.agents.voice import OpenAISTTTranscriptionSession, StreamedAudioInput, STTModelSettings
+    from cai.sdk.agents.voice.exceptions import STTWebsocketConnectionError
+    from cai.sdk.agents.voice.models.openai_stt import EVENT_INACTIVITY_TIMEOUT
 
-    from .fake_models import FakeStreamedAudioInput
+    from tests.fake_models import FakeStreamedAudioInput
 except ImportError:
     pass
 

tests/voice/test_pipeline.py

@@ -5,10 +5,10 @@ import numpy.typing as npt
 import pytest
 
 try:
-    from agents.voice import AudioInput, TTSModelSettings, VoicePipeline, VoicePipelineConfig
+    from cai.sdk.agents.voice import AudioInput, TTSModelSettings, VoicePipeline, VoicePipelineConfig
 
-    from .fake_models import FakeStreamedAudioInput, FakeSTT, FakeTTS, FakeWorkflow
-    from .helpers import extract_events
+    from tests.fake_models import FakeStreamedAudioInput, FakeSTT, FakeTTS, FakeWorkflow
+    from tests.helpers import extract_events
 except ImportError:
     pass
 

tests/voice/test_workflow.py

@@ -8,10 +8,10 @@ from inline_snapshot import snapshot
 from openai.types.responses import ResponseCompletedEvent
 from openai.types.responses.response_text_delta_event import ResponseTextDeltaEvent
 
-from agents import Agent, Model, ModelSettings, ModelTracing, Tool
-from agents.agent_output import AgentOutputSchema
-from agents.handoffs import Handoff
-from agents.items import (
+from cai.sdk.agents import Agent, Model, ModelSettings, ModelTracing, Tool
+from cai.sdk.agents.agent_output import AgentOutputSchema
+from cai.sdk.agents.handoffs import Handoff
+from cai.sdk.agents.items import (
     ModelResponse,
     TResponseInputItem,
     TResponseOutputItem,
@@ -19,10 +19,10 @@ from agents.items import (
 )
 
 try:
-    from agents.voice import SingleAgentVoiceWorkflow
+    from cai.sdk.agents.voice import SingleAgentVoiceWorkflow
 
-    from ..fake_model import get_response_obj
-    from ..test_responses import get_function_tool, get_function_tool_call, get_text_message
+    from tests.fake_model import get_response_obj
+    from tests.test_responses import get_function_tool, get_function_tool_call, get_text_message
 except ImportError:
     pass