unified-agent-api 0.3.5

Agent-agnostic facade and registry for wrapper backends
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

use std::{collections::BTreeSet, future::Future, pin::Pin, sync::Arc};

use crate::{
    backend_harness::{run_harnessed_backend, BackendDefaults},
    AgentWrapperBackend, AgentWrapperCapabilities, AgentWrapperError, AgentWrapperKind,
    AgentWrapperRunHandle, AgentWrapperRunRequest,
};

use super::GeminiCliBackend;

impl AgentWrapperBackend for GeminiCliBackend {
    fn kind(&self) -> AgentWrapperKind {
        AgentWrapperKind(super::AGENT_KIND.to_string())
    }

    fn capabilities(&self) -> AgentWrapperCapabilities {
        let ids = BTreeSet::from([
            super::CAP_RUN_V1.to_string(),
            super::CAP_EVENTS_V1.to_string(),
            super::CAP_EVENTS_LIVE_V1.to_string(),
            crate::EXT_AGENT_API_CONFIG_MODEL_V1.to_string(),
        ]);
        AgentWrapperCapabilities { ids }
    }

    fn run(
        &self,
        request: AgentWrapperRunRequest,
    ) -> Pin<Box<dyn Future<Output = Result<AgentWrapperRunHandle, AgentWrapperError>> + Send + '_>>
    {
        let adapter = Arc::new(self.clone());
        let defaults = BackendDefaults {
            env: self.config.env.clone(),
            default_timeout: self.config.default_timeout,
        };

        Box::pin(async move { run_harnessed_backend(adapter, defaults, request).await })
    }
}