phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-08 11:07:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

improve memory test, but it fails on chromadb :/

Browse source
This commit is contained in:
Ashwin Bharambe 2024-10-07 23:09:31 -07:00 • committed by Ashwin Bharambe
parent 4ab6e1b81a
commit f21ad1173e
2 changed files with 56 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/adapters/memory/chroma/chroma.py
View file

@ -99,7 +99,7 @@ class ChromaMemoryAdapter(Memory):
memory_bank.type == MemoryBankType.vector.value
), f"Only vector banks are supported {memory_bank.type}"
collection = await self.client.create_collection(
collection = await self.client.get_or_create_collection(
name=memory_bank.identifier,
)
bank_index = BankWithIndex(