mentra-provider 0.2.0

Shared provider core for Mentra
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141

pub mod anthropic;
mod auth;
mod definition;
mod error;
pub mod gemini;
mod model;
mod registry;
mod request;
mod response;
pub mod responses;
mod stream;
mod tool;

pub use auth::AuthScheme;
pub use auth::CredentialSource;
pub use auth::ProviderCredentials;
pub use auth::StaticCredentialSource;
pub use definition::BuiltinProvider;
pub use definition::ProviderCapabilities;
pub use definition::ProviderDefinition;
pub use definition::ProviderDescriptor;
pub use definition::ProviderId;
pub use definition::RetryPolicy;
pub use definition::WireApi;
pub use error::ProviderError;
pub use model::ContentBlock;
pub use model::HostedToolSearchCall;
pub use model::HostedWebSearchCall;
pub use model::ImageGenerationCall;
pub use model::ImageGenerationResult;
pub use model::ImageSource;
pub use model::Message;
pub use model::ModelInfo;
pub use model::ModelSelector;
pub use model::Role;
pub use model::TokenUsage;
pub use model::ToolChoice;
pub use model::ToolResultContent;
pub use model::WebSearchAction;
pub use registry::ModelCatalog;
pub use registry::Provider;
pub use registry::ProviderRegistry;
pub use registry::ProviderSession;
pub use registry::ProviderSessionFactory;
pub use registry::RegisteredProvider;
pub use request::AnthropicRequestOptions;
pub use request::CompactionInputItem;
pub use request::CompactionRequest;
pub use request::GeminiRequestOptions;
pub use request::MemorySummarizeRequest;
pub use request::ProviderRequestOptions;
pub use request::RawMemory;
pub use request::RawMemoryMetadata;
pub use request::ReasoningEffort;
pub use request::ReasoningOptions;
pub use request::ReasoningSummary;
pub use request::Request;
pub use request::ResponsesRequestCompression;
pub use request::ResponsesRequestOptions;
pub use request::ResponsesTextControls;
pub use request::ResponsesTextFormat;
pub use request::ResponsesTextFormatType;
pub use request::ResponsesVerbosity;
pub use request::SessionRequestOptions;
pub use request::ToolSearchMode;
pub use response::CompactionResponse;
pub use response::MemorySummarizeOutput;
pub use response::MemorySummarizeResponse;
pub use response::Response;
pub use response::collect_response_from_stream;
pub use response::provider_event_stream_from_response;
pub use stream::ContentBlockDelta;
pub use stream::ContentBlockStart;
pub use stream::ProviderEvent;
pub use stream::ProviderEventStream;
pub use stream::ResponseHeaders;
pub use tool::ProviderToolKind;
pub use tool::ToolLoadingPolicy;
pub use tool::ToolSpec;
pub use tool::ToolSpecBuilder;

pub type OpenAIRequestOptions = ResponsesRequestOptions;

pub mod provider {
    pub use crate::Provider;

    pub mod model {
        pub use crate::AnthropicRequestOptions;
        pub use crate::ContentBlock;
        pub use crate::ContentBlockDelta;
        pub use crate::ContentBlockStart;
        pub use crate::HostedToolSearchCall;
        pub use crate::HostedWebSearchCall;
        pub use crate::ImageGenerationCall;
        pub use crate::ImageGenerationResult;
        pub use crate::ImageSource;
        pub use crate::Message;
        pub use crate::ModelInfo;
        pub use crate::OpenAIRequestOptions;
        pub use crate::ProviderError;
        pub use crate::ProviderEvent;
        pub use crate::ProviderEventStream;
        pub use crate::ProviderId;
        pub use crate::ProviderRequestOptions;
        pub use crate::ReasoningEffort;
        pub use crate::ReasoningOptions;
        pub use crate::ReasoningSummary;
        pub use crate::Request;
        pub use crate::Response;
        pub use crate::ResponsesTextControls;
        pub use crate::ResponsesTextFormat;
        pub use crate::ResponsesTextFormatType;
        pub use crate::ResponsesVerbosity;
        pub use crate::Role;
        pub use crate::SessionRequestOptions;
        pub use crate::TokenUsage;
        pub use crate::ToolChoice;
        pub use crate::ToolResultContent;
        pub use crate::ToolSearchMode;
        pub use crate::WebSearchAction;
        pub use crate::collect_response_from_stream;
        pub use crate::provider_event_stream_from_response;
    }
}

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

    #[test]
    fn provider_definition_defaults_to_responses_wire_api_and_websockets_disabled() {
        let definition = ProviderDefinition::new(BuiltinProvider::OpenAI);

        assert_eq!(
            definition.descriptor.id,
            ProviderId::from(BuiltinProvider::OpenAI)
        );
        assert_eq!(definition.wire_api, WireApi::Responses);
        assert!(!definition.capabilities.supports_websockets);
    }
}