1ed6cfd918
<!-- .github/pull_request_template.md --> ## Description <!-- Provide a clear description of the changes in this PR --> ## DCO Affirmation I affirm that all code in every commit of this pull request conforms to the terms of the Topoteretes Developer Certificate of Origin. --------- Co-authored-by: Boris Arzentar <borisarzentar@gmail.com> Co-authored-by: Boris <boris@topoteretes.com>
19 lines
778 B
Python
19 lines
778 B
Python
from datetime import datetime, timezone
|
|
|
|
from sqlalchemy import Column, DateTime, String, UUID, ForeignKey
|
|
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=True, nullable=False)
|
|
graph_database_name = Column(String, unique=True, nullable=False)
|
|
|
|
created_at = Column(DateTime(timezone=True), default=lambda: datetime.now(timezone.utc))
|
|
updated_at = Column(DateTime(timezone=True), onupdate=lambda: datetime.now(timezone.utc))
|