JDSS/mem0
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: drop unused vector column from pgvector get() and list() queries (#6483)

Browse Source
This commit is contained in:
Gyubin Son
2026-07-23 00:11:40 +05:30
committed by GitHub
parent 5e7adc4d12
commit 7bf84b8d38
2 changed files with 28 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mem0/vector_stores/pgvector.py
+4 -4
View File
@@ -459,13 +459,13 @@ class PGVector(VectorStoreBase):
self._ensure_collection()
with self._get_cursor() as cur:
cur.execute(
sql.SQL("SELECT id, vector, payload FROM {} WHERE id = %s").format(self._col()),
sql.SQL("SELECT id, payload FROM {} WHERE id = %s").format(self._col()),
(vector_id,),
)
result = cur.fetchone()
if not result:
return None
return OutputData(id=str(result[0]), score=None, payload=result[2])
return OutputData(id=str(result[0]), score=None, payload=result[1])
def list_cols(self) -> List[str]:
"""
@@ -528,7 +528,7 @@ class PGVector(VectorStoreBase):
with self._get_cursor() as cur:
cur.execute(
sql.SQL("""
SELECT id, vector, payload
SELECT id, payload
FROM {}
{}
LIMIT %s
@@ -536,7 +536,7 @@ class PGVector(VectorStoreBase):
(*filter_params, top_k),
)
results = cur.fetchall()
return [[OutputData(id=str(r[0]), score=None, payload=r[2]) for r in results]]
return [[OutputData(id=str(r[0]), score=None, payload=r[1]) for r in results]]
def __del__(self) -> None:
"""