diff --git a/src/app/endpoints/a2a.py b/src/app/endpoints/a2a.py index c87da2dcd..f183cbf4f 100644 --- a/src/app/endpoints/a2a.py +++ b/src/app/endpoints/a2a.py @@ -63,7 +63,7 @@ from utils.agents.query import map_agent_inference_error from utils.conversation_compaction import apply_compaction_blocking from utils.mcp_headers import McpHeaders, mcp_headers_dependency -from utils.pydantic_ai import build_agent +from utils.pydantic_ai_helpers import build_agent from utils.responses import prepare_responses_params from utils.suid import normalize_conversation_id from version import __version__ diff --git a/src/app/endpoints/tools.py b/src/app/endpoints/tools.py index 2e0f11a1f..f0cf99511 100644 --- a/src/app/endpoints/tools.py +++ b/src/app/endpoints/tools.py @@ -28,7 +28,7 @@ mcp_headers_dependency, ) from utils.mcp_oauth_probe import check_mcp_auth -from utils.pydantic_ai import get_agent_capability_tools +from utils.pydantic_ai_helpers import get_agent_capability_tools from utils.tool_formatter import format_tools_list logger = get_logger(__name__) diff --git a/src/utils/agents/query.py b/src/utils/agents/query.py index 4edf05d15..1fc2d0492 100644 --- a/src/utils/agents/query.py +++ b/src/utils/agents/query.py @@ -42,7 +42,7 @@ process_native_tool_result, ) from utils.conversations import append_turn_items_to_conversation -from utils.pydantic_ai import build_agent +from utils.pydantic_ai_helpers import build_agent from utils.query import ( extract_provider_and_model_from_model_id, handle_known_apistatus_errors, diff --git a/src/utils/agents/streaming.py b/src/utils/agents/streaming.py index b0ee7452e..1954d61c3 100644 --- a/src/utils/agents/streaming.py +++ b/src/utils/agents/streaming.py @@ -57,7 +57,7 @@ process_native_tool_result, ) from utils.conversations import append_turn_items_to_conversation -from utils.pydantic_ai import build_agent +from utils.pydantic_ai_helpers import build_agent from utils.query import consume_query_tokens, store_query_results from utils.quota_utils import get_available_quotas from utils.responses import ( diff --git a/src/utils/pydantic_ai.py b/src/utils/pydantic_ai_helpers.py similarity index 100% rename from src/utils/pydantic_ai.py rename to src/utils/pydantic_ai_helpers.py diff --git a/tests/unit/utils/test_pydantic_ai.py b/tests/unit/utils/test_pydantic_ai.py index b4089f9e3..cb4e528c1 100644 --- a/tests/unit/utils/test_pydantic_ai.py +++ b/tests/unit/utils/test_pydantic_ai.py @@ -10,7 +10,7 @@ from models.common.responses.responses_api_params import ResponsesApiParams from models.config import SkillsConfiguration -from utils.pydantic_ai import ( +from utils.pydantic_ai_helpers import ( _agent_capabilities, _skills_capability, build_agent,