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

fix: Update responses unit tests to reflect llama_stack_api structure changes

Browse source
This commit is contained in:
Mike Sager 2025-11-14 11:15:43 -05:00
parent ff60bb31e6
commit 62ea40cb85
6 changed files with 13 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

2
tests/unit/providers/agents/meta_reference/fixtures/__init__.py
View file

@ -8,7 +8,7 @@ import os
import yaml
from llama_stack.apis.inference import (
from llama_stack_api.inference import (
OpenAIChatCompletion,
)

8
tests/unit/providers/agents/meta_reference/test_openai_responses.py
View file

@ -15,8 +15,8 @@ from openai.types.chat.chat_completion_chunk import (
ChoiceDeltaToolCallFunction,
)
from llama_stack.apis.agents import Order
from llama_stack.apis.agents.openai_responses import (
from llama_stack_api.agents import Order
from llama_stack_api.openai_responses import (
ListOpenAIResponseInputItem,
OpenAIResponseInputMessageContentText,
OpenAIResponseInputToolFunction,
@ -31,7 +31,7 @@ from llama_stack.apis.agents.openai_responses import (
OpenAIResponseTextFormat,
WebSearchToolTypes,
)
from llama_stack.apis.inference import (
from llama_stack_api.inference import (
OpenAIAssistantMessageParam,
OpenAIChatCompletionContentPartTextParam,
OpenAIChatCompletionRequestWithExtraBody,
@ -41,7 +41,7 @@ from llama_stack.apis.inference import (
OpenAIResponseFormatJSONSchema,
OpenAIUserMessageParam,
)
from llama_stack.apis.tools.tools import ListToolDefsResponse, ToolDef, ToolGroups, ToolInvocationResult, ToolRuntime
from llama_stack_api.tools import ListToolDefsResponse, ToolDef, ToolGroups, ToolInvocationResult, ToolRuntime
from llama_stack.core.access_control.access_control import default_policy
from llama_stack.core.storage.datatypes import ResponsesStoreReference, SqliteSqlStoreConfig
from llama_stack.providers.inline.agents.meta_reference.responses.openai_responses import (

6
tests/unit/providers/agents/meta_reference/test_openai_responses_conversations.py
View file

@ -7,18 +7,18 @@
import pytest
from llama_stack.apis.agents.openai_responses import (
from llama_stack_api.openai_responses import (
OpenAIResponseMessage,
OpenAIResponseObject,
OpenAIResponseObjectStreamResponseCompleted,
OpenAIResponseObjectStreamResponseOutputItemDone,
OpenAIResponseOutputMessageContentOutputText,
)
from llama_stack.apis.common.errors import (
from llama_stack_api.common.errors import (
ConversationNotFoundError,
InvalidConversationIdError,
)
from llama_stack.apis.conversations.conversations import (
from llama_stack_api.conversations import (
ConversationItemList,
)

4
tests/unit/providers/agents/meta_reference/test_response_conversion_utils.py
View file

@ -7,7 +7,7 @@
import pytest
from llama_stack.apis.agents.openai_responses import (
from llama_stack_api.openai_responses import (
OpenAIResponseAnnotationFileCitation,
OpenAIResponseInputFunctionToolCallOutput,
OpenAIResponseInputMessageContentImage,
@ -20,7 +20,7 @@ from llama_stack.apis.agents.openai_responses import (
OpenAIResponseText,
OpenAIResponseTextFormat,
)
from llama_stack.apis.inference import (
from llama_stack_api.inference import (
OpenAIAssistantMessageParam,
OpenAIChatCompletionContentPartImageParam,
OpenAIChatCompletionContentPartTextParam,

2
tests/unit/providers/agents/meta_reference/test_response_tool_context.py
View file

@ -5,7 +5,7 @@
# the root directory of this source tree.
from llama_stack.apis.agents.openai_responses import (
from llama_stack_api.openai_responses import (
MCPListToolsTool,
OpenAIResponseInputToolFileSearch,
OpenAIResponseInputToolFunction,

4
tests/unit/providers/agents/meta_reference/test_responses_safety_utils.py
View file

@ -8,8 +8,8 @@ from unittest.mock import AsyncMock
import pytest
from llama_stack.apis.agents.agents import ResponseGuardrailSpec
from llama_stack.apis.safety import ModerationObject, ModerationObjectResults
from llama_stack_api.agents import ResponseGuardrailSpec
from llama_stack_api.safety import ModerationObject, ModerationObjectResults
from llama_stack.providers.inline.agents.meta_reference.responses.openai_responses import (
OpenAIResponsesImpl,
)