claude-agent 0.2.25

Rust SDK for building AI agents with Anthropic's Claude - Direct API, no CLI dependency
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

//! Provider adapters for different cloud platforms.

mod anthropic;
#[cfg(any(feature = "aws", feature = "gcp", feature = "azure"))]
mod base;
mod config;
mod request;
mod traits;

#[cfg(any(feature = "aws", feature = "gcp", feature = "azure"))]
mod token_cache;

#[cfg(feature = "aws")]
mod bedrock;
#[cfg(feature = "azure")]
mod foundry;
#[cfg(feature = "gcp")]
mod vertex;

pub use anthropic::AnthropicAdapter;
pub use config::{
    BetaConfig, BetaFeature, DEFAULT_MODEL, DEFAULT_REASONING_MODEL, DEFAULT_SMALL_MODEL,
    FRONTIER_MODEL, ModelConfig, ModelType, ProviderConfig,
};
pub use traits::ProviderAdapter;

#[cfg(feature = "aws")]
pub use bedrock::BedrockAdapter;
#[cfg(feature = "azure")]
pub use foundry::FoundryAdapter;
#[cfg(feature = "gcp")]
pub use vertex::VertexAdapter;

use crate::Result;

#[derive(Clone, Copy, Debug, Default, PartialEq, Eq)]
pub enum CloudProvider {
    #[default]
    Anthropic,
    #[cfg(feature = "aws")]
    Bedrock,
    #[cfg(feature = "gcp")]
    Vertex,
    #[cfg(feature = "azure")]
    Foundry,
}

impl CloudProvider {
    pub fn from_env() -> Self {
        #[cfg(feature = "aws")]
        if std::env::var("CLAUDE_CODE_USE_BEDROCK").is_ok() {
            return Self::Bedrock;
        }
        #[cfg(feature = "gcp")]
        if std::env::var("CLAUDE_CODE_USE_VERTEX").is_ok() {
            return Self::Vertex;
        }
        #[cfg(feature = "azure")]
        if std::env::var("CLAUDE_CODE_USE_FOUNDRY").is_ok() {
            return Self::Foundry;
        }
        Self::Anthropic
    }

    pub fn default_models(&self) -> ModelConfig {
        match self {
            Self::Anthropic => ModelConfig::anthropic(),
            #[cfg(feature = "aws")]
            Self::Bedrock => ModelConfig::bedrock(),
            #[cfg(feature = "gcp")]
            Self::Vertex => ModelConfig::vertex(),
            #[cfg(feature = "azure")]
            Self::Foundry => ModelConfig::foundry(),
        }
    }
}

pub async fn create_adapter(
    provider: CloudProvider,
    config: ProviderConfig,
) -> Result<Box<dyn ProviderAdapter>> {
    match provider {
        CloudProvider::Anthropic => Ok(Box::new(AnthropicAdapter::new(config))),
        #[cfg(feature = "aws")]
        CloudProvider::Bedrock => Ok(Box::new(BedrockAdapter::from_env(config).await?)),
        #[cfg(feature = "gcp")]
        CloudProvider::Vertex => Ok(Box::new(VertexAdapter::from_env(config).await?)),
        #[cfg(feature = "azure")]
        CloudProvider::Foundry => Ok(Box::new(FoundryAdapter::from_env(config).await?)),
    }
}