anthropic-rust-sdk 0.1.0

Anthropic Claude API client for Rust, synced from anthropic-sdk-typescript
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

//! Beta Messages API.

use crate::client::Anthropic;
use crate::core::error::Error;
use crate::resources::messages::{
    Message, MessageCountTokensParams, MessageCreateParams, MessageTokensCount,
};
use serde_json::Value;

pub struct BetaMessages<'a> {
    client: &'a Anthropic,
    beta_headers: Vec<String>,
}

impl<'a> BetaMessages<'a> {
    pub(crate) fn new(client: &'a Anthropic, beta_headers: Vec<String>) -> Self {
        Self {
            client,
            beta_headers,
        }
    }

    pub async fn create(&self, params: MessageCreateParams) -> Result<Message, Error> {
        let mut body = serde_json::to_value(&params).map_err(|e| {
            crate::core::error::AnthropicError(format!("serialize error: {e}"))
        })?;
        if let Value::Object(ref mut map) = body {
            map.insert("stream".to_string(), Value::Bool(false));
        }
        self.client
            .post_beta("/v1/messages", &body, &self.beta_headers)
            .await
    }

    pub async fn count_tokens(
        &self,
        params: MessageCountTokensParams,
    ) -> Result<MessageTokensCount, Error> {
        self.client
            .post_beta("/v1/messages/count_tokens", &params, &self.beta_headers)
            .await
    }
}