diff --git a/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py b/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py index d9aecec90..a911d4785 100644 --- a/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py +++ b/cognee/infrastructure/databases/vector/pgvector/PGVectorAdapter.py @@ -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) ]