phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

Fix precommit check after moving to ruff (#927)

Browse source 
Lint check in main branch is failing. This fixes the lint check after we
moved to ruff in https://github.com/meta-llama/llama-stack/pull/921. We
need to move to a `ruff.toml` file as well as fixing and ignoring some
additional checks.

Signed-off-by: Yuan Tang <terrytangyuan@gmail.com>
This commit is contained in:
Yuan Tang 2025-02-02 09:46:45 -05:00 • committed by GitHub
parent 4773092dd1
commit 34ab7a3b6c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
217 changed files with 981 additions and 2681 deletions
Download patch file Download diff file Expand all files Collapse all files

20
tests/client-sdk/tool_runtime/test_rag_tool.py
View file

@ -13,9 +13,7 @@ from llama_stack_client.types import Document
@pytest.fixture(scope="function")
def empty_vector_db_registry(llama_stack_client):
vector_dbs = [
vector_db.identifier for vector_db in llama_stack_client.vector_dbs.list()
]
vector_dbs = [vector_db.identifier for vector_db in llama_stack_client.vector_dbs.list()]
for vector_db_id in vector_dbs:
llama_stack_client.vector_dbs.unregister(vector_db_id=vector_db_id)
@ -29,9 +27,7 @@ def single_entry_vector_db_registry(llama_stack_client, empty_vector_db_registry
embedding_dimension=384,
provider_id="faiss",
)
vector_dbs = [
vector_db.identifier for vector_db in llama_stack_client.vector_dbs.list()
]
vector_dbs = [vector_db.identifier for vector_db in llama_stack_client.vector_dbs.list()]
return vector_dbs
@ -69,9 +65,7 @@ def assert_valid_response(response):
assert isinstance(chunk.content, str)
def test_vector_db_insert_inline_and_query(
llama_stack_client, single_entry_vector_db_registry, sample_documents
):
def test_vector_db_insert_inline_and_query(llama_stack_client, single_entry_vector_db_registry, sample_documents):
vector_db_id = single_entry_vector_db_registry[0]
llama_stack_client.tool_runtime.rag_tool.insert(
documents=sample_documents,
@ -118,9 +112,7 @@ def test_vector_db_insert_inline_and_query(
assert all(score >= 0.01 for score in response4.scores)
def test_vector_db_insert_from_url_and_query(
llama_stack_client, empty_vector_db_registry
):
def test_vector_db_insert_from_url_and_query(llama_stack_client, empty_vector_db_registry):
providers = [p for p in llama_stack_client.providers.list() if p.api == "vector_io"]
assert len(providers) > 0
@ -134,9 +126,7 @@ def test_vector_db_insert_from_url_and_query(
)
# list to check memory bank is successfully registered
available_vector_dbs = [
vector_db.identifier for vector_db in llama_stack_client.vector_dbs.list()
]
available_vector_dbs = [vector_db.identifier for vector_db in llama_stack_client.vector_dbs.list()]
assert vector_db_id in available_vector_dbs
# URLs of documents to insert