nexus-sdk 2.0.0

Rust SDK for Nexus protocol types, RPC access, and transaction construction.
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148

//! SDK conveniences for generated workflow and verifier bindings.
//!
//! Workflow code turns generated events and execution objects into SDK lookup keys, display
//! values, and timeout decisions. These helpers keep that translation local to generated types so
//! BCS layout and runtime behavior remain coupled to the Move ABI.

use crate::{
    move_bindings::{
        interface::graph::{PostFailureAction, RuntimeVertex},
        workflow::{
            execution::{DAGExecution, DAGExecutionInnerV1, DAGWalk},
            execution_events::RequestWalkExecutionEvent,
            execution_failure::WorkflowFailureClass,
        },
    },
    sui,
    types::{RequestWalkContext, SkillRevisionLookupKey},
};

impl RequestWalkExecutionEvent {
    pub fn skill_revision_key(&self) -> Option<SkillRevisionLookupKey> {
        Some(SkillRevisionLookupKey {
            agent_id: self.agent_id.into(),
            skill_id: self.skill_id,
            interface_revision: self.interface_version,
        })
    }

    pub fn to_context(&self) -> anyhow::Result<Option<RequestWalkContext>> {
        Ok(Some(RequestWalkContext {
            agent_id: self.agent_id.into(),
            skill_id: self.skill_id,
            interface_revision: self.interface_version,
            task_id: self.task_id.bytes,
            occurrence_id: self.occurrence_id,
        }))
    }
}

impl PostFailureAction {
    pub const fn as_str(&self) -> &'static str {
        match self {
            Self::Terminate => "terminate",
            Self::TransientContinue => "continue",
        }
    }
}

impl std::fmt::Display for PostFailureAction {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.write_str(self.as_str())
    }
}

impl std::fmt::Display for WorkflowFailureClass {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        let value = match self {
            Self::Retryable => "retryable",
            Self::TerminalToolFailure => "terminal_tool_failure",
            Self::TerminalSubmissionFailure => "terminal_submission_failure",
        };

        f.write_str(value)
    }
}

impl DAGExecution {
    /// Returns the stable execution object ID.
    pub fn object_id(&self) -> sui::types::Address {
        self.id.id.bytes
    }
}

impl DAGExecutionInnerV1 {
    pub fn dag_id(&self) -> sui::types::Address {
        self.dag.bytes
    }

    pub fn agent_id_address(&self) -> sui::types::Address {
        self.agent_id.bytes
    }

    pub fn task_id_address(&self) -> sui::types::Address {
        self.task_id.bytes
    }

    pub fn occurrence_id_value(&self) -> u64 {
        self.occurrence_id
    }

    pub fn skill_revision_key(&self) -> Option<SkillRevisionLookupKey> {
        Some(SkillRevisionLookupKey {
            agent_id: self.agent_id_address(),
            skill_id: self.skill_id,
            interface_revision: self.interface_version,
        })
    }

    pub fn to_context(&self) -> anyhow::Result<Option<RequestWalkContext>> {
        Ok(Some(RequestWalkContext {
            agent_id: self.agent_id_address(),
            skill_id: self.skill_id,
            interface_revision: self.interface_version,
            task_id: self.task_id_address(),
            occurrence_id: self.occurrence_id_value(),
        }))
    }
}

impl DAGWalk {
    pub fn timeout_expired_vertex(&self, clock_ms: u64) -> Option<&RuntimeVertex> {
        match self {
            Self::Active {
                next_vertex,
                timeout_ms,
                created_at,
                ..
            }
            | Self::PendingSettlement {
                next_vertex,
                timeout_ms,
                created_at,
                ..
            } if clock_ms >= created_at.saturating_add(timeout_ms.saturating_mul(2)) => {
                Some(next_vertex)
            }
            _ => None,
        }
    }

    pub fn abortable_timeout_expired_vertex(&self, clock_ms: u64) -> Option<&RuntimeVertex> {
        match self {
            Self::Active {
                next_vertex,
                timeout_ms,
                created_at,
                ..
            } if clock_ms >= created_at.saturating_add(timeout_ms.saturating_mul(2)) => {
                Some(next_vertex)
            }
            _ => None,
        }
    }

    pub fn expired_active_vertex(&self, clock_ms: u64) -> Option<&RuntimeVertex> {
        self.abortable_timeout_expired_vertex(clock_ms)
    }
}