From 62ea40cb854b8793a02cec7656012b3a9c1b6414 Mon Sep 17 00:00:00 2001 From: Mike Sager Date: Fri, 14 Nov 2025 11:15:43 -0500 Subject: [PATCH] fix: Update responses unit tests to reflect llama_stack_api structure changes --- .../providers/agents/meta_reference/fixtures/__init__.py | 2 +- .../agents/meta_reference/test_openai_responses.py | 8 ++++---- .../meta_reference/test_openai_responses_conversations.py | 6 +++--- .../meta_reference/test_response_conversion_utils.py | 4 ++-- .../agents/meta_reference/test_response_tool_context.py | 2 +- .../agents/meta_reference/test_responses_safety_utils.py | 4 ++-- 6 files changed, 13 insertions(+), 13 deletions(-) diff --git a/tests/unit/providers/agents/meta_reference/fixtures/__init__.py b/tests/unit/providers/agents/meta_reference/fixtures/__init__.py index 2ebcd9970..130c46f6d 100644 --- a/tests/unit/providers/agents/meta_reference/fixtures/__init__.py +++ b/tests/unit/providers/agents/meta_reference/fixtures/__init__.py @@ -8,7 +8,7 @@ import os import yaml -from llama_stack.apis.inference import ( +from llama_stack_api.inference import ( OpenAIChatCompletion, ) diff --git a/tests/unit/providers/agents/meta_reference/test_openai_responses.py b/tests/unit/providers/agents/meta_reference/test_openai_responses.py index ba914d808..5f0cbab9c 100644 --- a/tests/unit/providers/agents/meta_reference/test_openai_responses.py +++ b/tests/unit/providers/agents/meta_reference/test_openai_responses.py @@ -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 ( diff --git a/tests/unit/providers/agents/meta_reference/test_openai_responses_conversations.py b/tests/unit/providers/agents/meta_reference/test_openai_responses_conversations.py index c2c113c1b..598346bf9 100644 --- a/tests/unit/providers/agents/meta_reference/test_openai_responses_conversations.py +++ b/tests/unit/providers/agents/meta_reference/test_openai_responses_conversations.py @@ -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, ) diff --git a/tests/unit/providers/agents/meta_reference/test_response_conversion_utils.py b/tests/unit/providers/agents/meta_reference/test_response_conversion_utils.py index 2698b88c8..50cd6ef1d 100644 --- a/tests/unit/providers/agents/meta_reference/test_response_conversion_utils.py +++ b/tests/unit/providers/agents/meta_reference/test_response_conversion_utils.py @@ -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, diff --git a/tests/unit/providers/agents/meta_reference/test_response_tool_context.py b/tests/unit/providers/agents/meta_reference/test_response_tool_context.py index e966ad41e..8d5312ec8 100644 --- a/tests/unit/providers/agents/meta_reference/test_response_tool_context.py +++ b/tests/unit/providers/agents/meta_reference/test_response_tool_context.py @@ -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, diff --git a/tests/unit/providers/agents/meta_reference/test_responses_safety_utils.py b/tests/unit/providers/agents/meta_reference/test_responses_safety_utils.py index 9c5cc853c..f3404f6cb 100644 --- a/tests/unit/providers/agents/meta_reference/test_responses_safety_utils.py +++ b/tests/unit/providers/agents/meta_reference/test_responses_safety_utils.py @@ -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, )