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
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 .test_responses import (

4
tests/test_guardrails.py
View File

@ -4,7 +4,7 @@ from typing import Any
import pytest
from agents import (
from cai.sdk.agents import (
Agent,
GuardrailFunctionOutput,
InputGuardrail,
@ -13,7 +13,7 @@ from agents import (
TResponseInputItem,
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):

2
tests/test_handoff_tool.py
View File

@ -4,7 +4,7 @@ import pytest
from openai.types.responses import ResponseOutputMessage, ResponseOutputText
from pydantic import BaseModel
from agents import (
from cai.sdk.agents import (
Agent,
Handoff,
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_param import ResponseReasoningItemParam
from agents import (
from cai.sdk.agents import (
Agent,
ItemHelpers,
MessageOutputItem,

2
tests/test_max_turns.py
View File

@ -5,7 +5,7 @@ import json
import pytest
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 .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,
)
from agents import (
from cai.sdk.agents import (
ModelResponse,
ModelSettings,
ModelTracing,
@ -30,7 +30,7 @@ from agents import (
OpenAIProvider,
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

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 agents.agent_output import AgentOutputSchema
from agents.exceptions import UserError
from agents.items import TResponseInputItem
from agents.models.fake_id import FAKE_RESPONSES_ID
from agents.models.openai_chatcompletions import _Converter
from cai.sdk.agents.agent_output import AgentOutputSchema
from cai.sdk.agents.exceptions import UserError
from cai.sdk.agents.items import TResponseInputItem
from cai.sdk.agents.models.fake_id import FAKE_RESPONSES_ID
from cai.sdk.agents.models.openai_chatcompletions import _Converter
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,
)
from agents.model_settings import ModelSettings
from agents.models.interface import ModelTracing
from agents.models.openai_chatcompletions import OpenAIChatCompletionsModel
from agents.models.openai_provider import OpenAIProvider
from cai.sdk.agents.model_settings import ModelSettings
from cai.sdk.agents.models.interface import ModelTracing
from cai.sdk.agents.models.openai_chatcompletions import OpenAIChatCompletionsModel
from cai.sdk.agents.models.openai_provider import OpenAIProvider
@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 pydantic import BaseModel
from agents import (
from cai.sdk.agents import (
Agent,
AgentOutputSchema,
Computer,
@ -40,7 +40,7 @@ from agents import (
function_tool,
handoff,
)
from agents.models.openai_responses import Converter
from cai.sdk.agents.models.openai_responses import Converter
def test_convert_tool_choice_standard_values():