phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-03 18:00:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

imports

Browse source
This commit is contained in:
Ashwin Bharambe 2025-11-06 13:41:27 -08:00
parent 4343bc880c
commit 286a94ba8a
10 changed files with 10 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/integration/inference/test_provider_data_routing.py
View file

@ -16,7 +16,6 @@ from unittest.mock import AsyncMock, patch
import pytest import pytest
from llama_stack import LlamaStackAsLibraryClient
from llama_stack.apis.datatypes import Api from llama_stack.apis.datatypes import Api
from llama_stack.apis.inference.inference import ( from llama_stack.apis.inference.inference import (
OpenAIAssistantMessageParam, OpenAIAssistantMessageParam,
@ -24,6 +23,7 @@ from llama_stack.apis.inference.inference import (
OpenAIChatCompletionUsage, OpenAIChatCompletionUsage,
OpenAIChoice, OpenAIChoice,
) )
from llama_stack.core.library_client import LlamaStackAsLibraryClient
from llama_stack.core.telemetry.telemetry import MetricEvent from llama_stack.core.telemetry.telemetry import MetricEvent

2
tests/integration/inference/test_tools_with_schemas.py
View file

@ -13,7 +13,7 @@ import json
import pytest import pytest
from llama_stack import LlamaStackAsLibraryClient from llama_stack.core.library_client import LlamaStackAsLibraryClient
from llama_stack.models.llama.datatypes import ToolDefinition from llama_stack.models.llama.datatypes import ToolDefinition
from tests.common.mcp import make_mcp_server from tests.common.mcp import make_mcp_server

2
tests/integration/inspect/test_inspect.py
View file

@ -7,7 +7,7 @@
import pytest import pytest
from llama_stack_client import LlamaStackClient from llama_stack_client import LlamaStackClient
from llama_stack import LlamaStackAsLibraryClient from llama_stack.core.library_client import LlamaStackAsLibraryClient
class TestInspect: class TestInspect:

2
tests/integration/providers/test_providers.py
View file

@ -6,7 +6,7 @@
from llama_stack_client import LlamaStackClient from llama_stack_client import LlamaStackClient
from llama_stack import LlamaStackAsLibraryClient from llama_stack.core.library_client import LlamaStackAsLibraryClient
class TestProviders: class TestProviders:

2
tests/integration/responses/fixtures/fixtures.py
View file

@ -11,7 +11,7 @@ import pytest
import yaml import yaml
from openai import OpenAI from openai import OpenAI
from llama_stack import LlamaStackAsLibraryClient from llama_stack.core.library_client import LlamaStackAsLibraryClient
# --- Helper Functions --- # --- Helper Functions ---

2
tests/integration/responses/test_file_search.py
View file

@ -9,7 +9,7 @@ import time
import pytest import pytest
from llama_stack import LlamaStackAsLibraryClient from llama_stack.core.library_client import LlamaStackAsLibraryClient
from .helpers import new_vector_store, upload_file from .helpers import new_vector_store, upload_file

2
tests/integration/responses/test_tool_responses.py
View file

@ -12,8 +12,8 @@ import httpx
import openai import openai
import pytest import pytest
from llama_stack import LlamaStackAsLibraryClient
from llama_stack.core.datatypes import AuthenticationRequiredError 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 tests.common.mcp import dependency_tools, make_mcp_server
from .fixtures.test_cases import ( from .fixtures.test_cases import (

2
tests/integration/tool_runtime/test_mcp.py
View file

@ -10,7 +10,7 @@ import pytest
from llama_stack_client.lib.agents.agent import Agent from llama_stack_client.lib.agents.agent import Agent
from llama_stack_client.lib.agents.turn_events import StepCompleted, StepProgress, ToolCallIssuedDelta 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" AUTH_TOKEN = "test-token"

2
tests/integration/tool_runtime/test_mcp_json_schema.py
View file

@ -13,7 +13,7 @@ import json
import pytest import pytest
from llama_stack import LlamaStackAsLibraryClient from llama_stack.core.library_client import LlamaStackAsLibraryClient
from tests.common.mcp import make_mcp_server from tests.common.mcp import make_mcp_server
AUTH_TOKEN = "test-token" AUTH_TOKEN = "test-token"

2
tests/integration/tool_runtime/test_registration.py
View file

@ -8,8 +8,8 @@ import re
import pytest import pytest
from llama_stack import LlamaStackAsLibraryClient
from llama_stack.apis.common.errors import ToolGroupNotFoundError 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 from tests.common.mcp import MCP_TOOLGROUP_ID, make_mcp_server