text-to-cypher 0.1.15

A library and REST API for translating natural language text to Cypher queries using AI models
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46

use serde::{Deserialize, Serialize};
#[cfg(feature = "server")]
use utoipa::ToSchema;

pub use super::attribute::Attribute;

#[derive(Debug, Clone, Serialize, Deserialize)]
#[cfg_attr(feature = "server", derive(ToSchema))]
pub struct Entity {
    pub label: String,
    #[serde(skip_serializing_if = "Vec::is_empty")]
    pub attributes: Vec<Attribute>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub description: Option<String>,
}

impl Entity {
    #[must_use]
    pub const fn new(
        label: String,
        attributes: Vec<Attribute>,
        description: Option<String>,
    ) -> Self {
        Self {
            label,
            attributes,
            description,
        }
    }
}

impl std::fmt::Display for Entity {
    fn fmt(
        &self,
        f: &mut std::fmt::Formatter<'_>,
    ) -> std::fmt::Result {
        let description = self.description.as_ref().map_or("None", |d| d.as_str());
        write!(
            f,
            "Entity: {} ({} attributes, description: {})",
            self.label,
            self.attributes.len(),
            description
        )
    }
}