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

refactor: set default value to empty dictionary

Browse source
This commit is contained in:
Igor Ilic 2025-11-26 16:45:18 +01:00
parent cf9edf2663
commit 1ff6a72fc7
2 changed files with 9 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
alembic/versions/46a6ce2bd2b2_expand_dataset_database_with_json_.py
View file

@ -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("'{}'"),
),
)

10
cognee/modules/users/models/DatasetDatabase.py
View file

@ -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))