rskit-llm-openai 0.2.0-alpha.3

OpenAI provider implementation for rskit
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
use super::*;

fn config() -> Config {
    Config {
        api_key: rskit_util::SecretString::new("sk-test"),
        base_url: "https://api.openai.com/v1".into(),
        model: "gpt-4o".into(),
        embedding_model: "text-embedding-3-small".into(),
        embedding_dimensions: Some(1536),
    }
}

#[test]
fn embedding_provider_builders_construct_providers() {
    let cfg = config();
    embedding_provider(&cfg).unwrap();
    embedding_provider_with_policy(&cfg, rskit_resilience::Policy::default()).unwrap();
}