agent_control_specification_core 0.3.1-beta.0

Stateless Rust core for Agent Control Specification
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
73
74
75
76
77
78
79
80
81
82
83

use crate::{constants::annotation as annotation_key, JsonValue, RuntimeError};
use serde::{Deserialize, Serialize};
use std::collections::BTreeMap;

#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize)]
#[serde(rename_all = "snake_case")]
pub enum AnnotatorType {
    Classifier,
    Llm,
    Endpoint,
}

impl AnnotatorType {
    pub fn as_str(self) -> &'static str {
        match self {
            Self::Classifier => "classifier",
            Self::Llm => "llm",
            Self::Endpoint => "endpoint",
        }
    }
}

#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct AnnotatorConfig {
    #[serde(rename = "type")]
    pub annotator_type: AnnotatorType,
    #[serde(default, flatten, skip_serializing_if = "BTreeMap::is_empty")]
    pub fields: BTreeMap<String, JsonValue>,
}

#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct AnnotationConfig {
    pub from: String,
    #[serde(default, flatten, skip_serializing_if = "BTreeMap::is_empty")]
    pub fields: BTreeMap<String, JsonValue>,
}

#[derive(Debug, Clone, Default, PartialEq, Serialize, Deserialize)]
pub struct AnnotatorInvocation {
    #[serde(flatten)]
    pub fields: BTreeMap<String, JsonValue>,
}

impl AnnotatorInvocation {
    pub fn from_annotation(annotator: &AnnotatorConfig, annotation: &AnnotationConfig) -> Self {
        let mut fields = BTreeMap::new();
        fields.insert(
            annotation_key::TYPE.to_string(),
            JsonValue::String(annotator.annotator_type.as_str().to_string()),
        );
        for (key, value) in &annotator.fields {
            fields.insert(key.clone(), value.clone());
        }
        fields.insert(
            annotation_key::FROM.to_string(),
            JsonValue::String(annotation.from.clone()),
        );
        for (key, value) in &annotation.fields {
            fields.insert(key.clone(), value.clone());
        }
        Self { fields }
    }

    pub fn input_from(&self) -> Option<&str> {
        self.fields
            .get(annotation_key::FROM)
            .or_else(|| self.fields.get(annotation_key::INPUT_FROM))
            .and_then(JsonValue::as_str)
    }

    pub fn field(&self, name: &str) -> Option<&JsonValue> {
        self.fields.get(name)
    }
}

pub trait AnnotatorDispatcher: Send + Sync {
    fn dispatch(
        &self,
        annotator_name: &str,
        annotator: &AnnotatorInvocation,
        preliminary_policy_input: &JsonValue,
    ) -> Result<JsonValue, RuntimeError>;
}