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

fix rag test

Browse source
This commit is contained in:
Xi Yan 2025-02-15 14:56:14 -08:00
parent 532f607941
commit 0293b18f55
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -484,10 +484,12 @@ def test_rag_and_code_agent(llama_stack_client, agent_config):
for i, url in enumerate(urls)
]
vector_db_id = f"test-vector-db-{uuid4()}"
embdding_models = [x for x in llama_stack_client.models.list() if x.model_type == "embedding"]
embedding_model = embdding_models[0]
llama_stack_client.vector_dbs.register(
vector_db_id=vector_db_id,
embedding_model="all-MiniLM-L6-v2",
embedding_dimension=384,
embedding_model=embedding_model.identifier,
embedding_dimension=embedding_model.metadata["embedding_dimension"],
)
llama_stack_client.tool_runtime.rag_tool.insert(
documents=documents,