ras-types 2.1.0

Shared types and ID newtypes
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

use indexmap::IndexMap;
use serde::{Deserialize, Serialize};

#[derive(Debug, Clone, Default, Serialize, Deserialize)]
pub struct ActionResult {
    pub extracted_content: Option<String>,
    pub include_in_memory: bool,
    pub error: Option<String>,
    pub is_done: bool,
    pub long_term_memory: Option<String>,
    pub attachments: Vec<String>,
    pub images: Vec<String>,
    pub metadata: IndexMap<String, serde_json::Value>,
}

impl ActionResult {
    #[must_use]
    pub fn ok(content: impl Into<String>) -> Self {
        Self {
            extracted_content: Some(content.into()),
            include_in_memory: true,
            ..Self::default()
        }
    }

    #[must_use]
    pub fn err(message: impl Into<String>) -> Self {
        Self {
            error: Some(message.into()),
            include_in_memory: true,
            ..Self::default()
        }
    }

    #[must_use]
    pub fn done(content: impl Into<String>) -> Self {
        Self {
            extracted_content: Some(content.into()),
            include_in_memory: true,
            is_done: true,
            ..Self::default()
        }
    }

    #[must_use]
    pub fn with_attachment(mut self, path: impl Into<String>) -> Self {
        self.attachments.push(path.into());
        self
    }

    #[must_use]
    pub fn with_image(mut self, b64: impl Into<String>) -> Self {
        self.images.push(b64.into());
        self
    }

    #[must_use]
    pub fn is_error(&self) -> bool {
        self.error.is_some()
    }
}