From 45be9f3b856693064ee07e8bf86954b3a8987fbc Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Tue, 24 Sep 2024 22:49:30 -0700 Subject: [PATCH] fix agent's embedding model config --- .../providers/impls/meta_reference/agents/agent_instance.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/providers/impls/meta_reference/agents/agent_instance.py b/llama_stack/providers/impls/meta_reference/agents/agent_instance.py index 797a1bc7f..952946a1e 100644 --- a/llama_stack/providers/impls/meta_reference/agents/agent_instance.py +++ b/llama_stack/providers/impls/meta_reference/agents/agent_instance.py @@ -627,7 +627,7 @@ class ChatAgent(ShieldRunnerMixin): memory_bank = await self.memory_api.create_memory_bank( name=f"memory_bank_{session_id}", config=VectorMemoryBankConfig( - embedding_model="sentence-transformer/all-MiniLM-L6-v2", + embedding_model="all-MiniLM-L6-v2", chunk_size_in_tokens=512, ), )