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

fix more test

This commit is contained in:
Mery-Sanz 2025-04-09 16:07:58 +02:00
parent c083ecf69f
commit 73de470fb0
9 changed files with 19 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_global_hooks.py
View File

@ -7,7 +7,7 @@ from typing import Any
import pytest import pytest
from typing_extensions import TypedDict from typing_extensions import TypedDict
from agents import Agent, RunContextWrapper, RunHooks, Runner, TContext, Tool from cai.sdk.agents import Agent, RunContextWrapper, RunHooks, Runner, TContext, Tool
from .fake_model import FakeModel from .fake_model import FakeModel
from .test_responses import ( from .test_responses import (

4
tests/test_guardrails.py
View File

@ -4,7 +4,7 @@ from typing import Any
import pytest import pytest
from agents import ( from cai.sdk.agents import (
Agent, Agent,
GuardrailFunctionOutput, GuardrailFunctionOutput,
InputGuardrail, InputGuardrail,
@ -13,7 +13,7 @@ from agents import (
TResponseInputItem, TResponseInputItem,
UserError, UserError,
) )
from agents.guardrail import input_guardrail, output_guardrail from cai.sdk.agents.guardrail import input_guardrail, output_guardrail
def get_sync_guardrail(triggers: bool, output_info: Any | None = None): def get_sync_guardrail(triggers: bool, output_info: Any | None = None):

2
tests/test_handoff_tool.py
View File

@ -4,7 +4,7 @@ import pytest
from openai.types.responses import ResponseOutputMessage, ResponseOutputText from openai.types.responses import ResponseOutputMessage, ResponseOutputText
from pydantic import BaseModel from pydantic import BaseModel
from agents import ( from cai.sdk.agents import (
Agent, Agent,
Handoff, Handoff,
HandoffInputData, HandoffInputData,

2
tests/test_items_helpers.py
View File

@ -20,7 +20,7 @@ from openai.types.responses.response_output_text import ResponseOutputText
from openai.types.responses.response_reasoning_item import ResponseReasoningItem, Summary from openai.types.responses.response_reasoning_item import ResponseReasoningItem, Summary
from openai.types.responses.response_reasoning_item_param import ResponseReasoningItemParam from openai.types.responses.response_reasoning_item_param import ResponseReasoningItemParam
from agents import ( from cai.sdk.agents import (
Agent, Agent,
ItemHelpers, ItemHelpers,
MessageOutputItem, MessageOutputItem,

2
tests/test_max_turns.py
View File

@ -5,7 +5,7 @@ import json
import pytest import pytest
from typing_extensions import TypedDict from typing_extensions import TypedDict
from agents import Agent, MaxTurnsExceeded, Runner from cai.sdk.agents import Agent, MaxTurnsExceeded, Runner
from .fake_model import FakeModel from .fake_model import FakeModel
from .test_responses import get_function_tool, get_function_tool_call, get_text_message from .test_responses import get_function_tool, get_function_tool_call, get_text_message

4
tests/test_openai_chatcompletions.py
View File

@ -22,7 +22,7 @@ from openai.types.responses import (
ResponseOutputText, ResponseOutputText,
) )
from agents import ( from cai.sdk.agents import (
ModelResponse, ModelResponse,
ModelSettings, ModelSettings,
ModelTracing, ModelTracing,
@ -30,7 +30,7 @@ from agents import (
OpenAIProvider, OpenAIProvider,
generation_span, generation_span,
) )
from agents.models.fake_id import FAKE_RESPONSES_ID from cai.sdk.agents.models.fake_id import FAKE_RESPONSES_ID
@pytest.mark.allow_call_model_methods @pytest.mark.allow_call_model_methods

10
tests/test_openai_chatcompletions_converter.py
View File

@ -38,11 +38,11 @@ from openai.types.responses import (
) )
from openai.types.responses.response_input_item_param import FunctionCallOutput from openai.types.responses.response_input_item_param import FunctionCallOutput
from agents.agent_output import AgentOutputSchema from cai.sdk.agents.agent_output import AgentOutputSchema
from agents.exceptions import UserError from cai.sdk.agents.exceptions import UserError
from agents.items import TResponseInputItem from cai.sdk.agents.items import TResponseInputItem
from agents.models.fake_id import FAKE_RESPONSES_ID from cai.sdk.agents.models.fake_id import FAKE_RESPONSES_ID
from agents.models.openai_chatcompletions import _Converter from cai.sdk.agents.models.openai_chatcompletions import _Converter
def test_message_to_output_items_with_text_only(): def test_message_to_output_items_with_text_only():

8
tests/test_openai_chatcompletions_stream.py
View File

@ -17,10 +17,10 @@ from openai.types.responses import (
ResponseOutputText, ResponseOutputText,
) )
from agents.model_settings import ModelSettings from cai.sdk.agents.model_settings import ModelSettings
from agents.models.interface import ModelTracing from cai.sdk.agents.models.interface import ModelTracing
from agents.models.openai_chatcompletions import OpenAIChatCompletionsModel from cai.sdk.agents.models.openai_chatcompletions import OpenAIChatCompletionsModel
from agents.models.openai_provider import OpenAIProvider from cai.sdk.agents.models.openai_provider import OpenAIProvider
@pytest.mark.allow_call_model_methods @pytest.mark.allow_call_model_methods

4
tests/test_openai_responses_converter.py
View File

@ -27,7 +27,7 @@ import pytest
from openai import NOT_GIVEN from openai import NOT_GIVEN
from pydantic import BaseModel from pydantic import BaseModel
from agents import ( from cai.sdk.agents import (
Agent, Agent,
AgentOutputSchema, AgentOutputSchema,
Computer, Computer,
@ -40,7 +40,7 @@ from agents import (
function_tool, function_tool,
handoff, handoff,
) )
from agents.models.openai_responses import Converter from cai.sdk.agents.models.openai_responses import Converter
def test_convert_tool_choice_standard_values(): def test_convert_tool_choice_standard_values():