swiftide-integrations 0.33.0

Composable LLM agents, typed task graphs, and streaming RAG pipelines in 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

use config::DashscopeConfig;

use crate::openai;

mod config;

pub type Dashscope = openai::GenericOpenAI<DashscopeConfig>;
impl Dashscope {
    pub fn builder() -> DashscopeBuilder {
        DashscopeBuilder::default()
    }
}

pub type DashscopeBuilder = openai::GenericOpenAIBuilder<DashscopeConfig>;
pub type DashscopeBuilderError = openai::GenericOpenAIBuilderError;
pub use openai::{Options, OptionsBuilder, OptionsBuilderError};

impl Default for Dashscope {
    fn default() -> Self {
        Dashscope::builder().build().unwrap()
    }
}

#[cfg(test)]
mod test {
    use super::*;
    #[test]
    fn test_default_prompt_model() {
        let openai = Dashscope::builder()
            .default_prompt_model("qwen-long")
            .build()
            .unwrap();
        assert_eq!(
            openai.default_options.prompt_model,
            Some("qwen-long".to_string())
        );

        let openai = Dashscope::builder()
            .default_prompt_model("qwen-turbo")
            .build()
            .unwrap();
        assert_eq!(
            openai.default_options.prompt_model,
            Some("qwen-turbo".to_string())
        );
    }
}