langchainrust 0.2.13

A LangChain-inspired framework for building LLM applications in Rust. Supports OpenAI, Agents, Tools, Memory, Chains, RAG, BM25, Hybrid Retrieval, LangGraph, HyDE, Reranking, MultiQuery, and native Function Calling.
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

// tests/unit/providers.rs
//! Unit tests for LLM providers

use langchainrust::{
    AnthropicChat, AnthropicConfig, DeepSeekChat, DeepSeekConfig, MoonshotChat, MoonshotConfig,
    QwenChat, QwenConfig, ZhipuChat, ZhipuConfig,
};

#[test]
fn test_deepseek_config_default() {
    let config = DeepSeekConfig::default();
    assert_eq!(config.base_url, "https://api.deepseek.com/v1");
    assert_eq!(config.model, "deepseek-chat");
}

#[test]
fn test_deepseek_config_new() {
    let config = DeepSeekConfig::new("test-api-key");
    assert_eq!(config.api_key, "test-api-key");
}

#[test]
fn test_deepseek_config_with_model() {
    let config = DeepSeekConfig::new("test-key").with_model("deepseek-coder");
    assert_eq!(config.model, "deepseek-coder");
}

#[test]
fn test_deepseek_config_with_temperature() {
    let config = DeepSeekConfig::new("test-key").with_temperature(0.5);
    assert_eq!(config.temperature, Some(0.5));
}

#[test]
fn test_moonshot_config_default() {
    let config = MoonshotConfig::default();
    assert_eq!(config.base_url, "https://api.moonshot.cn/v1");
    assert_eq!(config.model, "moonshot-v1-8k");
}

#[test]
fn test_moonshot_config_with_model() {
    let config = MoonshotConfig::new("test-key").with_model("moonshot-v1-128k");
    assert_eq!(config.model, "moonshot-v1-128k");
}

#[test]
fn test_zhipu_config_default() {
    let config = ZhipuConfig::default();
    assert_eq!(config.base_url, "https://open.bigmodel.cn/api/paas/v4");
    assert_eq!(config.model, "glm-4-flash");
}

#[test]
fn test_zhipu_config_with_model() {
    let config = ZhipuConfig::new("test-key").with_model("glm-4");
    assert_eq!(config.model, "glm-4");
}

#[test]
fn test_qwen_config_default() {
    let config = QwenConfig::default();
    assert_eq!(
        config.base_url,
        "https://dashscope.aliyuncs.com/compatible-mode/v1"
    );
    assert_eq!(config.model, "qwen-plus");
}

#[test]
fn test_qwen_config_with_model() {
    let config = QwenConfig::new("test-key").with_model("qwen-max");
    assert_eq!(config.model, "qwen-max");
}

#[test]
fn test_anthropic_config_default() {
    let config = AnthropicConfig::default();
    assert_eq!(config.base_url, "https://api.anthropic.com/v1");
    assert_eq!(config.model, "claude-3-5-sonnet-20241022");
    assert_eq!(config.max_tokens, 4096);
}

#[test]
fn test_anthropic_config_new() {
    let config = AnthropicConfig::new("test-api-key");
    assert_eq!(config.api_key, "test-api-key");
}

#[test]
fn test_anthropic_config_with_model() {
    let config = AnthropicConfig::new("test-key").with_model("claude-3-opus-20240229");
    assert_eq!(config.model, "claude-3-opus-20240229");
}

#[test]
fn test_anthropic_config_with_max_tokens() {
    let config = AnthropicConfig::new("test-key").with_max_tokens(8192);
    assert_eq!(config.max_tokens, 8192);
}

#[test]
fn test_anthropic_config_with_temperature() {
    let config = AnthropicConfig::new("test-key").with_temperature(0.7);
    assert_eq!(config.temperature, Some(0.7));
}

#[test]
fn test_anthropic_config_with_system_prompt() {
    let config =
        AnthropicConfig::new("test-key").with_system_prompt("You are a helpful assistant.");
    assert_eq!(
        config.system_prompt,
        Some("You are a helpful assistant.".to_string())
    );
}