oneiros-model 0.0.6

Shared domain types for oneiros
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

use serde::{Deserialize, Serialize};

use crate::*;

#[derive(Debug, thiserror::Error)]
pub enum AgentConstructionError {
    #[error("invalid agent id: {0}")]
    InvalidId(#[from] IdParseError),
}

#[derive(Clone, Debug, Serialize, Deserialize, PartialEq)]
pub struct Agent {
    pub id: AgentId,
    pub name: AgentName,
    pub persona: PersonaName,
    pub description: Description,
    pub prompt: Prompt,
}

impl Agent {
    pub fn init(
        description: impl Into<Description>,
        prompt: impl Into<Prompt>,
        name: AgentName,
        persona: PersonaName,
    ) -> Self {
        Self {
            id: AgentId::new(),
            name,
            persona,
            description: description.into(),
            prompt: prompt.into(),
        }
    }

    pub fn construct(
        id: impl Into<AgentId>,
        description: impl Into<Description>,
        prompt: impl Into<Prompt>,
        name: AgentName,
        persona: PersonaName,
    ) -> Self {
        Self {
            id: id.into(),
            name,
            persona,
            description: description.into(),
            prompt: prompt.into(),
        }
    }

    pub fn construct_from_db(
        (id, name, persona, description, prompt): (
            impl AsRef<str>,
            impl AsRef<str>,
            impl AsRef<str>,
            impl AsRef<str>,
            impl AsRef<str>,
        ),
    ) -> Result<Agent, AgentConstructionError> {
        Ok(Agent {
            id: id.as_ref().parse()?,
            name: AgentName::new(name),
            persona: PersonaName::new(persona),
            description: Description::new(description),
            prompt: Prompt::new(prompt),
        })
    }
}

domain_id!(AgentId);
domain_name!(AgentName);