gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refactor: Remove conversion to string

Browse source
This commit is contained in:
Igor Ilic 2025-01-24 19:03:57 +01:00
parent 9f6a0ba783
commit b0cec3fcaa
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py
View file

@ -208,7 +208,7 @@ class PGVectorAdapter(SQLAlchemyAdapter, VectorDBInterface):
# Create and return ScoredResult objects # Create and return ScoredResult objects
return [ 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 for row in vector_list
] ]
@ -249,7 +249,7 @@ class PGVectorAdapter(SQLAlchemyAdapter, VectorDBInterface):
for vector in closest_items: for vector in closest_items:
vector_list.append( vector_list.append(
{ {
"id": UUID(str(vector.id)), "id": UUID(vector.id),
"payload": vector.payload, "payload": vector.payload,
"_distance": vector.similarity, "_distance": vector.similarity,
} }