bamboo_infrastructure/llm/protocol/
mod.rs

1//! Protocol conversion traits and types.
2//!
3//! This module defines the hub-and-spoke conversion architecture where all provider-specific
4//! types convert to/from internal types (agent_core).
5//!
6//! # Architecture
7//!
8//! ```text
9//! Provider Types (OpenAI, Anthropic, etc.)
10//!     ↕
11//! Internal Types (bamboo_domain::Message, ToolSchema)
12//! ```
13
14mod anthropic;
15mod errors;
16pub mod gemini;
17mod openai;
18
19pub use anthropic::AnthropicProtocol;
20pub use errors::{ProtocolError, ProtocolResult};
21pub use gemini::GeminiProtocol;
22pub use openai::OpenAIProtocol;
23
24use bamboo_domain::Message;
25
26/// Trait for converting provider-specific types to internal types.
27///
28/// This is the "spoke → hub" conversion.
29pub trait FromProvider<T>: Sized {
30    /// Convert from provider-specific type to internal type.
31    fn from_provider(value: T) -> ProtocolResult<Self>;
32}
33
34/// Trait for converting internal types to provider-specific types.
35///
36/// This is the "hub → spoke" conversion.
37pub trait ToProvider<T>: Sized {
38    /// Convert from internal type to provider-specific type.
39    fn to_provider(&self) -> ProtocolResult<T>;
40}
41
42/// Batch conversion for multiple messages.
43pub trait FromProviderBatch<T>: Sized {
44    fn from_provider_batch(values: Vec<T>) -> ProtocolResult<Vec<Self>>;
45}
46
47/// Batch conversion for multiple messages.
48pub trait ToProviderBatch<T>: Sized {
49    fn to_provider_batch(&self) -> ProtocolResult<Vec<T>>;
50}
51
52// Implement batch conversion for specific types
53
54impl FromProviderBatch<crate::llm::api::models::ChatMessage> for Message {
55    fn from_provider_batch(
56        values: Vec<crate::llm::api::models::ChatMessage>,
57    ) -> ProtocolResult<Vec<Self>> {
58        values.into_iter().map(Self::from_provider).collect()
59    }
60}
61
62impl ToProviderBatch<crate::llm::api::models::ChatMessage> for Vec<Message> {
63    fn to_provider_batch(&self) -> ProtocolResult<Vec<crate::llm::api::models::ChatMessage>> {
64        self.iter().map(|msg| msg.to_provider()).collect()
65    }
66}
67
68#[cfg(test)]
69mod tests {
70    use super::*;
71
72    #[test]
73    fn test_trait_bounds() {
74        // This test just verifies the trait design compiles
75        fn assert_from_provider<T, U>()
76        where
77            T: FromProvider<U>,
78        {
79        }
80
81        fn assert_to_provider<T, U>()
82        where
83            T: ToProvider<U>,
84        {
85        }
86
87        // Ensure at least one concrete spoke type satisfies the bounds.
88        assert_from_provider::<Message, crate::llm::api::models::ChatMessage>();
89        assert_to_provider::<Message, crate::llm::api::models::ChatMessage>();
90    }
91}
bamboo_infrastructure/llm/protocol/mod.rs

bamboo_infrastructure/llm/protocol/
mod.rs
