From 286a94ba8a99ac5e110525b3230cec0d4a018be9 Mon Sep 17 00:00:00 2001 From: Ashwin Bharambe Date: Thu, 6 Nov 2025 13:41:27 -0800 Subject: [PATCH] imports --- tests/integration/inference/test_provider_data_routing.py | 2 +- tests/integration/inference/test_tools_with_schemas.py | 2 +- tests/integration/inspect/test_inspect.py | 2 +- tests/integration/providers/test_providers.py | 2 +- tests/integration/responses/fixtures/fixtures.py | 2 +- tests/integration/responses/test_file_search.py | 2 +- tests/integration/responses/test_tool_responses.py | 2 +- tests/integration/tool_runtime/test_mcp.py | 2 +- tests/integration/tool_runtime/test_mcp_json_schema.py | 2 +- tests/integration/tool_runtime/test_registration.py | 2 +- 10 files changed, 10 insertions(+), 10 deletions(-) diff --git a/tests/integration/inference/test_provider_data_routing.py b/tests/integration/inference/test_provider_data_routing.py index 34ee2672f..99aa75395 100644 --- a/tests/integration/inference/test_provider_data_routing.py +++ b/tests/integration/inference/test_provider_data_routing.py @@ -16,7 +16,6 @@ from unittest.mock import AsyncMock, patch import pytest -from llama_stack import LlamaStackAsLibraryClient from llama_stack.apis.datatypes import Api from llama_stack.apis.inference.inference import ( OpenAIAssistantMessageParam, @@ -24,6 +23,7 @@ from llama_stack.apis.inference.inference import ( OpenAIChatCompletionUsage, OpenAIChoice, ) +from llama_stack.core.library_client import LlamaStackAsLibraryClient from llama_stack.core.telemetry.telemetry import MetricEvent diff --git a/tests/integration/inference/test_tools_with_schemas.py b/tests/integration/inference/test_tools_with_schemas.py index b144a5196..f30e9ece5 100644 --- a/tests/integration/inference/test_tools_with_schemas.py +++ b/tests/integration/inference/test_tools_with_schemas.py @@ -13,7 +13,7 @@ import json import pytest -from llama_stack import LlamaStackAsLibraryClient +from llama_stack.core.library_client import LlamaStackAsLibraryClient from llama_stack.models.llama.datatypes import ToolDefinition from tests.common.mcp import make_mcp_server diff --git a/tests/integration/inspect/test_inspect.py b/tests/integration/inspect/test_inspect.py index 8c62c85a1..3a62068c9 100644 --- a/tests/integration/inspect/test_inspect.py +++ b/tests/integration/inspect/test_inspect.py @@ -7,7 +7,7 @@ import pytest from llama_stack_client import LlamaStackClient -from llama_stack import LlamaStackAsLibraryClient +from llama_stack.core.library_client import LlamaStackAsLibraryClient class TestInspect: diff --git a/tests/integration/providers/test_providers.py b/tests/integration/providers/test_providers.py index fc65e2a10..959abf195 100644 --- a/tests/integration/providers/test_providers.py +++ b/tests/integration/providers/test_providers.py @@ -6,7 +6,7 @@ from llama_stack_client import LlamaStackClient -from llama_stack import LlamaStackAsLibraryClient +from llama_stack.core.library_client import LlamaStackAsLibraryClient class TestProviders: diff --git a/tests/integration/responses/fixtures/fixtures.py b/tests/integration/responses/fixtures/fixtures.py index 1783a5622..dbf67e138 100644 --- a/tests/integration/responses/fixtures/fixtures.py +++ b/tests/integration/responses/fixtures/fixtures.py @@ -11,7 +11,7 @@ import pytest import yaml from openai import OpenAI -from llama_stack import LlamaStackAsLibraryClient +from llama_stack.core.library_client import LlamaStackAsLibraryClient # --- Helper Functions --- diff --git a/tests/integration/responses/test_file_search.py b/tests/integration/responses/test_file_search.py index e8d4fe359..dde5fd7f6 100644 --- a/tests/integration/responses/test_file_search.py +++ b/tests/integration/responses/test_file_search.py @@ -9,7 +9,7 @@ import time import pytest -from llama_stack import LlamaStackAsLibraryClient +from llama_stack.core.library_client import LlamaStackAsLibraryClient from .helpers import new_vector_store, upload_file diff --git a/tests/integration/responses/test_tool_responses.py b/tests/integration/responses/test_tool_responses.py index 3f1c35214..9bf58c6ff 100644 --- a/tests/integration/responses/test_tool_responses.py +++ b/tests/integration/responses/test_tool_responses.py @@ -12,8 +12,8 @@ import httpx import openai import pytest -from llama_stack import LlamaStackAsLibraryClient from llama_stack.core.datatypes import AuthenticationRequiredError +from llama_stack.core.library_client import LlamaStackAsLibraryClient from tests.common.mcp import dependency_tools, make_mcp_server from .fixtures.test_cases import ( diff --git a/tests/integration/tool_runtime/test_mcp.py b/tests/integration/tool_runtime/test_mcp.py index 59f558d2c..3a8fde37f 100644 --- a/tests/integration/tool_runtime/test_mcp.py +++ b/tests/integration/tool_runtime/test_mcp.py @@ -10,7 +10,7 @@ import pytest from llama_stack_client.lib.agents.agent import Agent from llama_stack_client.lib.agents.turn_events import StepCompleted, StepProgress, ToolCallIssuedDelta -from llama_stack import LlamaStackAsLibraryClient +from llama_stack.core.library_client import LlamaStackAsLibraryClient AUTH_TOKEN = "test-token" diff --git a/tests/integration/tool_runtime/test_mcp_json_schema.py b/tests/integration/tool_runtime/test_mcp_json_schema.py index 240ec403a..def0b27b8 100644 --- a/tests/integration/tool_runtime/test_mcp_json_schema.py +++ b/tests/integration/tool_runtime/test_mcp_json_schema.py @@ -13,7 +13,7 @@ import json import pytest -from llama_stack import LlamaStackAsLibraryClient +from llama_stack.core.library_client import LlamaStackAsLibraryClient from tests.common.mcp import make_mcp_server AUTH_TOKEN = "test-token" diff --git a/tests/integration/tool_runtime/test_registration.py b/tests/integration/tool_runtime/test_registration.py index c8c9cd046..4d532ed87 100644 --- a/tests/integration/tool_runtime/test_registration.py +++ b/tests/integration/tool_runtime/test_registration.py @@ -8,8 +8,8 @@ import re import pytest -from llama_stack import LlamaStackAsLibraryClient from llama_stack.apis.common.errors import ToolGroupNotFoundError +from llama_stack.core.library_client import LlamaStackAsLibraryClient from tests.common.mcp import MCP_TOOLGROUP_ID, make_mcp_server