From 5eae392dbd4fc789829ac2f47a24614dbc5db5e9 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Wed, 1 May 2024 06:01:10 +0800
Subject: refactore: add models for openai-compatible platforms (#471)

---
 src/client/common.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'src/client/common.rs')

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)]
-- 
cgit v1.2.3