From 23ecf245edb6ee1a1a0912002535d09c02563aed Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Fri, 24 Jan 2025 19:20:55 +0100 Subject: [PATCH] fix: Return string conversion to resolve traceback --- .../databases/vector/pgvector/PGVectorAdapter.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py b/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py index 6e7ea6318..3700fd0fa 100644 --- a/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py +++ b/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py @@ -208,7 +208,7 @@ class PGVectorAdapter(SQLAlchemyAdapter, VectorDBInterface): # Create and return ScoredResult objects return [ - ScoredResult(id=UUID(row.id), payload=row.payload, score=row.similarity) + ScoredResult(id=UUID(str(row.id)), payload=row.payload, score=row.similarity) for row in vector_list ] @@ -249,7 +249,7 @@ class PGVectorAdapter(SQLAlchemyAdapter, VectorDBInterface): for vector in closest_items: vector_list.append( { - "id": UUID(vector.id), + "id": UUID(str(vector.id)), "payload": vector.payload, "_distance": vector.similarity, }