phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-17 23:17:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

remove auto discovery

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-11-11 11:13:33 -08:00
parent f1a2996812
commit 4206f07ada
4 changed files with 25 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -158,7 +158,7 @@ class PGVectorMemoryAdapter(Memory, MemoryBanksProtocolPrivate):
async def register_memory_bank(
self,
memory_bank: VectorMemoryBank,
memory_bank: MemoryBank,
) -> None:
assert (
memory_bank.memory_bank_type == MemoryBankType.vector.value
@ -177,7 +177,7 @@ class PGVectorMemoryAdapter(Memory, MemoryBanksProtocolPrivate):
)
self.cache[memory_bank.identifier] = index
async def list_memory_banks(self) -> List[VectorMemoryBank]:
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: