siumai 0.10.3

A unified LLM interface library for Rust
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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142

//! Tests for OpenAiCompatibleBuilder base_url functionality
//!
//! This module tests the custom base_url method that allows users to override
//! the default provider base URL for self-deployed services and alternative endpoints.

use crate::builder::LlmBuilder;
use crate::providers::openai_compatible::builder::OpenAiCompatibleBuilder;

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_base_url_method_exists() {
        // Test that the base_url method exists and can be called
        let builder = LlmBuilder::new();
        let openai_compatible_builder = OpenAiCompatibleBuilder::new(builder, "deepseek");

        let builder_with_url = openai_compatible_builder.base_url("https://custom-server.com/v1");

        // The method should return Self for chaining
        assert!(std::mem::size_of_val(&builder_with_url) > 0);
    }

    #[test]
    fn test_base_url_chaining() {
        // Test that base_url method supports method chaining
        let builder = LlmBuilder::new();
        let result = OpenAiCompatibleBuilder::new(builder, "deepseek")
            .api_key("test-key")
            .base_url("https://custom-server.com/v1")
            .model("deepseek-chat")
            .temperature(0.7);

        // Should be able to chain methods after base_url
        assert!(std::mem::size_of_val(&result) > 0);
    }

    #[test]
    fn test_base_url_accepts_string_types() {
        // Test that base_url accepts different string types
        let builder1 = LlmBuilder::new();
        let builder2 = LlmBuilder::new();
        let builder3 = LlmBuilder::new();

        // String literal
        let _result1 =
            OpenAiCompatibleBuilder::new(builder1, "deepseek").base_url("https://example.com/v1");

        // String
        let url = String::from("https://example.com/v1");
        let _result2 = OpenAiCompatibleBuilder::new(builder2, "deepseek").base_url(url);

        // &str
        let url_ref = "https://example.com/v1";
        let _result3 = OpenAiCompatibleBuilder::new(builder3, "deepseek").base_url(url_ref);
    }

    #[tokio::test]
    async fn test_base_url_integration_with_build() {
        // Test that custom base_url is used during build process
        // The build should succeed with valid configuration

        let builder = LlmBuilder::new();
        let openai_compatible_builder = OpenAiCompatibleBuilder::new(builder, "deepseek")
            .base_url("https://custom-deepseek-server.com/v1")
            .api_key("test-key")
            .model("deepseek-chat");

        // The build should succeed with proper configuration
        let result = openai_compatible_builder.build().await;

        // Should succeed in creating the client (configuration is valid)
        assert!(
            result.is_ok(),
            "Build should succeed with valid configuration"
        );

        // Verify the client was created successfully
        let _client = result.unwrap();
    }

    #[test]
    fn test_base_url_documentation_examples() {
        // Test the examples from the documentation

        // Example 1: Self-deployed server
        let builder1 = LlmBuilder::new();
        let _client_builder1 = builder1
            .deepseek()
            .api_key("your-api-key")
            .base_url("https://my-deepseek-server.com/v1")
            .model("deepseek-chat");

        // Example 2: Alternative endpoint
        let builder2 = LlmBuilder::new();
        let _client_builder2 = builder2
            .openrouter()
            .api_key("your-api-key")
            .base_url("https://openrouter.ai/api/v1")
            .model("openai/gpt-4");
    }

    #[test]
    fn test_base_url_empty_string() {
        // Test behavior with empty string (should be allowed but may cause issues later)
        let builder = LlmBuilder::new();
        let _result = OpenAiCompatibleBuilder::new(builder, "deepseek").base_url("");

        // Should not panic during builder creation
    }

    #[test]
    fn test_base_url_with_trailing_slash() {
        // Test that URLs with trailing slashes are handled correctly
        let builder = LlmBuilder::new();
        let _result =
            OpenAiCompatibleBuilder::new(builder, "deepseek").base_url("https://example.com/v1/");

        // Should not panic during builder creation
    }

    #[test]
    fn test_base_url_with_different_protocols() {
        // Test different URL protocols
        let builder1 = LlmBuilder::new();
        let builder2 = LlmBuilder::new();
        let builder3 = LlmBuilder::new();

        // HTTPS
        let _result1 =
            OpenAiCompatibleBuilder::new(builder1, "deepseek").base_url("https://example.com/v1");

        // HTTP (for local development)
        let _result2 =
            OpenAiCompatibleBuilder::new(builder2, "deepseek").base_url("http://localhost:8080/v1");

        // Custom port
        let _result3 = OpenAiCompatibleBuilder::new(builder3, "deepseek")
            .base_url("https://example.com:8443/v1");
    }
}