RomanNum3ral
/
cai
mirror of https://github.com/aliasrobotics/cai.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix test

This commit is contained in:
Mery-Sanz 2025-04-11 09:34:06 +02:00
parent 58985f9d57
commit 00a50a03ae
6 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/core/test_openai_chatcompletions_converter.py
View File

@ -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 or "none" unchanged, and translate any other string into a function
selection dict. 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("auto") == "auto"
assert _Converter.convert_tool_choice("required") == "required" assert _Converter.convert_tool_choice("required") == "required"
assert _Converter.convert_tool_choice("none") == "none" 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`. strict flag from the provided `AgentOutputSchema`.
""" """
# when output is plain text (schema None or output_type str), do not include response_format # 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 ( 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 # For e.g. integer output, we expect a response_format dict
schema = AgentOutputSchema(int) schema = AgentOutputSchema(int)

2
tests/voice/fake_models.py
View File

@ -7,7 +7,7 @@ import numpy as np
import numpy.typing as npt import numpy.typing as npt
try: try:
from agents.voice import ( from cai.sdk.agents.voice import (
AudioInput, AudioInput,
StreamedAudioInput, StreamedAudioInput,
StreamedTranscriptionSession, StreamedTranscriptionSession,

2
tests/voice/test_openai_stt.py
View File

@ -13,7 +13,7 @@ try:
from cai.sdk.agents.voice.exceptions import STTWebsocketConnectionError from cai.sdk.agents.voice.exceptions import STTWebsocketConnectionError
from cai.sdk.agents.voice.models.openai_stt import EVENT_INACTIVITY_TIMEOUT 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: except ImportError:
pass pass

2
tests/voice/test_openai_tts.py
View File

@ -6,7 +6,7 @@ from typing import Any
import pytest import pytest
try: try:
from agents.voice import OpenAITTSModel, TTSModelSettings from cai.sdk.agents.voice import OpenAITTSModel, TTSModelSettings
except ImportError: except ImportError:
pass pass

4
tests/voice/test_pipeline.py
View File

@ -7,8 +7,8 @@ import pytest
try: try:
from cai.sdk.agents.voice import AudioInput, TTSModelSettings, VoicePipeline, VoicePipelineConfig from cai.sdk.agents.voice import AudioInput, TTSModelSettings, VoicePipeline, VoicePipelineConfig
from tests.fake_models import FakeStreamedAudioInput, FakeSTT, FakeTTS, FakeWorkflow from tests.voice.fake_models import FakeStreamedAudioInput, FakeSTT, FakeTTS, FakeWorkflow
from tests.helpers import extract_events from tests.mcp.helpers import extract_events
except ImportError: except ImportError:
pass pass

2
tests/voice/test_workflow.py
View File

@ -22,7 +22,7 @@ try:
from cai.sdk.agents.voice import SingleAgentVoiceWorkflow from cai.sdk.agents.voice import SingleAgentVoiceWorkflow
from tests.fake_model import get_response_obj 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: except ImportError:
pass pass