Skip to main content

cognee_database/entities/
dataset.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 = "datasets")]
6pub struct Model {
7    #[sea_orm(primary_key, auto_increment = false)]
8    pub id: String,
9    pub name: String,
10    #[sea_orm(indexed)]
11    pub owner_id: String,
12    pub tenant_id: Option<String>,
13    pub created_at: DateTimeUtc,
14    pub updated_at: Option<DateTimeUtc>,
15}
16
17#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
18pub enum Relation {}
19
20impl Related<super::data::Entity> for Entity {
21    fn to() -> RelationDef {
22        super::dataset_data::Relation::Data.def()
23    }
24    fn via() -> Option<RelationDef> {
25        Some(super::dataset_data::Relation::Dataset.def().rev())
26    }
27}
28
29impl ActiveModelBehavior for ActiveModel {}