llm-trait 0.1.0

Trait definitions and core types for a unified LLM provider layer
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103

//! Protocol type definitions.

use std::fmt;
use std::str::FromStr;

/// Wire protocol type.
///
/// Describes the API format used to communicate with the LLM provider.
/// This is distinct from the provider name (e.g., "deepseek" uses OpenAi protocol).
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum Protocol {
    /// OpenAI Chat Completions API
    OpenAi,
    /// OpenAI Responses API (future)
    OpenAiResponses,
    /// Anthropic Messages API
    Anthropic,
}

/// Error returned when parsing an invalid protocol string.
#[derive(Debug, Clone)]
pub struct ProtocolParseError(String);

impl fmt::Display for ProtocolParseError {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "unknown protocol: '{}'", self.0)
    }
}

impl std::error::Error for ProtocolParseError {}

impl FromStr for Protocol {
    type Err = ProtocolParseError;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        match s.to_lowercase().as_str() {
            "openai" | "openai-chat" => Ok(Self::OpenAi),
            "openai-responses" | "responses" => Ok(Self::OpenAiResponses),
            "anthropic" | "claude" => Ok(Self::Anthropic),
            _ => Err(ProtocolParseError(s.to_string())),
        }
    }
}

impl Protocol {
    pub fn as_str(&self) -> &str {
        match self {
            Self::OpenAi => "openai",
            Self::OpenAiResponses => "openai-responses",
            Self::Anthropic => "anthropic",
        }
    }
}

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

    #[test]
    fn protocol_from_str() {
        assert_eq!("openai".parse::<Protocol>().unwrap(), Protocol::OpenAi);
        assert_eq!("openai-chat".parse::<Protocol>().unwrap(), Protocol::OpenAi);
        assert_eq!(
            "anthropic".parse::<Protocol>().unwrap(),
            Protocol::Anthropic
        );
        assert_eq!("claude".parse::<Protocol>().unwrap(), Protocol::Anthropic);
        assert_eq!(
            "openai-responses".parse::<Protocol>().unwrap(),
            Protocol::OpenAiResponses
        );
        assert_eq!(
            "responses".parse::<Protocol>().unwrap(),
            Protocol::OpenAiResponses
        );
        assert!("unknown".parse::<Protocol>().is_err());
    }

    #[test]
    fn protocol_as_str() {
        assert_eq!(Protocol::OpenAi.as_str(), "openai");
        assert_eq!(Protocol::Anthropic.as_str(), "anthropic");
        assert_eq!(Protocol::OpenAiResponses.as_str(), "openai-responses");
    }

    #[test]
    fn protocol_parse_error_display_names_input() {
        let err = "grpc".parse::<Protocol>().unwrap_err();
        assert_eq!(err.to_string(), "unknown protocol: 'grpc'");
        // Usable as a std error and still Debug-printable.
        let dyn_err: &dyn std::error::Error = &err;
        assert!(dyn_err.to_string().contains("grpc"));
    }

    #[test]
    fn protocol_from_str_case_insensitive() {
        assert_eq!("OpenAI".parse::<Protocol>().unwrap(), Protocol::OpenAi);
        assert_eq!(
            "ANTHROPIC".parse::<Protocol>().unwrap(),
            Protocol::Anthropic
        );
    }
}