fix agent's embedding model config

2024-09-24 22:49:30 -07:00 · 2024-09-24 22:49:30 -07:00 · 45be9f3b85
commit 45be9f3b85
parent f45705cd10
1 changed files with 1 additions and 1 deletions
--- 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,
                ),
            )