c889a57b6b
Test Suites / Build CI Environment (push) Has been cancelled
Test Suites / Basic Tests (push) Has been cancelled
Test Suites / End-to-End Tests (push) Has been cancelled
Test Suites / CLI Tests (push) Has been cancelled
Test Suites / Slow End-to-End Tests (push) Has been cancelled
Test Suites / Graph Database Tests (push) Has been cancelled
Test Suites / Vector DB Tests (push) Has been cancelled
Test Suites / Temporal Graph Test (push) Has been cancelled
Test Suites / Search Test on Different DBs (push) Has been cancelled
Test Suites / Example Tests (push) Has been cancelled
Test Suites / Notebook Tests (push) Has been cancelled
Test Suites / OS and Python Tests Ubuntu (push) Has been cancelled
Test Suites / OS and Python Tests Extended (push) Has been cancelled
Test Suites / LLM Test Suite (push) Has been cancelled
Test Suites / S3 File Storage Test (push) Has been cancelled
Test Suites / Run Integration Tests (push) Has been cancelled
Test Suites / MCP Tests (push) Has been cancelled
Test Suites / Docker Compose Test (push) Has been cancelled
Test Suites / Docker CI test (push) Has been cancelled
Test Suites / Relational DB Migration Tests (push) Has been cancelled
Test Suites / Distributed Cognee Test (push) Has been cancelled
Test Suites / DB Examples Tests (push) Has been cancelled
Test Suites / Test Completion Status (push) Has been cancelled
Test Suites / Claude Code Review (push) Has been cancelled
Test Suites / basic checks (push) Has been cancelled
build | Build and Push Cognee MCP Docker Image to dockerhub / docker-build-and-push (push) Has been cancelled
Scorecard supply-chain security / Scorecard analysis (push) Has been cancelled
build | Build and Push Docker Image to dockerhub / docker-build-and-push (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges Core Functionality (3.11) (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges Core Functionality (3.12) (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges with Different Graph Databases (kuzu, kuzu) (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges with Different Graph Databases (neo4j, neo4j) (push) Has been cancelled
Weighted Edges Tests / Test Weighted Edges Examples (push) Has been cancelled
Weighted Edges Tests / Code Quality for Weighted Edges (push) Has been cancelled
52 lines
2.4 KiB
Python
52 lines
2.4 KiB
Python
from datetime import datetime, timezone
|
|
|
|
from sqlalchemy import Column, DateTime, String, UUID, ForeignKey, JSON, text
|
|
from cognee.infrastructure.databases.relational import Base
|
|
|
|
|
|
class DatasetDatabase(Base):
|
|
__tablename__ = "dataset_database"
|
|
|
|
owner_id = Column(UUID, ForeignKey("principals.id", ondelete="CASCADE"), index=True)
|
|
dataset_id = Column(
|
|
UUID, ForeignKey("datasets.id", ondelete="CASCADE"), primary_key=True, index=True
|
|
)
|
|
|
|
vector_database_name = Column(String, unique=False, nullable=False)
|
|
graph_database_name = Column(String, unique=False, nullable=False)
|
|
|
|
vector_database_provider = Column(String, unique=False, nullable=False)
|
|
graph_database_provider = Column(String, unique=False, nullable=False)
|
|
|
|
graph_dataset_database_handler = Column(String, unique=False, nullable=False)
|
|
vector_dataset_database_handler = Column(String, unique=False, nullable=False)
|
|
|
|
vector_database_url = Column(String, unique=False, nullable=True)
|
|
graph_database_url = Column(String, unique=False, nullable=True)
|
|
|
|
vector_database_key = Column(String, unique=False, nullable=True)
|
|
graph_database_key = Column(String, unique=False, nullable=True)
|
|
|
|
# Cognee release that last touched these databases (informational/audit only).
|
|
cognee_version = Column(String, unique=False, nullable=True)
|
|
|
|
# Last-applied data-migration revision for this dataset's database pair
|
|
# (one Alembic-style chain covers graph + vector + relational-ledger
|
|
# changes). NULL means "no recorded revision" -> all migrations run.
|
|
migration_revision = Column(String, unique=False, nullable=True)
|
|
# Why the last migration attempt failed (NULL when healthy). Written by the
|
|
# runner on failure, cleared on success; surfaced by `cognee-cli current`.
|
|
migration_last_error = Column(String, unique=False, nullable=True)
|
|
|
|
# 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, 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))
|