diff --git a/alembic/versions/46a6ce2bd2b2_expand_dataset_database_with_json_.py b/alembic/versions/46a6ce2bd2b2_expand_dataset_database_with_json_.py index becd29226..a9e804f88 100644 --- a/alembic/versions/46a6ce2bd2b2_expand_dataset_database_with_json_.py +++ b/alembic/versions/46a6ce2bd2b2_expand_dataset_database_with_json_.py @@ -41,7 +41,7 @@ def upgrade() -> None: sa.JSON(), unique=False, nullable=False, - default={}, + server_default=sa.text("'{}'"), ), ) @@ -56,7 +56,7 @@ def upgrade() -> None: sa.JSON(), unique=False, nullable=False, - default={}, + server_default=sa.text("'{}'"), ), ) diff --git a/cognee/modules/users/models/DatasetDatabase.py b/cognee/modules/users/models/DatasetDatabase.py index fee323d2f..577828921 100644 --- a/cognee/modules/users/models/DatasetDatabase.py +++ b/cognee/modules/users/models/DatasetDatabase.py @@ -1,6 +1,6 @@ from datetime import datetime, timezone -from sqlalchemy import Column, DateTime, String, UUID, ForeignKey, JSON +from sqlalchemy import Column, DateTime, String, UUID, ForeignKey, JSON, text from cognee.infrastructure.databases.relational import Base @@ -27,8 +27,12 @@ class DatasetDatabase(Base): # TODO: Instead of specifying and forwawrding all these individual fields, consider using a JSON field to store # configuration details for different database types. This would make it more flexible to add new database types # without changing the database schema. - graph_database_connection_info = Column(JSON, unique=False, nullable=False, default={}) - vector_database_connection_info = Column(JSON, unique=False, nullable=False, default={}) + graph_database_connection_info = Column( + JSON, unique=False, nullable=False, server_default=text("'{}'") + ) + vector_database_connection_info = Column( + JSON, unique=False, nullable=False, server_default=text("'{}'") + ) created_at = Column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc)) updated_at = Column(DateTime(timezone=True), onupdate=lambda: datetime.now(timezone.utc))