mirror of
				https://github.com/songquanpeng/one-api.git
				synced 2025-11-04 15:53:42 +08:00 
			
		
		
		
	Compare commits
	
		
			30 Commits
		
	
	
		
			v0.6.6-alp
			...
			v0.6.7-alp
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 
						 | 
					b53e00a9b3 | ||
| 
						 | 
					332c8db0b3 | ||
| 
						 | 
					3be28da57b | ||
| 
						 | 
					fa74ba0eaa | ||
| 
						 | 
					a9211d66f6 | ||
| 
						 | 
					07b2fd58d6 | ||
| 
						 | 
					0acee9a065 | ||
| 
						 | 
					f965469e8a | ||
| 
						 | 
					03ea60532a | ||
| 
						 | 
					2457d00afb | ||
| 
						 | 
					91b80ae879 | ||
| 
						 | 
					2720e1a358 | ||
| 
						 | 
					71f4403fd5 | ||
| 
						 | 
					1f76c80553 | ||
| 
						 | 
					7e027d2bd0 | ||
| 
						 | 
					30f373b623 | ||
| 
						 | 
					1c2654320e | ||
| 
						 | 
					6cffb116b7 | ||
| 
						 | 
					a84c7b38b7 | ||
| 
						 | 
					1bd14af47b | ||
| 
						 | 
					6170b91d1c | ||
| 
						 | 
					04b49aa0ec | ||
| 
						 | 
					ef88497f25 | ||
| 
						 | 
					007906216d | ||
| 
						 | 
					e64e7707a0 | ||
| 
						 | 
					ea210b6ed7 | ||
| 
						 | 
					9026ec7510 | ||
| 
						 | 
					c317872097 | ||
| 
						 | 
					da0842272c | ||
| 
						 | 
					0a650b85b4 | 
