From f94768dcf07518ce523931fa04d176b934bede48 Mon Sep 17 00:00:00 2001
From: lidongjie <limingliulise@outlook.com>
Date: Sat, 12 Oct 2024 18:32:50 +0800
Subject: [PATCH] =?UTF-8?q?add:=E6=B7=BB=E5=8A=A0=E9=98=B6=E8=B7=83?=
 =?UTF-8?q?=E6=98=9F=E8=BE=B0=E4=BE=9B=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/stepfun/stepfun.yaml        | 2 +-
 gateway/apinto/profession.go                                  | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/ai-provider/model-runtime/model-providers/stepfun/stepfun.yaml b/ai-provider/model-runtime/model-providers/stepfun/stepfun.yaml
index 6f49d733..5466ec4e 100644
--- a/ai-provider/model-runtime/model-providers/stepfun/stepfun.yaml
+++ b/ai-provider/model-runtime/model-providers/stepfun/stepfun.yaml
@@ -79,4 +79,4 @@ model_credential_schema:
           label:
             en_US: Tool Call
             zh_Hans: Tool Call
-address: https://api.openai.com
\ No newline at end of file
+address: https://api.stepfun.com
\ No newline at end of file
diff --git a/gateway/apinto/profession.go b/gateway/apinto/profession.go
index 20433e95..174ca078 100644
--- a/gateway/apinto/profession.go
+++ b/gateway/apinto/profession.go
@@ -79,6 +79,10 @@ var dynamicResourceMap = map[string]Worker{
 		Profession: ProfessionAIProvider,
 		Driver:     "baichuan",
 	},
+	"stepfun": {
+		Profession: ProfessionAIProvider,
+		Driver:     "stepfun",
+	},
 }
 
 type Worker struct {