behest 0.2.1

A Rust-native cloud agent runtime with typed tools, pluggable memory, queues, and observability.
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
142
143
144
145
146
147
148
149
150
151

//! Provider connection configuration.

use std::time::Duration;

use secrecy::SecretString;
use serde::{Deserialize, Serialize};

use crate::provider::{ModelName, ProviderId};

/// Identifies which concrete adapter implementation to use for a provider.
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
#[serde(rename_all = "lowercase")]
pub enum ProviderType {
    /// OpenAI-compatible provider (chat + embeddings).
    #[cfg(feature = "openai")]
    OpenAi,

    /// Anthropic-compatible provider (chat only).
    #[cfg(feature = "anthropic")]
    Anthropic,
}

/// Configuration for a single model provider.
///
/// Supports `env:VAR_NAME` syntax for `api_key` to reference
/// environment variables instead of embedding secrets directly.
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ProviderConfig {
    /// Which adapter implementation to use.
    #[serde(default)]
    pub provider_type: Option<ProviderType>,

    /// Provider API base URL.
    pub base_url: String,

    /// Default model name for this provider.
    #[serde(default)]
    pub model: Option<ModelName>,

    /// Known model names for this provider (model catalog).
    #[serde(default)]
    pub models: Vec<ModelName>,

    /// Model to use for compaction. Falls back to `model` when `None`.
    #[serde(default)]
    pub compaction_model: Option<ModelName>,

    /// API key or bearer token.
    ///
    /// Use `"env:MY_VAR"` to read from the environment variable `MY_VAR`.
    #[serde(default)]
    pub api_key: Option<String>,

    /// Organization, workspace, or tenant identifier.
    #[serde(default)]
    pub organization: Option<String>,

    /// End-to-end request timeout in seconds.
    #[serde(default = "default_timeout_secs")]
    pub timeout_secs: u64,

    /// TCP connection timeout in seconds.
    #[serde(default = "default_connect_timeout_secs")]
    pub connect_timeout_secs: u64,

    /// Maximum retry attempts for retryable failures.
    #[serde(default = "default_max_retries")]
    pub max_retries: usize,
}

impl ProviderConfig {
    /// Creates a minimal provider config with only the base URL.
    #[must_use]
    pub fn new(base_url: impl Into<String>) -> Self {
        Self {
            provider_type: None,
            base_url: base_url.into(),
            model: None,
            models: Vec::new(),
            compaction_model: None,
            api_key: None,
            organization: None,
            timeout_secs: default_timeout_secs(),
            connect_timeout_secs: default_connect_timeout_secs(),
            max_retries: default_max_retries(),
        }
    }

    /// Sets the provider adapter type.
    #[must_use]
    pub fn with_provider_type(mut self, provider_type: ProviderType) -> Self {
        self.provider_type = Some(provider_type);
        self
    }

    /// Sets the default model name.
    #[must_use]
    pub fn with_model(mut self, model: impl Into<ModelName>) -> Self {
        self.model = Some(model.into());
        self
    }

    /// Sets the API key directly.
    #[must_use]
    pub fn with_api_key(mut self, api_key: impl Into<String>) -> Self {
        self.api_key = Some(api_key.into());
        self
    }

    /// Resolves the API key, expanding `env:VAR_NAME` references.
    ///
    /// Returns `None` when no key is configured.
    #[must_use]
    pub fn resolve_api_key(&self) -> Option<SecretString> {
        self.api_key.as_ref().and_then(|key| {
            if let Some(var_name) = key.strip_prefix("env:") {
                std::env::var(var_name)
                    .ok()
                    .map(|s| SecretString::new(s.into_boxed_str()))
            } else {
                Some(SecretString::new(key.clone().into_boxed_str()))
            }
        })
    }

    /// Converts to the provider HTTP config used by adapters.
    #[must_use]
    pub fn to_http_config(&self, id: ProviderId) -> crate::provider::ProviderHttpConfig {
        crate::provider::ProviderHttpConfig {
            id,
            base_url: self.base_url.clone(),
            api_key: self.resolve_api_key(),
            organization: self.organization.clone(),
            timeout: Duration::from_secs(self.timeout_secs),
            connect_timeout: Duration::from_secs(self.connect_timeout_secs),
            max_retries: self.max_retries,
        }
    }
}

const fn default_timeout_secs() -> u64 {
    60
}

const fn default_connect_timeout_secs() -> u64 {
    10
}

const fn default_max_retries() -> usize {
    2
}