ai-lib-core 1.0.0

AI-Protocol execution runtime core (protocol, client, pipeline, transport)
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

//! ProviderContract types — content_block_mapping for manifest-driven encoding.
//!
//! PT-079 / ALR-DOC-002：与 ai-protocol `schemas/v2/provider-contract.json` 对齐的子集。

use serde::Deserialize;
use std::collections::HashMap;

/// Parsed ProviderContract (encoding-relevant fields).
#[derive(Debug, Clone, Deserialize)]
pub struct ProviderContract {
    pub contract_version: String,
    pub provider_id: String,
    pub api_style: String,
    pub request_mapping: RequestMappingContract,
}

/// Request mapping section used by manifest encoder.
#[derive(Debug, Clone, Deserialize)]
pub struct RequestMappingContract {
    pub message_format: String,
    #[serde(default)]
    pub role_mapping: HashMap<String, String>,
    #[serde(default)]
    pub content_block_mapping: Option<ContentBlockMapping>,
}

/// Declarative ContentBlock → wire JSON mapping.
#[derive(Debug, Clone, Deserialize, Default)]
pub struct ContentBlockMapping {
    #[serde(default)]
    pub text: Option<TextBlockMapping>,
    #[serde(default)]
    pub image: Option<ImageBlockMapping>,
    #[serde(default)]
    pub document: Option<DocumentBlockMapping>,
}

#[derive(Debug, Clone, Deserialize)]
pub struct TextBlockMapping {
    #[serde(default)]
    pub field: Option<String>,
    #[serde(default)]
    pub wrapper: Option<String>,
}

#[derive(Debug, Clone, Deserialize)]
pub struct ImageBlockMapping {
    #[serde(default)]
    pub format: Option<String>,
    #[serde(default)]
    pub base64_field: Option<String>,
    #[serde(default)]
    pub url_field: Option<String>,
}

/// Document block mapping per PT-079-R1 schema.
#[derive(Debug, Clone, Deserialize)]
pub struct DocumentBlockMapping {
    pub format: String,
    #[serde(default)]
    pub type_field: Option<String>,
    #[serde(default)]
    pub default_mime_type: Option<String>,
    #[serde(default)]
    pub ref_resolution: Option<String>,
}

impl DocumentBlockMapping {
    pub fn default_mime(&self) -> &str {
        self.default_mime_type
            .as_deref()
            .unwrap_or("application/pdf")
    }

    pub fn rejects_ref_before_encode(&self) -> bool {
        self.ref_resolution
            .as_deref()
            .unwrap_or("error_before_encode")
            == "error_before_encode"
    }
}

#[cfg(test)]
mod tests {
    use crate::protocol::v2::contracts;

    #[test]
    fn anthropic_contract_has_document_mapping() {
        let contract = contracts::anthropic_messages_contract().unwrap();
        let mapping = contract
            .request_mapping
            .content_block_mapping
            .as_ref()
            .and_then(|m| m.document.as_ref())
            .expect("document mapping");
        assert_eq!(mapping.format, "anthropic_document");
        assert!(mapping.rejects_ref_before_encode());
    }

    #[test]
    fn gemini_contract_has_document_mapping() {
        let contract = contracts::gemini_generate_contract().unwrap();
        let mapping = contract
            .request_mapping
            .content_block_mapping
            .as_ref()
            .and_then(|m| m.document.as_ref())
            .expect("document mapping");
        assert_eq!(mapping.format, "gemini_inline_data");
    }
}