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

fix: serialize UUID in pgvector data point payload

Browse source
This commit is contained in:
Boris Arzentar 2024-11-07 11:36:31 +01:00 • committed by Leon Luithlen
parent a2b1087c84
commit 63900f6b0a
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

@ -8,7 +8,7 @@ from sqlalchemy.ext.asyncio import create_async_engine, async_sessionmaker
from cognee.infrastructure.engine import DataPoint
from .serialize_datetime import serialize_datetime
from .serialize_data import serialize_data
from ..models.ScoredResult import ScoredResult
from ..vector_db_interface import VectorDBInterface
from ..embeddings.EmbeddingEngine import EmbeddingEngine
@ -111,7 +111,7 @@ class PGVectorAdapter(SQLAlchemyAdapter, VectorDBInterface):
PGVectorDataPoint(
id=data_point.id,
vector=data_vectors[data_index],
payload=serialize_datetime(data_point.model_dump()),
payload=serialize_data(data_point.model_dump()),
)
for (data_index, data_point) in enumerate(data_points)
]