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

test: dont hardcode faiss as provider in the tests please

Browse source
This commit is contained in:
Ashwin Bharambe 2025-02-27 22:54:34 -08:00
parent 4c8a0fa8dc
commit ece354eedd
2 changed files with 0 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

1
tests/client-sdk/agents/test_agents.py
View file

@ -458,7 +458,6 @@ def test_rag_agent(llama_stack_client, agent_config, rag_tool_name):
vector_db_id=vector_db_id,
embedding_model="all-MiniLM-L6-v2",
embedding_dimension=384,
provider_id="faiss",
)
llama_stack_client.tool_runtime.rag_tool.insert(
documents=documents,

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

@ -24,7 +24,6 @@ def single_entry_vector_db_registry(llama_stack_client, empty_vector_db_registry
vector_db_id=vector_db_id,
embedding_model="all-MiniLM-L6-v2",
embedding_dimension=384,
provider_id="faiss",
)
vector_dbs = [vector_db.identifier for vector_db in llama_stack_client.vector_dbs.list()]
return vector_dbs
@ -121,7 +120,6 @@ def test_vector_db_insert_from_url_and_query(llama_stack_client, empty_vector_db
vector_db_id=vector_db_id,
embedding_model="all-MiniLM-L6-v2",
embedding_dimension=384,
provider_id="faiss",
)
# list to check memory bank is successfully registered