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

pgvector fix v2

Browse source
This commit is contained in:
Dinesh Yeduguru 2024-11-11 10:50:27 -08:00
parent b2b49fbdb9
commit f1a2996812
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -121,6 +121,7 @@ class PGVectorMemoryAdapter(Memory, MemoryBanksProtocolPrivate):
self.cache = {} self.cache = {}
async def initialize(self) -> None: async def initialize(self) -> None:
print(f"Initializing PGVector memory adapter with config: {self.config}")
try: try:
self.conn = psycopg2.connect( self.conn = psycopg2.connect(
host=self.config.host, host=self.config.host,
@ -160,8 +161,8 @@ class PGVectorMemoryAdapter(Memory, MemoryBanksProtocolPrivate):
memory_bank: VectorMemoryBank, memory_bank: VectorMemoryBank,
) -> None: ) -> None:
assert ( assert (
memory_bank.type == MemoryBankType.vector.value memory_bank.memory_bank_type == MemoryBankType.vector.value
), f"Only vector banks are supported {memory_bank.type}" ), f"Only vector banks are supported {memory_bank.memory_bank_type}"
upsert_models( upsert_models(
self.cursor, self.cursor,