aion-integrations 0.10.0

Harness-integration SDK for Aion: the AgentHarness trait plus reusable building blocks for making an agent harness a first-class Aion integration.
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

//! The neutral run identity handed to a harness at [`crate::AgentHarness::start`].

use aion_core::{ActivityId, Payload, WorkflowId};

/// The neutral identity + input for one agent run (one activity attempt).
///
/// Carries only what every harness needs to run an attempt: the `(workflow, activity, attempt)`
/// key, the activity-type name the engine dispatched, and the input [`Payload`]. It names **no**
/// harness-specific configuration — an adapter holds any harness-specific settings itself
/// (constructed when the [`crate::AgentHarness`] is built), keeping this spec harness-blind.
#[derive(Clone, Debug, PartialEq, Eq)]
pub struct AgentRunSpec {
    /// The workflow this activity attempt belongs to.
    pub workflow_id: WorkflowId,
    /// The activity within the workflow.
    pub activity_id: ActivityId,
    /// The attempt number of the activity being run.
    pub attempt: u32,
    /// The activity-type name the engine dispatched to this run (e.g. `"dev"`).
    ///
    /// Neutral run identity, not harness configuration: it is the same type name the workflow
    /// definition routed on, so an adapter can label or parameterise a run by what the engine
    /// asked for without the spec learning anything harness-specific.
    pub activity_type: String,
    /// The activity input handed to the agent.
    pub input: Payload,
}

impl AgentRunSpec {
    /// Builds a run spec from the neutral run identity and input payload.
    #[must_use]
    pub fn new(
        workflow_id: WorkflowId,
        activity_id: ActivityId,
        attempt: u32,
        activity_type: impl Into<String>,
        input: Payload,
    ) -> Self {
        Self {
            workflow_id,
            activity_id,
            attempt,
            activity_type: activity_type.into(),
            input,
        }
    }
}

#[cfg(test)]
mod tests {
    use aion_core::{ActivityId, ContentType, Payload, WorkflowId};

    use super::AgentRunSpec;

    #[test]
    fn spec_carries_run_identity_and_input() {
        let workflow_id = WorkflowId::new_v4();
        let activity_id = ActivityId::from_sequence_position(4);
        let input = Payload::new(ContentType::Json, b"{}".to_vec());

        let spec = AgentRunSpec::new(
            workflow_id.clone(),
            activity_id.clone(),
            2,
            "dev",
            input.clone(),
        );

        assert_eq!(spec.workflow_id, workflow_id);
        assert_eq!(spec.activity_id, activity_id);
        assert_eq!(spec.attempt, 2);
        assert_eq!(spec.activity_type, "dev");
        assert_eq!(spec.input, input);
    }
}