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

migrate router for memory wip

Browse source
This commit is contained in:
Xi Yan 2024-09-20 12:19:33 -07:00
parent 7d4135d5fd
commit cda61119ce
8 changed files with 213 additions and 45 deletions
Download patch file Download diff file Expand all files Collapse all files

53
llama_stack/providers/adapters/memory/pgvector/pgvector.py
View file

@ -128,35 +128,36 @@ class PGVectorMemoryAdapter(Memory):
self.cache = {}
async def initialize(self) -> None:
try:
self.conn = psycopg2.connect(
host=self.config.host,
port=self.config.port,
database=self.config.db,
user=self.config.user,
password=self.config.password,
)
self.cursor = self.conn.cursor()
print("Init PGVector!")
# try:
# self.conn = psycopg2.connect(
# host=self.config.host,
# port=self.config.port,
# database=self.config.db,
# user=self.config.user,
# password=self.config.password,
# )
# self.cursor = self.conn.cursor()
version = check_extension_version(self.cursor)
if version:
print(f"Vector extension version: {version}")
else:
raise RuntimeError("Vector extension is not installed.")
# version = check_extension_version(self.cursor)
# if version:
# print(f"Vector extension version: {version}")
# else:
# raise RuntimeError("Vector extension is not installed.")
self.cursor.execute(
"""
CREATE TABLE IF NOT EXISTS metadata_store (
key TEXT PRIMARY KEY,
data JSONB
)
"""
)
except Exception as e:
import traceback
# self.cursor.execute(
# """
# CREATE TABLE IF NOT EXISTS metadata_store (
# key TEXT PRIMARY KEY,
# data JSONB
# )
# """
# )
# except Exception as e:
# import traceback
traceback.print_exc()
raise RuntimeError("Could not connect to PGVector database server") from e
# traceback.print_exc()
# raise RuntimeError("Could not connect to PGVector database server") from e
async def shutdown(self) -> None:
pass