litellm-rs 0.4.16

A high-performance AI Gateway written in Rust, providing OpenAI-compatible APIs with intelligent routing, load balancing, and enterprise features
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

//! Tests for Codestral provider

use super::*;

#[test]
fn test_config_default() {
    let config = CodestralConfig::default();
    assert!(config.api_key.is_none());
    assert_eq!(config.timeout, 60);
}

#[test]
fn test_get_api_base_default() {
    let config = CodestralConfig::default();
    assert_eq!(config.get_api_base(), "https://codestral.mistral.ai/v1");
}

#[test]
fn test_model_info() {
    let info = get_model_info("codestral-latest");
    assert!(info.is_some());
    assert!(info.unwrap().supports_fim);
}

// Note: Error conversion tests removed - CodestralError is now a type alias to ProviderError

#[test]
fn test_fim_request_serialization() {
    use super::provider::FimRequest;
    let request = FimRequest {
        model: "codestral-latest".to_string(),
        prompt: "def hello():".to_string(),
        suffix: Some("    return greeting".to_string()),
        temperature: Some(0.7),
        max_tokens: Some(100),
        stop: None,
    };

    let json = serde_json::to_value(&request).unwrap();
    assert_eq!(json["model"], "codestral-latest");
    assert_eq!(json["prompt"], "def hello():");
}