fix mcp test

tests/helpers.py

@@ -5,7 +5,7 @@ from typing import Any
 from mcp import Tool as MCPTool
 from mcp.types import CallToolResult, TextContent
 
-from agents.mcp import MCPServer
+from cai.sdk.agents.mcp import MCPServer
 
 tee = shutil.which("tee") or ""
 assert tee, "tee not found"

tests/mcp/helpers.py

@@ -0,0 +1,21 @@
+try:
+    from cai.sdk.agents.voice import StreamedAudioResult
+except ImportError:
+    pass
+
+
+async def extract_events(result: StreamedAudioResult) -> tuple[list[str], list[bytes]]:
+    """Collapse pipeline stream events to simple labels for ordering assertions."""
+    flattened: list[str] = []
+    audio_chunks: list[bytes] = []
+
+    async for ev in result.stream():
+        if ev.type == "voice_stream_event_audio":
+            if ev.data is not None:
+                audio_chunks.append(ev.data.tobytes())
+            flattened.append("audio")
+        elif ev.type == "voice_stream_event_lifecycle":
+            flattened.append(ev.event)
+        elif ev.type == "voice_stream_event_error":
+            flattened.append("error")
+    return flattened, audio_chunks

tests/mcp/test_caching.py

@@ -5,7 +5,7 @@ from mcp.types import ListToolsResult, Tool as MCPTool
 
 from cai.sdk.agents.mcp import MCPServerStdio
 
-from tests.mcp.helpers import DummyStreamsContextManager, tee
+from tests.helpers import DummyStreamsContextManager, tee
 
 
 @pytest.mark.asyncio

tests/mcp/test_connect_disconnect.py

@@ -5,7 +5,7 @@ from mcp.types import ListToolsResult, Tool as MCPTool
 
 from cai.sdk.agents.mcp import MCPServerStdio
 
-from .helpers import DummyStreamsContextManager, tee
+from tests.helpers import DummyStreamsContextManager, tee
 
 
 @pytest.mark.asyncio

tests/mcp/test_mcp_tracing.py

@@ -6,7 +6,7 @@ from cai.sdk.agents import Agent, Runner
 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
+from tests.helpers import FakeMCPServer
 
 
 @pytest.mark.asyncio

tests/mcp/test_mcp_util.py

@@ -9,7 +9,7 @@ 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
+from tests.helpers import FakeMCPServer
 
 
 class Foo(BaseModel):

tests/mcp/test_runner_calls_mcp.py

@@ -7,7 +7,7 @@ from cai.sdk.agents import Agent, ModelBehaviorError, Runner, UserError
 
 from tests.fake_model import FakeModel
 from tests.core.test_responses import get_function_tool_call, get_text_message
-from .helpers import FakeMCPServer
+from tests.helpers import FakeMCPServer
 
 
 @pytest.mark.asyncio

tests/voice/helpers.py

@@ -1,5 +1,5 @@
 try:
-    from agents.voice import StreamedAudioResult
+    from cai.sdk.agents.voice import StreamedAudioResult
 except ImportError:
     pass