adk-awp 0.8.2

Agentic Web Protocol (AWP) implementation for ADK-Rust
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

//! Discovery document generation from [`BusinessContext`].

use awp_types::{AwpDiscoveryDocument, BusinessContext, CURRENT_VERSION, TrustLevel};

/// Generate an AWP discovery document from a business context.
///
/// URLs are derived from the context's `domain` field.
///
/// # Example
///
/// ```
/// use awp_types::{BusinessContext, TrustLevel, CURRENT_VERSION};
/// use adk_awp::generate_discovery_document;
///
/// let ctx = BusinessContext::core("My Site", "A site", "example.com");
/// let doc = generate_discovery_document(&ctx);
/// assert_eq!(doc.version, CURRENT_VERSION);
/// assert_eq!(doc.site_name, "My Site");
/// ```
pub fn generate_discovery_document(ctx: &BusinessContext) -> AwpDiscoveryDocument {
    AwpDiscoveryDocument {
        version: CURRENT_VERSION,
        site_name: ctx.site_name.clone(),
        site_description: ctx.site_description.clone(),
        capability_manifest_url: format!("https://{}/awp/manifest", ctx.domain),
        a2a_endpoint_url: format!("https://{}/awp/a2a", ctx.domain),
        events_endpoint_url: format!("https://{}/awp/events/subscribe", ctx.domain),
        health_endpoint_url: format!("https://{}/awp/health", ctx.domain),
        supported_trust_levels: vec![
            TrustLevel::Anonymous,
            TrustLevel::Known,
            TrustLevel::Partner,
            TrustLevel::Internal,
        ],
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    fn sample_context() -> BusinessContext {
        BusinessContext::core("Test Site", "A test site", "example.com")
    }

    #[test]
    fn test_generate_discovery_document_version() {
        let doc = generate_discovery_document(&sample_context());
        assert_eq!(doc.version, CURRENT_VERSION);
    }

    #[test]
    fn test_generate_discovery_document_urls() {
        let doc = generate_discovery_document(&sample_context());
        assert_eq!(doc.capability_manifest_url, "https://example.com/awp/manifest");
        assert_eq!(doc.a2a_endpoint_url, "https://example.com/awp/a2a");
        assert_eq!(doc.events_endpoint_url, "https://example.com/awp/events/subscribe");
        assert_eq!(doc.health_endpoint_url, "https://example.com/awp/health");
    }

    #[test]
    fn test_generate_discovery_document_trust_levels() {
        let doc = generate_discovery_document(&sample_context());
        assert_eq!(
            doc.supported_trust_levels,
            vec![
                TrustLevel::Anonymous,
                TrustLevel::Known,
                TrustLevel::Partner,
                TrustLevel::Internal,
            ]
        );
    }

    #[test]
    fn test_generate_discovery_document_site_info() {
        let doc = generate_discovery_document(&sample_context());
        assert_eq!(doc.site_name, "Test Site");
        assert_eq!(doc.site_description, "A test site");
    }
}