phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

fix agent's embedding model config

Browse source
This commit is contained in:
Xi Yan 2024-09-24 22:49:30 -07:00
parent f45705cd10
commit 45be9f3b85
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/impls/meta_reference/agents/agent_instance.py
View file

@ -627,7 +627,7 @@ class ChatAgent(ShieldRunnerMixin):
memory_bank = await self.memory_api.create_memory_bank( memory_bank = await self.memory_api.create_memory_bank(
name=f"memory_bank_{session_id}", name=f"memory_bank_{session_id}",
config=VectorMemoryBankConfig( config=VectorMemoryBankConfig(
embedding_model="sentence-transformer/all-MiniLM-L6-v2", embedding_model="all-MiniLM-L6-v2",
chunk_size_in_tokens=512, chunk_size_in_tokens=512,
), ),
) )