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

Allow using an "inline" version of Chroma using PersistentClient

Browse source
This commit is contained in:
Ashwin Bharambe 2024-12-04 12:54:39 -08:00
parent e2054d53e4
commit 44ab7d93fb
9 changed files with 71 additions and 69 deletions
Download patch file Download diff file Expand all files Collapse all files

11
llama_stack/providers/remote/memory/pgvector/pgvector.py
View file

@ -185,17 +185,6 @@ class PGVectorMemoryAdapter(Memory, MemoryBanksProtocolPrivate):
await self.cache[memory_bank_id].index.delete()
del self.cache[memory_bank_id]
async def list_memory_banks(self) -> List[MemoryBank]:
banks = load_models(self.cursor, VectorMemoryBank)
for bank in banks:
if bank.identifier not in self.cache:
index = BankWithIndex(
bank=bank,
index=PGVectorIndex(bank, ALL_MINILM_L6_V2_DIMENSION, self.cursor),
)
self.cache[bank.identifier] = index
return banks
async def insert_documents(
self,
bank_id: str,