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

Revert "migrate router for memory wip"

Browse source 
This reverts commit 08379f5214.
This commit is contained in:
Xi Yan 2024-09-21 12:42:09 -07:00
parent 3939611676
commit cf8bd10989
8 changed files with 45 additions and 213 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,36 +128,35 @@ class PGVectorMemoryAdapter(Memory):
self.cache = {}
async def initialize(self) -> None:
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()
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