This commit is contained in:
IAN MILLER 2025-10-03 04:57:49 -07:00 committed by GitHub
commit 7774d9e9f3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
80 changed files with 32197 additions and 186 deletions

View file

@ -458,8 +458,8 @@ def test_rag_agent(llama_stack_client, agent_config, rag_tool_name):
vector_db_id = f"test-vector-db-{uuid4()}"
llama_stack_client.vector_dbs.register(
vector_db_id=vector_db_id,
embedding_model="all-MiniLM-L6-v2",
embedding_dimension=384,
embedding_model="nomic-embed-text-v1.5",
embedding_dimension=768,
)
llama_stack_client.tool_runtime.rag_tool.insert(
documents=documents,