agent-proxy-rust-core 1.0.0

Core middleware trait, axum server engine, upstream forwarding, and auth for agent-proxy
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

//! Shared test helpers for the core crate and downstream middleware crates.

use bytes::Bytes;
use http::{HeaderMap, Method};

use crate::types::{AgentType, ApiFormat, ChannelConfig, ConnectionContext, ProxyRequest};

/// Creates a minimal test [`ProxyRequest`] for `/v1/messages`.
#[must_use]
pub fn test_proxy_request() -> ProxyRequest {
    ProxyRequest::new(
        Method::POST,
        "/v1/messages".into(),
        HeaderMap::new(),
        Bytes::from(
            r#"{"model":"claude-sonnet","max_tokens":1024,"messages":[{"role":"user","content":"hello"}]}"#,
        ),
    )
}

/// Creates a streaming test [`ProxyRequest`].
#[must_use]
pub fn test_proxy_request_streaming() -> ProxyRequest {
    ProxyRequest::new(
        Method::POST,
        "/v1/messages".into(),
        HeaderMap::new(),
        Bytes::from(
            r#"{"model":"claude-sonnet","max_tokens":1024,"messages":[{"role":"user","content":"hello"}],"stream":true}"#,
        ),
    )
}

/// Creates a test [`ProxyRequest`] with a custom body.
#[must_use]
pub fn test_proxy_request_with_body(body: impl Into<Bytes>) -> ProxyRequest {
    ProxyRequest::new(
        Method::POST,
        "/v1/messages".into(),
        HeaderMap::new(),
        body.into(),
    )
}

/// Creates a minimal test [`ConnectionContext`].
#[must_use]
pub fn test_connection_context() -> ConnectionContext {
    ConnectionContext::new(
        1,
        AgentType::Claude,
        None,
        Some(ApiFormat::AnthropicMessages),
    )
}

/// Creates a test [`ConnectionContext`] with a specific role.
#[must_use]
pub fn test_connection_context_with_role(role: impl Into<String>) -> ConnectionContext {
    ConnectionContext::new(
        1,
        AgentType::Claude,
        Some(role.into()),
        Some(ApiFormat::AnthropicMessages),
    )
}

/// Creates a test [`ChannelConfig`] pointing to a local upstream.
#[must_use]
pub fn test_channel_config(base_url: impl Into<String>) -> ChannelConfig {
    ChannelConfig {
        url: base_url.into(),
        api_key: secrecy::SecretString::from("sk-test-channel-key"),
        protocol: ApiFormat::AnthropicMessages,
        name: "test-channel".into(),
        rewrite_path: None,
    }
}

/// Creates a test [`ChannelConfig`] for `OpenAI` Chat.
#[must_use]
pub fn test_channel_config_openai(base_url: impl Into<String>) -> ChannelConfig {
    ChannelConfig {
        url: base_url.into(),
        api_key: secrecy::SecretString::from("sk-test-openai-key"),
        protocol: ApiFormat::OpenaiChat,
        name: "test-openai-channel".into(),
        rewrite_path: None,
    }
}