From 73de470fb04baf04ddfa7cbbf8f9e85cf49c38b5 Mon Sep 17 00:00:00 2001 From: Mery-Sanz Date: Wed, 9 Apr 2025 16:07:58 +0200 Subject: [PATCH] fix more test --- tests/test_global_hooks.py | 2 +- tests/test_guardrails.py | 4 ++-- tests/test_handoff_tool.py | 2 +- tests/test_items_helpers.py | 2 +- tests/test_max_turns.py | 2 +- tests/test_openai_chatcompletions.py | 4 ++-- tests/test_openai_chatcompletions_converter.py | 10 +++++----- tests/test_openai_chatcompletions_stream.py | 8 ++++---- tests/test_openai_responses_converter.py | 4 ++-- 9 files changed, 19 insertions(+), 19 deletions(-) diff --git a/tests/test_global_hooks.py b/tests/test_global_hooks.py index 45854410..0cf12612 100644 --- a/tests/test_global_hooks.py +++ b/tests/test_global_hooks.py @@ -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 ( diff --git a/tests/test_guardrails.py b/tests/test_guardrails.py index c9f318c3..c1be6a4b 100644 --- a/tests/test_guardrails.py +++ b/tests/test_guardrails.py @@ -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): diff --git a/tests/test_handoff_tool.py b/tests/test_handoff_tool.py index a2a06208..54288163 100644 --- a/tests/test_handoff_tool.py +++ b/tests/test_handoff_tool.py @@ -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, diff --git a/tests/test_items_helpers.py b/tests/test_items_helpers.py index 90fe6475..eb4a2fd0 100644 --- a/tests/test_items_helpers.py +++ b/tests/test_items_helpers.py @@ -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, diff --git a/tests/test_max_turns.py b/tests/test_max_turns.py index f01bb18f..1a827590 100644 --- a/tests/test_max_turns.py +++ b/tests/test_max_turns.py @@ -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 diff --git a/tests/test_openai_chatcompletions.py b/tests/test_openai_chatcompletions.py index 9a53e2b7..c7f33b66 100644 --- a/tests/test_openai_chatcompletions.py +++ b/tests/test_openai_chatcompletions.py @@ -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 diff --git a/tests/test_openai_chatcompletions_converter.py b/tests/test_openai_chatcompletions_converter.py index 73acb8ab..cdb67e7d 100644 --- a/tests/test_openai_chatcompletions_converter.py +++ b/tests/test_openai_chatcompletions_converter.py @@ -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(): diff --git a/tests/test_openai_chatcompletions_stream.py b/tests/test_openai_chatcompletions_stream.py index 7add92a6..e2227b7e 100644 --- a/tests/test_openai_chatcompletions_stream.py +++ b/tests/test_openai_chatcompletions_stream.py @@ -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 diff --git a/tests/test_openai_responses_converter.py b/tests/test_openai_responses_converter.py index 34cbac5c..3da2a207 100644 --- a/tests/test_openai_responses_converter.py +++ b/tests/test_openai_responses_converter.py @@ -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():