From 974c5cf8a2de94cc22d0ed833eb5cee8600db382 Mon Sep 17 00:00:00 2001
From: lidongjie <limingliulise@outlook.com>
Date: Sat, 12 Oct 2024 18:03:19 +0800
Subject: [PATCH] =?UTF-8?q?add:=E6=B7=BB=E5=8A=A0baichuan=E4=BE=9B?=
 =?UTF-8?q?=E5=BA=94=E5=95=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../model-runtime/model-providers/baichuan/baichuan.yaml      | 2 +-
 gateway/apinto/profession.go                                  | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/ai-provider/model-runtime/model-providers/baichuan/baichuan.yaml b/ai-provider/model-runtime/model-providers/baichuan/baichuan.yaml
index 285ac656..dcadb657 100644
--- a/ai-provider/model-runtime/model-providers/baichuan/baichuan.yaml
+++ b/ai-provider/model-runtime/model-providers/baichuan/baichuan.yaml
@@ -27,4 +27,4 @@ provider_credential_schema:
       placeholder:
         zh_Hans: 在此输入您的 API Key
         en_US: Enter your API Key
-address: https://api.openai.com
+address: https://api.baichuan-ai.com
diff --git a/gateway/apinto/profession.go b/gateway/apinto/profession.go
index 664a482b..20433e95 100644
--- a/gateway/apinto/profession.go
+++ b/gateway/apinto/profession.go
@@ -75,6 +75,10 @@ var dynamicResourceMap = map[string]Worker{
 		Profession: ProfessionAIProvider,
 		Driver:     "mistralai",
 	},
+	"baichuan": {
+		Profession: ProfessionAIProvider,
+		Driver:     "baichuan",
+	},
 }
 
 type Worker struct {