From 8568957afae2570e2ac963d742d32354e1ca100f Mon Sep 17 00:00:00 2001
From: lidongjie <limingliulise@outlook.com>
Date: Fri, 11 Oct 2024 17:16:16 +0800
Subject: [PATCH] =?UTF-8?q?add:=E6=B7=BB=E5=8A=A0fireworks?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

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

diff --git a/ai-provider/model-runtime/model-providers/fireworks/fireworks.yaml b/ai-provider/model-runtime/model-providers/fireworks/fireworks.yaml
index fcfba8e4..be734ac4 100644
--- a/ai-provider/model-runtime/model-providers/fireworks/fireworks.yaml
+++ b/ai-provider/model-runtime/model-providers/fireworks/fireworks.yaml
@@ -28,4 +28,4 @@ provider_credential_schema:
       placeholder:
         zh_Hans: 在此输入您的 API Key
         en_US: Enter your API Key
-address: https://api.openai.com
\ No newline at end of file
+address: https://api.fireworks.ai
\ No newline at end of file
diff --git a/gateway/apinto/profession.go b/gateway/apinto/profession.go
index 4152879c..506346c4 100644
--- a/gateway/apinto/profession.go
+++ b/gateway/apinto/profession.go
@@ -63,6 +63,10 @@ var dynamicResourceMap = map[string]Worker{
 		Profession: ProfessionAIProvider,
 		Driver:     "zhipuai",
 	},
+	"fireworks": {
+		Profession: ProfessionAIProvider,
+		Driver:     "fireworks",
+	},
 }
 
 type Worker struct {