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

test: catch BadRequestError for non-library client

Browse source 
# What does this PR do?


## Test Plan
This commit is contained in:
Eric Huang 2025-05-16 12:19:59 -07:00
parent 7f1f21fd6c
commit 9a835c1452
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
tests/integration/tool_runtime/test_rag_tool.py
View file

@ -5,6 +5,7 @@
# the root directory of this source tree. # the root directory of this source tree.
import pytest import pytest
from llama_stack_client import BadRequestError
from llama_stack_client.types import Document from llama_stack_client.types import Document
@ -223,7 +224,7 @@ def test_rag_tool_insert_and_query(client_with_empty_registry, embedding_model_i
assert_valid_text_response(response_without_metadata) assert_valid_text_response(response_without_metadata)
assert not any("metadata:" in chunk.text.lower() for chunk in response_without_metadata.content) assert not any("metadata:" in chunk.text.lower() for chunk in response_without_metadata.content)
with pytest.raises(ValueError): with pytest.raises((ValueError, BadRequestError)):
client_with_empty_registry.tool_runtime.rag_tool.query( client_with_empty_registry.tool_runtime.rag_tool.query(
vector_db_ids=[vector_db_id], vector_db_ids=[vector_db_id],
content="What is the name of the method used for fine-tuning?", content="What is the name of the method used for fine-tuning?",