distri 0.3.8

Rust client for the Distri A2A agent platform
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

//! Builds the `distri_request` dynamic tool factory from client config.
//!
//! Any client (CLI, SDK, gateway) that talks to a distri agent can call
//! `build_distri_request_factory` to create a `DynamicToolFactory` that
//! the agent will use to call back into the platform API.  The factory
//! is sent as a `DefinitionOverrides.dynamic_tools` entry in the A2A
//! metadata so the server injects it at agent-config level.

use std::collections::HashMap;

use distri_types::DistriConfig;
use distri_types::configuration::DefinitionOverrides;
use distri_types::dynamic_tool::DynamicToolFactory;
use distri_types::http_request::HttpFactoryConfig;

/// Build a `DynamicToolFactory` named `distri_request` from the client's
/// connection config.  The factory is an HTTP tool whose base_url and auth
/// headers mirror the client's own credentials so the agent can call the
/// platform API on behalf of the logged-in user.
pub fn build_distri_request_factory(config: &DistriConfig) -> DynamicToolFactory {
    let mut headers = HashMap::new();
    if let Some(ref api_key) = config.api_key {
        headers.insert("x-api-key".to_string(), api_key.clone());
    }
    if let Some(ref workspace_id) = config.workspace_id {
        headers.insert("x-workspace-id".to_string(), workspace_id.clone());
    }

    let factory_config = HttpFactoryConfig {
        base_url: config.base_url.clone(),
        headers,
    };

    DynamicToolFactory {
        name: "distri_request".to_string(),
        factory_type: "http".to_string(),
        config: serde_json::to_value(factory_config).expect("HttpFactoryConfig serialization"),
        description: Some(
            "Call the Distri platform REST API. Input: {path, method, headers?, body?}".to_string(),
        ),
    }
}

/// Build `DefinitionOverrides` containing the `distri_request` dynamic tool.
pub fn build_platform_overrides(config: &DistriConfig) -> DefinitionOverrides {
    DefinitionOverrides::new().with_dynamic_tools(vec![build_distri_request_factory(config)])
}

/// Build a `distri_request` factory with an explicit workspace ID override.
/// Used by the gateway which handles requests for different workspaces.
pub fn build_distri_request_factory_for_workspace(
    config: &DistriConfig,
    workspace_id: uuid::Uuid,
) -> DynamicToolFactory {
    let mut headers = HashMap::new();
    if let Some(ref api_key) = config.api_key {
        headers.insert("x-api-key".to_string(), api_key.clone());
    }
    // Always set the workspace header to the specific workspace
    headers.insert("x-workspace-id".to_string(), workspace_id.to_string());

    let factory_config = HttpFactoryConfig {
        base_url: config.base_url.clone(),
        headers,
    };

    DynamicToolFactory {
        name: "distri_request".to_string(),
        factory_type: "http".to_string(),
        config: serde_json::to_value(factory_config).expect("HttpFactoryConfig serialization"),
        description: Some(
            "Call the Distri platform REST API. Input: {path, method, headers?, body?}".to_string(),
        ),
    }
}