ai_chain_openai_compatible/chatgpt/
config.rs

1use async_openai::config::{Config, OPENAI_API_BASE};
2use ai_chain::tokens::Tokenizer;
3
4/// config extension
5pub trait OAIConfig: Config + Send + Sync + 'static {
6
7
8    fn create() -> Self;
9
10    fn with_api_key<S: Into<String>>(&mut self, api_key: S) -> Self;
11
12    /// To use a API base url different from default [OPENAI_API_BASE]
13    fn with_api_base<S: Into<String>>(&mut self, api_base: S) -> Self;
14
15    fn model_config() ->(String, Vec<String>);
16
17
18    fn tokenizer(&self)->Box<dyn Tokenizer>;
19}
20
21trait OpenAICompatibleTokenizer: Tokenizer {
22
23}
24