Skip to main content

cognee_database/entities/
dataset_data.rs

1use sea_orm::entity::prelude::*;
2use serde::{Deserialize, Serialize};
3
4#[derive(Clone, Debug, PartialEq, Eq, DeriveEntityModel, Serialize, Deserialize)]
5#[sea_orm(table_name = "dataset_data")]
6pub struct Model {
7    #[sea_orm(primary_key, auto_increment = false)]
8    pub dataset_id: String,
9    #[sea_orm(primary_key, auto_increment = false)]
10    pub data_id: String,
11    pub created_at: DateTimeUtc,
12}
13
14#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
15pub enum Relation {
16    #[sea_orm(
17        belongs_to = "super::dataset::Entity",
18        from = "Column::DatasetId",
19        to = "super::dataset::Column::Id",
20        on_delete = "Cascade"
21    )]
22    Dataset,
23    #[sea_orm(
24        belongs_to = "super::data::Entity",
25        from = "Column::DataId",
26        to = "super::data::Column::Id",
27        on_delete = "Cascade"
28    )]
29    Data,
30}
31
32impl ActiveModelBehavior for ActiveModel {}