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(), sa.JSON(),
unique=False, unique=False,
nullable=False, nullable=False,
default={}, server_default=sa.text("'{}'"),
), ),
) )
@ -56,7 +56,7 @@ def upgrade() -> None:
sa.JSON(), sa.JSON(),
unique=False, unique=False,
nullable=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 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 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 # 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 # configuration details for different database types. This would make it more flexible to add new database types
# without changing the database schema. # without changing the database schema.
graph_database_connection_info = Column(JSON, unique=False, nullable=False, default={}) graph_database_connection_info = Column(
vector_database_connection_info = Column(JSON, unique=False, nullable=False, default={}) 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)) created_at = Column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc))
updated_at = Column(DateTime(timezone=True), onupdate=lambda: datetime.now(timezone.utc)) updated_at = Column(DateTime(timezone=True), onupdate=lambda: datetime.now(timezone.utc))