mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-10-04 12:07:34 +00:00
Merge 376eb94def
into d266c59c2a
This commit is contained in:
commit
7774d9e9f3
80 changed files with 32197 additions and 186 deletions
|
@ -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,
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue