diff options
| author | sigoden <sigoden@gmail.com> | 2024-05-01 06:01:10 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-05-01 06:01:10 +0800 |
| commit | 5eae392dbd4fc789829ac2f47a24614dbc5db5e9 (patch) | |
| tree | 331aded9b1badd5b66eb6b7afbbbe8f8227b87b8 /src/client/common.rs | |
| parent | 8aa18b59f5be467fcadc9aa679bce58d66e62793 (diff) | |
| download | aichat-5eae392dbd4fc789829ac2f47a24614dbc5db5e9.tar.gz | |
refactore: add models for openai-compatible platforms (#471)
Diffstat (limited to 'src/client/common.rs')
| -rw-r--r-- | src/client/common.rs | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/client/common.rs b/src/client/common.rs index 91d6bca..e889bc8 100644 --- a/src/client/common.rs +++ b/src/client/common.rs @@ -390,10 +390,11 @@ pub fn create_openai_compatible_client_config(client: &str) -> Result<Option<(St .find(|(name, _)| client == *name) { None => Ok(None), - Some((name, _)) => { + Some((name, api_base)) => { let mut config = json!({ "type": "openai-compatible", "name": name, + "api_base": api_base, }); let prompts = if ALL_CLIENT_MODELS.iter().any(|v| &v.platform == name) { vec![("api_key", "API Key:", false, PromptKind::String)] |