@@ -84,6 +84,10 @@ _✨ 通过标准的 OpenAI API 格式访问所有的大模型,开箱即用 
 | 
			
		||||
   + [x] [阶跃星辰](https://platform.stepfun.com/)
 | 
			
		||||
   + [x] [Coze](https://www.coze.com/)
 | 
			
		||||
   + [x] [Cohere](https://cohere.com/)
 | 
			
		||||
   + [x] [DeepSeek](https://www.deepseek.com/)
 | 
			
		||||
   + [x] [Cloudflare Workers AI](https://developers.cloudflare.com/workers-ai/)
 | 
			
		||||
   + [x] [DeepL](https://www.deepl.com/)
 | 
			
		||||
   + [x] [together.ai](https://www.together.ai/)
 | 
			
		||||
2. 支持配置镜像以及众多[第三方代理服务](https://iamazing.cn/page/openai-api-third-party-services)。
 | 
			
		||||
3. 支持通过**负载均衡**的方式访问多个渠道。
 | 
			
		||||
4. 支持 **stream 模式**,可以通过流式传输实现打字机效果。
 | 
			
		||||
 
 | 
			
		||||
@@ -1,13 +0,0 @@
 | 
			
		||||
package ctxkey
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	ConfigPrefix = "cfg_"
 | 
			
		||||
 | 
			
		||||
	ConfigAPIVersion = ConfigPrefix + "api_version"
 | 
			
		||||
	ConfigLibraryID  = ConfigPrefix + "library_id"
 | 
			
		||||
	ConfigPlugin     = ConfigPrefix + "plugin"
 | 
			
		||||
	ConfigSK         = ConfigPrefix + "sk"
 | 
			
		||||
	ConfigAK         = ConfigPrefix + "ak"
 | 
			
		||||
	ConfigRegion     = ConfigPrefix + "region"
 | 
			
		||||
	ConfigUserID     = ConfigPrefix + "user_id"
 | 
			
		||||
)
 | 
			
		||||
@@ -1,6 +1,7 @@
 | 
			
		||||
package ctxkey
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	Config            = "config"
 | 
			
		||||
	Id                = "id"
 | 
			
		||||
	Username          = "username"
 | 
			
		||||
	Role              = "role"
 | 
			
		||||
 
 | 
			
		||||
@@ -2,6 +2,7 @@ package helper
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/random"
 | 
			
		||||
	"html/template"
 | 
			
		||||
	"log"
 | 
			
		||||
@@ -105,6 +106,11 @@ func GenRequestID() string {
 | 
			
		||||
	return GetTimeString() + random.GetRandomNumberString(8)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func GetResponseID(c *gin.Context) string {
 | 
			
		||||
	logID := c.GetString(RequestIdKey)
 | 
			
		||||
	return fmt.Sprintf("chatcmpl-%s", logID)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func Max(a int, b int) int {
 | 
			
		||||
	if a >= b {
 | 
			
		||||
		return a
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										5
									
								
								common/helper/key.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										5
									
								
								common/helper/key.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,5 @@
 | 
			
		||||
package helper
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	RequestIdKey = "X-Oneapi-Request-Id"
 | 
			
		||||
)
 | 
			
		||||
@@ -1,7 +1,3 @@
 | 
			
		||||
package logger
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	RequestIdKey = "X-Oneapi-Request-Id"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var LogDir string
 | 
			
		||||
 
 | 
			
		||||
@@ -87,7 +87,7 @@ func logHelper(ctx context.Context, level string, msg string) {
 | 
			
		||||
	if level == loggerINFO {
 | 
			
		||||
		writer = gin.DefaultWriter
 | 
			
		||||
	}
 | 
			
		||||
	id := ctx.Value(RequestIdKey)
 | 
			
		||||
	id := ctx.Value(helper.RequestIdKey)
 | 
			
		||||
	if id == nil {
 | 
			
		||||
		id = helper.GenRequestID()
 | 
			
		||||
	}
 | 
			
		||||
 
 | 
			
		||||
@@ -5,6 +5,15 @@ import (
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"errors"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"net/http/httptest"
 | 
			
		||||
	"net/url"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"sync"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/config"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/ctxkey"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/logger"
 | 
			
		||||
@@ -18,14 +27,6 @@ import (
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/meta"
 | 
			
		||||
	relaymodel "github.com/songquanpeng/one-api/relay/model"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/relaymode"
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"net/http/httptest"
 | 
			
		||||
	"net/url"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"strings"
 | 
			
		||||
	"sync"
 | 
			
		||||
	"time"
 | 
			
		||||
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
)
 | 
			
		||||
@@ -57,6 +58,8 @@ func testChannel(channel *model.Channel) (err error, openaiErr *relaymodel.Error
 | 
			
		||||
	c.Request.Header.Set("Content-Type", "application/json")
 | 
			
		||||
	c.Set(ctxkey.Channel, channel.Type)
 | 
			
		||||
	c.Set(ctxkey.BaseURL, channel.GetBaseURL())
 | 
			
		||||
	cfg, _ := channel.LoadConfig()
 | 
			
		||||
	c.Set(ctxkey.Config, cfg)
 | 
			
		||||
	middleware.SetupContextForSelectedChannel(c, channel, "")
 | 
			
		||||
	meta := meta.GetByContext(c)
 | 
			
		||||
	apiType := channeltype.ToAPIType(channel.Type)
 | 
			
		||||
@@ -67,6 +70,7 @@ func testChannel(channel *model.Channel) (err error, openaiErr *relaymodel.Error
 | 
			
		||||
	adaptor.Init(meta)
 | 
			
		||||
	var modelName string
 | 
			
		||||
	modelList := adaptor.GetModelList()
 | 
			
		||||
	modelMap := channel.GetModelMapping()
 | 
			
		||||
	if len(modelList) != 0 {
 | 
			
		||||
		modelName = modelList[0]
 | 
			
		||||
	}
 | 
			
		||||
@@ -75,6 +79,9 @@ func testChannel(channel *model.Channel) (err error, openaiErr *relaymodel.Error
 | 
			
		||||
		if len(modelNames) > 0 {
 | 
			
		||||
			modelName = modelNames[0]
 | 
			
		||||
		}
 | 
			
		||||
		if modelMap != nil && modelMap[modelName] != "" {
 | 
			
		||||
			modelName = modelMap[modelName]
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	request := buildTestRequest()
 | 
			
		||||
	request.Model = modelName
 | 
			
		||||
 
 | 
			
		||||
@@ -4,6 +4,9 @@ import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"context"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/config"
 | 
			
		||||
@@ -16,8 +19,6 @@ import (
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/controller"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/model"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/relaymode"
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// https://platform.openai.com/docs/api-reference/chat
 | 
			
		||||
@@ -47,6 +48,7 @@ func Relay(c *gin.Context) {
 | 
			
		||||
		logger.Debugf(ctx, "request body: %s", string(requestBody))
 | 
			
		||||
	}
 | 
			
		||||
	channelId := c.GetInt(ctxkey.ChannelId)
 | 
			
		||||
	userId := c.GetInt("id")
 | 
			
		||||
	bizErr := relayHelper(c, relayMode)
 | 
			
		||||
	if bizErr == nil {
 | 
			
		||||
		monitor.Emit(channelId, true)
 | 
			
		||||
@@ -56,8 +58,8 @@ func Relay(c *gin.Context) {
 | 
			
		||||
	channelName := c.GetString(ctxkey.ChannelName)
 | 
			
		||||
	group := c.GetString(ctxkey.Group)
 | 
			
		||||
	originalModel := c.GetString(ctxkey.OriginalModel)
 | 
			
		||||
	go processChannelRelayError(ctx, channelId, channelName, bizErr)
 | 
			
		||||
	requestId := c.GetString(logger.RequestIdKey)
 | 
			
		||||
	go processChannelRelayError(ctx, userId, channelId, channelName, bizErr)
 | 
			
		||||
	requestId := c.GetString(helper.RequestIdKey)
 | 
			
		||||
	retryTimes := config.RetryTimes
 | 
			
		||||
	if !shouldRetry(c, bizErr.StatusCode) {
 | 
			
		||||
		logger.Errorf(ctx, "relay error happen, status code is %d, won't retry in this case", bizErr.StatusCode)
 | 
			
		||||
@@ -83,7 +85,7 @@ func Relay(c *gin.Context) {
 | 
			
		||||
		channelId := c.GetInt(ctxkey.ChannelId)
 | 
			
		||||
		lastFailedChannelId = channelId
 | 
			
		||||
		channelName := c.GetString(ctxkey.ChannelName)
 | 
			
		||||
		go processChannelRelayError(ctx, channelId, channelName, bizErr)
 | 
			
		||||
		go processChannelRelayError(ctx, userId, channelId, channelName, bizErr)
 | 
			
		||||
	}
 | 
			
		||||
	if bizErr != nil {
 | 
			
		||||
		if bizErr.StatusCode == http.StatusTooManyRequests {
 | 
			
		||||
@@ -115,8 +117,8 @@ func shouldRetry(c *gin.Context, statusCode int) bool {
 | 
			
		||||
	return true
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func processChannelRelayError(ctx context.Context, channelId int, channelName string, err *model.ErrorWithStatusCode) {
 | 
			
		||||
	logger.Errorf(ctx, "relay error (channel #%d): %s", channelId, err.Message)
 | 
			
		||||
func processChannelRelayError(ctx context.Context, userId int, channelId int, channelName string, err *model.ErrorWithStatusCode) {
 | 
			
		||||
	logger.Errorf(ctx, "relay error (channel id %d, user id: %d): %s", channelId, userId, err.Message)
 | 
			
		||||
	// https://platform.openai.com/docs/guides/error-codes/api-errors
 | 
			
		||||
	if monitor.ShouldDisableChannel(&err.Error, err.StatusCode) {
 | 
			
		||||
		monitor.DisableChannel(channelId, channelName, err.Message)
 | 
			
		||||
 
 | 
			
		||||
@@ -6,6 +6,8 @@ import (
 | 
			
		||||
	"github.com/songquanpeng/one-api/common"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/config"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/ctxkey"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/helper"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/logger"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/random"
 | 
			
		||||
	"github.com/songquanpeng/one-api/model"
 | 
			
		||||
	"net/http"
 | 
			
		||||
@@ -109,6 +111,7 @@ func Logout(c *gin.Context) {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func Register(c *gin.Context) {
 | 
			
		||||
	ctx := c.Request.Context()
 | 
			
		||||
	if !config.RegisterEnabled {
 | 
			
		||||
		c.JSON(http.StatusOK, gin.H{
 | 
			
		||||
			"message": "管理员关闭了新用户注册",
 | 
			
		||||
@@ -173,6 +176,28 @@ func Register(c *gin.Context) {
 | 
			
		||||
		})
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	go func() {
 | 
			
		||||
		err := user.ValidateAndFill()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			logger.Errorf(ctx, "user.ValidateAndFill failed: %w", err)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
		cleanToken := model.Token{
 | 
			
		||||
			UserId:         user.Id,
 | 
			
		||||
			Name:           "default",
 | 
			
		||||
			Key:            random.GenerateKey(),
 | 
			
		||||
			CreatedTime:    helper.GetTimestamp(),
 | 
			
		||||
			AccessedTime:   helper.GetTimestamp(),
 | 
			
		||||
			ExpiredTime:    -1,
 | 
			
		||||
			RemainQuota:    -1,
 | 
			
		||||
			UnlimitedQuota: true,
 | 
			
		||||
		}
 | 
			
		||||
		err = cleanToken.Insert()
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			logger.Errorf(ctx, "cleanToken.Insert failed: %w", err)
 | 
			
		||||
			return
 | 
			
		||||
		}
 | 
			
		||||
	}()
 | 
			
		||||
	c.JSON(http.StatusOK, gin.H{
 | 
			
		||||
		"success": true,
 | 
			
		||||
		"message": "",
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										48
									
								
								go.mod
									
									
									
									
									
								
							
							
						
						
									
										48
									
								
								go.mod
									
									
									
									
									
								
							@@ -4,42 +4,42 @@ module github.com/songquanpeng/one-api
 | 
			
		||||
go 1.20
 | 
			
		||||
 | 
			
		||||
require (
 | 
			
		||||
	github.com/aws/aws-sdk-go-v2 v1.26.1
 | 
			
		||||
	github.com/aws/aws-sdk-go-v2/credentials v1.17.11
 | 
			
		||||
	github.com/aws/aws-sdk-go-v2/service/bedrockruntime v1.7.4
 | 
			
		||||
	github.com/gin-contrib/cors v1.7.1
 | 
			
		||||
	github.com/gin-contrib/gzip v1.0.0
 | 
			
		||||
	github.com/gin-contrib/sessions v1.0.0
 | 
			
		||||
	github.com/gin-contrib/static v1.1.1
 | 
			
		||||
	github.com/gin-gonic/gin v1.9.1
 | 
			
		||||
	github.com/go-playground/validator/v10 v10.19.0
 | 
			
		||||
	github.com/aws/aws-sdk-go-v2 v1.27.0
 | 
			
		||||
	github.com/aws/aws-sdk-go-v2/credentials v1.17.15
 | 
			
		||||
	github.com/aws/aws-sdk-go-v2/service/bedrockruntime v1.8.3
 | 
			
		||||
	github.com/gin-contrib/cors v1.7.2
 | 
			
		||||
	github.com/gin-contrib/gzip v1.0.1
 | 
			
		||||
	github.com/gin-contrib/sessions v1.0.1
 | 
			
		||||
	github.com/gin-contrib/static v1.1.2
 | 
			
		||||
	github.com/gin-gonic/gin v1.10.0
 | 
			
		||||
	github.com/go-playground/validator/v10 v10.20.0
 | 
			
		||||
	github.com/go-redis/redis/v8 v8.11.5
 | 
			
		||||
	github.com/golang-jwt/jwt v3.2.2+incompatible
 | 
			
		||||
	github.com/google/uuid v1.6.0
 | 
			
		||||
	github.com/gorilla/websocket v1.5.1
 | 
			
		||||
	github.com/jinzhu/copier v0.4.0
 | 
			
		||||
	github.com/pkg/errors v0.9.1
 | 
			
		||||
	github.com/pkoukk/tiktoken-go v0.1.6
 | 
			
		||||
	github.com/pkoukk/tiktoken-go v0.1.7
 | 
			
		||||
	github.com/smartystreets/goconvey v1.8.1
 | 
			
		||||
	github.com/stretchr/testify v1.9.0
 | 
			
		||||
	golang.org/x/crypto v0.22.0
 | 
			
		||||
	golang.org/x/image v0.15.0
 | 
			
		||||
	golang.org/x/crypto v0.23.0
 | 
			
		||||
	golang.org/x/image v0.16.0
 | 
			
		||||
	gorm.io/driver/mysql v1.5.6
 | 
			
		||||
	gorm.io/driver/postgres v1.5.7
 | 
			
		||||
	gorm.io/driver/sqlite v1.5.5
 | 
			
		||||
	gorm.io/gorm v1.25.9
 | 
			
		||||
	gorm.io/gorm v1.25.10
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
require (
 | 
			
		||||
	filippo.io/edwards25519 v1.1.0 // indirect
 | 
			
		||||
	github.com/aws/aws-sdk-go-v2/aws/protocol/eventstream v1.6.2 // indirect
 | 
			
		||||
	github.com/aws/aws-sdk-go-v2/internal/configsources v1.3.5 // indirect
 | 
			
		||||
	github.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.6.5 // indirect
 | 
			
		||||
	github.com/aws/aws-sdk-go-v2/internal/configsources v1.3.7 // indirect
 | 
			
		||||
	github.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.6.7 // indirect
 | 
			
		||||
	github.com/aws/smithy-go v1.20.2 // indirect
 | 
			
		||||
	github.com/bytedance/sonic v1.11.5 // indirect
 | 
			
		||||
	github.com/bytedance/sonic v1.11.6 // indirect
 | 
			
		||||
	github.com/bytedance/sonic/loader v0.1.1 // indirect
 | 
			
		||||
	github.com/cespare/xxhash/v2 v2.3.0 // indirect
 | 
			
		||||
	github.com/cloudwego/base64x v0.1.3 // indirect
 | 
			
		||||
	github.com/cloudwego/base64x v0.1.4 // indirect
 | 
			
		||||
	github.com/cloudwego/iasm v0.2.0 // indirect
 | 
			
		||||
	github.com/davecgh/go-spew v1.1.1 // indirect
 | 
			
		||||
	github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect
 | 
			
		||||
@@ -50,7 +50,7 @@ require (
 | 
			
		||||
	github.com/go-playground/locales v0.14.1 // indirect
 | 
			
		||||
	github.com/go-playground/universal-translator v0.18.1 // indirect
 | 
			
		||||
	github.com/go-sql-driver/mysql v1.8.1 // indirect
 | 
			
		||||
	github.com/goccy/go-json v0.10.2 // indirect
 | 
			
		||||
	github.com/goccy/go-json v0.10.3 // indirect
 | 
			
		||||
	github.com/gopherjs/gopherjs v1.17.2 // indirect
 | 
			
		||||
	github.com/gorilla/context v1.1.2 // indirect
 | 
			
		||||
	github.com/gorilla/securecookie v1.1.2 // indirect
 | 
			
		||||
@@ -70,16 +70,16 @@ require (
 | 
			
		||||
	github.com/mattn/go-sqlite3 v2.0.3+incompatible // indirect
 | 
			
		||||
	github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
 | 
			
		||||
	github.com/modern-go/reflect2 v1.0.2 // indirect
 | 
			
		||||
	github.com/pelletier/go-toml/v2 v2.2.1 // indirect
 | 
			
		||||
	github.com/pelletier/go-toml/v2 v2.2.2 // indirect
 | 
			
		||||
	github.com/pmezard/go-difflib v1.0.0 // indirect
 | 
			
		||||
	github.com/smarty/assertions v1.15.0 // indirect
 | 
			
		||||
	github.com/twitchyliquid64/golang-asm v0.15.1 // indirect
 | 
			
		||||
	github.com/ugorji/go/codec v1.2.12 // indirect
 | 
			
		||||
	golang.org/x/arch v0.7.0 // indirect
 | 
			
		||||
	golang.org/x/net v0.24.0 // indirect
 | 
			
		||||
	golang.org/x/arch v0.8.0 // indirect
 | 
			
		||||
	golang.org/x/net v0.25.0 // indirect
 | 
			
		||||
	golang.org/x/sync v0.7.0 // indirect
 | 
			
		||||
	golang.org/x/sys v0.19.0 // indirect
 | 
			
		||||
	golang.org/x/text v0.14.0 // indirect
 | 
			
		||||
	google.golang.org/protobuf v1.33.0 // indirect
 | 
			
		||||
	golang.org/x/sys v0.20.0 // indirect
 | 
			
		||||
	golang.org/x/text v0.15.0 // indirect
 | 
			
		||||
	google.golang.org/protobuf v1.34.1 // indirect
 | 
			
		||||
	gopkg.in/yaml.v3 v3.0.1 // indirect
 | 
			
		||||
)
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										48
									
								
								go.sum
									
									
									
									
									
								
							
							
						
						
									
										48
									
								
								go.sum
									
									
									
									
									
								
							@@ -2,20 +2,32 @@ filippo.io/edwards25519 v1.1.0 h1:FNf4tywRC1HmFuKW5xopWpigGjJKiJSV0Cqo0cJWDaA=
 | 
			
		||||
filippo.io/edwards25519 v1.1.0/go.mod h1:BxyFTGdWcka3PhytdK4V28tE5sGfRvvvRV7EaN4VDT4=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2 v1.26.1 h1:5554eUqIYVWpU0YmeeYZ0wU64H2VLBs8TlhRB2L+EkA=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2 v1.26.1/go.mod h1:ffIFB97e2yNsv4aTSGkqtHnppsIJzw7G7BReUZ3jCXM=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2 v1.27.0 h1:7bZWKoXhzI+mMR/HjdMx8ZCC5+6fY0lS5tr0bbgiLlo=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2 v1.27.0/go.mod h1:ffIFB97e2yNsv4aTSGkqtHnppsIJzw7G7BReUZ3jCXM=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2/aws/protocol/eventstream v1.6.2 h1:x6xsQXGSmW6frevwDA+vi/wqhp1ct18mVXYN08/93to=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2/aws/protocol/eventstream v1.6.2/go.mod h1:lPprDr1e6cJdyYeGXnRaJoP4Md+cDBvi2eOj00BlGmg=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2/credentials v1.17.11 h1:YuIB1dJNf1Re822rriUOTxopaHHvIq0l/pX3fwO+Tzs=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2/credentials v1.17.11/go.mod h1:AQtFPsDH9bI2O+71anW6EKL+NcD7LG3dpKGMV4SShgo=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2/credentials v1.17.15 h1:YDexlvDRCA8ems2T5IP1xkMtOZ1uLJOCJdTr0igs5zo=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2/credentials v1.17.15/go.mod h1:vxHggqW6hFNaeNC0WyXS3VdyjcV0a4KMUY4dKJ96buU=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2/internal/configsources v1.3.5 h1:aw39xVGeRWlWx9EzGVnhOR4yOjQDHPQ6o6NmBlscyQg=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2/internal/configsources v1.3.5/go.mod h1:FSaRudD0dXiMPK2UjknVwwTYyZMRsHv3TtkabsZih5I=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2/internal/configsources v1.3.7 h1:lf/8VTF2cM+N4SLzaYJERKEWAXq8MOMpZfU6wEPWsPk=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2/internal/configsources v1.3.7/go.mod h1:4SjkU7QiqK2M9oozyMzfZ/23LmUY+h3oFqhdeP5OMiI=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.6.5 h1:PG1F3OD1szkuQPzDw3CIQsRIrtTlUC3lP84taWzHlq0=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.6.5/go.mod h1:jU1li6RFryMz+so64PpKtudI+QzbKoIEivqdf6LNpOc=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.6.7 h1:4OYVp0705xu8yjdyoWix0r9wPIRXnIzzOoUpQVHIJ/g=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.6.7/go.mod h1:vd7ESTEvI76T2Na050gODNmNU7+OyKrIKroYTu4ABiI=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2/service/bedrockruntime v1.7.4 h1:JgHnonzbnA3pbqj76wYsSZIZZQYBxkmMEjvL6GHy8XU=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2/service/bedrockruntime v1.7.4/go.mod h1:nZspkhg+9p8iApLFoyAqfyuMP0F38acy2Hm3r5r95Cg=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2/service/bedrockruntime v1.8.3 h1:Fihjyd6DeNjcawBEGLH9dkIEUi6AdhucDKPE9nJ4QiY=
 | 
			
		||||
github.com/aws/aws-sdk-go-v2/service/bedrockruntime v1.8.3/go.mod h1:opvUj3ismqSCxYc+m4WIjPL0ewZGtvp0ess7cKvBPOQ=
 | 
			
		||||
github.com/aws/smithy-go v1.20.2 h1:tbp628ireGtzcHDDmLT/6ADHidqnwgF57XOXZe6tp4Q=
 | 
			
		||||
github.com/aws/smithy-go v1.20.2/go.mod h1:krry+ya/rV9RDcV/Q16kpu6ypI4K2czasz0NC3qS14E=
 | 
			
		||||
github.com/bytedance/sonic v1.11.5 h1:G00FYjjqll5iQ1PYXynbg/hyzqBqavH8Mo9/oTopd9k=
 | 
			
		||||
github.com/bytedance/sonic v1.11.5/go.mod h1:X2PC2giUdj/Cv2lliWFLk6c/DUQok5rViJSemeB0wDw=
 | 
			
		||||
github.com/bytedance/sonic v1.11.6 h1:oUp34TzMlL+OY1OUWxHqsdkgC/Zfc85zGqw9siXjrc0=
 | 
			
		||||
github.com/bytedance/sonic v1.11.6/go.mod h1:LysEHSvpvDySVdC2f87zGWf6CIKJcAvqab1ZaiQtds4=
 | 
			
		||||
github.com/bytedance/sonic/loader v0.1.0/go.mod h1:UmRT+IRTGKz/DAkzcEGzyVqQFJ7H9BqwBO3pm9H/+HY=
 | 
			
		||||
github.com/bytedance/sonic/loader v0.1.1 h1:c+e5Pt1k/cy5wMveRDyk2X4B9hF4g7an8N3zCYjJFNM=
 | 
			
		||||
github.com/bytedance/sonic/loader v0.1.1/go.mod h1:ncP89zfokxS5LZrJxl5z0UJcsk4M4yY2JpfqGeCtNLU=
 | 
			
		||||
@@ -23,6 +35,8 @@ github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UF
 | 
			
		||||
github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
 | 
			
		||||
github.com/cloudwego/base64x v0.1.3 h1:b5J/l8xolB7dyDTTmhJP2oTs5LdrjyrUFuNxdfq5hAg=
 | 
			
		||||
github.com/cloudwego/base64x v0.1.3/go.mod h1:1+1K5BUHIQzyapgpF7LwvOGAEDicKtt1umPV+aN8pi8=
 | 
			
		||||
github.com/cloudwego/base64x v0.1.4 h1:jwCgWpFanWmN8xoIUHa2rtzmkd5J2plF/dnLS6Xd/0Y=
 | 
			
		||||
github.com/cloudwego/base64x v0.1.4/go.mod h1:0zlkT4Wn5C6NdauXdJRhSKRlJvmclQ1hhJgA0rcu/8w=
 | 
			
		||||
github.com/cloudwego/iasm v0.2.0 h1:1KNIy1I1H9hNNFEEH3DVnI4UujN+1zjpuk6gwHLTssg=
 | 
			
		||||
github.com/cloudwego/iasm v0.2.0/go.mod h1:8rXZaNYT2n95jn+zTI1sDr+IgcD2GVs0nlbbQPiEFhY=
 | 
			
		||||
github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=
 | 
			
		||||
@@ -39,16 +53,26 @@ github.com/gabriel-vasile/mimetype v1.4.3 h1:in2uUcidCuFcDKtdcBxlR0rJ1+fsokWf+uq
 | 
			
		||||
github.com/gabriel-vasile/mimetype v1.4.3/go.mod h1:d8uq/6HKRL6CGdk+aubisF/M5GcPfT7nKyLpA0lbSSk=
 | 
			
		||||
github.com/gin-contrib/cors v1.7.1 h1:s9SIppU/rk8enVvkzwiC2VK3UZ/0NNGsWfUKvV55rqs=
 | 
			
		||||
github.com/gin-contrib/cors v1.7.1/go.mod h1:n/Zj7B4xyrgk/cX1WCX2dkzFfaNm/xJb6oIUk7WTtps=
 | 
			
		||||
github.com/gin-contrib/cors v1.7.2 h1:oLDHxdg8W/XDoN/8zamqk/Drgt4oVZDvaV0YmvVICQw=
 | 
			
		||||
github.com/gin-contrib/cors v1.7.2/go.mod h1:SUJVARKgQ40dmrzgXEVxj2m7Ig1v1qIboQkPDTQ9t2E=
 | 
			
		||||
github.com/gin-contrib/gzip v1.0.0 h1:UKN586Po/92IDX6ie5CWLgMI81obiIp5nSP85T3wlTk=
 | 
			
		||||
github.com/gin-contrib/gzip v1.0.0/go.mod h1:CtG7tQrPB3vIBo6Gat9FVUsis+1emjvQqd66ME5TdnE=
 | 
			
		||||
github.com/gin-contrib/gzip v1.0.1 h1:HQ8ENHODeLY7a4g1Au/46Z92bdGFl74OhxcZble9WJE=
 | 
			
		||||
github.com/gin-contrib/gzip v1.0.1/go.mod h1:njt428fdUNRvjuJf16tZMYZ2Yl+WQB53X5wmhDwXvC4=
 | 
			
		||||
github.com/gin-contrib/sessions v1.0.0 h1:r5GLta4Oy5xo9rAwMHx8B4wLpeRGHMdz9NafzJAdP8Y=
 | 
			
		||||
github.com/gin-contrib/sessions v1.0.0/go.mod h1:DN0f4bvpqMQElDdi+gNGScrP2QEI04IErRyMFyorUOI=
 | 
			
		||||
github.com/gin-contrib/sessions v1.0.1 h1:3hsJyNs7v7N8OtelFmYXFrulAf6zSR7nW/putcPEHxI=
 | 
			
		||||
github.com/gin-contrib/sessions v1.0.1/go.mod h1:ouxSFM24/OgIud5MJYQJLpy6AwxQ5EYO9yLhbtObGkM=
 | 
			
		||||
github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=
 | 
			
		||||
github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=
 | 
			
		||||
github.com/gin-contrib/static v1.1.1 h1:XEvBd4DDLG1HBlyPBQU1XO8NlTpw6mgdqcPteetYA5k=
 | 
			
		||||
github.com/gin-contrib/static v1.1.1/go.mod h1:yRGmar7+JYvbMLRPIi4H5TVVSBwULfT9vetnVD0IO74=
 | 
			
		||||
github.com/gin-contrib/static v1.1.2 h1:c3kT4bFkUJn2aoRU3s6XnMjJT8J6nNWJkR0NglqmlZ4=
 | 
			
		||||
github.com/gin-contrib/static v1.1.2/go.mod h1:Fw90ozjHCmZBWbgrsqrDvO28YbhKEKzKp8GixhR4yLw=
 | 
			
		||||
github.com/gin-gonic/gin v1.9.1 h1:4idEAncQnU5cB7BeOkPtxjfCSye0AAm1R0RVIqJ+Jmg=
 | 
			
		||||
github.com/gin-gonic/gin v1.9.1/go.mod h1:hPrL7YrpYKXt5YId3A/Tnip5kqbEAP+KLuI3SUcPTeU=
 | 
			
		||||
github.com/gin-gonic/gin v1.10.0 h1:nTuyha1TYqgedzytsKYqna+DfLos46nTv2ygFy86HFU=
 | 
			
		||||
github.com/gin-gonic/gin v1.10.0/go.mod h1:4PMNQiOhvDRa013RKVbsiNwoyezlm2rm0uX/T7kzp5Y=
 | 
			
		||||
github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s=
 | 
			
		||||
github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA=
 | 
			
		||||
github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY=
 | 
			
		||||
@@ -56,6 +80,8 @@ github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJn
 | 
			
		||||
github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
 | 
			
		||||
github.com/go-playground/validator/v10 v10.19.0 h1:ol+5Fu+cSq9JD7SoSqe04GMI92cbn0+wvQ3bZ8b/AU4=
 | 
			
		||||
github.com/go-playground/validator/v10 v10.19.0/go.mod h1:dbuPbCMFw/DrkbEynArYaCwl3amGuJotoKCe95atGMM=
 | 
			
		||||
github.com/go-playground/validator/v10 v10.20.0 h1:K9ISHbSaI0lyB2eWMPJo+kOS/FBExVwjEviJTixqxL8=
 | 
			
		||||
github.com/go-playground/validator/v10 v10.20.0/go.mod h1:dbuPbCMFw/DrkbEynArYaCwl3amGuJotoKCe95atGMM=
 | 
			
		||||
github.com/go-redis/redis/v8 v8.11.5 h1:AcZZR7igkdvfVmQTPnu9WE37LRrO/YrBH5zWyjDC0oI=
 | 
			
		||||
github.com/go-redis/redis/v8 v8.11.5/go.mod h1:gREzHqY1hg6oD9ngVRbLStwAWKhA0FEgq8Jd4h5lpwo=
 | 
			
		||||
github.com/go-sql-driver/mysql v1.7.0/go.mod h1:OXbVy3sEdcQ2Doequ6Z5BW6fXNQTmx+9S1MCJN5yJMI=
 | 
			
		||||
@@ -63,6 +89,8 @@ github.com/go-sql-driver/mysql v1.8.1 h1:LedoTUt/eveggdHS9qUFC1EFSa8bU2+1pZjSRpv
 | 
			
		||||
github.com/go-sql-driver/mysql v1.8.1/go.mod h1:wEBSXgmK//2ZFJyE+qWnIsVGmvmEKlqwuVSjsCm7DZg=
 | 
			
		||||
github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU=
 | 
			
		||||
github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I=
 | 
			
		||||
github.com/goccy/go-json v0.10.3 h1:KZ5WoDbxAIgm2HNbYckL0se1fHD6rz5j4ywS6ebzDqA=
 | 
			
		||||
github.com/goccy/go-json v0.10.3/go.mod h1:oq7eo15ShAhp70Anwd5lgX2pLfOS3QCiwU/PULtXL6M=
 | 
			
		||||
github.com/golang-jwt/jwt v3.2.2+incompatible h1:IfV12K8xAKAnZqdXVzCZ+TOjboZ2keLg81eXfW3O+oY=
 | 
			
		||||
github.com/golang-jwt/jwt v3.2.2+incompatible/go.mod h1:8pz2t5EyA70fFQQSrl6XZXzqecmYZeUEB8OUGHkxJ+I=
 | 
			
		||||
github.com/google/go-cmp v0.5.6 h1:BKbKCqvP6I+rmFHt06ZmyQtvB8xAkWdhFyr0ZUNZcxQ=
 | 
			
		||||
@@ -121,10 +149,14 @@ github.com/onsi/ginkgo v1.16.5 h1:8xi0RTUf59SOSfEtZMvwTvXYMzG4gV23XVHOZiXNtnE=
 | 
			
		||||
github.com/onsi/gomega v1.18.1 h1:M1GfJqGRrBrrGGsbxzV5dqM2U2ApXefZCQpkukxYRLE=
 | 
			
		||||
github.com/pelletier/go-toml/v2 v2.2.1 h1:9TA9+T8+8CUCO2+WYnDLCgrYi9+omqKXyjDtosvtEhg=
 | 
			
		||||
github.com/pelletier/go-toml/v2 v2.2.1/go.mod h1:1t835xjRzz80PqgE6HHgN2JOsmgYu/h4qDAS4n929Rs=
 | 
			
		||||
github.com/pelletier/go-toml/v2 v2.2.2 h1:aYUidT7k73Pcl9nb2gScu7NSrKCSHIDE89b3+6Wq+LM=
 | 
			
		||||
github.com/pelletier/go-toml/v2 v2.2.2/go.mod h1:1t835xjRzz80PqgE6HHgN2JOsmgYu/h4qDAS4n929Rs=
 | 
			
		||||
github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
 | 
			
		||||
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
 | 
			
		||||
github.com/pkoukk/tiktoken-go v0.1.6 h1:JF0TlJzhTbrI30wCvFuiw6FzP2+/bR+FIxUdgEAcUsw=
 | 
			
		||||
github.com/pkoukk/tiktoken-go v0.1.6/go.mod h1:9NiV+i9mJKGj1rYOT+njbv+ZwA/zJxYdewGl6qVatpg=
 | 
			
		||||
github.com/pkoukk/tiktoken-go v0.1.7 h1:qOBHXX4PHtvIvmOtyg1EeKlwFRiMKAcoMp4Q+bLQDmw=
 | 
			
		||||
github.com/pkoukk/tiktoken-go v0.1.7/go.mod h1:9NiV+i9mJKGj1rYOT+njbv+ZwA/zJxYdewGl6qVatpg=
 | 
			
		||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 | 
			
		||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 | 
			
		||||
github.com/rogpeppe/go-internal v1.8.0 h1:FCbCCtXNOY3UtUuHUYaghJg4y7Fd14rXifAYUAtL9R8=
 | 
			
		||||
@@ -151,23 +183,37 @@ github.com/ugorji/go/codec v1.2.12/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZ
 | 
			
		||||
golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=
 | 
			
		||||
golang.org/x/arch v0.7.0 h1:pskyeJh/3AmoQ8CPE95vxHLqp1G1GfGNXTmcl9NEKTc=
 | 
			
		||||
golang.org/x/arch v0.7.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys=
 | 
			
		||||
golang.org/x/arch v0.8.0 h1:3wRIsP3pM4yUptoR96otTUOXI367OS0+c9eeRi9doIc=
 | 
			
		||||
golang.org/x/arch v0.8.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys=
 | 
			
		||||
golang.org/x/crypto v0.22.0 h1:g1v0xeRhjcugydODzvb3mEM9SQ0HGp9s/nh3COQ/C30=
 | 
			
		||||
golang.org/x/crypto v0.22.0/go.mod h1:vr6Su+7cTlO45qkww3VDJlzDn0ctJvRgYbC2NvXHt+M=
 | 
			
		||||
golang.org/x/crypto v0.23.0 h1:dIJU/v2J8Mdglj/8rJ6UUOM3Zc9zLZxVZwwxMooUSAI=
 | 
			
		||||
golang.org/x/crypto v0.23.0/go.mod h1:CKFgDieR+mRhux2Lsu27y0fO304Db0wZe70UKqHu0v8=
 | 
			
		||||
golang.org/x/image v0.15.0 h1:kOELfmgrmJlw4Cdb7g/QGuB3CvDrXbqEIww/pNtNBm8=
 | 
			
		||||
golang.org/x/image v0.15.0/go.mod h1:HUYqC05R2ZcZ3ejNQsIHQDQiwWM4JBqmm6MKANTp4LE=
 | 
			
		||||
golang.org/x/image v0.16.0 h1:9kloLAKhUufZhA12l5fwnx2NZW39/we1UhBesW433jw=
 | 
			
		||||
golang.org/x/image v0.16.0/go.mod h1:ugSZItdV4nOxyqp56HmXwH0Ry0nBCpjnZdpDaIHdoPs=
 | 
			
		||||
golang.org/x/net v0.24.0 h1:1PcaxkF854Fu3+lvBIx5SYn9wRlBzzcnHZSiaFFAb0w=
 | 
			
		||||
golang.org/x/net v0.24.0/go.mod h1:2Q7sJY5mzlzWjKtYUEXSlBWCdyaioyXzRB2RtU8KVE8=
 | 
			
		||||
golang.org/x/net v0.25.0 h1:d/OCCoBEUq33pjydKrGQhw7IlUPI2Oylr+8qLx49kac=
 | 
			
		||||
golang.org/x/net v0.25.0/go.mod h1:JkAGAh7GEvH74S6FOH42FLoXpXbE/aqXSrIQjXgsiwM=
 | 
			
		||||
golang.org/x/sync v0.7.0 h1:YsImfSBoP9QPYL0xyKJPq0gcaJdG3rInoqxTWbfQu9M=
 | 
			
		||||
golang.org/x/sync v0.7.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
 | 
			
		||||
golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 | 
			
		||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
 | 
			
		||||
golang.org/x/sys v0.19.0 h1:q5f1RH2jigJ1MoAWp2KTp3gm5zAGFUTarQZ5U386+4o=
 | 
			
		||||
golang.org/x/sys v0.19.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
 | 
			
		||||
golang.org/x/sys v0.20.0 h1:Od9JTbYCk261bKm4M/mw7AklTlFYIa0bIp9BgSm1S8Y=
 | 
			
		||||
golang.org/x/sys v0.20.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
 | 
			
		||||
golang.org/x/text v0.14.0 h1:ScX5w1eTa3QqT8oi6+ziP7dTV1S2+ALU0bI+0zXKWiQ=
 | 
			
		||||
golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
 | 
			
		||||
golang.org/x/text v0.15.0 h1:h1V/4gjBv8v9cjcR6+AR5+/cIYK5N/WAgiv4xlsEtAk=
 | 
			
		||||
golang.org/x/text v0.15.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
 | 
			
		||||
golang.org/x/xerrors v0.0.0-20200804184101-5ec99f83aff1 h1:go1bK/D/BFZV2I8cIQd1NKEZ+0owSTG1fDTci4IqFcE=
 | 
			
		||||
google.golang.org/protobuf v1.33.0 h1:uNO2rsAINq/JlFpSdYEKIZ0uKD/R9cpdv0T+yoGwGmI=
 | 
			
		||||
google.golang.org/protobuf v1.33.0/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=
 | 
			
		||||
google.golang.org/protobuf v1.34.1 h1:9ddQBjfCyZPOHPUiPxpYESBLc+T8P3E+Vo4IbKZgFWg=
 | 
			
		||||
google.golang.org/protobuf v1.34.1/go.mod h1:c6P6GXX6sHbq/GpV6MGZEdwhWPcYBgnhAHhKbcUYpos=
 | 
			
		||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 | 
			
		||||
gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
 | 
			
		||||
gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 h1:uRGJdciOHaEIrze2W8Q3AKkepLTh2hOroT7a+7czfdQ=
 | 
			
		||||
@@ -184,5 +230,7 @@ gorm.io/driver/sqlite v1.5.5/go.mod h1:6NgQ7sQWAIFsPrJJl1lSNSu2TABh0ZZ/zm5fosATa
 | 
			
		||||
gorm.io/gorm v1.25.7/go.mod h1:hbnx/Oo0ChWMn1BIhpy1oYozzpM15i4YPuHDmfYtwg8=
 | 
			
		||||
gorm.io/gorm v1.25.9 h1:wct0gxZIELDk8+ZqF/MVnHLkA1rvYlBWUMv2EdsK1g8=
 | 
			
		||||
gorm.io/gorm v1.25.9/go.mod h1:hbnx/Oo0ChWMn1BIhpy1oYozzpM15i4YPuHDmfYtwg8=
 | 
			
		||||
gorm.io/gorm v1.25.10 h1:dQpO+33KalOA+aFYGlK+EfxcI5MbO7EP2yYygwh9h+s=
 | 
			
		||||
gorm.io/gorm v1.25.10/go.mod h1:hbnx/Oo0ChWMn1BIhpy1oYozzpM15i4YPuHDmfYtwg8=
 | 
			
		||||
nullprogram.com/x/optparse v1.0.0/go.mod h1:KdyPE+Igbe0jQUrVfMqDMeJQIJZEuyV7pjYmp6pbG50=
 | 
			
		||||
rsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4=
 | 
			
		||||
 
 | 
			
		||||
@@ -65,21 +65,29 @@ func SetupContextForSelectedChannel(c *gin.Context, channel *model.Channel, mode
 | 
			
		||||
	c.Set(ctxkey.OriginalModel, modelName) // for retry
 | 
			
		||||
	c.Request.Header.Set("Authorization", fmt.Sprintf("Bearer %s", channel.Key))
 | 
			
		||||
	c.Set(ctxkey.BaseURL, channel.GetBaseURL())
 | 
			
		||||
	cfg, _ := channel.LoadConfig()
 | 
			
		||||
	// this is for backward compatibility
 | 
			
		||||
	switch channel.Type {
 | 
			
		||||
	case channeltype.Azure:
 | 
			
		||||
		c.Set(ctxkey.ConfigAPIVersion, channel.Other)
 | 
			
		||||
		if cfg.APIVersion == "" {
 | 
			
		||||
			cfg.APIVersion = channel.Other
 | 
			
		||||
		}
 | 
			
		||||
	case channeltype.Xunfei:
 | 
			
		||||
		c.Set(ctxkey.ConfigAPIVersion, channel.Other)
 | 
			
		||||
		if cfg.APIVersion == "" {
 | 
			
		||||
			cfg.APIVersion = channel.Other
 | 
			
		||||
		}
 | 
			
		||||
	case channeltype.Gemini:
 | 
			
		||||
		c.Set(ctxkey.ConfigAPIVersion, channel.Other)
 | 
			
		||||
		if cfg.APIVersion == "" {
 | 
			
		||||
			cfg.APIVersion = channel.Other
 | 
			
		||||
		}
 | 
			
		||||
	case channeltype.AIProxyLibrary:
 | 
			
		||||
		c.Set(ctxkey.ConfigLibraryID, channel.Other)
 | 
			
		||||
		if cfg.LibraryID == "" {
 | 
			
		||||
			cfg.LibraryID = channel.Other
 | 
			
		||||
		}
 | 
			
		||||
	case channeltype.Ali:
 | 
			
		||||
		c.Set(ctxkey.ConfigPlugin, channel.Other)
 | 
			
		||||
	}
 | 
			
		||||
	cfg, _ := channel.LoadConfig()
 | 
			
		||||
	for k, v := range cfg {
 | 
			
		||||
		c.Set(ctxkey.ConfigPrefix+k, v)
 | 
			
		||||
		if cfg.Plugin == "" {
 | 
			
		||||
			cfg.Plugin = channel.Other
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	c.Set(ctxkey.Config, cfg)
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -3,14 +3,14 @@ package middleware
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/logger"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/helper"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func SetUpLogger(server *gin.Engine) {
 | 
			
		||||
	server.Use(gin.LoggerWithFormatter(func(param gin.LogFormatterParams) string {
 | 
			
		||||
		var requestID string
 | 
			
		||||
		if param.Keys != nil {
 | 
			
		||||
			requestID = param.Keys[logger.RequestIdKey].(string)
 | 
			
		||||
			requestID = param.Keys[helper.RequestIdKey].(string)
 | 
			
		||||
		}
 | 
			
		||||
		return fmt.Sprintf("[GIN] %s | %s | %3d | %13v | %15s | %7s %s\n",
 | 
			
		||||
			param.TimeStamp.Format("2006/01/02 - 15:04:05"),
 | 
			
		||||
 
 | 
			
		||||
@@ -4,16 +4,15 @@ import (
 | 
			
		||||
	"context"
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/helper"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/logger"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func RequestId() func(c *gin.Context) {
 | 
			
		||||
	return func(c *gin.Context) {
 | 
			
		||||
		id := helper.GenRequestID()
 | 
			
		||||
		c.Set(logger.RequestIdKey, id)
 | 
			
		||||
		ctx := context.WithValue(c.Request.Context(), logger.RequestIdKey, id)
 | 
			
		||||
		c.Set(helper.RequestIdKey, id)
 | 
			
		||||
		ctx := context.WithValue(c.Request.Context(), helper.RequestIdKey, id)
 | 
			
		||||
		c.Request = c.Request.WithContext(ctx)
 | 
			
		||||
		c.Header(logger.RequestIdKey, id)
 | 
			
		||||
		c.Header(helper.RequestIdKey, id)
 | 
			
		||||
		c.Next()
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -12,7 +12,7 @@ import (
 | 
			
		||||
func abortWithMessage(c *gin.Context, statusCode int, message string) {
 | 
			
		||||
	c.JSON(statusCode, gin.H{
 | 
			
		||||
		"error": gin.H{
 | 
			
		||||
			"message": helper.MessageWithRequestId(message, c.GetString(logger.RequestIdKey)),
 | 
			
		||||
			"message": helper.MessageWithRequestId(message, c.GetString(helper.RequestIdKey)),
 | 
			
		||||
			"type":    "one_api_error",
 | 
			
		||||
		},
 | 
			
		||||
	})
 | 
			
		||||
 
 | 
			
		||||
@@ -38,6 +38,16 @@ type Channel struct {
 | 
			
		||||
	Config             string  `json:"config"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type ChannelConfig struct {
 | 
			
		||||
	Region     string `json:"region,omitempty"`
 | 
			
		||||
	SK         string `json:"sk,omitempty"`
 | 
			
		||||
	AK         string `json:"ak,omitempty"`
 | 
			
		||||
	UserID     string `json:"user_id,omitempty"`
 | 
			
		||||
	APIVersion string `json:"api_version,omitempty"`
 | 
			
		||||
	LibraryID  string `json:"library_id,omitempty"`
 | 
			
		||||
	Plugin     string `json:"plugin,omitempty"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func GetAllChannels(startIdx int, num int, scope string) ([]*Channel, error) {
 | 
			
		||||
	var channels []*Channel
 | 
			
		||||
	var err error
 | 
			
		||||
@@ -161,14 +171,14 @@ func (channel *Channel) Delete() error {
 | 
			
		||||
	return err
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (channel *Channel) LoadConfig() (map[string]string, error) {
 | 
			
		||||
func (channel *Channel) LoadConfig() (ChannelConfig, error) {
 | 
			
		||||
	var cfg ChannelConfig
 | 
			
		||||
	if channel.Config == "" {
 | 
			
		||||
		return nil, nil
 | 
			
		||||
		return cfg, nil
 | 
			
		||||
	}
 | 
			
		||||
	cfg := make(map[string]string)
 | 
			
		||||
	err := json.Unmarshal([]byte(channel.Config), &cfg)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return nil, err
 | 
			
		||||
		return cfg, err
 | 
			
		||||
	}
 | 
			
		||||
	return cfg, nil
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -7,8 +7,10 @@ import (
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/anthropic"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/aws"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/baidu"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/cloudflare"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/cohere"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/coze"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/deepl"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/gemini"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/ollama"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/openai"
 | 
			
		||||
@@ -49,6 +51,10 @@ func GetAdaptor(apiType int) adaptor.Adaptor {
 | 
			
		||||
		return &coze.Adaptor{}
 | 
			
		||||
	case apitype.Cohere:
 | 
			
		||||
		return &cohere.Adaptor{}
 | 
			
		||||
	case apitype.Cloudflare:
 | 
			
		||||
		return &cloudflare.Adaptor{}
 | 
			
		||||
	case apitype.DeepL:
 | 
			
		||||
		return &deepl.Adaptor{}
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -4,7 +4,6 @@ import (
 | 
			
		||||
	"errors"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/ctxkey"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/meta"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/model"
 | 
			
		||||
@@ -13,10 +12,11 @@ import (
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Adaptor struct {
 | 
			
		||||
	meta *meta.Meta
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) Init(meta *meta.Meta) {
 | 
			
		||||
 | 
			
		||||
	a.meta = meta
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {
 | 
			
		||||
@@ -34,7 +34,7 @@ func (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.G
 | 
			
		||||
		return nil, errors.New("request is nil")
 | 
			
		||||
	}
 | 
			
		||||
	aiProxyLibraryRequest := ConvertRequest(*request)
 | 
			
		||||
	aiProxyLibraryRequest.LibraryId = c.GetString(ctxkey.ConfigLibraryID)
 | 
			
		||||
	aiProxyLibraryRequest.LibraryId = a.meta.Config.LibraryID
 | 
			
		||||
	return aiProxyLibraryRequest, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
@@ -4,7 +4,6 @@ import (
 | 
			
		||||
	"errors"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/ctxkey"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/meta"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/model"
 | 
			
		||||
@@ -16,10 +15,11 @@ import (
 | 
			
		||||
// https://help.aliyun.com/zh/dashscope/developer-reference/api-details
 | 
			
		||||
 | 
			
		||||
type Adaptor struct {
 | 
			
		||||
	meta *meta.Meta
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) Init(meta *meta.Meta) {
 | 
			
		||||
 | 
			
		||||
	a.meta = meta
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {
 | 
			
		||||
@@ -47,8 +47,8 @@ func (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *me
 | 
			
		||||
	if meta.Mode == relaymode.ImagesGenerations {
 | 
			
		||||
		req.Header.Set("X-DashScope-Async", "enable")
 | 
			
		||||
	}
 | 
			
		||||
	if c.GetString(ctxkey.ConfigPlugin) != "" {
 | 
			
		||||
		req.Header.Set("X-DashScope-Plugin", c.GetString(ctxkey.ConfigPlugin))
 | 
			
		||||
	if a.meta.Config.Plugin != "" {
 | 
			
		||||
		req.Header.Set("X-DashScope-Plugin", a.meta.Config.Plugin)
 | 
			
		||||
	}
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -4,6 +4,10 @@ import (
 | 
			
		||||
	"bufio"
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"strings"
 | 
			
		||||
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/helper"
 | 
			
		||||
@@ -11,9 +15,6 @@ import (
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/logger"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/openai"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/model"
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func stopReasonClaude2OpenAI(reason *string) string {
 | 
			
		||||
@@ -176,10 +177,10 @@ func StreamHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusC
 | 
			
		||||
			if len(data) < 6 {
 | 
			
		||||
				continue
 | 
			
		||||
			}
 | 
			
		||||
			if !strings.HasPrefix(data, "data: ") {
 | 
			
		||||
			if !strings.HasPrefix(data, "data:") {
 | 
			
		||||
				continue
 | 
			
		||||
			}
 | 
			
		||||
			data = strings.TrimPrefix(data, "data: ")
 | 
			
		||||
			data = strings.TrimPrefix(data, "data:")
 | 
			
		||||
			dataChan <- data
 | 
			
		||||
		}
 | 
			
		||||
		stopChan <- true
 | 
			
		||||
@@ -192,7 +193,7 @@ func StreamHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusC
 | 
			
		||||
		select {
 | 
			
		||||
		case data := <-dataChan:
 | 
			
		||||
			// some implementations may add \r at the end of data
 | 
			
		||||
			data = strings.TrimSuffix(data, "\r")
 | 
			
		||||
			data = strings.TrimSpace(data)
 | 
			
		||||
			var claudeResponse StreamResponse
 | 
			
		||||
			err := json.Unmarshal([]byte(data), &claudeResponse)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
 
 | 
			
		||||
@@ -1,6 +1,9 @@
 | 
			
		||||
package aws
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"github.com/aws/aws-sdk-go-v2/aws"
 | 
			
		||||
	"github.com/aws/aws-sdk-go-v2/credentials"
 | 
			
		||||
	"github.com/aws/aws-sdk-go-v2/service/bedrockruntime"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/ctxkey"
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
@@ -16,10 +19,16 @@ import (
 | 
			
		||||
var _ adaptor.Adaptor = new(Adaptor)
 | 
			
		||||
 | 
			
		||||
type Adaptor struct {
 | 
			
		||||
	meta      *meta.Meta
 | 
			
		||||
	awsClient *bedrockruntime.Client
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) Init(meta *meta.Meta) {
 | 
			
		||||
 | 
			
		||||
	a.meta = meta
 | 
			
		||||
	a.awsClient = bedrockruntime.New(bedrockruntime.Options{
 | 
			
		||||
		Region:      meta.Config.Region,
 | 
			
		||||
		Credentials: aws.NewCredentialsCache(credentials.NewStaticCredentialsProvider(meta.Config.AK, meta.Config.SK, "")),
 | 
			
		||||
	})
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {
 | 
			
		||||
@@ -54,9 +63,9 @@ func (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Read
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {
 | 
			
		||||
	if meta.IsStream {
 | 
			
		||||
		err, usage = StreamHandler(c, resp)
 | 
			
		||||
		err, usage = StreamHandler(c, a.awsClient)
 | 
			
		||||
	} else {
 | 
			
		||||
		err, usage = Handler(c, resp, meta.PromptTokens, meta.ActualModelName)
 | 
			
		||||
		err, usage = Handler(c, a.awsClient, meta.ActualModelName)
 | 
			
		||||
	}
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
@@ -65,7 +74,6 @@ func (a *Adaptor) GetModelList() (models []string) {
 | 
			
		||||
	for n := range awsModelIDMap {
 | 
			
		||||
		models = append(models, n)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
@@ -10,7 +10,6 @@ import (
 | 
			
		||||
	"net/http"
 | 
			
		||||
 | 
			
		||||
	"github.com/aws/aws-sdk-go-v2/aws"
 | 
			
		||||
	"github.com/aws/aws-sdk-go-v2/credentials"
 | 
			
		||||
	"github.com/aws/aws-sdk-go-v2/service/bedrockruntime"
 | 
			
		||||
	"github.com/aws/aws-sdk-go-v2/service/bedrockruntime/types"
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
@@ -23,18 +22,6 @@ import (
 | 
			
		||||
	relaymodel "github.com/songquanpeng/one-api/relay/model"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func newAwsClient(c *gin.Context) (*bedrockruntime.Client, error) {
 | 
			
		||||
	ak := c.GetString(ctxkey.ConfigAK)
 | 
			
		||||
	sk := c.GetString(ctxkey.ConfigSK)
 | 
			
		||||
	region := c.GetString(ctxkey.ConfigRegion)
 | 
			
		||||
	client := bedrockruntime.New(bedrockruntime.Options{
 | 
			
		||||
		Region:      region,
 | 
			
		||||
		Credentials: aws.NewCredentialsCache(credentials.NewStaticCredentialsProvider(ak, sk, "")),
 | 
			
		||||
	})
 | 
			
		||||
 | 
			
		||||
	return client, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func wrapErr(err error) *relaymodel.ErrorWithStatusCode {
 | 
			
		||||
	return &relaymodel.ErrorWithStatusCode{
 | 
			
		||||
		StatusCode: http.StatusInternalServerError,
 | 
			
		||||
@@ -62,12 +49,7 @@ func awsModelID(requestModel string) (string, error) {
 | 
			
		||||
	return "", errors.Errorf("model %s not found", requestModel)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func Handler(c *gin.Context, resp *http.Response, promptTokens int, modelName string) (*relaymodel.ErrorWithStatusCode, *relaymodel.Usage) {
 | 
			
		||||
	awsCli, err := newAwsClient(c)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return wrapErr(errors.Wrap(err, "newAwsClient")), nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
func Handler(c *gin.Context, awsCli *bedrockruntime.Client, modelName string) (*relaymodel.ErrorWithStatusCode, *relaymodel.Usage) {
 | 
			
		||||
	awsModelId, err := awsModelID(c.GetString(ctxkey.RequestModel))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return wrapErr(errors.Wrap(err, "awsModelID")), nil
 | 
			
		||||
@@ -120,13 +102,8 @@ func Handler(c *gin.Context, resp *http.Response, promptTokens int, modelName st
 | 
			
		||||
	return nil, &usage
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func StreamHandler(c *gin.Context, resp *http.Response) (*relaymodel.ErrorWithStatusCode, *relaymodel.Usage) {
 | 
			
		||||
func StreamHandler(c *gin.Context, awsCli *bedrockruntime.Client) (*relaymodel.ErrorWithStatusCode, *relaymodel.Usage) {
 | 
			
		||||
	createdTime := helper.GetTimestamp()
 | 
			
		||||
	awsCli, err := newAwsClient(c)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return wrapErr(errors.Wrap(err, "newAwsClient")), nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	awsModelId, err := awsModelID(c.GetString(ctxkey.RequestModel))
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return wrapErr(errors.Wrap(err, "awsModelID")), nil
 | 
			
		||||
 
 | 
			
		||||
@@ -1,15 +0,0 @@
 | 
			
		||||
package azure
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/ctxkey"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func GetAPIVersion(c *gin.Context) string {
 | 
			
		||||
	query := c.Request.URL.Query()
 | 
			
		||||
	apiVersion := query.Get("api-version")
 | 
			
		||||
	if apiVersion == "" {
 | 
			
		||||
		apiVersion = c.GetString(ctxkey.ConfigAPIVersion)
 | 
			
		||||
	}
 | 
			
		||||
	return apiVersion
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										66
									
								
								relay/adaptor/cloudflare/adaptor.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										66
									
								
								relay/adaptor/cloudflare/adaptor.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,66 @@
 | 
			
		||||
package cloudflare
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"errors"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/meta"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/model"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Adaptor struct {
 | 
			
		||||
	meta *meta.Meta
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ConvertImageRequest implements adaptor.Adaptor.
 | 
			
		||||
func (*Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {
 | 
			
		||||
	return nil, errors.New("not implemented")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// ConvertImageRequest implements adaptor.Adaptor.
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) Init(meta *meta.Meta) {
 | 
			
		||||
	a.meta = meta
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {
 | 
			
		||||
	return fmt.Sprintf("%s/client/v4/accounts/%s/ai/run/%s", meta.BaseURL, meta.Config.UserID, meta.ActualModelName), nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {
 | 
			
		||||
	adaptor.SetupCommonRequestHeader(c, req, meta)
 | 
			
		||||
	req.Header.Set("Authorization", "Bearer "+meta.APIKey)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {
 | 
			
		||||
	if request == nil {
 | 
			
		||||
		return nil, errors.New("request is nil")
 | 
			
		||||
	}
 | 
			
		||||
	return ConvertRequest(*request), nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {
 | 
			
		||||
	return adaptor.DoRequestHelper(a, c, meta, requestBody)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {
 | 
			
		||||
	if meta.IsStream {
 | 
			
		||||
		err, usage = StreamHandler(c, resp, meta.PromptTokens, meta.ActualModelName)
 | 
			
		||||
	} else {
 | 
			
		||||
		err, usage = Handler(c, resp, meta.PromptTokens, meta.ActualModelName)
 | 
			
		||||
	}
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) GetModelList() []string {
 | 
			
		||||
	return ModelList
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) GetChannelName() string {
 | 
			
		||||
	return "cloudflare"
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										36
									
								
								relay/adaptor/cloudflare/constant.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										36
									
								
								relay/adaptor/cloudflare/constant.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,36 @@
 | 
			
		||||
package cloudflare
 | 
			
		||||
 | 
			
		||||
var ModelList = []string{
 | 
			
		||||
	"@cf/meta/llama-2-7b-chat-fp16",
 | 
			
		||||
	"@cf/meta/llama-2-7b-chat-int8",
 | 
			
		||||
	"@cf/mistral/mistral-7b-instruct-v0.1",
 | 
			
		||||
	"@hf/thebloke/deepseek-coder-6.7b-base-awq",
 | 
			
		||||
	"@hf/thebloke/deepseek-coder-6.7b-instruct-awq",
 | 
			
		||||
	"@cf/deepseek-ai/deepseek-math-7b-base",
 | 
			
		||||
	"@cf/deepseek-ai/deepseek-math-7b-instruct",
 | 
			
		||||
	"@cf/thebloke/discolm-german-7b-v1-awq",
 | 
			
		||||
	"@cf/tiiuae/falcon-7b-instruct",
 | 
			
		||||
	"@cf/google/gemma-2b-it-lora",
 | 
			
		||||
	"@hf/google/gemma-7b-it",
 | 
			
		||||
	"@cf/google/gemma-7b-it-lora",
 | 
			
		||||
	"@hf/nousresearch/hermes-2-pro-mistral-7b",
 | 
			
		||||
	"@hf/thebloke/llama-2-13b-chat-awq",
 | 
			
		||||
	"@cf/meta-llama/llama-2-7b-chat-hf-lora",
 | 
			
		||||
	"@cf/meta/llama-3-8b-instruct",
 | 
			
		||||
	"@hf/thebloke/llamaguard-7b-awq",
 | 
			
		||||
	"@hf/thebloke/mistral-7b-instruct-v0.1-awq",
 | 
			
		||||
	"@hf/mistralai/mistral-7b-instruct-v0.2",
 | 
			
		||||
	"@cf/mistral/mistral-7b-instruct-v0.2-lora",
 | 
			
		||||
	"@hf/thebloke/neural-chat-7b-v3-1-awq",
 | 
			
		||||
	"@cf/openchat/openchat-3.5-0106",
 | 
			
		||||
	"@hf/thebloke/openhermes-2.5-mistral-7b-awq",
 | 
			
		||||
	"@cf/microsoft/phi-2",
 | 
			
		||||
	"@cf/qwen/qwen1.5-0.5b-chat",
 | 
			
		||||
	"@cf/qwen/qwen1.5-1.8b-chat",
 | 
			
		||||
	"@cf/qwen/qwen1.5-14b-chat-awq",
 | 
			
		||||
	"@cf/qwen/qwen1.5-7b-chat-awq",
 | 
			
		||||
	"@cf/defog/sqlcoder-7b-2",
 | 
			
		||||
	"@hf/nexusflow/starling-lm-7b-beta",
 | 
			
		||||
	"@cf/tinyllama/tinyllama-1.1b-chat-v1.0",
 | 
			
		||||
	"@hf/thebloke/zephyr-7b-beta-awq",
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										152
									
								
								relay/adaptor/cloudflare/main.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										152
									
								
								relay/adaptor/cloudflare/main.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,152 @@
 | 
			
		||||
package cloudflare
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"bufio"
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"strings"
 | 
			
		||||
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/helper"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/logger"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/openai"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/model"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func ConvertRequest(textRequest model.GeneralOpenAIRequest) *Request {
 | 
			
		||||
	lastMessage := textRequest.Messages[len(textRequest.Messages)-1]
 | 
			
		||||
	return &Request{
 | 
			
		||||
		MaxTokens:   textRequest.MaxTokens,
 | 
			
		||||
		Prompt:      lastMessage.StringContent(),
 | 
			
		||||
		Stream:      textRequest.Stream,
 | 
			
		||||
		Temperature: textRequest.Temperature,
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func ResponseCloudflare2OpenAI(cloudflareResponse *Response) *openai.TextResponse {
 | 
			
		||||
	choice := openai.TextResponseChoice{
 | 
			
		||||
		Index: 0,
 | 
			
		||||
		Message: model.Message{
 | 
			
		||||
			Role:    "assistant",
 | 
			
		||||
			Content: cloudflareResponse.Result.Response,
 | 
			
		||||
		},
 | 
			
		||||
		FinishReason: "stop",
 | 
			
		||||
	}
 | 
			
		||||
	fullTextResponse := openai.TextResponse{
 | 
			
		||||
		Object:  "chat.completion",
 | 
			
		||||
		Created: helper.GetTimestamp(),
 | 
			
		||||
		Choices: []openai.TextResponseChoice{choice},
 | 
			
		||||
	}
 | 
			
		||||
	return &fullTextResponse
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func StreamResponseCloudflare2OpenAI(cloudflareResponse *StreamResponse) *openai.ChatCompletionsStreamResponse {
 | 
			
		||||
	var choice openai.ChatCompletionsStreamResponseChoice
 | 
			
		||||
	choice.Delta.Content = cloudflareResponse.Response
 | 
			
		||||
	choice.Delta.Role = "assistant"
 | 
			
		||||
	openaiResponse := openai.ChatCompletionsStreamResponse{
 | 
			
		||||
		Object:  "chat.completion.chunk",
 | 
			
		||||
		Choices: []openai.ChatCompletionsStreamResponseChoice{choice},
 | 
			
		||||
		Created: helper.GetTimestamp(),
 | 
			
		||||
	}
 | 
			
		||||
	return &openaiResponse
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func StreamHandler(c *gin.Context, resp *http.Response, promptTokens int, modelName string) (*model.ErrorWithStatusCode, *model.Usage) {
 | 
			
		||||
	scanner := bufio.NewScanner(resp.Body)
 | 
			
		||||
	scanner.Split(func(data []byte, atEOF bool) (advance int, token []byte, err error) {
 | 
			
		||||
		if atEOF && len(data) == 0 {
 | 
			
		||||
			return 0, nil, nil
 | 
			
		||||
		}
 | 
			
		||||
		if i := bytes.IndexByte(data, '\n'); i >= 0 {
 | 
			
		||||
			return i + 1, data[0:i], nil
 | 
			
		||||
		}
 | 
			
		||||
		if atEOF {
 | 
			
		||||
			return len(data), data, nil
 | 
			
		||||
		}
 | 
			
		||||
		return 0, nil, nil
 | 
			
		||||
	})
 | 
			
		||||
 | 
			
		||||
	dataChan := make(chan string)
 | 
			
		||||
	stopChan := make(chan bool)
 | 
			
		||||
	go func() {
 | 
			
		||||
		for scanner.Scan() {
 | 
			
		||||
			data := scanner.Text()
 | 
			
		||||
			if len(data) < len("data: ") {
 | 
			
		||||
				continue
 | 
			
		||||
			}
 | 
			
		||||
			data = strings.TrimPrefix(data, "data: ")
 | 
			
		||||
			dataChan <- data
 | 
			
		||||
		}
 | 
			
		||||
		stopChan <- true
 | 
			
		||||
	}()
 | 
			
		||||
	common.SetEventStreamHeaders(c)
 | 
			
		||||
	id := helper.GetResponseID(c)
 | 
			
		||||
	responseModel := c.GetString("original_model")
 | 
			
		||||
	var responseText string
 | 
			
		||||
	c.Stream(func(w io.Writer) bool {
 | 
			
		||||
		select {
 | 
			
		||||
		case data := <-dataChan:
 | 
			
		||||
			// some implementations may add \r at the end of data
 | 
			
		||||
			data = strings.TrimSuffix(data, "\r")
 | 
			
		||||
			var cloudflareResponse StreamResponse
 | 
			
		||||
			err := json.Unmarshal([]byte(data), &cloudflareResponse)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				logger.SysError("error unmarshalling stream response: " + err.Error())
 | 
			
		||||
				return true
 | 
			
		||||
			}
 | 
			
		||||
			response := StreamResponseCloudflare2OpenAI(&cloudflareResponse)
 | 
			
		||||
			if response == nil {
 | 
			
		||||
				return true
 | 
			
		||||
			}
 | 
			
		||||
			responseText += cloudflareResponse.Response
 | 
			
		||||
			response.Id = id
 | 
			
		||||
			response.Model = responseModel
 | 
			
		||||
			jsonStr, err := json.Marshal(response)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				logger.SysError("error marshalling stream response: " + err.Error())
 | 
			
		||||
				return true
 | 
			
		||||
			}
 | 
			
		||||
			c.Render(-1, common.CustomEvent{Data: "data: " + string(jsonStr)})
 | 
			
		||||
			return true
 | 
			
		||||
		case <-stopChan:
 | 
			
		||||
			c.Render(-1, common.CustomEvent{Data: "data: [DONE]"})
 | 
			
		||||
			return false
 | 
			
		||||
		}
 | 
			
		||||
	})
 | 
			
		||||
	_ = resp.Body.Close()
 | 
			
		||||
	usage := openai.ResponseText2Usage(responseText, responseModel, promptTokens)
 | 
			
		||||
	return nil, usage
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func Handler(c *gin.Context, resp *http.Response, promptTokens int, modelName string) (*model.ErrorWithStatusCode, *model.Usage) {
 | 
			
		||||
	responseBody, err := io.ReadAll(resp.Body)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return openai.ErrorWrapper(err, "read_response_body_failed", http.StatusInternalServerError), nil
 | 
			
		||||
	}
 | 
			
		||||
	err = resp.Body.Close()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return openai.ErrorWrapper(err, "close_response_body_failed", http.StatusInternalServerError), nil
 | 
			
		||||
	}
 | 
			
		||||
	var cloudflareResponse Response
 | 
			
		||||
	err = json.Unmarshal(responseBody, &cloudflareResponse)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return openai.ErrorWrapper(err, "unmarshal_response_body_failed", http.StatusInternalServerError), nil
 | 
			
		||||
	}
 | 
			
		||||
	fullTextResponse := ResponseCloudflare2OpenAI(&cloudflareResponse)
 | 
			
		||||
	fullTextResponse.Model = modelName
 | 
			
		||||
	usage := openai.ResponseText2Usage(cloudflareResponse.Result.Response, modelName, promptTokens)
 | 
			
		||||
	fullTextResponse.Usage = *usage
 | 
			
		||||
	fullTextResponse.Id = helper.GetResponseID(c)
 | 
			
		||||
	jsonResponse, err := json.Marshal(fullTextResponse)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return openai.ErrorWrapper(err, "marshal_response_body_failed", http.StatusInternalServerError), nil
 | 
			
		||||
	}
 | 
			
		||||
	c.Writer.Header().Set("Content-Type", "application/json")
 | 
			
		||||
	c.Writer.WriteHeader(resp.StatusCode)
 | 
			
		||||
	_, err = c.Writer.Write(jsonResponse)
 | 
			
		||||
	return nil, usage
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										25
									
								
								relay/adaptor/cloudflare/model.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										25
									
								
								relay/adaptor/cloudflare/model.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,25 @@
 | 
			
		||||
package cloudflare
 | 
			
		||||
 | 
			
		||||
type Request struct {
 | 
			
		||||
	Lora        string  `json:"lora,omitempty"`
 | 
			
		||||
	MaxTokens   int     `json:"max_tokens,omitempty"`
 | 
			
		||||
	Prompt      string  `json:"prompt,omitempty"`
 | 
			
		||||
	Raw         bool    `json:"raw,omitempty"`
 | 
			
		||||
	Stream      bool    `json:"stream,omitempty"`
 | 
			
		||||
	Temperature float64 `json:"temperature,omitempty"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Result struct {
 | 
			
		||||
	Response string `json:"response"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Response struct {
 | 
			
		||||
	Result   Result   `json:"result"`
 | 
			
		||||
	Success  bool     `json:"success"`
 | 
			
		||||
	Errors   []string `json:"errors"`
 | 
			
		||||
	Messages []string `json:"messages"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type StreamResponse struct {
 | 
			
		||||
	Response string `json:"response"`
 | 
			
		||||
}
 | 
			
		||||
@@ -5,3 +5,10 @@ var ModelList = []string{
 | 
			
		||||
	"command-light", "command-light-nightly",
 | 
			
		||||
	"command-r", "command-r-plus",
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
	num := len(ModelList)
 | 
			
		||||
	for i := 0; i < num; i++ {
 | 
			
		||||
		ModelList = append(ModelList, ModelList[i]+"-internet")
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -17,6 +17,10 @@ import (
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/model"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
var (
 | 
			
		||||
	WebSearchConnector = Connector{ID: "web-search"}
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func stopReasonCohere2OpenAI(reason *string) string {
 | 
			
		||||
	if reason == nil {
 | 
			
		||||
		return ""
 | 
			
		||||
@@ -45,6 +49,10 @@ func ConvertRequest(textRequest model.GeneralOpenAIRequest) *Request {
 | 
			
		||||
	if cohereRequest.Model == "" {
 | 
			
		||||
		cohereRequest.Model = "command-r"
 | 
			
		||||
	}
 | 
			
		||||
	if strings.HasSuffix(cohereRequest.Model, "-internet") {
 | 
			
		||||
		cohereRequest.Model = strings.TrimSuffix(cohereRequest.Model, "-internet")
 | 
			
		||||
		cohereRequest.Connectors = append(cohereRequest.Connectors, WebSearchConnector)
 | 
			
		||||
	}
 | 
			
		||||
	for _, message := range textRequest.Messages {
 | 
			
		||||
		if message.Role == "user" {
 | 
			
		||||
			cohereRequest.Message = message.Content.(string)
 | 
			
		||||
 
 | 
			
		||||
@@ -4,7 +4,6 @@ import (
 | 
			
		||||
	"errors"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/ctxkey"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/openai"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/meta"
 | 
			
		||||
@@ -14,10 +13,11 @@ import (
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Adaptor struct {
 | 
			
		||||
	meta *meta.Meta
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) Init(meta *meta.Meta) {
 | 
			
		||||
 | 
			
		||||
	a.meta = meta
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {
 | 
			
		||||
@@ -34,7 +34,7 @@ func (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.G
 | 
			
		||||
	if request == nil {
 | 
			
		||||
		return nil, errors.New("request is nil")
 | 
			
		||||
	}
 | 
			
		||||
	request.User = c.GetString(ctxkey.ConfigUserID)
 | 
			
		||||
	request.User = a.meta.Config.UserID
 | 
			
		||||
	return ConvertRequest(*request), nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										73
									
								
								relay/adaptor/deepl/adaptor.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										73
									
								
								relay/adaptor/deepl/adaptor.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,73 @@
 | 
			
		||||
package deepl
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"errors"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/meta"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/model"
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Adaptor struct {
 | 
			
		||||
	meta       *meta.Meta
 | 
			
		||||
	promptText string
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) Init(meta *meta.Meta) {
 | 
			
		||||
	a.meta = meta
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {
 | 
			
		||||
	return fmt.Sprintf("%s/v2/translate", meta.BaseURL), nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {
 | 
			
		||||
	adaptor.SetupCommonRequestHeader(c, req, meta)
 | 
			
		||||
	req.Header.Set("Authorization", "DeepL-Auth-Key "+meta.APIKey)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {
 | 
			
		||||
	if request == nil {
 | 
			
		||||
		return nil, errors.New("request is nil")
 | 
			
		||||
	}
 | 
			
		||||
	convertedRequest, text := ConvertRequest(*request)
 | 
			
		||||
	a.promptText = text
 | 
			
		||||
	return convertedRequest, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {
 | 
			
		||||
	if request == nil {
 | 
			
		||||
		return nil, errors.New("request is nil")
 | 
			
		||||
	}
 | 
			
		||||
	return request, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {
 | 
			
		||||
	return adaptor.DoRequestHelper(a, c, meta, requestBody)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {
 | 
			
		||||
	if meta.IsStream {
 | 
			
		||||
		err = StreamHandler(c, resp, meta.ActualModelName)
 | 
			
		||||
	} else {
 | 
			
		||||
		err = Handler(c, resp, meta.ActualModelName)
 | 
			
		||||
	}
 | 
			
		||||
	promptTokens := len(a.promptText)
 | 
			
		||||
	usage = &model.Usage{
 | 
			
		||||
		PromptTokens: promptTokens,
 | 
			
		||||
		TotalTokens:  promptTokens,
 | 
			
		||||
	}
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) GetModelList() []string {
 | 
			
		||||
	return ModelList
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) GetChannelName() string {
 | 
			
		||||
	return "deepl"
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										9
									
								
								relay/adaptor/deepl/constants.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										9
									
								
								relay/adaptor/deepl/constants.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,9 @@
 | 
			
		||||
package deepl
 | 
			
		||||
 | 
			
		||||
// https://developers.deepl.com/docs/api-reference/glossaries
 | 
			
		||||
 | 
			
		||||
var ModelList = []string{
 | 
			
		||||
	"deepl-zh",
 | 
			
		||||
	"deepl-en",
 | 
			
		||||
	"deepl-ja",
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										11
									
								
								relay/adaptor/deepl/helper.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										11
									
								
								relay/adaptor/deepl/helper.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,11 @@
 | 
			
		||||
package deepl
 | 
			
		||||
 | 
			
		||||
import "strings"
 | 
			
		||||
 | 
			
		||||
func parseLangFromModelName(modelName string) string {
 | 
			
		||||
	parts := strings.Split(modelName, "-")
 | 
			
		||||
	if len(parts) == 1 {
 | 
			
		||||
		return "ZH"
 | 
			
		||||
	}
 | 
			
		||||
	return parts[1]
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										137
									
								
								relay/adaptor/deepl/main.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										137
									
								
								relay/adaptor/deepl/main.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,137 @@
 | 
			
		||||
package deepl
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/helper"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/openai"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/constant"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/constant/finishreason"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/constant/role"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/model"
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
// https://developers.deepl.com/docs/getting-started/your-first-api-request
 | 
			
		||||
 | 
			
		||||
func ConvertRequest(textRequest model.GeneralOpenAIRequest) (*Request, string) {
 | 
			
		||||
	var text string
 | 
			
		||||
	if len(textRequest.Messages) != 0 {
 | 
			
		||||
		text = textRequest.Messages[len(textRequest.Messages)-1].StringContent()
 | 
			
		||||
	}
 | 
			
		||||
	deeplRequest := Request{
 | 
			
		||||
		TargetLang: parseLangFromModelName(textRequest.Model),
 | 
			
		||||
		Text:       []string{text},
 | 
			
		||||
	}
 | 
			
		||||
	return &deeplRequest, text
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func StreamResponseDeepL2OpenAI(deeplResponse *Response) *openai.ChatCompletionsStreamResponse {
 | 
			
		||||
	var choice openai.ChatCompletionsStreamResponseChoice
 | 
			
		||||
	if len(deeplResponse.Translations) != 0 {
 | 
			
		||||
		choice.Delta.Content = deeplResponse.Translations[0].Text
 | 
			
		||||
	}
 | 
			
		||||
	choice.Delta.Role = role.Assistant
 | 
			
		||||
	choice.FinishReason = &constant.StopFinishReason
 | 
			
		||||
	openaiResponse := openai.ChatCompletionsStreamResponse{
 | 
			
		||||
		Object:  constant.StreamObject,
 | 
			
		||||
		Created: helper.GetTimestamp(),
 | 
			
		||||
		Choices: []openai.ChatCompletionsStreamResponseChoice{choice},
 | 
			
		||||
	}
 | 
			
		||||
	return &openaiResponse
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func ResponseDeepL2OpenAI(deeplResponse *Response) *openai.TextResponse {
 | 
			
		||||
	var responseText string
 | 
			
		||||
	if len(deeplResponse.Translations) != 0 {
 | 
			
		||||
		responseText = deeplResponse.Translations[0].Text
 | 
			
		||||
	}
 | 
			
		||||
	choice := openai.TextResponseChoice{
 | 
			
		||||
		Index: 0,
 | 
			
		||||
		Message: model.Message{
 | 
			
		||||
			Role:    role.Assistant,
 | 
			
		||||
			Content: responseText,
 | 
			
		||||
			Name:    nil,
 | 
			
		||||
		},
 | 
			
		||||
		FinishReason: finishreason.Stop,
 | 
			
		||||
	}
 | 
			
		||||
	fullTextResponse := openai.TextResponse{
 | 
			
		||||
		Object:  constant.NonStreamObject,
 | 
			
		||||
		Created: helper.GetTimestamp(),
 | 
			
		||||
		Choices: []openai.TextResponseChoice{choice},
 | 
			
		||||
	}
 | 
			
		||||
	return &fullTextResponse
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func StreamHandler(c *gin.Context, resp *http.Response, modelName string) *model.ErrorWithStatusCode {
 | 
			
		||||
	responseBody, err := io.ReadAll(resp.Body)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return openai.ErrorWrapper(err, "read_response_body_failed", http.StatusInternalServerError)
 | 
			
		||||
	}
 | 
			
		||||
	err = resp.Body.Close()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return openai.ErrorWrapper(err, "close_response_body_failed", http.StatusInternalServerError)
 | 
			
		||||
	}
 | 
			
		||||
	var deeplResponse Response
 | 
			
		||||
	err = json.Unmarshal(responseBody, &deeplResponse)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return openai.ErrorWrapper(err, "unmarshal_response_body_failed", http.StatusInternalServerError)
 | 
			
		||||
	}
 | 
			
		||||
	fullTextResponse := StreamResponseDeepL2OpenAI(&deeplResponse)
 | 
			
		||||
	fullTextResponse.Model = modelName
 | 
			
		||||
	fullTextResponse.Id = helper.GetResponseID(c)
 | 
			
		||||
	jsonData, err := json.Marshal(fullTextResponse)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return openai.ErrorWrapper(err, "marshal_response_body_failed", http.StatusInternalServerError)
 | 
			
		||||
	}
 | 
			
		||||
	common.SetEventStreamHeaders(c)
 | 
			
		||||
	c.Stream(func(w io.Writer) bool {
 | 
			
		||||
		if jsonData != nil {
 | 
			
		||||
			c.Render(-1, common.CustomEvent{Data: "data: " + string(jsonData)})
 | 
			
		||||
			jsonData = nil
 | 
			
		||||
			return true
 | 
			
		||||
		}
 | 
			
		||||
		c.Render(-1, common.CustomEvent{Data: "data: [DONE]"})
 | 
			
		||||
		return false
 | 
			
		||||
	})
 | 
			
		||||
	_ = resp.Body.Close()
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func Handler(c *gin.Context, resp *http.Response, modelName string) *model.ErrorWithStatusCode {
 | 
			
		||||
	responseBody, err := io.ReadAll(resp.Body)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return openai.ErrorWrapper(err, "read_response_body_failed", http.StatusInternalServerError)
 | 
			
		||||
	}
 | 
			
		||||
	err = resp.Body.Close()
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return openai.ErrorWrapper(err, "close_response_body_failed", http.StatusInternalServerError)
 | 
			
		||||
	}
 | 
			
		||||
	var deeplResponse Response
 | 
			
		||||
	err = json.Unmarshal(responseBody, &deeplResponse)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return openai.ErrorWrapper(err, "unmarshal_response_body_failed", http.StatusInternalServerError)
 | 
			
		||||
	}
 | 
			
		||||
	if deeplResponse.Message != "" {
 | 
			
		||||
		return &model.ErrorWithStatusCode{
 | 
			
		||||
			Error: model.Error{
 | 
			
		||||
				Message: deeplResponse.Message,
 | 
			
		||||
				Code:    "deepl_error",
 | 
			
		||||
			},
 | 
			
		||||
			StatusCode: resp.StatusCode,
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	fullTextResponse := ResponseDeepL2OpenAI(&deeplResponse)
 | 
			
		||||
	fullTextResponse.Model = modelName
 | 
			
		||||
	fullTextResponse.Id = helper.GetResponseID(c)
 | 
			
		||||
	jsonResponse, err := json.Marshal(fullTextResponse)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return openai.ErrorWrapper(err, "marshal_response_body_failed", http.StatusInternalServerError)
 | 
			
		||||
	}
 | 
			
		||||
	c.Writer.Header().Set("Content-Type", "application/json")
 | 
			
		||||
	c.Writer.WriteHeader(resp.StatusCode)
 | 
			
		||||
	_, err = c.Writer.Write(jsonResponse)
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										16
									
								
								relay/adaptor/deepl/model.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										16
									
								
								relay/adaptor/deepl/model.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,16 @@
 | 
			
		||||
package deepl
 | 
			
		||||
 | 
			
		||||
type Request struct {
 | 
			
		||||
	Text       []string `json:"text"`
 | 
			
		||||
	TargetLang string   `json:"target_lang"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Translation struct {
 | 
			
		||||
	DetectedSourceLanguage string `json:"detected_source_language,omitempty"`
 | 
			
		||||
	Text                   string `json:"text,omitempty"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type Response struct {
 | 
			
		||||
	Translations []Translation `json:"translations,omitempty"`
 | 
			
		||||
	Message      string        `json:"message,omitempty"`
 | 
			
		||||
}
 | 
			
		||||
							
								
								
									
										6
									
								
								relay/adaptor/deepseek/constants.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										6
									
								
								relay/adaptor/deepseek/constants.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,6 @@
 | 
			
		||||
package deepseek
 | 
			
		||||
 | 
			
		||||
var ModelList = []string{
 | 
			
		||||
	"deepseek-chat",
 | 
			
		||||
	"deepseek-coder",
 | 
			
		||||
}
 | 
			
		||||
@@ -3,6 +3,9 @@ package gemini
 | 
			
		||||
import (
 | 
			
		||||
	"errors"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/config"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/helper"
 | 
			
		||||
@@ -10,8 +13,6 @@ import (
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/openai"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/meta"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/model"
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
type Adaptor struct {
 | 
			
		||||
@@ -22,10 +23,10 @@ func (a *Adaptor) Init(meta *meta.Meta) {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {
 | 
			
		||||
	version := helper.AssignOrDefault(meta.APIVersion, config.GeminiVersion)
 | 
			
		||||
	version := helper.AssignOrDefault(meta.Config.APIVersion, config.GeminiVersion)
 | 
			
		||||
	action := "generateContent"
 | 
			
		||||
	if meta.IsStream {
 | 
			
		||||
		action = "streamGenerateContent"
 | 
			
		||||
		action = "streamGenerateContent?alt=sse"
 | 
			
		||||
	}
 | 
			
		||||
	return fmt.Sprintf("%s/%s/models/%s:%s", meta.BaseURL, version, meta.ActualModelName, action), nil
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -232,8 +232,6 @@ func streamResponseGeminiChat2OpenAI(geminiResponse *ChatResponse) *openai.ChatC
 | 
			
		||||
 | 
			
		||||
func StreamHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, string) {
 | 
			
		||||
	responseText := ""
 | 
			
		||||
	dataChan := make(chan string)
 | 
			
		||||
	stopChan := make(chan bool)
 | 
			
		||||
	scanner := bufio.NewScanner(resp.Body)
 | 
			
		||||
	scanner.Split(func(data []byte, atEOF bool) (advance int, token []byte, err error) {
 | 
			
		||||
		if atEOF && len(data) == 0 {
 | 
			
		||||
@@ -247,14 +245,16 @@ func StreamHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusC
 | 
			
		||||
		}
 | 
			
		||||
		return 0, nil, nil
 | 
			
		||||
	})
 | 
			
		||||
	dataChan := make(chan string)
 | 
			
		||||
	stopChan := make(chan bool)
 | 
			
		||||
	go func() {
 | 
			
		||||
		for scanner.Scan() {
 | 
			
		||||
			data := scanner.Text()
 | 
			
		||||
			data = strings.TrimSpace(data)
 | 
			
		||||
			if !strings.HasPrefix(data, "\"text\": \"") {
 | 
			
		||||
			if !strings.HasPrefix(data, "data: ") {
 | 
			
		||||
				continue
 | 
			
		||||
			}
 | 
			
		||||
			data = strings.TrimPrefix(data, "\"text\": \"")
 | 
			
		||||
			data = strings.TrimPrefix(data, "data: ")
 | 
			
		||||
			data = strings.TrimSuffix(data, "\"")
 | 
			
		||||
			dataChan <- data
 | 
			
		||||
		}
 | 
			
		||||
@@ -264,23 +264,17 @@ func StreamHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusC
 | 
			
		||||
	c.Stream(func(w io.Writer) bool {
 | 
			
		||||
		select {
 | 
			
		||||
		case data := <-dataChan:
 | 
			
		||||
			// this is used to prevent annoying \ related format bug
 | 
			
		||||
			data = fmt.Sprintf("{\"content\": \"%s\"}", data)
 | 
			
		||||
			type dummyStruct struct {
 | 
			
		||||
				Content string `json:"content"`
 | 
			
		||||
			var geminiResponse ChatResponse
 | 
			
		||||
			err := json.Unmarshal([]byte(data), &geminiResponse)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				logger.SysError("error unmarshalling stream response: " + err.Error())
 | 
			
		||||
				return true
 | 
			
		||||
			}
 | 
			
		||||
			var dummy dummyStruct
 | 
			
		||||
			err := json.Unmarshal([]byte(data), &dummy)
 | 
			
		||||
			responseText += dummy.Content
 | 
			
		||||
			var choice openai.ChatCompletionsStreamResponseChoice
 | 
			
		||||
			choice.Delta.Content = dummy.Content
 | 
			
		||||
			response := openai.ChatCompletionsStreamResponse{
 | 
			
		||||
				Id:      fmt.Sprintf("chatcmpl-%s", random.GetUUID()),
 | 
			
		||||
				Object:  "chat.completion.chunk",
 | 
			
		||||
				Created: helper.GetTimestamp(),
 | 
			
		||||
				Model:   "gemini-pro",
 | 
			
		||||
				Choices: []openai.ChatCompletionsStreamResponseChoice{choice},
 | 
			
		||||
			response := streamResponseGeminiChat2OpenAI(&geminiResponse)
 | 
			
		||||
			if response == nil {
 | 
			
		||||
				return true
 | 
			
		||||
			}
 | 
			
		||||
			responseText += response.Choices[0].Delta.StringContent()
 | 
			
		||||
			jsonResponse, err := json.Marshal(response)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				logger.SysError("error marshalling stream response: " + err.Error())
 | 
			
		||||
 
 | 
			
		||||
@@ -1,7 +1,11 @@
 | 
			
		||||
package minimax
 | 
			
		||||
 | 
			
		||||
// https://www.minimaxi.com/document/guides/chat-model/V2?id=65e0736ab2845de20908e2dd
 | 
			
		||||
 | 
			
		||||
var ModelList = []string{
 | 
			
		||||
	"abab5.5s-chat",
 | 
			
		||||
	"abab5.5-chat",
 | 
			
		||||
	"abab6.5-chat",
 | 
			
		||||
	"abab6.5s-chat",
 | 
			
		||||
	"abab6-chat",
 | 
			
		||||
	"abab5.5-chat",
 | 
			
		||||
	"abab5.5s-chat",
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -1,5 +1,11 @@
 | 
			
		||||
package ollama
 | 
			
		||||
 | 
			
		||||
var ModelList = []string{
 | 
			
		||||
	"codellama:7b-instruct",
 | 
			
		||||
	"llama2:7b",
 | 
			
		||||
	"llama2:latest",
 | 
			
		||||
	"llama3:latest",
 | 
			
		||||
	"phi3:latest",
 | 
			
		||||
	"qwen:0.5b-chat",
 | 
			
		||||
	"qwen:7b",
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -13,6 +13,7 @@ import (
 | 
			
		||||
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/image"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/logger"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/openai"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/constant"
 | 
			
		||||
@@ -32,9 +33,22 @@ func ConvertRequest(request model.GeneralOpenAIRequest) *ChatRequest {
 | 
			
		||||
		Stream: request.Stream,
 | 
			
		||||
	}
 | 
			
		||||
	for _, message := range request.Messages {
 | 
			
		||||
		openaiContent := message.ParseContent()
 | 
			
		||||
		var imageUrls []string
 | 
			
		||||
		var contentText string
 | 
			
		||||
		for _, part := range openaiContent {
 | 
			
		||||
			switch part.Type {
 | 
			
		||||
			case model.ContentTypeText:
 | 
			
		||||
				contentText = part.Text
 | 
			
		||||
			case model.ContentTypeImageURL:
 | 
			
		||||
				_, data, _ := image.GetImageFromUrl(part.ImageURL.Url)
 | 
			
		||||
				imageUrls = append(imageUrls, data)
 | 
			
		||||
			}
 | 
			
		||||
		}
 | 
			
		||||
		ollamaRequest.Messages = append(ollamaRequest.Messages, Message{
 | 
			
		||||
			Role:    message.Role,
 | 
			
		||||
			Content: message.StringContent(),
 | 
			
		||||
			Content: contentText,
 | 
			
		||||
			Images:  imageUrls,
 | 
			
		||||
		})
 | 
			
		||||
	}
 | 
			
		||||
	return &ollamaRequest
 | 
			
		||||
@@ -53,6 +67,7 @@ func responseOllama2OpenAI(response *ChatResponse) *openai.TextResponse {
 | 
			
		||||
	}
 | 
			
		||||
	fullTextResponse := openai.TextResponse{
 | 
			
		||||
		Id:      fmt.Sprintf("chatcmpl-%s", random.GetUUID()),
 | 
			
		||||
		Model:   response.Model,
 | 
			
		||||
		Object:  "chat.completion",
 | 
			
		||||
		Created: helper.GetTimestamp(),
 | 
			
		||||
		Choices: []openai.TextResponseChoice{choice},
 | 
			
		||||
 
 | 
			
		||||
@@ -29,13 +29,13 @@ func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {
 | 
			
		||||
		if meta.Mode == relaymode.ImagesGenerations {
 | 
			
		||||
			// https://learn.microsoft.com/en-us/azure/ai-services/openai/dall-e-quickstart?tabs=dalle3%2Ccommand-line&pivots=rest-api
 | 
			
		||||
			// https://{resource_name}.openai.azure.com/openai/deployments/dall-e-3/images/generations?api-version=2024-03-01-preview
 | 
			
		||||
			fullRequestURL := fmt.Sprintf("%s/openai/deployments/%s/images/generations?api-version=%s", meta.BaseURL, meta.ActualModelName, meta.APIVersion)
 | 
			
		||||
			fullRequestURL := fmt.Sprintf("%s/openai/deployments/%s/images/generations?api-version=%s", meta.BaseURL, meta.ActualModelName, meta.Config.APIVersion)
 | 
			
		||||
			return fullRequestURL, nil
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		// https://learn.microsoft.com/en-us/azure/cognitive-services/openai/chatgpt-quickstart?pivots=rest-api&tabs=command-line#rest-api
 | 
			
		||||
		requestURL := strings.Split(meta.RequestURLPath, "?")[0]
 | 
			
		||||
		requestURL = fmt.Sprintf("%s?api-version=%s", requestURL, meta.APIVersion)
 | 
			
		||||
		requestURL = fmt.Sprintf("%s?api-version=%s", requestURL, meta.Config.APIVersion)
 | 
			
		||||
		task := strings.TrimPrefix(requestURL, "/v1/")
 | 
			
		||||
		model_ := meta.ActualModelName
 | 
			
		||||
		model_ = strings.Replace(model_, ".", "", -1)
 | 
			
		||||
@@ -86,9 +86,13 @@ func (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Met
 | 
			
		||||
	if meta.IsStream {
 | 
			
		||||
		var responseText string
 | 
			
		||||
		err, responseText, usage = StreamHandler(c, resp, meta.Mode)
 | 
			
		||||
		if usage == nil {
 | 
			
		||||
		if usage == nil || usage.TotalTokens == 0 {
 | 
			
		||||
			usage = ResponseText2Usage(responseText, meta.ActualModelName, meta.PromptTokens)
 | 
			
		||||
		}
 | 
			
		||||
		if usage.TotalTokens != 0 && usage.PromptTokens == 0 { // some channels don't return prompt tokens & completion tokens
 | 
			
		||||
			usage.PromptTokens = meta.PromptTokens
 | 
			
		||||
			usage.CompletionTokens = usage.TotalTokens - meta.PromptTokens
 | 
			
		||||
		}
 | 
			
		||||
	} else {
 | 
			
		||||
		switch meta.Mode {
 | 
			
		||||
		case relaymode.ImagesGenerations:
 | 
			
		||||
 
 | 
			
		||||
@@ -3,12 +3,14 @@ package openai
 | 
			
		||||
import (
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/ai360"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/baichuan"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/deepseek"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/groq"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/lingyiwanwu"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/minimax"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/mistral"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/moonshot"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/stepfun"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/togetherai"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/channeltype"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
@@ -22,6 +24,8 @@ var CompatibleChannels = []int{
 | 
			
		||||
	channeltype.Groq,
 | 
			
		||||
	channeltype.LingYiWanWu,
 | 
			
		||||
	channeltype.StepFun,
 | 
			
		||||
	channeltype.DeepSeek,
 | 
			
		||||
	channeltype.TogetherAI,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func GetCompatibleChannelMeta(channelType int) (string, []string) {
 | 
			
		||||
@@ -44,6 +48,10 @@ func GetCompatibleChannelMeta(channelType int) (string, []string) {
 | 
			
		||||
		return "lingyiwanwu", lingyiwanwu.ModelList
 | 
			
		||||
	case channeltype.StepFun:
 | 
			
		||||
		return "stepfun", stepfun.ModelList
 | 
			
		||||
	case channeltype.DeepSeek:
 | 
			
		||||
		return "deepseek", deepseek.ModelList
 | 
			
		||||
	case channeltype.TogetherAI:
 | 
			
		||||
		return "together.ai", togetherai.ModelList
 | 
			
		||||
	default:
 | 
			
		||||
		return "openai", ModelList
 | 
			
		||||
	}
 | 
			
		||||
 
 | 
			
		||||
@@ -7,6 +7,7 @@ var ModelList = []string{
 | 
			
		||||
	"gpt-4", "gpt-4-0314", "gpt-4-0613", "gpt-4-1106-preview", "gpt-4-0125-preview",
 | 
			
		||||
	"gpt-4-32k", "gpt-4-32k-0314", "gpt-4-32k-0613",
 | 
			
		||||
	"gpt-4-turbo-preview", "gpt-4-turbo", "gpt-4-turbo-2024-04-09",
 | 
			
		||||
	"gpt-4o", "gpt-4o-2024-05-13",
 | 
			
		||||
	"gpt-4-vision-preview",
 | 
			
		||||
	"text-embedding-ada-002", "text-embedding-3-small", "text-embedding-3-large",
 | 
			
		||||
	"text-curie-001", "text-babbage-001", "text-ada-001", "text-davinci-002", "text-davinci-003",
 | 
			
		||||
 
 | 
			
		||||
@@ -134,7 +134,7 @@ type ChatCompletionsStreamResponse struct {
 | 
			
		||||
	Created int64                                 `json:"created"`
 | 
			
		||||
	Model   string                                `json:"model"`
 | 
			
		||||
	Choices []ChatCompletionsStreamResponseChoice `json:"choices"`
 | 
			
		||||
	Usage   *model.Usage                          `json:"usage"`
 | 
			
		||||
	Usage   *model.Usage                          `json:"usage,omitempty"`
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
type CompletionsStreamResponse struct {
 | 
			
		||||
 
 | 
			
		||||
@@ -24,6 +24,10 @@ func InitTokenEncoders() {
 | 
			
		||||
		logger.FatalLog(fmt.Sprintf("failed to get gpt-3.5-turbo token encoder: %s", err.Error()))
 | 
			
		||||
	}
 | 
			
		||||
	defaultTokenEncoder = gpt35TokenEncoder
 | 
			
		||||
	gpt4oTokenEncoder, err := tiktoken.EncodingForModel("gpt-4o")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logger.FatalLog(fmt.Sprintf("failed to get gpt-4o token encoder: %s", err.Error()))
 | 
			
		||||
	}
 | 
			
		||||
	gpt4TokenEncoder, err := tiktoken.EncodingForModel("gpt-4")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		logger.FatalLog(fmt.Sprintf("failed to get gpt-4 token encoder: %s", err.Error()))
 | 
			
		||||
@@ -31,6 +35,8 @@ func InitTokenEncoders() {
 | 
			
		||||
	for model := range billingratio.ModelRatio {
 | 
			
		||||
		if strings.HasPrefix(model, "gpt-3.5") {
 | 
			
		||||
			tokenEncoderMap[model] = gpt35TokenEncoder
 | 
			
		||||
		} else if strings.HasPrefix(model, "gpt-4o") {
 | 
			
		||||
			tokenEncoderMap[model] = gpt4oTokenEncoder
 | 
			
		||||
		} else if strings.HasPrefix(model, "gpt-4") {
 | 
			
		||||
			tokenEncoderMap[model] = gpt4TokenEncoder
 | 
			
		||||
		} else {
 | 
			
		||||
@@ -206,3 +212,7 @@ func CountTokenText(text string, model string) int {
 | 
			
		||||
	tokenEncoder := getTokenEncoder(model)
 | 
			
		||||
	return getTokenNum(tokenEncoder, text)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func CountToken(text string) int {
 | 
			
		||||
	return CountTokenInput(text, "gpt-3.5-turbo")
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										10
									
								
								relay/adaptor/togetherai/constants.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										10
									
								
								relay/adaptor/togetherai/constants.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,10 @@
 | 
			
		||||
package togetherai
 | 
			
		||||
 | 
			
		||||
// https://docs.together.ai/docs/inference-models
 | 
			
		||||
 | 
			
		||||
var ModelList = []string{
 | 
			
		||||
	"meta-llama/Llama-3-70b-chat-hf",
 | 
			
		||||
	"deepseek-ai/deepseek-coder-33b-instruct",
 | 
			
		||||
	"mistralai/Mixtral-8x22B-Instruct-v0.1",
 | 
			
		||||
	"Qwen/Qwen1.5-72B-Chat",
 | 
			
		||||
}
 | 
			
		||||
@@ -14,10 +14,11 @@ import (
 | 
			
		||||
 | 
			
		||||
type Adaptor struct {
 | 
			
		||||
	request *model.GeneralOpenAIRequest
 | 
			
		||||
	meta    *meta.Meta
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) Init(meta *meta.Meta) {
 | 
			
		||||
 | 
			
		||||
	a.meta = meta
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {
 | 
			
		||||
@@ -26,6 +27,14 @@ func (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {
 | 
			
		||||
	adaptor.SetupCommonRequestHeader(c, req, meta)
 | 
			
		||||
	version := parseAPIVersionByModelName(meta.ActualModelName)
 | 
			
		||||
	if version == "" {
 | 
			
		||||
		version = a.meta.Config.APIVersion
 | 
			
		||||
	}
 | 
			
		||||
	if version == "" {
 | 
			
		||||
		version = "v1.1"
 | 
			
		||||
	}
 | 
			
		||||
	a.meta.Config.APIVersion = version
 | 
			
		||||
	// check DoResponse for auth part
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
@@ -61,9 +70,9 @@ func (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Met
 | 
			
		||||
		return nil, openai.ErrorWrapper(errors.New("request is nil"), "request_is_nil", http.StatusBadRequest)
 | 
			
		||||
	}
 | 
			
		||||
	if meta.IsStream {
 | 
			
		||||
		err, usage = StreamHandler(c, *a.request, splits[0], splits[1], splits[2])
 | 
			
		||||
		err, usage = StreamHandler(c, meta, *a.request, splits[0], splits[1], splits[2])
 | 
			
		||||
	} else {
 | 
			
		||||
		err, usage = Handler(c, *a.request, splits[0], splits[1], splits[2])
 | 
			
		||||
		err, usage = Handler(c, meta, *a.request, splits[0], splits[1], splits[2])
 | 
			
		||||
	}
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -9,12 +9,12 @@ import (
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
	"github.com/gorilla/websocket"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/ctxkey"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/helper"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/logger"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/random"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/openai"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/constant"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/meta"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/model"
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
@@ -149,8 +149,8 @@ func buildXunfeiAuthUrl(hostUrl string, apiKey, apiSecret string) string {
 | 
			
		||||
	return callUrl
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func StreamHandler(c *gin.Context, textRequest model.GeneralOpenAIRequest, appId string, apiSecret string, apiKey string) (*model.ErrorWithStatusCode, *model.Usage) {
 | 
			
		||||
	domain, authUrl := getXunfeiAuthUrl(c, apiKey, apiSecret, textRequest.Model)
 | 
			
		||||
func StreamHandler(c *gin.Context, meta *meta.Meta, textRequest model.GeneralOpenAIRequest, appId string, apiSecret string, apiKey string) (*model.ErrorWithStatusCode, *model.Usage) {
 | 
			
		||||
	domain, authUrl := getXunfeiAuthUrl(meta.Config.APIVersion, apiKey, apiSecret)
 | 
			
		||||
	dataChan, stopChan, err := xunfeiMakeRequest(textRequest, domain, authUrl, appId)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return openai.ErrorWrapper(err, "xunfei_request_failed", http.StatusInternalServerError), nil
 | 
			
		||||
@@ -179,8 +179,8 @@ func StreamHandler(c *gin.Context, textRequest model.GeneralOpenAIRequest, appId
 | 
			
		||||
	return nil, &usage
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func Handler(c *gin.Context, textRequest model.GeneralOpenAIRequest, appId string, apiSecret string, apiKey string) (*model.ErrorWithStatusCode, *model.Usage) {
 | 
			
		||||
	domain, authUrl := getXunfeiAuthUrl(c, apiKey, apiSecret, textRequest.Model)
 | 
			
		||||
func Handler(c *gin.Context, meta *meta.Meta, textRequest model.GeneralOpenAIRequest, appId string, apiSecret string, apiKey string) (*model.ErrorWithStatusCode, *model.Usage) {
 | 
			
		||||
	domain, authUrl := getXunfeiAuthUrl(meta.Config.APIVersion, apiKey, apiSecret)
 | 
			
		||||
	dataChan, stopChan, err := xunfeiMakeRequest(textRequest, domain, authUrl, appId)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		return openai.ErrorWrapper(err, "xunfei_request_failed", http.StatusInternalServerError), nil
 | 
			
		||||
@@ -268,25 +268,12 @@ func xunfeiMakeRequest(textRequest model.GeneralOpenAIRequest, domain, authUrl,
 | 
			
		||||
	return dataChan, stopChan, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func getAPIVersion(c *gin.Context, modelName string) string {
 | 
			
		||||
	query := c.Request.URL.Query()
 | 
			
		||||
	apiVersion := query.Get("api-version")
 | 
			
		||||
	if apiVersion != "" {
 | 
			
		||||
		return apiVersion
 | 
			
		||||
	}
 | 
			
		||||
func parseAPIVersionByModelName(modelName string) string {
 | 
			
		||||
	parts := strings.Split(modelName, "-")
 | 
			
		||||
	if len(parts) == 2 {
 | 
			
		||||
		apiVersion = parts[1]
 | 
			
		||||
		return apiVersion
 | 
			
		||||
 | 
			
		||||
		return parts[1]
 | 
			
		||||
	}
 | 
			
		||||
	apiVersion = c.GetString(ctxkey.ConfigAPIVersion)
 | 
			
		||||
	if apiVersion != "" {
 | 
			
		||||
		return apiVersion
 | 
			
		||||
	}
 | 
			
		||||
	apiVersion = "v1.1"
 | 
			
		||||
	logger.SysLog("api_version not found, using default: " + apiVersion)
 | 
			
		||||
	return apiVersion
 | 
			
		||||
	return ""
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// https://www.xfyun.cn/doc/spark/Web.html#_1-%E6%8E%A5%E5%8F%A3%E8%AF%B4%E6%98%8E
 | 
			
		||||
@@ -304,8 +291,7 @@ func apiVersion2domain(apiVersion string) string {
 | 
			
		||||
	return "general" + apiVersion
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func getXunfeiAuthUrl(c *gin.Context, apiKey string, apiSecret string, modelName string) (string, string) {
 | 
			
		||||
	apiVersion := getAPIVersion(c, modelName)
 | 
			
		||||
func getXunfeiAuthUrl(apiVersion string, apiKey string, apiSecret string) (string, string) {
 | 
			
		||||
	domain := apiVersion2domain(apiVersion)
 | 
			
		||||
	authUrl := buildXunfeiAuthUrl(fmt.Sprintf("wss://spark-api.xf-yun.com/%s/chat", apiVersion), apiKey, apiSecret)
 | 
			
		||||
	return domain, authUrl
 | 
			
		||||
 
 | 
			
		||||
@@ -62,8 +62,8 @@ func (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.G
 | 
			
		||||
	}
 | 
			
		||||
	switch relayMode {
 | 
			
		||||
	case relaymode.Embeddings:
 | 
			
		||||
		baiduEmbeddingRequest := ConvertEmbeddingRequest(*request)
 | 
			
		||||
		return baiduEmbeddingRequest, nil
 | 
			
		||||
		baiduEmbeddingRequest, err := ConvertEmbeddingRequest(*request)
 | 
			
		||||
		return baiduEmbeddingRequest, err
 | 
			
		||||
	default:
 | 
			
		||||
		// TopP (0.0, 1.0)
 | 
			
		||||
		request.TopP = math.Min(0.99, request.TopP)
 | 
			
		||||
@@ -129,11 +129,15 @@ func (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Met
 | 
			
		||||
	return
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func ConvertEmbeddingRequest(request model.GeneralOpenAIRequest) *EmbeddingRequest {
 | 
			
		||||
	return &EmbeddingRequest{
 | 
			
		||||
		Model: "embedding-2",
 | 
			
		||||
		Input: request.Input.(string),
 | 
			
		||||
func ConvertEmbeddingRequest(request model.GeneralOpenAIRequest) (*EmbeddingRequest, error) {
 | 
			
		||||
	inputs := request.ParseInput()
 | 
			
		||||
	if len(inputs) != 1 {
 | 
			
		||||
		return nil, errors.New("invalid input length, zhipu only support one input")
 | 
			
		||||
	}
 | 
			
		||||
	return &EmbeddingRequest{
 | 
			
		||||
		Model: request.Model,
 | 
			
		||||
		Input: inputs[0],
 | 
			
		||||
	}, nil
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func (a *Adaptor) GetModelList() []string {
 | 
			
		||||
 
 | 
			
		||||
@@ -15,6 +15,8 @@ const (
 | 
			
		||||
	AwsClaude
 | 
			
		||||
	Coze
 | 
			
		||||
	Cohere
 | 
			
		||||
	Cloudflare
 | 
			
		||||
	DeepL
 | 
			
		||||
 | 
			
		||||
	Dummy // this one is only for count, do not add any channel after this
 | 
			
		||||
)
 | 
			
		||||
 
 | 
			
		||||
@@ -49,3 +49,8 @@ var ImagePromptLengthLimitations = map[string]int{
 | 
			
		||||
	"wanx-v1":                   4000,
 | 
			
		||||
	"cogview-3":                 833,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var ImageOriginModelName = map[string]string{
 | 
			
		||||
	"ali-stable-diffusion-xl":   "stable-diffusion-xl",
 | 
			
		||||
	"ali-stable-diffusion-v1.5": "stable-diffusion-v1.5",
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -32,6 +32,8 @@ var ModelRatio = map[string]float64{
 | 
			
		||||
	"gpt-4-turbo-preview":     5,    // $0.01 / 1K tokens
 | 
			
		||||
	"gpt-4-turbo":             5,    // $0.01 / 1K tokens
 | 
			
		||||
	"gpt-4-turbo-2024-04-09":  5,    // $0.01 / 1K tokens
 | 
			
		||||
	"gpt-4o":                  2.5,  // $0.005 / 1K tokens
 | 
			
		||||
	"gpt-4o-2024-05-13":       2.5,  // $0.005 / 1K tokens
 | 
			
		||||
	"gpt-4-vision-preview":    5,    // $0.01 / 1K tokens
 | 
			
		||||
	"gpt-3.5-turbo":           0.25, // $0.0005 / 1K tokens
 | 
			
		||||
	"gpt-3.5-turbo-0301":      0.75,
 | 
			
		||||
@@ -138,6 +140,8 @@ var ModelRatio = map[string]float64{
 | 
			
		||||
	"Baichuan2-Turbo-192k": 0.016 * RMB,
 | 
			
		||||
	"Baichuan2-53B":        0.02 * RMB,
 | 
			
		||||
	// https://api.minimax.chat/document/price
 | 
			
		||||
	"abab6.5-chat":  0.03 * RMB,
 | 
			
		||||
	"abab6.5s-chat": 0.01 * RMB,
 | 
			
		||||
	"abab6-chat":    0.1 * RMB,
 | 
			
		||||
	"abab5.5-chat":  0.015 * RMB,
 | 
			
		||||
	"abab5.5s-chat": 0.005 * RMB,
 | 
			
		||||
@@ -169,7 +173,14 @@ var ModelRatio = map[string]float64{
 | 
			
		||||
	"command-light":         0.5,
 | 
			
		||||
	"command-light-nightly": 0.5,
 | 
			
		||||
	"command-r":             0.5 / 1000 * USD,
 | 
			
		||||
	"command-r-plus	":       3.0 / 1000 * USD,
 | 
			
		||||
	"command-r-plus":        3.0 / 1000 * USD,
 | 
			
		||||
	// https://platform.deepseek.com/api-docs/pricing/
 | 
			
		||||
	"deepseek-chat":  1.0 / 1000 * RMB,
 | 
			
		||||
	"deepseek-coder": 1.0 / 1000 * RMB,
 | 
			
		||||
	// https://www.deepl.com/pro?cta=header-prices
 | 
			
		||||
	"deepl-zh": 25.0 / 1000 * USD,
 | 
			
		||||
	"deepl-en": 25.0 / 1000 * USD,
 | 
			
		||||
	"deepl-ja": 25.0 / 1000 * USD,
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
var CompletionRatio = map[string]float64{}
 | 
			
		||||
@@ -225,6 +236,9 @@ func GetModelRatio(name string) float64 {
 | 
			
		||||
	if strings.HasPrefix(name, "qwen-") && strings.HasSuffix(name, "-internet") {
 | 
			
		||||
		name = strings.TrimSuffix(name, "-internet")
 | 
			
		||||
	}
 | 
			
		||||
	if strings.HasPrefix(name, "command-") && strings.HasSuffix(name, "-internet") {
 | 
			
		||||
		name = strings.TrimSuffix(name, "-internet")
 | 
			
		||||
	}
 | 
			
		||||
	ratio, ok := ModelRatio[name]
 | 
			
		||||
	if !ok {
 | 
			
		||||
		ratio, ok = DefaultModelRatio[name]
 | 
			
		||||
@@ -268,7 +282,9 @@ func GetCompletionRatio(name string) float64 {
 | 
			
		||||
		return 4.0 / 3.0
 | 
			
		||||
	}
 | 
			
		||||
	if strings.HasPrefix(name, "gpt-4") {
 | 
			
		||||
		if strings.HasPrefix(name, "gpt-4-turbo") || strings.HasSuffix(name, "preview") {
 | 
			
		||||
		if strings.HasPrefix(name, "gpt-4-turbo") ||
 | 
			
		||||
			strings.HasPrefix(name, "gpt-4o") ||
 | 
			
		||||
			strings.HasSuffix(name, "preview") {
 | 
			
		||||
			return 3
 | 
			
		||||
		}
 | 
			
		||||
		return 2
 | 
			
		||||
@@ -285,6 +301,9 @@ func GetCompletionRatio(name string) float64 {
 | 
			
		||||
	if strings.HasPrefix(name, "gemini-") {
 | 
			
		||||
		return 3
 | 
			
		||||
	}
 | 
			
		||||
	if strings.HasPrefix(name, "deepseek-") {
 | 
			
		||||
		return 2
 | 
			
		||||
	}
 | 
			
		||||
	switch name {
 | 
			
		||||
	case "llama2-70b-4096":
 | 
			
		||||
		return 0.8 / 0.64
 | 
			
		||||
 
 | 
			
		||||
@@ -37,6 +37,10 @@ const (
 | 
			
		||||
	AwsClaude
 | 
			
		||||
	Coze
 | 
			
		||||
	Cohere
 | 
			
		||||
	DeepSeek
 | 
			
		||||
	Cloudflare
 | 
			
		||||
	DeepL
 | 
			
		||||
	TogetherAI
 | 
			
		||||
 | 
			
		||||
	Dummy
 | 
			
		||||
)
 | 
			
		||||
 
 | 
			
		||||
@@ -31,6 +31,10 @@ func ToAPIType(channelType int) int {
 | 
			
		||||
		apiType = apitype.Coze
 | 
			
		||||
	case Cohere:
 | 
			
		||||
		apiType = apitype.Cohere
 | 
			
		||||
	case Cloudflare:
 | 
			
		||||
		apiType = apitype.Cloudflare
 | 
			
		||||
	case DeepL:
 | 
			
		||||
		apiType = apitype.DeepL
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return apiType
 | 
			
		||||
 
 | 
			
		||||
@@ -36,7 +36,11 @@ var ChannelBaseURLs = []string{
 | 
			
		||||
	"https://api.stepfun.com",                   // 32
 | 
			
		||||
	"",                                          // 33
 | 
			
		||||
	"https://api.coze.com",                      // 34
 | 
			
		||||
	"https://api.cohere.ai",                     //35
 | 
			
		||||
	"https://api.cohere.ai",                     // 35
 | 
			
		||||
	"https://api.deepseek.com",                  // 36
 | 
			
		||||
	"https://api.cloudflare.com",                // 37
 | 
			
		||||
	"https://api-free.deepl.com",                // 38
 | 
			
		||||
	"https://api.together.xyz",                  // 39
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func init() {
 | 
			
		||||
 
 | 
			
		||||
@@ -1,3 +1,5 @@
 | 
			
		||||
package constant
 | 
			
		||||
 | 
			
		||||
var StopFinishReason = "stop"
 | 
			
		||||
var StreamObject = "chat.completion.chunk"
 | 
			
		||||
var NonStreamObject = "chat.completion"
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										5
									
								
								relay/constant/finishreason/define.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										5
									
								
								relay/constant/finishreason/define.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,5 @@
 | 
			
		||||
package finishreason
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	Stop = "stop"
 | 
			
		||||
)
 | 
			
		||||
							
								
								
									
										5
									
								
								relay/constant/role/define.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										5
									
								
								relay/constant/role/define.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,5 @@
 | 
			
		||||
package role
 | 
			
		||||
 | 
			
		||||
const (
 | 
			
		||||
	Assistant = "assistant"
 | 
			
		||||
)
 | 
			
		||||
@@ -13,12 +13,12 @@ import (
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/ctxkey"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/logger"
 | 
			
		||||
	"github.com/songquanpeng/one-api/model"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/azure"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/openai"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/billing"
 | 
			
		||||
	billingratio "github.com/songquanpeng/one-api/relay/billing/ratio"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/channeltype"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/client"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/meta"
 | 
			
		||||
	relaymodel "github.com/songquanpeng/one-api/relay/model"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/relaymode"
 | 
			
		||||
	"io"
 | 
			
		||||
@@ -28,6 +28,7 @@ import (
 | 
			
		||||
 | 
			
		||||
func RelayAudioHelper(c *gin.Context, relayMode int) *relaymodel.ErrorWithStatusCode {
 | 
			
		||||
	ctx := c.Request.Context()
 | 
			
		||||
	meta := meta.GetByContext(c)
 | 
			
		||||
	audioModel := "whisper-1"
 | 
			
		||||
 | 
			
		||||
	tokenId := c.GetInt(ctxkey.TokenId)
 | 
			
		||||
@@ -128,7 +129,7 @@ func RelayAudioHelper(c *gin.Context, relayMode int) *relaymodel.ErrorWithStatus
 | 
			
		||||
 | 
			
		||||
	fullRequestURL := openai.GetFullRequestURL(baseURL, requestURL, channelType)
 | 
			
		||||
	if channelType == channeltype.Azure {
 | 
			
		||||
		apiVersion := azure.GetAPIVersion(c)
 | 
			
		||||
		apiVersion := meta.Config.APIVersion
 | 
			
		||||
		if relayMode == relaymode.AudioTranscription {
 | 
			
		||||
			// https://learn.microsoft.com/en-us/azure/ai-services/openai/whisper-quickstart?tabs=command-line#rest-api
 | 
			
		||||
			fullRequestURL = fmt.Sprintf("%s/openai/deployments/%s/audio/transcriptions?api-version=%s", baseURL, audioModel, apiVersion)
 | 
			
		||||
 
 | 
			
		||||
@@ -53,6 +53,16 @@ func (e GeneralErrorResponse) ToMessage() string {
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func RelayErrorHandler(resp *http.Response) (ErrorWithStatusCode *model.ErrorWithStatusCode) {
 | 
			
		||||
	if resp == nil {
 | 
			
		||||
		return &model.ErrorWithStatusCode{
 | 
			
		||||
			StatusCode: 500,
 | 
			
		||||
			Error: model.Error{
 | 
			
		||||
				Message: "resp is nil",
 | 
			
		||||
				Type:    "upstream_error",
 | 
			
		||||
				Code:    "bad_response",
 | 
			
		||||
			},
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	ErrorWithStatusCode = &model.ErrorWithStatusCode{
 | 
			
		||||
		StatusCode: resp.StatusCode,
 | 
			
		||||
		Error: model.Error{
 | 
			
		||||
 
 | 
			
		||||
@@ -18,6 +18,7 @@ import (
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/relaymode"
 | 
			
		||||
	"math"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"strings"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func getAndValidateTextRequest(c *gin.Context, relayMode int) (*relaymodel.GeneralOpenAIRequest, error) {
 | 
			
		||||
@@ -124,9 +125,9 @@ func getPromptTokens(textRequest *relaymodel.GeneralOpenAIRequest, relayMode int
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func getPreConsumedQuota(textRequest *relaymodel.GeneralOpenAIRequest, promptTokens int, ratio float64) int64 {
 | 
			
		||||
	preConsumedTokens := config.PreConsumedQuota
 | 
			
		||||
	preConsumedTokens := config.PreConsumedQuota + int64(promptTokens)
 | 
			
		||||
	if textRequest.MaxTokens != 0 {
 | 
			
		||||
		preConsumedTokens = int64(promptTokens) + int64(textRequest.MaxTokens)
 | 
			
		||||
		preConsumedTokens += int64(textRequest.MaxTokens)
 | 
			
		||||
	}
 | 
			
		||||
	return int64(float64(preConsumedTokens) * ratio)
 | 
			
		||||
}
 | 
			
		||||
@@ -204,3 +205,23 @@ func getMappedModelName(modelName string, mapping map[string]string) (string, bo
 | 
			
		||||
	}
 | 
			
		||||
	return modelName, false
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func isErrorHappened(meta *meta.Meta, resp *http.Response) bool {
 | 
			
		||||
	if resp == nil {
 | 
			
		||||
		if meta.ChannelType == channeltype.AwsClaude {
 | 
			
		||||
			return false
 | 
			
		||||
		}
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
	if resp.StatusCode != http.StatusOK {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
	if meta.ChannelType == channeltype.DeepL {
 | 
			
		||||
		// skip stream check for deepl
 | 
			
		||||
		return false
 | 
			
		||||
	}
 | 
			
		||||
	if meta.IsStream && strings.HasPrefix(resp.Header.Get("Content-Type"), "application/json") {
 | 
			
		||||
		return true
 | 
			
		||||
	}
 | 
			
		||||
	return false
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -55,6 +55,11 @@ func RelayImageHelper(c *gin.Context, relayMode int) *relaymodel.ErrorWithStatus
 | 
			
		||||
		return openai.ErrorWrapper(err, "get_image_cost_ratio_failed", http.StatusInternalServerError)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	imageModel := imageRequest.Model
 | 
			
		||||
	// Convert the original image model
 | 
			
		||||
	imageRequest.Model, _ = getMappedModelName(imageRequest.Model, billingratio.ImageOriginModelName)
 | 
			
		||||
	c.Set("response_format", imageRequest.ResponseFormat)
 | 
			
		||||
 | 
			
		||||
	var requestBody io.Reader
 | 
			
		||||
	if isModelMapped || meta.ChannelType == channeltype.Azure { // make Azure channel request body
 | 
			
		||||
		jsonStr, err := json.Marshal(imageRequest)
 | 
			
		||||
@@ -70,6 +75,7 @@ func RelayImageHelper(c *gin.Context, relayMode int) *relaymodel.ErrorWithStatus
 | 
			
		||||
	if adaptor == nil {
 | 
			
		||||
		return openai.ErrorWrapper(fmt.Errorf("invalid api type: %d", meta.APIType), "invalid_api_type", http.StatusBadRequest)
 | 
			
		||||
	}
 | 
			
		||||
	adaptor.Init(meta)
 | 
			
		||||
 | 
			
		||||
	switch meta.ChannelType {
 | 
			
		||||
	case channeltype.Ali:
 | 
			
		||||
@@ -88,7 +94,7 @@ func RelayImageHelper(c *gin.Context, relayMode int) *relaymodel.ErrorWithStatus
 | 
			
		||||
		requestBody = bytes.NewBuffer(jsonStr)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	modelRatio := billingratio.GetModelRatio(imageRequest.Model)
 | 
			
		||||
	modelRatio := billingratio.GetModelRatio(imageModel)
 | 
			
		||||
	groupRatio := billingratio.GetGroupRatio(meta.Group)
 | 
			
		||||
	ratio := modelRatio * groupRatio
 | 
			
		||||
	userQuota, err := model.CacheGetUserQuota(ctx, meta.UserId)
 | 
			
		||||
 
 | 
			
		||||
@@ -4,10 +4,6 @@ import (
 | 
			
		||||
	"bytes"
 | 
			
		||||
	"encoding/json"
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
	"strings"
 | 
			
		||||
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/logger"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay"
 | 
			
		||||
@@ -18,6 +14,8 @@ import (
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/channeltype"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/meta"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/model"
 | 
			
		||||
	"io"
 | 
			
		||||
	"net/http"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func RelayTextHelper(c *gin.Context) *model.ErrorWithStatusCode {
 | 
			
		||||
@@ -53,6 +51,7 @@ func RelayTextHelper(c *gin.Context) *model.ErrorWithStatusCode {
 | 
			
		||||
	if adaptor == nil {
 | 
			
		||||
		return openai.ErrorWrapper(fmt.Errorf("invalid api type: %d", meta.APIType), "invalid_api_type", http.StatusBadRequest)
 | 
			
		||||
	}
 | 
			
		||||
	adaptor.Init(meta)
 | 
			
		||||
 | 
			
		||||
	// get request body
 | 
			
		||||
	var requestBody io.Reader
 | 
			
		||||
@@ -87,12 +86,9 @@ func RelayTextHelper(c *gin.Context) *model.ErrorWithStatusCode {
 | 
			
		||||
		logger.Errorf(ctx, "DoRequest failed: %s", err.Error())
 | 
			
		||||
		return openai.ErrorWrapper(err, "do_request_failed", http.StatusInternalServerError)
 | 
			
		||||
	}
 | 
			
		||||
	if resp != nil {
 | 
			
		||||
		errorHappened := (resp.StatusCode != http.StatusOK) || (meta.IsStream && strings.HasPrefix(resp.Header.Get("Content-Type"), "application/json"))
 | 
			
		||||
		if errorHappened {
 | 
			
		||||
			billing.ReturnPreConsumedQuota(ctx, preConsumedQuota, meta.TokenId)
 | 
			
		||||
			return RelayErrorHandler(resp)
 | 
			
		||||
		}
 | 
			
		||||
	if isErrorHappened(meta, resp) {
 | 
			
		||||
		billing.ReturnPreConsumedQuota(ctx, preConsumedQuota, meta.TokenId)
 | 
			
		||||
		return RelayErrorHandler(resp)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	// do response
 | 
			
		||||
 
 | 
			
		||||
@@ -3,7 +3,7 @@ package meta
 | 
			
		||||
import (
 | 
			
		||||
	"github.com/gin-gonic/gin"
 | 
			
		||||
	"github.com/songquanpeng/one-api/common/ctxkey"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/adaptor/azure"
 | 
			
		||||
	"github.com/songquanpeng/one-api/model"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/channeltype"
 | 
			
		||||
	"github.com/songquanpeng/one-api/relay/relaymode"
 | 
			
		||||
	"strings"
 | 
			
		||||
@@ -19,10 +19,9 @@ type Meta struct {
 | 
			
		||||
	Group           string
 | 
			
		||||
	ModelMapping    map[string]string
 | 
			
		||||
	BaseURL         string
 | 
			
		||||
	APIVersion      string
 | 
			
		||||
	APIKey          string
 | 
			
		||||
	APIType         int
 | 
			
		||||
	Config          map[string]string
 | 
			
		||||
	Config          model.ChannelConfig
 | 
			
		||||
	IsStream        bool
 | 
			
		||||
	OriginModelName string
 | 
			
		||||
	ActualModelName string
 | 
			
		||||
@@ -32,22 +31,22 @@ type Meta struct {
 | 
			
		||||
 | 
			
		||||
func GetByContext(c *gin.Context) *Meta {
 | 
			
		||||
	meta := Meta{
 | 
			
		||||
		Mode:           relaymode.GetByPath(c.Request.URL.Path),
 | 
			
		||||
		ChannelType:    c.GetInt(ctxkey.Channel),
 | 
			
		||||
		ChannelId:      c.GetInt(ctxkey.ChannelId),
 | 
			
		||||
		TokenId:        c.GetInt(ctxkey.TokenId),
 | 
			
		||||
		TokenName:      c.GetString(ctxkey.TokenName),
 | 
			
		||||
		UserId:         c.GetInt(ctxkey.Id),
 | 
			
		||||
		Group:          c.GetString(ctxkey.Group),
 | 
			
		||||
		ModelMapping:   c.GetStringMapString(ctxkey.ModelMapping),
 | 
			
		||||
		BaseURL:        c.GetString(ctxkey.BaseURL),
 | 
			
		||||
		APIVersion:     c.GetString(ctxkey.ConfigAPIVersion),
 | 
			
		||||
		APIKey:         strings.TrimPrefix(c.Request.Header.Get("Authorization"), "Bearer "),
 | 
			
		||||
		Config:         nil,
 | 
			
		||||
		RequestURLPath: c.Request.URL.String(),
 | 
			
		||||
		Mode:            relaymode.GetByPath(c.Request.URL.Path),
 | 
			
		||||
		ChannelType:     c.GetInt(ctxkey.Channel),
 | 
			
		||||
		ChannelId:       c.GetInt(ctxkey.ChannelId),
 | 
			
		||||
		TokenId:         c.GetInt(ctxkey.TokenId),
 | 
			
		||||
		TokenName:       c.GetString(ctxkey.TokenName),
 | 
			
		||||
		UserId:          c.GetInt(ctxkey.Id),
 | 
			
		||||
		Group:           c.GetString(ctxkey.Group),
 | 
			
		||||
		ModelMapping:    c.GetStringMapString(ctxkey.ModelMapping),
 | 
			
		||||
		OriginModelName: c.GetString(ctxkey.RequestModel),
 | 
			
		||||
		BaseURL:         c.GetString(ctxkey.BaseURL),
 | 
			
		||||
		APIKey:          strings.TrimPrefix(c.Request.Header.Get("Authorization"), "Bearer "),
 | 
			
		||||
		RequestURLPath:  c.Request.URL.String(),
 | 
			
		||||
	}
 | 
			
		||||
	if meta.ChannelType == channeltype.Azure {
 | 
			
		||||
		meta.APIVersion = azure.GetAPIVersion(c)
 | 
			
		||||
	cfg, ok := c.Get(ctxkey.Config)
 | 
			
		||||
	if ok {
 | 
			
		||||
		meta.Config = cfg.(model.ChannelConfig)
 | 
			
		||||
	}
 | 
			
		||||
	if meta.BaseURL == "" {
 | 
			
		||||
		meta.BaseURL = channeltype.ChannelBaseURLs[meta.ChannelType]
 | 
			
		||||
 
 | 
			
		||||
@@ -1,5 +1,5 @@
 | 
			
		||||
<!DOCTYPE html>
 | 
			
		||||
<html lang="en">
 | 
			
		||||
<html lang="zh-CN">
 | 
			
		||||
<head>
 | 
			
		||||
  <meta charset="utf-8" />
 | 
			
		||||
  <link rel="icon" href="logo.png" />
 | 
			
		||||
 
 | 
			
		||||
@@ -15,7 +15,7 @@
 | 
			
		||||
    "@mui/utils": "^5.8.6",
 | 
			
		||||
    "@mui/x-date-pickers": "^6.18.5",
 | 
			
		||||
    "@tabler/icons-react": "^2.44.0",
 | 
			
		||||
    "apexcharts": "^3.35.3",
 | 
			
		||||
    "apexcharts": "3.35.3",
 | 
			
		||||
    "axios": "^0.27.2",
 | 
			
		||||
    "dayjs": "^1.11.10",
 | 
			
		||||
    "formik": "^2.2.9",
 | 
			
		||||
@@ -26,7 +26,7 @@
 | 
			
		||||
    "notistack": "^3.0.1",
 | 
			
		||||
    "prop-types": "^15.8.1",
 | 
			
		||||
    "react": "^18.2.0",
 | 
			
		||||
    "react-apexcharts": "^1.4.0",
 | 
			
		||||
    "react-apexcharts": "1.4.0",
 | 
			
		||||
    "react-device-detect": "^2.2.2",
 | 
			
		||||
    "react-dom": "^18.2.0",
 | 
			
		||||
    "react-perfect-scrollbar": "^1.5.8",
 | 
			
		||||
 
 | 
			
		||||
@@ -1,5 +1,5 @@
 | 
			
		||||
<!DOCTYPE html>
 | 
			
		||||
<html lang="en">
 | 
			
		||||
<html lang="zh-CN">
 | 
			
		||||
  <head>
 | 
			
		||||
    <title>One API</title>
 | 
			
		||||
    <link rel="icon" href="/favicon.ico" />
 | 
			
		||||
 
 | 
			
		||||
@@ -11,6 +11,18 @@ export const CHANNEL_OPTIONS = {
 | 
			
		||||
    value: 14,
 | 
			
		||||
    color: 'primary'
 | 
			
		||||
  },
 | 
			
		||||
  33: {
 | 
			
		||||
    key: 33,
 | 
			
		||||
    text: 'AWS Claude',
 | 
			
		||||
    value: 33,
 | 
			
		||||
    color: 'primary'
 | 
			
		||||
  },
 | 
			
		||||
  37: {
 | 
			
		||||
    key: 37,
 | 
			
		||||
    text: 'Cloudflare',
 | 
			
		||||
    value: 37,
 | 
			
		||||
    color: 'success'
 | 
			
		||||
  },
 | 
			
		||||
  3: {
 | 
			
		||||
    key: 3,
 | 
			
		||||
    text: 'Azure OpenAI',
 | 
			
		||||
@@ -113,6 +125,36 @@ export const CHANNEL_OPTIONS = {
 | 
			
		||||
    value: 32,
 | 
			
		||||
    color: 'primary'
 | 
			
		||||
  },
 | 
			
		||||
  34: {
 | 
			
		||||
    key: 34,
 | 
			
		||||
    text: 'Coze',
 | 
			
		||||
    value: 34,
 | 
			
		||||
    color: 'primary'
 | 
			
		||||
  },
 | 
			
		||||
  35: {
 | 
			
		||||
    key: 35,
 | 
			
		||||
    text: 'Cohere',
 | 
			
		||||
    value: 35,
 | 
			
		||||
    color: 'primary'
 | 
			
		||||
  },
 | 
			
		||||
  36: {
 | 
			
		||||
    key: 36,
 | 
			
		||||
    text: 'DeepSeek',
 | 
			
		||||
    value: 36,
 | 
			
		||||
    color: 'primary'
 | 
			
		||||
  },
 | 
			
		||||
  38: {
 | 
			
		||||
    key: 38,
 | 
			
		||||
    text: 'DeepL',
 | 
			
		||||
    value: 38,
 | 
			
		||||
    color: 'primary'
 | 
			
		||||
  },
 | 
			
		||||
  39: {
 | 
			
		||||
    key: 39,
 | 
			
		||||
    text: 'together.ai',
 | 
			
		||||
    value: 39,
 | 
			
		||||
    color: 'primary'
 | 
			
		||||
  },
 | 
			
		||||
  8: {
 | 
			
		||||
    key: 8,
 | 
			
		||||
    text: '自定义渠道',
 | 
			
		||||
 
 | 
			
		||||
@@ -1,24 +1,56 @@
 | 
			
		||||
import { closeSnackbar } from 'notistack';
 | 
			
		||||
import { IconX } from '@tabler/icons-react';
 | 
			
		||||
import { IconButton } from '@mui/material';
 | 
			
		||||
const action = (snackbarId) => (
 | 
			
		||||
  <>
 | 
			
		||||
    <IconButton
 | 
			
		||||
      onClick={() => {
 | 
			
		||||
        closeSnackbar(snackbarId);
 | 
			
		||||
      }}
 | 
			
		||||
    >
 | 
			
		||||
      <IconX stroke={1.5} size="1.25rem" />
 | 
			
		||||
    </IconButton>
 | 
			
		||||
  </>
 | 
			
		||||
);
 | 
			
		||||
 | 
			
		||||
export const snackbarConstants = {
 | 
			
		||||
  Common: {
 | 
			
		||||
    ERROR: {
 | 
			
		||||
      variant: 'error',
 | 
			
		||||
      autoHideDuration: 5000
 | 
			
		||||
      autoHideDuration: 5000,
 | 
			
		||||
      preventDuplicate: true,
 | 
			
		||||
      action
 | 
			
		||||
    },
 | 
			
		||||
    WARNING: {
 | 
			
		||||
      variant: 'warning',
 | 
			
		||||
      autoHideDuration: 10000
 | 
			
		||||
      autoHideDuration: 10000,
 | 
			
		||||
      preventDuplicate: true,
 | 
			
		||||
      action
 | 
			
		||||
    },
 | 
			
		||||
    SUCCESS: {
 | 
			
		||||
      variant: 'success',
 | 
			
		||||
      autoHideDuration: 1500
 | 
			
		||||
      autoHideDuration: 1500,
 | 
			
		||||
      preventDuplicate: true,
 | 
			
		||||
      action
 | 
			
		||||
    },
 | 
			
		||||
    INFO: {
 | 
			
		||||
      variant: 'info',
 | 
			
		||||
      autoHideDuration: 3000
 | 
			
		||||
      autoHideDuration: 3000,
 | 
			
		||||
      preventDuplicate: true,
 | 
			
		||||
      action
 | 
			
		||||
    },
 | 
			
		||||
    NOTICE: {
 | 
			
		||||
      variant: 'info',
 | 
			
		||||
      autoHideDuration: 7000
 | 
			
		||||
      autoHideDuration: 20000,
 | 
			
		||||
      preventDuplicate: true,
 | 
			
		||||
      action
 | 
			
		||||
    },
 | 
			
		||||
    COPY: {
 | 
			
		||||
      variant: 'copy',
 | 
			
		||||
      persist: true,
 | 
			
		||||
      preventDuplicate: true,
 | 
			
		||||
      allowDownload: true,
 | 
			
		||||
      action
 | 
			
		||||
    }
 | 
			
		||||
  },
 | 
			
		||||
  Mobile: {
 | 
			
		||||
 
 | 
			
		||||
@@ -193,3 +193,40 @@ export function removeTrailingSlash(url) {
 | 
			
		||||
    return url;
 | 
			
		||||
  }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
let channelModels = undefined;
 | 
			
		||||
export async function loadChannelModels() {
 | 
			
		||||
  const res = await API.get('/api/models');
 | 
			
		||||
  const { success, data } = res.data;
 | 
			
		||||
  if (!success) {
 | 
			
		||||
    return;
 | 
			
		||||
  }
 | 
			
		||||
  channelModels = data;
 | 
			
		||||
  localStorage.setItem('channel_models', JSON.stringify(data));
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
export function getChannelModels(type) {
 | 
			
		||||
  if (channelModels !== undefined && type in channelModels) {
 | 
			
		||||
    return channelModels[type];
 | 
			
		||||
  }
 | 
			
		||||
  let models = localStorage.getItem('channel_models');
 | 
			
		||||
  if (!models) {
 | 
			
		||||
    return [];
 | 
			
		||||
  }
 | 
			
		||||
  channelModels = JSON.parse(models);
 | 
			
		||||
  if (type in channelModels) {
 | 
			
		||||
    return channelModels[type];
 | 
			
		||||
  }
 | 
			
		||||
  return [];
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
export function copy(text, name = '') {
 | 
			
		||||
  try {
 | 
			
		||||
    navigator.clipboard.writeText(text);
 | 
			
		||||
  } catch (error) {
 | 
			
		||||
    text = `复制${name}失败,请手动复制:<br /><br />${text}`;
 | 
			
		||||
    enqueueSnackbar(<SnackbarHTMLContent htmlContent={text} />, getSnackbarOptions('COPY'));
 | 
			
		||||
    return;
 | 
			
		||||
  }
 | 
			
		||||
  showSuccess(`复制${name}成功!`);
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
@@ -1,22 +1,22 @@
 | 
			
		||||
import { useState, useEffect } from "react";
 | 
			
		||||
import { useSearchParams } from "react-router-dom";
 | 
			
		||||
import { useState, useEffect } from 'react';
 | 
			
		||||
import { useSearchParams } from 'react-router-dom';
 | 
			
		||||
 | 
			
		||||
// material-ui
 | 
			
		||||
import { Button, Stack, Typography, Alert } from "@mui/material";
 | 
			
		||||
import { Button, Stack, Typography, Alert } from '@mui/material';
 | 
			
		||||
 | 
			
		||||
// assets
 | 
			
		||||
import { showError, showInfo } from "utils/common";
 | 
			
		||||
import { API } from "utils/api";
 | 
			
		||||
import { showError, copy } from 'utils/common';
 | 
			
		||||
import { API } from 'utils/api';
 | 
			
		||||
 | 
			
		||||
// ===========================|| FIREBASE - REGISTER ||=========================== //
 | 
			
		||||
 | 
			
		||||
const ResetPasswordForm = () => {
 | 
			
		||||
  const [searchParams] = useSearchParams();
 | 
			
		||||
  const [inputs, setInputs] = useState({
 | 
			
		||||
    email: "",
 | 
			
		||||
    token: "",
 | 
			
		||||
    email: '',
 | 
			
		||||
    token: ''
 | 
			
		||||
  });
 | 
			
		||||
  const [newPassword, setNewPassword] = useState("");
 | 
			
		||||
  const [newPassword, setNewPassword] = useState('');
 | 
			
		||||
 | 
			
		||||
  const submit = async () => {
 | 
			
		||||
    const res = await API.post(`/api/user/reset`, inputs);
 | 
			
		||||
@@ -24,31 +24,25 @@ const ResetPasswordForm = () => {
 | 
			
		||||
    if (success) {
 | 
			
		||||
      let password = res.data.data;
 | 
			
		||||
      setNewPassword(password);
 | 
			
		||||
      navigator.clipboard.writeText(password);
 | 
			
		||||
      showInfo(`新密码已复制到剪贴板:${password}`);
 | 
			
		||||
      copy(password, '新密码');
 | 
			
		||||
    } else {
 | 
			
		||||
      showError(message);
 | 
			
		||||
    }
 | 
			
		||||
  };
 | 
			
		||||
 | 
			
		||||
  useEffect(() => {
 | 
			
		||||
    let email = searchParams.get("email");
 | 
			
		||||
    let token = searchParams.get("token");
 | 
			
		||||
    let email = searchParams.get('email');
 | 
			
		||||
    let token = searchParams.get('token');
 | 
			
		||||
    setInputs({
 | 
			
		||||
      token,
 | 
			
		||||
      email,
 | 
			
		||||
      email
 | 
			
		||||
    });
 | 
			
		||||
  }, []);
 | 
			
		||||
 | 
			
		||||
  return (
 | 
			
		||||
    <Stack
 | 
			
		||||
      spacing={3}
 | 
			
		||||
      padding={"24px"}
 | 
			
		||||
      justifyContent={"center"}
 | 
			
		||||
      alignItems={"center"}
 | 
			
		||||
    >
 | 
			
		||||
    <Stack spacing={3} padding={'24px'} justifyContent={'center'} alignItems={'center'}>
 | 
			
		||||
      {!inputs.email || !inputs.token ? (
 | 
			
		||||
        <Typography variant="h3" sx={{ textDecoration: "none" }}>
 | 
			
		||||
        <Typography variant="h3" sx={{ textDecoration: 'none' }}>
 | 
			
		||||
          无效的链接
 | 
			
		||||
        </Typography>
 | 
			
		||||
      ) : newPassword ? (
 | 
			
		||||
@@ -57,14 +51,7 @@ const ResetPasswordForm = () => {
 | 
			
		||||
          请登录后及时修改密码
 | 
			
		||||
        </Alert>
 | 
			
		||||
      ) : (
 | 
			
		||||
        <Button
 | 
			
		||||
          fullWidth
 | 
			
		||||
          onClick={submit}
 | 
			
		||||
          size="large"
 | 
			
		||||
          type="submit"
 | 
			
		||||
          variant="contained"
 | 
			
		||||
          color="primary"
 | 
			
		||||
        >
 | 
			
		||||
        <Button fullWidth onClick={submit} size="large" type="submit" variant="contained" color="primary">
 | 
			
		||||
          点击重置密码
 | 
			
		||||
        </Button>
 | 
			
		||||
      )}
 | 
			
		||||
 
 | 
			
		||||
@@ -1,9 +1,9 @@
 | 
			
		||||
import PropTypes from "prop-types";
 | 
			
		||||
import { useState, useEffect } from "react";
 | 
			
		||||
import { CHANNEL_OPTIONS } from "constants/ChannelConstants";
 | 
			
		||||
import { useTheme } from "@mui/material/styles";
 | 
			
		||||
import { API } from "utils/api";
 | 
			
		||||
import { showError, showSuccess } from "utils/common";
 | 
			
		||||
import PropTypes from 'prop-types';
 | 
			
		||||
import { useState, useEffect } from 'react';
 | 
			
		||||
import { CHANNEL_OPTIONS } from 'constants/ChannelConstants';
 | 
			
		||||
import { useTheme } from '@mui/material/styles';
 | 
			
		||||
import { API } from 'utils/api';
 | 
			
		||||
import { showError, showSuccess, getChannelModels } from 'utils/common';
 | 
			
		||||
import {
 | 
			
		||||
  Dialog,
 | 
			
		||||
  DialogTitle,
 | 
			
		||||
@@ -22,15 +22,15 @@ import {
 | 
			
		||||
  Autocomplete,
 | 
			
		||||
  FormHelperText,
 | 
			
		||||
  Switch,
 | 
			
		||||
  Checkbox,
 | 
			
		||||
} from "@mui/material";
 | 
			
		||||
  Checkbox
 | 
			
		||||
} from '@mui/material';
 | 
			
		||||
 | 
			
		||||
import { Formik } from "formik";
 | 
			
		||||
import * as Yup from "yup";
 | 
			
		||||
import { defaultConfig, typeConfig } from "../type/Config"; //typeConfig
 | 
			
		||||
import { createFilterOptions } from "@mui/material/Autocomplete";
 | 
			
		||||
import CheckBoxOutlineBlankIcon from "@mui/icons-material/CheckBoxOutlineBlank";
 | 
			
		||||
import CheckBoxIcon from "@mui/icons-material/CheckBox";
 | 
			
		||||
import { Formik } from 'formik';
 | 
			
		||||
import * as Yup from 'yup';
 | 
			
		||||
import { defaultConfig, typeConfig } from '../type/Config'; //typeConfig
 | 
			
		||||
import { createFilterOptions } from '@mui/material/Autocomplete';
 | 
			
		||||
import CheckBoxOutlineBlankIcon from '@mui/icons-material/CheckBoxOutlineBlank';
 | 
			
		||||
import CheckBoxIcon from '@mui/icons-material/CheckBox';
 | 
			
		||||
 | 
			
		||||
const icon = <CheckBoxOutlineBlankIcon fontSize="small" />;
 | 
			
		||||
const checkedIcon = <CheckBoxIcon fontSize="small" />;
 | 
			
		||||
@@ -38,38 +38,34 @@ const checkedIcon = <CheckBoxIcon fontSize="small" />;
 | 
			
		||||
const filter = createFilterOptions();
 | 
			
		||||
const validationSchema = Yup.object().shape({
 | 
			
		||||
  is_edit: Yup.boolean(),
 | 
			
		||||
  name: Yup.string().required("名称 不能为空"),
 | 
			
		||||
  type: Yup.number().required("渠道 不能为空"),
 | 
			
		||||
  key: Yup.string().when("is_edit", {
 | 
			
		||||
    is: false,
 | 
			
		||||
    then: Yup.string().required("密钥 不能为空"),
 | 
			
		||||
  name: Yup.string().required('名称 不能为空'),
 | 
			
		||||
  type: Yup.number().required('渠道 不能为空'),
 | 
			
		||||
  key: Yup.string().when(['is_edit', 'type'], {
 | 
			
		||||
    is: (is_edit, type) => !is_edit && type !== 33,
 | 
			
		||||
    then: Yup.string().required('密钥 不能为空')
 | 
			
		||||
  }),
 | 
			
		||||
  other: Yup.string(),
 | 
			
		||||
  models: Yup.array().min(1, "模型 不能为空"),
 | 
			
		||||
  groups: Yup.array().min(1, "用户组 不能为空"),
 | 
			
		||||
  base_url: Yup.string().when("type", {
 | 
			
		||||
  models: Yup.array().min(1, '模型 不能为空'),
 | 
			
		||||
  groups: Yup.array().min(1, '用户组 不能为空'),
 | 
			
		||||
  base_url: Yup.string().when('type', {
 | 
			
		||||
    is: (value) => [3, 8].includes(value),
 | 
			
		||||
    then: Yup.string().required("渠道API地址 不能为空"), // base_url 是必需的
 | 
			
		||||
    otherwise: Yup.string(), // 在其他情况下,base_url 可以是任意字符串
 | 
			
		||||
    then: Yup.string().required('渠道API地址 不能为空'), // base_url 是必需的
 | 
			
		||||
    otherwise: Yup.string() // 在其他情况下,base_url 可以是任意字符串
 | 
			
		||||
  }),
 | 
			
		||||
  model_mapping: Yup.string().test(
 | 
			
		||||
    "is-json",
 | 
			
		||||
    "必须是有效的JSON字符串",
 | 
			
		||||
    function (value) {
 | 
			
		||||
      try {
 | 
			
		||||
        if (value === "" || value === null || value === undefined) {
 | 
			
		||||
          return true;
 | 
			
		||||
        }
 | 
			
		||||
        const parsedValue = JSON.parse(value);
 | 
			
		||||
        if (typeof parsedValue === "object") {
 | 
			
		||||
          return true;
 | 
			
		||||
        }
 | 
			
		||||
      } catch (e) {
 | 
			
		||||
        return false;
 | 
			
		||||
  model_mapping: Yup.string().test('is-json', '必须是有效的JSON字符串', function (value) {
 | 
			
		||||
    try {
 | 
			
		||||
      if (value === '' || value === null || value === undefined) {
 | 
			
		||||
        return true;
 | 
			
		||||
      }
 | 
			
		||||
      const parsedValue = JSON.parse(value);
 | 
			
		||||
      if (typeof parsedValue === 'object') {
 | 
			
		||||
        return true;
 | 
			
		||||
      }
 | 
			
		||||
    } catch (e) {
 | 
			
		||||
      return false;
 | 
			
		||||
    }
 | 
			
		||||
  ),
 | 
			
		||||
    return false;
 | 
			
		||||
  })
 | 
			
		||||
});
 | 
			
		||||
 | 
			
		||||
const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
@@ -81,12 +77,13 @@ const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
  const [groupOptions, setGroupOptions] = useState([]);
 | 
			
		||||
  const [modelOptions, setModelOptions] = useState([]);
 | 
			
		||||
  const [batchAdd, setBatchAdd] = useState(false);
 | 
			
		||||
  const [basicModels, setBasicModels] = useState([]);
 | 
			
		||||
 | 
			
		||||
  const initChannel = (typeValue) => {
 | 
			
		||||
    if (typeConfig[typeValue]?.inputLabel) {
 | 
			
		||||
      setInputLabel({
 | 
			
		||||
        ...defaultConfig.inputLabel,
 | 
			
		||||
        ...typeConfig[typeValue].inputLabel,
 | 
			
		||||
        ...typeConfig[typeValue].inputLabel
 | 
			
		||||
      });
 | 
			
		||||
    } else {
 | 
			
		||||
      setInputLabel(defaultConfig.inputLabel);
 | 
			
		||||
@@ -95,7 +92,7 @@ const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
    if (typeConfig[typeValue]?.prompt) {
 | 
			
		||||
      setInputPrompt({
 | 
			
		||||
        ...defaultConfig.prompt,
 | 
			
		||||
        ...typeConfig[typeValue].prompt,
 | 
			
		||||
        ...typeConfig[typeValue].prompt
 | 
			
		||||
      });
 | 
			
		||||
    } else {
 | 
			
		||||
      setInputPrompt(defaultConfig.prompt);
 | 
			
		||||
@@ -104,40 +101,14 @@ const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
    return typeConfig[typeValue]?.input;
 | 
			
		||||
  };
 | 
			
		||||
  const handleTypeChange = (setFieldValue, typeValue, values) => {
 | 
			
		||||
    const newInput = initChannel(typeValue);
 | 
			
		||||
 | 
			
		||||
    if (newInput) {
 | 
			
		||||
      Object.keys(newInput).forEach((key) => {
 | 
			
		||||
        if (
 | 
			
		||||
          (!Array.isArray(values[key]) &&
 | 
			
		||||
            values[key] !== null &&
 | 
			
		||||
            values[key] !== undefined &&
 | 
			
		||||
            values[key] !== "") ||
 | 
			
		||||
          (Array.isArray(values[key]) && values[key].length > 0)
 | 
			
		||||
        ) {
 | 
			
		||||
          return;
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        if (key === "models") {
 | 
			
		||||
          setFieldValue(key, initialModel(newInput[key]));
 | 
			
		||||
          return;
 | 
			
		||||
        }
 | 
			
		||||
        setFieldValue(key, newInput[key]);
 | 
			
		||||
      });
 | 
			
		||||
    initChannel(typeValue);
 | 
			
		||||
    let localModels = getChannelModels(typeValue);
 | 
			
		||||
    setBasicModels(localModels);
 | 
			
		||||
    if (localModels.length > 0 && Array.isArray(values['models']) && values['models'].length == 0) {
 | 
			
		||||
      setFieldValue('models', initialModel(localModels));
 | 
			
		||||
    }
 | 
			
		||||
  };
 | 
			
		||||
 | 
			
		||||
  const basicModels = (channelType) => {
 | 
			
		||||
    let modelGroup =
 | 
			
		||||
      typeConfig[channelType]?.modelGroup || defaultConfig.modelGroup;
 | 
			
		||||
    // 循环 modelOptions,找到 modelGroup 对应的模型
 | 
			
		||||
    let modelList = [];
 | 
			
		||||
    modelOptions.forEach((model) => {
 | 
			
		||||
      if (model.group === modelGroup) {
 | 
			
		||||
        modelList.push(model);
 | 
			
		||||
      }
 | 
			
		||||
    });
 | 
			
		||||
    return modelList;
 | 
			
		||||
    setFieldValue('config', {});
 | 
			
		||||
  };
 | 
			
		||||
 | 
			
		||||
  const fetchGroups = async () => {
 | 
			
		||||
@@ -155,7 +126,7 @@ const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
      const { data } = res.data;
 | 
			
		||||
      data.forEach((item) => {
 | 
			
		||||
        if (!item.owned_by) {
 | 
			
		||||
          item.owned_by = "未知";
 | 
			
		||||
          item.owned_by = '未知';
 | 
			
		||||
        }
 | 
			
		||||
      });
 | 
			
		||||
      // 先对data排序
 | 
			
		||||
@@ -171,7 +142,7 @@ const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
        data.map((model) => {
 | 
			
		||||
          return {
 | 
			
		||||
            id: model.id,
 | 
			
		||||
            group: model.owned_by,
 | 
			
		||||
            group: model.owned_by
 | 
			
		||||
          };
 | 
			
		||||
        })
 | 
			
		||||
      );
 | 
			
		||||
@@ -182,33 +153,41 @@ const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
 | 
			
		||||
  const submit = async (values, { setErrors, setStatus, setSubmitting }) => {
 | 
			
		||||
    setSubmitting(true);
 | 
			
		||||
    if (values.base_url && values.base_url.endsWith("/")) {
 | 
			
		||||
    if (values.base_url && values.base_url.endsWith('/')) {
 | 
			
		||||
      values.base_url = values.base_url.slice(0, values.base_url.length - 1);
 | 
			
		||||
    }
 | 
			
		||||
    if (values.type === 3 && values.other === "") {
 | 
			
		||||
      values.other = "2023-09-01-preview";
 | 
			
		||||
    if (values.type === 3 && values.other === '') {
 | 
			
		||||
      values.other = '2023-09-01-preview';
 | 
			
		||||
    }
 | 
			
		||||
    if (values.type === 18 && values.other === "") {
 | 
			
		||||
      values.other = "v2.1";
 | 
			
		||||
    if (values.type === 18 && values.other === '') {
 | 
			
		||||
      values.other = 'v2.1';
 | 
			
		||||
    }
 | 
			
		||||
    if (values.key === '') {
 | 
			
		||||
      if (values.config.ak !== '' && values.config.sk !== '' && values.config.region !== '') {
 | 
			
		||||
        values.key = `${values.config.ak}|${values.config.sk}|${values.config.region}`;
 | 
			
		||||
      }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    let res;
 | 
			
		||||
    const modelsStr = values.models.map((model) => model.id).join(",");
 | 
			
		||||
    values.group = values.groups.join(",");
 | 
			
		||||
    const modelsStr = values.models.map((model) => model.id).join(',');
 | 
			
		||||
    const configStr = JSON.stringify(values.config);
 | 
			
		||||
    values.group = values.groups.join(',');
 | 
			
		||||
    if (channelId) {
 | 
			
		||||
      res = await API.put(`/api/channel/`, {
 | 
			
		||||
        ...values,
 | 
			
		||||
        id: parseInt(channelId),
 | 
			
		||||
        models: modelsStr,
 | 
			
		||||
        config: configStr
 | 
			
		||||
      });
 | 
			
		||||
    } else {
 | 
			
		||||
      res = await API.post(`/api/channel/`, { ...values, models: modelsStr });
 | 
			
		||||
      res = await API.post(`/api/channel/`, { ...values, models: modelsStr, config: configStr });
 | 
			
		||||
    }
 | 
			
		||||
    const { success, message } = res.data;
 | 
			
		||||
    if (success) {
 | 
			
		||||
      if (channelId) {
 | 
			
		||||
        showSuccess("渠道更新成功!");
 | 
			
		||||
        showSuccess('渠道更新成功!');
 | 
			
		||||
      } else {
 | 
			
		||||
        showSuccess("渠道创建成功!");
 | 
			
		||||
        showSuccess('渠道创建成功!');
 | 
			
		||||
      }
 | 
			
		||||
      setSubmitting(false);
 | 
			
		||||
      setStatus({ success: true });
 | 
			
		||||
@@ -226,15 +205,15 @@ const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    // 如果 channelModel 是一个字符串
 | 
			
		||||
    if (typeof channelModel === "string") {
 | 
			
		||||
      channelModel = channelModel.split(",");
 | 
			
		||||
    if (typeof channelModel === 'string') {
 | 
			
		||||
      channelModel = channelModel.split(',');
 | 
			
		||||
    }
 | 
			
		||||
    let modelList = channelModel.map((model) => {
 | 
			
		||||
      const modelOption = modelOptions.find((option) => option.id === model);
 | 
			
		||||
      if (modelOption) {
 | 
			
		||||
        return modelOption;
 | 
			
		||||
      }
 | 
			
		||||
      return { id: model, group: "自定义:点击或回车输入" };
 | 
			
		||||
      return { id: model, group: '自定义:点击或回车输入' };
 | 
			
		||||
    });
 | 
			
		||||
    return modelList;
 | 
			
		||||
  }
 | 
			
		||||
@@ -243,24 +222,24 @@ const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
    let res = await API.get(`/api/channel/${channelId}`);
 | 
			
		||||
    const { success, message, data } = res.data;
 | 
			
		||||
    if (success) {
 | 
			
		||||
      if (data.models === "") {
 | 
			
		||||
      if (data.models === '') {
 | 
			
		||||
        data.models = [];
 | 
			
		||||
      } else {
 | 
			
		||||
        data.models = initialModel(data.models);
 | 
			
		||||
      }
 | 
			
		||||
      if (data.group === "") {
 | 
			
		||||
      if (data.group === '') {
 | 
			
		||||
        data.groups = [];
 | 
			
		||||
      } else {
 | 
			
		||||
        data.groups = data.group.split(",");
 | 
			
		||||
        data.groups = data.group.split(',');
 | 
			
		||||
      }
 | 
			
		||||
      if (data.model_mapping !== "") {
 | 
			
		||||
        data.model_mapping = JSON.stringify(
 | 
			
		||||
          JSON.parse(data.model_mapping),
 | 
			
		||||
          null,
 | 
			
		||||
          2
 | 
			
		||||
        );
 | 
			
		||||
      if (data.model_mapping !== '') {
 | 
			
		||||
        data.model_mapping = JSON.stringify(JSON.parse(data.model_mapping), null, 2);
 | 
			
		||||
      }
 | 
			
		||||
      data.base_url = data.base_url ?? "";
 | 
			
		||||
      if (data.config !== '') {
 | 
			
		||||
        data.config = JSON.parse(data.config);
 | 
			
		||||
      }
 | 
			
		||||
 | 
			
		||||
      data.base_url = data.base_url ?? '';
 | 
			
		||||
      data.is_edit = true;
 | 
			
		||||
      initChannel(data.type);
 | 
			
		||||
      setInitialInput(data);
 | 
			
		||||
@@ -286,45 +265,25 @@ const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
  }, [channelId]);
 | 
			
		||||
 | 
			
		||||
  return (
 | 
			
		||||
    <Dialog open={open} onClose={onCancel} fullWidth maxWidth={"md"}>
 | 
			
		||||
    <Dialog open={open} onClose={onCancel} fullWidth maxWidth={'md'}>
 | 
			
		||||
      <DialogTitle
 | 
			
		||||
        sx={{
 | 
			
		||||
          margin: "0px",
 | 
			
		||||
          margin: '0px',
 | 
			
		||||
          fontWeight: 700,
 | 
			
		||||
          lineHeight: "1.55556",
 | 
			
		||||
          padding: "24px",
 | 
			
		||||
          fontSize: "1.125rem",
 | 
			
		||||
          lineHeight: '1.55556',
 | 
			
		||||
          padding: '24px',
 | 
			
		||||
          fontSize: '1.125rem'
 | 
			
		||||
        }}
 | 
			
		||||
      >
 | 
			
		||||
        {channelId ? "编辑渠道" : "新建渠道"}
 | 
			
		||||
        {channelId ? '编辑渠道' : '新建渠道'}
 | 
			
		||||
      </DialogTitle>
 | 
			
		||||
      <Divider />
 | 
			
		||||
      <DialogContent>
 | 
			
		||||
        <Formik
 | 
			
		||||
          initialValues={initialInput}
 | 
			
		||||
          enableReinitialize
 | 
			
		||||
          validationSchema={validationSchema}
 | 
			
		||||
          onSubmit={submit}
 | 
			
		||||
        >
 | 
			
		||||
          {({
 | 
			
		||||
            errors,
 | 
			
		||||
            handleBlur,
 | 
			
		||||
            handleChange,
 | 
			
		||||
            handleSubmit,
 | 
			
		||||
            isSubmitting,
 | 
			
		||||
            touched,
 | 
			
		||||
            values,
 | 
			
		||||
            setFieldValue,
 | 
			
		||||
          }) => (
 | 
			
		||||
        <Formik initialValues={initialInput} enableReinitialize validationSchema={validationSchema} onSubmit={submit}>
 | 
			
		||||
          {({ errors, handleBlur, handleChange, handleSubmit, isSubmitting, touched, values, setFieldValue }) => (
 | 
			
		||||
            <form noValidate onSubmit={handleSubmit}>
 | 
			
		||||
              <FormControl
 | 
			
		||||
                fullWidth
 | 
			
		||||
                error={Boolean(touched.type && errors.type)}
 | 
			
		||||
                sx={{ ...theme.typography.otherInput }}
 | 
			
		||||
              >
 | 
			
		||||
                <InputLabel htmlFor="channel-type-label">
 | 
			
		||||
                  {inputLabel.type}
 | 
			
		||||
                </InputLabel>
 | 
			
		||||
              <FormControl fullWidth error={Boolean(touched.type && errors.type)} sx={{ ...theme.typography.otherInput }}>
 | 
			
		||||
                <InputLabel htmlFor="channel-type-label">{inputLabel.type}</InputLabel>
 | 
			
		||||
                <Select
 | 
			
		||||
                  id="channel-type-label"
 | 
			
		||||
                  label={inputLabel.type}
 | 
			
		||||
@@ -338,9 +297,9 @@ const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
                  MenuProps={{
 | 
			
		||||
                    PaperProps: {
 | 
			
		||||
                      style: {
 | 
			
		||||
                        maxHeight: 200,
 | 
			
		||||
                      },
 | 
			
		||||
                    },
 | 
			
		||||
                        maxHeight: 200
 | 
			
		||||
                      }
 | 
			
		||||
                    }
 | 
			
		||||
                  }}
 | 
			
		||||
                >
 | 
			
		||||
                  {Object.values(CHANNEL_OPTIONS)
 | 
			
		||||
@@ -360,21 +319,12 @@ const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
                    {errors.type}
 | 
			
		||||
                  </FormHelperText>
 | 
			
		||||
                ) : (
 | 
			
		||||
                  <FormHelperText id="helper-tex-channel-type-label">
 | 
			
		||||
                    {" "}
 | 
			
		||||
                    {inputPrompt.type}{" "}
 | 
			
		||||
                  </FormHelperText>
 | 
			
		||||
                  <FormHelperText id="helper-tex-channel-type-label"> {inputPrompt.type} </FormHelperText>
 | 
			
		||||
                )}
 | 
			
		||||
              </FormControl>
 | 
			
		||||
 | 
			
		||||
              <FormControl
 | 
			
		||||
                fullWidth
 | 
			
		||||
                error={Boolean(touched.name && errors.name)}
 | 
			
		||||
                sx={{ ...theme.typography.otherInput }}
 | 
			
		||||
              >
 | 
			
		||||
                <InputLabel htmlFor="channel-name-label">
 | 
			
		||||
                  {inputLabel.name}
 | 
			
		||||
                </InputLabel>
 | 
			
		||||
              <FormControl fullWidth error={Boolean(touched.name && errors.name)} sx={{ ...theme.typography.otherInput }}>
 | 
			
		||||
                <InputLabel htmlFor="channel-name-label">{inputLabel.name}</InputLabel>
 | 
			
		||||
                <OutlinedInput
 | 
			
		||||
                  id="channel-name-label"
 | 
			
		||||
                  label={inputLabel.name}
 | 
			
		||||
@@ -383,7 +333,7 @@ const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
                  name="name"
 | 
			
		||||
                  onBlur={handleBlur}
 | 
			
		||||
                  onChange={handleChange}
 | 
			
		||||
                  inputProps={{ autoComplete: "name" }}
 | 
			
		||||
                  inputProps={{ autoComplete: 'name' }}
 | 
			
		||||
                  aria-describedby="helper-text-channel-name-label"
 | 
			
		||||
                />
 | 
			
		||||
                {touched.name && errors.name ? (
 | 
			
		||||
@@ -391,21 +341,12 @@ const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
                    {errors.name}
 | 
			
		||||
                  </FormHelperText>
 | 
			
		||||
                ) : (
 | 
			
		||||
                  <FormHelperText id="helper-tex-channel-name-label">
 | 
			
		||||
                    {" "}
 | 
			
		||||
                    {inputPrompt.name}{" "}
 | 
			
		||||
                  </FormHelperText>
 | 
			
		||||
                  <FormHelperText id="helper-tex-channel-name-label"> {inputPrompt.name} </FormHelperText>
 | 
			
		||||
                )}
 | 
			
		||||
              </FormControl>
 | 
			
		||||
 | 
			
		||||
              <FormControl
 | 
			
		||||
                fullWidth
 | 
			
		||||
                error={Boolean(touched.base_url && errors.base_url)}
 | 
			
		||||
                sx={{ ...theme.typography.otherInput }}
 | 
			
		||||
              >
 | 
			
		||||
                <InputLabel htmlFor="channel-base_url-label">
 | 
			
		||||
                  {inputLabel.base_url}
 | 
			
		||||
                </InputLabel>
 | 
			
		||||
              <FormControl fullWidth error={Boolean(touched.base_url && errors.base_url)} sx={{ ...theme.typography.otherInput }}>
 | 
			
		||||
                <InputLabel htmlFor="channel-base_url-label">{inputLabel.base_url}</InputLabel>
 | 
			
		||||
                <OutlinedInput
 | 
			
		||||
                  id="channel-base_url-label"
 | 
			
		||||
                  label={inputLabel.base_url}
 | 
			
		||||
@@ -422,22 +363,13 @@ const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
                    {errors.base_url}
 | 
			
		||||
                  </FormHelperText>
 | 
			
		||||
                ) : (
 | 
			
		||||
                  <FormHelperText id="helper-tex-channel-base_url-label">
 | 
			
		||||
                    {" "}
 | 
			
		||||
                    {inputPrompt.base_url}{" "}
 | 
			
		||||
                  </FormHelperText>
 | 
			
		||||
                  <FormHelperText id="helper-tex-channel-base_url-label"> {inputPrompt.base_url} </FormHelperText>
 | 
			
		||||
                )}
 | 
			
		||||
              </FormControl>
 | 
			
		||||
 | 
			
		||||
              {inputPrompt.other && (
 | 
			
		||||
                <FormControl
 | 
			
		||||
                  fullWidth
 | 
			
		||||
                  error={Boolean(touched.other && errors.other)}
 | 
			
		||||
                  sx={{ ...theme.typography.otherInput }}
 | 
			
		||||
                >
 | 
			
		||||
                  <InputLabel htmlFor="channel-other-label">
 | 
			
		||||
                    {inputLabel.other}
 | 
			
		||||
                  </InputLabel>
 | 
			
		||||
                <FormControl fullWidth error={Boolean(touched.other && errors.other)} sx={{ ...theme.typography.otherInput }}>
 | 
			
		||||
                  <InputLabel htmlFor="channel-other-label">{inputLabel.other}</InputLabel>
 | 
			
		||||
                  <OutlinedInput
 | 
			
		||||
                    id="channel-other-label"
 | 
			
		||||
                    label={inputLabel.other}
 | 
			
		||||
@@ -454,10 +386,7 @@ const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
                      {errors.other}
 | 
			
		||||
                    </FormHelperText>
 | 
			
		||||
                  ) : (
 | 
			
		||||
                    <FormHelperText id="helper-tex-channel-other-label">
 | 
			
		||||
                      {" "}
 | 
			
		||||
                      {inputPrompt.other}{" "}
 | 
			
		||||
                    </FormHelperText>
 | 
			
		||||
                    <FormHelperText id="helper-tex-channel-other-label"> {inputPrompt.other} </FormHelperText>
 | 
			
		||||
                  )}
 | 
			
		||||
                </FormControl>
 | 
			
		||||
              )}
 | 
			
		||||
@@ -471,22 +400,15 @@ const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
                  onChange={(e, value) => {
 | 
			
		||||
                    const event = {
 | 
			
		||||
                      target: {
 | 
			
		||||
                        name: "groups",
 | 
			
		||||
                        value: value,
 | 
			
		||||
                      },
 | 
			
		||||
                        name: 'groups',
 | 
			
		||||
                        value: value
 | 
			
		||||
                      }
 | 
			
		||||
                    };
 | 
			
		||||
                    handleChange(event);
 | 
			
		||||
                  }}
 | 
			
		||||
                  onBlur={handleBlur}
 | 
			
		||||
                  filterSelectedOptions
 | 
			
		||||
                  renderInput={(params) => (
 | 
			
		||||
                    <TextField
 | 
			
		||||
                      {...params}
 | 
			
		||||
                      name="groups"
 | 
			
		||||
                      error={Boolean(errors.groups)}
 | 
			
		||||
                      label={inputLabel.groups}
 | 
			
		||||
                    />
 | 
			
		||||
                  )}
 | 
			
		||||
                  renderInput={(params) => <TextField {...params} name="groups" error={Boolean(errors.groups)} label={inputLabel.groups} />}
 | 
			
		||||
                  aria-describedby="helper-text-channel-groups-label"
 | 
			
		||||
                />
 | 
			
		||||
                {errors.groups ? (
 | 
			
		||||
@@ -494,10 +416,7 @@ const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
                    {errors.groups}
 | 
			
		||||
                  </FormHelperText>
 | 
			
		||||
                ) : (
 | 
			
		||||
                  <FormHelperText id="helper-tex-channel-groups-label">
 | 
			
		||||
                    {" "}
 | 
			
		||||
                    {inputPrompt.groups}{" "}
 | 
			
		||||
                  </FormHelperText>
 | 
			
		||||
                  <FormHelperText id="helper-tex-channel-groups-label"> {inputPrompt.groups} </FormHelperText>
 | 
			
		||||
                )}
 | 
			
		||||
              </FormControl>
 | 
			
		||||
 | 
			
		||||
@@ -511,30 +430,19 @@ const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
                  onChange={(e, value) => {
 | 
			
		||||
                    const event = {
 | 
			
		||||
                      target: {
 | 
			
		||||
                        name: "models",
 | 
			
		||||
                        value: value.map((item) =>
 | 
			
		||||
                          typeof item === "string"
 | 
			
		||||
                            ? { id: item, group: "自定义:点击或回车输入" }
 | 
			
		||||
                            : item
 | 
			
		||||
                        ),
 | 
			
		||||
                      },
 | 
			
		||||
                        name: 'models',
 | 
			
		||||
                        value: value.map((item) => (typeof item === 'string' ? { id: item, group: '自定义:点击或回车输入' } : item))
 | 
			
		||||
                      }
 | 
			
		||||
                    };
 | 
			
		||||
                    handleChange(event);
 | 
			
		||||
                  }}
 | 
			
		||||
                  onBlur={handleBlur}
 | 
			
		||||
                  // filterSelectedOptions
 | 
			
		||||
                  disableCloseOnSelect
 | 
			
		||||
                  renderInput={(params) => (
 | 
			
		||||
                    <TextField
 | 
			
		||||
                      {...params}
 | 
			
		||||
                      name="models"
 | 
			
		||||
                      error={Boolean(errors.models)}
 | 
			
		||||
                      label={inputLabel.models}
 | 
			
		||||
                    />
 | 
			
		||||
                  )}
 | 
			
		||||
                  renderInput={(params) => <TextField {...params} name="models" error={Boolean(errors.models)} label={inputLabel.models} />}
 | 
			
		||||
                  groupBy={(option) => option.group}
 | 
			
		||||
                  getOptionLabel={(option) => {
 | 
			
		||||
                    if (typeof option === "string") {
 | 
			
		||||
                    if (typeof option === 'string') {
 | 
			
		||||
                      return option;
 | 
			
		||||
                    }
 | 
			
		||||
                    if (option.inputValue) {
 | 
			
		||||
@@ -545,25 +453,18 @@ const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
                  filterOptions={(options, params) => {
 | 
			
		||||
                    const filtered = filter(options, params);
 | 
			
		||||
                    const { inputValue } = params;
 | 
			
		||||
                    const isExisting = options.some(
 | 
			
		||||
                      (option) => inputValue === option.id
 | 
			
		||||
                    );
 | 
			
		||||
                    if (inputValue !== "" && !isExisting) {
 | 
			
		||||
                    const isExisting = options.some((option) => inputValue === option.id);
 | 
			
		||||
                    if (inputValue !== '' && !isExisting) {
 | 
			
		||||
                      filtered.push({
 | 
			
		||||
                        id: inputValue,
 | 
			
		||||
                        group: "自定义:点击或回车输入",
 | 
			
		||||
                        group: '自定义:点击或回车输入'
 | 
			
		||||
                      });
 | 
			
		||||
                    }
 | 
			
		||||
                    return filtered;
 | 
			
		||||
                  }}
 | 
			
		||||
                  renderOption={(props, option, { selected }) => (
 | 
			
		||||
                    <li {...props}>
 | 
			
		||||
                      <Checkbox
 | 
			
		||||
                        icon={icon}
 | 
			
		||||
                        checkedIcon={checkedIcon}
 | 
			
		||||
                        style={{ marginRight: 8 }}
 | 
			
		||||
                        checked={selected}
 | 
			
		||||
                      />
 | 
			
		||||
                      <Checkbox icon={icon} checkedIcon={checkedIcon} style={{ marginRight: 8 }} checked={selected} />
 | 
			
		||||
                      {option.id}
 | 
			
		||||
                    </li>
 | 
			
		||||
                  )}
 | 
			
		||||
@@ -573,103 +474,104 @@ const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
                    {errors.models}
 | 
			
		||||
                  </FormHelperText>
 | 
			
		||||
                ) : (
 | 
			
		||||
                  <FormHelperText id="helper-tex-channel-models-label">
 | 
			
		||||
                    {" "}
 | 
			
		||||
                    {inputPrompt.models}{" "}
 | 
			
		||||
                  </FormHelperText>
 | 
			
		||||
                  <FormHelperText id="helper-tex-channel-models-label"> {inputPrompt.models} </FormHelperText>
 | 
			
		||||
                )}
 | 
			
		||||
              </FormControl>
 | 
			
		||||
              <Container
 | 
			
		||||
                sx={{
 | 
			
		||||
                  textAlign: "right",
 | 
			
		||||
                  textAlign: 'right'
 | 
			
		||||
                }}
 | 
			
		||||
              >
 | 
			
		||||
                <ButtonGroup
 | 
			
		||||
                  variant="outlined"
 | 
			
		||||
                  aria-label="small outlined primary button group"
 | 
			
		||||
                >
 | 
			
		||||
                <ButtonGroup variant="outlined" aria-label="small outlined primary button group">
 | 
			
		||||
                  <Button
 | 
			
		||||
                    onClick={() => {
 | 
			
		||||
                      setFieldValue("models", basicModels(values.type));
 | 
			
		||||
                      setFieldValue('models', initialModel(basicModels));
 | 
			
		||||
                    }}
 | 
			
		||||
                  >
 | 
			
		||||
                    填入渠道支持模型
 | 
			
		||||
                    填入相关模型
 | 
			
		||||
                  </Button>
 | 
			
		||||
                  <Button
 | 
			
		||||
                    onClick={() => {
 | 
			
		||||
                      setFieldValue("models", modelOptions);
 | 
			
		||||
                      setFieldValue('models', modelOptions);
 | 
			
		||||
                    }}
 | 
			
		||||
                  >
 | 
			
		||||
                    填入所有模型
 | 
			
		||||
                  </Button>
 | 
			
		||||
                </ButtonGroup>
 | 
			
		||||
              </Container>
 | 
			
		||||
              <FormControl
 | 
			
		||||
                fullWidth
 | 
			
		||||
                error={Boolean(touched.key && errors.key)}
 | 
			
		||||
                sx={{ ...theme.typography.otherInput }}
 | 
			
		||||
              >
 | 
			
		||||
                {!batchAdd ? (
 | 
			
		||||
                  <>
 | 
			
		||||
                    <InputLabel htmlFor="channel-key-label">
 | 
			
		||||
                      {inputLabel.key}
 | 
			
		||||
                    </InputLabel>
 | 
			
		||||
                    <OutlinedInput
 | 
			
		||||
                      id="channel-key-label"
 | 
			
		||||
                      label={inputLabel.key}
 | 
			
		||||
                      type="text"
 | 
			
		||||
                      value={values.key}
 | 
			
		||||
                      name="key"
 | 
			
		||||
                      onBlur={handleBlur}
 | 
			
		||||
                      onChange={handleChange}
 | 
			
		||||
                      inputProps={{}}
 | 
			
		||||
                      aria-describedby="helper-text-channel-key-label"
 | 
			
		||||
                    />
 | 
			
		||||
                  </>
 | 
			
		||||
                ) : (
 | 
			
		||||
                  <TextField
 | 
			
		||||
                    multiline
 | 
			
		||||
                    id="channel-key-label"
 | 
			
		||||
                    label={inputLabel.key}
 | 
			
		||||
                    value={values.key}
 | 
			
		||||
                    name="key"
 | 
			
		||||
                    onBlur={handleBlur}
 | 
			
		||||
                    onChange={handleChange}
 | 
			
		||||
                    aria-describedby="helper-text-channel-key-label"
 | 
			
		||||
                    minRows={5}
 | 
			
		||||
                    placeholder={inputPrompt.key + ",一行一个密钥"}
 | 
			
		||||
                  />
 | 
			
		||||
                )}
 | 
			
		||||
              {inputLabel.key && (
 | 
			
		||||
                <>
 | 
			
		||||
                  <FormControl fullWidth error={Boolean(touched.key && errors.key)} sx={{ ...theme.typography.otherInput }}>
 | 
			
		||||
                    {!batchAdd ? (
 | 
			
		||||
                      <>
 | 
			
		||||
                        <InputLabel htmlFor="channel-key-label">{inputLabel.key}</InputLabel>
 | 
			
		||||
                        <OutlinedInput
 | 
			
		||||
                          id="channel-key-label"
 | 
			
		||||
                          label={inputLabel.key}
 | 
			
		||||
                          type="text"
 | 
			
		||||
                          value={values.key}
 | 
			
		||||
                          name="key"
 | 
			
		||||
                          onBlur={handleBlur}
 | 
			
		||||
                          onChange={handleChange}
 | 
			
		||||
                          inputProps={{}}
 | 
			
		||||
                          aria-describedby="helper-text-channel-key-label"
 | 
			
		||||
                        />
 | 
			
		||||
                      </>
 | 
			
		||||
                    ) : (
 | 
			
		||||
                      <TextField
 | 
			
		||||
                        multiline
 | 
			
		||||
                        id="channel-key-label"
 | 
			
		||||
                        label={inputLabel.key}
 | 
			
		||||
                        value={values.key}
 | 
			
		||||
                        name="key"
 | 
			
		||||
                        onBlur={handleBlur}
 | 
			
		||||
                        onChange={handleChange}
 | 
			
		||||
                        aria-describedby="helper-text-channel-key-label"
 | 
			
		||||
                        minRows={5}
 | 
			
		||||
                        placeholder={inputPrompt.key + ',一行一个密钥'}
 | 
			
		||||
                      />
 | 
			
		||||
                    )}
 | 
			
		||||
 | 
			
		||||
                {touched.key && errors.key ? (
 | 
			
		||||
                  <FormHelperText error id="helper-tex-channel-key-label">
 | 
			
		||||
                    {errors.key}
 | 
			
		||||
                  </FormHelperText>
 | 
			
		||||
                ) : (
 | 
			
		||||
                  <FormHelperText id="helper-tex-channel-key-label">
 | 
			
		||||
                    {" "}
 | 
			
		||||
                    {inputPrompt.key}{" "}
 | 
			
		||||
                  </FormHelperText>
 | 
			
		||||
                )}
 | 
			
		||||
              </FormControl>
 | 
			
		||||
              {channelId === 0 && (
 | 
			
		||||
                <Container
 | 
			
		||||
                  sx={{
 | 
			
		||||
                    textAlign: "right",
 | 
			
		||||
                  }}
 | 
			
		||||
                >
 | 
			
		||||
                  <Switch
 | 
			
		||||
                    checked={batchAdd}
 | 
			
		||||
                    onChange={(e) => setBatchAdd(e.target.checked)}
 | 
			
		||||
                  />
 | 
			
		||||
                  批量添加
 | 
			
		||||
                </Container>
 | 
			
		||||
                    {touched.key && errors.key ? (
 | 
			
		||||
                      <FormHelperText error id="helper-tex-channel-key-label">
 | 
			
		||||
                        {errors.key}
 | 
			
		||||
                      </FormHelperText>
 | 
			
		||||
                    ) : (
 | 
			
		||||
                      <FormHelperText id="helper-tex-channel-key-label"> {inputPrompt.key} </FormHelperText>
 | 
			
		||||
                    )}
 | 
			
		||||
                  </FormControl>
 | 
			
		||||
                  {channelId === 0 && (
 | 
			
		||||
                    <Container
 | 
			
		||||
                      sx={{
 | 
			
		||||
                        textAlign: 'right'
 | 
			
		||||
                      }}
 | 
			
		||||
                    >
 | 
			
		||||
                      <Switch checked={batchAdd} onChange={(e) => setBatchAdd(e.target.checked)} />
 | 
			
		||||
                      批量添加
 | 
			
		||||
                    </Container>
 | 
			
		||||
                  )}
 | 
			
		||||
                </>
 | 
			
		||||
              )}
 | 
			
		||||
              <FormControl
 | 
			
		||||
                fullWidth
 | 
			
		||||
                error={Boolean(touched.model_mapping && errors.model_mapping)}
 | 
			
		||||
                sx={{ ...theme.typography.otherInput }}
 | 
			
		||||
              >
 | 
			
		||||
 | 
			
		||||
              {inputLabel.config &&
 | 
			
		||||
                Object.keys(inputLabel.config).map((configName) => {
 | 
			
		||||
                  return (
 | 
			
		||||
                    <FormControl key={'config.' + configName} fullWidth sx={{ ...theme.typography.otherInput }}>
 | 
			
		||||
                      <TextField
 | 
			
		||||
                        multiline
 | 
			
		||||
                        key={'config.' + configName}
 | 
			
		||||
                        name={'config.' + configName}
 | 
			
		||||
                        value={values.config?.[configName] || ''}
 | 
			
		||||
                        label={configName}
 | 
			
		||||
                        placeholder={inputPrompt.config[configName]}
 | 
			
		||||
                        onChange={handleChange}
 | 
			
		||||
                      />
 | 
			
		||||
                      <FormHelperText id={`helper-tex-config.${configName}-label`}> {inputPrompt.config[configName]} </FormHelperText>
 | 
			
		||||
                    </FormControl>
 | 
			
		||||
                  );
 | 
			
		||||
                })}
 | 
			
		||||
 | 
			
		||||
              <FormControl fullWidth error={Boolean(touched.model_mapping && errors.model_mapping)} sx={{ ...theme.typography.otherInput }}>
 | 
			
		||||
                {/* <InputLabel htmlFor="channel-model_mapping-label">{inputLabel.model_mapping}</InputLabel> */}
 | 
			
		||||
                <TextField
 | 
			
		||||
                  multiline
 | 
			
		||||
@@ -684,28 +586,16 @@ const EditModal = ({ open, channelId, onCancel, onOk }) => {
 | 
			
		||||
                  placeholder={inputPrompt.model_mapping}
 | 
			
		||||
                />
 | 
			
		||||
                {touched.model_mapping && errors.model_mapping ? (
 | 
			
		||||
                  <FormHelperText
 | 
			
		||||
                    error
 | 
			
		||||
                    id="helper-tex-channel-model_mapping-label"
 | 
			
		||||
                  >
 | 
			
		||||
                  <FormHelperText error id="helper-tex-channel-model_mapping-label">
 | 
			
		||||
                    {errors.model_mapping}
 | 
			
		||||
                  </FormHelperText>
 | 
			
		||||
                ) : (
 | 
			
		||||
                  <FormHelperText id="helper-tex-channel-model_mapping-label">
 | 
			
		||||
                    {" "}
 | 
			
		||||
                    {inputPrompt.model_mapping}{" "}
 | 
			
		||||
                  </FormHelperText>
 | 
			
		||||
                  <FormHelperText id="helper-tex-channel-model_mapping-label"> {inputPrompt.model_mapping} </FormHelperText>
 | 
			
		||||
                )}
 | 
			
		||||
              </FormControl>
 | 
			
		||||
              <DialogActions>
 | 
			
		||||
                <Button onClick={onCancel}>取消</Button>
 | 
			
		||||
                <Button
 | 
			
		||||
                  disableElevation
 | 
			
		||||
                  disabled={isSubmitting}
 | 
			
		||||
                  type="submit"
 | 
			
		||||
                  variant="contained"
 | 
			
		||||
                  color="primary"
 | 
			
		||||
                >
 | 
			
		||||
                <Button disableElevation disabled={isSubmitting} type="submit" variant="contained" color="primary">
 | 
			
		||||
                  提交
 | 
			
		||||
                </Button>
 | 
			
		||||
              </DialogActions>
 | 
			
		||||
@@ -723,5 +613,5 @@ EditModal.propTypes = {
 | 
			
		||||
  open: PropTypes.bool,
 | 
			
		||||
  channelId: PropTypes.number,
 | 
			
		||||
  onCancel: PropTypes.func,
 | 
			
		||||
  onOk: PropTypes.func,
 | 
			
		||||
  onOk: PropTypes.func
 | 
			
		||||
};
 | 
			
		||||
 
 | 
			
		||||
@@ -1,20 +1,20 @@
 | 
			
		||||
import PropTypes from "prop-types";
 | 
			
		||||
import { Tooltip, Stack, Container } from "@mui/material";
 | 
			
		||||
import Label from "ui-component/Label";
 | 
			
		||||
import { styled } from "@mui/material/styles";
 | 
			
		||||
import { showSuccess } from "utils/common";
 | 
			
		||||
import PropTypes from 'prop-types';
 | 
			
		||||
import { Tooltip, Stack, Container } from '@mui/material';
 | 
			
		||||
import Label from 'ui-component/Label';
 | 
			
		||||
import { styled } from '@mui/material/styles';
 | 
			
		||||
import { showSuccess, copy } from 'utils/common';
 | 
			
		||||
 | 
			
		||||
const TooltipContainer = styled(Container)({
 | 
			
		||||
  maxHeight: "250px",
 | 
			
		||||
  overflow: "auto",
 | 
			
		||||
  "&::-webkit-scrollbar": {
 | 
			
		||||
    width: "0px", // Set the width to 0 to hide the scrollbar
 | 
			
		||||
  },
 | 
			
		||||
  maxHeight: '250px',
 | 
			
		||||
  overflow: 'auto',
 | 
			
		||||
  '&::-webkit-scrollbar': {
 | 
			
		||||
    width: '0px' // Set the width to 0 to hide the scrollbar
 | 
			
		||||
  }
 | 
			
		||||
});
 | 
			
		||||
 | 
			
		||||
const NameLabel = ({ name, models }) => {
 | 
			
		||||
  let modelMap = [];
 | 
			
		||||
  modelMap = models.split(",");
 | 
			
		||||
  modelMap = models.split(',');
 | 
			
		||||
  modelMap.sort();
 | 
			
		||||
 | 
			
		||||
  return (
 | 
			
		||||
@@ -28,8 +28,7 @@ const NameLabel = ({ name, models }) => {
 | 
			
		||||
                  variant="ghost"
 | 
			
		||||
                  key={index}
 | 
			
		||||
                  onClick={() => {
 | 
			
		||||
                    navigator.clipboard.writeText(item);
 | 
			
		||||
                    showSuccess("复制模型名称成功!");
 | 
			
		||||
                    copy(item, '模型名称');
 | 
			
		||||
                  }}
 | 
			
		||||
                >
 | 
			
		||||
                  {item}
 | 
			
		||||
@@ -48,7 +47,7 @@ const NameLabel = ({ name, models }) => {
 | 
			
		||||
 | 
			
		||||
NameLabel.propTypes = {
 | 
			
		||||
  name: PropTypes.string,
 | 
			
		||||
  models: PropTypes.string,
 | 
			
		||||
  models: PropTypes.string
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
export default NameLabel;
 | 
			
		||||
 
 | 
			
		||||
@@ -1,5 +1,5 @@
 | 
			
		||||
import { useState, useEffect } from 'react';
 | 
			
		||||
import { showError, showSuccess, showInfo } from 'utils/common';
 | 
			
		||||
import { showError, showSuccess, showInfo, loadChannelModels } from 'utils/common';
 | 
			
		||||
 | 
			
		||||
import { useTheme } from '@mui/material/styles';
 | 
			
		||||
import Table from '@mui/material/Table';
 | 
			
		||||
@@ -8,7 +8,6 @@ import TableContainer from '@mui/material/TableContainer';
 | 
			
		||||
import PerfectScrollbar from 'react-perfect-scrollbar';
 | 
			
		||||
import TablePagination from '@mui/material/TablePagination';
 | 
			
		||||
import LinearProgress from '@mui/material/LinearProgress';
 | 
			
		||||
import Alert from '@mui/material/Alert';
 | 
			
		||||
import ButtonGroup from '@mui/material/ButtonGroup';
 | 
			
		||||
import Toolbar from '@mui/material/Toolbar';
 | 
			
		||||
import useMediaQuery from '@mui/material/useMediaQuery';
 | 
			
		||||
@@ -189,6 +188,7 @@ export default function ChannelPage() {
 | 
			
		||||
      .catch((reason) => {
 | 
			
		||||
        showError(reason);
 | 
			
		||||
      });
 | 
			
		||||
    loadChannelModels().then();
 | 
			
		||||
  }, []);
 | 
			
		||||
 | 
			
		||||
  return (
 | 
			
		||||
@@ -200,7 +200,7 @@ export default function ChannelPage() {
 | 
			
		||||
        </Button>
 | 
			
		||||
      </Stack>
 | 
			
		||||
      <Card>
 | 
			
		||||
        <Box component="form" onSubmit={searchChannels} noValidate sx={{marginTop: 2}}>
 | 
			
		||||
        <Box component="form" onSubmit={searchChannels} noValidate sx={{ marginTop: 2 }}>
 | 
			
		||||
          <TableToolBar filterName={searchKeyword} handleFilterName={handleSearchKeyword} placeholder={'搜索渠道的 ID,名称和密钥 ...'} />
 | 
			
		||||
        </Box>
 | 
			
		||||
        <Toolbar
 | 
			
		||||
@@ -214,7 +214,7 @@ export default function ChannelPage() {
 | 
			
		||||
        >
 | 
			
		||||
          <Container>
 | 
			
		||||
            {matchUpMd ? (
 | 
			
		||||
              <ButtonGroup variant="outlined" aria-label="outlined small primary button group" sx={{marginBottom: 2}}>
 | 
			
		||||
              <ButtonGroup variant="outlined" aria-label="outlined small primary button group" sx={{ marginBottom: 2 }}>
 | 
			
		||||
                <Button onClick={handleRefresh} startIcon={<IconRefresh width={'18px'} />}>
 | 
			
		||||
                  刷新
 | 
			
		||||
                </Button>
 | 
			
		||||
 
 | 
			
		||||
@@ -1,177 +1,209 @@
 | 
			
		||||
const defaultConfig = {
 | 
			
		||||
  input: {
 | 
			
		||||
    name: "",
 | 
			
		||||
    name: '',
 | 
			
		||||
    type: 1,
 | 
			
		||||
    key: "",
 | 
			
		||||
    base_url: "",
 | 
			
		||||
    other: "",
 | 
			
		||||
    model_mapping: "",
 | 
			
		||||
    key: '',
 | 
			
		||||
    base_url: '',
 | 
			
		||||
    other: '',
 | 
			
		||||
    model_mapping: '',
 | 
			
		||||
    models: [],
 | 
			
		||||
    groups: ["default"],
 | 
			
		||||
    groups: ['default'],
 | 
			
		||||
    config: {}
 | 
			
		||||
  },
 | 
			
		||||
  inputLabel: {
 | 
			
		||||
    name: "渠道名称",
 | 
			
		||||
    type: "渠道类型",
 | 
			
		||||
    base_url: "渠道API地址",
 | 
			
		||||
    key: "密钥",
 | 
			
		||||
    other: "其他参数",
 | 
			
		||||
    models: "模型",
 | 
			
		||||
    model_mapping: "模型映射关系",
 | 
			
		||||
    groups: "用户组",
 | 
			
		||||
    name: '渠道名称',
 | 
			
		||||
    type: '渠道类型',
 | 
			
		||||
    base_url: '渠道API地址',
 | 
			
		||||
    key: '密钥',
 | 
			
		||||
    other: '其他参数',
 | 
			
		||||
    models: '模型',
 | 
			
		||||
    model_mapping: '模型映射关系',
 | 
			
		||||
    groups: '用户组',
 | 
			
		||||
    config: null
 | 
			
		||||
  },
 | 
			
		||||
  prompt: {
 | 
			
		||||
    type: "请选择渠道类型",
 | 
			
		||||
    name: "请为渠道命名",
 | 
			
		||||
    base_url: "可空,请输入中转API地址,例如通过cloudflare中转",
 | 
			
		||||
    key: "请输入渠道对应的鉴权密钥",
 | 
			
		||||
    other: "",
 | 
			
		||||
    models: "请选择该渠道所支持的模型",
 | 
			
		||||
    type: '请选择渠道类型',
 | 
			
		||||
    name: '请为渠道命名',
 | 
			
		||||
    base_url: '可空,请输入中转API地址,例如通过cloudflare中转',
 | 
			
		||||
    key: '请输入渠道对应的鉴权密钥',
 | 
			
		||||
    other: '',
 | 
			
		||||
    models: '请选择该渠道所支持的模型',
 | 
			
		||||
    model_mapping:
 | 
			
		||||
      '请输入要修改的模型映射关系,格式为:api请求模型ID:实际转发给渠道的模型ID,使用JSON数组表示,例如:{"gpt-3.5": "gpt-35"}',
 | 
			
		||||
    groups: "请选择该渠道所支持的用户组",
 | 
			
		||||
    groups: '请选择该渠道所支持的用户组',
 | 
			
		||||
    config: null
 | 
			
		||||
  },
 | 
			
		||||
  modelGroup: "openai",
 | 
			
		||||
  modelGroup: 'openai'
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
const typeConfig = {
 | 
			
		||||
  3: {
 | 
			
		||||
    inputLabel: {
 | 
			
		||||
      base_url: "AZURE_OPENAI_ENDPOINT",
 | 
			
		||||
      other: "默认 API 版本",
 | 
			
		||||
      base_url: 'AZURE_OPENAI_ENDPOINT',
 | 
			
		||||
      other: '默认 API 版本'
 | 
			
		||||
    },
 | 
			
		||||
    prompt: {
 | 
			
		||||
      base_url: "请填写AZURE_OPENAI_ENDPOINT",
 | 
			
		||||
      other: "请输入默认API版本,例如:2024-03-01-preview",
 | 
			
		||||
    },
 | 
			
		||||
      base_url: '请填写AZURE_OPENAI_ENDPOINT',
 | 
			
		||||
      other: '请输入默认API版本,例如:2024-03-01-preview'
 | 
			
		||||
    }
 | 
			
		||||
  },
 | 
			
		||||
  11: {
 | 
			
		||||
    input: {
 | 
			
		||||
      models: ["PaLM-2"],
 | 
			
		||||
      models: ['PaLM-2']
 | 
			
		||||
    },
 | 
			
		||||
    modelGroup: "google palm",
 | 
			
		||||
    modelGroup: 'google palm'
 | 
			
		||||
  },
 | 
			
		||||
  14: {
 | 
			
		||||
    input: {
 | 
			
		||||
      models: ["claude-instant-1", "claude-2", "claude-2.0", "claude-2.1"],
 | 
			
		||||
      models: ['claude-instant-1', 'claude-2', 'claude-2.0', 'claude-2.1']
 | 
			
		||||
    },
 | 
			
		||||
    modelGroup: "anthropic",
 | 
			
		||||
    modelGroup: 'anthropic'
 | 
			
		||||
  },
 | 
			
		||||
  15: {
 | 
			
		||||
    input: {
 | 
			
		||||
      models: ["ERNIE-Bot", "ERNIE-Bot-turbo", "ERNIE-Bot-4", "Embedding-V1"],
 | 
			
		||||
      models: ['ERNIE-Bot', 'ERNIE-Bot-turbo', 'ERNIE-Bot-4', 'Embedding-V1']
 | 
			
		||||
    },
 | 
			
		||||
    prompt: {
 | 
			
		||||
      key: "按照如下格式输入:APIKey|SecretKey",
 | 
			
		||||
      key: '按照如下格式输入:APIKey|SecretKey'
 | 
			
		||||
    },
 | 
			
		||||
    modelGroup: "baidu",
 | 
			
		||||
    modelGroup: 'baidu'
 | 
			
		||||
  },
 | 
			
		||||
  16: {
 | 
			
		||||
    input: {
 | 
			
		||||
      models: ["glm-4", "glm-4v", "glm-3-turbo", "chatglm_turbo", "chatglm_pro", "chatglm_std", "chatglm_lite"],
 | 
			
		||||
      models: ['glm-4', 'glm-4v', 'glm-3-turbo', 'chatglm_turbo', 'chatglm_pro', 'chatglm_std', 'chatglm_lite']
 | 
			
		||||
    },
 | 
			
		||||
    modelGroup: "zhipu",
 | 
			
		||||
    modelGroup: 'zhipu'
 | 
			
		||||
  },
 | 
			
		||||
  17: {
 | 
			
		||||
    inputLabel: {
 | 
			
		||||
      other: "插件参数",
 | 
			
		||||
      other: '插件参数'
 | 
			
		||||
    },
 | 
			
		||||
    input: {
 | 
			
		||||
      models: [
 | 
			
		||||
        "qwen-turbo",
 | 
			
		||||
        "qwen-plus",
 | 
			
		||||
        "qwen-max",
 | 
			
		||||
        "qwen-max-longcontext",
 | 
			
		||||
        "text-embedding-v1",
 | 
			
		||||
      ],
 | 
			
		||||
      models: ['qwen-turbo', 'qwen-plus', 'qwen-max', 'qwen-max-longcontext', 'text-embedding-v1']
 | 
			
		||||
    },
 | 
			
		||||
    prompt: {
 | 
			
		||||
      other: "请输入插件参数,即 X-DashScope-Plugin 请求头的取值",
 | 
			
		||||
      other: '请输入插件参数,即 X-DashScope-Plugin 请求头的取值'
 | 
			
		||||
    },
 | 
			
		||||
    modelGroup: "ali",
 | 
			
		||||
    modelGroup: 'ali'
 | 
			
		||||
  },
 | 
			
		||||
  18: {
 | 
			
		||||
    inputLabel: {
 | 
			
		||||
      other: "版本号",
 | 
			
		||||
      other: '版本号'
 | 
			
		||||
    },
 | 
			
		||||
    input: {
 | 
			
		||||
      models: [
 | 
			
		||||
          "SparkDesk",
 | 
			
		||||
        'SparkDesk-v1.1',
 | 
			
		||||
        'SparkDesk-v2.1',
 | 
			
		||||
        'SparkDesk-v3.1',
 | 
			
		||||
        'SparkDesk-v3.5'
 | 
			
		||||
      ],
 | 
			
		||||
      models: ['SparkDesk', 'SparkDesk-v1.1', 'SparkDesk-v2.1', 'SparkDesk-v3.1', 'SparkDesk-v3.5']
 | 
			
		||||
    },
 | 
			
		||||
    prompt: {
 | 
			
		||||
      key: "按照如下格式输入:APPID|APISecret|APIKey",
 | 
			
		||||
      other: "请输入版本号,例如:v3.1",
 | 
			
		||||
      key: '按照如下格式输入:APPID|APISecret|APIKey',
 | 
			
		||||
      other: '请输入版本号,例如:v3.1'
 | 
			
		||||
    },
 | 
			
		||||
    modelGroup: "xunfei",
 | 
			
		||||
    modelGroup: 'xunfei'
 | 
			
		||||
  },
 | 
			
		||||
  19: {
 | 
			
		||||
    input: {
 | 
			
		||||
      models: [
 | 
			
		||||
        "360GPT_S2_V9",
 | 
			
		||||
        "embedding-bert-512-v1",
 | 
			
		||||
        "embedding_s1_v1",
 | 
			
		||||
        "semantic_similarity_s1_v1",
 | 
			
		||||
      ],
 | 
			
		||||
      models: ['360GPT_S2_V9', 'embedding-bert-512-v1', 'embedding_s1_v1', 'semantic_similarity_s1_v1']
 | 
			
		||||
    },
 | 
			
		||||
    modelGroup: "360",
 | 
			
		||||
    modelGroup: '360'
 | 
			
		||||
  },
 | 
			
		||||
  22: {
 | 
			
		||||
    prompt: {
 | 
			
		||||
      key: "按照如下格式输入:APIKey-AppId,例如:fastgpt-0sp2gtvfdgyi4k30jwlgwf1i-64f335d84283f05518e9e041",
 | 
			
		||||
    },
 | 
			
		||||
      key: '按照如下格式输入:APIKey-AppId,例如:fastgpt-0sp2gtvfdgyi4k30jwlgwf1i-64f335d84283f05518e9e041'
 | 
			
		||||
    }
 | 
			
		||||
  },
 | 
			
		||||
  23: {
 | 
			
		||||
    input: {
 | 
			
		||||
      models: ["hunyuan"],
 | 
			
		||||
      models: ['hunyuan']
 | 
			
		||||
    },
 | 
			
		||||
    prompt: {
 | 
			
		||||
      key: "按照如下格式输入:AppId|SecretId|SecretKey",
 | 
			
		||||
      key: '按照如下格式输入:AppId|SecretId|SecretKey'
 | 
			
		||||
    },
 | 
			
		||||
    modelGroup: "tencent",
 | 
			
		||||
    modelGroup: 'tencent'
 | 
			
		||||
  },
 | 
			
		||||
  24: {
 | 
			
		||||
    inputLabel: {
 | 
			
		||||
      other: "版本号",
 | 
			
		||||
      other: '版本号'
 | 
			
		||||
    },
 | 
			
		||||
    input: {
 | 
			
		||||
      models: ["gemini-pro"],
 | 
			
		||||
      models: ['gemini-pro']
 | 
			
		||||
    },
 | 
			
		||||
    prompt: {
 | 
			
		||||
      other: "请输入版本号,例如:v1",
 | 
			
		||||
      other: '请输入版本号,例如:v1'
 | 
			
		||||
    },
 | 
			
		||||
    modelGroup: "google gemini",
 | 
			
		||||
    modelGroup: 'google gemini'
 | 
			
		||||
  },
 | 
			
		||||
  25: {
 | 
			
		||||
    input: {
 | 
			
		||||
      models: ['moonshot-v1-8k', 'moonshot-v1-32k', 'moonshot-v1-128k'],
 | 
			
		||||
      models: ['moonshot-v1-8k', 'moonshot-v1-32k', 'moonshot-v1-128k']
 | 
			
		||||
    },
 | 
			
		||||
    modelGroup: "moonshot",
 | 
			
		||||
    modelGroup: 'moonshot'
 | 
			
		||||
  },
 | 
			
		||||
  26: {
 | 
			
		||||
    input: {
 | 
			
		||||
      models: ['Baichuan2-Turbo', 'Baichuan2-Turbo-192k', 'Baichuan-Text-Embedding'],
 | 
			
		||||
      models: ['Baichuan2-Turbo', 'Baichuan2-Turbo-192k', 'Baichuan-Text-Embedding']
 | 
			
		||||
    },
 | 
			
		||||
    modelGroup: "baichuan",
 | 
			
		||||
    modelGroup: 'baichuan'
 | 
			
		||||
  },
 | 
			
		||||
  27: {
 | 
			
		||||
    input: {
 | 
			
		||||
      models: ['abab5.5s-chat', 'abab5.5-chat', 'abab6-chat'],
 | 
			
		||||
      models: ['abab5.5s-chat', 'abab5.5-chat', 'abab6-chat']
 | 
			
		||||
    },
 | 
			
		||||
    modelGroup: "minimax",
 | 
			
		||||
    modelGroup: 'minimax'
 | 
			
		||||
  },
 | 
			
		||||
  29: {
 | 
			
		||||
    modelGroup: "groq",
 | 
			
		||||
    modelGroup: 'groq'
 | 
			
		||||
  },
 | 
			
		||||
  30: {
 | 
			
		||||
    modelGroup: "ollama",
 | 
			
		||||
    modelGroup: 'ollama'
 | 
			
		||||
  },
 | 
			
		||||
  31: {
 | 
			
		||||
    modelGroup: "lingyiwanwu",
 | 
			
		||||
    modelGroup: 'lingyiwanwu'
 | 
			
		||||
  },
 | 
			
		||||
  33: {
 | 
			
		||||
    inputLabel: {
 | 
			
		||||
      key: '',
 | 
			
		||||
      config: {
 | 
			
		||||
        region: 'Region',
 | 
			
		||||
        ak: 'Access Key',
 | 
			
		||||
        sk: 'Secret Key'
 | 
			
		||||
      }
 | 
			
		||||
    },
 | 
			
		||||
    prompt: {
 | 
			
		||||
      key: '',
 | 
			
		||||
      config: {
 | 
			
		||||
        region: 'region,e.g. us-west-2',
 | 
			
		||||
        ak: 'AWS IAM Access Key',
 | 
			
		||||
        sk: 'AWS IAM Secret Key'
 | 
			
		||||
      }
 | 
			
		||||
    },
 | 
			
		||||
    modelGroup: 'anthropic'
 | 
			
		||||
  },
 | 
			
		||||
  37: {
 | 
			
		||||
    inputLabel: {
 | 
			
		||||
      config: {
 | 
			
		||||
        user_id: 'Account ID'
 | 
			
		||||
      }
 | 
			
		||||
    },
 | 
			
		||||
    prompt: {
 | 
			
		||||
      config: {
 | 
			
		||||
        user_id: '请输入 Account ID,例如:d8d7c61dbc334c32d3ced580e4bf42b4'
 | 
			
		||||
      }
 | 
			
		||||
    },
 | 
			
		||||
    modelGroup: 'Cloudflare'
 | 
			
		||||
  },
 | 
			
		||||
  34: {
 | 
			
		||||
    inputLabel: {
 | 
			
		||||
      config: {
 | 
			
		||||
        user_id: 'User ID'
 | 
			
		||||
      }
 | 
			
		||||
    },
 | 
			
		||||
    prompt: {
 | 
			
		||||
      models: '对于 Coze 而言,模型名称即 Bot ID,你可以添加一个前缀 `bot-`,例如:`bot-123456`',
 | 
			
		||||
      config: {
 | 
			
		||||
        user_id: '生成该密钥的用户 ID'
 | 
			
		||||
      }
 | 
			
		||||
    },
 | 
			
		||||
    modelGroup: 'Coze'
 | 
			
		||||
  }
 | 
			
		||||
};
 | 
			
		||||
 | 
			
		||||
export { defaultConfig, typeConfig };
 | 
			
		||||
 
 | 
			
		||||
@@ -21,7 +21,7 @@ import { IconBrandWechat, IconBrandGithub, IconMail } from '@tabler/icons-react'
 | 
			
		||||
import Label from 'ui-component/Label';
 | 
			
		||||
import { API } from 'utils/api';
 | 
			
		||||
import { showError, showSuccess } from 'utils/common';
 | 
			
		||||
import { onGitHubOAuthClicked, onLarkOAuthClicked } from 'utils/common';
 | 
			
		||||
import { onGitHubOAuthClicked, onLarkOAuthClicked, copy } from 'utils/common';
 | 
			
		||||
import * as Yup from 'yup';
 | 
			
		||||
import WechatModal from 'views/Authentication/AuthForms/WechatModal';
 | 
			
		||||
import { useSelector } from 'react-redux';
 | 
			
		||||
@@ -90,8 +90,7 @@ export default function Profile() {
 | 
			
		||||
    const { success, message, data } = res.data;
 | 
			
		||||
    if (success) {
 | 
			
		||||
      setInputs((inputs) => ({ ...inputs, access_token: data }));
 | 
			
		||||
      navigator.clipboard.writeText(data);
 | 
			
		||||
      showSuccess(`令牌已重置并已复制到剪贴板`);
 | 
			
		||||
      copy(data, '访问令牌');
 | 
			
		||||
    } else {
 | 
			
		||||
      showError(message);
 | 
			
		||||
    }
 | 
			
		||||
 
 | 
			
		||||
@@ -18,7 +18,7 @@ import {
 | 
			
		||||
 | 
			
		||||
import Label from 'ui-component/Label';
 | 
			
		||||
import TableSwitch from 'ui-component/Switch';
 | 
			
		||||
import { timestamp2string, renderQuota, showSuccess } from 'utils/common';
 | 
			
		||||
import { timestamp2string, renderQuota, copy } from 'utils/common';
 | 
			
		||||
 | 
			
		||||
import { IconDotsVertical, IconEdit, IconTrash } from '@tabler/icons-react';
 | 
			
		||||
 | 
			
		||||
@@ -83,8 +83,7 @@ export default function RedemptionTableRow({ item, manageRedemption, handleOpenM
 | 
			
		||||
              variant="contained"
 | 
			
		||||
              color="primary"
 | 
			
		||||
              onClick={() => {
 | 
			
		||||
                navigator.clipboard.writeText(item.key);
 | 
			
		||||
                showSuccess('已复制到剪贴板!');
 | 
			
		||||
                copy(item.key, '兑换码');
 | 
			
		||||
              }}
 | 
			
		||||
            >
 | 
			
		||||
              复制
 | 
			
		||||
 
 | 
			
		||||
@@ -20,7 +20,7 @@ import {
 | 
			
		||||
} from '@mui/material';
 | 
			
		||||
 | 
			
		||||
import TableSwitch from 'ui-component/Switch';
 | 
			
		||||
import { renderQuota, showSuccess, timestamp2string } from 'utils/common';
 | 
			
		||||
import { renderQuota, timestamp2string, copy } from 'utils/common';
 | 
			
		||||
 | 
			
		||||
import { IconDotsVertical, IconEdit, IconTrash, IconCaretDownFilled } from '@tabler/icons-react';
 | 
			
		||||
 | 
			
		||||
@@ -141,8 +141,7 @@ export default function TokensTableRow({ item, manageToken, handleOpenModal, set
 | 
			
		||||
    if (type === 'link') {
 | 
			
		||||
      window.open(text);
 | 
			
		||||
    } else {
 | 
			
		||||
      navigator.clipboard.writeText(text);
 | 
			
		||||
      showSuccess('已复制到剪贴板!');
 | 
			
		||||
      copy(text);
 | 
			
		||||
    }
 | 
			
		||||
    handleCloseMenu();
 | 
			
		||||
  };
 | 
			
		||||
@@ -192,7 +191,7 @@ export default function TokensTableRow({ item, manageToken, handleOpenModal, set
 | 
			
		||||
              id={`switch-${item.id}`}
 | 
			
		||||
              checked={statusSwitch === 1}
 | 
			
		||||
              onChange={handleStatus}
 | 
			
		||||
            // disabled={statusSwitch !== 1 && statusSwitch !== 2}
 | 
			
		||||
              // disabled={statusSwitch !== 1 && statusSwitch !== 2}
 | 
			
		||||
            />
 | 
			
		||||
          </Tooltip>
 | 
			
		||||
        </TableCell>
 | 
			
		||||
@@ -211,8 +210,7 @@ export default function TokensTableRow({ item, manageToken, handleOpenModal, set
 | 
			
		||||
              <Button
 | 
			
		||||
                color="primary"
 | 
			
		||||
                onClick={() => {
 | 
			
		||||
                  navigator.clipboard.writeText(`sk-${item.key}`);
 | 
			
		||||
                  showSuccess('已复制到剪贴板!');
 | 
			
		||||
                  copy(`sk-${item.key}`);
 | 
			
		||||
                }}
 | 
			
		||||
              >
 | 
			
		||||
                复制
 | 
			
		||||
@@ -222,7 +220,9 @@ export default function TokensTableRow({ item, manageToken, handleOpenModal, set
 | 
			
		||||
              </Button>
 | 
			
		||||
            </ButtonGroup>
 | 
			
		||||
            <ButtonGroup size="small" aria-label="split button">
 | 
			
		||||
              <Button color="primary" onClick={(e) => handleCopy(COPY_OPTIONS[0], 'link')}>聊天</Button>
 | 
			
		||||
              <Button color="primary" onClick={(e) => handleCopy(COPY_OPTIONS[0], 'link')}>
 | 
			
		||||
                聊天
 | 
			
		||||
              </Button>
 | 
			
		||||
              <Button size="small" onClick={(e) => handleOpenMenu(e, 'link')}>
 | 
			
		||||
                <IconCaretDownFilled size={'16px'} />
 | 
			
		||||
              </Button>
 | 
			
		||||
 
 | 
			
		||||
@@ -4,7 +4,7 @@ import SubCard from 'ui-component/cards/SubCard';
 | 
			
		||||
import inviteImage from 'assets/images/invite/cwok_casual_19.webp';
 | 
			
		||||
import { useState } from 'react';
 | 
			
		||||
import { API } from 'utils/api';
 | 
			
		||||
import { showError, showSuccess } from 'utils/common';
 | 
			
		||||
import { showError, copy } from 'utils/common';
 | 
			
		||||
 | 
			
		||||
const InviteCard = () => {
 | 
			
		||||
  const theme = useTheme();
 | 
			
		||||
@@ -12,8 +12,7 @@ const InviteCard = () => {
 | 
			
		||||
 | 
			
		||||
  const handleInviteUrl = async () => {
 | 
			
		||||
    if (inviteUl) {
 | 
			
		||||
      navigator.clipboard.writeText(inviteUl);
 | 
			
		||||
      showSuccess(`邀请链接已复制到剪切板`);
 | 
			
		||||
      copy(inviteUl, '邀请链接');
 | 
			
		||||
      return;
 | 
			
		||||
    }
 | 
			
		||||
    const res = await API.get('/api/user/aff');
 | 
			
		||||
@@ -21,8 +20,7 @@ const InviteCard = () => {
 | 
			
		||||
    if (success) {
 | 
			
		||||
      let link = `${window.location.origin}/register?aff=${data}`;
 | 
			
		||||
      setInviteUrl(link);
 | 
			
		||||
      navigator.clipboard.writeText(link);
 | 
			
		||||
      showSuccess(`邀请链接已复制到剪切板`);
 | 
			
		||||
      copy(link, '邀请链接');
 | 
			
		||||
    } else {
 | 
			
		||||
      showError(message);
 | 
			
		||||
    }
 | 
			
		||||
 
 | 
			
		||||
@@ -1,5 +1,5 @@
 | 
			
		||||
<!DOCTYPE html>
 | 
			
		||||
<html lang="en">
 | 
			
		||||
<html lang="zh-CN">
 | 
			
		||||
  <head>
 | 
			
		||||
    <meta charset="utf-8" />
 | 
			
		||||
    <link rel="icon" href="logo.png" />
 | 
			
		||||
 
 | 
			
		||||
@@ -1,37 +1,41 @@
 | 
			
		||||
export const CHANNEL_OPTIONS = [
 | 
			
		||||
  { key: 1, text: 'OpenAI', value: 1, color: 'green' },
 | 
			
		||||
  { key: 14, text: 'Anthropic Claude', value: 14, color: 'black' },
 | 
			
		||||
  { key: 33, text: 'AWS Claude', value: 33, color: 'black' },
 | 
			
		||||
  { key: 3, text: 'Azure OpenAI', value: 3, color: 'olive' },
 | 
			
		||||
  { key: 11, text: 'Google PaLM2', value: 11, color: 'orange' },
 | 
			
		||||
  { key: 24, text: 'Google Gemini', value: 24, color: 'orange' },
 | 
			
		||||
  { key: 28, text: 'Mistral AI', value: 28, color: 'orange' },
 | 
			
		||||
  { key: 15, text: '百度文心千帆', value: 15, color: 'blue' },
 | 
			
		||||
  { key: 17, text: '阿里通义千问', value: 17, color: 'orange' },
 | 
			
		||||
  { key: 18, text: '讯飞星火认知', value: 18, color: 'blue' },
 | 
			
		||||
  { key: 16, text: '智谱 ChatGLM', value: 16, color: 'violet' },
 | 
			
		||||
  { key: 19, text: '360 智脑', value: 19, color: 'blue' },
 | 
			
		||||
  { key: 25, text: 'Moonshot AI', value: 25, color: 'black' },
 | 
			
		||||
  { key: 23, text: '腾讯混元', value: 23, color: 'teal' },
 | 
			
		||||
  { key: 26, text: '百川大模型', value: 26, color: 'orange' },
 | 
			
		||||
  { key: 27, text: 'MiniMax', value: 27, color: 'red' },
 | 
			
		||||
  { key: 29, text: 'Groq', value: 29, color: 'orange' },
 | 
			
		||||
  { key: 30, text: 'Ollama', value: 30, color: 'black' },
 | 
			
		||||
  { key: 31, text: '零一万物', value: 31, color: 'green' },
 | 
			
		||||
  { key: 32, text: '阶跃星辰', value: 32, color: 'blue' },
 | 
			
		||||
  { key: 34, text: 'Coze', value: 34, color: 'blue' },
 | 
			
		||||
  { key: 35, text: 'Cohere', value: 35, color: 'blue' },
 | 
			
		||||
  { 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: 2, text: '代理:API2D', value: 2, color: 'blue' },
 | 
			
		||||
  { key: 5, text: '代理:OpenAI-SB', value: 5, color: 'brown' },
 | 
			
		||||
  { key: 7, text: '代理:OhMyGPT', value: 7, color: 'purple' },
 | 
			
		||||
  { key: 10, text: '代理:AI Proxy', value: 10, color: 'purple' },
 | 
			
		||||
  { key: 4, text: '代理:CloseAI', value: 4, color: 'teal' },
 | 
			
		||||
  { key: 6, text: '代理:OpenAI Max', value: 6, color: 'violet' },
 | 
			
		||||
  { key: 9, text: '代理:AI.LS', value: 9, color: 'yellow' },
 | 
			
		||||
  { key: 12, text: '代理:API2GPT', value: 12, color: 'blue' },
 | 
			
		||||
  { key: 13, text: '代理:AIGC2D', value: 13, color: 'purple' }
 | 
			
		||||
    {key: 1, text: 'OpenAI', value: 1, color: 'green'},
 | 
			
		||||
    {key: 14, text: 'Anthropic Claude', value: 14, color: 'black'},
 | 
			
		||||
    {key: 33, text: 'AWS Claude', value: 33, color: 'black'},
 | 
			
		||||
    {key: 3, text: 'Azure OpenAI', value: 3, color: 'olive'},
 | 
			
		||||
    {key: 11, text: 'Google PaLM2', value: 11, color: 'orange'},
 | 
			
		||||
    {key: 24, text: 'Google Gemini', value: 24, color: 'orange'},
 | 
			
		||||
    {key: 28, text: 'Mistral AI', value: 28, color: 'orange'},
 | 
			
		||||
    {key: 15, text: '百度文心千帆', value: 15, color: 'blue'},
 | 
			
		||||
    {key: 17, text: '阿里通义千问', value: 17, color: 'orange'},
 | 
			
		||||
    {key: 18, text: '讯飞星火认知', value: 18, color: 'blue'},
 | 
			
		||||
    {key: 16, text: '智谱 ChatGLM', value: 16, color: 'violet'},
 | 
			
		||||
    {key: 19, text: '360 智脑', value: 19, color: 'blue'},
 | 
			
		||||
    {key: 25, text: 'Moonshot AI', value: 25, color: 'black'},
 | 
			
		||||
    {key: 23, text: '腾讯混元', value: 23, color: 'teal'},
 | 
			
		||||
    {key: 26, text: '百川大模型', value: 26, color: 'orange'},
 | 
			
		||||
    {key: 27, text: 'MiniMax', value: 27, color: 'red'},
 | 
			
		||||
    {key: 29, text: 'Groq', value: 29, color: 'orange'},
 | 
			
		||||
    {key: 30, text: 'Ollama', value: 30, color: 'black'},
 | 
			
		||||
    {key: 31, text: '零一万物', value: 31, color: 'green'},
 | 
			
		||||
    {key: 32, text: '阶跃星辰', value: 32, color: 'blue'},
 | 
			
		||||
    {key: 34, text: 'Coze', value: 34, color: 'blue'},
 | 
			
		||||
    {key: 35, text: 'Cohere', value: 35, color: 'blue'},
 | 
			
		||||
    {key: 36, text: 'DeepSeek', value: 36, color: 'black'},
 | 
			
		||||
    {key: 37, text: 'Cloudflare', value: 37, color: 'orange'},
 | 
			
		||||
    {key: 38, text: 'DeepL', value: 38, color: 'black'},
 | 
			
		||||
    {key: 39, text: 'together.ai', value: 39, color: 'blue'},
 | 
			
		||||
    {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: 2, text: '代理:API2D', value: 2, color: 'blue'},
 | 
			
		||||
    {key: 5, text: '代理:OpenAI-SB', value: 5, color: 'brown'},
 | 
			
		||||
    {key: 7, text: '代理:OhMyGPT', value: 7, color: 'purple'},
 | 
			
		||||
    {key: 10, text: '代理:AI Proxy', value: 10, color: 'purple'},
 | 
			
		||||
    {key: 4, text: '代理:CloseAI', value: 4, color: 'teal'},
 | 
			
		||||
    {key: 6, text: '代理:OpenAI Max', value: 6, color: 'violet'},
 | 
			
		||||
    {key: 9, text: '代理:AI.LS', value: 9, color: 'yellow'},
 | 
			
		||||
    {key: 12, text: '代理:API2GPT', value: 12, color: 'blue'},
 | 
			
		||||
    {key: 13, text: '代理:AIGC2D', value: 13, color: 'purple'}
 | 
			
		||||
];
 | 
			
		||||
 
 | 
			
		||||
@@ -488,6 +488,21 @@ const EditChannel = () => {
 | 
			
		||||
              />
 | 
			
		||||
            </Form.Field>)
 | 
			
		||||
          }
 | 
			
		||||
          {
 | 
			
		||||
            inputs.type === 37 && (
 | 
			
		||||
              <Form.Field>
 | 
			
		||||
                <Form.Input
 | 
			
		||||
                  label='Account ID'
 | 
			
		||||
                  name='user_id'
 | 
			
		||||
                  required
 | 
			
		||||
                  placeholder={'请输入 Account ID,例如:d8d7c61dbc334c32d3ced580e4bf42b4'}
 | 
			
		||||
                  onChange={handleConfigChange}
 | 
			
		||||
                  value={config.user_id}
 | 
			
		||||
                  autoComplete=''
 | 
			
		||||
                />
 | 
			
		||||
              </Form.Field>
 | 
			
		||||
            )
 | 
			
		||||
          }
 | 
			
		||||
          {
 | 
			
		||||
            inputs.type !== 33 && !isEdit && (
 | 
			
		||||
              <Form.Checkbox
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user