systemprompt-models 0.1.18

Shared data models and types for systemprompt.io OS
Documentation 
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
47
48
49
50
51
52
53
54

use crate::artifacts::card::PresentationCardResponse;
use schemars::JsonSchema;
use serde::{Deserialize, Serialize};

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema)]
pub struct ResearchArtifact {
    #[serde(flatten)]
    pub card: PresentationCardResponse,
    pub topic: String,
    pub sources: Vec<SourceCitation>,
    pub query_count: u32,
    pub source_count: u32,
}

impl ResearchArtifact {
    pub const ARTIFACT_TYPE: &'static str = "presentation_card";

    pub fn new(
        topic: impl Into<String>,
        card: PresentationCardResponse,
        sources: Vec<SourceCitation>,
    ) -> Self {
        let source_count = u32::try_from(sources.len()).unwrap_or(u32::MAX);
        Self {
            card,
            topic: topic.into(),
            sources,
            query_count: 0,
            source_count,
        }
    }

    pub const fn with_query_count(mut self, count: u32) -> Self {
        self.query_count = count;
        self
    }
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema)]
pub struct SourceCitation {
    pub title: String,
    pub uri: String,
    pub relevance: f32,
}

impl SourceCitation {
    pub fn new(title: impl Into<String>, uri: impl Into<String>, relevance: f32) -> Self {
        Self {
            title: title.into(),
            uri: uri.into(),
            relevance,
        }
    }
}