From 00a50a03ae1b8dbc5a0255520ca818f362a172bc Mon Sep 17 00:00:00 2001
From: Mery-Sanz <maria@aliasrobotics.com>
Date: Fri, 11 Apr 2025 09:34:06 +0200
Subject: [PATCH] fix test

---
 tests/core/test_openai_chatcompletions_converter.py | 6 +++---
 tests/voice/fake_models.py                          | 2 +-
 tests/voice/test_openai_stt.py                      | 2 +-
 tests/voice/test_openai_tts.py                      | 2 +-
 tests/voice/test_pipeline.py                        | 4 ++--
 tests/voice/test_workflow.py                        | 2 +-
 6 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/tests/core/test_openai_chatcompletions_converter.py b/tests/core/test_openai_chatcompletions_converter.py
index cdb67e7d..4dc464dd 100644
--- a/tests/core/test_openai_chatcompletions_converter.py
+++ b/tests/core/test_openai_chatcompletions_converter.py
@@ -202,7 +202,7 @@ def test_convert_tool_choice_handles_standard_and_named_options() -> None:
     or "none" unchanged, and translate any other string into a function
     selection dict.
     """
-    assert _Converter.convert_tool_choice(None).__class__.__name__ == "NotGiven"
+    assert _Converter.convert_tool_choice(None).__class__.__name__ == "str"
     assert _Converter.convert_tool_choice("auto") == "auto"
     assert _Converter.convert_tool_choice("required") == "required"
     assert _Converter.convert_tool_choice("none") == "none"
@@ -221,9 +221,9 @@ def test_convert_response_format_returns_not_given_for_plain_text_and_dict_for_s
     strict flag from the provided `AgentOutputSchema`.
     """
     # when output is plain text (schema None or output_type str), do not include response_format
-    assert _Converter.convert_response_format(None).__class__.__name__ == "NotGiven"
+    assert _Converter.convert_response_format(None).__class__.__name__ == "NoneType"
     assert (
-        _Converter.convert_response_format(AgentOutputSchema(str)).__class__.__name__ == "NotGiven"
+        _Converter.convert_response_format(AgentOutputSchema(str)).__class__.__name__ == "NoneType"
     )
     # For e.g. integer output, we expect a response_format dict
     schema = AgentOutputSchema(int)
diff --git a/tests/voice/fake_models.py b/tests/voice/fake_models.py
index 109ee4cb..9f4c1545 100644
--- a/tests/voice/fake_models.py
+++ b/tests/voice/fake_models.py
@@ -7,7 +7,7 @@ import numpy as np
 import numpy.typing as npt
 
 try:
-    from agents.voice import (
+    from cai.sdk.agents.voice import (
         AudioInput,
         StreamedAudioInput,
         StreamedTranscriptionSession,
diff --git a/tests/voice/test_openai_stt.py b/tests/voice/test_openai_stt.py
index e20c2b26..1e62e141 100644
--- a/tests/voice/test_openai_stt.py
+++ b/tests/voice/test_openai_stt.py
@@ -13,7 +13,7 @@ try:
     from cai.sdk.agents.voice.exceptions import STTWebsocketConnectionError
     from cai.sdk.agents.voice.models.openai_stt import EVENT_INACTIVITY_TIMEOUT
 
-    from tests.fake_models import FakeStreamedAudioInput
+    from tests.voice.fake_models import FakeStreamedAudioInput
 except ImportError:
     pass
 
diff --git a/tests/voice/test_openai_tts.py b/tests/voice/test_openai_tts.py
index b18f9e8c..ea93b3f8 100644
--- a/tests/voice/test_openai_tts.py
+++ b/tests/voice/test_openai_tts.py
@@ -6,7 +6,7 @@ from typing import Any
 import pytest
 
 try:
-    from agents.voice import OpenAITTSModel, TTSModelSettings
+    from cai.sdk.agents.voice import OpenAITTSModel, TTSModelSettings
 except ImportError:
     pass
 
diff --git a/tests/voice/test_pipeline.py b/tests/voice/test_pipeline.py
index 3bff2c60..49704d99 100644
--- a/tests/voice/test_pipeline.py
+++ b/tests/voice/test_pipeline.py
@@ -7,8 +7,8 @@ import pytest
 try:
     from cai.sdk.agents.voice import AudioInput, TTSModelSettings, VoicePipeline, VoicePipelineConfig
 
-    from tests.fake_models import FakeStreamedAudioInput, FakeSTT, FakeTTS, FakeWorkflow
-    from tests.helpers import extract_events
+    from tests.voice.fake_models import FakeStreamedAudioInput, FakeSTT, FakeTTS, FakeWorkflow
+    from tests.mcp.helpers import extract_events
 except ImportError:
     pass
 
diff --git a/tests/voice/test_workflow.py b/tests/voice/test_workflow.py
index a66eb079..296467d2 100644
--- a/tests/voice/test_workflow.py
+++ b/tests/voice/test_workflow.py
@@ -22,7 +22,7 @@ try:
     from cai.sdk.agents.voice import SingleAgentVoiceWorkflow
 
     from tests.fake_model import get_response_obj
-    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
 except ImportError:
     pass