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

formatting issues

Browse source
This commit is contained in:
Omar Abdelwahab 2025-11-06 12:43:24 -08:00
parent 5ce48d2c6a
commit d08c529ac0
1 changed files with 6 additions and 31 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -24,12 +24,7 @@ from .fixtures.test_cases import (
multi_turn_tool_execution_test_cases,
web_search_test_cases,
)
from .helpers import (
new_vector_store,
setup_mcp_tools,
upload_file,
wait_for_file_attachment,
)
from .helpers import new_vector_store, setup_mcp_tools, upload_file, wait_for_file_attachment
from .streaming_assertions import StreamingValidator
@ -53,12 +48,7 @@ def test_response_non_streaming_web_search(compat_client, text_model_id, case):
@pytest.mark.parametrize("case", file_search_test_cases)
def test_response_non_streaming_file_search(
compat_client,
text_model_id,
embedding_model_id,
embedding_dimension,
tmp_path,
case,
compat_client, text_model_id, embedding_model_id, embedding_dimension, tmp_path, case
):
if isinstance(compat_client, LlamaStackAsLibraryClient):
pytest.skip("Responses API file search is not yet supported in library client.")
@ -258,8 +248,7 @@ def test_response_non_streaming_mcp_tool(compat_client, text_model_id, case, cap
)
# Suppress expected auth error logs only for the failing auth attempt
with caplog.at_level(
logging.CRITICAL,
logger="llama_stack.providers.inline.agents.meta_reference.responses.streaming",
logging.CRITICAL, logger="llama_stack.providers.inline.agents.meta_reference.responses.streaming"
):
with pytest.raises(exc_type):
compat_client.responses.create(
@ -323,11 +312,7 @@ def test_response_sequential_mcp_tool(compat_client, text_model_id, case):
assert "boiling point" in text_content.lower()
response2 = compat_client.responses.create(
model=text_model_id,
input=case.input,
tools=tools,
stream=False,
previous_response_id=response.id,
model=text_model_id, input=case.input, tools=tools, stream=False, previous_response_id=response.id
)
assert len(response2.output) >= 1
@ -376,13 +361,7 @@ def test_response_mcp_tool_approval(compat_client, text_model_id, case, approve)
response = compat_client.responses.create(
previous_response_id=response.id,
model=text_model_id,
input=[
{
"type": "mcp_approval_response",
"approval_request_id": approval_request.id,
"approve": approve,
}
],
input=[{"type": "mcp_approval_response", "approval_request_id": approval_request.id, "approve": approve}],
tools=tools,
stream=False,
)
@ -459,11 +438,7 @@ def test_response_function_call_ordering_1(compat_client, text_model_id, case):
}
)
response = compat_client.responses.create(
model=text_model_id,
input=inputs,
tools=case.tools,
stream=False,
previous_response_id=response.id,
model=text_model_id, input=inputs, tools=case.tools, stream=False, previous_response_id=response.id
)
assert len(response.output) == 1