Skip to main content

cognee_database/entities/
query.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 = "queries")]
6pub struct Model {
7    #[sea_orm(primary_key, auto_increment = false)]
8    pub id: String,
9    pub query_text: String,
10    pub query_type: String,
11    pub user_id: Option<String>,
12    pub created_at: DateTimeUtc,
13}
14
15#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
16pub enum Relation {
17    #[sea_orm(has_many = "super::result_log::Entity")]
18    Results,
19}
20
21impl Related<super::result_log::Entity> for Entity {
22    fn to() -> RelationDef {
23        Relation::Results.def()
24    }
25}
26
27impl ActiveModelBehavior for ActiveModel {}