From b0cec3fcaa9d298e63f4f394640fe55c53cb7f25 Mon Sep 17 00:00:00 2001 From: Igor Ilic Date: Fri, 24 Jan 2025 19:03:57 +0100 Subject: [PATCH] refactor: Remove conversion to string --- .../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 3700fd0fa..6e7ea6318 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(str(row.id)), payload=row.payload, score=row.similarity) + ScoredResult(id=UUID(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(str(vector.id)), + "id": UUID(vector.id), "payload": vector.payload, "_distance": vector.similarity, }