From eb96aa635e693d620201e73f7a4a824150b07e3e Mon Sep 17 00:00:00 2001
From: JustSong <quanpengsong@gmail.com>
Date: Tue, 11 Feb 2025 00:20:55 +0800
Subject: [PATCH] feat: update OpenRouter channel name and add model list for
 OpenRouter adaptor

---
 relay/adaptor/openai/compatible.go            |  3 +++
 relay/adaptor/openrouter/constants.go         | 20 +++++++++++++++++++
 web/air/src/constants/channel.constants.js    |  2 +-
 web/berry/src/constants/ChannelConstants.js   |  2 +-
 .../src/constants/channel.constants.js        |  2 +-
 5 files changed, 26 insertions(+), 3 deletions(-)
 create mode 100644 relay/adaptor/openrouter/constants.go

diff --git a/relay/adaptor/openai/compatible.go b/relay/adaptor/openai/compatible.go
index bcbc825e..af65365a 100644
--- a/relay/adaptor/openai/compatible.go
+++ b/relay/adaptor/openai/compatible.go
@@ -12,6 +12,7 @@ import (
 	"github.com/songquanpeng/one-api/relay/adaptor/mistral"
 	"github.com/songquanpeng/one-api/relay/adaptor/moonshot"
 	"github.com/songquanpeng/one-api/relay/adaptor/novita"
+	"github.com/songquanpeng/one-api/relay/adaptor/openrouter"
 	"github.com/songquanpeng/one-api/relay/adaptor/siliconflow"
 	"github.com/songquanpeng/one-api/relay/adaptor/stepfun"
 	"github.com/songquanpeng/one-api/relay/adaptor/togetherai"
@@ -76,6 +77,8 @@ func GetCompatibleChannelMeta(channelType int) (string, []string) {
 		return "baiduv2", baiduv2.ModelList
 	case channeltype.XunfeiV2:
 		return "xunfeiv2", xunfeiv2.ModelList
+	case channeltype.OpenRouter:
+		return "openrouter", openrouter.ModelList
 	default:
 		return "openai", ModelList
 	}
diff --git a/relay/adaptor/openrouter/constants.go b/relay/adaptor/openrouter/constants.go
new file mode 100644
index 00000000..3ad8ed55
--- /dev/null
+++ b/relay/adaptor/openrouter/constants.go
@@ -0,0 +1,20 @@
+package openrouter
+
+var ModelList = []string{
+	"openai/gpt-3.5-turbo",
+	"openai/chatgpt-4o-latest",
+	"openai/o1",
+	"openai/o1-preview",
+	"openai/o1-mini",
+	"openai/o3-mini",
+	"google/gemini-2.0-flash-001",
+	"google/gemini-2.0-flash-thinking-exp:free",
+	"google/gemini-2.0-flash-lite-preview-02-05:free",
+	"google/gemini-2.0-pro-exp-02-05:free",
+	"google/gemini-flash-1.5-8b",
+	"anthropic/claude-3.5-sonnet",
+	"anthropic/claude-3.5-haiku",
+	"deepseek/deepseek-r1:free",
+	"deepseek/deepseek-r1",
+	"qwen/qwen-vl-plus:free",
+}
diff --git a/web/air/src/constants/channel.constants.js b/web/air/src/constants/channel.constants.js
index 9c256ba2..00a1d528 100644
--- a/web/air/src/constants/channel.constants.js
+++ b/web/air/src/constants/channel.constants.js
@@ -35,7 +35,7 @@ export const CHANNEL_OPTIONS = [
   { key: 8, text: '自定义渠道', value: 8, color: 'pink' },
   { key: 22, text: '知识库：FastGPT', value: 22, color: 'blue' },
   { key: 21, text: '知识库：AI Proxy', value: 21, color: 'purple' },
-  { key: 20, text: '代理：OpenRouter', value: 20, color: 'black' },
+  {key: 20, text: 'OpenRouter', value: 20, color: 'black'},
   { key: 2, text: '代理：API2D', value: 2, color: 'blue' },
   { key: 5, text: '代理：OpenAI-SB', value: 5, color: 'brown' },
   { key: 7, text: '代理：OhMyGPT', value: 7, color: 'purple' },
diff --git a/web/berry/src/constants/ChannelConstants.js b/web/berry/src/constants/ChannelConstants.js
index 83b60837..7907c8d4 100644
--- a/web/berry/src/constants/ChannelConstants.js
+++ b/web/berry/src/constants/ChannelConstants.js
@@ -217,7 +217,7 @@ export const CHANNEL_OPTIONS = {
   },
   20: {
     key: 20,
-    text: '代理：OpenRouter',
+      text: 'OpenRouter',
     value: 20,
     color: 'success'
   },
diff --git a/web/default/src/constants/channel.constants.js b/web/default/src/constants/channel.constants.js
index 99e605d6..f6c7d61c 100644
--- a/web/default/src/constants/channel.constants.js
+++ b/web/default/src/constants/channel.constants.js
@@ -67,7 +67,7 @@ export const CHANNEL_OPTIONS = [
   {key: 8, text: '自定义渠道', value: 8, color: 'pink'},
   {key: 22, text: '知识库：FastGPT', value: 22, color: 'blue'},
   {key: 21, text: '知识库：AI Proxy', value: 21, color: 'purple'},
-  {key: 20, text: '代理：OpenRouter', value: 20, color: 'black'},
+    {key: 20, text: 'OpenRouter', value: 20, color: 'black'},
   {key: 2, text: '代理：API2D', value: 2, color: 'blue'},
   {key: 5, text: '代理：OpenAI-SB', value: 5, color: 'brown'},
   {key: 7, text: '代理：OhMyGPT', value: 7, color: 'purple'},