aquaregia 0.1.6

Provider-agnostic Rust toolkit for AI apps and agents.
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

use aquaregia::{ErrorCode, GenerateTextRequest, LlmClient, Message, anthropic};
use wiremock::matchers::{method, path};
use wiremock::{Mock, MockServer, ResponseTemplate};

fn anthropic_request() -> GenerateTextRequest<aquaregia::Anthropic> {
    GenerateTextRequest::builder(anthropic("claude-3-5-haiku-latest"))
        .message(Message::user_text("hi"))
        .temperature(0.2)
        .max_output_tokens(32)
        .build()
        .expect("request should build")
}

#[tokio::test]
async fn anthropic_429_maps_to_rate_limited() {
    let server = MockServer::start().await;
    Mock::given(method("POST"))
        .and(path("/v1/messages"))
        .respond_with(ResponseTemplate::new(429).set_body_string("rate limited"))
        .expect(1)
        .mount(&server)
        .await;

    let client = LlmClient::anthropic("test-anthropic-key")
        .base_url(server.uri())
        .api_version("2023-06-01")
        .max_retries(0)
        .build()
        .expect("client should build");

    let err = client
        .generate(anthropic_request())
        .await
        .expect_err("request should fail");

    assert_eq!(err.code, ErrorCode::RateLimited);
    assert_eq!(err.status, Some(429));
    assert!(err.retryable);
}