Skip to main content

cognee_database/entities/
graph_metrics.rs

1use sea_orm::entity::prelude::*;
2use serde::{Deserialize, Serialize};
3
4// Note: no Eq because f64 fields don't implement Eq.
5#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Serialize, Deserialize)]
6#[sea_orm(table_name = "graph_metrics")]
7pub struct Model {
8    #[sea_orm(primary_key, auto_increment = false)]
9    pub id: String,
10    pub num_tokens: Option<i32>,
11    pub num_nodes: Option<i32>,
12    pub num_edges: Option<i32>,
13    pub mean_degree: Option<f64>,
14    pub edge_density: Option<f64>,
15    pub num_connected_components: Option<i32>,
16    #[sea_orm(column_type = "Json", nullable)]
17    pub sizes_of_connected_components: Option<Json>,
18    pub num_selfloops: Option<i32>,
19    pub diameter: Option<i32>,
20    pub avg_shortest_path_length: Option<f64>,
21    pub avg_clustering: Option<f64>,
22    pub created_at: DateTimeUtc,
23    pub updated_at: Option<DateTimeUtc>,
24}
25
26#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
27pub enum Relation {}
28
29impl ActiveModelBehavior for ActiveModel {}