rig-core 0.36.0

An opinionated library for building LLM powered applications.
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

mod anthropic;
mod coding;
mod general;

use rig::providers::zai;

pub(crate) fn api_key() -> String {
    std::env::var("ZAI_API_KEY").expect("ZAI_API_KEY should be set")
}

pub(crate) fn general_client() -> zai::Client {
    zai::Client::builder()
        .api_key(api_key())
        .general()
        .build()
        .expect("Z.AI general client should build")
}

pub(crate) fn coding_client() -> zai::Client {
    zai::Client::builder()
        .api_key(api_key())
        .coding()
        .build()
        .expect("Z.AI coding client should build")
}

pub(crate) fn anthropic_client() -> zai::AnthropicClient {
    zai::AnthropicClient::builder()
        .api_key(api_key())
        .general()
        .build()
        .expect("Z.AI Anthropic-compatible client should build")
}