opendev-http 0.1.4

HTTP client, authentication, and provider adapters for OpenDev
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
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

use super::*;
use crate::adapters::base::ProviderAdapter;

#[test]
fn test_provider_name() {
    let adapter = BedrockAdapter::new("anthropic.claude-3-sonnet-20240229-v1:0");
    assert_eq!(adapter.provider_name(), "bedrock");
}

#[test]
fn test_api_url_format() {
    let adapter =
        BedrockAdapter::with_region("anthropic.claude-3-sonnet-20240229-v1:0", "us-west-2");
    assert_eq!(
        adapter.api_url(),
        "https://bedrock-runtime.us-west-2.amazonaws.com/model/anthropic.claude-3-sonnet-20240229-v1:0/invoke"
    );
}

#[test]
fn test_api_url_default_region() {
    let adapter =
        BedrockAdapter::with_region("anthropic.claude-3-haiku-20240307-v1:0", "us-east-1");
    assert!(adapter.api_url().contains("us-east-1"));
}

#[test]
fn test_model_id() {
    let adapter = BedrockAdapter::new("anthropic.claude-3-sonnet-20240229-v1:0");
    assert_eq!(
        adapter.model_id(),
        "anthropic.claude-3-sonnet-20240229-v1:0"
    );
}

#[test]
fn test_region() {
    let adapter = BedrockAdapter::with_region("model", "eu-west-1");
    assert_eq!(adapter.region(), "eu-west-1");
}

#[test]
fn test_convert_request_removes_unsupported_fields() {
    let adapter = BedrockAdapter::with_region("model-id", "us-east-1");
    let payload = json!({
        "model": "model-id",
        "messages": [{"role": "user", "content": "Hi"}],
        "n": 1,
        "frequency_penalty": 0.5,
        "presence_penalty": 0.5,
        "logprobs": true,
        "stream": true
    });
    let result = adapter.convert_request(payload);
    assert!(result.get("model").is_none());
    assert!(result.get("n").is_none());
    assert!(result.get("frequency_penalty").is_none());
    assert!(result.get("presence_penalty").is_none());
    assert!(result.get("logprobs").is_none());
    assert!(result.get("stream").is_none());
    assert_eq!(result["anthropic_version"], "bedrock-2023-05-31");
}

#[test]
fn test_convert_request_sets_max_tokens() {
    let adapter = BedrockAdapter::with_region("model-id", "us-east-1");
    let payload = json!({
        "messages": [{"role": "user", "content": "Hi"}]
    });
    let result = adapter.convert_request(payload);
    assert_eq!(result["max_tokens"], 4096);
}

#[test]
fn test_convert_request_preserves_custom_max_tokens() {
    let adapter = BedrockAdapter::with_region("model-id", "us-east-1");
    let payload = json!({
        "messages": [{"role": "user", "content": "Hi"}],
        "max_tokens": 8192
    });
    let result = adapter.convert_request(payload);
    assert_eq!(result["max_tokens"], 8192);
}

#[test]
fn test_convert_request_converts_max_completion_tokens() {
    let adapter = BedrockAdapter::with_region("model-id", "us-east-1");
    let payload = json!({
        "messages": [{"role": "user", "content": "Hi"}],
        "max_completion_tokens": 2048
    });
    let result = adapter.convert_request(payload);
    assert_eq!(result["max_tokens"], 2048);
    assert!(result.get("max_completion_tokens").is_none());
}

#[test]
fn test_extra_headers() {
    let adapter = BedrockAdapter::with_region("model-id", "us-east-1");
    let headers = adapter.extra_headers();
    assert!(
        headers
            .iter()
            .any(|(k, v)| k == "Content-Type" && v == "application/json")
    );
    assert!(
        headers
            .iter()
            .any(|(k, v)| k == "Accept" && v == "application/json")
    );
}

#[test]
fn test_build_url() {
    let url = BedrockAdapter::build_url("ap-southeast-1", "anthropic.claude-v2");
    assert_eq!(
        url,
        "https://bedrock-runtime.ap-southeast-1.amazonaws.com/model/anthropic.claude-v2/invoke"
    );
}