phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-31 16:01:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

uncomment

Browse source 
# What does this PR do?


## Test Plan
This commit is contained in:
Eric Huang 2025-05-27 14:48:56 -07:00
parent 484abe3116
commit 089a1578e9
1 changed files with 8 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

16
tests/integration/agents/test_openai_responses.py
View file

@ -41,7 +41,6 @@ def openai_client(client_with_models):
], ],
], ],
) )
@pytest.mark.skip(reason="Very flaky, sometimes there is a message not a function call, standard tool calling issues")
def test_responses_store(openai_client, client_with_models, text_model_id, stream, tools): def test_responses_store(openai_client, client_with_models, text_model_id, stream, tools):
if isinstance(client_with_models, LlamaStackAsLibraryClient): if isinstance(client_with_models, LlamaStackAsLibraryClient):
pytest.skip("OpenAI responses are not supported when testing with library client yet.") pytest.skip("OpenAI responses are not supported when testing with library client yet.")
@ -68,13 +67,15 @@ def test_responses_store(openai_client, client_with_models, text_model_id, strea
for chunk in response: for chunk in response:
if response_id is None: if response_id is None:
response_id = chunk.response.id response_id = chunk.response.id
if not tools: if chunk.type == "response.completed":
if chunk.type == "response.completed": response_id = chunk.response.id
response_id = chunk.response.id output_type = chunk.response.output[0].type
if output_type == "message":
content = chunk.response.output[0].content[0].text content = chunk.response.output[0].content[0].text
else: else:
response_id = response.id response_id = response.id
if not tools: output_type = response.output[0].type
if output_type == "message":
content = response.output[0].content[0].text content = response.output[0].content[0].text
# list responses - use the underlying HTTP client for endpoints not in SDK # list responses - use the underlying HTTP client for endpoints not in SDK
@ -87,9 +88,8 @@ def test_responses_store(openai_client, client_with_models, text_model_id, strea
retrieved_response = client.responses.retrieve(response_id) retrieved_response = client.responses.retrieve(response_id)
assert retrieved_response.id == response_id assert retrieved_response.id == response_id
assert retrieved_response.model == text_model_id assert retrieved_response.model == text_model_id
if tools: assert retrieved_response.output[0].type == output_type, retrieved_response
assert retrieved_response.output[0].type == "function_call" if output_type == "message":
else:
assert retrieved_response.output[0].content[0].text == content assert retrieved_response.output[0].content[0].text == content