feat: add OpenRouter balance update functionality and improve code formatting

controller/channel-billing.go

@@ -112,6 +112,13 @@ type DeepSeekUsageResponse struct {
 	} `json:"balance_infos"`
 }
 
+type OpenRouterResponse struct {
+	Data struct {
+		TotalCredits float64 `json:"total_credits"`
+		TotalUsage   float64 `json:"total_usage"`
+	} `json:"data"`
+}
+
 // GetAuthHeader get auth header
 func GetAuthHeader(token string) http.Header {
 	h := http.Header{}
@@ -285,6 +292,22 @@ func updateChannelDeepSeekBalance(channel *model.Channel) (float64, error) {
 	return balance, nil
 }
 
+func updateChannelOpenRouterBalance(channel *model.Channel) (float64, error) {
+	url := "https://openrouter.ai/api/v1/credits"
+	body, err := GetResponseBody("GET", url, channel, GetAuthHeader(channel.Key))
+	if err != nil {
+		return 0, err
+	}
+	response := OpenRouterResponse{}
+	err = json.Unmarshal(body, &response)
+	if err != nil {
+		return 0, err
+	}
+	balance := response.Data.TotalCredits - response.Data.TotalUsage
+	channel.UpdateBalance(balance)
+	return balance, nil
+}
+
 func updateChannelBalance(channel *model.Channel) (float64, error) {
 	baseURL := channeltype.ChannelBaseURLs[channel.Type]
 	if channel.GetBaseURL() == "" {
@@ -313,6 +336,8 @@ func updateChannelBalance(channel *model.Channel) (float64, error) {
 		return updateChannelSiliconFlowBalance(channel)
 	case channeltype.DeepSeek:
 		return updateChannelDeepSeekBalance(channel)
+	case channeltype.OpenRouter:
+		return updateChannelOpenRouterBalance(channel)
 	default:
 		return 0, errors.New("尚未实现")
 	}

controller/channel-test.go

@@ -153,6 +153,7 @@ func testChannel(ctx context.Context, channel *model.Channel, request *relaymode
 	rawResponse := w.Body.String()
 	_, responseMessage, err = parseTestResponse(rawResponse)
 	if err != nil {
+		logger.SysError(fmt.Sprintf("failed to parse error: %s, \nresponse: %s", err.Error(), rawResponse))
 		return "", err, nil
 	}
 	result := w.Result()

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
 	}

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",
+}

relay/adaptor/xai/constants.go

@@ -1,5 +1,14 @@
 package xai
 
+//https://console.x.ai/
+
 var ModelList = []string{
+	"grok-2",
+	"grok-vision-beta",
+	"grok-2-vision-1212",
+	"grok-2-vision",
+	"grok-2-vision-latest",
+	"grok-2-1212",
+	"grok-2-latest",
 	"grok-beta",
 }

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' },

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'
   },

web/default/src/components/ChannelsTable.js

@@ -57,6 +57,8 @@ function renderBalance(type, balance, t) {
       return <span>¥{balance.toFixed(2)}</span>;
     case 13: // AIGC2D
       return <span>{renderNumber(balance)}</span>;
+    case 20: // OpenRouter
+      return <span>${balance.toFixed(2)}</span>;
     case 36: // DeepSeek
       return <span>¥{balance.toFixed(2)}</span>;
     case 44: // SiliconFlow

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'},