] [--version] [--help]\")\n}\n\nfunc Init() {\n\tflag.Parse()\n\n\tif *PrintVersion {\n\t\tfmt.Println(Version)\n\t\tos.Exit(0)\n\t}\n\n\tif *PrintHelp {\n\t\tprintHelp()\n\t\tos.Exit(0)\n\t}\n\n\tif os.Getenv(\"SESSION_SECRET\") != \"\" {\n\t\tif os.Getenv(\"SESSION_SECRET\") == \"random_string\" {\n\t\t\tlogger.SysError(\"SESSION_SECRET is set to an example value, please change it to a random string.\")\n\t\t} else {\n\t\t\tconfig.SessionSecret = os.Getenv(\"SESSION_SECRET\")\n\t\t}\n\t}\n\tif os.Getenv(\"SQLITE_PATH\") != \"\" {\n\t\tSQLitePath = os.Getenv(\"SQLITE_PATH\")\n\t}\n\tif *LogDir != \"\" {\n\t\tvar err error\n\t\t*LogDir, err = filepath.Abs(*LogDir)\n\t\tif err != nil {\n\t\t\tlog.Fatal(err)\n\t\t}\n\t\tif _, err := os.Stat(*LogDir); os.IsNotExist(err) {\n\t\t\terr = os.Mkdir(*LogDir, 0777)\n\t\t\tif err != nil {\n\t\t\t\tlog.Fatal(err)\n\t\t\t}\n\t\t}\n\t\tlogger.LogDir = *LogDir\n\t}\n}\n"
},
{
"path": "common/logger/constants.go",
"content": "package logger\n\nvar LogDir string\n"
},
{
"path": "common/logger/logger.go",
"content": "package logger\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"io\"\n\t\"log\"\n\t\"os\"\n\t\"path/filepath\"\n\t\"runtime\"\n\t\"strings\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/gin-gonic/gin\"\n\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n)\n\ntype loggerLevel string\n\nconst (\n\tloggerDEBUG loggerLevel = \"DEBUG\"\n\tloggerINFO loggerLevel = \"INFO\"\n\tloggerWarn loggerLevel = \"WARN\"\n\tloggerError loggerLevel = \"ERROR\"\n\tloggerFatal loggerLevel = \"FATAL\"\n)\n\nvar setupLogOnce sync.Once\n\nfunc SetupLogger() {\n\tsetupLogOnce.Do(func() {\n\t\tif LogDir != \"\" {\n\t\t\tvar logPath string\n\t\t\tif config.OnlyOneLogFile {\n\t\t\t\tlogPath = filepath.Join(LogDir, \"oneapi.log\")\n\t\t\t} else {\n\t\t\t\tlogPath = filepath.Join(LogDir, fmt.Sprintf(\"oneapi-%s.log\", time.Now().Format(\"20060102\")))\n\t\t\t}\n\t\t\tfd, err := os.OpenFile(logPath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)\n\t\t\tif err != nil {\n\t\t\t\tlog.Fatal(\"failed to open log file\")\n\t\t\t}\n\t\t\tgin.DefaultWriter = io.MultiWriter(os.Stdout, fd)\n\t\t\tgin.DefaultErrorWriter = io.MultiWriter(os.Stderr, fd)\n\t\t}\n\t})\n}\n\nfunc SysLog(s string) {\n\tlogHelper(nil, loggerINFO, s)\n}\n\nfunc SysLogf(format string, a ...any) {\n\tlogHelper(nil, loggerINFO, fmt.Sprintf(format, a...))\n}\n\nfunc SysWarn(s string) {\n\tlogHelper(nil, loggerWarn, s)\n}\n\nfunc SysWarnf(format string, a ...any) {\n\tlogHelper(nil, loggerWarn, fmt.Sprintf(format, a...))\n}\n\nfunc SysError(s string) {\n\tlogHelper(nil, loggerError, s)\n}\n\nfunc SysErrorf(format string, a ...any) {\n\tlogHelper(nil, loggerError, fmt.Sprintf(format, a...))\n}\n\nfunc Debug(ctx context.Context, msg string) {\n\tif !config.DebugEnabled {\n\t\treturn\n\t}\n\tlogHelper(ctx, loggerDEBUG, msg)\n}\n\nfunc Info(ctx context.Context, msg string) {\n\tlogHelper(ctx, loggerINFO, msg)\n}\n\nfunc Warn(ctx context.Context, msg string) {\n\tlogHelper(ctx, loggerWarn, msg)\n}\n\nfunc Error(ctx context.Context, msg string) {\n\tlogHelper(ctx, loggerError, msg)\n}\n\nfunc Debugf(ctx context.Context, format string, a ...any) {\n\tif !config.DebugEnabled {\n\t\treturn\n\t}\n\tlogHelper(ctx, loggerDEBUG, fmt.Sprintf(format, a...))\n}\n\nfunc Infof(ctx context.Context, format string, a ...any) {\n\tlogHelper(ctx, loggerINFO, fmt.Sprintf(format, a...))\n}\n\nfunc Warnf(ctx context.Context, format string, a ...any) {\n\tlogHelper(ctx, loggerWarn, fmt.Sprintf(format, a...))\n}\n\nfunc Errorf(ctx context.Context, format string, a ...any) {\n\tlogHelper(ctx, loggerError, fmt.Sprintf(format, a...))\n}\n\nfunc FatalLog(s string) {\n\tlogHelper(nil, loggerFatal, s)\n}\n\nfunc FatalLogf(format string, a ...any) {\n\tlogHelper(nil, loggerFatal, fmt.Sprintf(format, a...))\n}\n\nfunc logHelper(ctx context.Context, level loggerLevel, msg string) {\n\twriter := gin.DefaultErrorWriter\n\tif level == loggerINFO {\n\t\twriter = gin.DefaultWriter\n\t}\n\tvar requestId string\n\tif ctx != nil {\n\t\trawRequestId := helper.GetRequestID(ctx)\n\t\tif rawRequestId != \"\" {\n\t\t\trequestId = fmt.Sprintf(\" | %s\", rawRequestId)\n\t\t}\n\t}\n\tlineInfo, funcName := getLineInfo()\n\tnow := time.Now()\n\t_, _ = fmt.Fprintf(writer, \"[%s] %v%s%s %s%s \\n\", level, now.Format(\"2006/01/02 - 15:04:05\"), requestId, lineInfo, funcName, msg)\n\tSetupLogger()\n\tif level == loggerFatal {\n\t\tos.Exit(1)\n\t}\n}\n\nfunc getLineInfo() (string, string) {\n\tfuncName := \"[unknown] \"\n\tpc, file, line, ok := runtime.Caller(3)\n\tif ok {\n\t\tif fn := runtime.FuncForPC(pc); fn != nil {\n\t\t\tparts := strings.Split(fn.Name(), \".\")\n\t\t\tfuncName = \"[\" + parts[len(parts)-1] + \"] \"\n\t\t}\n\t} else {\n\t\tfile = \"unknown\"\n\t\tline = 0\n\t}\n\tparts := strings.Split(file, \"one-api/\")\n\tif len(parts) > 1 {\n\t\tfile = parts[1]\n\t}\n\treturn fmt.Sprintf(\" | %s:%d\", file, line), funcName\n}\n"
},
{
"path": "common/message/email.go",
"content": "package message\n\nimport (\n\t\"crypto/rand\"\n\t\"crypto/tls\"\n\t\"encoding/base64\"\n\t\"fmt\"\n\t\"net\"\n\t\"net/smtp\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n)\n\nfunc shouldAuth() bool {\n\treturn config.SMTPAccount != \"\" || config.SMTPToken != \"\"\n}\n\nfunc SendEmail(subject string, receiver string, content string) error {\n\tif receiver == \"\" {\n\t\treturn fmt.Errorf(\"receiver is empty\")\n\t}\n\tif config.SMTPFrom == \"\" { // for compatibility\n\t\tconfig.SMTPFrom = config.SMTPAccount\n\t}\n\tencodedSubject := fmt.Sprintf(\"=?UTF-8?B?%s?=\", base64.StdEncoding.EncodeToString([]byte(subject)))\n\n\t// Extract domain from SMTPFrom\n\tparts := strings.Split(config.SMTPFrom, \"@\")\n\tvar domain string\n\tif len(parts) > 1 {\n\t\tdomain = parts[1]\n\t}\n\t// Generate a unique Message-ID\n\tbuf := make([]byte, 16)\n\t_, err := rand.Read(buf)\n\tif err != nil {\n\t\treturn err\n\t}\n\tmessageId := fmt.Sprintf(\"<%x@%s>\", buf, domain)\n\n\tmail := []byte(fmt.Sprintf(\"To: %s\\r\\n\"+\n\t\t\"From: %s<%s>\\r\\n\"+\n\t\t\"Subject: %s\\r\\n\"+\n\t\t\"Message-ID: %s\\r\\n\"+ // add Message-ID header to avoid being treated as spam, RFC 5322\n\t\t\"Date: %s\\r\\n\"+\n\t\t\"Content-Type: text/html; charset=UTF-8\\r\\n\\r\\n%s\\r\\n\",\n\t\treceiver, config.SystemName, config.SMTPFrom, encodedSubject, messageId, time.Now().Format(time.RFC1123Z), content))\n\n\tauth := smtp.PlainAuth(\"\", config.SMTPAccount, config.SMTPToken, config.SMTPServer)\n\taddr := fmt.Sprintf(\"%s:%d\", config.SMTPServer, config.SMTPPort)\n\tto := strings.Split(receiver, \";\")\n\n\tif config.SMTPPort == 465 || !shouldAuth() {\n\t\t// need advanced client\n\t\tvar conn net.Conn\n\t\tvar err error\n\t\tif config.SMTPPort == 465 {\n\t\t\ttlsConfig := &tls.Config{\n\t\t\t\tInsecureSkipVerify: true,\n\t\t\t\tServerName: config.SMTPServer,\n\t\t\t}\n\t\t\tconn, err = tls.Dial(\"tcp\", fmt.Sprintf(\"%s:%d\", config.SMTPServer, config.SMTPPort), tlsConfig)\n\t\t} else {\n\t\t\tconn, err = net.Dial(\"tcp\", fmt.Sprintf(\"%s:%d\", config.SMTPServer, config.SMTPPort))\n\t\t}\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tclient, err := smtp.NewClient(conn, config.SMTPServer)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tdefer client.Close()\n\t\tif shouldAuth() {\n\t\t\tif err = client.Auth(auth); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\t\tif err = client.Mail(config.SMTPFrom); err != nil {\n\t\t\treturn err\n\t\t}\n\t\treceiverEmails := strings.Split(receiver, \";\")\n\t\tfor _, receiver := range receiverEmails {\n\t\t\tif err = client.Rcpt(receiver); err != nil {\n\t\t\t\treturn err\n\t\t\t}\n\t\t}\n\t\tw, err := client.Data()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\t_, err = w.Write(mail)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\terr = w.Close()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\treturn nil\n\t}\n\terr = smtp.SendMail(addr, auth, config.SMTPAccount, to, mail)\n\tif err != nil && strings.Contains(err.Error(), \"short response\") { // 部分提供商返回该错误，但实际上邮件已经发送成功\n\t\tlogger.SysWarnf(\"short response from SMTP server, return nil instead of error: %s\", err.Error())\n\t\treturn nil\n\t}\n\treturn err\n}\n"
},
{
"path": "common/message/main.go",
"content": "package message\n\nimport (\n\t\"fmt\"\n\t\"github.com/songquanpeng/one-api/common/config\"\n)\n\nconst (\n\tByAll = \"all\"\n\tByEmail = \"email\"\n\tByMessagePusher = \"message_pusher\"\n)\n\nfunc Notify(by string, title string, description string, content string) error {\n\tif by == ByEmail {\n\t\treturn SendEmail(title, config.RootUserEmail, content)\n\t}\n\tif by == ByMessagePusher {\n\t\treturn SendMessage(title, description, content)\n\t}\n\treturn fmt.Errorf(\"unknown notify method: %s\", by)\n}\n"
},
{
"path": "common/message/message-pusher.go",
"content": "package message\n\nimport (\n\t\"bytes\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"net/http\"\n)\n\ntype request struct {\n\tTitle string `json:\"title\"`\n\tDescription string `json:\"description\"`\n\tContent string `json:\"content\"`\n\tURL string `json:\"url\"`\n\tChannel string `json:\"channel\"`\n\tToken string `json:\"token\"`\n}\n\ntype response struct {\n\tSuccess bool `json:\"success\"`\n\tMessage string `json:\"message\"`\n}\n\nfunc SendMessage(title string, description string, content string) error {\n\tif config.MessagePusherAddress == \"\" {\n\t\treturn errors.New(\"message pusher address is not set\")\n\t}\n\treq := request{\n\t\tTitle: title,\n\t\tDescription: description,\n\t\tContent: content,\n\t\tToken: config.MessagePusherToken,\n\t}\n\tdata, err := json.Marshal(req)\n\tif err != nil {\n\t\treturn err\n\t}\n\tresp, err := http.Post(config.MessagePusherAddress,\n\t\t\"application/json\", bytes.NewBuffer(data))\n\tif err != nil {\n\t\treturn err\n\t}\n\tvar res response\n\terr = json.NewDecoder(resp.Body).Decode(&res)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !res.Success {\n\t\treturn errors.New(res.Message)\n\t}\n\treturn nil\n}\n"
},
{
"path": "common/message/template.go",
"content": "package message\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/songquanpeng/one-api/common/config\"\n)\n\n// EmailTemplate 生成美观的 HTML 邮件内容\nfunc EmailTemplate(title, content string) string {\n\treturn fmt.Sprintf(`\n\n\n\n \n
\n
%s \n \n
\n %s\n
\n
\n
此邮件由系统自动发送，请勿直接回复
\n
%s
\n
\n
您好！
\n\t\t\t您正在进行 %s 邮箱验证。
\n\t\t\t您的验证码为：
\n\t\t\t%s
\n\t\t\t验证码 %d 分钟内有效，如果不是本人操作，请忽略。
\n\t\t`, config.SystemName, code, common.VerificationValidMinutes),\n\t)\n\terr := message.SendEmail(subject, email, content)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t})\n\treturn\n}\n\nfunc SendPasswordResetEmail(c *gin.Context) {\n\temail := c.Query(\"email\")\n\tif err := common.Validate.Var(email, \"required,email\"); err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": i18n.Translate(c, \"invalid_parameter\"),\n\t\t})\n\t\treturn\n\t}\n\tif !model.IsEmailAlreadyTaken(email) {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": \"该邮箱地址未注册\",\n\t\t})\n\t\treturn\n\t}\n\tcode := common.GenerateVerificationCode(0)\n\tcommon.RegisterVerificationCodeWithKey(email, code, common.PasswordResetPurpose)\n\tlink := fmt.Sprintf(\"%s/user/reset?email=%s&token=%s\", config.ServerAddress, email, code)\n\tsubject := fmt.Sprintf(\"%s 密码重置\", config.SystemName)\n\tcontent := message.EmailTemplate(\n\t\tsubject,\n\t\tfmt.Sprintf(`\n\t\t\t您好！
\n\t\t\t您正在进行 %s 密码重置。
\n\t\t\t请点击下面的按钮进行密码重置：
\n\t\t\t\n\t\t\t\t重置密码 \n\t\t\t
\n\t\t\t如果按钮无法点击，请复制以下链接到浏览器中打开：
\n\t\t\t%s
\n\t\t\t重置链接 %d 分钟内有效，如果不是本人操作，请忽略。
\n\t\t`, config.SystemName, link, link, common.VerificationValidMinutes),\n\t)\n\terr := message.SendEmail(subject, email, content)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": fmt.Sprintf(\"%s%s\", i18n.Translate(c, \"send_email_failed\"), err.Error()),\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t})\n\treturn\n}\n\ntype PasswordResetRequest struct {\n\tEmail string `json:\"email\"`\n\tToken string `json:\"token\"`\n}\n\nfunc ResetPassword(c *gin.Context) {\n\tvar req PasswordResetRequest\n\terr := json.NewDecoder(c.Request.Body).Decode(&req)\n\tif req.Email == \"\" || req.Token == \"\" {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": i18n.Translate(c, \"invalid_parameter\"),\n\t\t})\n\t\treturn\n\t}\n\tif !common.VerifyCodeWithKey(req.Email, req.Token, common.PasswordResetPurpose) {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": \"重置链接非法或已过期\",\n\t\t})\n\t\treturn\n\t}\n\tpassword := common.GenerateVerificationCode(12)\n\terr = model.ResetUserPasswordByEmail(req.Email, password)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tcommon.DeleteKey(req.Email, common.PasswordResetPurpose)\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": password,\n\t})\n\treturn\n}\n"
},
{
"path": "controller/model.go",
"content": "package controller\n\nimport (\n\t\"fmt\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common/ctxkey\"\n\t\"github.com/songquanpeng/one-api/model\"\n\trelay \"github.com/songquanpeng/one-api/relay\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/apitype\"\n\t\"github.com/songquanpeng/one-api/relay/channeltype\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\trelaymodel \"github.com/songquanpeng/one-api/relay/model\"\n\t\"net/http\"\n\t\"strings\"\n)\n\n// https://platform.openai.com/docs/api-reference/models/list\n\ntype OpenAIModelPermission struct {\n\tId string `json:\"id\"`\n\tObject string `json:\"object\"`\n\tCreated int `json:\"created\"`\n\tAllowCreateEngine bool `json:\"allow_create_engine\"`\n\tAllowSampling bool `json:\"allow_sampling\"`\n\tAllowLogprobs bool `json:\"allow_logprobs\"`\n\tAllowSearchIndices bool `json:\"allow_search_indices\"`\n\tAllowView bool `json:\"allow_view\"`\n\tAllowFineTuning bool `json:\"allow_fine_tuning\"`\n\tOrganization string `json:\"organization\"`\n\tGroup *string `json:\"group\"`\n\tIsBlocking bool `json:\"is_blocking\"`\n}\n\ntype OpenAIModels struct {\n\tId string `json:\"id\"`\n\tObject string `json:\"object\"`\n\tCreated int `json:\"created\"`\n\tOwnedBy string `json:\"owned_by\"`\n\tPermission []OpenAIModelPermission `json:\"permission\"`\n\tRoot string `json:\"root\"`\n\tParent *string `json:\"parent\"`\n}\n\nvar models []OpenAIModels\nvar modelsMap map[string]OpenAIModels\nvar channelId2Models map[int][]string\n\nfunc init() {\n\tvar permission []OpenAIModelPermission\n\tpermission = append(permission, OpenAIModelPermission{\n\t\tId: \"modelperm-LwHkVFn8AcMItP432fKKDIKJ\",\n\t\tObject: \"model_permission\",\n\t\tCreated: 1626777600,\n\t\tAllowCreateEngine: true,\n\t\tAllowSampling: true,\n\t\tAllowLogprobs: true,\n\t\tAllowSearchIndices: false,\n\t\tAllowView: true,\n\t\tAllowFineTuning: false,\n\t\tOrganization: \"*\",\n\t\tGroup: nil,\n\t\tIsBlocking: false,\n\t})\n\t// https://platform.openai.com/docs/models/model-endpoint-compatibility\n\tfor i := 0; i < apitype.Dummy; i++ {\n\t\tif i == apitype.AIProxyLibrary {\n\t\t\tcontinue\n\t\t}\n\t\tadaptor := relay.GetAdaptor(i)\n\t\tchannelName := adaptor.GetChannelName()\n\t\tmodelNames := adaptor.GetModelList()\n\t\tfor _, modelName := range modelNames {\n\t\t\tmodels = append(models, OpenAIModels{\n\t\t\t\tId: modelName,\n\t\t\t\tObject: \"model\",\n\t\t\t\tCreated: 1626777600,\n\t\t\t\tOwnedBy: channelName,\n\t\t\t\tPermission: permission,\n\t\t\t\tRoot: modelName,\n\t\t\t\tParent: nil,\n\t\t\t})\n\t\t}\n\t}\n\tfor _, channelType := range openai.CompatibleChannels {\n\t\tif channelType == channeltype.Azure {\n\t\t\tcontinue\n\t\t}\n\t\tchannelName, channelModelList := openai.GetCompatibleChannelMeta(channelType)\n\t\tfor _, modelName := range channelModelList {\n\t\t\tmodels = append(models, OpenAIModels{\n\t\t\t\tId: modelName,\n\t\t\t\tObject: \"model\",\n\t\t\t\tCreated: 1626777600,\n\t\t\t\tOwnedBy: channelName,\n\t\t\t\tPermission: permission,\n\t\t\t\tRoot: modelName,\n\t\t\t\tParent: nil,\n\t\t\t})\n\t\t}\n\t}\n\tmodelsMap = make(map[string]OpenAIModels)\n\tfor _, model := range models {\n\t\tmodelsMap[model.Id] = model\n\t}\n\tchannelId2Models = make(map[int][]string)\n\tfor i := 1; i < channeltype.Dummy; i++ {\n\t\tadaptor := relay.GetAdaptor(channeltype.ToAPIType(i))\n\t\tmeta := &meta.Meta{\n\t\t\tChannelType: i,\n\t\t}\n\t\tadaptor.Init(meta)\n\t\tchannelId2Models[i] = adaptor.GetModelList()\n\t}\n}\n\nfunc DashboardListModels(c *gin.Context) {\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": channelId2Models,\n\t})\n}\n\nfunc ListAllModels(c *gin.Context) {\n\tc.JSON(200, gin.H{\n\t\t\"object\": \"list\",\n\t\t\"data\": models,\n\t})\n}\n\nfunc ListModels(c *gin.Context) {\n\tctx := c.Request.Context()\n\tvar availableModels []string\n\tif c.GetString(ctxkey.AvailableModels) != \"\" {\n\t\tavailableModels = strings.Split(c.GetString(ctxkey.AvailableModels), \",\")\n\t} else {\n\t\tuserId := c.GetInt(ctxkey.Id)\n\t\tuserGroup, _ := model.CacheGetUserGroup(userId)\n\t\tavailableModels, _ = model.CacheGetGroupModels(ctx, userGroup)\n\t}\n\tmodelSet := make(map[string]bool)\n\tfor _, availableModel := range availableModels {\n\t\tmodelSet[availableModel] = true\n\t}\n\tavailableOpenAIModels := make([]OpenAIModels, 0)\n\tfor _, model := range models {\n\t\tif _, ok := modelSet[model.Id]; ok {\n\t\t\tmodelSet[model.Id] = false\n\t\t\tavailableOpenAIModels = append(availableOpenAIModels, model)\n\t\t}\n\t}\n\tfor modelName, ok := range modelSet {\n\t\tif ok {\n\t\t\tavailableOpenAIModels = append(availableOpenAIModels, OpenAIModels{\n\t\t\t\tId: modelName,\n\t\t\t\tObject: \"model\",\n\t\t\t\tCreated: 1626777600,\n\t\t\t\tOwnedBy: \"custom\",\n\t\t\t\tRoot: modelName,\n\t\t\t\tParent: nil,\n\t\t\t})\n\t\t}\n\t}\n\tc.JSON(200, gin.H{\n\t\t\"object\": \"list\",\n\t\t\"data\": availableOpenAIModels,\n\t})\n}\n\nfunc RetrieveModel(c *gin.Context) {\n\tmodelId := c.Param(\"model\")\n\tif model, ok := modelsMap[modelId]; ok {\n\t\tc.JSON(200, model)\n\t} else {\n\t\tError := relaymodel.Error{\n\t\t\tMessage: fmt.Sprintf(\"The model '%s' does not exist\", modelId),\n\t\t\tType: \"invalid_request_error\",\n\t\t\tParam: \"model\",\n\t\t\tCode: \"model_not_found\",\n\t\t}\n\t\tc.JSON(200, gin.H{\n\t\t\t\"error\": Error,\n\t\t})\n\t}\n}\n\nfunc GetUserAvailableModels(c *gin.Context) {\n\tctx := c.Request.Context()\n\tid := c.GetInt(ctxkey.Id)\n\tuserGroup, err := model.CacheGetUserGroup(id)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tmodels, err := model.CacheGetGroupModels(ctx, userGroup)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": models,\n\t})\n\treturn\n}\n"
},
{
"path": "controller/option.go",
"content": "package controller\n\nimport (\n\t\"encoding/json\"\n\t\"net/http\"\n\t\"strings\"\n\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/i18n\"\n\t\"github.com/songquanpeng/one-api/model\"\n\n\t\"github.com/gin-gonic/gin\"\n)\n\nfunc GetOptions(c *gin.Context) {\n\tvar options []*model.Option\n\tconfig.OptionMapRWMutex.Lock()\n\tfor k, v := range config.OptionMap {\n\t\tif strings.HasSuffix(k, \"Token\") || strings.HasSuffix(k, \"Secret\") {\n\t\t\tcontinue\n\t\t}\n\t\toptions = append(options, &model.Option{\n\t\t\tKey: k,\n\t\t\tValue: helper.Interface2String(v),\n\t\t})\n\t}\n\tconfig.OptionMapRWMutex.Unlock()\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": options,\n\t})\n\treturn\n}\n\nfunc UpdateOption(c *gin.Context) {\n\tvar option model.Option\n\terr := json.NewDecoder(c.Request.Body).Decode(&option)\n\tif err != nil {\n\t\tc.JSON(http.StatusBadRequest, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": i18n.Translate(c, \"invalid_parameter\"),\n\t\t})\n\t\treturn\n\t}\n\tswitch option.Key {\n\tcase \"Theme\":\n\t\tif !config.ValidThemes[option.Value] {\n\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\"success\": false,\n\t\t\t\t\"message\": \"无效的主题\",\n\t\t\t})\n\t\t\treturn\n\t\t}\n\tcase \"GitHubOAuthEnabled\":\n\t\tif option.Value == \"true\" && config.GitHubClientId == \"\" {\n\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\"success\": false,\n\t\t\t\t\"message\": \"无法启用 GitHub OAuth，请先填入 GitHub Client Id 以及 GitHub Client Secret！\",\n\t\t\t})\n\t\t\treturn\n\t\t}\n\tcase \"EmailDomainRestrictionEnabled\":\n\t\tif option.Value == \"true\" && len(config.EmailDomainWhitelist) == 0 {\n\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\"success\": false,\n\t\t\t\t\"message\": \"无法启用邮箱域名限制，请先填入限制的邮箱域名！\",\n\t\t\t})\n\t\t\treturn\n\t\t}\n\tcase \"WeChatAuthEnabled\":\n\t\tif option.Value == \"true\" && config.WeChatServerAddress == \"\" {\n\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\"success\": false,\n\t\t\t\t\"message\": \"无法启用微信登录，请先填入微信登录相关配置信息！\",\n\t\t\t})\n\t\t\treturn\n\t\t}\n\tcase \"TurnstileCheckEnabled\":\n\t\tif option.Value == \"true\" && config.TurnstileSiteKey == \"\" {\n\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\"success\": false,\n\t\t\t\t\"message\": \"无法启用 Turnstile 校验，请先填入 Turnstile 校验相关配置信息！\",\n\t\t\t})\n\t\t\treturn\n\t\t}\n\t}\n\terr = model.UpdateOption(option.Key, option.Value)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t})\n\treturn\n}\n"
},
{
"path": "controller/redemption.go",
"content": "package controller\n\nimport (\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/ctxkey\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/random\"\n\t\"github.com/songquanpeng/one-api/model\"\n\t\"net/http\"\n\t\"strconv\"\n)\n\nfunc GetAllRedemptions(c *gin.Context) {\n\tp, _ := strconv.Atoi(c.Query(\"p\"))\n\tif p < 0 {\n\t\tp = 0\n\t}\n\tredemptions, err := model.GetAllRedemptions(p*config.ItemsPerPage, config.ItemsPerPage)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": redemptions,\n\t})\n\treturn\n}\n\nfunc SearchRedemptions(c *gin.Context) {\n\tkeyword := c.Query(\"keyword\")\n\tredemptions, err := model.SearchRedemptions(keyword)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": redemptions,\n\t})\n\treturn\n}\n\nfunc GetRedemption(c *gin.Context) {\n\tid, err := strconv.Atoi(c.Param(\"id\"))\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tredemption, err := model.GetRedemptionById(id)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": redemption,\n\t})\n\treturn\n}\n\nfunc AddRedemption(c *gin.Context) {\n\tredemption := model.Redemption{}\n\terr := c.ShouldBindJSON(&redemption)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tif len(redemption.Name) == 0 || len(redemption.Name) > 20 {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": \"兑换码名称长度必须在1-20之间\",\n\t\t})\n\t\treturn\n\t}\n\tif redemption.Count <= 0 {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": \"兑换码个数必须大于0\",\n\t\t})\n\t\treturn\n\t}\n\tif redemption.Count > 100 {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": \"一次兑换码批量生成的个数不能大于 100\",\n\t\t})\n\t\treturn\n\t}\n\tvar keys []string\n\tfor i := 0; i < redemption.Count; i++ {\n\t\tkey := random.GetUUID()\n\t\tcleanRedemption := model.Redemption{\n\t\t\tUserId: c.GetInt(ctxkey.Id),\n\t\t\tName: redemption.Name,\n\t\t\tKey: key,\n\t\t\tCreatedTime: helper.GetTimestamp(),\n\t\t\tQuota: redemption.Quota,\n\t\t}\n\t\terr = cleanRedemption.Insert()\n\t\tif err != nil {\n\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\"success\": false,\n\t\t\t\t\"message\": err.Error(),\n\t\t\t\t\"data\": keys,\n\t\t\t})\n\t\t\treturn\n\t\t}\n\t\tkeys = append(keys, key)\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": keys,\n\t})\n\treturn\n}\n\nfunc DeleteRedemption(c *gin.Context) {\n\tid, _ := strconv.Atoi(c.Param(\"id\"))\n\terr := model.DeleteRedemptionById(id)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t})\n\treturn\n}\n\nfunc UpdateRedemption(c *gin.Context) {\n\tstatusOnly := c.Query(\"status_only\")\n\tredemption := model.Redemption{}\n\terr := c.ShouldBindJSON(&redemption)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tcleanRedemption, err := model.GetRedemptionById(redemption.Id)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tif statusOnly != \"\" {\n\t\tcleanRedemption.Status = redemption.Status\n\t} else {\n\t\t// If you add more fields, please also update redemption.Update()\n\t\tcleanRedemption.Name = redemption.Name\n\t\tcleanRedemption.Quota = redemption.Quota\n\t}\n\terr = cleanRedemption.Update()\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": cleanRedemption,\n\t})\n\treturn\n}\n"
},
{
"path": "controller/relay.go",
"content": "package controller\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"fmt\"\n\t\"io\"\n\t\"net/http\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/ctxkey\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/middleware\"\n\tdbmodel \"github.com/songquanpeng/one-api/model\"\n\t\"github.com/songquanpeng/one-api/monitor\"\n\t\"github.com/songquanpeng/one-api/relay/controller\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"github.com/songquanpeng/one-api/relay/relaymode\"\n)\n\n// https://platform.openai.com/docs/api-reference/chat\n\nfunc relayHelper(c *gin.Context, relayMode int) *model.ErrorWithStatusCode {\n\tvar err *model.ErrorWithStatusCode\n\tswitch relayMode {\n\tcase relaymode.ImagesGenerations:\n\t\terr = controller.RelayImageHelper(c, relayMode)\n\tcase relaymode.AudioSpeech:\n\t\tfallthrough\n\tcase relaymode.AudioTranslation:\n\t\tfallthrough\n\tcase relaymode.AudioTranscription:\n\t\terr = controller.RelayAudioHelper(c, relayMode)\n\tcase relaymode.Proxy:\n\t\terr = controller.RelayProxyHelper(c, relayMode)\n\tdefault:\n\t\terr = controller.RelayTextHelper(c)\n\t}\n\treturn err\n}\n\nfunc Relay(c *gin.Context) {\n\tctx := c.Request.Context()\n\trelayMode := relaymode.GetByPath(c.Request.URL.Path)\n\tif config.DebugEnabled {\n\t\trequestBody, _ := common.GetRequestBody(c)\n\t\tlogger.Debugf(ctx, \"request body: %s\", string(requestBody))\n\t}\n\tchannelId := c.GetInt(ctxkey.ChannelId)\n\tuserId := c.GetInt(ctxkey.Id)\n\tbizErr := relayHelper(c, relayMode)\n\tif bizErr == nil {\n\t\tmonitor.Emit(channelId, true)\n\t\treturn\n\t}\n\tlastFailedChannelId := channelId\n\tchannelName := c.GetString(ctxkey.ChannelName)\n\tgroup := c.GetString(ctxkey.Group)\n\toriginalModel := c.GetString(ctxkey.OriginalModel)\n\tgo processChannelRelayError(ctx, userId, channelId, channelName, *bizErr)\n\trequestId := c.GetString(helper.RequestIdKey)\n\tretryTimes := config.RetryTimes\n\tif !shouldRetry(c, bizErr.StatusCode) {\n\t\tlogger.Errorf(ctx, \"relay error happen, status code is %d, won't retry in this case\", bizErr.StatusCode)\n\t\tretryTimes = 0\n\t}\n\tfor i := retryTimes; i > 0; i-- {\n\t\tchannel, err := dbmodel.CacheGetRandomSatisfiedChannel(group, originalModel, i != retryTimes)\n\t\tif err != nil {\n\t\t\tlogger.Errorf(ctx, \"CacheGetRandomSatisfiedChannel failed: %+v\", err)\n\t\t\tbreak\n\t\t}\n\t\tlogger.Infof(ctx, \"using channel #%d to retry (remain times %d)\", channel.Id, i)\n\t\tif channel.Id == lastFailedChannelId {\n\t\t\tcontinue\n\t\t}\n\t\tmiddleware.SetupContextForSelectedChannel(c, channel, originalModel)\n\t\trequestBody, err := common.GetRequestBody(c)\n\t\tc.Request.Body = io.NopCloser(bytes.NewBuffer(requestBody))\n\t\tbizErr = relayHelper(c, relayMode)\n\t\tif bizErr == nil {\n\t\t\treturn\n\t\t}\n\t\tchannelId := c.GetInt(ctxkey.ChannelId)\n\t\tlastFailedChannelId = channelId\n\t\tchannelName := c.GetString(ctxkey.ChannelName)\n\t\tgo processChannelRelayError(ctx, userId, channelId, channelName, *bizErr)\n\t}\n\tif bizErr != nil {\n\t\tif bizErr.StatusCode == http.StatusTooManyRequests {\n\t\t\tbizErr.Error.Message = \"当前分组上游负载已饱和，请稍后再试\"\n\t\t}\n\n\t\t// BUG: bizErr is in race condition\n\t\tbizErr.Error.Message = helper.MessageWithRequestId(bizErr.Error.Message, requestId)\n\t\tc.JSON(bizErr.StatusCode, gin.H{\n\t\t\t\"error\": bizErr.Error,\n\t\t})\n\t}\n}\n\nfunc shouldRetry(c *gin.Context, statusCode int) bool {\n\tif _, ok := c.Get(ctxkey.SpecificChannelId); ok {\n\t\treturn false\n\t}\n\tif statusCode == http.StatusTooManyRequests {\n\t\treturn true\n\t}\n\tif statusCode/100 == 5 {\n\t\treturn true\n\t}\n\tif statusCode == http.StatusBadRequest {\n\t\treturn false\n\t}\n\tif statusCode/100 == 2 {\n\t\treturn false\n\t}\n\treturn true\n}\n\nfunc processChannelRelayError(ctx context.Context, userId int, channelId int, channelName string, err model.ErrorWithStatusCode) {\n\tlogger.Errorf(ctx, \"relay error (channel id %d, user id: %d): %s\", channelId, userId, err.Message)\n\t// https://platform.openai.com/docs/guides/error-codes/api-errors\n\tif monitor.ShouldDisableChannel(&err.Error, err.StatusCode) {\n\t\tmonitor.DisableChannel(channelId, channelName, err.Message)\n\t} else {\n\t\tmonitor.Emit(channelId, false)\n\t}\n}\n\nfunc RelayNotImplemented(c *gin.Context) {\n\terr := model.Error{\n\t\tMessage: \"API not implemented\",\n\t\tType: \"one_api_error\",\n\t\tParam: \"\",\n\t\tCode: \"api_not_implemented\",\n\t}\n\tc.JSON(http.StatusNotImplemented, gin.H{\n\t\t\"error\": err,\n\t})\n}\n\nfunc RelayNotFound(c *gin.Context) {\n\terr := model.Error{\n\t\tMessage: fmt.Sprintf(\"Invalid URL (%s %s)\", c.Request.Method, c.Request.URL.Path),\n\t\tType: \"invalid_request_error\",\n\t\tParam: \"\",\n\t\tCode: \"\",\n\t}\n\tc.JSON(http.StatusNotFound, gin.H{\n\t\t\"error\": err,\n\t})\n}\n"
},
{
"path": "controller/token.go",
"content": "package controller\n\nimport (\n\t\"fmt\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/ctxkey\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/network\"\n\t\"github.com/songquanpeng/one-api/common/random\"\n\t\"github.com/songquanpeng/one-api/model\"\n\t\"net/http\"\n\t\"strconv\"\n)\n\nfunc GetAllTokens(c *gin.Context) {\n\tuserId := c.GetInt(ctxkey.Id)\n\tp, _ := strconv.Atoi(c.Query(\"p\"))\n\tif p < 0 {\n\t\tp = 0\n\t}\n\n\torder := c.Query(\"order\")\n\ttokens, err := model.GetAllUserTokens(userId, p*config.ItemsPerPage, config.ItemsPerPage, order)\n\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": tokens,\n\t})\n\treturn\n}\n\nfunc SearchTokens(c *gin.Context) {\n\tuserId := c.GetInt(ctxkey.Id)\n\tkeyword := c.Query(\"keyword\")\n\ttokens, err := model.SearchUserTokens(userId, keyword)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": tokens,\n\t})\n\treturn\n}\n\nfunc GetToken(c *gin.Context) {\n\tid, err := strconv.Atoi(c.Param(\"id\"))\n\tuserId := c.GetInt(ctxkey.Id)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\ttoken, err := model.GetTokenByIds(id, userId)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": token,\n\t})\n\treturn\n}\n\nfunc GetTokenStatus(c *gin.Context) {\n\ttokenId := c.GetInt(ctxkey.TokenId)\n\tuserId := c.GetInt(ctxkey.Id)\n\ttoken, err := model.GetTokenByIds(tokenId, userId)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\texpiredAt := token.ExpiredTime\n\tif expiredAt == -1 {\n\t\texpiredAt = 0\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"object\": \"credit_summary\",\n\t\t\"total_granted\": token.RemainQuota,\n\t\t\"total_used\": 0, // not supported currently\n\t\t\"total_available\": token.RemainQuota,\n\t\t\"expires_at\": expiredAt * 1000,\n\t})\n}\n\nfunc validateToken(c *gin.Context, token model.Token) error {\n\tif len(token.Name) > 30 {\n\t\treturn fmt.Errorf(\"令牌名称过长\")\n\t}\n\tif token.Subnet != nil && *token.Subnet != \"\" {\n\t\terr := network.IsValidSubnets(*token.Subnet)\n\t\tif err != nil {\n\t\t\treturn fmt.Errorf(\"无效的网段：%s\", err.Error())\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc AddToken(c *gin.Context) {\n\ttoken := model.Token{}\n\terr := c.ShouldBindJSON(&token)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\terr = validateToken(c, token)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": fmt.Sprintf(\"参数错误：%s\", err.Error()),\n\t\t})\n\t\treturn\n\t}\n\n\tcleanToken := model.Token{\n\t\tUserId: c.GetInt(ctxkey.Id),\n\t\tName: token.Name,\n\t\tKey: random.GenerateKey(),\n\t\tCreatedTime: helper.GetTimestamp(),\n\t\tAccessedTime: helper.GetTimestamp(),\n\t\tExpiredTime: token.ExpiredTime,\n\t\tRemainQuota: token.RemainQuota,\n\t\tUnlimitedQuota: token.UnlimitedQuota,\n\t\tModels: token.Models,\n\t\tSubnet: token.Subnet,\n\t}\n\terr = cleanToken.Insert()\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": cleanToken,\n\t})\n\treturn\n}\n\nfunc DeleteToken(c *gin.Context) {\n\tid, _ := strconv.Atoi(c.Param(\"id\"))\n\tuserId := c.GetInt(ctxkey.Id)\n\terr := model.DeleteTokenById(id, userId)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t})\n\treturn\n}\n\nfunc UpdateToken(c *gin.Context) {\n\tuserId := c.GetInt(ctxkey.Id)\n\tstatusOnly := c.Query(\"status_only\")\n\ttoken := model.Token{}\n\terr := c.ShouldBindJSON(&token)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\terr = validateToken(c, token)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": fmt.Sprintf(\"参数错误：%s\", err.Error()),\n\t\t})\n\t\treturn\n\t}\n\tcleanToken, err := model.GetTokenByIds(token.Id, userId)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tif token.Status == model.TokenStatusEnabled {\n\t\tif cleanToken.Status == model.TokenStatusExpired && cleanToken.ExpiredTime <= helper.GetTimestamp() && cleanToken.ExpiredTime != -1 {\n\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\"success\": false,\n\t\t\t\t\"message\": \"令牌已过期，无法启用，请先修改令牌过期时间，或者设置为永不过期\",\n\t\t\t})\n\t\t\treturn\n\t\t}\n\t\tif cleanToken.Status == model.TokenStatusExhausted && cleanToken.RemainQuota <= 0 && !cleanToken.UnlimitedQuota {\n\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\"success\": false,\n\t\t\t\t\"message\": \"令牌可用额度已用尽，无法启用，请先修改令牌剩余额度，或者设置为无限额度\",\n\t\t\t})\n\t\t\treturn\n\t\t}\n\t}\n\tif statusOnly != \"\" {\n\t\tcleanToken.Status = token.Status\n\t} else {\n\t\t// If you add more fields, please also update token.Update()\n\t\tcleanToken.Name = token.Name\n\t\tcleanToken.ExpiredTime = token.ExpiredTime\n\t\tcleanToken.RemainQuota = token.RemainQuota\n\t\tcleanToken.UnlimitedQuota = token.UnlimitedQuota\n\t\tcleanToken.Models = token.Models\n\t\tcleanToken.Subnet = token.Subnet\n\t}\n\terr = cleanToken.Update()\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": cleanToken,\n\t})\n\treturn\n}\n"
},
{
"path": "controller/user.go",
"content": "package controller\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"net/http\"\n\t\"strconv\"\n\t\"time\"\n\n\t\"github.com/gin-contrib/sessions\"\n\t\"github.com/gin-gonic/gin\"\n\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/ctxkey\"\n\t\"github.com/songquanpeng/one-api/common/i18n\"\n\t\"github.com/songquanpeng/one-api/common/random\"\n\t\"github.com/songquanpeng/one-api/model\"\n)\n\ntype LoginRequest struct {\n\tUsername string `json:\"username\"`\n\tPassword string `json:\"password\"`\n}\n\nfunc Login(c *gin.Context) {\n\tif !config.PasswordLoginEnabled {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"message\": \"管理员关闭了密码登录\",\n\t\t\t\"success\": false,\n\t\t})\n\t\treturn\n\t}\n\tvar loginRequest LoginRequest\n\terr := json.NewDecoder(c.Request.Body).Decode(&loginRequest)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"message\": i18n.Translate(c, \"invalid_parameter\"),\n\t\t\t\"success\": false,\n\t\t})\n\t\treturn\n\t}\n\tusername := loginRequest.Username\n\tpassword := loginRequest.Password\n\tif username == \"\" || password == \"\" {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"message\": i18n.Translate(c, \"invalid_parameter\"),\n\t\t\t\"success\": false,\n\t\t})\n\t\treturn\n\t}\n\tuser := model.User{\n\t\tUsername: username,\n\t\tPassword: password,\n\t}\n\terr = user.ValidateAndFill()\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"message\": err.Error(),\n\t\t\t\"success\": false,\n\t\t})\n\t\treturn\n\t}\n\tSetupLogin(&user, c)\n}\n\n// setup session & cookies and then return user info\nfunc SetupLogin(user *model.User, c *gin.Context) {\n\tsession := sessions.Default(c)\n\tsession.Set(\"id\", user.Id)\n\tsession.Set(\"username\", user.Username)\n\tsession.Set(\"role\", user.Role)\n\tsession.Set(\"status\", user.Status)\n\terr := session.Save()\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"message\": \"无法保存会话信息，请重试\",\n\t\t\t\"success\": false,\n\t\t})\n\t\treturn\n\t}\n\tcleanUser := model.User{\n\t\tId: user.Id,\n\t\tUsername: user.Username,\n\t\tDisplayName: user.DisplayName,\n\t\tRole: user.Role,\n\t\tStatus: user.Status,\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"message\": \"\",\n\t\t\"success\": true,\n\t\t\"data\": cleanUser,\n\t})\n}\n\nfunc Logout(c *gin.Context) {\n\tsession := sessions.Default(c)\n\tsession.Clear()\n\terr := session.Save()\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"message\": err.Error(),\n\t\t\t\"success\": false,\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"message\": \"\",\n\t\t\"success\": true,\n\t})\n}\n\nfunc Register(c *gin.Context) {\n\tctx := c.Request.Context()\n\tif !config.RegisterEnabled {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"message\": \"管理员关闭了新用户注册\",\n\t\t\t\"success\": false,\n\t\t})\n\t\treturn\n\t}\n\tif !config.PasswordRegisterEnabled {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"message\": \"管理员关闭了通过密码进行注册，请使用第三方账户验证的形式进行注册\",\n\t\t\t\"success\": false,\n\t\t})\n\t\treturn\n\t}\n\tvar user model.User\n\terr := json.NewDecoder(c.Request.Body).Decode(&user)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": i18n.Translate(c, \"invalid_parameter\"),\n\t\t})\n\t\treturn\n\t}\n\tif err := common.Validate.Struct(&user); err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": i18n.Translate(c, \"invalid_input\"),\n\t\t})\n\t\treturn\n\t}\n\tif config.EmailVerificationEnabled {\n\t\tif user.Email == \"\" || user.VerificationCode == \"\" {\n\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\"success\": false,\n\t\t\t\t\"message\": \"管理员开启了邮箱验证，请输入邮箱地址和验证码\",\n\t\t\t})\n\t\t\treturn\n\t\t}\n\t\tif !common.VerifyCodeWithKey(user.Email, user.VerificationCode, common.EmailVerificationPurpose) {\n\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\"success\": false,\n\t\t\t\t\"message\": \"验证码错误或已过期\",\n\t\t\t})\n\t\t\treturn\n\t\t}\n\t}\n\taffCode := user.AffCode // this code is the inviter's code, not the user's own code\n\tinviterId, _ := model.GetUserIdByAffCode(affCode)\n\tcleanUser := model.User{\n\t\tUsername: user.Username,\n\t\tPassword: user.Password,\n\t\tDisplayName: user.Username,\n\t\tInviterId: inviterId,\n\t}\n\tif config.EmailVerificationEnabled {\n\t\tcleanUser.Email = user.Email\n\t}\n\tif err := cleanUser.Insert(ctx, inviterId); err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t})\n\treturn\n}\n\nfunc GetAllUsers(c *gin.Context) {\n\tp, _ := strconv.Atoi(c.Query(\"p\"))\n\tif p < 0 {\n\t\tp = 0\n\t}\n\n\torder := c.DefaultQuery(\"order\", \"\")\n\tusers, err := model.GetAllUsers(p*config.ItemsPerPage, config.ItemsPerPage, order)\n\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": users,\n\t})\n}\n\nfunc SearchUsers(c *gin.Context) {\n\tkeyword := c.Query(\"keyword\")\n\tusers, err := model.SearchUsers(keyword)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": users,\n\t})\n\treturn\n}\n\nfunc GetUser(c *gin.Context) {\n\tid, err := strconv.Atoi(c.Param(\"id\"))\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tuser, err := model.GetUserById(id, false)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tmyRole := c.GetInt(ctxkey.Role)\n\tif myRole <= user.Role && myRole != model.RoleRootUser {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": \"无权获取同级或更高等级用户的信息\",\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": user,\n\t})\n\treturn\n}\n\nfunc GetUserDashboard(c *gin.Context) {\n\tid := c.GetInt(ctxkey.Id)\n\tnow := time.Now()\n\tstartOfDay := now.Truncate(24*time.Hour).AddDate(0, 0, -6).Unix()\n\tendOfDay := now.Truncate(24 * time.Hour).Add(24*time.Hour - time.Second).Unix()\n\n\tdashboards, err := model.SearchLogsByDayAndModel(id, int(startOfDay), int(endOfDay))\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": \"无法获取统计信息\",\n\t\t\t\"data\": nil,\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": dashboards,\n\t})\n\treturn\n}\n\nfunc GenerateAccessToken(c *gin.Context) {\n\tid := c.GetInt(ctxkey.Id)\n\tuser, err := model.GetUserById(id, true)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tuser.AccessToken = random.GetUUID()\n\n\tif model.DB.Where(\"access_token = ?\", user.AccessToken).First(user).RowsAffected != 0 {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": \"请重试，系统生成的 UUID 竟然重复了！\",\n\t\t})\n\t\treturn\n\t}\n\n\tif err := user.Update(false); err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": user.AccessToken,\n\t})\n\treturn\n}\n\nfunc GetAffCode(c *gin.Context) {\n\tid := c.GetInt(ctxkey.Id)\n\tuser, err := model.GetUserById(id, true)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tif user.AffCode == \"\" {\n\t\tuser.AffCode = random.GetRandomString(4)\n\t\tif err := user.Update(false); err != nil {\n\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\"success\": false,\n\t\t\t\t\"message\": err.Error(),\n\t\t\t})\n\t\t\treturn\n\t\t}\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": user.AffCode,\n\t})\n\treturn\n}\n\nfunc GetSelf(c *gin.Context) {\n\tid := c.GetInt(ctxkey.Id)\n\tuser, err := model.GetUserById(id, false)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": user,\n\t})\n\treturn\n}\n\nfunc UpdateUser(c *gin.Context) {\n\tctx := c.Request.Context()\n\tvar updatedUser model.User\n\terr := json.NewDecoder(c.Request.Body).Decode(&updatedUser)\n\tif err != nil || updatedUser.Id == 0 {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": i18n.Translate(c, \"invalid_parameter\"),\n\t\t})\n\t\treturn\n\t}\n\tif updatedUser.Password == \"\" {\n\t\tupdatedUser.Password = \"$I_LOVE_U\" // make Validator happy :)\n\t}\n\tif err := common.Validate.Struct(&updatedUser); err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": i18n.Translate(c, \"invalid_input\"),\n\t\t})\n\t\treturn\n\t}\n\toriginUser, err := model.GetUserById(updatedUser.Id, false)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tmyRole := c.GetInt(ctxkey.Role)\n\tif myRole <= originUser.Role && myRole != model.RoleRootUser {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": \"无权更新同权限等级或更高权限等级的用户信息\",\n\t\t})\n\t\treturn\n\t}\n\tif myRole <= updatedUser.Role && myRole != model.RoleRootUser {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": \"无权将其他用户权限等级提升到大于等于自己的权限等级\",\n\t\t})\n\t\treturn\n\t}\n\tif updatedUser.Password == \"$I_LOVE_U\" {\n\t\tupdatedUser.Password = \"\" // rollback to what it should be\n\t}\n\tupdatePassword := updatedUser.Password != \"\"\n\tif err := updatedUser.Update(updatePassword); err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tif originUser.Quota != updatedUser.Quota {\n\t\tmodel.RecordLog(ctx, originUser.Id, model.LogTypeManage, fmt.Sprintf(\"管理员将用户额度从 %s修改为 %s\", common.LogQuota(originUser.Quota), common.LogQuota(updatedUser.Quota)))\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t})\n\treturn\n}\n\nfunc UpdateSelf(c *gin.Context) {\n\tvar user model.User\n\terr := json.NewDecoder(c.Request.Body).Decode(&user)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": i18n.Translate(c, \"invalid_parameter\"),\n\t\t})\n\t\treturn\n\t}\n\tif user.Password == \"\" {\n\t\tuser.Password = \"$I_LOVE_U\" // make Validator happy :)\n\t}\n\tif err := common.Validate.Struct(&user); err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": \"输入不合法 \" + err.Error(),\n\t\t})\n\t\treturn\n\t}\n\n\tcleanUser := model.User{\n\t\tId: c.GetInt(ctxkey.Id),\n\t\tUsername: user.Username,\n\t\tPassword: user.Password,\n\t\tDisplayName: user.DisplayName,\n\t}\n\tif user.Password == \"$I_LOVE_U\" {\n\t\tuser.Password = \"\" // rollback to what it should be\n\t\tcleanUser.Password = \"\"\n\t}\n\tupdatePassword := user.Password != \"\"\n\tif err := cleanUser.Update(updatePassword); err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t})\n\treturn\n}\n\nfunc DeleteUser(c *gin.Context) {\n\tid, err := strconv.Atoi(c.Param(\"id\"))\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\toriginUser, err := model.GetUserById(id, false)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tmyRole := c.GetInt(\"role\")\n\tif myRole <= originUser.Role {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": \"无权删除同权限等级或更高权限等级的用户\",\n\t\t})\n\t\treturn\n\t}\n\terr = model.DeleteUserById(id)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": true,\n\t\t\t\"message\": \"\",\n\t\t})\n\t\treturn\n\t}\n}\n\nfunc DeleteSelf(c *gin.Context) {\n\tid := c.GetInt(\"id\")\n\tuser, _ := model.GetUserById(id, false)\n\n\tif user.Role == model.RoleRootUser {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": \"不能删除超级管理员账户\",\n\t\t})\n\t\treturn\n\t}\n\n\terr := model.DeleteUserById(id)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t})\n\treturn\n}\n\nfunc CreateUser(c *gin.Context) {\n\tctx := c.Request.Context()\n\tvar user model.User\n\terr := json.NewDecoder(c.Request.Body).Decode(&user)\n\tif err != nil || user.Username == \"\" || user.Password == \"\" {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": i18n.Translate(c, \"invalid_parameter\"),\n\t\t})\n\t\treturn\n\t}\n\tif err := common.Validate.Struct(&user); err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": i18n.Translate(c, \"invalid_input\"),\n\t\t})\n\t\treturn\n\t}\n\tif user.DisplayName == \"\" {\n\t\tuser.DisplayName = user.Username\n\t}\n\tmyRole := c.GetInt(\"role\")\n\tif user.Role >= myRole {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": \"无法创建权限大于等于自己的用户\",\n\t\t})\n\t\treturn\n\t}\n\t// Even for admin users, we cannot fully trust them!\n\tcleanUser := model.User{\n\t\tUsername: user.Username,\n\t\tPassword: user.Password,\n\t\tDisplayName: user.DisplayName,\n\t}\n\tif err := cleanUser.Insert(ctx, 0); err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t})\n\treturn\n}\n\ntype ManageRequest struct {\n\tUsername string `json:\"username\"`\n\tAction string `json:\"action\"`\n}\n\n// ManageUser Only admin user can do this\nfunc ManageUser(c *gin.Context) {\n\tvar req ManageRequest\n\terr := json.NewDecoder(c.Request.Body).Decode(&req)\n\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": i18n.Translate(c, \"invalid_parameter\"),\n\t\t})\n\t\treturn\n\t}\n\tuser := model.User{\n\t\tUsername: req.Username,\n\t}\n\t// Fill attributes\n\tmodel.DB.Where(&user).First(&user)\n\tif user.Id == 0 {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": \"用户不存在\",\n\t\t})\n\t\treturn\n\t}\n\tmyRole := c.GetInt(\"role\")\n\tif myRole <= user.Role && myRole != model.RoleRootUser {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": \"无权更新同权限等级或更高权限等级的用户信息\",\n\t\t})\n\t\treturn\n\t}\n\tswitch req.Action {\n\tcase \"disable\":\n\t\tuser.Status = model.UserStatusDisabled\n\t\tif user.Role == model.RoleRootUser {\n\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\"success\": false,\n\t\t\t\t\"message\": \"无法禁用超级管理员用户\",\n\t\t\t})\n\t\t\treturn\n\t\t}\n\tcase \"enable\":\n\t\tuser.Status = model.UserStatusEnabled\n\tcase \"delete\":\n\t\tif user.Role == model.RoleRootUser {\n\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\"success\": false,\n\t\t\t\t\"message\": \"无法删除超级管理员用户\",\n\t\t\t})\n\t\t\treturn\n\t\t}\n\t\tif err := user.Delete(); err != nil {\n\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\"success\": false,\n\t\t\t\t\"message\": err.Error(),\n\t\t\t})\n\t\t\treturn\n\t\t}\n\tcase \"promote\":\n\t\tif myRole != model.RoleRootUser {\n\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\"success\": false,\n\t\t\t\t\"message\": \"普通管理员用户无法提升其他用户为管理员\",\n\t\t\t})\n\t\t\treturn\n\t\t}\n\t\tif user.Role >= model.RoleAdminUser {\n\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\"success\": false,\n\t\t\t\t\"message\": \"该用户已经是管理员\",\n\t\t\t})\n\t\t\treturn\n\t\t}\n\t\tuser.Role = model.RoleAdminUser\n\tcase \"demote\":\n\t\tif user.Role == model.RoleRootUser {\n\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\"success\": false,\n\t\t\t\t\"message\": \"无法降级超级管理员用户\",\n\t\t\t})\n\t\t\treturn\n\t\t}\n\t\tif user.Role == model.RoleCommonUser {\n\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\"success\": false,\n\t\t\t\t\"message\": \"该用户已经是普通用户\",\n\t\t\t})\n\t\t\treturn\n\t\t}\n\t\tuser.Role = model.RoleCommonUser\n\t}\n\n\tif err := user.Update(false); err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tclearUser := model.User{\n\t\tRole: user.Role,\n\t\tStatus: user.Status,\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": clearUser,\n\t})\n\treturn\n}\n\nfunc EmailBind(c *gin.Context) {\n\temail := c.Query(\"email\")\n\tcode := c.Query(\"code\")\n\tif !common.VerifyCodeWithKey(email, code, common.EmailVerificationPurpose) {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": \"验证码错误或已过期\",\n\t\t})\n\t\treturn\n\t}\n\tid := c.GetInt(\"id\")\n\tuser := model.User{\n\t\tId: id,\n\t}\n\terr := user.FillUserById()\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tuser.Email = email\n\t// no need to check if this email already taken, because we have used verification code to check it\n\terr = user.Update(false)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tif user.Role == model.RoleRootUser {\n\t\tconfig.RootUserEmail = email\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t})\n\treturn\n}\n\ntype topUpRequest struct {\n\tKey string `json:\"key\"`\n}\n\nfunc TopUp(c *gin.Context) {\n\tctx := c.Request.Context()\n\treq := topUpRequest{}\n\terr := c.ShouldBindJSON(&req)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tid := c.GetInt(\"id\")\n\tquota, err := model.Redeem(ctx, req.Key, id)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t\t\"data\": quota,\n\t})\n\treturn\n}\n\ntype adminTopUpRequest struct {\n\tUserId int `json:\"user_id\"`\n\tQuota int `json:\"quota\"`\n\tRemark string `json:\"remark\"`\n}\n\nfunc AdminTopUp(c *gin.Context) {\n\tctx := c.Request.Context()\n\treq := adminTopUpRequest{}\n\terr := c.ShouldBindJSON(&req)\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\terr = model.IncreaseUserQuota(req.UserId, int64(req.Quota))\n\tif err != nil {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": err.Error(),\n\t\t})\n\t\treturn\n\t}\n\tif req.Remark == \"\" {\n\t\treq.Remark = fmt.Sprintf(\"通过 API 充值 %s\", common.LogQuota(int64(req.Quota)))\n\t}\n\tmodel.RecordTopupLog(ctx, req.UserId, req.Remark, req.Quota)\n\tc.JSON(http.StatusOK, gin.H{\n\t\t\"success\": true,\n\t\t\"message\": \"\",\n\t})\n\treturn\n}\n"
},
{
"path": "docker-compose.yml",
"content": "version: '3.4'\n\nservices:\n one-api:\n image: \"${REGISTRY:-docker.io}/justsong/one-api:latest\"\n container_name: one-api\n restart: always\n command: --log-dir /app/logs\n ports:\n - \"3000:3000\"\n volumes:\n - ./data/oneapi:/data\n - ./logs:/app/logs\n environment:\n - SQL_DSN=oneapi:123456@tcp(db:3306)/one-api # 修改此行，或注释掉以使用 SQLite 作为数据库\n - REDIS_CONN_STRING=redis://redis\n - SESSION_SECRET=random_string # 修改为随机字符串\n - TZ=Asia/Shanghai\n# - NODE_TYPE=slave # 多机部署时从节点取消注释该行\n# - SYNC_FREQUENCY=60 # 需要定期从数据库加载数据时取消注释该行\n# - FRONTEND_BASE_URL=https://openai.justsong.cn # 多机部署时从节点取消注释该行\n depends_on:\n - redis\n - db\n healthcheck:\n test: [ \"CMD-SHELL\", \"wget -q -O - http://localhost:3000/api/status | grep -o '\\\"success\\\":\\\\s*true' | awk -F: '{print $2}'\" ]\n interval: 30s\n timeout: 10s\n retries: 3\n\n redis:\n image: \"${REGISTRY:-docker.io}/redis:latest\"\n container_name: redis\n restart: always\n\n db:\n image: \"${REGISTRY:-docker.io}/mysql:8.2.0\"\n restart: always\n container_name: mysql\n volumes:\n - ./data/mysql:/var/lib/mysql # 挂载目录，持久化存储\n ports:\n - '3306:3306'\n environment:\n TZ: Asia/Shanghai # 设置时区\n MYSQL_ROOT_PASSWORD: 'OneAPI@justsong' # 设置 root 用户的密码\n MYSQL_USER: oneapi # 创建专用用户\n MYSQL_PASSWORD: '123456' # 设置专用用户密码\n MYSQL_DATABASE: one-api # 自动创建数据库"
},
{
"path": "docs/API.md",
"content": "# 使用 API 操控 & 扩展 One API\n> 欢迎提交 PR 在此放上你的拓展项目。\n\n例如，虽然 One API 本身没有直接支持支付，但是你可以通过系统扩展的 API 来实现支付功能。\n\n又或者你想自定义渠道管理策略，也可以通过 API 来实现渠道的禁用与启用。\n\n## 鉴权\nOne API 支持两种鉴权方式：Cookie 和 Token，对于 Token，参照下图获取：\n\n\n\n之后，将 Token 作为请求头的 Authorization 字段的值即可，例如下面使用 Token 调用测试渠道的 API：\n\n\n## 请求格式与响应格式\nOne API 使用 JSON 格式进行请求和响应。\n\n对于响应体，一般格式如下：\n```json\n{\n \"message\": \"请求信息\",\n \"success\": true,\n \"data\": {}\n}\n```\n\n## API 列表\n> 当前 API 列表不全，请自行通过浏览器抓取前端请求\n\n如果现有的 API 没有办法满足你的需求，欢迎提交 issue 讨论。\n\n### 获取当前登录用户信息\n**GET** `/api/user/self`\n\n### 为给定用户充值额度\n**POST** `/api/topup`\n```json\n{\n \"user_id\": 1,\n \"quota\": 100000,\n \"remark\": \"充值 100000 额度\"\n}\n```\n\n## 其他\n### 充值链接上的附加参数\nOne API 会在用户点击充值按钮的时候，将用户的信息和充值信息附加在链接上，例如：\n`https://example.com?username=root&user_id=1&transaction_id=4b3eed80-55d5-443f-bd44-fb18c648c837`\n\n你可以通过解析链接上的参数来获取用户信息和充值信息，然后调用 API 来为用户充值。\n\n注意，不是所有主题都支持该功能，欢迎 PR 补齐。"
},
{
"path": "go.mod",
"content": "module github.com/songquanpeng/one-api\n\ngo 1.20\n\nrequire (\n\tcloud.google.com/go/iam v1.1.10\n\tgithub.com/aws/aws-sdk-go-v2 v1.27.0\n\tgithub.com/aws/aws-sdk-go-v2/credentials v1.17.15\n\tgithub.com/aws/aws-sdk-go-v2/service/bedrockruntime v1.8.3\n\tgithub.com/gin-contrib/cors v1.7.2\n\tgithub.com/gin-contrib/gzip v1.0.1\n\tgithub.com/gin-contrib/sessions v1.0.1\n\tgithub.com/gin-contrib/static v1.1.2\n\tgithub.com/gin-gonic/gin v1.10.0\n\tgithub.com/go-playground/validator/v10 v10.20.0\n\tgithub.com/go-redis/redis/v8 v8.11.5\n\tgithub.com/golang-jwt/jwt v3.2.2+incompatible\n\tgithub.com/google/uuid v1.6.0\n\tgithub.com/gorilla/websocket v1.5.1\n\tgithub.com/jinzhu/copier v0.4.0\n\tgithub.com/joho/godotenv v1.5.1\n\tgithub.com/patrickmn/go-cache v2.1.0+incompatible\n\tgithub.com/pkg/errors v0.9.1\n\tgithub.com/pkoukk/tiktoken-go v0.1.7\n\tgithub.com/smartystreets/goconvey v1.8.1\n\tgithub.com/stretchr/testify v1.9.0\n\tgolang.org/x/crypto v0.31.0\n\tgolang.org/x/image v0.18.0\n\tgolang.org/x/sync v0.10.0\n\tgoogle.golang.org/api v0.187.0\n\tgorm.io/driver/mysql v1.5.6\n\tgorm.io/driver/postgres v1.5.7\n\tgorm.io/driver/sqlite v1.5.1\n\tgorm.io/gorm v1.25.10\n)\n\nrequire (\n\tcloud.google.com/go/auth v0.6.1 // indirect\n\tcloud.google.com/go/auth/oauth2adapt v0.2.2 // indirect\n\tcloud.google.com/go/compute/metadata v0.3.0 // indirect\n\tfilippo.io/edwards25519 v1.1.0 // indirect\n\tgithub.com/aws/aws-sdk-go-v2/aws/protocol/eventstream v1.6.2 // indirect\n\tgithub.com/aws/aws-sdk-go-v2/internal/configsources v1.3.7 // indirect\n\tgithub.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.6.7 // indirect\n\tgithub.com/aws/smithy-go v1.20.2 // indirect\n\tgithub.com/bytedance/sonic v1.11.6 // indirect\n\tgithub.com/bytedance/sonic/loader v0.1.1 // indirect\n\tgithub.com/cespare/xxhash/v2 v2.3.0 // indirect\n\tgithub.com/cloudwego/base64x v0.1.4 // indirect\n\tgithub.com/cloudwego/iasm v0.2.0 // indirect\n\tgithub.com/davecgh/go-spew v1.1.1 // indirect\n\tgithub.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect\n\tgithub.com/dlclark/regexp2 v1.11.0 // indirect\n\tgithub.com/felixge/httpsnoop v1.0.4 // indirect\n\tgithub.com/fsnotify/fsnotify v1.7.0 // indirect\n\tgithub.com/gabriel-vasile/mimetype v1.4.3 // indirect\n\tgithub.com/gin-contrib/sse v0.1.0 // indirect\n\tgithub.com/go-logr/logr v1.4.1 // indirect\n\tgithub.com/go-logr/stdr v1.2.2 // indirect\n\tgithub.com/go-playground/locales v0.14.1 // indirect\n\tgithub.com/go-playground/universal-translator v0.18.1 // indirect\n\tgithub.com/go-sql-driver/mysql v1.8.1 // indirect\n\tgithub.com/goccy/go-json v0.10.3 // indirect\n\tgithub.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da // indirect\n\tgithub.com/golang/protobuf v1.5.4 // indirect\n\tgithub.com/google/s2a-go v0.1.7 // indirect\n\tgithub.com/googleapis/enterprise-certificate-proxy v0.3.2 // indirect\n\tgithub.com/googleapis/gax-go/v2 v2.12.5 // indirect\n\tgithub.com/gopherjs/gopherjs v1.17.2 // indirect\n\tgithub.com/gorilla/context v1.1.2 // indirect\n\tgithub.com/gorilla/securecookie v1.1.2 // indirect\n\tgithub.com/gorilla/sessions v1.2.2 // indirect\n\tgithub.com/jackc/pgpassfile v1.0.0 // indirect\n\tgithub.com/jackc/pgservicefile v0.0.0-20231201235250-de7065d80cb9 // indirect\n\tgithub.com/jackc/pgx/v5 v5.5.5 // indirect\n\tgithub.com/jackc/puddle/v2 v2.2.1 // indirect\n\tgithub.com/jinzhu/inflection v1.0.0 // indirect\n\tgithub.com/jinzhu/now v1.1.5 // indirect\n\tgithub.com/json-iterator/go v1.1.12 // indirect\n\tgithub.com/jtolds/gls v4.20.0+incompatible // indirect\n\tgithub.com/klauspost/cpuid/v2 v2.2.7 // indirect\n\tgithub.com/kr/text v0.2.0 // indirect\n\tgithub.com/leodido/go-urn v1.4.0 // indirect\n\tgithub.com/mattn/go-isatty v0.0.20 // indirect\n\tgithub.com/mattn/go-sqlite3 v1.14.24 // indirect\n\tgithub.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect\n\tgithub.com/modern-go/reflect2 v1.0.2 // indirect\n\tgithub.com/pelletier/go-toml/v2 v2.2.2 // indirect\n\tgithub.com/pmezard/go-difflib v1.0.0 // indirect\n\tgithub.com/smarty/assertions v1.15.0 // indirect\n\tgithub.com/twitchyliquid64/golang-asm v0.15.1 // indirect\n\tgithub.com/ugorji/go/codec v1.2.12 // indirect\n\tgo.opencensus.io v0.24.0 // indirect\n\tgo.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.49.0 // indirect\n\tgo.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.49.0 // indirect\n\tgo.opentelemetry.io/otel v1.24.0 // indirect\n\tgo.opentelemetry.io/otel/metric v1.24.0 // indirect\n\tgo.opentelemetry.io/otel/trace v1.24.0 // indirect\n\tgolang.org/x/arch v0.8.0 // indirect\n\tgolang.org/x/net v0.26.0 // indirect\n\tgolang.org/x/oauth2 v0.21.0 // indirect\n\tgolang.org/x/sys v0.28.0 // indirect\n\tgolang.org/x/text v0.21.0 // indirect\n\tgolang.org/x/time v0.5.0 // indirect\n\tgoogle.golang.org/genproto/googleapis/api v0.0.0-20240617180043-68d350f18fd4 // indirect\n\tgoogle.golang.org/genproto/googleapis/rpc v0.0.0-20240624140628-dc46fd24d27d // indirect\n\tgoogle.golang.org/grpc v1.64.1 // indirect\n\tgoogle.golang.org/protobuf v1.34.2 // indirect\n\tgopkg.in/yaml.v3 v3.0.1 // indirect\n)\n"
},
{
"path": "go.sum",
"content": "cloud.google.com/go v0.26.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw=\ncloud.google.com/go/auth v0.6.1 h1:T0Zw1XM5c1GlpN2HYr2s+m3vr1p2wy+8VN+Z1FKxW38=\ncloud.google.com/go/auth v0.6.1/go.mod h1:eFHG7zDzbXHKmjJddFG/rBlcGp6t25SwRUiEQSlO4x4=\ncloud.google.com/go/auth/oauth2adapt v0.2.2 h1:+TTV8aXpjeChS9M+aTtN/TjdQnzJvmzKFt//oWu7HX4=\ncloud.google.com/go/auth/oauth2adapt v0.2.2/go.mod h1:wcYjgpZI9+Yu7LyYBg4pqSiaRkfEK3GQcpb7C/uyF1Q=\ncloud.google.com/go/compute/metadata v0.3.0 h1:Tz+eQXMEqDIKRsmY3cHTL6FVaynIjX2QxYC4trgAKZc=\ncloud.google.com/go/compute/metadata v0.3.0/go.mod h1:zFmK7XCadkQkj6TtorcaGlCW1hT1fIilQDwofLpJ20k=\ncloud.google.com/go/iam v1.1.10 h1:ZSAr64oEhQSClwBL670MsJAW5/RLiC6kfw3Bqmd5ZDI=\ncloud.google.com/go/iam v1.1.10/go.mod h1:iEgMq62sg8zx446GCaijmA2Miwg5o3UbO+nI47WHJps=\nfilippo.io/edwards25519 v1.1.0 h1:FNf4tywRC1HmFuKW5xopWpigGjJKiJSV0Cqo0cJWDaA=\nfilippo.io/edwards25519 v1.1.0/go.mod h1:BxyFTGdWcka3PhytdK4V28tE5sGfRvvvRV7EaN4VDT4=\ngithub.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=\ngithub.com/aws/aws-sdk-go-v2 v1.27.0 h1:7bZWKoXhzI+mMR/HjdMx8ZCC5+6fY0lS5tr0bbgiLlo=\ngithub.com/aws/aws-sdk-go-v2 v1.27.0/go.mod h1:ffIFB97e2yNsv4aTSGkqtHnppsIJzw7G7BReUZ3jCXM=\ngithub.com/aws/aws-sdk-go-v2/aws/protocol/eventstream v1.6.2 h1:x6xsQXGSmW6frevwDA+vi/wqhp1ct18mVXYN08/93to=\ngithub.com/aws/aws-sdk-go-v2/aws/protocol/eventstream v1.6.2/go.mod h1:lPprDr1e6cJdyYeGXnRaJoP4Md+cDBvi2eOj00BlGmg=\ngithub.com/aws/aws-sdk-go-v2/credentials v1.17.15 h1:YDexlvDRCA8ems2T5IP1xkMtOZ1uLJOCJdTr0igs5zo=\ngithub.com/aws/aws-sdk-go-v2/credentials v1.17.15/go.mod h1:vxHggqW6hFNaeNC0WyXS3VdyjcV0a4KMUY4dKJ96buU=\ngithub.com/aws/aws-sdk-go-v2/internal/configsources v1.3.7 h1:lf/8VTF2cM+N4SLzaYJERKEWAXq8MOMpZfU6wEPWsPk=\ngithub.com/aws/aws-sdk-go-v2/internal/configsources v1.3.7/go.mod h1:4SjkU7QiqK2M9oozyMzfZ/23LmUY+h3oFqhdeP5OMiI=\ngithub.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.6.7 h1:4OYVp0705xu8yjdyoWix0r9wPIRXnIzzOoUpQVHIJ/g=\ngithub.com/aws/aws-sdk-go-v2/internal/endpoints/v2 v2.6.7/go.mod h1:vd7ESTEvI76T2Na050gODNmNU7+OyKrIKroYTu4ABiI=\ngithub.com/aws/aws-sdk-go-v2/service/bedrockruntime v1.8.3 h1:Fihjyd6DeNjcawBEGLH9dkIEUi6AdhucDKPE9nJ4QiY=\ngithub.com/aws/aws-sdk-go-v2/service/bedrockruntime v1.8.3/go.mod h1:opvUj3ismqSCxYc+m4WIjPL0ewZGtvp0ess7cKvBPOQ=\ngithub.com/aws/smithy-go v1.20.2 h1:tbp628ireGtzcHDDmLT/6ADHidqnwgF57XOXZe6tp4Q=\ngithub.com/aws/smithy-go v1.20.2/go.mod h1:krry+ya/rV9RDcV/Q16kpu6ypI4K2czasz0NC3qS14E=\ngithub.com/bytedance/sonic v1.11.6 h1:oUp34TzMlL+OY1OUWxHqsdkgC/Zfc85zGqw9siXjrc0=\ngithub.com/bytedance/sonic v1.11.6/go.mod h1:LysEHSvpvDySVdC2f87zGWf6CIKJcAvqab1ZaiQtds4=\ngithub.com/bytedance/sonic/loader v0.1.1 h1:c+e5Pt1k/cy5wMveRDyk2X4B9hF4g7an8N3zCYjJFNM=\ngithub.com/bytedance/sonic/loader v0.1.1/go.mod h1:ncP89zfokxS5LZrJxl5z0UJcsk4M4yY2JpfqGeCtNLU=\ngithub.com/census-instrumentation/opencensus-proto v0.2.1/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU=\ngithub.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs=\ngithub.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=\ngithub.com/client9/misspell v0.3.4/go.mod h1:qj6jICC3Q7zFZvVWo7KLAzC3yx5G7kyvSDkc90ppPyw=\ngithub.com/cloudwego/base64x v0.1.4 h1:jwCgWpFanWmN8xoIUHa2rtzmkd5J2plF/dnLS6Xd/0Y=\ngithub.com/cloudwego/base64x v0.1.4/go.mod h1:0zlkT4Wn5C6NdauXdJRhSKRlJvmclQ1hhJgA0rcu/8w=\ngithub.com/cloudwego/iasm v0.2.0 h1:1KNIy1I1H9hNNFEEH3DVnI4UujN+1zjpuk6gwHLTssg=\ngithub.com/cloudwego/iasm v0.2.0/go.mod h1:8rXZaNYT2n95jn+zTI1sDr+IgcD2GVs0nlbbQPiEFhY=\ngithub.com/cncf/udpa/go v0.0.0-20191209042840-269d4d468f6f/go.mod h1:M8M6+tZqaGXZJjfX53e64911xZQV5JYwmTeXPW+k8Sc=\ngithub.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E=\ngithub.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=\ngithub.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=\ngithub.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=\ngithub.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f h1:lO4WD4F/rVNCu3HqELle0jiPLLBs70cWOduZpkS1E78=\ngithub.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f/go.mod h1:cuUVRXasLTGF7a8hSLbxyZXjz+1KgoB3wDUb6vlszIc=\ngithub.com/dlclark/regexp2 v1.11.0 h1:G/nrcoOa7ZXlpoa/91N3X7mM3r8eIlMBBJZvsz/mxKI=\ngithub.com/dlclark/regexp2 v1.11.0/go.mod h1:DHkYz0B9wPfa6wondMfaivmHpzrQ3v9q8cnmRbL6yW8=\ngithub.com/envoyproxy/go-control-plane v0.9.0/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=\ngithub.com/envoyproxy/go-control-plane v0.9.1-0.20191026205805-5f8ba28d4473/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=\ngithub.com/envoyproxy/go-control-plane v0.9.4/go.mod h1:6rpuAdCZL397s3pYoYcLgu1mIlRU8Am5FuJP05cCM98=\ngithub.com/envoyproxy/protoc-gen-validate v0.1.0/go.mod h1:iSmxcyjqTsJpI2R4NaDN7+kN2VEUnK/pcBlmesArF7c=\ngithub.com/felixge/httpsnoop v1.0.4 h1:NFTV2Zj1bL4mc9sqWACXbQFVBBg2W3GPvqp8/ESS2Wg=\ngithub.com/felixge/httpsnoop v1.0.4/go.mod h1:m8KPJKqk1gH5J9DgRY2ASl2lWCfGKXixSwevea8zH2U=\ngithub.com/fsnotify/fsnotify v1.7.0 h1:8JEhPFa5W2WU7YfeZzPNqzMP6Lwt7L2715Ggo0nosvA=\ngithub.com/fsnotify/fsnotify v1.7.0/go.mod h1:40Bi/Hjc2AVfZrqy+aj+yEI+/bRxZnMJyTJwOpGvigM=\ngithub.com/gabriel-vasile/mimetype v1.4.3 h1:in2uUcidCuFcDKtdcBxlR0rJ1+fsokWf+uqxgUFjbI0=\ngithub.com/gabriel-vasile/mimetype v1.4.3/go.mod h1:d8uq/6HKRL6CGdk+aubisF/M5GcPfT7nKyLpA0lbSSk=\ngithub.com/gin-contrib/cors v1.7.2 h1:oLDHxdg8W/XDoN/8zamqk/Drgt4oVZDvaV0YmvVICQw=\ngithub.com/gin-contrib/cors v1.7.2/go.mod h1:SUJVARKgQ40dmrzgXEVxj2m7Ig1v1qIboQkPDTQ9t2E=\ngithub.com/gin-contrib/gzip v1.0.1 h1:HQ8ENHODeLY7a4g1Au/46Z92bdGFl74OhxcZble9WJE=\ngithub.com/gin-contrib/gzip v1.0.1/go.mod h1:njt428fdUNRvjuJf16tZMYZ2Yl+WQB53X5wmhDwXvC4=\ngithub.com/gin-contrib/sessions v1.0.1 h1:3hsJyNs7v7N8OtelFmYXFrulAf6zSR7nW/putcPEHxI=\ngithub.com/gin-contrib/sessions v1.0.1/go.mod h1:ouxSFM24/OgIud5MJYQJLpy6AwxQ5EYO9yLhbtObGkM=\ngithub.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE=\ngithub.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI=\ngithub.com/gin-contrib/static v1.1.2 h1:c3kT4bFkUJn2aoRU3s6XnMjJT8J6nNWJkR0NglqmlZ4=\ngithub.com/gin-contrib/static v1.1.2/go.mod h1:Fw90ozjHCmZBWbgrsqrDvO28YbhKEKzKp8GixhR4yLw=\ngithub.com/gin-gonic/gin v1.10.0 h1:nTuyha1TYqgedzytsKYqna+DfLos46nTv2ygFy86HFU=\ngithub.com/gin-gonic/gin v1.10.0/go.mod h1:4PMNQiOhvDRa013RKVbsiNwoyezlm2rm0uX/T7kzp5Y=\ngithub.com/go-logr/logr v1.2.2/go.mod h1:jdQByPbusPIv2/zmleS9BjJVeZ6kBagPoEUsqbVz/1A=\ngithub.com/go-logr/logr v1.4.1 h1:pKouT5E8xu9zeFC39JXRDukb6JFQPXM5p5I91188VAQ=\ngithub.com/go-logr/logr v1.4.1/go.mod h1:9T104GzyrTigFIr8wt5mBrctHMim0Nb2HLGrmQ40KvY=\ngithub.com/go-logr/stdr v1.2.2 h1:hSWxHoqTgW2S2qGc0LTAI563KZ5YKYRhT3MFKZMbjag=\ngithub.com/go-logr/stdr v1.2.2/go.mod h1:mMo/vtBO5dYbehREoey6XUKy/eSumjCCveDpRre4VKE=\ngithub.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s=\ngithub.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA=\ngithub.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY=\ngithub.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY=\ngithub.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=\ngithub.com/go-playground/validator/v10 v10.20.0 h1:K9ISHbSaI0lyB2eWMPJo+kOS/FBExVwjEviJTixqxL8=\ngithub.com/go-playground/validator/v10 v10.20.0/go.mod h1:dbuPbCMFw/DrkbEynArYaCwl3amGuJotoKCe95atGMM=\ngithub.com/go-redis/redis/v8 v8.11.5 h1:AcZZR7igkdvfVmQTPnu9WE37LRrO/YrBH5zWyjDC0oI=\ngithub.com/go-redis/redis/v8 v8.11.5/go.mod h1:gREzHqY1hg6oD9ngVRbLStwAWKhA0FEgq8Jd4h5lpwo=\ngithub.com/go-sql-driver/mysql v1.7.0/go.mod h1:OXbVy3sEdcQ2Doequ6Z5BW6fXNQTmx+9S1MCJN5yJMI=\ngithub.com/go-sql-driver/mysql v1.8.1 h1:LedoTUt/eveggdHS9qUFC1EFSa8bU2+1pZjSRpvNJ1Y=\ngithub.com/go-sql-driver/mysql v1.8.1/go.mod h1:wEBSXgmK//2ZFJyE+qWnIsVGmvmEKlqwuVSjsCm7DZg=\ngithub.com/goccy/go-json v0.10.3 h1:KZ5WoDbxAIgm2HNbYckL0se1fHD6rz5j4ywS6ebzDqA=\ngithub.com/goccy/go-json v0.10.3/go.mod h1:oq7eo15ShAhp70Anwd5lgX2pLfOS3QCiwU/PULtXL6M=\ngithub.com/golang-jwt/jwt v3.2.2+incompatible h1:IfV12K8xAKAnZqdXVzCZ+TOjboZ2keLg81eXfW3O+oY=\ngithub.com/golang-jwt/jwt v3.2.2+incompatible/go.mod h1:8pz2t5EyA70fFQQSrl6XZXzqecmYZeUEB8OUGHkxJ+I=\ngithub.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q=\ngithub.com/golang/groupcache v0.0.0-20200121045136-8c9f03a8e57e/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=\ngithub.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da h1:oI5xCqsCo564l8iNU+DwB5epxmsaqB+rhGL0m5jtYqE=\ngithub.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da/go.mod h1:cIg4eruTrX1D+g88fzRXU5OdNfaM+9IcxsU14FzY7Hc=\ngithub.com/golang/mock v1.1.1/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A=\ngithub.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=\ngithub.com/golang/protobuf v1.3.2/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U=\ngithub.com/golang/protobuf v1.4.0-rc.1/go.mod h1:ceaxUfeHdC40wWswd/P6IGgMaK3YpKi5j83Wpe3EHw8=\ngithub.com/golang/protobuf v1.4.0-rc.1.0.20200221234624-67d41d38c208/go.mod h1:xKAWHe0F5eneWXFV3EuXVDTCmh+JuBKY0li0aMyXATA=\ngithub.com/golang/protobuf v1.4.0-rc.2/go.mod h1:LlEzMj4AhA7rCAGe4KMBDvJI+AwstrUpVNzEA03Pprs=\ngithub.com/golang/protobuf v1.4.0-rc.4.0.20200313231945-b860323f09d0/go.mod h1:WU3c8KckQ9AFe+yFwt9sWVRKCVIyN9cPHBJSNnbL67w=\ngithub.com/golang/protobuf v1.4.0/go.mod h1:jodUvKwWbYaEsadDk5Fwe5c77LiNKVO9IDvqG2KuDX0=\ngithub.com/golang/protobuf v1.4.1/go.mod h1:U8fpvMrcmy5pZrNK1lt4xCsGvpyWQ/VVv6QDs8UjoX8=\ngithub.com/golang/protobuf v1.4.3/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI=\ngithub.com/golang/protobuf v1.5.4 h1:i7eJL8qZTpSEXOPTxNKhASYpMn+8e5Q6AdndVa1dWek=\ngithub.com/golang/protobuf v1.5.4/go.mod h1:lnTiLA8Wa4RWRcIUkrtSVa5nRhsEGBg48fD6rSs7xps=\ngithub.com/google/go-cmp v0.2.0/go.mod h1:oXzfMopK8JAjlY9xF4vHSVASa0yLyX7SntLO5aqRK0M=\ngithub.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=\ngithub.com/google/go-cmp v0.3.1/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=\ngithub.com/google/go-cmp v0.4.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.5.3/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=\ngithub.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI=\ngithub.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=\ngithub.com/google/gofuzz v1.2.0 h1:xRy4A+RhZaiKjJ1bPfwQ8sedCA+YS2YcCHW6ec7JMi0=\ngithub.com/google/s2a-go v0.1.7 h1:60BLSyTrOV4/haCDW4zb1guZItoSq8foHCXrAnjBo/o=\ngithub.com/google/s2a-go v0.1.7/go.mod h1:50CgR4k1jNlWBu4UfS4AcfhVe1r6pdZPygJ3R8F0Qdw=\ngithub.com/google/uuid v1.1.2/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=\ngithub.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=\ngithub.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=\ngithub.com/googleapis/enterprise-certificate-proxy v0.3.2 h1:Vie5ybvEvT75RniqhfFxPRy3Bf7vr3h0cechB90XaQs=\ngithub.com/googleapis/enterprise-certificate-proxy v0.3.2/go.mod h1:VLSiSSBs/ksPL8kq3OBOQ6WRI2QnaFynd1DCjZ62+V0=\ngithub.com/googleapis/gax-go/v2 v2.12.5 h1:8gw9KZK8TiVKB6q3zHY3SBzLnrGp6HQjyfYBYGmXdxA=\ngithub.com/googleapis/gax-go/v2 v2.12.5/go.mod h1:BUDKcWo+RaKq5SC9vVYL0wLADa3VcfswbOMMRmB9H3E=\ngithub.com/gopherjs/gopherjs v1.17.2 h1:fQnZVsXk8uxXIStYb0N4bGk7jeyTalG/wsZjQ25dO0g=\ngithub.com/gopherjs/gopherjs v1.17.2/go.mod h1:pRRIvn/QzFLrKfvEz3qUuEhtE/zLCWfreZ6J5gM2i+k=\ngithub.com/gorilla/context v1.1.2 h1:WRkNAv2uoa03QNIc1A6u4O7DAGMUVoopZhkiXWA2V1o=\ngithub.com/gorilla/context v1.1.2/go.mod h1:KDPwT9i/MeWHiLl90fuTgrt4/wPcv75vFAZLaOOcbxM=\ngithub.com/gorilla/securecookie v1.1.2 h1:YCIWL56dvtr73r6715mJs5ZvhtnY73hBvEF8kXD8ePA=\ngithub.com/gorilla/securecookie v1.1.2/go.mod h1:NfCASbcHqRSY+3a8tlWJwsQap2VX5pwzwo4h3eOamfo=\ngithub.com/gorilla/sessions v1.2.2 h1:lqzMYz6bOfvn2WriPUjNByzeXIlVzURcPmgMczkmTjY=\ngithub.com/gorilla/sessions v1.2.2/go.mod h1:ePLdVu+jbEgHH+KWw8I1z2wqd0BAdAQh/8LRvBeoNcQ=\ngithub.com/gorilla/websocket v1.5.1 h1:gmztn0JnHVt9JZquRuzLw3g4wouNVzKL15iLr/zn/QY=\ngithub.com/gorilla/websocket v1.5.1/go.mod h1:x3kM2JMyaluk02fnUJpQuwD2dCS5NDG2ZHL0uE0tcaY=\ngithub.com/jackc/pgpassfile v1.0.0 h1:/6Hmqy13Ss2zCq62VdNG8tM1wchn8zjSGOBJ6icpsIM=\ngithub.com/jackc/pgpassfile v1.0.0/go.mod h1:CEx0iS5ambNFdcRtxPj5JhEz+xB6uRky5eyVu/W2HEg=\ngithub.com/jackc/pgservicefile v0.0.0-20231201235250-de7065d80cb9 h1:L0QtFUgDarD7Fpv9jeVMgy/+Ec0mtnmYuImjTz6dtDA=\ngithub.com/jackc/pgservicefile v0.0.0-20231201235250-de7065d80cb9/go.mod h1:5TJZWKEWniPve33vlWYSoGYefn3gLQRzjfDlhSJ9ZKM=\ngithub.com/jackc/pgx/v5 v5.5.5 h1:amBjrZVmksIdNjxGW/IiIMzxMKZFelXbUoPNb+8sjQw=\ngithub.com/jackc/pgx/v5 v5.5.5/go.mod h1:ez9gk+OAat140fv9ErkZDYFWmXLfV+++K0uAOiwgm1A=\ngithub.com/jackc/puddle/v2 v2.2.1 h1:RhxXJtFG022u4ibrCSMSiu5aOq1i77R3OHKNJj77OAk=\ngithub.com/jackc/puddle/v2 v2.2.1/go.mod h1:vriiEXHvEE654aYKXXjOvZM39qJ0q+azkZFrfEOc3H4=\ngithub.com/jinzhu/copier v0.4.0 h1:w3ciUoD19shMCRargcpm0cm91ytaBhDvuRpz1ODO/U8=\ngithub.com/jinzhu/copier v0.4.0/go.mod h1:DfbEm0FYsaqBcKcFuvmOZb218JkPGtvSHsKg8S8hyyg=\ngithub.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E=\ngithub.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc=\ngithub.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ=\ngithub.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8=\ngithub.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=\ngithub.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=\ngithub.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=\ngithub.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo=\ngithub.com/jtolds/gls v4.20.0+incompatible h1:xdiiI2gbIgH/gLH7ADydsJ1uDOEzR8yvV7C0MuV77Wo=\ngithub.com/jtolds/gls v4.20.0+incompatible/go.mod h1:QJZ7F/aHp+rZTRtaJ1ow/lLfFfVYBRgL+9YlvaHOwJU=\ngithub.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg=\ngithub.com/klauspost/cpuid/v2 v2.2.7 h1:ZWSB3igEs+d0qvnxR/ZBzXVmxkgt8DdzP6m9pfuVLDM=\ngithub.com/klauspost/cpuid/v2 v2.2.7/go.mod h1:Lcz8mBdAVJIBVzewtcLocK12l3Y+JytZYpaMropDUws=\ngithub.com/knz/go-libedit v1.10.1/go.mod h1:MZTVkCWyz0oBc7JOWP3wNAzd002ZbM/5hgShxwh4x8M=\ngithub.com/kr/pretty v0.3.0 h1:WgNl7dwNpEZ6jJ9k1snq4pZsg7DOEN8hP9Xw0Tsjwk0=\ngithub.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=\ngithub.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=\ngithub.com/leodido/go-urn v1.4.0 h1:WT9HwE9SGECu3lg4d/dIA+jxlljEa1/ffXKmRjqdmIQ=\ngithub.com/leodido/go-urn v1.4.0/go.mod h1:bvxc+MVxLKB4z00jd1z+Dvzr47oO32F/QSNjSBOlFxI=\ngithub.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=\ngithub.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=\ngithub.com/mattn/go-sqlite3 v1.14.24 h1:tpSp2G2KyMnnQu99ngJ47EIkWVmliIizyZBfPrBWDRM=\ngithub.com/mattn/go-sqlite3 v1.14.24/go.mod h1:Uh1q+B4BYcTPb+yiD3kU8Ct7aC0hY9fxUwlHK0RXw+Y=\ngithub.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=\ngithub.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg=\ngithub.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q=\ngithub.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M=\ngithub.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk=\ngithub.com/nxadm/tail v1.4.8 h1:nPr65rt6Y5JFSKQO7qToXr7pePgD6Gwiw05lkbyAQTE=\ngithub.com/onsi/ginkgo v1.16.5 h1:8xi0RTUf59SOSfEtZMvwTvXYMzG4gV23XVHOZiXNtnE=\ngithub.com/onsi/gomega v1.18.1 h1:M1GfJqGRrBrrGGsbxzV5dqM2U2ApXefZCQpkukxYRLE=\ngithub.com/patrickmn/go-cache v2.1.0+incompatible h1:HRMgzkcYKYpi3C8ajMPV8OFXaaRUnok+kx1WdO15EQc=\ngithub.com/patrickmn/go-cache v2.1.0+incompatible/go.mod h1:3Qf8kWWT7OJRJbdiICTKqZju1ZixQ/KpMGzzAfe6+WQ=\ngithub.com/pelletier/go-toml/v2 v2.2.2 h1:aYUidT7k73Pcl9nb2gScu7NSrKCSHIDE89b3+6Wq+LM=\ngithub.com/pelletier/go-toml/v2 v2.2.2/go.mod h1:1t835xjRzz80PqgE6HHgN2JOsmgYu/h4qDAS4n929Rs=\ngithub.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=\ngithub.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=\ngithub.com/pkoukk/tiktoken-go v0.1.7 h1:qOBHXX4PHtvIvmOtyg1EeKlwFRiMKAcoMp4Q+bLQDmw=\ngithub.com/pkoukk/tiktoken-go v0.1.7/go.mod h1:9NiV+i9mJKGj1rYOT+njbv+ZwA/zJxYdewGl6qVatpg=\ngithub.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=\ngithub.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=\ngithub.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA=\ngithub.com/rogpeppe/go-internal v1.8.0 h1:FCbCCtXNOY3UtUuHUYaghJg4y7Fd14rXifAYUAtL9R8=\ngithub.com/smarty/assertions v1.15.0 h1:cR//PqUBUiQRakZWqBiFFQ9wb8emQGDb0HeGdqGByCY=\ngithub.com/smarty/assertions v1.15.0/go.mod h1:yABtdzeQs6l1brC900WlRNwj6ZR55d7B+E8C6HtKdec=\ngithub.com/smartystreets/goconvey v1.8.1 h1:qGjIddxOk4grTu9JPOU31tVfq3cNdBlNa5sSznIX1xY=\ngithub.com/smartystreets/goconvey v1.8.1/go.mod h1:+/u4qLyY6x1jReYOp7GOM2FSt8aP9CzCZL03bI28W60=\ngithub.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=\ngithub.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw=\ngithub.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo=\ngithub.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA=\ngithub.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=\ngithub.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=\ngithub.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=\ngithub.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=\ngithub.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4=\ngithub.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo=\ngithub.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=\ngithub.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=\ngithub.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI=\ngithub.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08=\ngithub.com/ugorji/go/codec v1.2.12 h1:9LC83zGrHhuUA9l16C9AHXAqEV/2wBQ4nkvumAE65EE=\ngithub.com/ugorji/go/codec v1.2.12/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg=\ngo.opencensus.io v0.24.0 h1:y73uSU6J157QMP2kn2r30vwW1A2W2WFwSCGnAVxeaD0=\ngo.opencensus.io v0.24.0/go.mod h1:vNK8G9p7aAivkbmorf4v+7Hgx+Zs0yY+0fOtgBfjQKo=\ngo.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.49.0 h1:4Pp6oUg3+e/6M4C0A/3kJ2VYa++dsWVTtGgLVj5xtHg=\ngo.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.49.0/go.mod h1:Mjt1i1INqiaoZOMGR1RIUJN+i3ChKoFRqzrRQhlkbs0=\ngo.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.49.0 h1:jq9TW8u3so/bN+JPT166wjOI6/vQPF6Xe7nMNIltagk=\ngo.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.49.0/go.mod h1:p8pYQP+m5XfbZm9fxtSKAbM6oIllS7s2AfxrChvc7iw=\ngo.opentelemetry.io/otel v1.24.0 h1:0LAOdjNmQeSTzGBzduGe/rU4tZhMwL5rWgtp9Ku5Jfo=\ngo.opentelemetry.io/otel v1.24.0/go.mod h1:W7b9Ozg4nkF5tWI5zsXkaKKDjdVjpD4oAt9Qi/MArHo=\ngo.opentelemetry.io/otel/metric v1.24.0 h1:6EhoGWWK28x1fbpA4tYTOWBkPefTDQnb8WSGXlc88kI=\ngo.opentelemetry.io/otel/metric v1.24.0/go.mod h1:VYhLe1rFfxuTXLgj4CBiyz+9WYBA8pNGJgDcSFRKBco=\ngo.opentelemetry.io/otel/trace v1.24.0 h1:CsKnnL4dUAr/0llH9FKuc698G04IrpWV0MQA/Y1YELI=\ngo.opentelemetry.io/otel/trace v1.24.0/go.mod h1:HPc3Xr/cOApsBI154IU0OI0HJexz+aw5uPdbs3UCjNU=\ngolang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8=\ngolang.org/x/arch v0.8.0 h1:3wRIsP3pM4yUptoR96otTUOXI367OS0+c9eeRi9doIc=\ngolang.org/x/arch v0.8.0/go.mod h1:FEVrYAQjsQXMVJ1nsMoVVXPZg6p2JE2mx8psSWTDQys=\ngolang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=\ngolang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto=\ngolang.org/x/crypto v0.31.0 h1:ihbySMvVjLAeSH1IbfcRTkD/iNscyz8rGzjF/E5hV6U=\ngolang.org/x/crypto v0.31.0/go.mod h1:kDsLvtWBEx7MV9tJOj9bnXsPbxwJQ6csT/x4KIN4Ssk=\ngolang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=\ngolang.org/x/image v0.18.0 h1:jGzIakQa/ZXI1I0Fxvaa9W7yP25TqT6cHIHn+6CqvSQ=\ngolang.org/x/image v0.18.0/go.mod h1:4yyo5vMFQjVjUcVk4jEQcU9MGy/rulF5WvUILseCM2E=\ngolang.org/x/lint v0.0.0-20181026193005-c67002cb31c3/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE=\ngolang.org/x/lint v0.0.0-20190227174305-5b3e6a55c961/go.mod h1:wehouNa3lNwaWXcvxsM5YxQ5yQlVC4a0KAMCusXpPoU=\ngolang.org/x/lint v0.0.0-20190313153728-d0100b6bd8b3/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc=\ngolang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20190213061140-3a22650c66bd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=\ngolang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=\ngolang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=\ngolang.org/x/net v0.0.0-20201110031124-69a78807bb2b/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU=\ngolang.org/x/net v0.26.0 h1:soB7SVo0PWrY4vPW/+ay0jKDNScG2X9wFeYlXIvJsOQ=\ngolang.org/x/net v0.26.0/go.mod h1:5YKkiSynbBIh3p6iOc/vibscux0x38BZDkn8sCUPxHE=\ngolang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=\ngolang.org/x/oauth2 v0.21.0 h1:tsimM75w1tF/uws5rbeHzIWxEqElMehnc+iW793zsZs=\ngolang.org/x/oauth2 v0.21.0/go.mod h1:XYTD2NtWslqkgxebSiOHnXEap4TF09sJSc7H1sXbhtI=\ngolang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20181108010431-42b317875d0f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=\ngolang.org/x/sync v0.10.0 h1:3NQrjDixjgGwUOCaF8w2+VYHv0Ve/vGYSbdkTa98gmQ=\ngolang.org/x/sync v0.10.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=\ngolang.org/x/sys v0.0.0-20180830151530-49385e6e1522/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=\ngolang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=\ngolang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=\ngolang.org/x/sys v0.28.0 h1:Fksou7UEQUWlKvIdsqzJmUmCX3cZuD2+P3XyyzwMhlA=\ngolang.org/x/sys v0.28.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=\ngolang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=\ngolang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=\ngolang.org/x/text v0.21.0 h1:zyQAAkrwaneQ066sspRyJaG9VNi/YJ1NfzcGB3hZ/qo=\ngolang.org/x/text v0.21.0/go.mod h1:4IBbMaMmOPCJ8SecivzSH54+73PCFmPWxNTLm+vZkEQ=\ngolang.org/x/time v0.5.0 h1:o7cqy6amK/52YcAKIPlM3a+Fpj35zvRj2TP+e1xFSfk=\ngolang.org/x/time v0.5.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM=\ngolang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20190114222345-bf090417da8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=\ngolang.org/x/tools v0.0.0-20190226205152-f727befe758c/go.mod h1:9Yl7xja0Znq3iFh3HoIrodX9oNMXvdceNzlUR8zjMvY=\ngolang.org/x/tools v0.0.0-20190311212946-11955173bddd/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=\ngolang.org/x/tools v0.0.0-20190524140312-2c0ae7006135/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q=\ngolang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=\ngoogle.golang.org/api v0.187.0 h1:Mxs7VATVC2v7CY+7Xwm4ndkX71hpElcvx0D1Ji/p1eo=\ngoogle.golang.org/api v0.187.0/go.mod h1:KIHlTc4x7N7gKKuVsdmfBXN13yEEWXWFURWY6SBp2gk=\ngoogle.golang.org/appengine v1.1.0/go.mod h1:EbEs0AVv82hx2wNQdGPgUI5lhzA/G0D9YwlJXL52JkM=\ngoogle.golang.org/appengine v1.4.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4=\ngoogle.golang.org/genproto v0.0.0-20180817151627-c66870c02cf8/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc=\ngoogle.golang.org/genproto v0.0.0-20190819201941-24fa4b261c55/go.mod h1:DMBHOl98Agz4BDEuKkezgsaosCRResVns1a3J2ZsMNc=\ngoogle.golang.org/genproto v0.0.0-20200526211855-cb27e3aa2013/go.mod h1:NbSheEEYHJ7i3ixzK3sjbqSGDJWnxyFXZblF3eUsNvo=\ngoogle.golang.org/genproto/googleapis/api v0.0.0-20240617180043-68d350f18fd4 h1:MuYw1wJzT+ZkybKfaOXKp5hJiZDn2iHaXRw0mRYdHSc=\ngoogle.golang.org/genproto/googleapis/api v0.0.0-20240617180043-68d350f18fd4/go.mod h1:px9SlOOZBg1wM1zdnr8jEL4CNGUBZ+ZKYtNPApNQc4c=\ngoogle.golang.org/genproto/googleapis/rpc v0.0.0-20240624140628-dc46fd24d27d h1:k3zyW3BYYR30e8v3x0bTDdE9vpYFjZHK+HcyqkrppWk=\ngoogle.golang.org/genproto/googleapis/rpc v0.0.0-20240624140628-dc46fd24d27d/go.mod h1:Ue6ibwXGpU+dqIcODieyLOcgj7z8+IcskoNIgZxtrFY=\ngoogle.golang.org/grpc v1.19.0/go.mod h1:mqu4LbDTu4XGKhr4mRzUsmM4RtVoemTSY81AxZiDr8c=\ngoogle.golang.org/grpc v1.23.0/go.mod h1:Y5yQAOtifL1yxbo5wqy6BxZv8vAUGQwXBOALyacEbxg=\ngoogle.golang.org/grpc v1.25.1/go.mod h1:c3i+UQWmh7LiEpx4sFZnkU36qjEYZ0imhYfXVyQciAY=\ngoogle.golang.org/grpc v1.27.0/go.mod h1:qbnxyOmOxrQa7FizSgH+ReBfzJrCY1pSN7KXBS8abTk=\ngoogle.golang.org/grpc v1.33.2/go.mod h1:JMHMWHQWaTccqQQlmk3MJZS+GWXOdAesneDmEnv2fbc=\ngoogle.golang.org/grpc v1.64.1 h1:LKtvyfbX3UGVPFcGqJ9ItpVWW6oN/2XqTxfAnwRRXiA=\ngoogle.golang.org/grpc v1.64.1/go.mod h1:hiQF4LFZelK2WKaP6W0L92zGHtiQdZxk8CrSdvyjeP0=\ngoogle.golang.org/protobuf v0.0.0-20200109180630-ec00e32a8dfd/go.mod h1:DFci5gLYBciE7Vtevhsrf46CRTquxDuWsQurQQe4oz8=\ngoogle.golang.org/protobuf v0.0.0-20200221191635-4d8936d0db64/go.mod h1:kwYJMbMJ01Woi6D6+Kah6886xMZcty6N08ah7+eCXa0=\ngoogle.golang.org/protobuf v0.0.0-20200228230310-ab0ca4ff8a60/go.mod h1:cfTl7dwQJ+fmap5saPgwCLgHXTUD7jkjRqWcaiX5VyM=\ngoogle.golang.org/protobuf v1.20.1-0.20200309200217-e05f789c0967/go.mod h1:A+miEFZTKqfCUM6K7xSMQL9OKL/b6hQv+e19PK+JZNE=\ngoogle.golang.org/protobuf v1.21.0/go.mod h1:47Nbq4nVaFHyn7ilMalzfO3qCViNmqZ2kzikPIcrTAo=\ngoogle.golang.org/protobuf v1.22.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=\ngoogle.golang.org/protobuf v1.23.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=\ngoogle.golang.org/protobuf v1.23.1-0.20200526195155-81db48ad09cc/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU=\ngoogle.golang.org/protobuf v1.25.0/go.mod h1:9JNX74DMeImyA3h4bdi1ymwjUzf21/xIlbajtzgsN7c=\ngoogle.golang.org/protobuf v1.34.2 h1:6xV6lTsCfpGD21XK49h7MhtcApnLqkfYgPcdHftf6hg=\ngoogle.golang.org/protobuf v1.34.2/go.mod h1:qYOHts0dSfpeUzUFpOMr/WGzszTmLH+DiWniOlNbLDw=\ngopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=\ngopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=\ngopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 h1:uRGJdciOHaEIrze2W8Q3AKkepLTh2hOroT7a+7czfdQ=\ngopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=\ngopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=\ngopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=\ngopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=\ngorm.io/driver/mysql v1.5.6 h1:Ld4mkIickM+EliaQZQx3uOJDJHtrd70MxAUqWqlx3Y8=\ngorm.io/driver/mysql v1.5.6/go.mod h1:sEtPWMiqiN1N1cMXoXmBbd8C6/l+TESwriotuRRpkDM=\ngorm.io/driver/postgres v1.5.7 h1:8ptbNJTDbEmhdr62uReG5BGkdQyeasu/FZHxI0IMGnM=\ngorm.io/driver/postgres v1.5.7/go.mod h1:3e019WlBaYI5o5LIdNV+LyxCMNtLOQETBXL2h4chKpA=\ngorm.io/driver/sqlite v1.5.1 h1:hYyrLkAWE71bcarJDPdZNTLWtr8XrSjOWyjUYI6xdL4=\ngorm.io/driver/sqlite v1.5.1/go.mod h1:7MZZ2Z8bqyfSQA1gYEV6MagQWj3cpUkJj9Z+d1HEMEQ=\ngorm.io/gorm v1.25.7/go.mod h1:hbnx/Oo0ChWMn1BIhpy1oYozzpM15i4YPuHDmfYtwg8=\ngorm.io/gorm v1.25.10 h1:dQpO+33KalOA+aFYGlK+EfxcI5MbO7EP2yYygwh9h+s=\ngorm.io/gorm v1.25.10/go.mod h1:hbnx/Oo0ChWMn1BIhpy1oYozzpM15i4YPuHDmfYtwg8=\nhonnef.co/go/tools v0.0.0-20190102054323-c2f93a96b099/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=\nhonnef.co/go/tools v0.0.0-20190523083050-ea95bdfd59fc/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4=\nnullprogram.com/x/optparse v1.0.0/go.mod h1:KdyPE+Igbe0jQUrVfMqDMeJQIJZEuyV7pjYmp6pbG50=\nrsc.io/pdf v0.1.1/go.mod h1:n8OzWcQ6Sp37PL01nO98y4iUCRdTGarVfzxY20ICaU4=\n"
},
{
"path": "main.go",
"content": "package main\n\nimport (\n\t\"embed\"\n\t\"fmt\"\n\t\"os\"\n\t\"strconv\"\n\n\t\"github.com/gin-contrib/sessions\"\n\t\"github.com/gin-contrib/sessions/cookie\"\n\t\"github.com/gin-gonic/gin\"\n\t_ \"github.com/joho/godotenv/autoload\"\n\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/client\"\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/i18n\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/controller\"\n\t\"github.com/songquanpeng/one-api/middleware\"\n\t\"github.com/songquanpeng/one-api/model\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/router\"\n)\n\n//go:embed web/build/*\nvar buildFS embed.FS\n\nfunc main() {\n\tcommon.Init()\n\tlogger.SetupLogger()\n\tlogger.SysLogf(\"One API %s started\", common.Version)\n\n\tif os.Getenv(\"GIN_MODE\") != gin.DebugMode {\n\t\tgin.SetMode(gin.ReleaseMode)\n\t}\n\tif config.DebugEnabled {\n\t\tlogger.SysLog(\"running in debug mode\")\n\t}\n\n\t// Initialize SQL Database\n\tmodel.InitDB()\n\tmodel.InitLogDB()\n\n\tvar err error\n\terr = model.CreateRootAccountIfNeed()\n\tif err != nil {\n\t\tlogger.FatalLog(\"database init error: \" + err.Error())\n\t}\n\tdefer func() {\n\t\terr := model.CloseDB()\n\t\tif err != nil {\n\t\t\tlogger.FatalLog(\"failed to close database: \" + err.Error())\n\t\t}\n\t}()\n\n\t// Initialize Redis\n\terr = common.InitRedisClient()\n\tif err != nil {\n\t\tlogger.FatalLog(\"failed to initialize Redis: \" + err.Error())\n\t}\n\n\t// Initialize options\n\tmodel.InitOptionMap()\n\tlogger.SysLog(fmt.Sprintf(\"using theme %s\", config.Theme))\n\tif common.RedisEnabled {\n\t\t// for compatibility with old versions\n\t\tconfig.MemoryCacheEnabled = true\n\t}\n\tif config.MemoryCacheEnabled {\n\t\tlogger.SysLog(\"memory cache enabled\")\n\t\tlogger.SysLog(fmt.Sprintf(\"sync frequency: %d seconds\", config.SyncFrequency))\n\t\tmodel.InitChannelCache()\n\t}\n\tif config.MemoryCacheEnabled {\n\t\tgo model.SyncOptions(config.SyncFrequency)\n\t\tgo model.SyncChannelCache(config.SyncFrequency)\n\t}\n\tif os.Getenv(\"CHANNEL_TEST_FREQUENCY\") != \"\" {\n\t\tfrequency, err := strconv.Atoi(os.Getenv(\"CHANNEL_TEST_FREQUENCY\"))\n\t\tif err != nil {\n\t\t\tlogger.FatalLog(\"failed to parse CHANNEL_TEST_FREQUENCY: \" + err.Error())\n\t\t}\n\t\tgo controller.AutomaticallyTestChannels(frequency)\n\t}\n\tif os.Getenv(\"BATCH_UPDATE_ENABLED\") == \"true\" {\n\t\tconfig.BatchUpdateEnabled = true\n\t\tlogger.SysLog(\"batch update enabled with interval \" + strconv.Itoa(config.BatchUpdateInterval) + \"s\")\n\t\tmodel.InitBatchUpdater()\n\t}\n\tif config.EnableMetric {\n\t\tlogger.SysLog(\"metric enabled, will disable channel if too much request failed\")\n\t}\n\topenai.InitTokenEncoders()\n\tclient.Init()\n\n\t// Initialize i18n\n\tif err := i18n.Init(); err != nil {\n\t\tlogger.FatalLog(\"failed to initialize i18n: \" + err.Error())\n\t}\n\n\t// Initialize HTTP server\n\tserver := gin.New()\n\tserver.Use(gin.Recovery())\n\t// This will cause SSE not to work!!!\n\t//server.Use(gzip.Gzip(gzip.DefaultCompression))\n\tserver.Use(middleware.RequestId())\n\tserver.Use(middleware.Language())\n\tmiddleware.SetUpLogger(server)\n\t// Initialize session store\n\tstore := cookie.NewStore([]byte(config.SessionSecret))\n\tserver.Use(sessions.Sessions(\"session\", store))\n\n\trouter.SetRouter(server, buildFS)\n\tvar port = os.Getenv(\"PORT\")\n\tif port == \"\" {\n\t\tport = strconv.Itoa(*common.Port)\n\t}\n\tlogger.SysLogf(\"server started on http://localhost:%s\", port)\n\terr = server.Run(\":\" + port)\n\tif err != nil {\n\t\tlogger.FatalLog(\"failed to start HTTP server: \" + err.Error())\n\t}\n}\n"
},
{
"path": "middleware/auth.go",
"content": "package middleware\n\nimport (\n\t\"fmt\"\n\t\"github.com/gin-contrib/sessions\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common/blacklist\"\n\t\"github.com/songquanpeng/one-api/common/ctxkey\"\n\t\"github.com/songquanpeng/one-api/common/network\"\n\t\"github.com/songquanpeng/one-api/model\"\n\t\"net/http\"\n\t\"strings\"\n)\n\nfunc authHelper(c *gin.Context, minRole int) {\n\tsession := sessions.Default(c)\n\tusername := session.Get(\"username\")\n\trole := session.Get(\"role\")\n\tid := session.Get(\"id\")\n\tstatus := session.Get(\"status\")\n\tif username == nil {\n\t\t// Check access token\n\t\taccessToken := c.Request.Header.Get(\"Authorization\")\n\t\tif accessToken == \"\" {\n\t\t\tc.JSON(http.StatusUnauthorized, gin.H{\n\t\t\t\t\"success\": false,\n\t\t\t\t\"message\": \"无权进行此操作，未登录且未提供 access token\",\n\t\t\t})\n\t\t\tc.Abort()\n\t\t\treturn\n\t\t}\n\t\tuser := model.ValidateAccessToken(accessToken)\n\t\tif user != nil && user.Username != \"\" {\n\t\t\t// Token is valid\n\t\t\tusername = user.Username\n\t\t\trole = user.Role\n\t\t\tid = user.Id\n\t\t\tstatus = user.Status\n\t\t} else {\n\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\"success\": false,\n\t\t\t\t\"message\": \"无权进行此操作，access token 无效\",\n\t\t\t})\n\t\t\tc.Abort()\n\t\t\treturn\n\t\t}\n\t}\n\tif status.(int) == model.UserStatusDisabled || blacklist.IsUserBanned(id.(int)) {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": \"用户已被封禁\",\n\t\t})\n\t\tsession := sessions.Default(c)\n\t\tsession.Clear()\n\t\t_ = session.Save()\n\t\tc.Abort()\n\t\treturn\n\t}\n\tif role.(int) < minRole {\n\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\"success\": false,\n\t\t\t\"message\": \"无权进行此操作，权限不足\",\n\t\t})\n\t\tc.Abort()\n\t\treturn\n\t}\n\tc.Set(\"username\", username)\n\tc.Set(\"role\", role)\n\tc.Set(\"id\", id)\n\tc.Next()\n}\n\nfunc UserAuth() func(c *gin.Context) {\n\treturn func(c *gin.Context) {\n\t\tauthHelper(c, model.RoleCommonUser)\n\t}\n}\n\nfunc AdminAuth() func(c *gin.Context) {\n\treturn func(c *gin.Context) {\n\t\tauthHelper(c, model.RoleAdminUser)\n\t}\n}\n\nfunc RootAuth() func(c *gin.Context) {\n\treturn func(c *gin.Context) {\n\t\tauthHelper(c, model.RoleRootUser)\n\t}\n}\n\nfunc TokenAuth() func(c *gin.Context) {\n\treturn func(c *gin.Context) {\n\t\tctx := c.Request.Context()\n\t\tkey := c.Request.Header.Get(\"Authorization\")\n\t\tkey = strings.TrimPrefix(key, \"Bearer \")\n\t\tkey = strings.TrimPrefix(key, \"sk-\")\n\t\tparts := strings.Split(key, \"-\")\n\t\tkey = parts[0]\n\t\ttoken, err := model.ValidateUserToken(key)\n\t\tif err != nil {\n\t\t\tabortWithMessage(c, http.StatusUnauthorized, err.Error())\n\t\t\treturn\n\t\t}\n\t\tif token.Subnet != nil && *token.Subnet != \"\" {\n\t\t\tif !network.IsIpInSubnets(ctx, c.ClientIP(), *token.Subnet) {\n\t\t\t\tabortWithMessage(c, http.StatusForbidden, fmt.Sprintf(\"该令牌只能在指定网段使用：%s，当前 ip：%s\", *token.Subnet, c.ClientIP()))\n\t\t\t\treturn\n\t\t\t}\n\t\t}\n\t\tuserEnabled, err := model.CacheIsUserEnabled(token.UserId)\n\t\tif err != nil {\n\t\t\tabortWithMessage(c, http.StatusInternalServerError, err.Error())\n\t\t\treturn\n\t\t}\n\t\tif !userEnabled || blacklist.IsUserBanned(token.UserId) {\n\t\t\tabortWithMessage(c, http.StatusForbidden, \"用户已被封禁\")\n\t\t\treturn\n\t\t}\n\t\trequestModel, err := getRequestModel(c)\n\t\tif err != nil && shouldCheckModel(c) {\n\t\t\tabortWithMessage(c, http.StatusBadRequest, err.Error())\n\t\t\treturn\n\t\t}\n\t\tc.Set(ctxkey.RequestModel, requestModel)\n\t\tif token.Models != nil && *token.Models != \"\" {\n\t\t\tc.Set(ctxkey.AvailableModels, *token.Models)\n\t\t\tif requestModel != \"\" && !isModelInList(requestModel, *token.Models) {\n\t\t\t\tabortWithMessage(c, http.StatusForbidden, fmt.Sprintf(\"该令牌无权使用模型：%s\", requestModel))\n\t\t\t\treturn\n\t\t\t}\n\t\t}\n\t\tc.Set(ctxkey.Id, token.UserId)\n\t\tc.Set(ctxkey.TokenId, token.Id)\n\t\tc.Set(ctxkey.TokenName, token.Name)\n\t\tif len(parts) > 1 {\n\t\t\tif model.IsAdmin(token.UserId) {\n\t\t\t\tc.Set(ctxkey.SpecificChannelId, parts[1])\n\t\t\t} else {\n\t\t\t\tabortWithMessage(c, http.StatusForbidden, \"普通用户不支持指定渠道\")\n\t\t\t\treturn\n\t\t\t}\n\t\t}\n\n\t\t// set channel id for proxy relay\n\t\tif channelId := c.Param(\"channelid\"); channelId != \"\" {\n\t\t\tc.Set(ctxkey.SpecificChannelId, channelId)\n\t\t}\n\n\t\tc.Next()\n\t}\n}\n\nfunc shouldCheckModel(c *gin.Context) bool {\n\tif strings.HasPrefix(c.Request.URL.Path, \"/v1/completions\") {\n\t\treturn true\n\t}\n\tif strings.HasPrefix(c.Request.URL.Path, \"/v1/chat/completions\") {\n\t\treturn true\n\t}\n\tif strings.HasPrefix(c.Request.URL.Path, \"/v1/images\") {\n\t\treturn true\n\t}\n\tif strings.HasPrefix(c.Request.URL.Path, \"/v1/audio\") {\n\t\treturn true\n\t}\n\treturn false\n}\n"
},
{
"path": "middleware/cache.go",
"content": "package middleware\n\nimport (\n\t\"github.com/gin-gonic/gin\"\n)\n\nfunc Cache() func(c *gin.Context) {\n\treturn func(c *gin.Context) {\n\t\tif c.Request.RequestURI == \"/\" {\n\t\t\tc.Header(\"Cache-Control\", \"no-cache\")\n\t\t} else {\n\t\t\tc.Header(\"Cache-Control\", \"max-age=604800\") // one week\n\t\t}\n\t\tc.Next()\n\t}\n}\n"
},
{
"path": "middleware/cors.go",
"content": "package middleware\n\nimport (\n\t\"github.com/gin-contrib/cors\"\n\t\"github.com/gin-gonic/gin\"\n)\n\nfunc CORS() gin.HandlerFunc {\n\tconfig := cors.DefaultConfig()\n\tconfig.AllowAllOrigins = true\n\tconfig.AllowCredentials = true\n\tconfig.AllowMethods = []string{\"GET\", \"POST\", \"PUT\", \"DELETE\", \"OPTIONS\"}\n\tconfig.AllowHeaders = []string{\"*\"}\n\treturn cors.New(config)\n}\n"
},
{
"path": "middleware/distributor.go",
"content": "package middleware\n\nimport (\n\t\"fmt\"\n\t\"net/http\"\n\t\"strconv\"\n\n\t\"github.com/gin-gonic/gin\"\n\n\t\"github.com/songquanpeng/one-api/common/ctxkey\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/model\"\n\t\"github.com/songquanpeng/one-api/relay/channeltype\"\n)\n\ntype ModelRequest struct {\n\tModel string `json:\"model\" form:\"model\"`\n}\n\nfunc Distribute() func(c *gin.Context) {\n\treturn func(c *gin.Context) {\n\t\tctx := c.Request.Context()\n\t\tuserId := c.GetInt(ctxkey.Id)\n\t\tuserGroup, _ := model.CacheGetUserGroup(userId)\n\t\tc.Set(ctxkey.Group, userGroup)\n\t\tvar requestModel string\n\t\tvar channel *model.Channel\n\t\tchannelId, ok := c.Get(ctxkey.SpecificChannelId)\n\t\tif ok {\n\t\t\tid, err := strconv.Atoi(channelId.(string))\n\t\t\tif err != nil {\n\t\t\t\tabortWithMessage(c, http.StatusBadRequest, \"无效的渠道 Id\")\n\t\t\t\treturn\n\t\t\t}\n\t\t\tchannel, err = model.GetChannelById(id, true)\n\t\t\tif err != nil {\n\t\t\t\tabortWithMessage(c, http.StatusBadRequest, \"无效的渠道 Id\")\n\t\t\t\treturn\n\t\t\t}\n\t\t\tif channel.Status != model.ChannelStatusEnabled {\n\t\t\t\tabortWithMessage(c, http.StatusForbidden, \"该渠道已被禁用\")\n\t\t\t\treturn\n\t\t\t}\n\t\t} else {\n\t\t\trequestModel = c.GetString(ctxkey.RequestModel)\n\t\t\tvar err error\n\t\t\tchannel, err = model.CacheGetRandomSatisfiedChannel(userGroup, requestModel, false)\n\t\t\tif err != nil {\n\t\t\t\tmessage := fmt.Sprintf(\"当前分组 %s 下对于模型 %s 无可用渠道\", userGroup, requestModel)\n\t\t\t\tif channel != nil {\n\t\t\t\t\tlogger.SysError(fmt.Sprintf(\"渠道不存在：%d\", channel.Id))\n\t\t\t\t\tmessage = \"数据库一致性已被破坏，请联系管理员\"\n\t\t\t\t}\n\t\t\t\tabortWithMessage(c, http.StatusServiceUnavailable, message)\n\t\t\t\treturn\n\t\t\t}\n\t\t}\n\t\tlogger.Debugf(ctx, \"user id %d, user group: %s, request model: %s, using channel #%d\", userId, userGroup, requestModel, channel.Id)\n\t\tSetupContextForSelectedChannel(c, channel, requestModel)\n\t\tc.Next()\n\t}\n}\n\nfunc SetupContextForSelectedChannel(c *gin.Context, channel *model.Channel, modelName string) {\n\tc.Set(ctxkey.Channel, channel.Type)\n\tc.Set(ctxkey.ChannelId, channel.Id)\n\tc.Set(ctxkey.ChannelName, channel.Name)\n\tif channel.SystemPrompt != nil && *channel.SystemPrompt != \"\" {\n\t\tc.Set(ctxkey.SystemPrompt, *channel.SystemPrompt)\n\t}\n\tc.Set(ctxkey.ModelMapping, channel.GetModelMapping())\n\tc.Set(ctxkey.OriginalModel, modelName) // for retry\n\tc.Request.Header.Set(\"Authorization\", fmt.Sprintf(\"Bearer %s\", channel.Key))\n\tc.Set(ctxkey.BaseURL, channel.GetBaseURL())\n\tcfg, _ := channel.LoadConfig()\n\t// this is for backward compatibility\n\tif channel.Other != nil {\n\t\tswitch channel.Type {\n\t\tcase channeltype.Azure:\n\t\t\tif cfg.APIVersion == \"\" {\n\t\t\t\tcfg.APIVersion = *channel.Other\n\t\t\t}\n\t\tcase channeltype.Xunfei:\n\t\t\tif cfg.APIVersion == \"\" {\n\t\t\t\tcfg.APIVersion = *channel.Other\n\t\t\t}\n\t\tcase channeltype.Gemini:\n\t\t\tif cfg.APIVersion == \"\" {\n\t\t\t\tcfg.APIVersion = *channel.Other\n\t\t\t}\n\t\tcase channeltype.AIProxyLibrary:\n\t\t\tif cfg.LibraryID == \"\" {\n\t\t\t\tcfg.LibraryID = *channel.Other\n\t\t\t}\n\t\tcase channeltype.Ali:\n\t\t\tif cfg.Plugin == \"\" {\n\t\t\t\tcfg.Plugin = *channel.Other\n\t\t\t}\n\t\t}\n\t}\n\tc.Set(ctxkey.Config, cfg)\n}\n"
},
{
"path": "middleware/gzip.go",
"content": "package middleware\n\nimport (\n\t\"compress/gzip\"\n\t\"github.com/gin-gonic/gin\"\n\t\"io\"\n\t\"net/http\"\n)\n\nfunc GzipDecodeMiddleware() gin.HandlerFunc {\n\treturn func(c *gin.Context) {\n\t\tif c.GetHeader(\"Content-Encoding\") == \"gzip\" {\n\t\t\tgzipReader, err := gzip.NewReader(c.Request.Body)\n\t\t\tif err != nil {\n\t\t\t\tc.AbortWithStatus(http.StatusBadRequest)\n\t\t\t\treturn\n\t\t\t}\n\t\t\tdefer gzipReader.Close()\n\n\t\t\t// Replace the request body with the decompressed data\n\t\t\tc.Request.Body = io.NopCloser(gzipReader)\n\t\t}\n\n\t\t// Continue processing the request\n\t\tc.Next()\n\t}\n}\n"
},
{
"path": "middleware/language.go",
"content": "package middleware\n\nimport (\n\t\"strings\"\n\n\t\"github.com/gin-gonic/gin\"\n\n\t\"github.com/songquanpeng/one-api/common/i18n\"\n)\n\nfunc Language() gin.HandlerFunc {\n\treturn func(c *gin.Context) {\n\t\tlang := c.GetHeader(\"Accept-Language\")\n\t\tif lang == \"\" {\n\t\t\tlang = \"en\"\n\t\t}\n\t\tif strings.HasPrefix(strings.ToLower(lang), \"zh\") {\n\t\t\tlang = \"zh-CN\"\n\t\t} else {\n\t\t\tlang = \"en\"\n\t\t}\n\t\tc.Set(i18n.ContextKey, lang)\n\t\tc.Next()\n\t}\n}\n"
},
{
"path": "middleware/logger.go",
"content": "package middleware\n\nimport (\n\t\"fmt\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n)\n\nfunc SetUpLogger(server *gin.Engine) {\n\tserver.Use(gin.LoggerWithFormatter(func(param gin.LogFormatterParams) string {\n\t\tvar requestID string\n\t\tif param.Keys != nil {\n\t\t\trequestID = param.Keys[helper.RequestIdKey].(string)\n\t\t}\n\t\treturn fmt.Sprintf(\"[GIN] %s | %s | %3d | %13v | %15s | %7s %s\\n\",\n\t\t\tparam.TimeStamp.Format(\"2006/01/02 - 15:04:05\"),\n\t\t\trequestID,\n\t\t\tparam.StatusCode,\n\t\t\tparam.Latency,\n\t\t\tparam.ClientIP,\n\t\t\tparam.Method,\n\t\t\tparam.Path,\n\t\t)\n\t}))\n}\n"
},
{
"path": "middleware/rate-limit.go",
"content": "package middleware\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\t\"net/http\"\n\t\"time\"\n\n\t\"github.com/gin-gonic/gin\"\n\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/config\"\n)\n\nvar timeFormat = \"2006-01-02T15:04:05.000Z\"\n\nvar inMemoryRateLimiter common.InMemoryRateLimiter\n\nfunc redisRateLimiter(c *gin.Context, maxRequestNum int, duration int64, mark string) {\n\tctx := context.Background()\n\trdb := common.RDB\n\tkey := \"rateLimit:\" + mark + c.ClientIP()\n\tlistLength, err := rdb.LLen(ctx, key).Result()\n\tif err != nil {\n\t\tfmt.Println(err.Error())\n\t\tc.Status(http.StatusInternalServerError)\n\t\tc.Abort()\n\t\treturn\n\t}\n\tif listLength < int64(maxRequestNum) {\n\t\trdb.LPush(ctx, key, time.Now().Format(timeFormat))\n\t\trdb.Expire(ctx, key, config.RateLimitKeyExpirationDuration)\n\t} else {\n\t\toldTimeStr, _ := rdb.LIndex(ctx, key, -1).Result()\n\t\toldTime, err := time.Parse(timeFormat, oldTimeStr)\n\t\tif err != nil {\n\t\t\tfmt.Println(err)\n\t\t\tc.Status(http.StatusInternalServerError)\n\t\t\tc.Abort()\n\t\t\treturn\n\t\t}\n\t\tnowTimeStr := time.Now().Format(timeFormat)\n\t\tnowTime, err := time.Parse(timeFormat, nowTimeStr)\n\t\tif err != nil {\n\t\t\tfmt.Println(err)\n\t\t\tc.Status(http.StatusInternalServerError)\n\t\t\tc.Abort()\n\t\t\treturn\n\t\t}\n\t\t// time.Since will return negative number!\n\t\t// See: https://stackoverflow.com/questions/50970900/why-is-time-since-returning-negative-durations-on-windows\n\t\tif int64(nowTime.Sub(oldTime).Seconds()) < duration {\n\t\t\trdb.Expire(ctx, key, config.RateLimitKeyExpirationDuration)\n\t\t\tc.Status(http.StatusTooManyRequests)\n\t\t\tc.Abort()\n\t\t\treturn\n\t\t} else {\n\t\t\trdb.LPush(ctx, key, time.Now().Format(timeFormat))\n\t\t\trdb.LTrim(ctx, key, 0, int64(maxRequestNum-1))\n\t\t\trdb.Expire(ctx, key, config.RateLimitKeyExpirationDuration)\n\t\t}\n\t}\n}\n\nfunc memoryRateLimiter(c *gin.Context, maxRequestNum int, duration int64, mark string) {\n\tkey := mark + c.ClientIP()\n\tif !inMemoryRateLimiter.Request(key, maxRequestNum, duration) {\n\t\tc.Status(http.StatusTooManyRequests)\n\t\tc.Abort()\n\t\treturn\n\t}\n}\n\nfunc rateLimitFactory(maxRequestNum int, duration int64, mark string) func(c *gin.Context) {\n\tif maxRequestNum == 0 || config.DebugEnabled {\n\t\treturn func(c *gin.Context) {\n\t\t\tc.Next()\n\t\t}\n\t}\n\tif common.RedisEnabled {\n\t\treturn func(c *gin.Context) {\n\t\t\tredisRateLimiter(c, maxRequestNum, duration, mark)\n\t\t}\n\t} else {\n\t\t// It's safe to call multi times.\n\t\tinMemoryRateLimiter.Init(config.RateLimitKeyExpirationDuration)\n\t\treturn func(c *gin.Context) {\n\t\t\tmemoryRateLimiter(c, maxRequestNum, duration, mark)\n\t\t}\n\t}\n}\n\nfunc GlobalWebRateLimit() func(c *gin.Context) {\n\treturn rateLimitFactory(config.GlobalWebRateLimitNum, config.GlobalWebRateLimitDuration, \"GW\")\n}\n\nfunc GlobalAPIRateLimit() func(c *gin.Context) {\n\treturn rateLimitFactory(config.GlobalApiRateLimitNum, config.GlobalApiRateLimitDuration, \"GA\")\n}\n\nfunc CriticalRateLimit() func(c *gin.Context) {\n\treturn rateLimitFactory(config.CriticalRateLimitNum, config.CriticalRateLimitDuration, \"CT\")\n}\n\nfunc DownloadRateLimit() func(c *gin.Context) {\n\treturn rateLimitFactory(config.DownloadRateLimitNum, config.DownloadRateLimitDuration, \"DW\")\n}\n\nfunc UploadRateLimit() func(c *gin.Context) {\n\treturn rateLimitFactory(config.UploadRateLimitNum, config.UploadRateLimitDuration, \"UP\")\n}\n"
},
{
"path": "middleware/recover.go",
"content": "package middleware\n\nimport (\n\t\"fmt\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"net/http\"\n\t\"runtime/debug\"\n)\n\nfunc RelayPanicRecover() gin.HandlerFunc {\n\treturn func(c *gin.Context) {\n\t\tdefer func() {\n\t\t\tif err := recover(); err != nil {\n\t\t\t\tctx := c.Request.Context()\n\t\t\t\tlogger.Errorf(ctx, fmt.Sprintf(\"panic detected: %v\", err))\n\t\t\t\tlogger.Errorf(ctx, fmt.Sprintf(\"stacktrace from panic: %s\", string(debug.Stack())))\n\t\t\t\tlogger.Errorf(ctx, fmt.Sprintf(\"request: %s %s\", c.Request.Method, c.Request.URL.Path))\n\t\t\t\tbody, _ := common.GetRequestBody(c)\n\t\t\t\tlogger.Errorf(ctx, fmt.Sprintf(\"request body: %s\", string(body)))\n\t\t\t\tc.JSON(http.StatusInternalServerError, gin.H{\n\t\t\t\t\t\"error\": gin.H{\n\t\t\t\t\t\t\"message\": fmt.Sprintf(\"Panic detected, error: %v. Please submit an issue with the related log here: https://github.com/songquanpeng/one-api\", err),\n\t\t\t\t\t\t\"type\": \"one_api_panic\",\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t\tc.Abort()\n\t\t\t}\n\t\t}()\n\t\tc.Next()\n\t}\n}\n"
},
{
"path": "middleware/request-id.go",
"content": "package middleware\n\nimport (\n\t\"github.com/gin-gonic/gin\"\n\n\t\"github.com/songquanpeng/one-api/common/helper\"\n)\n\nfunc RequestId() func(c *gin.Context) {\n\treturn func(c *gin.Context) {\n\t\tid := helper.GenRequestID()\n\t\tc.Set(helper.RequestIdKey, id)\n\t\tctx := helper.SetRequestID(c.Request.Context(), id)\n\t\tc.Request = c.Request.WithContext(ctx)\n\t\tc.Header(helper.RequestIdKey, id)\n\t\tc.Next()\n\t}\n}\n"
},
{
"path": "middleware/turnstile-check.go",
"content": "package middleware\n\nimport (\n\t\"encoding/json\"\n\t\"github.com/gin-contrib/sessions\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"net/http\"\n\t\"net/url\"\n)\n\ntype turnstileCheckResponse struct {\n\tSuccess bool `json:\"success\"`\n}\n\nfunc TurnstileCheck() gin.HandlerFunc {\n\treturn func(c *gin.Context) {\n\t\tif config.TurnstileCheckEnabled {\n\t\t\tsession := sessions.Default(c)\n\t\t\tturnstileChecked := session.Get(\"turnstile\")\n\t\t\tif turnstileChecked != nil {\n\t\t\t\tc.Next()\n\t\t\t\treturn\n\t\t\t}\n\t\t\tresponse := c.Query(\"turnstile\")\n\t\t\tif response == \"\" {\n\t\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\t\"success\": false,\n\t\t\t\t\t\"message\": \"Turnstile token 为空\",\n\t\t\t\t})\n\t\t\t\tc.Abort()\n\t\t\t\treturn\n\t\t\t}\n\t\t\trawRes, err := http.PostForm(\"https://challenges.cloudflare.com/turnstile/v0/siteverify\", url.Values{\n\t\t\t\t\"secret\": {config.TurnstileSecretKey},\n\t\t\t\t\"response\": {response},\n\t\t\t\t\"remoteip\": {c.ClientIP()},\n\t\t\t})\n\t\t\tif err != nil {\n\t\t\t\tlogger.SysError(err.Error())\n\t\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\t\"success\": false,\n\t\t\t\t\t\"message\": err.Error(),\n\t\t\t\t})\n\t\t\t\tc.Abort()\n\t\t\t\treturn\n\t\t\t}\n\t\t\tdefer rawRes.Body.Close()\n\t\t\tvar res turnstileCheckResponse\n\t\t\terr = json.NewDecoder(rawRes.Body).Decode(&res)\n\t\t\tif err != nil {\n\t\t\t\tlogger.SysError(err.Error())\n\t\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\t\"success\": false,\n\t\t\t\t\t\"message\": err.Error(),\n\t\t\t\t})\n\t\t\t\tc.Abort()\n\t\t\t\treturn\n\t\t\t}\n\t\t\tif !res.Success {\n\t\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\t\"success\": false,\n\t\t\t\t\t\"message\": \"Turnstile 校验失败，请刷新重试！\",\n\t\t\t\t})\n\t\t\t\tc.Abort()\n\t\t\t\treturn\n\t\t\t}\n\t\t\tsession.Set(\"turnstile\", true)\n\t\t\terr = session.Save()\n\t\t\tif err != nil {\n\t\t\t\tc.JSON(http.StatusOK, gin.H{\n\t\t\t\t\t\"message\": \"无法保存会话信息，请重试\",\n\t\t\t\t\t\"success\": false,\n\t\t\t\t})\n\t\t\t\treturn\n\t\t\t}\n\t\t}\n\t\tc.Next()\n\t}\n}\n"
},
{
"path": "middleware/utils.go",
"content": "package middleware\n\nimport (\n\t\"fmt\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"strings\"\n)\n\nfunc abortWithMessage(c *gin.Context, statusCode int, message string) {\n\tc.JSON(statusCode, gin.H{\n\t\t\"error\": gin.H{\n\t\t\t\"message\": helper.MessageWithRequestId(message, c.GetString(helper.RequestIdKey)),\n\t\t\t\"type\": \"one_api_error\",\n\t\t},\n\t})\n\tc.Abort()\n\tlogger.Error(c.Request.Context(), message)\n}\n\nfunc getRequestModel(c *gin.Context) (string, error) {\n\tvar modelRequest ModelRequest\n\terr := common.UnmarshalBodyReusable(c, &modelRequest)\n\tif err != nil {\n\t\treturn \"\", fmt.Errorf(\"common.UnmarshalBodyReusable failed: %w\", err)\n\t}\n\tif strings.HasPrefix(c.Request.URL.Path, \"/v1/moderations\") {\n\t\tif modelRequest.Model == \"\" {\n\t\t\tmodelRequest.Model = \"text-moderation-stable\"\n\t\t}\n\t}\n\tif strings.HasSuffix(c.Request.URL.Path, \"embeddings\") {\n\t\tif modelRequest.Model == \"\" {\n\t\t\tmodelRequest.Model = c.Param(\"model\")\n\t\t}\n\t}\n\tif strings.HasPrefix(c.Request.URL.Path, \"/v1/images/generations\") {\n\t\tif modelRequest.Model == \"\" {\n\t\t\tmodelRequest.Model = \"dall-e-2\"\n\t\t}\n\t}\n\tif strings.HasPrefix(c.Request.URL.Path, \"/v1/audio/transcriptions\") || strings.HasPrefix(c.Request.URL.Path, \"/v1/audio/translations\") {\n\t\tif modelRequest.Model == \"\" {\n\t\t\tmodelRequest.Model = \"whisper-1\"\n\t\t}\n\t}\n\treturn modelRequest.Model, nil\n}\n\nfunc isModelInList(modelName string, models string) bool {\n\tmodelList := strings.Split(models, \",\")\n\tfor _, model := range modelList {\n\t\tif modelName == model {\n\t\t\treturn true\n\t\t}\n\t}\n\treturn false\n}\n"
},
{
"path": "model/ability.go",
"content": "package model\n\nimport (\n\t\"context\"\n\t\"sort\"\n\t\"strings\"\n\n\t\"gorm.io/gorm\"\n\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/utils\"\n)\n\ntype Ability struct {\n\tGroup string `json:\"group\" gorm:\"type:varchar(32);primaryKey;autoIncrement:false\"`\n\tModel string `json:\"model\" gorm:\"primaryKey;autoIncrement:false\"`\n\tChannelId int `json:\"channel_id\" gorm:\"primaryKey;autoIncrement:false;index\"`\n\tEnabled bool `json:\"enabled\"`\n\tPriority *int64 `json:\"priority\" gorm:\"bigint;default:0;index\"`\n}\n\nfunc GetRandomSatisfiedChannel(group string, model string, ignoreFirstPriority bool) (*Channel, error) {\n\tability := Ability{}\n\tgroupCol := \"`group`\"\n\ttrueVal := \"1\"\n\tif common.UsingPostgreSQL {\n\t\tgroupCol = `\"group\"`\n\t\ttrueVal = \"true\"\n\t}\n\n\tvar err error = nil\n\tvar channelQuery *gorm.DB\n\tif ignoreFirstPriority {\n\t\tchannelQuery = DB.Where(groupCol+\" = ? and model = ? and enabled = \"+trueVal, group, model)\n\t} else {\n\t\tmaxPrioritySubQuery := DB.Model(&Ability{}).Select(\"MAX(priority)\").Where(groupCol+\" = ? and model = ? and enabled = \"+trueVal, group, model)\n\t\tchannelQuery = DB.Where(groupCol+\" = ? and model = ? and enabled = \"+trueVal+\" and priority = (?)\", group, model, maxPrioritySubQuery)\n\t}\n\tif common.UsingSQLite || common.UsingPostgreSQL {\n\t\terr = channelQuery.Order(\"RANDOM()\").First(&ability).Error\n\t} else {\n\t\terr = channelQuery.Order(\"RAND()\").First(&ability).Error\n\t}\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tchannel := Channel{}\n\tchannel.Id = ability.ChannelId\n\terr = DB.First(&channel, \"id = ?\", ability.ChannelId).Error\n\treturn &channel, err\n}\n\nfunc (channel *Channel) AddAbilities() error {\n\tmodels_ := strings.Split(channel.Models, \",\")\n\tmodels_ = utils.DeDuplication(models_)\n\tgroups_ := strings.Split(channel.Group, \",\")\n\tabilities := make([]Ability, 0, len(models_))\n\tfor _, model := range models_ {\n\t\tfor _, group := range groups_ {\n\t\t\tability := Ability{\n\t\t\t\tGroup: group,\n\t\t\t\tModel: model,\n\t\t\t\tChannelId: channel.Id,\n\t\t\t\tEnabled: channel.Status == ChannelStatusEnabled,\n\t\t\t\tPriority: channel.Priority,\n\t\t\t}\n\t\t\tabilities = append(abilities, ability)\n\t\t}\n\t}\n\treturn DB.Create(&abilities).Error\n}\n\nfunc (channel *Channel) DeleteAbilities() error {\n\treturn DB.Where(\"channel_id = ?\", channel.Id).Delete(&Ability{}).Error\n}\n\n// UpdateAbilities updates abilities of this channel.\n// Make sure the channel is completed before calling this function.\nfunc (channel *Channel) UpdateAbilities() error {\n\t// A quick and dirty way to update abilities\n\t// First delete all abilities of this channel\n\terr := channel.DeleteAbilities()\n\tif err != nil {\n\t\treturn err\n\t}\n\t// Then add new abilities\n\terr = channel.AddAbilities()\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc UpdateAbilityStatus(channelId int, status bool) error {\n\treturn DB.Model(&Ability{}).Where(\"channel_id = ?\", channelId).Select(\"enabled\").Update(\"enabled\", status).Error\n}\n\nfunc GetGroupModels(ctx context.Context, group string) ([]string, error) {\n\tgroupCol := \"`group`\"\n\ttrueVal := \"1\"\n\tif common.UsingPostgreSQL {\n\t\tgroupCol = `\"group\"`\n\t\ttrueVal = \"true\"\n\t}\n\tvar models []string\n\terr := DB.Model(&Ability{}).Distinct(\"model\").Where(groupCol+\" = ? and enabled = \"+trueVal, group).Pluck(\"model\", &models).Error\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tsort.Strings(models)\n\treturn models, err\n}\n"
},
{
"path": "model/cache.go",
"content": "package model\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/common/random\"\n\t\"math/rand\"\n\t\"sort\"\n\t\"strconv\"\n\t\"strings\"\n\t\"sync\"\n\t\"time\"\n)\n\nvar (\n\tTokenCacheSeconds = config.SyncFrequency\n\tUserId2GroupCacheSeconds = config.SyncFrequency\n\tUserId2QuotaCacheSeconds = config.SyncFrequency\n\tUserId2StatusCacheSeconds = config.SyncFrequency\n\tGroupModelsCacheSeconds = config.SyncFrequency\n)\n\nfunc CacheGetTokenByKey(key string) (*Token, error) {\n\tkeyCol := \"`key`\"\n\tif common.UsingPostgreSQL {\n\t\tkeyCol = `\"key\"`\n\t}\n\tvar token Token\n\tif !common.RedisEnabled {\n\t\terr := DB.Where(keyCol+\" = ?\", key).First(&token).Error\n\t\treturn &token, err\n\t}\n\ttokenObjectString, err := common.RedisGet(fmt.Sprintf(\"token:%s\", key))\n\tif err != nil {\n\t\terr := DB.Where(keyCol+\" = ?\", key).First(&token).Error\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\tjsonBytes, err := json.Marshal(token)\n\t\tif err != nil {\n\t\t\treturn nil, err\n\t\t}\n\t\terr = common.RedisSet(fmt.Sprintf(\"token:%s\", key), string(jsonBytes), time.Duration(TokenCacheSeconds)*time.Second)\n\t\tif err != nil {\n\t\t\tlogger.SysError(\"Redis set token error: \" + err.Error())\n\t\t}\n\t\treturn &token, nil\n\t}\n\terr = json.Unmarshal([]byte(tokenObjectString), &token)\n\treturn &token, err\n}\n\nfunc CacheGetUserGroup(id int) (group string, err error) {\n\tif !common.RedisEnabled {\n\t\treturn GetUserGroup(id)\n\t}\n\tgroup, err = common.RedisGet(fmt.Sprintf(\"user_group:%d\", id))\n\tif err != nil {\n\t\tgroup, err = GetUserGroup(id)\n\t\tif err != nil {\n\t\t\treturn \"\", err\n\t\t}\n\t\terr = common.RedisSet(fmt.Sprintf(\"user_group:%d\", id), group, time.Duration(UserId2GroupCacheSeconds)*time.Second)\n\t\tif err != nil {\n\t\t\tlogger.SysError(\"Redis set user group error: \" + err.Error())\n\t\t}\n\t}\n\treturn group, err\n}\n\nfunc fetchAndUpdateUserQuota(ctx context.Context, id int) (quota int64, err error) {\n\tquota, err = GetUserQuota(id)\n\tif err != nil {\n\t\treturn 0, err\n\t}\n\terr = common.RedisSet(fmt.Sprintf(\"user_quota:%d\", id), fmt.Sprintf(\"%d\", quota), time.Duration(UserId2QuotaCacheSeconds)*time.Second)\n\tif err != nil {\n\t\tlogger.Error(ctx, \"Redis set user quota error: \"+err.Error())\n\t}\n\treturn\n}\n\nfunc CacheGetUserQuota(ctx context.Context, id int) (quota int64, err error) {\n\tif !common.RedisEnabled {\n\t\treturn GetUserQuota(id)\n\t}\n\tquotaString, err := common.RedisGet(fmt.Sprintf(\"user_quota:%d\", id))\n\tif err != nil {\n\t\treturn fetchAndUpdateUserQuota(ctx, id)\n\t}\n\tquota, err = strconv.ParseInt(quotaString, 10, 64)\n\tif err != nil {\n\t\treturn 0, nil\n\t}\n\tif quota <= config.PreConsumedQuota { // when user's quota is less than pre-consumed quota, we need to fetch from db\n\t\tlogger.Infof(ctx, \"user %d's cached quota is too low: %d, refreshing from db\", quota, id)\n\t\treturn fetchAndUpdateUserQuota(ctx, id)\n\t}\n\treturn quota, nil\n}\n\nfunc CacheUpdateUserQuota(ctx context.Context, id int) error {\n\tif !common.RedisEnabled {\n\t\treturn nil\n\t}\n\tquota, err := CacheGetUserQuota(ctx, id)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = common.RedisSet(fmt.Sprintf(\"user_quota:%d\", id), fmt.Sprintf(\"%d\", quota), time.Duration(UserId2QuotaCacheSeconds)*time.Second)\n\treturn err\n}\n\nfunc CacheDecreaseUserQuota(id int, quota int64) error {\n\tif !common.RedisEnabled {\n\t\treturn nil\n\t}\n\terr := common.RedisDecrease(fmt.Sprintf(\"user_quota:%d\", id), int64(quota))\n\treturn err\n}\n\nfunc CacheIsUserEnabled(userId int) (bool, error) {\n\tif !common.RedisEnabled {\n\t\treturn IsUserEnabled(userId)\n\t}\n\tenabled, err := common.RedisGet(fmt.Sprintf(\"user_enabled:%d\", userId))\n\tif err == nil {\n\t\treturn enabled == \"1\", nil\n\t}\n\n\tuserEnabled, err := IsUserEnabled(userId)\n\tif err != nil {\n\t\treturn false, err\n\t}\n\tenabled = \"0\"\n\tif userEnabled {\n\t\tenabled = \"1\"\n\t}\n\terr = common.RedisSet(fmt.Sprintf(\"user_enabled:%d\", userId), enabled, time.Duration(UserId2StatusCacheSeconds)*time.Second)\n\tif err != nil {\n\t\tlogger.SysError(\"Redis set user enabled error: \" + err.Error())\n\t}\n\treturn userEnabled, err\n}\n\nfunc CacheGetGroupModels(ctx context.Context, group string) ([]string, error) {\n\tif !common.RedisEnabled {\n\t\treturn GetGroupModels(ctx, group)\n\t}\n\tmodelsStr, err := common.RedisGet(fmt.Sprintf(\"group_models:%s\", group))\n\tif err == nil {\n\t\treturn strings.Split(modelsStr, \",\"), nil\n\t}\n\tmodels, err := GetGroupModels(ctx, group)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\terr = common.RedisSet(fmt.Sprintf(\"group_models:%s\", group), strings.Join(models, \",\"), time.Duration(GroupModelsCacheSeconds)*time.Second)\n\tif err != nil {\n\t\tlogger.SysError(\"Redis set group models error: \" + err.Error())\n\t}\n\treturn models, nil\n}\n\nvar group2model2channels map[string]map[string][]*Channel\nvar channelSyncLock sync.RWMutex\n\nfunc InitChannelCache() {\n\tnewChannelId2channel := make(map[int]*Channel)\n\tvar channels []*Channel\n\tDB.Where(\"status = ?\", ChannelStatusEnabled).Find(&channels)\n\tfor _, channel := range channels {\n\t\tnewChannelId2channel[channel.Id] = channel\n\t}\n\tvar abilities []*Ability\n\tDB.Find(&abilities)\n\tgroups := make(map[string]bool)\n\tfor _, ability := range abilities {\n\t\tgroups[ability.Group] = true\n\t}\n\tnewGroup2model2channels := make(map[string]map[string][]*Channel)\n\tfor group := range groups {\n\t\tnewGroup2model2channels[group] = make(map[string][]*Channel)\n\t}\n\tfor _, channel := range channels {\n\t\tgroups := strings.Split(channel.Group, \",\")\n\t\tfor _, group := range groups {\n\t\t\tmodels := strings.Split(channel.Models, \",\")\n\t\t\tfor _, model := range models {\n\t\t\t\tif _, ok := newGroup2model2channels[group][model]; !ok {\n\t\t\t\t\tnewGroup2model2channels[group][model] = make([]*Channel, 0)\n\t\t\t\t}\n\t\t\t\tnewGroup2model2channels[group][model] = append(newGroup2model2channels[group][model], channel)\n\t\t\t}\n\t\t}\n\t}\n\n\t// sort by priority\n\tfor group, model2channels := range newGroup2model2channels {\n\t\tfor model, channels := range model2channels {\n\t\t\tsort.Slice(channels, func(i, j int) bool {\n\t\t\t\treturn channels[i].GetPriority() > channels[j].GetPriority()\n\t\t\t})\n\t\t\tnewGroup2model2channels[group][model] = channels\n\t\t}\n\t}\n\n\tchannelSyncLock.Lock()\n\tgroup2model2channels = newGroup2model2channels\n\tchannelSyncLock.Unlock()\n\tlogger.SysLog(\"channels synced from database\")\n}\n\nfunc SyncChannelCache(frequency int) {\n\tfor {\n\t\ttime.Sleep(time.Duration(frequency) * time.Second)\n\t\tlogger.SysLog(\"syncing channels from database\")\n\t\tInitChannelCache()\n\t}\n}\n\nfunc CacheGetRandomSatisfiedChannel(group string, model string, ignoreFirstPriority bool) (*Channel, error) {\n\tif !config.MemoryCacheEnabled {\n\t\treturn GetRandomSatisfiedChannel(group, model, ignoreFirstPriority)\n\t}\n\tchannelSyncLock.RLock()\n\tdefer channelSyncLock.RUnlock()\n\tchannels := group2model2channels[group][model]\n\tif len(channels) == 0 {\n\t\treturn nil, errors.New(\"channel not found\")\n\t}\n\tendIdx := len(channels)\n\t// choose by priority\n\tfirstChannel := channels[0]\n\tif firstChannel.GetPriority() > 0 {\n\t\tfor i := range channels {\n\t\t\tif channels[i].GetPriority() != firstChannel.GetPriority() {\n\t\t\t\tendIdx = i\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t}\n\tidx := rand.Intn(endIdx)\n\tif ignoreFirstPriority {\n\t\tif endIdx < len(channels) { // which means there are more than one priority\n\t\t\tidx = random.RandRange(endIdx, len(channels))\n\t\t}\n\t}\n\treturn channels[idx], nil\n}\n"
},
{
"path": "model/channel.go",
"content": "package model\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"gorm.io/gorm\"\n)\n\nconst (\n\tChannelStatusUnknown = 0\n\tChannelStatusEnabled = 1 // don't use 0, 0 is the default value!\n\tChannelStatusManuallyDisabled = 2 // also don't use 0\n\tChannelStatusAutoDisabled = 3\n)\n\ntype Channel struct {\n\tId int `json:\"id\"`\n\tType int `json:\"type\" gorm:\"default:0\"`\n\tKey string `json:\"key\" gorm:\"type:text\"`\n\tStatus int `json:\"status\" gorm:\"default:1\"`\n\tName string `json:\"name\" gorm:\"index\"`\n\tWeight *uint `json:\"weight\" gorm:\"default:0\"`\n\tCreatedTime int64 `json:\"created_time\" gorm:\"bigint\"`\n\tTestTime int64 `json:\"test_time\" gorm:\"bigint\"`\n\tResponseTime int `json:\"response_time\"` // in milliseconds\n\tBaseURL *string `json:\"base_url\" gorm:\"column:base_url;default:''\"`\n\tOther *string `json:\"other\"` // DEPRECATED: please save config to field Config\n\tBalance float64 `json:\"balance\"` // in USD\n\tBalanceUpdatedTime int64 `json:\"balance_updated_time\" gorm:\"bigint\"`\n\tModels string `json:\"models\"`\n\tGroup string `json:\"group\" gorm:\"type:varchar(32);default:'default'\"`\n\tUsedQuota int64 `json:\"used_quota\" gorm:\"bigint;default:0\"`\n\tModelMapping *string `json:\"model_mapping\" gorm:\"type:varchar(1024);default:''\"`\n\tPriority *int64 `json:\"priority\" gorm:\"bigint;default:0\"`\n\tConfig string `json:\"config\"`\n\tSystemPrompt *string `json:\"system_prompt\" gorm:\"type:text\"`\n}\n\ntype ChannelConfig struct {\n\tRegion string `json:\"region,omitempty\"`\n\tSK string `json:\"sk,omitempty\"`\n\tAK string `json:\"ak,omitempty\"`\n\tUserID string `json:\"user_id,omitempty\"`\n\tAPIVersion string `json:\"api_version,omitempty\"`\n\tLibraryID string `json:\"library_id,omitempty\"`\n\tPlugin string `json:\"plugin,omitempty\"`\n\tVertexAIProjectID string `json:\"vertex_ai_project_id,omitempty\"`\n\tVertexAIADC string `json:\"vertex_ai_adc,omitempty\"`\n}\n\nfunc GetAllChannels(startIdx int, num int, scope string) ([]*Channel, error) {\n\tvar channels []*Channel\n\tvar err error\n\tswitch scope {\n\tcase \"all\":\n\t\terr = DB.Order(\"id desc\").Find(&channels).Error\n\tcase \"disabled\":\n\t\terr = DB.Order(\"id desc\").Where(\"status = ? or status = ?\", ChannelStatusAutoDisabled, ChannelStatusManuallyDisabled).Find(&channels).Error\n\tdefault:\n\t\terr = DB.Order(\"id desc\").Limit(num).Offset(startIdx).Omit(\"key\").Find(&channels).Error\n\t}\n\treturn channels, err\n}\n\nfunc SearchChannels(keyword string) (channels []*Channel, err error) {\n\terr = DB.Omit(\"key\").Where(\"id = ? or name LIKE ?\", helper.String2Int(keyword), keyword+\"%\").Find(&channels).Error\n\treturn channels, err\n}\n\nfunc GetChannelById(id int, selectAll bool) (*Channel, error) {\n\tchannel := Channel{Id: id}\n\tvar err error = nil\n\tif selectAll {\n\t\terr = DB.First(&channel, \"id = ?\", id).Error\n\t} else {\n\t\terr = DB.Omit(\"key\").First(&channel, \"id = ?\", id).Error\n\t}\n\treturn &channel, err\n}\n\nfunc BatchInsertChannels(channels []Channel) error {\n\tvar err error\n\terr = DB.Create(&channels).Error\n\tif err != nil {\n\t\treturn err\n\t}\n\tfor _, channel_ := range channels {\n\t\terr = channel_.AddAbilities()\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc (channel *Channel) GetPriority() int64 {\n\tif channel.Priority == nil {\n\t\treturn 0\n\t}\n\treturn *channel.Priority\n}\n\nfunc (channel *Channel) GetBaseURL() string {\n\tif channel.BaseURL == nil {\n\t\treturn \"\"\n\t}\n\treturn *channel.BaseURL\n}\n\nfunc (channel *Channel) GetModelMapping() map[string]string {\n\tif channel.ModelMapping == nil || *channel.ModelMapping == \"\" || *channel.ModelMapping == \"{}\" {\n\t\treturn nil\n\t}\n\tmodelMapping := make(map[string]string)\n\terr := json.Unmarshal([]byte(*channel.ModelMapping), &modelMapping)\n\tif err != nil {\n\t\tlogger.SysError(fmt.Sprintf(\"failed to unmarshal model mapping for channel %d, error: %s\", channel.Id, err.Error()))\n\t\treturn nil\n\t}\n\treturn modelMapping\n}\n\nfunc (channel *Channel) Insert() error {\n\tvar err error\n\terr = DB.Create(channel).Error\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = channel.AddAbilities()\n\treturn err\n}\n\nfunc (channel *Channel) Update() error {\n\tvar err error\n\terr = DB.Model(channel).Updates(channel).Error\n\tif err != nil {\n\t\treturn err\n\t}\n\tDB.Model(channel).First(channel, \"id = ?\", channel.Id)\n\terr = channel.UpdateAbilities()\n\treturn err\n}\n\nfunc (channel *Channel) UpdateResponseTime(responseTime int64) {\n\terr := DB.Model(channel).Select(\"response_time\", \"test_time\").Updates(Channel{\n\t\tTestTime: helper.GetTimestamp(),\n\t\tResponseTime: int(responseTime),\n\t}).Error\n\tif err != nil {\n\t\tlogger.SysError(\"failed to update response time: \" + err.Error())\n\t}\n}\n\nfunc (channel *Channel) UpdateBalance(balance float64) {\n\terr := DB.Model(channel).Select(\"balance_updated_time\", \"balance\").Updates(Channel{\n\t\tBalanceUpdatedTime: helper.GetTimestamp(),\n\t\tBalance: balance,\n\t}).Error\n\tif err != nil {\n\t\tlogger.SysError(\"failed to update balance: \" + err.Error())\n\t}\n}\n\nfunc (channel *Channel) Delete() error {\n\tvar err error\n\terr = DB.Delete(channel).Error\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = channel.DeleteAbilities()\n\treturn err\n}\n\nfunc (channel *Channel) LoadConfig() (ChannelConfig, error) {\n\tvar cfg ChannelConfig\n\tif channel.Config == \"\" {\n\t\treturn cfg, nil\n\t}\n\terr := json.Unmarshal([]byte(channel.Config), &cfg)\n\tif err != nil {\n\t\treturn cfg, err\n\t}\n\treturn cfg, nil\n}\n\nfunc UpdateChannelStatusById(id int, status int) {\n\terr := UpdateAbilityStatus(id, status == ChannelStatusEnabled)\n\tif err != nil {\n\t\tlogger.SysError(\"failed to update ability status: \" + err.Error())\n\t}\n\terr = DB.Model(&Channel{}).Where(\"id = ?\", id).Update(\"status\", status).Error\n\tif err != nil {\n\t\tlogger.SysError(\"failed to update channel status: \" + err.Error())\n\t}\n}\n\nfunc UpdateChannelUsedQuota(id int, quota int64) {\n\tif config.BatchUpdateEnabled {\n\t\taddNewRecord(BatchUpdateTypeChannelUsedQuota, id, quota)\n\t\treturn\n\t}\n\tupdateChannelUsedQuota(id, quota)\n}\n\nfunc updateChannelUsedQuota(id int, quota int64) {\n\terr := DB.Model(&Channel{}).Where(\"id = ?\", id).Update(\"used_quota\", gorm.Expr(\"used_quota + ?\", quota)).Error\n\tif err != nil {\n\t\tlogger.SysError(\"failed to update channel used quota: \" + err.Error())\n\t}\n}\n\nfunc DeleteChannelByStatus(status int64) (int64, error) {\n\tresult := DB.Where(\"status = ?\", status).Delete(&Channel{})\n\treturn result.RowsAffected, result.Error\n}\n\nfunc DeleteDisabledChannel() (int64, error) {\n\tresult := DB.Where(\"status = ? or status = ?\", ChannelStatusAutoDisabled, ChannelStatusManuallyDisabled).Delete(&Channel{})\n\treturn result.RowsAffected, result.Error\n}\n"
},
{
"path": "model/log.go",
"content": "package model\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"gorm.io/gorm\"\n\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n)\n\ntype Log struct {\n\tId int `json:\"id\"`\n\tUserId int `json:\"user_id\" gorm:\"index\"`\n\tCreatedAt int64 `json:\"created_at\" gorm:\"bigint;index:idx_created_at_type\"`\n\tType int `json:\"type\" gorm:\"index:idx_created_at_type\"`\n\tContent string `json:\"content\"`\n\tUsername string `json:\"username\" gorm:\"index:index_username_model_name,priority:2;default:''\"`\n\tTokenName string `json:\"token_name\" gorm:\"index;default:''\"`\n\tModelName string `json:\"model_name\" gorm:\"index;index:index_username_model_name,priority:1;default:''\"`\n\tQuota int `json:\"quota\" gorm:\"default:0\"`\n\tPromptTokens int `json:\"prompt_tokens\" gorm:\"default:0\"`\n\tCompletionTokens int `json:\"completion_tokens\" gorm:\"default:0\"`\n\tChannelId int `json:\"channel\" gorm:\"index\"`\n\tRequestId string `json:\"request_id\" gorm:\"default:''\"`\n\tElapsedTime int64 `json:\"elapsed_time\" gorm:\"default:0\"` // unit is ms\n\tIsStream bool `json:\"is_stream\" gorm:\"default:false\"`\n\tSystemPromptReset bool `json:\"system_prompt_reset\" gorm:\"default:false\"`\n}\n\nconst (\n\tLogTypeUnknown = iota\n\tLogTypeTopup\n\tLogTypeConsume\n\tLogTypeManage\n\tLogTypeSystem\n\tLogTypeTest\n)\n\nfunc recordLogHelper(ctx context.Context, log *Log) {\n\trequestId := helper.GetRequestID(ctx)\n\tlog.RequestId = requestId\n\terr := LOG_DB.Create(log).Error\n\tif err != nil {\n\t\tlogger.Error(ctx, \"failed to record log: \"+err.Error())\n\t\treturn\n\t}\n\tlogger.Infof(ctx, \"record log: %+v\", log)\n}\n\nfunc RecordLog(ctx context.Context, userId int, logType int, content string) {\n\tif logType == LogTypeConsume && !config.LogConsumeEnabled {\n\t\treturn\n\t}\n\tlog := &Log{\n\t\tUserId: userId,\n\t\tUsername: GetUsernameById(userId),\n\t\tCreatedAt: helper.GetTimestamp(),\n\t\tType: logType,\n\t\tContent: content,\n\t}\n\trecordLogHelper(ctx, log)\n}\n\nfunc RecordTopupLog(ctx context.Context, userId int, content string, quota int) {\n\tlog := &Log{\n\t\tUserId: userId,\n\t\tUsername: GetUsernameById(userId),\n\t\tCreatedAt: helper.GetTimestamp(),\n\t\tType: LogTypeTopup,\n\t\tContent: content,\n\t\tQuota: quota,\n\t}\n\trecordLogHelper(ctx, log)\n}\n\nfunc RecordConsumeLog(ctx context.Context, log *Log) {\n\tif !config.LogConsumeEnabled {\n\t\treturn\n\t}\n\tlog.Username = GetUsernameById(log.UserId)\n\tlog.CreatedAt = helper.GetTimestamp()\n\tlog.Type = LogTypeConsume\n\trecordLogHelper(ctx, log)\n}\n\nfunc RecordTestLog(ctx context.Context, log *Log) {\n\tlog.CreatedAt = helper.GetTimestamp()\n\tlog.Type = LogTypeTest\n\trecordLogHelper(ctx, log)\n}\n\nfunc GetAllLogs(logType int, startTimestamp int64, endTimestamp int64, modelName string, username string, tokenName string, startIdx int, num int, channel int) (logs []*Log, err error) {\n\tvar tx *gorm.DB\n\tif logType == LogTypeUnknown {\n\t\ttx = LOG_DB\n\t} else {\n\t\ttx = LOG_DB.Where(\"type = ?\", logType)\n\t}\n\tif modelName != \"\" {\n\t\ttx = tx.Where(\"model_name = ?\", modelName)\n\t}\n\tif username != \"\" {\n\t\ttx = tx.Where(\"username = ?\", username)\n\t}\n\tif tokenName != \"\" {\n\t\ttx = tx.Where(\"token_name = ?\", tokenName)\n\t}\n\tif startTimestamp != 0 {\n\t\ttx = tx.Where(\"created_at >= ?\", startTimestamp)\n\t}\n\tif endTimestamp != 0 {\n\t\ttx = tx.Where(\"created_at <= ?\", endTimestamp)\n\t}\n\tif channel != 0 {\n\t\ttx = tx.Where(\"channel_id = ?\", channel)\n\t}\n\terr = tx.Order(\"id desc\").Limit(num).Offset(startIdx).Find(&logs).Error\n\treturn logs, err\n}\n\nfunc GetUserLogs(userId int, logType int, startTimestamp int64, endTimestamp int64, modelName string, tokenName string, startIdx int, num int) (logs []*Log, err error) {\n\tvar tx *gorm.DB\n\tif logType == LogTypeUnknown {\n\t\ttx = LOG_DB.Where(\"user_id = ?\", userId)\n\t} else {\n\t\ttx = LOG_DB.Where(\"user_id = ? and type = ?\", userId, logType)\n\t}\n\tif modelName != \"\" {\n\t\ttx = tx.Where(\"model_name = ?\", modelName)\n\t}\n\tif tokenName != \"\" {\n\t\ttx = tx.Where(\"token_name = ?\", tokenName)\n\t}\n\tif startTimestamp != 0 {\n\t\ttx = tx.Where(\"created_at >= ?\", startTimestamp)\n\t}\n\tif endTimestamp != 0 {\n\t\ttx = tx.Where(\"created_at <= ?\", endTimestamp)\n\t}\n\terr = tx.Order(\"id desc\").Limit(num).Offset(startIdx).Omit(\"id\").Find(&logs).Error\n\treturn logs, err\n}\n\nfunc SearchAllLogs(keyword string) (logs []*Log, err error) {\n\terr = LOG_DB.Where(\"type = ? or content LIKE ?\", keyword, keyword+\"%\").Order(\"id desc\").Limit(config.MaxRecentItems).Find(&logs).Error\n\treturn logs, err\n}\n\nfunc SearchUserLogs(userId int, keyword string) (logs []*Log, err error) {\n\terr = LOG_DB.Where(\"user_id = ? and type = ?\", userId, keyword).Order(\"id desc\").Limit(config.MaxRecentItems).Omit(\"id\").Find(&logs).Error\n\treturn logs, err\n}\n\nfunc SumUsedQuota(logType int, startTimestamp int64, endTimestamp int64, modelName string, username string, tokenName string, channel int) (quota int64) {\n\tifnull := \"ifnull\"\n\tif common.UsingPostgreSQL {\n\t\tifnull = \"COALESCE\"\n\t}\n\ttx := LOG_DB.Table(\"logs\").Select(fmt.Sprintf(\"%s(sum(quota),0)\", ifnull))\n\tif username != \"\" {\n\t\ttx = tx.Where(\"username = ?\", username)\n\t}\n\tif tokenName != \"\" {\n\t\ttx = tx.Where(\"token_name = ?\", tokenName)\n\t}\n\tif startTimestamp != 0 {\n\t\ttx = tx.Where(\"created_at >= ?\", startTimestamp)\n\t}\n\tif endTimestamp != 0 {\n\t\ttx = tx.Where(\"created_at <= ?\", endTimestamp)\n\t}\n\tif modelName != \"\" {\n\t\ttx = tx.Where(\"model_name = ?\", modelName)\n\t}\n\tif channel != 0 {\n\t\ttx = tx.Where(\"channel_id = ?\", channel)\n\t}\n\ttx.Where(\"type = ?\", LogTypeConsume).Scan("a)\n\treturn quota\n}\n\nfunc SumUsedToken(logType int, startTimestamp int64, endTimestamp int64, modelName string, username string, tokenName string) (token int) {\n\tifnull := \"ifnull\"\n\tif common.UsingPostgreSQL {\n\t\tifnull = \"COALESCE\"\n\t}\n\ttx := LOG_DB.Table(\"logs\").Select(fmt.Sprintf(\"%s(sum(prompt_tokens),0) + %s(sum(completion_tokens),0)\", ifnull, ifnull))\n\tif username != \"\" {\n\t\ttx = tx.Where(\"username = ?\", username)\n\t}\n\tif tokenName != \"\" {\n\t\ttx = tx.Where(\"token_name = ?\", tokenName)\n\t}\n\tif startTimestamp != 0 {\n\t\ttx = tx.Where(\"created_at >= ?\", startTimestamp)\n\t}\n\tif endTimestamp != 0 {\n\t\ttx = tx.Where(\"created_at <= ?\", endTimestamp)\n\t}\n\tif modelName != \"\" {\n\t\ttx = tx.Where(\"model_name = ?\", modelName)\n\t}\n\ttx.Where(\"type = ?\", LogTypeConsume).Scan(&token)\n\treturn token\n}\n\nfunc DeleteOldLog(targetTimestamp int64) (int64, error) {\n\tresult := LOG_DB.Where(\"created_at < ?\", targetTimestamp).Delete(&Log{})\n\treturn result.RowsAffected, result.Error\n}\n\ntype LogStatistic struct {\n\tDay string `gorm:\"column:day\"`\n\tModelName string `gorm:\"column:model_name\"`\n\tRequestCount int `gorm:\"column:request_count\"`\n\tQuota int `gorm:\"column:quota\"`\n\tPromptTokens int `gorm:\"column:prompt_tokens\"`\n\tCompletionTokens int `gorm:\"column:completion_tokens\"`\n}\n\nfunc SearchLogsByDayAndModel(userId, start, end int) (LogStatistics []*LogStatistic, err error) {\n\tgroupSelect := \"DATE_FORMAT(FROM_UNIXTIME(created_at), '%Y-%m-%d') as day\"\n\n\tif common.UsingPostgreSQL {\n\t\tgroupSelect = \"TO_CHAR(date_trunc('day', to_timestamp(created_at)), 'YYYY-MM-DD') as day\"\n\t}\n\n\tif common.UsingSQLite {\n\t\tgroupSelect = \"strftime('%Y-%m-%d', datetime(created_at, 'unixepoch')) as day\"\n\t}\n\n\terr = LOG_DB.Raw(`\n\t\tSELECT `+groupSelect+`,\n\t\tmodel_name, count(1) as request_count,\n\t\tsum(quota) as quota,\n\t\tsum(prompt_tokens) as prompt_tokens,\n\t\tsum(completion_tokens) as completion_tokens\n\t\tFROM logs\n\t\tWHERE type=2\n\t\tAND user_id= ?\n\t\tAND created_at BETWEEN ? AND ?\n\t\tGROUP BY day, model_name\n\t\tORDER BY day, model_name\n\t`, userId, start, end).Scan(&LogStatistics).Error\n\n\treturn LogStatistics, err\n}\n"
},
{
"path": "model/main.go",
"content": "package model\n\nimport (\n\t\"database/sql\"\n\t\"fmt\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/env\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/common/random\"\n\t\"gorm.io/driver/mysql\"\n\t\"gorm.io/driver/postgres\"\n\t\"gorm.io/driver/sqlite\"\n\t\"gorm.io/gorm\"\n\t\"os\"\n\t\"strings\"\n\t\"time\"\n)\n\nvar DB *gorm.DB\nvar LOG_DB *gorm.DB\n\nfunc CreateRootAccountIfNeed() error {\n\tvar user User\n\t//if user.Status != util.UserStatusEnabled {\n\tif err := DB.First(&user).Error; err != nil {\n\t\tlogger.SysLog(\"no user exists, creating a root user for you: username is root, password is 123456\")\n\t\thashedPassword, err := common.Password2Hash(\"123456\")\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\taccessToken := random.GetUUID()\n\t\tif config.InitialRootAccessToken != \"\" {\n\t\t\taccessToken = config.InitialRootAccessToken\n\t\t}\n\t\trootUser := User{\n\t\t\tUsername: \"root\",\n\t\t\tPassword: hashedPassword,\n\t\t\tRole: RoleRootUser,\n\t\t\tStatus: UserStatusEnabled,\n\t\t\tDisplayName: \"Root User\",\n\t\t\tAccessToken: accessToken,\n\t\t\tQuota: 500000000000000,\n\t\t}\n\t\tDB.Create(&rootUser)\n\t\tif config.InitialRootToken != \"\" {\n\t\t\tlogger.SysLog(\"creating initial root token as requested\")\n\t\t\ttoken := Token{\n\t\t\t\tId: 1,\n\t\t\t\tUserId: rootUser.Id,\n\t\t\t\tKey: config.InitialRootToken,\n\t\t\t\tStatus: TokenStatusEnabled,\n\t\t\t\tName: \"Initial Root Token\",\n\t\t\t\tCreatedTime: helper.GetTimestamp(),\n\t\t\t\tAccessedTime: helper.GetTimestamp(),\n\t\t\t\tExpiredTime: -1,\n\t\t\t\tRemainQuota: 500000000000000,\n\t\t\t\tUnlimitedQuota: true,\n\t\t\t}\n\t\t\tDB.Create(&token)\n\t\t}\n\t}\n\treturn nil\n}\n\nfunc chooseDB(envName string) (*gorm.DB, error) {\n\tdsn := os.Getenv(envName)\n\n\tswitch {\n\tcase strings.HasPrefix(dsn, \"postgres://\"):\n\t\t// Use PostgreSQL\n\t\treturn openPostgreSQL(dsn)\n\tcase dsn != \"\":\n\t\t// Use MySQL\n\t\treturn openMySQL(dsn)\n\tdefault:\n\t\t// Use SQLite\n\t\treturn openSQLite()\n\t}\n}\n\nfunc openPostgreSQL(dsn string) (*gorm.DB, error) {\n\tlogger.SysLog(\"using PostgreSQL as database\")\n\tcommon.UsingPostgreSQL = true\n\treturn gorm.Open(postgres.New(postgres.Config{\n\t\tDSN: dsn,\n\t\tPreferSimpleProtocol: true, // disables implicit prepared statement usage\n\t}), &gorm.Config{\n\t\tPrepareStmt: true, // precompile SQL\n\t})\n}\n\nfunc openMySQL(dsn string) (*gorm.DB, error) {\n\tlogger.SysLog(\"using MySQL as database\")\n\tcommon.UsingMySQL = true\n\treturn gorm.Open(mysql.Open(dsn), &gorm.Config{\n\t\tPrepareStmt: true, // precompile SQL\n\t})\n}\n\nfunc openSQLite() (*gorm.DB, error) {\n\tlogger.SysLog(\"SQL_DSN not set, using SQLite as database\")\n\tcommon.UsingSQLite = true\n\tdsn := fmt.Sprintf(\"%s?_busy_timeout=%d\", common.SQLitePath, common.SQLiteBusyTimeout)\n\treturn gorm.Open(sqlite.Open(dsn), &gorm.Config{\n\t\tPrepareStmt: true, // precompile SQL\n\t})\n}\n\nfunc InitDB() {\n\tvar err error\n\tDB, err = chooseDB(\"SQL_DSN\")\n\tif err != nil {\n\t\tlogger.FatalLog(\"failed to initialize database: \" + err.Error())\n\t\treturn\n\t}\n\n\tsqlDB := setDBConns(DB)\n\n\tif !config.IsMasterNode {\n\t\treturn\n\t}\n\n\tif common.UsingMySQL {\n\t\t_, _ = sqlDB.Exec(\"DROP INDEX idx_channels_key ON channels;\") // TODO: delete this line when most users have upgraded\n\t}\n\n\tlogger.SysLog(\"database migration started\")\n\tif err = migrateDB(); err != nil {\n\t\tlogger.FatalLog(\"failed to migrate database: \" + err.Error())\n\t\treturn\n\t}\n\tlogger.SysLog(\"database migrated\")\n}\n\nfunc migrateDB() error {\n\tvar err error\n\tif err = DB.AutoMigrate(&Channel{}); err != nil {\n\t\treturn err\n\t}\n\tif err = DB.AutoMigrate(&Token{}); err != nil {\n\t\treturn err\n\t}\n\tif err = DB.AutoMigrate(&User{}); err != nil {\n\t\treturn err\n\t}\n\tif err = DB.AutoMigrate(&Option{}); err != nil {\n\t\treturn err\n\t}\n\tif err = DB.AutoMigrate(&Redemption{}); err != nil {\n\t\treturn err\n\t}\n\tif err = DB.AutoMigrate(&Ability{}); err != nil {\n\t\treturn err\n\t}\n\tif err = DB.AutoMigrate(&Log{}); err != nil {\n\t\treturn err\n\t}\n\tif err = DB.AutoMigrate(&Channel{}); err != nil {\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc InitLogDB() {\n\tif os.Getenv(\"LOG_SQL_DSN\") == \"\" {\n\t\tLOG_DB = DB\n\t\treturn\n\t}\n\n\tlogger.SysLog(\"using secondary database for table logs\")\n\tvar err error\n\tLOG_DB, err = chooseDB(\"LOG_SQL_DSN\")\n\tif err != nil {\n\t\tlogger.FatalLog(\"failed to initialize secondary database: \" + err.Error())\n\t\treturn\n\t}\n\n\tsetDBConns(LOG_DB)\n\n\tif !config.IsMasterNode {\n\t\treturn\n\t}\n\n\tlogger.SysLog(\"secondary database migration started\")\n\terr = migrateLOGDB()\n\tif err != nil {\n\t\tlogger.FatalLog(\"failed to migrate secondary database: \" + err.Error())\n\t\treturn\n\t}\n\tlogger.SysLog(\"secondary database migrated\")\n}\n\nfunc migrateLOGDB() error {\n\tvar err error\n\tif err = LOG_DB.AutoMigrate(&Log{}); err != nil {\n\t\treturn err\n\t}\n\treturn nil\n}\n\nfunc setDBConns(db *gorm.DB) *sql.DB {\n\tif config.DebugSQLEnabled {\n\t\tdb = db.Debug()\n\t}\n\n\tsqlDB, err := db.DB()\n\tif err != nil {\n\t\tlogger.FatalLog(\"failed to connect database: \" + err.Error())\n\t\treturn nil\n\t}\n\n\tsqlDB.SetMaxIdleConns(env.Int(\"SQL_MAX_IDLE_CONNS\", 100))\n\tsqlDB.SetMaxOpenConns(env.Int(\"SQL_MAX_OPEN_CONNS\", 1000))\n\tsqlDB.SetConnMaxLifetime(time.Second * time.Duration(env.Int(\"SQL_MAX_LIFETIME\", 60)))\n\treturn sqlDB\n}\n\nfunc closeDB(db *gorm.DB) error {\n\tsqlDB, err := db.DB()\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = sqlDB.Close()\n\treturn err\n}\n\nfunc CloseDB() error {\n\tif LOG_DB != DB {\n\t\terr := closeDB(LOG_DB)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn closeDB(DB)\n}\n"
},
{
"path": "model/option.go",
"content": "package model\n\nimport (\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\tbillingratio \"github.com/songquanpeng/one-api/relay/billing/ratio\"\n\t\"strconv\"\n\t\"strings\"\n\t\"time\"\n)\n\ntype Option struct {\n\tKey string `json:\"key\" gorm:\"primaryKey\"`\n\tValue string `json:\"value\"`\n}\n\nfunc AllOption() ([]*Option, error) {\n\tvar options []*Option\n\tvar err error\n\terr = DB.Find(&options).Error\n\treturn options, err\n}\n\nfunc InitOptionMap() {\n\tconfig.OptionMapRWMutex.Lock()\n\tconfig.OptionMap = make(map[string]string)\n\tconfig.OptionMap[\"PasswordLoginEnabled\"] = strconv.FormatBool(config.PasswordLoginEnabled)\n\tconfig.OptionMap[\"PasswordRegisterEnabled\"] = strconv.FormatBool(config.PasswordRegisterEnabled)\n\tconfig.OptionMap[\"EmailVerificationEnabled\"] = strconv.FormatBool(config.EmailVerificationEnabled)\n\tconfig.OptionMap[\"GitHubOAuthEnabled\"] = strconv.FormatBool(config.GitHubOAuthEnabled)\n\tconfig.OptionMap[\"OidcEnabled\"] = strconv.FormatBool(config.OidcEnabled)\n\tconfig.OptionMap[\"WeChatAuthEnabled\"] = strconv.FormatBool(config.WeChatAuthEnabled)\n\tconfig.OptionMap[\"TurnstileCheckEnabled\"] = strconv.FormatBool(config.TurnstileCheckEnabled)\n\tconfig.OptionMap[\"RegisterEnabled\"] = strconv.FormatBool(config.RegisterEnabled)\n\tconfig.OptionMap[\"AutomaticDisableChannelEnabled\"] = strconv.FormatBool(config.AutomaticDisableChannelEnabled)\n\tconfig.OptionMap[\"AutomaticEnableChannelEnabled\"] = strconv.FormatBool(config.AutomaticEnableChannelEnabled)\n\tconfig.OptionMap[\"ApproximateTokenEnabled\"] = strconv.FormatBool(config.ApproximateTokenEnabled)\n\tconfig.OptionMap[\"LogConsumeEnabled\"] = strconv.FormatBool(config.LogConsumeEnabled)\n\tconfig.OptionMap[\"DisplayInCurrencyEnabled\"] = strconv.FormatBool(config.DisplayInCurrencyEnabled)\n\tconfig.OptionMap[\"DisplayTokenStatEnabled\"] = strconv.FormatBool(config.DisplayTokenStatEnabled)\n\tconfig.OptionMap[\"ChannelDisableThreshold\"] = strconv.FormatFloat(config.ChannelDisableThreshold, 'f', -1, 64)\n\tconfig.OptionMap[\"EmailDomainRestrictionEnabled\"] = strconv.FormatBool(config.EmailDomainRestrictionEnabled)\n\tconfig.OptionMap[\"EmailDomainWhitelist\"] = strings.Join(config.EmailDomainWhitelist, \",\")\n\tconfig.OptionMap[\"SMTPServer\"] = \"\"\n\tconfig.OptionMap[\"SMTPFrom\"] = \"\"\n\tconfig.OptionMap[\"SMTPPort\"] = strconv.Itoa(config.SMTPPort)\n\tconfig.OptionMap[\"SMTPAccount\"] = \"\"\n\tconfig.OptionMap[\"SMTPToken\"] = \"\"\n\tconfig.OptionMap[\"Notice\"] = \"\"\n\tconfig.OptionMap[\"About\"] = \"\"\n\tconfig.OptionMap[\"HomePageContent\"] = \"\"\n\tconfig.OptionMap[\"Footer\"] = config.Footer\n\tconfig.OptionMap[\"SystemName\"] = config.SystemName\n\tconfig.OptionMap[\"Logo\"] = config.Logo\n\tconfig.OptionMap[\"ServerAddress\"] = \"\"\n\tconfig.OptionMap[\"GitHubClientId\"] = \"\"\n\tconfig.OptionMap[\"GitHubClientSecret\"] = \"\"\n\tconfig.OptionMap[\"WeChatServerAddress\"] = \"\"\n\tconfig.OptionMap[\"WeChatServerToken\"] = \"\"\n\tconfig.OptionMap[\"WeChatAccountQRCodeImageURL\"] = \"\"\n\tconfig.OptionMap[\"MessagePusherAddress\"] = \"\"\n\tconfig.OptionMap[\"MessagePusherToken\"] = \"\"\n\tconfig.OptionMap[\"TurnstileSiteKey\"] = \"\"\n\tconfig.OptionMap[\"TurnstileSecretKey\"] = \"\"\n\tconfig.OptionMap[\"QuotaForNewUser\"] = strconv.FormatInt(config.QuotaForNewUser, 10)\n\tconfig.OptionMap[\"QuotaForInviter\"] = strconv.FormatInt(config.QuotaForInviter, 10)\n\tconfig.OptionMap[\"QuotaForInvitee\"] = strconv.FormatInt(config.QuotaForInvitee, 10)\n\tconfig.OptionMap[\"QuotaRemindThreshold\"] = strconv.FormatInt(config.QuotaRemindThreshold, 10)\n\tconfig.OptionMap[\"PreConsumedQuota\"] = strconv.FormatInt(config.PreConsumedQuota, 10)\n\tconfig.OptionMap[\"ModelRatio\"] = billingratio.ModelRatio2JSONString()\n\tconfig.OptionMap[\"GroupRatio\"] = billingratio.GroupRatio2JSONString()\n\tconfig.OptionMap[\"CompletionRatio\"] = billingratio.CompletionRatio2JSONString()\n\tconfig.OptionMap[\"TopUpLink\"] = config.TopUpLink\n\tconfig.OptionMap[\"ChatLink\"] = config.ChatLink\n\tconfig.OptionMap[\"QuotaPerUnit\"] = strconv.FormatFloat(config.QuotaPerUnit, 'f', -1, 64)\n\tconfig.OptionMap[\"RetryTimes\"] = strconv.Itoa(config.RetryTimes)\n\tconfig.OptionMap[\"Theme\"] = config.Theme\n\tconfig.OptionMapRWMutex.Unlock()\n\tloadOptionsFromDatabase()\n}\n\nfunc loadOptionsFromDatabase() {\n\toptions, _ := AllOption()\n\tfor _, option := range options {\n\t\tif option.Key == \"ModelRatio\" {\n\t\t\toption.Value = billingratio.AddNewMissingRatio(option.Value)\n\t\t}\n\t\terr := updateOptionMap(option.Key, option.Value)\n\t\tif err != nil {\n\t\t\tlogger.SysError(\"failed to update option map: \" + err.Error())\n\t\t}\n\t}\n}\n\nfunc SyncOptions(frequency int) {\n\tfor {\n\t\ttime.Sleep(time.Duration(frequency) * time.Second)\n\t\tlogger.SysLog(\"syncing options from database\")\n\t\tloadOptionsFromDatabase()\n\t}\n}\n\nfunc UpdateOption(key string, value string) error {\n\t// Save to database first\n\toption := Option{\n\t\tKey: key,\n\t}\n\t// https://gorm.io/docs/update.html#Save-All-Fields\n\tDB.FirstOrCreate(&option, Option{Key: key})\n\toption.Value = value\n\t// Save is a combination function.\n\t// If save value does not contain primary key, it will execute Create,\n\t// otherwise it will execute Update (with all fields).\n\tDB.Save(&option)\n\t// Update OptionMap\n\treturn updateOptionMap(key, value)\n}\n\nfunc updateOptionMap(key string, value string) (err error) {\n\tconfig.OptionMapRWMutex.Lock()\n\tdefer config.OptionMapRWMutex.Unlock()\n\tconfig.OptionMap[key] = value\n\tif strings.HasSuffix(key, \"Enabled\") {\n\t\tboolValue := value == \"true\"\n\t\tswitch key {\n\t\tcase \"PasswordRegisterEnabled\":\n\t\t\tconfig.PasswordRegisterEnabled = boolValue\n\t\tcase \"PasswordLoginEnabled\":\n\t\t\tconfig.PasswordLoginEnabled = boolValue\n\t\tcase \"EmailVerificationEnabled\":\n\t\t\tconfig.EmailVerificationEnabled = boolValue\n\t\tcase \"GitHubOAuthEnabled\":\n\t\t\tconfig.GitHubOAuthEnabled = boolValue\n\t\tcase \"OidcEnabled\":\n\t\t\tconfig.OidcEnabled = boolValue\n\t\tcase \"WeChatAuthEnabled\":\n\t\t\tconfig.WeChatAuthEnabled = boolValue\n\t\tcase \"TurnstileCheckEnabled\":\n\t\t\tconfig.TurnstileCheckEnabled = boolValue\n\t\tcase \"RegisterEnabled\":\n\t\t\tconfig.RegisterEnabled = boolValue\n\t\tcase \"EmailDomainRestrictionEnabled\":\n\t\t\tconfig.EmailDomainRestrictionEnabled = boolValue\n\t\tcase \"AutomaticDisableChannelEnabled\":\n\t\t\tconfig.AutomaticDisableChannelEnabled = boolValue\n\t\tcase \"AutomaticEnableChannelEnabled\":\n\t\t\tconfig.AutomaticEnableChannelEnabled = boolValue\n\t\tcase \"ApproximateTokenEnabled\":\n\t\t\tconfig.ApproximateTokenEnabled = boolValue\n\t\tcase \"LogConsumeEnabled\":\n\t\t\tconfig.LogConsumeEnabled = boolValue\n\t\tcase \"DisplayInCurrencyEnabled\":\n\t\t\tconfig.DisplayInCurrencyEnabled = boolValue\n\t\tcase \"DisplayTokenStatEnabled\":\n\t\t\tconfig.DisplayTokenStatEnabled = boolValue\n\t\t}\n\t}\n\tswitch key {\n\tcase \"EmailDomainWhitelist\":\n\t\tconfig.EmailDomainWhitelist = strings.Split(value, \",\")\n\tcase \"SMTPServer\":\n\t\tconfig.SMTPServer = value\n\tcase \"SMTPPort\":\n\t\tintValue, _ := strconv.Atoi(value)\n\t\tconfig.SMTPPort = intValue\n\tcase \"SMTPAccount\":\n\t\tconfig.SMTPAccount = value\n\tcase \"SMTPFrom\":\n\t\tconfig.SMTPFrom = value\n\tcase \"SMTPToken\":\n\t\tconfig.SMTPToken = value\n\tcase \"ServerAddress\":\n\t\tconfig.ServerAddress = value\n\tcase \"GitHubClientId\":\n\t\tconfig.GitHubClientId = value\n\tcase \"GitHubClientSecret\":\n\t\tconfig.GitHubClientSecret = value\n\tcase \"LarkClientId\":\n\t\tconfig.LarkClientId = value\n\tcase \"LarkClientSecret\":\n\t\tconfig.LarkClientSecret = value\n\tcase \"OidcClientId\":\n\t\tconfig.OidcClientId = value\n\tcase \"OidcClientSecret\":\n\t\tconfig.OidcClientSecret = value\n\tcase \"OidcWellKnown\":\n\t\tconfig.OidcWellKnown = value\n\tcase \"OidcAuthorizationEndpoint\":\n\t\tconfig.OidcAuthorizationEndpoint = value\n\tcase \"OidcTokenEndpoint\":\n\t\tconfig.OidcTokenEndpoint = value\n\tcase \"OidcUserinfoEndpoint\":\n\t\tconfig.OidcUserinfoEndpoint = value\n\tcase \"Footer\":\n\t\tconfig.Footer = value\n\tcase \"SystemName\":\n\t\tconfig.SystemName = value\n\tcase \"Logo\":\n\t\tconfig.Logo = value\n\tcase \"WeChatServerAddress\":\n\t\tconfig.WeChatServerAddress = value\n\tcase \"WeChatServerToken\":\n\t\tconfig.WeChatServerToken = value\n\tcase \"WeChatAccountQRCodeImageURL\":\n\t\tconfig.WeChatAccountQRCodeImageURL = value\n\tcase \"MessagePusherAddress\":\n\t\tconfig.MessagePusherAddress = value\n\tcase \"MessagePusherToken\":\n\t\tconfig.MessagePusherToken = value\n\tcase \"TurnstileSiteKey\":\n\t\tconfig.TurnstileSiteKey = value\n\tcase \"TurnstileSecretKey\":\n\t\tconfig.TurnstileSecretKey = value\n\tcase \"QuotaForNewUser\":\n\t\tconfig.QuotaForNewUser, _ = strconv.ParseInt(value, 10, 64)\n\tcase \"QuotaForInviter\":\n\t\tconfig.QuotaForInviter, _ = strconv.ParseInt(value, 10, 64)\n\tcase \"QuotaForInvitee\":\n\t\tconfig.QuotaForInvitee, _ = strconv.ParseInt(value, 10, 64)\n\tcase \"QuotaRemindThreshold\":\n\t\tconfig.QuotaRemindThreshold, _ = strconv.ParseInt(value, 10, 64)\n\tcase \"PreConsumedQuota\":\n\t\tconfig.PreConsumedQuota, _ = strconv.ParseInt(value, 10, 64)\n\tcase \"RetryTimes\":\n\t\tconfig.RetryTimes, _ = strconv.Atoi(value)\n\tcase \"ModelRatio\":\n\t\terr = billingratio.UpdateModelRatioByJSONString(value)\n\tcase \"GroupRatio\":\n\t\terr = billingratio.UpdateGroupRatioByJSONString(value)\n\tcase \"CompletionRatio\":\n\t\terr = billingratio.UpdateCompletionRatioByJSONString(value)\n\tcase \"TopUpLink\":\n\t\tconfig.TopUpLink = value\n\tcase \"ChatLink\":\n\t\tconfig.ChatLink = value\n\tcase \"ChannelDisableThreshold\":\n\t\tconfig.ChannelDisableThreshold, _ = strconv.ParseFloat(value, 64)\n\tcase \"QuotaPerUnit\":\n\t\tconfig.QuotaPerUnit, _ = strconv.ParseFloat(value, 64)\n\tcase \"Theme\":\n\t\tconfig.Theme = value\n\t}\n\treturn err\n}\n"
},
{
"path": "model/redemption.go",
"content": "package model\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\n\t\"gorm.io/gorm\"\n\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n)\n\nconst (\n\tRedemptionCodeStatusEnabled = 1 // don't use 0, 0 is the default value!\n\tRedemptionCodeStatusDisabled = 2 // also don't use 0\n\tRedemptionCodeStatusUsed = 3 // also don't use 0\n)\n\ntype Redemption struct {\n\tId int `json:\"id\"`\n\tUserId int `json:\"user_id\"`\n\tKey string `json:\"key\" gorm:\"type:char(32);uniqueIndex\"`\n\tStatus int `json:\"status\" gorm:\"default:1\"`\n\tName string `json:\"name\" gorm:\"index\"`\n\tQuota int64 `json:\"quota\" gorm:\"bigint;default:100\"`\n\tCreatedTime int64 `json:\"created_time\" gorm:\"bigint\"`\n\tRedeemedTime int64 `json:\"redeemed_time\" gorm:\"bigint\"`\n\tCount int `json:\"count\" gorm:\"-:all\"` // only for api request\n}\n\nfunc GetAllRedemptions(startIdx int, num int) ([]*Redemption, error) {\n\tvar redemptions []*Redemption\n\tvar err error\n\terr = DB.Order(\"id desc\").Limit(num).Offset(startIdx).Find(&redemptions).Error\n\treturn redemptions, err\n}\n\nfunc SearchRedemptions(keyword string) (redemptions []*Redemption, err error) {\n\terr = DB.Where(\"id = ? or name LIKE ?\", keyword, keyword+\"%\").Find(&redemptions).Error\n\treturn redemptions, err\n}\n\nfunc GetRedemptionById(id int) (*Redemption, error) {\n\tif id == 0 {\n\t\treturn nil, errors.New(\"id 为空！\")\n\t}\n\tredemption := Redemption{Id: id}\n\tvar err error = nil\n\terr = DB.First(&redemption, \"id = ?\", id).Error\n\treturn &redemption, err\n}\n\nfunc Redeem(ctx context.Context, key string, userId int) (quota int64, err error) {\n\tif key == \"\" {\n\t\treturn 0, errors.New(\"未提供兑换码\")\n\t}\n\tif userId == 0 {\n\t\treturn 0, errors.New(\"无效的 user id\")\n\t}\n\tredemption := &Redemption{}\n\n\tkeyCol := \"`key`\"\n\tif common.UsingPostgreSQL {\n\t\tkeyCol = `\"key\"`\n\t}\n\n\terr = DB.Transaction(func(tx *gorm.DB) error {\n\t\terr := tx.Set(\"gorm:query_option\", \"FOR UPDATE\").Where(keyCol+\" = ?\", key).First(redemption).Error\n\t\tif err != nil {\n\t\t\treturn errors.New(\"无效的兑换码\")\n\t\t}\n\t\tif redemption.Status != RedemptionCodeStatusEnabled {\n\t\t\treturn errors.New(\"该兑换码已被使用\")\n\t\t}\n\t\terr = tx.Model(&User{}).Where(\"id = ?\", userId).Update(\"quota\", gorm.Expr(\"quota + ?\", redemption.Quota)).Error\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t\tredemption.RedeemedTime = helper.GetTimestamp()\n\t\tredemption.Status = RedemptionCodeStatusUsed\n\t\terr = tx.Save(redemption).Error\n\t\treturn err\n\t})\n\tif err != nil {\n\t\treturn 0, errors.New(\"兑换失败，\" + err.Error())\n\t}\n\tRecordLog(ctx, userId, LogTypeTopup, fmt.Sprintf(\"通过兑换码充值 %s\", common.LogQuota(redemption.Quota)))\n\treturn redemption.Quota, nil\n}\n\nfunc (redemption *Redemption) Insert() error {\n\tvar err error\n\terr = DB.Create(redemption).Error\n\treturn err\n}\n\nfunc (redemption *Redemption) SelectUpdate() error {\n\t// This can update zero values\n\treturn DB.Model(redemption).Select(\"redeemed_time\", \"status\").Updates(redemption).Error\n}\n\n// Update Make sure your token's fields is completed, because this will update non-zero values\nfunc (redemption *Redemption) Update() error {\n\tvar err error\n\terr = DB.Model(redemption).Select(\"name\", \"status\", \"quota\", \"redeemed_time\").Updates(redemption).Error\n\treturn err\n}\n\nfunc (redemption *Redemption) Delete() error {\n\tvar err error\n\terr = DB.Delete(redemption).Error\n\treturn err\n}\n\nfunc DeleteRedemptionById(id int) (err error) {\n\tif id == 0 {\n\t\treturn errors.New(\"id 为空！\")\n\t}\n\tredemption := Redemption{Id: id}\n\terr = DB.Where(redemption).First(&redemption).Error\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn redemption.Delete()\n}\n"
},
{
"path": "model/token.go",
"content": "package model\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\n\t\"gorm.io/gorm\"\n\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/common/message\"\n)\n\nconst (\n\tTokenStatusEnabled = 1 // don't use 0, 0 is the default value!\n\tTokenStatusDisabled = 2 // also don't use 0\n\tTokenStatusExpired = 3\n\tTokenStatusExhausted = 4\n)\n\ntype Token struct {\n\tId int `json:\"id\"`\n\tUserId int `json:\"user_id\"`\n\tKey string `json:\"key\" gorm:\"type:char(48);uniqueIndex\"`\n\tStatus int `json:\"status\" gorm:\"default:1\"`\n\tName string `json:\"name\" gorm:\"index\" `\n\tCreatedTime int64 `json:\"created_time\" gorm:\"bigint\"`\n\tAccessedTime int64 `json:\"accessed_time\" gorm:\"bigint\"`\n\tExpiredTime int64 `json:\"expired_time\" gorm:\"bigint;default:-1\"` // -1 means never expired\n\tRemainQuota int64 `json:\"remain_quota\" gorm:\"bigint;default:0\"`\n\tUnlimitedQuota bool `json:\"unlimited_quota\" gorm:\"default:false\"`\n\tUsedQuota int64 `json:\"used_quota\" gorm:\"bigint;default:0\"` // used quota\n\tModels *string `json:\"models\" gorm:\"type:text\"` // allowed models\n\tSubnet *string `json:\"subnet\" gorm:\"default:''\"` // allowed subnet\n}\n\nfunc GetAllUserTokens(userId int, startIdx int, num int, order string) ([]*Token, error) {\n\tvar tokens []*Token\n\tvar err error\n\tquery := DB.Where(\"user_id = ?\", userId)\n\n\tswitch order {\n\tcase \"remain_quota\":\n\t\tquery = query.Order(\"unlimited_quota desc, remain_quota desc\")\n\tcase \"used_quota\":\n\t\tquery = query.Order(\"used_quota desc\")\n\tdefault:\n\t\tquery = query.Order(\"id desc\")\n\t}\n\n\terr = query.Limit(num).Offset(startIdx).Find(&tokens).Error\n\treturn tokens, err\n}\n\nfunc SearchUserTokens(userId int, keyword string) (tokens []*Token, err error) {\n\terr = DB.Where(\"user_id = ?\", userId).Where(\"name LIKE ?\", keyword+\"%\").Find(&tokens).Error\n\treturn tokens, err\n}\n\nfunc ValidateUserToken(key string) (token *Token, err error) {\n\tif key == \"\" {\n\t\treturn nil, errors.New(\"未提供令牌\")\n\t}\n\ttoken, err = CacheGetTokenByKey(key)\n\tif err != nil {\n\t\tlogger.SysError(\"CacheGetTokenByKey failed: \" + err.Error())\n\t\tif errors.Is(err, gorm.ErrRecordNotFound) {\n\t\t\treturn nil, errors.New(\"无效的令牌\")\n\t\t}\n\t\treturn nil, errors.New(\"令牌验证失败\")\n\t}\n\tif token.Status == TokenStatusExhausted {\n\t\treturn nil, fmt.Errorf(\"令牌 %s（#%d）额度已用尽\", token.Name, token.Id)\n\t} else if token.Status == TokenStatusExpired {\n\t\treturn nil, errors.New(\"该令牌已过期\")\n\t}\n\tif token.Status != TokenStatusEnabled {\n\t\treturn nil, errors.New(\"该令牌状态不可用\")\n\t}\n\tif token.ExpiredTime != -1 && token.ExpiredTime < helper.GetTimestamp() {\n\t\tif !common.RedisEnabled {\n\t\t\ttoken.Status = TokenStatusExpired\n\t\t\terr := token.SelectUpdate()\n\t\t\tif err != nil {\n\t\t\t\tlogger.SysError(\"failed to update token status\" + err.Error())\n\t\t\t}\n\t\t}\n\t\treturn nil, errors.New(\"该令牌已过期\")\n\t}\n\tif !token.UnlimitedQuota && token.RemainQuota <= 0 {\n\t\tif !common.RedisEnabled {\n\t\t\t// in this case, we can make sure the token is exhausted\n\t\t\ttoken.Status = TokenStatusExhausted\n\t\t\terr := token.SelectUpdate()\n\t\t\tif err != nil {\n\t\t\t\tlogger.SysError(\"failed to update token status\" + err.Error())\n\t\t\t}\n\t\t}\n\t\treturn nil, errors.New(\"该令牌额度已用尽\")\n\t}\n\treturn token, nil\n}\n\nfunc GetTokenByIds(id int, userId int) (*Token, error) {\n\tif id == 0 || userId == 0 {\n\t\treturn nil, errors.New(\"id 或 userId 为空！\")\n\t}\n\ttoken := Token{Id: id, UserId: userId}\n\tvar err error = nil\n\terr = DB.First(&token, \"id = ? and user_id = ?\", id, userId).Error\n\treturn &token, err\n}\n\nfunc GetTokenById(id int) (*Token, error) {\n\tif id == 0 {\n\t\treturn nil, errors.New(\"id 为空！\")\n\t}\n\ttoken := Token{Id: id}\n\tvar err error = nil\n\terr = DB.First(&token, \"id = ?\", id).Error\n\treturn &token, err\n}\n\nfunc (t *Token) Insert() error {\n\tvar err error\n\terr = DB.Create(t).Error\n\treturn err\n}\n\n// Update Make sure your token's fields is completed, because this will update non-zero values\nfunc (t *Token) Update() error {\n\tvar err error\n\terr = DB.Model(t).Select(\"name\", \"status\", \"expired_time\", \"remain_quota\", \"unlimited_quota\", \"models\", \"subnet\").Updates(t).Error\n\treturn err\n}\n\nfunc (t *Token) SelectUpdate() error {\n\t// This can update zero values\n\treturn DB.Model(t).Select(\"accessed_time\", \"status\").Updates(t).Error\n}\n\nfunc (t *Token) Delete() error {\n\tvar err error\n\terr = DB.Delete(t).Error\n\treturn err\n}\n\nfunc (t *Token) GetModels() string {\n\tif t == nil {\n\t\treturn \"\"\n\t}\n\tif t.Models == nil {\n\t\treturn \"\"\n\t}\n\treturn *t.Models\n}\n\nfunc DeleteTokenById(id int, userId int) (err error) {\n\t// Why we need userId here? In case user want to delete other's token.\n\tif id == 0 || userId == 0 {\n\t\treturn errors.New(\"id 或 userId 为空！\")\n\t}\n\ttoken := Token{Id: id, UserId: userId}\n\terr = DB.Where(token).First(&token).Error\n\tif err != nil {\n\t\treturn err\n\t}\n\treturn token.Delete()\n}\n\nfunc IncreaseTokenQuota(id int, quota int64) (err error) {\n\tif quota < 0 {\n\t\treturn errors.New(\"quota 不能为负数！\")\n\t}\n\tif config.BatchUpdateEnabled {\n\t\taddNewRecord(BatchUpdateTypeTokenQuota, id, quota)\n\t\treturn nil\n\t}\n\treturn increaseTokenQuota(id, quota)\n}\n\nfunc increaseTokenQuota(id int, quota int64) (err error) {\n\terr = DB.Model(&Token{}).Where(\"id = ?\", id).Updates(\n\t\tmap[string]interface{}{\n\t\t\t\"remain_quota\": gorm.Expr(\"remain_quota + ?\", quota),\n\t\t\t\"used_quota\": gorm.Expr(\"used_quota - ?\", quota),\n\t\t\t\"accessed_time\": helper.GetTimestamp(),\n\t\t},\n\t).Error\n\treturn err\n}\n\nfunc DecreaseTokenQuota(id int, quota int64) (err error) {\n\tif quota < 0 {\n\t\treturn errors.New(\"quota 不能为负数！\")\n\t}\n\tif config.BatchUpdateEnabled {\n\t\taddNewRecord(BatchUpdateTypeTokenQuota, id, -quota)\n\t\treturn nil\n\t}\n\treturn decreaseTokenQuota(id, quota)\n}\n\nfunc decreaseTokenQuota(id int, quota int64) (err error) {\n\terr = DB.Model(&Token{}).Where(\"id = ?\", id).Updates(\n\t\tmap[string]interface{}{\n\t\t\t\"remain_quota\": gorm.Expr(\"remain_quota - ?\", quota),\n\t\t\t\"used_quota\": gorm.Expr(\"used_quota + ?\", quota),\n\t\t\t\"accessed_time\": helper.GetTimestamp(),\n\t\t},\n\t).Error\n\treturn err\n}\n\nfunc PreConsumeTokenQuota(tokenId int, quota int64) (err error) {\n\tif quota < 0 {\n\t\treturn errors.New(\"quota 不能为负数！\")\n\t}\n\ttoken, err := GetTokenById(tokenId)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif !token.UnlimitedQuota && token.RemainQuota < quota {\n\t\treturn errors.New(\"令牌额度不足\")\n\t}\n\tuserQuota, err := GetUserQuota(token.UserId)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif userQuota < quota {\n\t\treturn errors.New(\"用户额度不足\")\n\t}\n\tquotaTooLow := userQuota >= config.QuotaRemindThreshold && userQuota-quota < config.QuotaRemindThreshold\n\tnoMoreQuota := userQuota-quota <= 0\n\tif quotaTooLow || noMoreQuota {\n\t\tgo func() {\n\t\t\temail, err := GetUserEmail(token.UserId)\n\t\t\tif err != nil {\n\t\t\t\tlogger.SysError(\"failed to fetch user email: \" + err.Error())\n\t\t\t}\n\t\t\tprompt := \"额度提醒\"\n\t\t\tvar contentText string\n\t\t\tif noMoreQuota {\n\t\t\t\tcontentText = \"您的额度已用尽\"\n\t\t\t} else {\n\t\t\t\tcontentText = \"您的额度即将用尽\"\n\t\t\t}\n\t\t\tif email != \"\" {\n\t\t\t\ttopUpLink := fmt.Sprintf(\"%s/topup\", config.ServerAddress)\n\t\t\t\tcontent := message.EmailTemplate(\n\t\t\t\t\tprompt,\n\t\t\t\t\tfmt.Sprintf(`\n\t\t\t\t\t\t您好！
\n\t\t\t\t\t\t%s，当前剩余额度为 %d 。
\n\t\t\t\t\t\t为了不影响您的使用，请及时充值。
\n\t\t\t\t\t\t\n\t\t\t\t\t\t\t立即充值 \n\t\t\t\t\t\t
\n\t\t\t\t\t\t如果按钮无法点击，请复制以下链接到浏览器中打开：
\n\t\t\t\t\t\t%s
\n\t\t\t\t\t`, contentText, userQuota, topUpLink, topUpLink),\n\t\t\t\t)\n\t\t\t\terr = message.SendEmail(prompt, email, content)\n\t\t\t\tif err != nil {\n\t\t\t\t\tlogger.SysError(\"failed to send email: \" + err.Error())\n\t\t\t\t}\n\t\t\t}\n\t\t}()\n\t}\n\tif !token.UnlimitedQuota {\n\t\terr = DecreaseTokenQuota(tokenId, quota)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\terr = DecreaseUserQuota(token.UserId, quota)\n\treturn err\n}\n\nfunc PostConsumeTokenQuota(tokenId int, quota int64) (err error) {\n\ttoken, err := GetTokenById(tokenId)\n\tif err != nil {\n\t\treturn err\n\t}\n\tif quota > 0 {\n\t\terr = DecreaseUserQuota(token.UserId, quota)\n\t} else {\n\t\terr = IncreaseUserQuota(token.UserId, -quota)\n\t}\n\tif !token.UnlimitedQuota {\n\t\tif quota > 0 {\n\t\t\terr = DecreaseTokenQuota(tokenId, quota)\n\t\t} else {\n\t\t\terr = IncreaseTokenQuota(tokenId, -quota)\n\t\t}\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\treturn nil\n}\n"
},
{
"path": "model/user.go",
"content": "package model\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"strings\"\n\n\t\"gorm.io/gorm\"\n\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/blacklist\"\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/common/random\"\n)\n\nconst (\n\tRoleGuestUser = 0\n\tRoleCommonUser = 1\n\tRoleAdminUser = 10\n\tRoleRootUser = 100\n)\n\nconst (\n\tUserStatusEnabled = 1 // don't use 0, 0 is the default value!\n\tUserStatusDisabled = 2 // also don't use 0\n\tUserStatusDeleted = 3\n)\n\n// User if you add sensitive fields, don't forget to clean them in setupLogin function.\n// Otherwise, the sensitive information will be saved on local storage in plain text!\ntype User struct {\n\tId int `json:\"id\"`\n\tUsername string `json:\"username\" gorm:\"unique;index\" validate:\"max=12\"`\n\tPassword string `json:\"password\" gorm:\"not null;\" validate:\"min=8,max=20\"`\n\tDisplayName string `json:\"display_name\" gorm:\"index\" validate:\"max=20\"`\n\tRole int `json:\"role\" gorm:\"type:int;default:1\"` // admin, util\n\tStatus int `json:\"status\" gorm:\"type:int;default:1\"` // enabled, disabled\n\tEmail string `json:\"email\" gorm:\"index\" validate:\"max=50\"`\n\tGitHubId string `json:\"github_id\" gorm:\"column:github_id;index\"`\n\tWeChatId string `json:\"wechat_id\" gorm:\"column:wechat_id;index\"`\n\tLarkId string `json:\"lark_id\" gorm:\"column:lark_id;index\"`\n\tOidcId string `json:\"oidc_id\" gorm:\"column:oidc_id;index\"`\n\tVerificationCode string `json:\"verification_code\" gorm:\"-:all\"` // this field is only for Email verification, don't save it to database!\n\tAccessToken string `json:\"access_token\" gorm:\"type:char(32);column:access_token;uniqueIndex\"` // this token is for system management\n\tQuota int64 `json:\"quota\" gorm:\"bigint;default:0\"`\n\tUsedQuota int64 `json:\"used_quota\" gorm:\"bigint;default:0;column:used_quota\"` // used quota\n\tRequestCount int `json:\"request_count\" gorm:\"type:int;default:0;\"` // request number\n\tGroup string `json:\"group\" gorm:\"type:varchar(32);default:'default'\"`\n\tAffCode string `json:\"aff_code\" gorm:\"type:varchar(32);column:aff_code;uniqueIndex\"`\n\tInviterId int `json:\"inviter_id\" gorm:\"type:int;column:inviter_id;index\"`\n}\n\nfunc GetMaxUserId() int {\n\tvar user User\n\tDB.Last(&user)\n\treturn user.Id\n}\n\nfunc GetAllUsers(startIdx int, num int, order string) (users []*User, err error) {\n\tquery := DB.Limit(num).Offset(startIdx).Omit(\"password\").Where(\"status != ?\", UserStatusDeleted)\n\n\tswitch order {\n\tcase \"quota\":\n\t\tquery = query.Order(\"quota desc\")\n\tcase \"used_quota\":\n\t\tquery = query.Order(\"used_quota desc\")\n\tcase \"request_count\":\n\t\tquery = query.Order(\"request_count desc\")\n\tdefault:\n\t\tquery = query.Order(\"id desc\")\n\t}\n\n\terr = query.Find(&users).Error\n\treturn users, err\n}\n\nfunc SearchUsers(keyword string) (users []*User, err error) {\n\tif !common.UsingPostgreSQL {\n\t\terr = DB.Omit(\"password\").Where(\"id = ? or username LIKE ? or email LIKE ? or display_name LIKE ?\", keyword, keyword+\"%\", keyword+\"%\", keyword+\"%\").Find(&users).Error\n\t} else {\n\t\terr = DB.Omit(\"password\").Where(\"username LIKE ? or email LIKE ? or display_name LIKE ?\", keyword+\"%\", keyword+\"%\", keyword+\"%\").Find(&users).Error\n\t}\n\treturn users, err\n}\n\nfunc GetUserById(id int, selectAll bool) (*User, error) {\n\tif id == 0 {\n\t\treturn nil, errors.New(\"id 为空！\")\n\t}\n\tuser := User{Id: id}\n\tvar err error = nil\n\tif selectAll {\n\t\terr = DB.First(&user, \"id = ?\", id).Error\n\t} else {\n\t\terr = DB.Omit(\"password\", \"access_token\").First(&user, \"id = ?\", id).Error\n\t}\n\treturn &user, err\n}\n\nfunc GetUserIdByAffCode(affCode string) (int, error) {\n\tif affCode == \"\" {\n\t\treturn 0, errors.New(\"affCode 为空！\")\n\t}\n\tvar user User\n\terr := DB.Select(\"id\").First(&user, \"aff_code = ?\", affCode).Error\n\treturn user.Id, err\n}\n\nfunc DeleteUserById(id int) (err error) {\n\tif id == 0 {\n\t\treturn errors.New(\"id 为空！\")\n\t}\n\tuser := User{Id: id}\n\treturn user.Delete()\n}\n\nfunc (user *User) Insert(ctx context.Context, inviterId int) error {\n\tvar err error\n\tif user.Password != \"\" {\n\t\tuser.Password, err = common.Password2Hash(user.Password)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\tuser.Quota = config.QuotaForNewUser\n\tuser.AccessToken = random.GetUUID()\n\tuser.AffCode = random.GetRandomString(4)\n\tresult := DB.Create(user)\n\tif result.Error != nil {\n\t\treturn result.Error\n\t}\n\tif config.QuotaForNewUser > 0 {\n\t\tRecordLog(ctx, user.Id, LogTypeSystem, fmt.Sprintf(\"新用户注册赠送 %s\", common.LogQuota(config.QuotaForNewUser)))\n\t}\n\tif inviterId != 0 {\n\t\tif config.QuotaForInvitee > 0 {\n\t\t\t_ = IncreaseUserQuota(user.Id, config.QuotaForInvitee)\n\t\t\tRecordLog(ctx, user.Id, LogTypeSystem, fmt.Sprintf(\"使用邀请码赠送 %s\", common.LogQuota(config.QuotaForInvitee)))\n\t\t}\n\t\tif config.QuotaForInviter > 0 {\n\t\t\t_ = IncreaseUserQuota(inviterId, config.QuotaForInviter)\n\t\t\tRecordLog(ctx, inviterId, LogTypeSystem, fmt.Sprintf(\"邀请用户赠送 %s\", common.LogQuota(config.QuotaForInviter)))\n\t\t}\n\t}\n\t// create default token\n\tcleanToken := Token{\n\t\tUserId: user.Id,\n\t\tName: \"default\",\n\t\tKey: random.GenerateKey(),\n\t\tCreatedTime: helper.GetTimestamp(),\n\t\tAccessedTime: helper.GetTimestamp(),\n\t\tExpiredTime: -1,\n\t\tRemainQuota: -1,\n\t\tUnlimitedQuota: true,\n\t}\n\tresult.Error = cleanToken.Insert()\n\tif result.Error != nil {\n\t\t// do not block\n\t\tlogger.SysError(fmt.Sprintf(\"create default token for user %d failed: %s\", user.Id, result.Error.Error()))\n\t}\n\treturn nil\n}\n\nfunc (user *User) Update(updatePassword bool) error {\n\tvar err error\n\tif updatePassword {\n\t\tuser.Password, err = common.Password2Hash(user.Password)\n\t\tif err != nil {\n\t\t\treturn err\n\t\t}\n\t}\n\tif user.Status == UserStatusDisabled {\n\t\tblacklist.BanUser(user.Id)\n\t} else if user.Status == UserStatusEnabled {\n\t\tblacklist.UnbanUser(user.Id)\n\t}\n\terr = DB.Model(user).Updates(user).Error\n\treturn err\n}\n\nfunc (user *User) Delete() error {\n\tif user.Id == 0 {\n\t\treturn errors.New(\"id 为空！\")\n\t}\n\tblacklist.BanUser(user.Id)\n\tuser.Username = fmt.Sprintf(\"deleted_%s\", random.GetUUID())\n\tuser.Status = UserStatusDeleted\n\terr := DB.Model(user).Updates(user).Error\n\treturn err\n}\n\n// ValidateAndFill check password & user status\nfunc (user *User) ValidateAndFill() (err error) {\n\t// When querying with struct, GORM will only query with non-zero fields,\n\t// that means if your field’s value is 0, '', false or other zero values,\n\t// it won’t be used to build query conditions\n\tpassword := user.Password\n\tif user.Username == \"\" || password == \"\" {\n\t\treturn errors.New(\"用户名或密码为空\")\n\t}\n\terr = DB.Where(\"username = ?\", user.Username).First(user).Error\n\tif err != nil {\n\t\t// we must make sure check username firstly\n\t\t// consider this case: a malicious user set his username as other's email\n\t\terr := DB.Where(\"email = ?\", user.Username).First(user).Error\n\t\tif err != nil {\n\t\t\treturn errors.New(\"用户名或密码错误，或用户已被封禁\")\n\t\t}\n\t}\n\tokay := common.ValidatePasswordAndHash(password, user.Password)\n\tif !okay || user.Status != UserStatusEnabled {\n\t\treturn errors.New(\"用户名或密码错误，或用户已被封禁\")\n\t}\n\treturn nil\n}\n\nfunc (user *User) FillUserById() error {\n\tif user.Id == 0 {\n\t\treturn errors.New(\"id 为空！\")\n\t}\n\tDB.Where(User{Id: user.Id}).First(user)\n\treturn nil\n}\n\nfunc (user *User) FillUserByEmail() error {\n\tif user.Email == \"\" {\n\t\treturn errors.New(\"email 为空！\")\n\t}\n\tDB.Where(User{Email: user.Email}).First(user)\n\treturn nil\n}\n\nfunc (user *User) FillUserByGitHubId() error {\n\tif user.GitHubId == \"\" {\n\t\treturn errors.New(\"GitHub id 为空！\")\n\t}\n\tDB.Where(User{GitHubId: user.GitHubId}).First(user)\n\treturn nil\n}\n\nfunc (user *User) FillUserByLarkId() error {\n\tif user.LarkId == \"\" {\n\t\treturn errors.New(\"lark id 为空！\")\n\t}\n\tDB.Where(User{LarkId: user.LarkId}).First(user)\n\treturn nil\n}\n\nfunc (user *User) FillUserByOidcId() error {\n\tif user.OidcId == \"\" {\n\t\treturn errors.New(\"oidc id 为空！\")\n\t}\n\tDB.Where(User{OidcId: user.OidcId}).First(user)\n\treturn nil\n}\n\nfunc (user *User) FillUserByWeChatId() error {\n\tif user.WeChatId == \"\" {\n\t\treturn errors.New(\"WeChat id 为空！\")\n\t}\n\tDB.Where(User{WeChatId: user.WeChatId}).First(user)\n\treturn nil\n}\n\nfunc (user *User) FillUserByUsername() error {\n\tif user.Username == \"\" {\n\t\treturn errors.New(\"username 为空！\")\n\t}\n\tDB.Where(User{Username: user.Username}).First(user)\n\treturn nil\n}\n\nfunc IsEmailAlreadyTaken(email string) bool {\n\treturn DB.Where(\"email = ?\", email).Find(&User{}).RowsAffected == 1\n}\n\nfunc IsWeChatIdAlreadyTaken(wechatId string) bool {\n\treturn DB.Where(\"wechat_id = ?\", wechatId).Find(&User{}).RowsAffected == 1\n}\n\nfunc IsGitHubIdAlreadyTaken(githubId string) bool {\n\treturn DB.Where(\"github_id = ?\", githubId).Find(&User{}).RowsAffected == 1\n}\n\nfunc IsLarkIdAlreadyTaken(githubId string) bool {\n\treturn DB.Where(\"lark_id = ?\", githubId).Find(&User{}).RowsAffected == 1\n}\n\nfunc IsOidcIdAlreadyTaken(oidcId string) bool {\n\treturn DB.Where(\"oidc_id = ?\", oidcId).Find(&User{}).RowsAffected == 1\n}\n\nfunc IsUsernameAlreadyTaken(username string) bool {\n\treturn DB.Where(\"username = ?\", username).Find(&User{}).RowsAffected == 1\n}\n\nfunc ResetUserPasswordByEmail(email string, password string) error {\n\tif email == \"\" || password == \"\" {\n\t\treturn errors.New(\"邮箱地址或密码为空！\")\n\t}\n\thashedPassword, err := common.Password2Hash(password)\n\tif err != nil {\n\t\treturn err\n\t}\n\terr = DB.Model(&User{}).Where(\"email = ?\", email).Update(\"password\", hashedPassword).Error\n\treturn err\n}\n\nfunc IsAdmin(userId int) bool {\n\tif userId == 0 {\n\t\treturn false\n\t}\n\tvar user User\n\terr := DB.Where(\"id = ?\", userId).Select(\"role\").Find(&user).Error\n\tif err != nil {\n\t\tlogger.SysError(\"no such user \" + err.Error())\n\t\treturn false\n\t}\n\treturn user.Role >= RoleAdminUser\n}\n\nfunc IsUserEnabled(userId int) (bool, error) {\n\tif userId == 0 {\n\t\treturn false, errors.New(\"user id is empty\")\n\t}\n\tvar user User\n\terr := DB.Where(\"id = ?\", userId).Select(\"status\").Find(&user).Error\n\tif err != nil {\n\t\treturn false, err\n\t}\n\treturn user.Status == UserStatusEnabled, nil\n}\n\nfunc ValidateAccessToken(token string) (user *User) {\n\tif token == \"\" {\n\t\treturn nil\n\t}\n\ttoken = strings.Replace(token, \"Bearer \", \"\", 1)\n\tuser = &User{}\n\tif DB.Where(\"access_token = ?\", token).First(user).RowsAffected == 1 {\n\t\treturn user\n\t}\n\treturn nil\n}\n\nfunc GetUserQuota(id int) (quota int64, err error) {\n\terr = DB.Model(&User{}).Where(\"id = ?\", id).Select(\"quota\").Find("a).Error\n\treturn quota, err\n}\n\nfunc GetUserUsedQuota(id int) (quota int64, err error) {\n\terr = DB.Model(&User{}).Where(\"id = ?\", id).Select(\"used_quota\").Find("a).Error\n\treturn quota, err\n}\n\nfunc GetUserEmail(id int) (email string, err error) {\n\terr = DB.Model(&User{}).Where(\"id = ?\", id).Select(\"email\").Find(&email).Error\n\treturn email, err\n}\n\nfunc GetUserGroup(id int) (group string, err error) {\n\tgroupCol := \"`group`\"\n\tif common.UsingPostgreSQL {\n\t\tgroupCol = `\"group\"`\n\t}\n\n\terr = DB.Model(&User{}).Where(\"id = ?\", id).Select(groupCol).Find(&group).Error\n\treturn group, err\n}\n\nfunc IncreaseUserQuota(id int, quota int64) (err error) {\n\tif quota < 0 {\n\t\treturn errors.New(\"quota 不能为负数！\")\n\t}\n\tif config.BatchUpdateEnabled {\n\t\taddNewRecord(BatchUpdateTypeUserQuota, id, quota)\n\t\treturn nil\n\t}\n\treturn increaseUserQuota(id, quota)\n}\n\nfunc increaseUserQuota(id int, quota int64) (err error) {\n\terr = DB.Model(&User{}).Where(\"id = ?\", id).Update(\"quota\", gorm.Expr(\"quota + ?\", quota)).Error\n\treturn err\n}\n\nfunc DecreaseUserQuota(id int, quota int64) (err error) {\n\tif quota < 0 {\n\t\treturn errors.New(\"quota 不能为负数！\")\n\t}\n\tif config.BatchUpdateEnabled {\n\t\taddNewRecord(BatchUpdateTypeUserQuota, id, -quota)\n\t\treturn nil\n\t}\n\treturn decreaseUserQuota(id, quota)\n}\n\nfunc decreaseUserQuota(id int, quota int64) (err error) {\n\terr = DB.Model(&User{}).Where(\"id = ?\", id).Update(\"quota\", gorm.Expr(\"quota - ?\", quota)).Error\n\treturn err\n}\n\nfunc GetRootUserEmail() (email string) {\n\tDB.Model(&User{}).Where(\"role = ?\", RoleRootUser).Select(\"email\").Find(&email)\n\treturn email\n}\n\nfunc UpdateUserUsedQuotaAndRequestCount(id int, quota int64) {\n\tif config.BatchUpdateEnabled {\n\t\taddNewRecord(BatchUpdateTypeUsedQuota, id, quota)\n\t\taddNewRecord(BatchUpdateTypeRequestCount, id, 1)\n\t\treturn\n\t}\n\tupdateUserUsedQuotaAndRequestCount(id, quota, 1)\n}\n\nfunc updateUserUsedQuotaAndRequestCount(id int, quota int64, count int) {\n\terr := DB.Model(&User{}).Where(\"id = ?\", id).Updates(\n\t\tmap[string]interface{}{\n\t\t\t\"used_quota\": gorm.Expr(\"used_quota + ?\", quota),\n\t\t\t\"request_count\": gorm.Expr(\"request_count + ?\", count),\n\t\t},\n\t).Error\n\tif err != nil {\n\t\tlogger.SysError(\"failed to update user used quota and request count: \" + err.Error())\n\t}\n}\n\nfunc updateUserUsedQuota(id int, quota int64) {\n\terr := DB.Model(&User{}).Where(\"id = ?\", id).Updates(\n\t\tmap[string]interface{}{\n\t\t\t\"used_quota\": gorm.Expr(\"used_quota + ?\", quota),\n\t\t},\n\t).Error\n\tif err != nil {\n\t\tlogger.SysError(\"failed to update user used quota: \" + err.Error())\n\t}\n}\n\nfunc updateUserRequestCount(id int, count int) {\n\terr := DB.Model(&User{}).Where(\"id = ?\", id).Update(\"request_count\", gorm.Expr(\"request_count + ?\", count)).Error\n\tif err != nil {\n\t\tlogger.SysError(\"failed to update user request count: \" + err.Error())\n\t}\n}\n\nfunc GetUsernameById(id int) (username string) {\n\tDB.Model(&User{}).Where(\"id = ?\", id).Select(\"username\").Find(&username)\n\treturn username\n}\n"
},
{
"path": "model/utils.go",
"content": "package model\n\nimport (\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"sync\"\n\t\"time\"\n)\n\nconst (\n\tBatchUpdateTypeUserQuota = iota\n\tBatchUpdateTypeTokenQuota\n\tBatchUpdateTypeUsedQuota\n\tBatchUpdateTypeChannelUsedQuota\n\tBatchUpdateTypeRequestCount\n\tBatchUpdateTypeCount // if you add a new type, you need to add a new map and a new lock\n)\n\nvar batchUpdateStores []map[int]int64\nvar batchUpdateLocks []sync.Mutex\n\nfunc init() {\n\tfor i := 0; i < BatchUpdateTypeCount; i++ {\n\t\tbatchUpdateStores = append(batchUpdateStores, make(map[int]int64))\n\t\tbatchUpdateLocks = append(batchUpdateLocks, sync.Mutex{})\n\t}\n}\n\nfunc InitBatchUpdater() {\n\tgo func() {\n\t\tfor {\n\t\t\ttime.Sleep(time.Duration(config.BatchUpdateInterval) * time.Second)\n\t\t\tbatchUpdate()\n\t\t}\n\t}()\n}\n\nfunc addNewRecord(type_ int, id int, value int64) {\n\tbatchUpdateLocks[type_].Lock()\n\tdefer batchUpdateLocks[type_].Unlock()\n\tif _, ok := batchUpdateStores[type_][id]; !ok {\n\t\tbatchUpdateStores[type_][id] = value\n\t} else {\n\t\tbatchUpdateStores[type_][id] += value\n\t}\n}\n\nfunc batchUpdate() {\n\tlogger.SysLog(\"batch update started\")\n\tfor i := 0; i < BatchUpdateTypeCount; i++ {\n\t\tbatchUpdateLocks[i].Lock()\n\t\tstore := batchUpdateStores[i]\n\t\tbatchUpdateStores[i] = make(map[int]int64)\n\t\tbatchUpdateLocks[i].Unlock()\n\t\t// TODO: maybe we can combine updates with same key?\n\t\tfor key, value := range store {\n\t\t\tswitch i {\n\t\t\tcase BatchUpdateTypeUserQuota:\n\t\t\t\terr := increaseUserQuota(key, value)\n\t\t\t\tif err != nil {\n\t\t\t\t\tlogger.SysError(\"failed to batch update user quota: \" + err.Error())\n\t\t\t\t}\n\t\t\tcase BatchUpdateTypeTokenQuota:\n\t\t\t\terr := increaseTokenQuota(key, value)\n\t\t\t\tif err != nil {\n\t\t\t\t\tlogger.SysError(\"failed to batch update token quota: \" + err.Error())\n\t\t\t\t}\n\t\t\tcase BatchUpdateTypeUsedQuota:\n\t\t\t\tupdateUserUsedQuota(key, value)\n\t\t\tcase BatchUpdateTypeRequestCount:\n\t\t\t\tupdateUserRequestCount(key, int(value))\n\t\t\tcase BatchUpdateTypeChannelUsedQuota:\n\t\t\t\tupdateChannelUsedQuota(key, value)\n\t\t\t}\n\t\t}\n\t}\n\tlogger.SysLog(\"batch update finished\")\n}\n"
},
{
"path": "monitor/channel.go",
"content": "package monitor\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/common/message\"\n\t\"github.com/songquanpeng/one-api/model\"\n)\n\nfunc notifyRootUser(subject string, content string) {\n\tif config.MessagePusherAddress != \"\" {\n\t\terr := message.SendMessage(subject, content, content)\n\t\tif err != nil {\n\t\t\tlogger.SysError(fmt.Sprintf(\"failed to send message: %s\", err.Error()))\n\t\t} else {\n\t\t\treturn\n\t\t}\n\t}\n\tif config.RootUserEmail == \"\" {\n\t\tconfig.RootUserEmail = model.GetRootUserEmail()\n\t}\n\terr := message.SendEmail(subject, config.RootUserEmail, content)\n\tif err != nil {\n\t\tlogger.SysError(fmt.Sprintf(\"failed to send email: %s\", err.Error()))\n\t}\n}\n\n// DisableChannel disable & notify\nfunc DisableChannel(channelId int, channelName string, reason string) {\n\tmodel.UpdateChannelStatusById(channelId, model.ChannelStatusAutoDisabled)\n\tlogger.SysLog(fmt.Sprintf(\"channel #%d has been disabled: %s\", channelId, reason))\n\tsubject := fmt.Sprintf(\"渠道状态变更提醒\")\n\tcontent := message.EmailTemplate(\n\t\tsubject,\n\t\tfmt.Sprintf(`\n\t\t\t您好！
\n\t\t\t渠道「%s 」（#%d）已被禁用。
\n\t\t\t禁用原因：
\n\t\t\t%s
\n\t\t`, channelName, channelId, reason),\n\t)\n\tnotifyRootUser(subject, content)\n}\n\nfunc MetricDisableChannel(channelId int, successRate float64) {\n\tmodel.UpdateChannelStatusById(channelId, model.ChannelStatusAutoDisabled)\n\tlogger.SysLog(fmt.Sprintf(\"channel #%d has been disabled due to low success rate: %.2f\", channelId, successRate*100))\n\tsubject := fmt.Sprintf(\"渠道状态变更提醒\")\n\tcontent := message.EmailTemplate(\n\t\tsubject,\n\t\tfmt.Sprintf(`\n\t\t\t您好！
\n\t\t\t渠道 #%d 已被系统自动禁用。
\n\t\t\t禁用原因：
\n\t\t\t该渠道在最近 %d 次调用中成功率为 %.2f%% ，低于系统阈值 %.2f%% 。
\n\t\t`, channelId, config.MetricQueueSize, successRate*100, config.MetricSuccessRateThreshold*100),\n\t)\n\tnotifyRootUser(subject, content)\n}\n\n// EnableChannel enable & notify\nfunc EnableChannel(channelId int, channelName string) {\n\tmodel.UpdateChannelStatusById(channelId, model.ChannelStatusEnabled)\n\tlogger.SysLog(fmt.Sprintf(\"channel #%d has been enabled\", channelId))\n\tsubject := fmt.Sprintf(\"渠道状态变更提醒\")\n\tcontent := message.EmailTemplate(\n\t\tsubject,\n\t\tfmt.Sprintf(`\n\t\t\t您好！
\n\t\t\t渠道「%s 」（#%d）已被重新启用。
\n\t\t\t您现在可以继续使用该渠道了。
\n\t\t`, channelName, channelId),\n\t)\n\tnotifyRootUser(subject, content)\n}\n"
},
{
"path": "monitor/manage.go",
"content": "package monitor\n\nimport (\n\t\"net/http\"\n\t\"strings\"\n\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\nfunc ShouldDisableChannel(err *model.Error, statusCode int) bool {\n\tif !config.AutomaticDisableChannelEnabled {\n\t\treturn false\n\t}\n\tif err == nil {\n\t\treturn false\n\t}\n\tif statusCode == http.StatusUnauthorized {\n\t\treturn true\n\t}\n\tswitch err.Type {\n\tcase \"insufficient_quota\", \"authentication_error\", \"permission_error\", \"forbidden\":\n\t\treturn true\n\t}\n\tif err.Code == \"invalid_api_key\" || err.Code == \"account_deactivated\" {\n\t\treturn true\n\t}\n\n\tlowerMessage := strings.ToLower(err.Message)\n\tif strings.Contains(lowerMessage, \"your access was terminated\") ||\n\t\tstrings.Contains(lowerMessage, \"violation of our policies\") ||\n\t\tstrings.Contains(lowerMessage, \"your credit balance is too low\") ||\n\t\tstrings.Contains(lowerMessage, \"organization has been disabled\") ||\n\t\tstrings.Contains(lowerMessage, \"credit\") ||\n\t\tstrings.Contains(lowerMessage, \"balance\") ||\n\t\tstrings.Contains(lowerMessage, \"permission denied\") ||\n\t\tstrings.Contains(lowerMessage, \"organization has been restricted\") || // groq\n\t\tstrings.Contains(lowerMessage, \"api key not valid\") || // gemini\n\t\tstrings.Contains(lowerMessage, \"api key expired\") || // gemini\n\t\tstrings.Contains(lowerMessage, \"已欠费\") {\n\t\treturn true\n\t}\n\treturn false\n}\n\nfunc ShouldEnableChannel(err error, openAIErr *model.Error) bool {\n\tif !config.AutomaticEnableChannelEnabled {\n\t\treturn false\n\t}\n\tif err != nil {\n\t\treturn false\n\t}\n\tif openAIErr != nil {\n\t\treturn false\n\t}\n\treturn true\n}\n"
},
{
"path": "monitor/metric.go",
"content": "package monitor\n\nimport (\n\t\"github.com/songquanpeng/one-api/common/config\"\n)\n\nvar store = make(map[int][]bool)\nvar metricSuccessChan = make(chan int, config.MetricSuccessChanSize)\nvar metricFailChan = make(chan int, config.MetricFailChanSize)\n\nfunc consumeSuccess(channelId int) {\n\tif len(store[channelId]) > config.MetricQueueSize {\n\t\tstore[channelId] = store[channelId][1:]\n\t}\n\tstore[channelId] = append(store[channelId], true)\n}\n\nfunc consumeFail(channelId int) (bool, float64) {\n\tif len(store[channelId]) > config.MetricQueueSize {\n\t\tstore[channelId] = store[channelId][1:]\n\t}\n\tstore[channelId] = append(store[channelId], false)\n\tsuccessCount := 0\n\tfor _, success := range store[channelId] {\n\t\tif success {\n\t\t\tsuccessCount++\n\t\t}\n\t}\n\tsuccessRate := float64(successCount) / float64(len(store[channelId]))\n\tif len(store[channelId]) < config.MetricQueueSize {\n\t\treturn false, successRate\n\t}\n\tif successRate < config.MetricSuccessRateThreshold {\n\t\tstore[channelId] = make([]bool, 0)\n\t\treturn true, successRate\n\t}\n\treturn false, successRate\n}\n\nfunc metricSuccessConsumer() {\n\tfor {\n\t\tselect {\n\t\tcase channelId := <-metricSuccessChan:\n\t\t\tconsumeSuccess(channelId)\n\t\t}\n\t}\n}\n\nfunc metricFailConsumer() {\n\tfor {\n\t\tselect {\n\t\tcase channelId := <-metricFailChan:\n\t\t\tdisable, successRate := consumeFail(channelId)\n\t\t\tif disable {\n\t\t\t\tgo MetricDisableChannel(channelId, successRate)\n\t\t\t}\n\t\t}\n\t}\n}\n\nfunc init() {\n\tif config.EnableMetric {\n\t\tgo metricSuccessConsumer()\n\t\tgo metricFailConsumer()\n\t}\n}\n\nfunc Emit(channelId int, success bool) {\n\tif !config.EnableMetric {\n\t\treturn\n\t}\n\tgo func() {\n\t\tif success {\n\t\t\tmetricSuccessChan <- channelId\n\t\t} else {\n\t\t\tmetricFailChan <- channelId\n\t\t}\n\t}()\n}\n"
},
{
"path": "one-api.service",
"content": "# File path: /etc/systemd/system/one-api.service\n# sudo systemctl daemon-reload\n# sudo systemctl start one-api\n# sudo systemctl enable one-api\n# sudo systemctl status one-api\n[Unit]\nDescription=One API Service\nAfter=network.target\n\n[Service]\nUser=ubuntu # 注意修改用户名\nWorkingDirectory=/path/to/one-api # 注意修改路径\nExecStart=/path/to/one-api/one-api --port 3000 --log-dir /path/to/one-api/logs # 注意修改路径和端口号\nRestart=always\nRestartSec=5\n\n[Install]\nWantedBy=multi-user.target\n"
},
{
"path": "pull_request_template.md",
"content": "[//]: # (请按照以下格式关联 issue)\n[//]: # (请在提交 PR 前确认所提交的功能可用，需要附上截图，谢谢)\n[//]: # (项目维护者一般仅在周末处理 PR，因此如若未能及时回复希望能理解)\n[//]: # (开发者交流群：910657413)\n[//]: # (请在提交 PR 之前删除上面的注释)\n\nclose #issue_number\n\n我已确认该 PR 已自测通过，相关截图如下：\n（此处放上测试通过的截图，如果不涉及前端改动或从 UI 上无法看出，请放终端启动成功的截图）\n"
},
{
"path": "relay/adaptor/ai360/constants.go",
"content": "package ai360\n\nvar ModelList = []string{\n\t\"360GPT_S2_V9\",\n\t\"embedding-bert-512-v1\",\n\t\"embedding_s1_v1\",\n\t\"semantic_similarity_s1_v1\",\n}\n"
},
{
"path": "relay/adaptor/aiproxy/adaptor.go",
"content": "package aiproxy\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"io\"\n\t\"net/http\"\n)\n\ntype Adaptor struct {\n\tmeta *meta.Meta\n}\n\nfunc (a *Adaptor) Init(meta *meta.Meta) {\n\ta.meta = meta\n}\n\nfunc (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {\n\treturn fmt.Sprintf(\"%s/api/library/ask\", meta.BaseURL), nil\n}\n\nfunc (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {\n\tadaptor.SetupCommonRequestHeader(c, req, meta)\n\treq.Header.Set(\"Authorization\", \"Bearer \"+meta.APIKey)\n\treturn nil\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\taiProxyLibraryRequest := ConvertRequest(*request)\n\taiProxyLibraryRequest.LibraryId = a.meta.Config.LibraryID\n\treturn aiProxyLibraryRequest, nil\n}\n\nfunc (a *Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\treturn request, nil\n}\n\nfunc (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {\n\treturn adaptor.DoRequestHelper(a, c, meta, requestBody)\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tif meta.IsStream {\n\t\terr, usage = StreamHandler(c, resp)\n\t} else {\n\t\terr, usage = Handler(c, resp)\n\t}\n\treturn\n}\n\nfunc (a *Adaptor) GetModelList() []string {\n\treturn ModelList\n}\n\nfunc (a *Adaptor) GetChannelName() string {\n\treturn \"aiproxy\"\n}\n"
},
{
"path": "relay/adaptor/aiproxy/constants.go",
"content": "package aiproxy\n\nimport \"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\nvar ModelList = []string{\"\"}\n\nfunc init() {\n\tModelList = openai.ModelList\n}\n"
},
{
"path": "relay/adaptor/aiproxy/main.go",
"content": "package aiproxy\n\nimport (\n\t\"bufio\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"github.com/songquanpeng/one-api/common/render\"\n\t\"io\"\n\t\"net/http\"\n\t\"strconv\"\n\t\"strings\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/common/random\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/constant\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\n// https://docs.aiproxy.io/dev/library#使用已经定制好的知识库进行对话问答\n\nfunc ConvertRequest(request model.GeneralOpenAIRequest) *LibraryRequest {\n\tquery := \"\"\n\tif len(request.Messages) != 0 {\n\t\tquery = request.Messages[len(request.Messages)-1].StringContent()\n\t}\n\treturn &LibraryRequest{\n\t\tModel: request.Model,\n\t\tStream: request.Stream,\n\t\tQuery: query,\n\t}\n}\n\nfunc aiProxyDocuments2Markdown(documents []LibraryDocument) string {\n\tif len(documents) == 0 {\n\t\treturn \"\"\n\t}\n\tcontent := \"\\n\\n参考文档：\\n\"\n\tfor i, document := range documents {\n\t\tcontent += fmt.Sprintf(\"%d. [%s](%s)\\n\", i+1, document.Title, document.URL)\n\t}\n\treturn content\n}\n\nfunc responseAIProxyLibrary2OpenAI(response *LibraryResponse) *openai.TextResponse {\n\tcontent := response.Answer + aiProxyDocuments2Markdown(response.Documents)\n\tchoice := openai.TextResponseChoice{\n\t\tIndex: 0,\n\t\tMessage: model.Message{\n\t\t\tRole: \"assistant\",\n\t\t\tContent: content,\n\t\t},\n\t\tFinishReason: \"stop\",\n\t}\n\tfullTextResponse := openai.TextResponse{\n\t\tId: fmt.Sprintf(\"chatcmpl-%s\", random.GetUUID()),\n\t\tObject: \"chat.completion\",\n\t\tCreated: helper.GetTimestamp(),\n\t\tChoices: []openai.TextResponseChoice{choice},\n\t}\n\treturn &fullTextResponse\n}\n\nfunc documentsAIProxyLibrary(documents []LibraryDocument) *openai.ChatCompletionsStreamResponse {\n\tvar choice openai.ChatCompletionsStreamResponseChoice\n\tchoice.Delta.Content = aiProxyDocuments2Markdown(documents)\n\tchoice.FinishReason = &constant.StopFinishReason\n\treturn &openai.ChatCompletionsStreamResponse{\n\t\tId: fmt.Sprintf(\"chatcmpl-%s\", random.GetUUID()),\n\t\tObject: \"chat.completion.chunk\",\n\t\tCreated: helper.GetTimestamp(),\n\t\tModel: \"\",\n\t\tChoices: []openai.ChatCompletionsStreamResponseChoice{choice},\n\t}\n}\n\nfunc streamResponseAIProxyLibrary2OpenAI(response *LibraryStreamResponse) *openai.ChatCompletionsStreamResponse {\n\tvar choice openai.ChatCompletionsStreamResponseChoice\n\tchoice.Delta.Content = response.Content\n\treturn &openai.ChatCompletionsStreamResponse{\n\t\tId: fmt.Sprintf(\"chatcmpl-%s\", random.GetUUID()),\n\t\tObject: \"chat.completion.chunk\",\n\t\tCreated: helper.GetTimestamp(),\n\t\tModel: response.Model,\n\t\tChoices: []openai.ChatCompletionsStreamResponseChoice{choice},\n\t}\n}\n\nfunc StreamHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tvar usage model.Usage\n\tvar documents []LibraryDocument\n\tscanner := bufio.NewScanner(resp.Body)\n\tscanner.Split(func(data []byte, atEOF bool) (advance int, token []byte, err error) {\n\t\tif atEOF && len(data) == 0 {\n\t\t\treturn 0, nil, nil\n\t\t}\n\t\tif i := strings.Index(string(data), \"\\n\"); i >= 0 {\n\t\t\treturn i + 1, data[0:i], nil\n\t\t}\n\t\tif atEOF {\n\t\t\treturn len(data), data, nil\n\t\t}\n\t\treturn 0, nil, nil\n\t})\n\n\tcommon.SetEventStreamHeaders(c)\n\n\tfor scanner.Scan() {\n\t\tdata := scanner.Text()\n\t\tif len(data) < 5 || data[:5] != \"data:\" {\n\t\t\tcontinue\n\t\t}\n\t\tdata = data[5:]\n\n\t\tvar AIProxyLibraryResponse LibraryStreamResponse\n\t\terr := json.Unmarshal([]byte(data), &AIProxyLibraryResponse)\n\t\tif err != nil {\n\t\t\tlogger.SysError(\"error unmarshalling stream response: \" + err.Error())\n\t\t\tcontinue\n\t\t}\n\t\tif len(AIProxyLibraryResponse.Documents) != 0 {\n\t\t\tdocuments = AIProxyLibraryResponse.Documents\n\t\t}\n\t\tresponse := streamResponseAIProxyLibrary2OpenAI(&AIProxyLibraryResponse)\n\t\terr = render.ObjectData(c, response)\n\t\tif err != nil {\n\t\t\tlogger.SysError(err.Error())\n\t\t}\n\t}\n\n\tif err := scanner.Err(); err != nil {\n\t\tlogger.SysError(\"error reading stream: \" + err.Error())\n\t}\n\n\tresponse := documentsAIProxyLibrary(documents)\n\terr := render.ObjectData(c, response)\n\tif err != nil {\n\t\tlogger.SysError(err.Error())\n\t}\n\trender.Done(c)\n\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\n\treturn nil, &usage\n}\n\nfunc Handler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tvar AIProxyLibraryResponse LibraryResponse\n\tresponseBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = json.Unmarshal(responseBody, &AIProxyLibraryResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tif AIProxyLibraryResponse.ErrCode != 0 {\n\t\treturn &model.ErrorWithStatusCode{\n\t\t\tError: model.Error{\n\t\t\t\tMessage: AIProxyLibraryResponse.Message,\n\t\t\t\tType: strconv.Itoa(AIProxyLibraryResponse.ErrCode),\n\t\t\t\tCode: AIProxyLibraryResponse.ErrCode,\n\t\t\t},\n\t\t\tStatusCode: resp.StatusCode,\n\t\t}, nil\n\t}\n\tfullTextResponse := responseAIProxyLibrary2OpenAI(&AIProxyLibraryResponse)\n\tjsonResponse, err := json.Marshal(fullTextResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tc.Writer.Header().Set(\"Content-Type\", \"application/json\")\n\tc.Writer.WriteHeader(resp.StatusCode)\n\t_, err = c.Writer.Write(jsonResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"write_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\treturn nil, &fullTextResponse.Usage\n}\n"
},
{
"path": "relay/adaptor/aiproxy/model.go",
"content": "package aiproxy\n\ntype LibraryRequest struct {\n\tModel string `json:\"model\"`\n\tQuery string `json:\"query\"`\n\tLibraryId string `json:\"libraryId\"`\n\tStream bool `json:\"stream\"`\n}\n\ntype LibraryError struct {\n\tErrCode int `json:\"errCode\"`\n\tMessage string `json:\"message\"`\n}\n\ntype LibraryDocument struct {\n\tTitle string `json:\"title\"`\n\tURL string `json:\"url\"`\n}\n\ntype LibraryResponse struct {\n\tSuccess bool `json:\"success\"`\n\tAnswer string `json:\"answer\"`\n\tDocuments []LibraryDocument `json:\"documents\"`\n\tLibraryError\n}\n\ntype LibraryStreamResponse struct {\n\tContent string `json:\"content\"`\n\tFinish bool `json:\"finish\"`\n\tModel string `json:\"model\"`\n\tDocuments []LibraryDocument `json:\"documents\"`\n}\n"
},
{
"path": "relay/adaptor/ali/adaptor.go",
"content": "package ali\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"github.com/songquanpeng/one-api/relay/relaymode\"\n\t\"io\"\n\t\"net/http\"\n)\n\n// https://help.aliyun.com/zh/dashscope/developer-reference/api-details\n\ntype Adaptor struct {\n\tmeta *meta.Meta\n}\n\nfunc (a *Adaptor) Init(meta *meta.Meta) {\n\ta.meta = meta\n}\n\nfunc (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {\n\tfullRequestURL := \"\"\n\tswitch meta.Mode {\n\tcase relaymode.Embeddings:\n\t\tfullRequestURL = fmt.Sprintf(\"%s/api/v1/services/embeddings/text-embedding/text-embedding\", meta.BaseURL)\n\tcase relaymode.ImagesGenerations:\n\t\tfullRequestURL = fmt.Sprintf(\"%s/api/v1/services/aigc/text2image/image-synthesis\", meta.BaseURL)\n\tdefault:\n\t\tfullRequestURL = fmt.Sprintf(\"%s/api/v1/services/aigc/text-generation/generation\", meta.BaseURL)\n\t}\n\n\treturn fullRequestURL, nil\n}\n\nfunc (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {\n\tadaptor.SetupCommonRequestHeader(c, req, meta)\n\tif meta.IsStream {\n\t\treq.Header.Set(\"Accept\", \"text/event-stream\")\n\t\treq.Header.Set(\"X-DashScope-SSE\", \"enable\")\n\t}\n\treq.Header.Set(\"Authorization\", \"Bearer \"+meta.APIKey)\n\n\tif meta.Mode == relaymode.ImagesGenerations {\n\t\treq.Header.Set(\"X-DashScope-Async\", \"enable\")\n\t}\n\tif a.meta.Config.Plugin != \"\" {\n\t\treq.Header.Set(\"X-DashScope-Plugin\", a.meta.Config.Plugin)\n\t}\n\treturn nil\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\tswitch relayMode {\n\tcase relaymode.Embeddings:\n\t\taliEmbeddingRequest := ConvertEmbeddingRequest(*request)\n\t\treturn aliEmbeddingRequest, nil\n\tdefault:\n\t\taliRequest := ConvertRequest(*request)\n\t\treturn aliRequest, nil\n\t}\n}\n\nfunc (a *Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\n\taliRequest := ConvertImageRequest(*request)\n\treturn aliRequest, nil\n}\n\nfunc (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {\n\treturn adaptor.DoRequestHelper(a, c, meta, requestBody)\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tif meta.IsStream {\n\t\terr, usage = StreamHandler(c, resp)\n\t} else {\n\t\tswitch meta.Mode {\n\t\tcase relaymode.Embeddings:\n\t\t\terr, usage = EmbeddingHandler(c, resp)\n\t\tcase relaymode.ImagesGenerations:\n\t\t\terr, usage = ImageHandler(c, resp)\n\t\tdefault:\n\t\t\terr, usage = Handler(c, resp)\n\t\t}\n\t}\n\treturn\n}\n\nfunc (a *Adaptor) GetModelList() []string {\n\treturn ModelList\n}\n\nfunc (a *Adaptor) GetChannelName() string {\n\treturn \"ali\"\n}\n"
},
{
"path": "relay/adaptor/ali/constants.go",
"content": "package ali\n\nvar ModelList = []string{\n\t\"qwen-turbo\", \"qwen-turbo-latest\",\n\t\"qwen-plus\", \"qwen-plus-latest\",\n\t\"qwen-max\", \"qwen-max-latest\",\n\t\"qwen-max-longcontext\",\n\t\"qwen-vl-max\", \"qwen-vl-max-latest\", \"qwen-vl-plus\", \"qwen-vl-plus-latest\",\n\t\"qwen-vl-ocr\", \"qwen-vl-ocr-latest\",\n\t\"qwen-audio-turbo\",\n\t\"qwen-math-plus\", \"qwen-math-plus-latest\", \"qwen-math-turbo\", \"qwen-math-turbo-latest\",\n\t\"qwen-coder-plus\", \"qwen-coder-plus-latest\", \"qwen-coder-turbo\", \"qwen-coder-turbo-latest\",\n\t\"qwq-32b-preview\", \"qwen2.5-72b-instruct\", \"qwen2.5-32b-instruct\", \"qwen2.5-14b-instruct\", \"qwen2.5-7b-instruct\", \"qwen2.5-3b-instruct\", \"qwen2.5-1.5b-instruct\", \"qwen2.5-0.5b-instruct\",\n\t\"qwen2-72b-instruct\", \"qwen2-57b-a14b-instruct\", \"qwen2-7b-instruct\", \"qwen2-1.5b-instruct\", \"qwen2-0.5b-instruct\",\n\t\"qwen1.5-110b-chat\", \"qwen1.5-72b-chat\", \"qwen1.5-32b-chat\", \"qwen1.5-14b-chat\", \"qwen1.5-7b-chat\", \"qwen1.5-1.8b-chat\", \"qwen1.5-0.5b-chat\",\n\t\"qwen-72b-chat\", \"qwen-14b-chat\", \"qwen-7b-chat\", \"qwen-1.8b-chat\", \"qwen-1.8b-longcontext-chat\",\n\t\"qvq-72b-preview\",\n\t\"qwen2.5-vl-72b-instruct\", \"qwen2.5-vl-7b-instruct\", \"qwen2.5-vl-2b-instruct\", \"qwen2.5-vl-1b-instruct\", \"qwen2.5-vl-0.5b-instruct\",\n\t\"qwen2-vl-7b-instruct\", \"qwen2-vl-2b-instruct\", \"qwen-vl-v1\", \"qwen-vl-chat-v1\",\n\t\"qwen2-audio-instruct\", \"qwen-audio-chat\",\n\t\"qwen2.5-math-72b-instruct\", \"qwen2.5-math-7b-instruct\", \"qwen2.5-math-1.5b-instruct\", \"qwen2-math-72b-instruct\", \"qwen2-math-7b-instruct\", \"qwen2-math-1.5b-instruct\",\n\t\"qwen2.5-coder-32b-instruct\", \"qwen2.5-coder-14b-instruct\", \"qwen2.5-coder-7b-instruct\", \"qwen2.5-coder-3b-instruct\", \"qwen2.5-coder-1.5b-instruct\", \"qwen2.5-coder-0.5b-instruct\",\n\t\"text-embedding-v1\", \"text-embedding-v3\", \"text-embedding-v2\", \"text-embedding-async-v2\", \"text-embedding-async-v1\",\n\t\"ali-stable-diffusion-xl\", \"ali-stable-diffusion-v1.5\", \"wanx-v1\",\n\t\"qwen-mt-plus\", \"qwen-mt-turbo\",\n\t\"deepseek-r1\", \"deepseek-v3\", \"deepseek-r1-distill-qwen-1.5b\", \"deepseek-r1-distill-qwen-7b\", \"deepseek-r1-distill-qwen-14b\", \"deepseek-r1-distill-qwen-32b\", \"deepseek-r1-distill-llama-8b\", \"deepseek-r1-distill-llama-70b\",\n}\n"
},
{
"path": "relay/adaptor/ali/image.go",
"content": "package ali\n\nimport (\n\t\"encoding/base64\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n\t\"time\"\n)\n\nfunc ImageHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tapiKey := c.Request.Header.Get(\"Authorization\")\n\tapiKey = strings.TrimPrefix(apiKey, \"Bearer \")\n\tresponseFormat := c.GetString(\"response_format\")\n\n\tvar aliTaskResponse TaskResponse\n\tresponseBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = json.Unmarshal(responseBody, &aliTaskResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\n\tif aliTaskResponse.Message != \"\" {\n\t\tlogger.SysError(\"aliAsyncTask err: \" + string(responseBody))\n\t\treturn openai.ErrorWrapper(errors.New(aliTaskResponse.Message), \"ali_async_task_failed\", http.StatusInternalServerError), nil\n\t}\n\n\taliResponse, _, err := asyncTaskWait(aliTaskResponse.Output.TaskId, apiKey)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"ali_async_task_wait_failed\", http.StatusInternalServerError), nil\n\t}\n\n\tif aliResponse.Output.TaskStatus != \"SUCCEEDED\" {\n\t\treturn &model.ErrorWithStatusCode{\n\t\t\tError: model.Error{\n\t\t\t\tMessage: aliResponse.Output.Message,\n\t\t\t\tType: \"ali_error\",\n\t\t\t\tParam: \"\",\n\t\t\t\tCode: aliResponse.Output.Code,\n\t\t\t},\n\t\t\tStatusCode: resp.StatusCode,\n\t\t}, nil\n\t}\n\n\tfullTextResponse := responseAli2OpenAIImage(aliResponse, responseFormat)\n\tjsonResponse, err := json.Marshal(fullTextResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tc.Writer.Header().Set(\"Content-Type\", \"application/json\")\n\tc.Writer.WriteHeader(resp.StatusCode)\n\t_, err = c.Writer.Write(jsonResponse)\n\treturn nil, nil\n}\n\nfunc asyncTask(taskID string, key string) (*TaskResponse, error, []byte) {\n\turl := fmt.Sprintf(\"https://dashscope.aliyuncs.com/api/v1/tasks/%s\", taskID)\n\n\tvar aliResponse TaskResponse\n\n\treq, err := http.NewRequest(\"GET\", url, nil)\n\tif err != nil {\n\t\treturn &aliResponse, err, nil\n\t}\n\n\treq.Header.Set(\"Authorization\", \"Bearer \"+key)\n\n\tclient := &http.Client{}\n\tresp, err := client.Do(req)\n\tif err != nil {\n\t\tlogger.SysError(\"aliAsyncTask client.Do err: \" + err.Error())\n\t\treturn &aliResponse, err, nil\n\t}\n\tdefer resp.Body.Close()\n\n\tresponseBody, err := io.ReadAll(resp.Body)\n\n\tvar response TaskResponse\n\terr = json.Unmarshal(responseBody, &response)\n\tif err != nil {\n\t\tlogger.SysError(\"aliAsyncTask NewDecoder err: \" + err.Error())\n\t\treturn &aliResponse, err, nil\n\t}\n\n\treturn &response, nil, responseBody\n}\n\nfunc asyncTaskWait(taskID string, key string) (*TaskResponse, []byte, error) {\n\twaitSeconds := 2\n\tstep := 0\n\tmaxStep := 20\n\n\tvar taskResponse TaskResponse\n\tvar responseBody []byte\n\n\tfor {\n\t\tstep++\n\t\trsp, err, body := asyncTask(taskID, key)\n\t\tresponseBody = body\n\t\tif err != nil {\n\t\t\treturn &taskResponse, responseBody, err\n\t\t}\n\n\t\tif rsp.Output.TaskStatus == \"\" {\n\t\t\treturn &taskResponse, responseBody, nil\n\t\t}\n\n\t\tswitch rsp.Output.TaskStatus {\n\t\tcase \"FAILED\":\n\t\t\tfallthrough\n\t\tcase \"CANCELED\":\n\t\t\tfallthrough\n\t\tcase \"SUCCEEDED\":\n\t\t\tfallthrough\n\t\tcase \"UNKNOWN\":\n\t\t\treturn rsp, responseBody, nil\n\t\t}\n\t\tif step >= maxStep {\n\t\t\tbreak\n\t\t}\n\t\ttime.Sleep(time.Duration(waitSeconds) * time.Second)\n\t}\n\n\treturn nil, nil, fmt.Errorf(\"aliAsyncTaskWait timeout\")\n}\n\nfunc responseAli2OpenAIImage(response *TaskResponse, responseFormat string) *openai.ImageResponse {\n\timageResponse := openai.ImageResponse{\n\t\tCreated: helper.GetTimestamp(),\n\t}\n\n\tfor _, data := range response.Output.Results {\n\t\tvar b64Json string\n\t\tif responseFormat == \"b64_json\" {\n\t\t\t// 读取 data.Url 的图片数据并转存到 b64Json\n\t\t\timageData, err := getImageData(data.Url)\n\t\t\tif err != nil {\n\t\t\t\t// 处理获取图片数据失败的情况\n\t\t\t\tlogger.SysError(\"getImageData Error getting image data: \" + err.Error())\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\t// 将图片数据转为 Base64 编码的字符串\n\t\t\tb64Json = Base64Encode(imageData)\n\t\t} else {\n\t\t\t// 如果 responseFormat 不是 \"b64_json\"，则直接使用 data.B64Image\n\t\t\tb64Json = data.B64Image\n\t\t}\n\n\t\timageResponse.Data = append(imageResponse.Data, openai.ImageData{\n\t\t\tUrl: data.Url,\n\t\t\tB64Json: b64Json,\n\t\t\tRevisedPrompt: \"\",\n\t\t})\n\t}\n\treturn &imageResponse\n}\n\nfunc getImageData(url string) ([]byte, error) {\n\tresponse, err := http.Get(url)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer response.Body.Close()\n\n\timageData, err := io.ReadAll(response.Body)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\n\treturn imageData, nil\n}\n\nfunc Base64Encode(data []byte) string {\n\tb64Json := base64.StdEncoding.EncodeToString(data)\n\treturn b64Json\n}\n"
},
{
"path": "relay/adaptor/ali/main.go",
"content": "package ali\n\nimport (\n\t\"bufio\"\n\t\"encoding/json\"\n\t\"github.com/songquanpeng/one-api/common/ctxkey\"\n\t\"github.com/songquanpeng/one-api/common/render\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\n// https://help.aliyun.com/document_detail/613695.html?spm=a2c4g.2399480.0.0.1adb778fAdzP9w#341800c0f8w0r\n\nconst EnableSearchModelSuffix = \"-internet\"\n\nfunc ConvertRequest(request model.GeneralOpenAIRequest) *ChatRequest {\n\tmessages := make([]Message, 0, len(request.Messages))\n\tfor i := 0; i < len(request.Messages); i++ {\n\t\tmessage := request.Messages[i]\n\t\tmessages = append(messages, Message{\n\t\t\tContent: message.StringContent(),\n\t\t\tRole: strings.ToLower(message.Role),\n\t\t})\n\t}\n\tenableSearch := false\n\taliModel := request.Model\n\tif strings.HasSuffix(aliModel, EnableSearchModelSuffix) {\n\t\tenableSearch = true\n\t\taliModel = strings.TrimSuffix(aliModel, EnableSearchModelSuffix)\n\t}\n\trequest.TopP = helper.Float64PtrMax(request.TopP, 0.9999)\n\treturn &ChatRequest{\n\t\tModel: aliModel,\n\t\tInput: Input{\n\t\t\tMessages: messages,\n\t\t},\n\t\tParameters: Parameters{\n\t\t\tEnableSearch: enableSearch,\n\t\t\tIncrementalOutput: request.Stream,\n\t\t\tSeed: uint64(request.Seed),\n\t\t\tMaxTokens: request.MaxTokens,\n\t\t\tTemperature: request.Temperature,\n\t\t\tTopP: request.TopP,\n\t\t\tTopK: request.TopK,\n\t\t\tResultFormat: \"message\",\n\t\t\tTools: request.Tools,\n\t\t},\n\t}\n}\n\nfunc ConvertEmbeddingRequest(request model.GeneralOpenAIRequest) *EmbeddingRequest {\n\treturn &EmbeddingRequest{\n\t\tModel: request.Model,\n\t\tInput: struct {\n\t\t\tTexts []string `json:\"texts\"`\n\t\t}{\n\t\t\tTexts: request.ParseInput(),\n\t\t},\n\t}\n}\n\nfunc ConvertImageRequest(request model.ImageRequest) *ImageRequest {\n\tvar imageRequest ImageRequest\n\timageRequest.Input.Prompt = request.Prompt\n\timageRequest.Model = request.Model\n\timageRequest.Parameters.Size = strings.Replace(request.Size, \"x\", \"*\", -1)\n\timageRequest.Parameters.N = request.N\n\timageRequest.ResponseFormat = request.ResponseFormat\n\n\treturn &imageRequest\n}\n\nfunc EmbeddingHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tvar aliResponse EmbeddingResponse\n\terr := json.NewDecoder(resp.Body).Decode(&aliResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\n\tif aliResponse.Code != \"\" {\n\t\treturn &model.ErrorWithStatusCode{\n\t\t\tError: model.Error{\n\t\t\t\tMessage: aliResponse.Message,\n\t\t\t\tType: aliResponse.Code,\n\t\t\t\tParam: aliResponse.RequestId,\n\t\t\t\tCode: aliResponse.Code,\n\t\t\t},\n\t\t\tStatusCode: resp.StatusCode,\n\t\t}, nil\n\t}\n\trequestModel := c.GetString(ctxkey.RequestModel)\n\tfullTextResponse := embeddingResponseAli2OpenAI(&aliResponse)\n\tfullTextResponse.Model = requestModel\n\tjsonResponse, err := json.Marshal(fullTextResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tc.Writer.Header().Set(\"Content-Type\", \"application/json\")\n\tc.Writer.WriteHeader(resp.StatusCode)\n\t_, err = c.Writer.Write(jsonResponse)\n\treturn nil, &fullTextResponse.Usage\n}\n\nfunc embeddingResponseAli2OpenAI(response *EmbeddingResponse) *openai.EmbeddingResponse {\n\topenAIEmbeddingResponse := openai.EmbeddingResponse{\n\t\tObject: \"list\",\n\t\tData: make([]openai.EmbeddingResponseItem, 0, len(response.Output.Embeddings)),\n\t\tModel: \"text-embedding-v1\",\n\t\tUsage: model.Usage{TotalTokens: response.Usage.TotalTokens},\n\t}\n\n\tfor _, item := range response.Output.Embeddings {\n\t\topenAIEmbeddingResponse.Data = append(openAIEmbeddingResponse.Data, openai.EmbeddingResponseItem{\n\t\t\tObject: `embedding`,\n\t\t\tIndex: item.TextIndex,\n\t\t\tEmbedding: item.Embedding,\n\t\t})\n\t}\n\treturn &openAIEmbeddingResponse\n}\n\nfunc responseAli2OpenAI(response *ChatResponse) *openai.TextResponse {\n\tfullTextResponse := openai.TextResponse{\n\t\tId: response.RequestId,\n\t\tObject: \"chat.completion\",\n\t\tCreated: helper.GetTimestamp(),\n\t\tChoices: response.Output.Choices,\n\t\tUsage: model.Usage{\n\t\t\tPromptTokens: response.Usage.InputTokens,\n\t\t\tCompletionTokens: response.Usage.OutputTokens,\n\t\t\tTotalTokens: response.Usage.InputTokens + response.Usage.OutputTokens,\n\t\t},\n\t}\n\treturn &fullTextResponse\n}\n\nfunc streamResponseAli2OpenAI(aliResponse *ChatResponse) *openai.ChatCompletionsStreamResponse {\n\tif len(aliResponse.Output.Choices) == 0 {\n\t\treturn nil\n\t}\n\taliChoice := aliResponse.Output.Choices[0]\n\tvar choice openai.ChatCompletionsStreamResponseChoice\n\tchoice.Delta = aliChoice.Message\n\tif aliChoice.FinishReason != \"null\" {\n\t\tfinishReason := aliChoice.FinishReason\n\t\tchoice.FinishReason = &finishReason\n\t}\n\tresponse := openai.ChatCompletionsStreamResponse{\n\t\tId: aliResponse.RequestId,\n\t\tObject: \"chat.completion.chunk\",\n\t\tCreated: helper.GetTimestamp(),\n\t\tModel: \"qwen\",\n\t\tChoices: []openai.ChatCompletionsStreamResponseChoice{choice},\n\t}\n\treturn &response\n}\n\nfunc StreamHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tvar usage model.Usage\n\tscanner := bufio.NewScanner(resp.Body)\n\tscanner.Split(func(data []byte, atEOF bool) (advance int, token []byte, err error) {\n\t\tif atEOF && len(data) == 0 {\n\t\t\treturn 0, nil, nil\n\t\t}\n\t\tif i := strings.Index(string(data), \"\\n\"); i >= 0 {\n\t\t\treturn i + 1, data[0:i], nil\n\t\t}\n\t\tif atEOF {\n\t\t\treturn len(data), data, nil\n\t\t}\n\t\treturn 0, nil, nil\n\t})\n\n\tcommon.SetEventStreamHeaders(c)\n\n\tfor scanner.Scan() {\n\t\tdata := scanner.Text()\n\t\tif len(data) < 5 || data[:5] != \"data:\" {\n\t\t\tcontinue\n\t\t}\n\t\tdata = data[5:]\n\n\t\tvar aliResponse ChatResponse\n\t\terr := json.Unmarshal([]byte(data), &aliResponse)\n\t\tif err != nil {\n\t\t\tlogger.SysError(\"error unmarshalling stream response: \" + err.Error())\n\t\t\tcontinue\n\t\t}\n\t\tif aliResponse.Usage.OutputTokens != 0 {\n\t\t\tusage.PromptTokens = aliResponse.Usage.InputTokens\n\t\t\tusage.CompletionTokens = aliResponse.Usage.OutputTokens\n\t\t\tusage.TotalTokens = aliResponse.Usage.InputTokens + aliResponse.Usage.OutputTokens\n\t\t}\n\t\tresponse := streamResponseAli2OpenAI(&aliResponse)\n\t\tif response == nil {\n\t\t\tcontinue\n\t\t}\n\t\terr = render.ObjectData(c, response)\n\t\tif err != nil {\n\t\t\tlogger.SysError(err.Error())\n\t\t}\n\t}\n\n\tif err := scanner.Err(); err != nil {\n\t\tlogger.SysError(\"error reading stream: \" + err.Error())\n\t}\n\n\trender.Done(c)\n\n\terr := resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\treturn nil, &usage\n}\n\nfunc Handler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tctx := c.Request.Context()\n\tvar aliResponse ChatResponse\n\tresponseBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tlogger.Debugf(ctx, \"response body: %s\\n\", responseBody)\n\terr = json.Unmarshal(responseBody, &aliResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tif aliResponse.Code != \"\" {\n\t\treturn &model.ErrorWithStatusCode{\n\t\t\tError: model.Error{\n\t\t\t\tMessage: aliResponse.Message,\n\t\t\t\tType: aliResponse.Code,\n\t\t\t\tParam: aliResponse.RequestId,\n\t\t\t\tCode: aliResponse.Code,\n\t\t\t},\n\t\t\tStatusCode: resp.StatusCode,\n\t\t}, nil\n\t}\n\tfullTextResponse := responseAli2OpenAI(&aliResponse)\n\tfullTextResponse.Model = \"qwen\"\n\tjsonResponse, err := json.Marshal(fullTextResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tc.Writer.Header().Set(\"Content-Type\", \"application/json\")\n\tc.Writer.WriteHeader(resp.StatusCode)\n\t_, err = c.Writer.Write(jsonResponse)\n\treturn nil, &fullTextResponse.Usage\n}\n"
},
{
"path": "relay/adaptor/ali/model.go",
"content": "package ali\n\nimport (\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\ntype Message struct {\n\tContent string `json:\"content\"`\n\tRole string `json:\"role\"`\n}\n\ntype Input struct {\n\t//Prompt string `json:\"prompt\"`\n\tMessages []Message `json:\"messages\"`\n}\n\ntype Parameters struct {\n\tTopP *float64 `json:\"top_p,omitempty\"`\n\tTopK int `json:\"top_k,omitempty\"`\n\tSeed uint64 `json:\"seed,omitempty\"`\n\tEnableSearch bool `json:\"enable_search,omitempty\"`\n\tIncrementalOutput bool `json:\"incremental_output,omitempty\"`\n\tMaxTokens int `json:\"max_tokens,omitempty\"`\n\tTemperature *float64 `json:\"temperature,omitempty\"`\n\tResultFormat string `json:\"result_format,omitempty\"`\n\tTools []model.Tool `json:\"tools,omitempty\"`\n}\n\ntype ChatRequest struct {\n\tModel string `json:\"model\"`\n\tInput Input `json:\"input\"`\n\tParameters Parameters `json:\"parameters,omitempty\"`\n}\n\ntype ImageRequest struct {\n\tModel string `json:\"model\"`\n\tInput struct {\n\t\tPrompt string `json:\"prompt\"`\n\t\tNegativePrompt string `json:\"negative_prompt,omitempty\"`\n\t} `json:\"input\"`\n\tParameters struct {\n\t\tSize string `json:\"size,omitempty\"`\n\t\tN int `json:\"n,omitempty\"`\n\t\tSteps string `json:\"steps,omitempty\"`\n\t\tScale string `json:\"scale,omitempty\"`\n\t} `json:\"parameters,omitempty\"`\n\tResponseFormat string `json:\"response_format,omitempty\"`\n}\n\ntype TaskResponse struct {\n\tStatusCode int `json:\"status_code,omitempty\"`\n\tRequestId string `json:\"request_id,omitempty\"`\n\tCode string `json:\"code,omitempty\"`\n\tMessage string `json:\"message,omitempty\"`\n\tOutput struct {\n\t\tTaskId string `json:\"task_id,omitempty\"`\n\t\tTaskStatus string `json:\"task_status,omitempty\"`\n\t\tCode string `json:\"code,omitempty\"`\n\t\tMessage string `json:\"message,omitempty\"`\n\t\tResults []struct {\n\t\t\tB64Image string `json:\"b64_image,omitempty\"`\n\t\t\tUrl string `json:\"url,omitempty\"`\n\t\t\tCode string `json:\"code,omitempty\"`\n\t\t\tMessage string `json:\"message,omitempty\"`\n\t\t} `json:\"results,omitempty\"`\n\t\tTaskMetrics struct {\n\t\t\tTotal int `json:\"TOTAL,omitempty\"`\n\t\t\tSucceeded int `json:\"SUCCEEDED,omitempty\"`\n\t\t\tFailed int `json:\"FAILED,omitempty\"`\n\t\t} `json:\"task_metrics,omitempty\"`\n\t} `json:\"output,omitempty\"`\n\tUsage Usage `json:\"usage\"`\n}\n\ntype Header struct {\n\tAction string `json:\"action,omitempty\"`\n\tStreaming string `json:\"streaming,omitempty\"`\n\tTaskID string `json:\"task_id,omitempty\"`\n\tEvent string `json:\"event,omitempty\"`\n\tErrorCode string `json:\"error_code,omitempty\"`\n\tErrorMessage string `json:\"error_message,omitempty\"`\n\tAttributes any `json:\"attributes,omitempty\"`\n}\n\ntype Payload struct {\n\tModel string `json:\"model,omitempty\"`\n\tTask string `json:\"task,omitempty\"`\n\tTaskGroup string `json:\"task_group,omitempty\"`\n\tFunction string `json:\"function,omitempty\"`\n\tParameters struct {\n\t\tSampleRate int `json:\"sample_rate,omitempty\"`\n\t\tRate float64 `json:\"rate,omitempty\"`\n\t\tFormat string `json:\"format,omitempty\"`\n\t} `json:\"parameters,omitempty\"`\n\tInput struct {\n\t\tText string `json:\"text,omitempty\"`\n\t} `json:\"input,omitempty\"`\n\tUsage struct {\n\t\tCharacters int `json:\"characters,omitempty\"`\n\t} `json:\"usage,omitempty\"`\n}\n\ntype WSSMessage struct {\n\tHeader Header `json:\"header,omitempty\"`\n\tPayload Payload `json:\"payload,omitempty\"`\n}\n\ntype EmbeddingRequest struct {\n\tModel string `json:\"model\"`\n\tInput struct {\n\t\tTexts []string `json:\"texts\"`\n\t} `json:\"input\"`\n\tParameters *struct {\n\t\tTextType string `json:\"text_type,omitempty\"`\n\t} `json:\"parameters,omitempty\"`\n}\n\ntype Embedding struct {\n\tEmbedding []float64 `json:\"embedding\"`\n\tTextIndex int `json:\"text_index\"`\n}\n\ntype EmbeddingResponse struct {\n\tOutput struct {\n\t\tEmbeddings []Embedding `json:\"embeddings\"`\n\t} `json:\"output\"`\n\tUsage Usage `json:\"usage\"`\n\tError\n}\n\ntype Error struct {\n\tCode string `json:\"code\"`\n\tMessage string `json:\"message\"`\n\tRequestId string `json:\"request_id\"`\n}\n\ntype Usage struct {\n\tInputTokens int `json:\"input_tokens\"`\n\tOutputTokens int `json:\"output_tokens\"`\n\tTotalTokens int `json:\"total_tokens\"`\n}\n\ntype Output struct {\n\t//Text string `json:\"text\"`\n\t//FinishReason string `json:\"finish_reason\"`\n\tChoices []openai.TextResponseChoice `json:\"choices\"`\n}\n\ntype ChatResponse struct {\n\tOutput Output `json:\"output\"`\n\tUsage Usage `json:\"usage\"`\n\tError\n}\n"
},
{
"path": "relay/adaptor/alibailian/constants.go",
"content": "package alibailian\n\n// https://help.aliyun.com/zh/model-studio/getting-started/models\n\nvar ModelList = []string{\n\t\"qwen-turbo\",\n\t\"qwen-plus\",\n\t\"qwen-long\",\n\t\"qwen-max\",\n\t\"qwen-coder-plus\",\n\t\"qwen-coder-plus-latest\",\n\t\"qwen-coder-turbo\",\n\t\"qwen-coder-turbo-latest\",\n\t\"qwen-mt-plus\",\n\t\"qwen-mt-turbo\",\n\t\"qwq-32b-preview\",\n\n\t\"deepseek-r1\",\n\t\"deepseek-v3\",\n}\n"
},
{
"path": "relay/adaptor/alibailian/main.go",
"content": "package alibailian\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/relaymode\"\n)\n\nfunc GetRequestURL(meta *meta.Meta) (string, error) {\n\tswitch meta.Mode {\n\tcase relaymode.ChatCompletions:\n\t\treturn fmt.Sprintf(\"%s/compatible-mode/v1/chat/completions\", meta.BaseURL), nil\n\tcase relaymode.Embeddings:\n\t\treturn fmt.Sprintf(\"%s/compatible-mode/v1/embeddings\", meta.BaseURL), nil\n\tdefault:\n\t}\n\treturn \"\", fmt.Errorf(\"unsupported relay mode %d for ali bailian\", meta.Mode)\n}\n"
},
{
"path": "relay/adaptor/anthropic/adaptor.go",
"content": "package anthropic\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\ntype Adaptor struct {\n}\n\nfunc (a *Adaptor) Init(meta *meta.Meta) {\n\n}\n\nfunc (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {\n\treturn fmt.Sprintf(\"%s/v1/messages\", meta.BaseURL), nil\n}\n\nfunc (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {\n\tadaptor.SetupCommonRequestHeader(c, req, meta)\n\treq.Header.Set(\"x-api-key\", meta.APIKey)\n\tanthropicVersion := c.Request.Header.Get(\"anthropic-version\")\n\tif anthropicVersion == \"\" {\n\t\tanthropicVersion = \"2023-06-01\"\n\t}\n\treq.Header.Set(\"anthropic-version\", anthropicVersion)\n\treq.Header.Set(\"anthropic-beta\", \"messages-2023-12-15\")\n\n\t// https://x.com/alexalbert__/status/1812921642143900036\n\t// claude-3-5-sonnet can support 8k context\n\tif strings.HasPrefix(meta.ActualModelName, \"claude-3-5-sonnet\") {\n\t\treq.Header.Set(\"anthropic-beta\", \"max-tokens-3-5-sonnet-2024-07-15\")\n\t}\n\n\treturn nil\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\treturn ConvertRequest(*request), nil\n}\n\nfunc (a *Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\treturn request, nil\n}\n\nfunc (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {\n\treturn adaptor.DoRequestHelper(a, c, meta, requestBody)\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tif meta.IsStream {\n\t\terr, usage = StreamHandler(c, resp)\n\t} else {\n\t\terr, usage = Handler(c, resp, meta.PromptTokens, meta.ActualModelName)\n\t}\n\treturn\n}\n\nfunc (a *Adaptor) GetModelList() []string {\n\treturn ModelList\n}\n\nfunc (a *Adaptor) GetChannelName() string {\n\treturn \"anthropic\"\n}\n"
},
{
"path": "relay/adaptor/anthropic/constants.go",
"content": "package anthropic\n\nvar ModelList = []string{\n\t\"claude-instant-1.2\", \"claude-2.0\", \"claude-2.1\",\n\t\"claude-3-haiku-20240307\",\n\t\"claude-3-5-haiku-20241022\",\n\t\"claude-3-5-haiku-latest\",\n\t\"claude-3-sonnet-20240229\",\n\t\"claude-3-opus-20240229\",\n\t\"claude-3-5-sonnet-20240620\",\n\t\"claude-3-5-sonnet-20241022\",\n\t\"claude-3-5-sonnet-latest\",\n}\n"
},
{
"path": "relay/adaptor/anthropic/main.go",
"content": "package anthropic\n\nimport (\n\t\"bufio\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"github.com/songquanpeng/one-api/common/render\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/image\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\nfunc stopReasonClaude2OpenAI(reason *string) string {\n\tif reason == nil {\n\t\treturn \"\"\n\t}\n\tswitch *reason {\n\tcase \"end_turn\":\n\t\treturn \"stop\"\n\tcase \"stop_sequence\":\n\t\treturn \"stop\"\n\tcase \"max_tokens\":\n\t\treturn \"length\"\n\tcase \"tool_use\":\n\t\treturn \"tool_calls\"\n\tdefault:\n\t\treturn *reason\n\t}\n}\n\nfunc ConvertRequest(textRequest model.GeneralOpenAIRequest) *Request {\n\tclaudeTools := make([]Tool, 0, len(textRequest.Tools))\n\n\tfor _, tool := range textRequest.Tools {\n\t\tif params, ok := tool.Function.Parameters.(map[string]any); ok {\n\t\t\tclaudeTools = append(claudeTools, Tool{\n\t\t\t\tName: tool.Function.Name,\n\t\t\t\tDescription: tool.Function.Description,\n\t\t\t\tInputSchema: InputSchema{\n\t\t\t\t\tType: params[\"type\"].(string),\n\t\t\t\t\tProperties: params[\"properties\"],\n\t\t\t\t\tRequired: params[\"required\"],\n\t\t\t\t},\n\t\t\t})\n\t\t}\n\t}\n\n\tclaudeRequest := Request{\n\t\tModel: textRequest.Model,\n\t\tMaxTokens: textRequest.MaxTokens,\n\t\tTemperature: textRequest.Temperature,\n\t\tTopP: textRequest.TopP,\n\t\tTopK: textRequest.TopK,\n\t\tStream: textRequest.Stream,\n\t\tTools: claudeTools,\n\t}\n\tif len(claudeTools) > 0 {\n\t\tclaudeToolChoice := struct {\n\t\t\tType string `json:\"type\"`\n\t\t\tName string `json:\"name,omitempty\"`\n\t\t}{Type: \"auto\"} // default value https://docs.anthropic.com/en/docs/build-with-claude/tool-use#controlling-claudes-output\n\t\tif choice, ok := textRequest.ToolChoice.(map[string]any); ok {\n\t\t\tif function, ok := choice[\"function\"].(map[string]any); ok {\n\t\t\t\tclaudeToolChoice.Type = \"tool\"\n\t\t\t\tclaudeToolChoice.Name = function[\"name\"].(string)\n\t\t\t}\n\t\t} else if toolChoiceType, ok := textRequest.ToolChoice.(string); ok {\n\t\t\tif toolChoiceType == \"any\" {\n\t\t\t\tclaudeToolChoice.Type = toolChoiceType\n\t\t\t}\n\t\t}\n\t\tclaudeRequest.ToolChoice = claudeToolChoice\n\t}\n\tif claudeRequest.MaxTokens == 0 {\n\t\tclaudeRequest.MaxTokens = 4096\n\t}\n\t// legacy model name mapping\n\tif claudeRequest.Model == \"claude-instant-1\" {\n\t\tclaudeRequest.Model = \"claude-instant-1.1\"\n\t} else if claudeRequest.Model == \"claude-2\" {\n\t\tclaudeRequest.Model = \"claude-2.1\"\n\t}\n\tfor _, message := range textRequest.Messages {\n\t\tif message.Role == \"system\" && claudeRequest.System == \"\" {\n\t\t\tclaudeRequest.System = message.StringContent()\n\t\t\tcontinue\n\t\t}\n\t\tclaudeMessage := Message{\n\t\t\tRole: message.Role,\n\t\t}\n\t\tvar content Content\n\t\tif message.IsStringContent() {\n\t\t\tcontent.Type = \"text\"\n\t\t\tcontent.Text = message.StringContent()\n\t\t\tif message.Role == \"tool\" {\n\t\t\t\tclaudeMessage.Role = \"user\"\n\t\t\t\tcontent.Type = \"tool_result\"\n\t\t\t\tcontent.Content = content.Text\n\t\t\t\tcontent.Text = \"\"\n\t\t\t\tcontent.ToolUseId = message.ToolCallId\n\t\t\t}\n\t\t\tclaudeMessage.Content = append(claudeMessage.Content, content)\n\t\t\tfor i := range message.ToolCalls {\n\t\t\t\tinputParam := make(map[string]any)\n\t\t\t\t_ = json.Unmarshal([]byte(message.ToolCalls[i].Function.Arguments.(string)), &inputParam)\n\t\t\t\tclaudeMessage.Content = append(claudeMessage.Content, Content{\n\t\t\t\t\tType: \"tool_use\",\n\t\t\t\t\tId: message.ToolCalls[i].Id,\n\t\t\t\t\tName: message.ToolCalls[i].Function.Name,\n\t\t\t\t\tInput: inputParam,\n\t\t\t\t})\n\t\t\t}\n\t\t\tclaudeRequest.Messages = append(claudeRequest.Messages, claudeMessage)\n\t\t\tcontinue\n\t\t}\n\t\tvar contents []Content\n\t\topenaiContent := message.ParseContent()\n\t\tfor _, part := range openaiContent {\n\t\t\tvar content Content\n\t\t\tif part.Type == model.ContentTypeText {\n\t\t\t\tcontent.Type = \"text\"\n\t\t\t\tcontent.Text = part.Text\n\t\t\t} else if part.Type == model.ContentTypeImageURL {\n\t\t\t\tcontent.Type = \"image\"\n\t\t\t\tcontent.Source = &ImageSource{\n\t\t\t\t\tType: \"base64\",\n\t\t\t\t}\n\t\t\t\tmimeType, data, _ := image.GetImageFromUrl(part.ImageURL.Url)\n\t\t\t\tcontent.Source.MediaType = mimeType\n\t\t\t\tcontent.Source.Data = data\n\t\t\t}\n\t\t\tcontents = append(contents, content)\n\t\t}\n\t\tclaudeMessage.Content = contents\n\t\tclaudeRequest.Messages = append(claudeRequest.Messages, claudeMessage)\n\t}\n\treturn &claudeRequest\n}\n\n// https://docs.anthropic.com/claude/reference/messages-streaming\nfunc StreamResponseClaude2OpenAI(claudeResponse *StreamResponse) (*openai.ChatCompletionsStreamResponse, *Response) {\n\tvar response *Response\n\tvar responseText string\n\tvar stopReason string\n\ttools := make([]model.Tool, 0)\n\n\tswitch claudeResponse.Type {\n\tcase \"message_start\":\n\t\treturn nil, claudeResponse.Message\n\tcase \"content_block_start\":\n\t\tif claudeResponse.ContentBlock != nil {\n\t\t\tresponseText = claudeResponse.ContentBlock.Text\n\t\t\tif claudeResponse.ContentBlock.Type == \"tool_use\" {\n\t\t\t\ttools = append(tools, model.Tool{\n\t\t\t\t\tId: claudeResponse.ContentBlock.Id,\n\t\t\t\t\tType: \"function\",\n\t\t\t\t\tFunction: model.Function{\n\t\t\t\t\t\tName: claudeResponse.ContentBlock.Name,\n\t\t\t\t\t\tArguments: \"\",\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t}\n\t\t}\n\tcase \"content_block_delta\":\n\t\tif claudeResponse.Delta != nil {\n\t\t\tresponseText = claudeResponse.Delta.Text\n\t\t\tif claudeResponse.Delta.Type == \"input_json_delta\" {\n\t\t\t\ttools = append(tools, model.Tool{\n\t\t\t\t\tFunction: model.Function{\n\t\t\t\t\t\tArguments: claudeResponse.Delta.PartialJson,\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t}\n\t\t}\n\tcase \"message_delta\":\n\t\tif claudeResponse.Usage != nil {\n\t\t\tresponse = &Response{\n\t\t\t\tUsage: *claudeResponse.Usage,\n\t\t\t}\n\t\t}\n\t\tif claudeResponse.Delta != nil && claudeResponse.Delta.StopReason != nil {\n\t\t\tstopReason = *claudeResponse.Delta.StopReason\n\t\t}\n\t}\n\tvar choice openai.ChatCompletionsStreamResponseChoice\n\tchoice.Delta.Content = responseText\n\tif len(tools) > 0 {\n\t\tchoice.Delta.Content = nil // compatible with other OpenAI derivative applications, like LobeOpenAICompatibleFactory ...\n\t\tchoice.Delta.ToolCalls = tools\n\t}\n\tchoice.Delta.Role = \"assistant\"\n\tfinishReason := stopReasonClaude2OpenAI(&stopReason)\n\tif finishReason != \"null\" {\n\t\tchoice.FinishReason = &finishReason\n\t}\n\tvar openaiResponse openai.ChatCompletionsStreamResponse\n\topenaiResponse.Object = \"chat.completion.chunk\"\n\topenaiResponse.Choices = []openai.ChatCompletionsStreamResponseChoice{choice}\n\treturn &openaiResponse, response\n}\n\nfunc ResponseClaude2OpenAI(claudeResponse *Response) *openai.TextResponse {\n\tvar responseText string\n\tif len(claudeResponse.Content) > 0 {\n\t\tresponseText = claudeResponse.Content[0].Text\n\t}\n\ttools := make([]model.Tool, 0)\n\tfor _, v := range claudeResponse.Content {\n\t\tif v.Type == \"tool_use\" {\n\t\t\targs, _ := json.Marshal(v.Input)\n\t\t\ttools = append(tools, model.Tool{\n\t\t\t\tId: v.Id,\n\t\t\t\tType: \"function\", // compatible with other OpenAI derivative applications\n\t\t\t\tFunction: model.Function{\n\t\t\t\t\tName: v.Name,\n\t\t\t\t\tArguments: string(args),\n\t\t\t\t},\n\t\t\t})\n\t\t}\n\t}\n\tchoice := openai.TextResponseChoice{\n\t\tIndex: 0,\n\t\tMessage: model.Message{\n\t\t\tRole: \"assistant\",\n\t\t\tContent: responseText,\n\t\t\tName: nil,\n\t\t\tToolCalls: tools,\n\t\t},\n\t\tFinishReason: stopReasonClaude2OpenAI(claudeResponse.StopReason),\n\t}\n\tfullTextResponse := openai.TextResponse{\n\t\tId: fmt.Sprintf(\"chatcmpl-%s\", claudeResponse.Id),\n\t\tModel: claudeResponse.Model,\n\t\tObject: \"chat.completion\",\n\t\tCreated: helper.GetTimestamp(),\n\t\tChoices: []openai.TextResponseChoice{choice},\n\t}\n\treturn &fullTextResponse\n}\n\nfunc StreamHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tcreatedTime := helper.GetTimestamp()\n\tscanner := bufio.NewScanner(resp.Body)\n\tscanner.Split(func(data []byte, atEOF bool) (advance int, token []byte, err error) {\n\t\tif atEOF && len(data) == 0 {\n\t\t\treturn 0, nil, nil\n\t\t}\n\t\tif i := strings.Index(string(data), \"\\n\"); i >= 0 {\n\t\t\treturn i + 1, data[0:i], nil\n\t\t}\n\t\tif atEOF {\n\t\t\treturn len(data), data, nil\n\t\t}\n\t\treturn 0, nil, nil\n\t})\n\n\tcommon.SetEventStreamHeaders(c)\n\n\tvar usage model.Usage\n\tvar modelName string\n\tvar id string\n\tvar lastToolCallChoice openai.ChatCompletionsStreamResponseChoice\n\n\tfor scanner.Scan() {\n\t\tdata := scanner.Text()\n\t\tif len(data) < 6 || !strings.HasPrefix(data, \"data:\") {\n\t\t\tcontinue\n\t\t}\n\t\tdata = strings.TrimPrefix(data, \"data:\")\n\t\tdata = strings.TrimSpace(data)\n\n\t\tvar claudeResponse StreamResponse\n\t\terr := json.Unmarshal([]byte(data), &claudeResponse)\n\t\tif err != nil {\n\t\t\tlogger.SysError(\"error unmarshalling stream response: \" + err.Error())\n\t\t\tcontinue\n\t\t}\n\n\t\tresponse, meta := StreamResponseClaude2OpenAI(&claudeResponse)\n\t\tif meta != nil {\n\t\t\tusage.PromptTokens += meta.Usage.InputTokens\n\t\t\tusage.CompletionTokens += meta.Usage.OutputTokens\n\t\t\tif len(meta.Id) > 0 { // only message_start has an id, otherwise it's a finish_reason event.\n\t\t\t\tmodelName = meta.Model\n\t\t\t\tid = fmt.Sprintf(\"chatcmpl-%s\", meta.Id)\n\t\t\t\tcontinue\n\t\t\t} else { // finish_reason case\n\t\t\t\tif len(lastToolCallChoice.Delta.ToolCalls) > 0 {\n\t\t\t\t\tlastArgs := &lastToolCallChoice.Delta.ToolCalls[len(lastToolCallChoice.Delta.ToolCalls)-1].Function\n\t\t\t\t\tif len(lastArgs.Arguments.(string)) == 0 { // compatible with OpenAI sending an empty object `{}` when no arguments.\n\t\t\t\t\t\tlastArgs.Arguments = \"{}\"\n\t\t\t\t\t\tresponse.Choices[len(response.Choices)-1].Delta.Content = nil\n\t\t\t\t\t\tresponse.Choices[len(response.Choices)-1].Delta.ToolCalls = lastToolCallChoice.Delta.ToolCalls\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t\tif response == nil {\n\t\t\tcontinue\n\t\t}\n\n\t\tresponse.Id = id\n\t\tresponse.Model = modelName\n\t\tresponse.Created = createdTime\n\n\t\tfor _, choice := range response.Choices {\n\t\t\tif len(choice.Delta.ToolCalls) > 0 {\n\t\t\t\tlastToolCallChoice = choice\n\t\t\t}\n\t\t}\n\t\terr = render.ObjectData(c, response)\n\t\tif err != nil {\n\t\t\tlogger.SysError(err.Error())\n\t\t}\n\t}\n\n\tif err := scanner.Err(); err != nil {\n\t\tlogger.SysError(\"error reading stream: \" + err.Error())\n\t}\n\n\trender.Done(c)\n\n\terr := resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\treturn nil, &usage\n}\n\nfunc Handler(c *gin.Context, resp *http.Response, promptTokens int, modelName string) (*model.ErrorWithStatusCode, *model.Usage) {\n\tresponseBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tvar claudeResponse Response\n\terr = json.Unmarshal(responseBody, &claudeResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tif claudeResponse.Error.Type != \"\" {\n\t\treturn &model.ErrorWithStatusCode{\n\t\t\tError: model.Error{\n\t\t\t\tMessage: claudeResponse.Error.Message,\n\t\t\t\tType: claudeResponse.Error.Type,\n\t\t\t\tParam: \"\",\n\t\t\t\tCode: claudeResponse.Error.Type,\n\t\t\t},\n\t\t\tStatusCode: resp.StatusCode,\n\t\t}, nil\n\t}\n\tfullTextResponse := ResponseClaude2OpenAI(&claudeResponse)\n\tfullTextResponse.Model = modelName\n\tusage := model.Usage{\n\t\tPromptTokens: claudeResponse.Usage.InputTokens,\n\t\tCompletionTokens: claudeResponse.Usage.OutputTokens,\n\t\tTotalTokens: claudeResponse.Usage.InputTokens + claudeResponse.Usage.OutputTokens,\n\t}\n\tfullTextResponse.Usage = usage\n\tjsonResponse, err := json.Marshal(fullTextResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tc.Writer.Header().Set(\"Content-Type\", \"application/json\")\n\tc.Writer.WriteHeader(resp.StatusCode)\n\t_, err = c.Writer.Write(jsonResponse)\n\treturn nil, &usage\n}\n"
},
{
"path": "relay/adaptor/anthropic/model.go",
"content": "package anthropic\n\n// https://docs.anthropic.com/claude/reference/messages_post\n\ntype Metadata struct {\n\tUserId string `json:\"user_id\"`\n}\n\ntype ImageSource struct {\n\tType string `json:\"type\"`\n\tMediaType string `json:\"media_type\"`\n\tData string `json:\"data\"`\n}\n\ntype Content struct {\n\tType string `json:\"type\"`\n\tText string `json:\"text,omitempty\"`\n\tSource *ImageSource `json:\"source,omitempty\"`\n\t// tool_calls\n\tId string `json:\"id,omitempty\"`\n\tName string `json:\"name,omitempty\"`\n\tInput any `json:\"input,omitempty\"`\n\tContent string `json:\"content,omitempty\"`\n\tToolUseId string `json:\"tool_use_id,omitempty\"`\n}\n\ntype Message struct {\n\tRole string `json:\"role\"`\n\tContent []Content `json:\"content\"`\n}\n\ntype Tool struct {\n\tName string `json:\"name\"`\n\tDescription string `json:\"description,omitempty\"`\n\tInputSchema InputSchema `json:\"input_schema\"`\n}\n\ntype InputSchema struct {\n\tType string `json:\"type\"`\n\tProperties any `json:\"properties,omitempty\"`\n\tRequired any `json:\"required,omitempty\"`\n}\n\ntype Request struct {\n\tModel string `json:\"model\"`\n\tMessages []Message `json:\"messages\"`\n\tSystem string `json:\"system,omitempty\"`\n\tMaxTokens int `json:\"max_tokens,omitempty\"`\n\tStopSequences []string `json:\"stop_sequences,omitempty\"`\n\tStream bool `json:\"stream,omitempty\"`\n\tTemperature *float64 `json:\"temperature,omitempty\"`\n\tTopP *float64 `json:\"top_p,omitempty\"`\n\tTopK int `json:\"top_k,omitempty\"`\n\tTools []Tool `json:\"tools,omitempty\"`\n\tToolChoice any `json:\"tool_choice,omitempty\"`\n\t//Metadata `json:\"metadata,omitempty\"`\n}\n\ntype Usage struct {\n\tInputTokens int `json:\"input_tokens\"`\n\tOutputTokens int `json:\"output_tokens\"`\n}\n\ntype Error struct {\n\tType string `json:\"type\"`\n\tMessage string `json:\"message\"`\n}\n\ntype Response struct {\n\tId string `json:\"id\"`\n\tType string `json:\"type\"`\n\tRole string `json:\"role\"`\n\tContent []Content `json:\"content\"`\n\tModel string `json:\"model\"`\n\tStopReason *string `json:\"stop_reason\"`\n\tStopSequence *string `json:\"stop_sequence\"`\n\tUsage Usage `json:\"usage\"`\n\tError Error `json:\"error\"`\n}\n\ntype Delta struct {\n\tType string `json:\"type\"`\n\tText string `json:\"text\"`\n\tPartialJson string `json:\"partial_json,omitempty\"`\n\tStopReason *string `json:\"stop_reason\"`\n\tStopSequence *string `json:\"stop_sequence\"`\n}\n\ntype StreamResponse struct {\n\tType string `json:\"type\"`\n\tMessage *Response `json:\"message\"`\n\tIndex int `json:\"index\"`\n\tContentBlock *Content `json:\"content_block\"`\n\tDelta *Delta `json:\"delta\"`\n\tUsage *Usage `json:\"usage\"`\n}\n"
},
{
"path": "relay/adaptor/aws/adaptor.go",
"content": "package aws\n\nimport (\n\t\"errors\"\n\t\"io\"\n\t\"net/http\"\n\n\t\"github.com/aws/aws-sdk-go-v2/aws\"\n\t\"github.com/aws/aws-sdk-go-v2/credentials\"\n\t\"github.com/aws/aws-sdk-go-v2/service/bedrockruntime\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/aws/utils\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\nvar _ adaptor.Adaptor = new(Adaptor)\n\ntype Adaptor struct {\n\tawsAdapter utils.AwsAdapter\n\n\tMeta *meta.Meta\n\tAwsClient *bedrockruntime.Client\n}\n\nfunc (a *Adaptor) Init(meta *meta.Meta) {\n\ta.Meta = meta\n\ta.AwsClient = bedrockruntime.New(bedrockruntime.Options{\n\t\tRegion: meta.Config.Region,\n\t\tCredentials: aws.NewCredentialsCache(credentials.NewStaticCredentialsProvider(meta.Config.AK, meta.Config.SK, \"\")),\n\t})\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\n\tadaptor := GetAdaptor(request.Model)\n\tif adaptor == nil {\n\t\treturn nil, errors.New(\"adaptor not found\")\n\t}\n\n\ta.awsAdapter = adaptor\n\treturn adaptor.ConvertRequest(c, relayMode, request)\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tif a.awsAdapter == nil {\n\t\treturn nil, utils.WrapErr(errors.New(\"awsAdapter is nil\"))\n\t}\n\treturn a.awsAdapter.DoResponse(c, a.AwsClient, meta)\n}\n\nfunc (a *Adaptor) GetModelList() (models []string) {\n\tfor model := range adaptors {\n\t\tmodels = append(models, model)\n\t}\n\treturn\n}\n\nfunc (a *Adaptor) GetChannelName() string {\n\treturn \"aws\"\n}\n\nfunc (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {\n\treturn \"\", nil\n}\n\nfunc (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {\n\treturn nil\n}\n\nfunc (a *Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\treturn request, nil\n}\n\nfunc (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {\n\treturn nil, nil\n}\n"
},
{
"path": "relay/adaptor/aws/claude/adapter.go",
"content": "package aws\n\nimport (\n\t\"github.com/aws/aws-sdk-go-v2/service/bedrockruntime\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/pkg/errors\"\n\t\"github.com/songquanpeng/one-api/common/ctxkey\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/anthropic\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/aws/utils\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\nvar _ utils.AwsAdapter = new(Adaptor)\n\ntype Adaptor struct {\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\n\tclaudeReq := anthropic.ConvertRequest(*request)\n\tc.Set(ctxkey.RequestModel, request.Model)\n\tc.Set(ctxkey.ConvertedRequest, claudeReq)\n\treturn claudeReq, nil\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, awsCli *bedrockruntime.Client, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tif meta.IsStream {\n\t\terr, usage = StreamHandler(c, awsCli)\n\t} else {\n\t\terr, usage = Handler(c, awsCli, meta.ActualModelName)\n\t}\n\treturn\n}\n"
},
{
"path": "relay/adaptor/aws/claude/main.go",
"content": "// Package aws provides the AWS adaptor for the relay service.\npackage aws\n\nimport (\n\t\"bytes\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"io\"\n\t\"net/http\"\n\n\t\"github.com/aws/aws-sdk-go-v2/aws\"\n\t\"github.com/aws/aws-sdk-go-v2/service/bedrockruntime\"\n\t\"github.com/aws/aws-sdk-go-v2/service/bedrockruntime/types\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/jinzhu/copier\"\n\t\"github.com/pkg/errors\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/ctxkey\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/anthropic\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/aws/utils\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\trelaymodel \"github.com/songquanpeng/one-api/relay/model\"\n)\n\n// https://docs.aws.amazon.com/bedrock/latest/userguide/model-ids.html\nvar AwsModelIDMap = map[string]string{\n\t\"claude-instant-1.2\": \"anthropic.claude-instant-v1\",\n\t\"claude-2.0\": \"anthropic.claude-v2\",\n\t\"claude-2.1\": \"anthropic.claude-v2:1\",\n\t\"claude-3-haiku-20240307\": \"anthropic.claude-3-haiku-20240307-v1:0\",\n\t\"claude-3-sonnet-20240229\": \"anthropic.claude-3-sonnet-20240229-v1:0\",\n\t\"claude-3-opus-20240229\": \"anthropic.claude-3-opus-20240229-v1:0\",\n\t\"claude-3-5-sonnet-20240620\": \"anthropic.claude-3-5-sonnet-20240620-v1:0\",\n\t\"claude-3-5-sonnet-20241022\": \"anthropic.claude-3-5-sonnet-20241022-v2:0\",\n\t\"claude-3-5-sonnet-latest\": \"anthropic.claude-3-5-sonnet-20241022-v2:0\",\n\t\"claude-3-5-haiku-20241022\": \"anthropic.claude-3-5-haiku-20241022-v1:0\",\n}\n\nfunc awsModelID(requestModel string) (string, error) {\n\tif awsModelID, ok := AwsModelIDMap[requestModel]; ok {\n\t\treturn awsModelID, nil\n\t}\n\n\treturn \"\", errors.Errorf(\"model %s not found\", requestModel)\n}\n\nfunc Handler(c *gin.Context, awsCli *bedrockruntime.Client, modelName string) (*relaymodel.ErrorWithStatusCode, *relaymodel.Usage) {\n\tawsModelId, err := awsModelID(c.GetString(ctxkey.RequestModel))\n\tif err != nil {\n\t\treturn utils.WrapErr(errors.Wrap(err, \"awsModelID\")), nil\n\t}\n\n\tawsReq := &bedrockruntime.InvokeModelInput{\n\t\tModelId: aws.String(awsModelId),\n\t\tAccept: aws.String(\"application/json\"),\n\t\tContentType: aws.String(\"application/json\"),\n\t}\n\n\tclaudeReq_, ok := c.Get(ctxkey.ConvertedRequest)\n\tif !ok {\n\t\treturn utils.WrapErr(errors.New(\"request not found\")), nil\n\t}\n\tclaudeReq := claudeReq_.(*anthropic.Request)\n\tawsClaudeReq := &Request{\n\t\tAnthropicVersion: \"bedrock-2023-05-31\",\n\t}\n\tif err = copier.Copy(awsClaudeReq, claudeReq); err != nil {\n\t\treturn utils.WrapErr(errors.Wrap(err, \"copy request\")), nil\n\t}\n\n\tawsReq.Body, err = json.Marshal(awsClaudeReq)\n\tif err != nil {\n\t\treturn utils.WrapErr(errors.Wrap(err, \"marshal request\")), nil\n\t}\n\n\tawsResp, err := awsCli.InvokeModel(c.Request.Context(), awsReq)\n\tif err != nil {\n\t\treturn utils.WrapErr(errors.Wrap(err, \"InvokeModel\")), nil\n\t}\n\n\tclaudeResponse := new(anthropic.Response)\n\terr = json.Unmarshal(awsResp.Body, claudeResponse)\n\tif err != nil {\n\t\treturn utils.WrapErr(errors.Wrap(err, \"unmarshal response\")), nil\n\t}\n\n\topenaiResp := anthropic.ResponseClaude2OpenAI(claudeResponse)\n\topenaiResp.Model = modelName\n\tusage := relaymodel.Usage{\n\t\tPromptTokens: claudeResponse.Usage.InputTokens,\n\t\tCompletionTokens: claudeResponse.Usage.OutputTokens,\n\t\tTotalTokens: claudeResponse.Usage.InputTokens + claudeResponse.Usage.OutputTokens,\n\t}\n\topenaiResp.Usage = usage\n\n\tc.JSON(http.StatusOK, openaiResp)\n\treturn nil, &usage\n}\n\nfunc StreamHandler(c *gin.Context, awsCli *bedrockruntime.Client) (*relaymodel.ErrorWithStatusCode, *relaymodel.Usage) {\n\tcreatedTime := helper.GetTimestamp()\n\tawsModelId, err := awsModelID(c.GetString(ctxkey.RequestModel))\n\tif err != nil {\n\t\treturn utils.WrapErr(errors.Wrap(err, \"awsModelID\")), nil\n\t}\n\n\tawsReq := &bedrockruntime.InvokeModelWithResponseStreamInput{\n\t\tModelId: aws.String(awsModelId),\n\t\tAccept: aws.String(\"application/json\"),\n\t\tContentType: aws.String(\"application/json\"),\n\t}\n\n\tclaudeReq_, ok := c.Get(ctxkey.ConvertedRequest)\n\tif !ok {\n\t\treturn utils.WrapErr(errors.New(\"request not found\")), nil\n\t}\n\tclaudeReq := claudeReq_.(*anthropic.Request)\n\n\tawsClaudeReq := &Request{\n\t\tAnthropicVersion: \"bedrock-2023-05-31\",\n\t}\n\tif err = copier.Copy(awsClaudeReq, claudeReq); err != nil {\n\t\treturn utils.WrapErr(errors.Wrap(err, \"copy request\")), nil\n\t}\n\tawsReq.Body, err = json.Marshal(awsClaudeReq)\n\tif err != nil {\n\t\treturn utils.WrapErr(errors.Wrap(err, \"marshal request\")), nil\n\t}\n\n\tawsResp, err := awsCli.InvokeModelWithResponseStream(c.Request.Context(), awsReq)\n\tif err != nil {\n\t\treturn utils.WrapErr(errors.Wrap(err, \"InvokeModelWithResponseStream\")), nil\n\t}\n\tstream := awsResp.GetStream()\n\tdefer stream.Close()\n\n\tc.Writer.Header().Set(\"Content-Type\", \"text/event-stream\")\n\tvar usage relaymodel.Usage\n\tvar id string\n\tvar lastToolCallChoice openai.ChatCompletionsStreamResponseChoice\n\n\tc.Stream(func(w io.Writer) bool {\n\t\tevent, ok := <-stream.Events()\n\t\tif !ok {\n\t\t\tc.Render(-1, common.CustomEvent{Data: \"data: [DONE]\"})\n\t\t\treturn false\n\t\t}\n\n\t\tswitch v := event.(type) {\n\t\tcase *types.ResponseStreamMemberChunk:\n\t\t\tclaudeResp := new(anthropic.StreamResponse)\n\t\t\terr := json.NewDecoder(bytes.NewReader(v.Value.Bytes)).Decode(claudeResp)\n\t\t\tif err != nil {\n\t\t\t\tlogger.SysError(\"error unmarshalling stream response: \" + err.Error())\n\t\t\t\treturn false\n\t\t\t}\n\n\t\t\tresponse, meta := anthropic.StreamResponseClaude2OpenAI(claudeResp)\n\t\t\tif meta != nil {\n\t\t\t\tusage.PromptTokens += meta.Usage.InputTokens\n\t\t\t\tusage.CompletionTokens += meta.Usage.OutputTokens\n\t\t\t\tif len(meta.Id) > 0 { // only message_start has an id, otherwise it's a finish_reason event.\n\t\t\t\t\tid = fmt.Sprintf(\"chatcmpl-%s\", meta.Id)\n\t\t\t\t\treturn true\n\t\t\t\t} else { // finish_reason case\n\t\t\t\t\tif len(lastToolCallChoice.Delta.ToolCalls) > 0 {\n\t\t\t\t\t\tlastArgs := &lastToolCallChoice.Delta.ToolCalls[len(lastToolCallChoice.Delta.ToolCalls)-1].Function\n\t\t\t\t\t\tif len(lastArgs.Arguments.(string)) == 0 { // compatible with OpenAI sending an empty object `{}` when no arguments.\n\t\t\t\t\t\t\tlastArgs.Arguments = \"{}\"\n\t\t\t\t\t\t\tresponse.Choices[len(response.Choices)-1].Delta.Content = nil\n\t\t\t\t\t\t\tresponse.Choices[len(response.Choices)-1].Delta.ToolCalls = lastToolCallChoice.Delta.ToolCalls\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t\tif response == nil {\n\t\t\t\treturn true\n\t\t\t}\n\t\t\tresponse.Id = id\n\t\t\tresponse.Model = c.GetString(ctxkey.OriginalModel)\n\t\t\tresponse.Created = createdTime\n\n\t\t\tfor _, choice := range response.Choices {\n\t\t\t\tif len(choice.Delta.ToolCalls) > 0 {\n\t\t\t\t\tlastToolCallChoice = choice\n\t\t\t\t}\n\t\t\t}\n\t\t\tjsonStr, err := json.Marshal(response)\n\t\t\tif err != nil {\n\t\t\t\tlogger.SysError(\"error marshalling stream response: \" + err.Error())\n\t\t\t\treturn true\n\t\t\t}\n\t\t\tc.Render(-1, common.CustomEvent{Data: \"data: \" + string(jsonStr)})\n\t\t\treturn true\n\t\tcase *types.UnknownUnionMember:\n\t\t\tfmt.Println(\"unknown tag:\", v.Tag)\n\t\t\treturn false\n\t\tdefault:\n\t\t\tfmt.Println(\"union is nil or unknown type\")\n\t\t\treturn false\n\t\t}\n\t})\n\n\treturn nil, &usage\n}\n"
},
{
"path": "relay/adaptor/aws/claude/model.go",
"content": "package aws\n\nimport \"github.com/songquanpeng/one-api/relay/adaptor/anthropic\"\n\n// Request is the request to AWS Claude\n//\n// https://docs.aws.amazon.com/bedrock/latest/userguide/model-parameters-anthropic-claude-messages.html\ntype Request struct {\n\t// AnthropicVersion should be \"bedrock-2023-05-31\"\n\tAnthropicVersion string `json:\"anthropic_version\"`\n\tMessages []anthropic.Message `json:\"messages\"`\n\tSystem string `json:\"system,omitempty\"`\n\tMaxTokens int `json:\"max_tokens,omitempty\"`\n\tTemperature *float64 `json:\"temperature,omitempty\"`\n\tTopP *float64 `json:\"top_p,omitempty\"`\n\tTopK int `json:\"top_k,omitempty\"`\n\tStopSequences []string `json:\"stop_sequences,omitempty\"`\n\tTools []anthropic.Tool `json:\"tools,omitempty\"`\n\tToolChoice any `json:\"tool_choice,omitempty\"`\n}\n"
},
{
"path": "relay/adaptor/aws/llama3/adapter.go",
"content": "package aws\n\nimport (\n\t\"github.com/aws/aws-sdk-go-v2/service/bedrockruntime\"\n\t\"github.com/songquanpeng/one-api/common/ctxkey\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/pkg/errors\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/aws/utils\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\nvar _ utils.AwsAdapter = new(Adaptor)\n\ntype Adaptor struct {\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\n\tllamaReq := ConvertRequest(*request)\n\tc.Set(ctxkey.RequestModel, request.Model)\n\tc.Set(ctxkey.ConvertedRequest, llamaReq)\n\treturn llamaReq, nil\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, awsCli *bedrockruntime.Client, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tif meta.IsStream {\n\t\terr, usage = StreamHandler(c, awsCli)\n\t} else {\n\t\terr, usage = Handler(c, awsCli, meta.ActualModelName)\n\t}\n\treturn\n}\n"
},
{
"path": "relay/adaptor/aws/llama3/main.go",
"content": "// Package aws provides the AWS adaptor for the relay service.\npackage aws\n\nimport (\n\t\"bytes\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"io\"\n\t\"net/http\"\n\t\"text/template\"\n\n\t\"github.com/songquanpeng/one-api/common/ctxkey\"\n\t\"github.com/songquanpeng/one-api/common/random\"\n\n\t\"github.com/aws/aws-sdk-go-v2/aws\"\n\t\"github.com/aws/aws-sdk-go-v2/service/bedrockruntime\"\n\t\"github.com/aws/aws-sdk-go-v2/service/bedrockruntime/types\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/pkg/errors\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/aws/utils\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\trelaymodel \"github.com/songquanpeng/one-api/relay/model\"\n)\n\n// Only support llama-3-8b and llama-3-70b instruction models\n// https://docs.aws.amazon.com/bedrock/latest/userguide/model-ids.html\nvar AwsModelIDMap = map[string]string{\n\t\"llama3-8b-8192\": \"meta.llama3-8b-instruct-v1:0\",\n\t\"llama3-70b-8192\": \"meta.llama3-70b-instruct-v1:0\",\n}\n\nfunc awsModelID(requestModel string) (string, error) {\n\tif awsModelID, ok := AwsModelIDMap[requestModel]; ok {\n\t\treturn awsModelID, nil\n\t}\n\n\treturn \"\", errors.Errorf(\"model %s not found\", requestModel)\n}\n\n// promptTemplate with range\nconst promptTemplate = `<|begin_of_text|>{{range .Messages}}<|start_header_id|>{{.Role}}<|end_header_id|>{{.StringContent}}<|eot_id|>{{end}}<|start_header_id|>assistant<|end_header_id|>\n`\n\nvar promptTpl = template.Must(template.New(\"llama3-chat\").Parse(promptTemplate))\n\nfunc RenderPrompt(messages []relaymodel.Message) string {\n\tvar buf bytes.Buffer\n\terr := promptTpl.Execute(&buf, struct{ Messages []relaymodel.Message }{messages})\n\tif err != nil {\n\t\tlogger.SysError(\"error rendering prompt messages: \" + err.Error())\n\t}\n\treturn buf.String()\n}\n\nfunc ConvertRequest(textRequest relaymodel.GeneralOpenAIRequest) *Request {\n\tllamaRequest := Request{\n\t\tMaxGenLen: textRequest.MaxTokens,\n\t\tTemperature: textRequest.Temperature,\n\t\tTopP: textRequest.TopP,\n\t}\n\tif llamaRequest.MaxGenLen == 0 {\n\t\tllamaRequest.MaxGenLen = 2048\n\t}\n\tprompt := RenderPrompt(textRequest.Messages)\n\tllamaRequest.Prompt = prompt\n\treturn &llamaRequest\n}\n\nfunc Handler(c *gin.Context, awsCli *bedrockruntime.Client, modelName string) (*relaymodel.ErrorWithStatusCode, *relaymodel.Usage) {\n\tawsModelId, err := awsModelID(c.GetString(ctxkey.RequestModel))\n\tif err != nil {\n\t\treturn utils.WrapErr(errors.Wrap(err, \"awsModelID\")), nil\n\t}\n\n\tawsReq := &bedrockruntime.InvokeModelInput{\n\t\tModelId: aws.String(awsModelId),\n\t\tAccept: aws.String(\"application/json\"),\n\t\tContentType: aws.String(\"application/json\"),\n\t}\n\n\tllamaReq, ok := c.Get(ctxkey.ConvertedRequest)\n\tif !ok {\n\t\treturn utils.WrapErr(errors.New(\"request not found\")), nil\n\t}\n\n\tawsReq.Body, err = json.Marshal(llamaReq)\n\tif err != nil {\n\t\treturn utils.WrapErr(errors.Wrap(err, \"marshal request\")), nil\n\t}\n\n\tawsResp, err := awsCli.InvokeModel(c.Request.Context(), awsReq)\n\tif err != nil {\n\t\treturn utils.WrapErr(errors.Wrap(err, \"InvokeModel\")), nil\n\t}\n\n\tvar llamaResponse Response\n\terr = json.Unmarshal(awsResp.Body, &llamaResponse)\n\tif err != nil {\n\t\treturn utils.WrapErr(errors.Wrap(err, \"unmarshal response\")), nil\n\t}\n\n\topenaiResp := ResponseLlama2OpenAI(&llamaResponse)\n\topenaiResp.Model = modelName\n\tusage := relaymodel.Usage{\n\t\tPromptTokens: llamaResponse.PromptTokenCount,\n\t\tCompletionTokens: llamaResponse.GenerationTokenCount,\n\t\tTotalTokens: llamaResponse.PromptTokenCount + llamaResponse.GenerationTokenCount,\n\t}\n\topenaiResp.Usage = usage\n\n\tc.JSON(http.StatusOK, openaiResp)\n\treturn nil, &usage\n}\n\nfunc ResponseLlama2OpenAI(llamaResponse *Response) *openai.TextResponse {\n\tvar responseText string\n\tif len(llamaResponse.Generation) > 0 {\n\t\tresponseText = llamaResponse.Generation\n\t}\n\tchoice := openai.TextResponseChoice{\n\t\tIndex: 0,\n\t\tMessage: relaymodel.Message{\n\t\t\tRole: \"assistant\",\n\t\t\tContent: responseText,\n\t\t\tName: nil,\n\t\t},\n\t\tFinishReason: llamaResponse.StopReason,\n\t}\n\tfullTextResponse := openai.TextResponse{\n\t\tId: fmt.Sprintf(\"chatcmpl-%s\", random.GetUUID()),\n\t\tObject: \"chat.completion\",\n\t\tCreated: helper.GetTimestamp(),\n\t\tChoices: []openai.TextResponseChoice{choice},\n\t}\n\treturn &fullTextResponse\n}\n\nfunc StreamHandler(c *gin.Context, awsCli *bedrockruntime.Client) (*relaymodel.ErrorWithStatusCode, *relaymodel.Usage) {\n\tcreatedTime := helper.GetTimestamp()\n\tawsModelId, err := awsModelID(c.GetString(ctxkey.RequestModel))\n\tif err != nil {\n\t\treturn utils.WrapErr(errors.Wrap(err, \"awsModelID\")), nil\n\t}\n\n\tawsReq := &bedrockruntime.InvokeModelWithResponseStreamInput{\n\t\tModelId: aws.String(awsModelId),\n\t\tAccept: aws.String(\"application/json\"),\n\t\tContentType: aws.String(\"application/json\"),\n\t}\n\n\tllamaReq, ok := c.Get(ctxkey.ConvertedRequest)\n\tif !ok {\n\t\treturn utils.WrapErr(errors.New(\"request not found\")), nil\n\t}\n\n\tawsReq.Body, err = json.Marshal(llamaReq)\n\tif err != nil {\n\t\treturn utils.WrapErr(errors.Wrap(err, \"marshal request\")), nil\n\t}\n\n\tawsResp, err := awsCli.InvokeModelWithResponseStream(c.Request.Context(), awsReq)\n\tif err != nil {\n\t\treturn utils.WrapErr(errors.Wrap(err, \"InvokeModelWithResponseStream\")), nil\n\t}\n\tstream := awsResp.GetStream()\n\tdefer stream.Close()\n\n\tc.Writer.Header().Set(\"Content-Type\", \"text/event-stream\")\n\tvar usage relaymodel.Usage\n\tc.Stream(func(w io.Writer) bool {\n\t\tevent, ok := <-stream.Events()\n\t\tif !ok {\n\t\t\tc.Render(-1, common.CustomEvent{Data: \"data: [DONE]\"})\n\t\t\treturn false\n\t\t}\n\n\t\tswitch v := event.(type) {\n\t\tcase *types.ResponseStreamMemberChunk:\n\t\t\tvar llamaResp StreamResponse\n\t\t\terr := json.NewDecoder(bytes.NewReader(v.Value.Bytes)).Decode(&llamaResp)\n\t\t\tif err != nil {\n\t\t\t\tlogger.SysError(\"error unmarshalling stream response: \" + err.Error())\n\t\t\t\treturn false\n\t\t\t}\n\n\t\t\tif llamaResp.PromptTokenCount > 0 {\n\t\t\t\tusage.PromptTokens = llamaResp.PromptTokenCount\n\t\t\t}\n\t\t\tif llamaResp.StopReason == \"stop\" {\n\t\t\t\tusage.CompletionTokens = llamaResp.GenerationTokenCount\n\t\t\t\tusage.TotalTokens = usage.PromptTokens + usage.CompletionTokens\n\t\t\t}\n\t\t\tresponse := StreamResponseLlama2OpenAI(&llamaResp)\n\t\t\tresponse.Id = fmt.Sprintf(\"chatcmpl-%s\", random.GetUUID())\n\t\t\tresponse.Model = c.GetString(ctxkey.OriginalModel)\n\t\t\tresponse.Created = createdTime\n\t\t\tjsonStr, err := json.Marshal(response)\n\t\t\tif err != nil {\n\t\t\t\tlogger.SysError(\"error marshalling stream response: \" + err.Error())\n\t\t\t\treturn true\n\t\t\t}\n\t\t\tc.Render(-1, common.CustomEvent{Data: \"data: \" + string(jsonStr)})\n\t\t\treturn true\n\t\tcase *types.UnknownUnionMember:\n\t\t\tfmt.Println(\"unknown tag:\", v.Tag)\n\t\t\treturn false\n\t\tdefault:\n\t\t\tfmt.Println(\"union is nil or unknown type\")\n\t\t\treturn false\n\t\t}\n\t})\n\n\treturn nil, &usage\n}\n\nfunc StreamResponseLlama2OpenAI(llamaResponse *StreamResponse) *openai.ChatCompletionsStreamResponse {\n\tvar choice openai.ChatCompletionsStreamResponseChoice\n\tchoice.Delta.Content = llamaResponse.Generation\n\tchoice.Delta.Role = \"assistant\"\n\tfinishReason := llamaResponse.StopReason\n\tif finishReason != \"null\" {\n\t\tchoice.FinishReason = &finishReason\n\t}\n\tvar openaiResponse openai.ChatCompletionsStreamResponse\n\topenaiResponse.Object = \"chat.completion.chunk\"\n\topenaiResponse.Choices = []openai.ChatCompletionsStreamResponseChoice{choice}\n\treturn &openaiResponse\n}\n"
},
{
"path": "relay/adaptor/aws/llama3/main_test.go",
"content": "package aws_test\n\nimport (\n\t\"testing\"\n\n\taws \"github.com/songquanpeng/one-api/relay/adaptor/aws/llama3\"\n\trelaymodel \"github.com/songquanpeng/one-api/relay/model\"\n\t\"github.com/stretchr/testify/assert\"\n)\n\nfunc TestRenderPrompt(t *testing.T) {\n\tmessages := []relaymodel.Message{\n\t\t{\n\t\t\tRole: \"user\",\n\t\t\tContent: \"What's your name?\",\n\t\t},\n\t}\n\tprompt := aws.RenderPrompt(messages)\n\texpected := `<|begin_of_text|><|start_header_id|>user<|end_header_id|>What's your name?<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n`\n\tassert.Equal(t, expected, prompt)\n\n\tmessages = []relaymodel.Message{\n\t\t{\n\t\t\tRole: \"system\",\n\t\t\tContent: \"Your name is Kat. You are a detective.\",\n\t\t},\n\t\t{\n\t\t\tRole: \"user\",\n\t\t\tContent: \"What's your name?\",\n\t\t},\n\t\t{\n\t\t\tRole: \"assistant\",\n\t\t\tContent: \"Kat\",\n\t\t},\n\t\t{\n\t\t\tRole: \"user\",\n\t\t\tContent: \"What's your job?\",\n\t\t},\n\t}\n\tprompt = aws.RenderPrompt(messages)\n\texpected = `<|begin_of_text|><|start_header_id|>system<|end_header_id|>Your name is Kat. You are a detective.<|eot_id|><|start_header_id|>user<|end_header_id|>What's your name?<|eot_id|><|start_header_id|>assistant<|end_header_id|>Kat<|eot_id|><|start_header_id|>user<|end_header_id|>What's your job?<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n`\n\tassert.Equal(t, expected, prompt)\n}\n"
},
{
"path": "relay/adaptor/aws/llama3/model.go",
"content": "package aws\n\n// Request is the request to AWS Llama3\n//\n// https://docs.aws.amazon.com/bedrock/latest/userguide/model-parameters-meta.html\ntype Request struct {\n\tPrompt string `json:\"prompt\"`\n\tMaxGenLen int `json:\"max_gen_len,omitempty\"`\n\tTemperature *float64 `json:\"temperature,omitempty\"`\n\tTopP *float64 `json:\"top_p,omitempty\"`\n}\n\n// Response is the response from AWS Llama3\n//\n// https://docs.aws.amazon.com/bedrock/latest/userguide/model-parameters-meta.html\ntype Response struct {\n\tGeneration string `json:\"generation\"`\n\tPromptTokenCount int `json:\"prompt_token_count\"`\n\tGenerationTokenCount int `json:\"generation_token_count\"`\n\tStopReason string `json:\"stop_reason\"`\n}\n\n// {'generation': 'Hi', 'prompt_token_count': 15, 'generation_token_count': 1, 'stop_reason': None}\ntype StreamResponse struct {\n\tGeneration string `json:\"generation\"`\n\tPromptTokenCount int `json:\"prompt_token_count\"`\n\tGenerationTokenCount int `json:\"generation_token_count\"`\n\tStopReason string `json:\"stop_reason\"`\n}\n"
},
{
"path": "relay/adaptor/aws/registry.go",
"content": "package aws\n\nimport (\n\tclaude \"github.com/songquanpeng/one-api/relay/adaptor/aws/claude\"\n\tllama3 \"github.com/songquanpeng/one-api/relay/adaptor/aws/llama3\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/aws/utils\"\n)\n\ntype AwsModelType int\n\nconst (\n\tAwsClaude AwsModelType = iota + 1\n\tAwsLlama3\n)\n\nvar (\n\tadaptors = map[string]AwsModelType{}\n)\n\nfunc init() {\n\tfor model := range claude.AwsModelIDMap {\n\t\tadaptors[model] = AwsClaude\n\t}\n\tfor model := range llama3.AwsModelIDMap {\n\t\tadaptors[model] = AwsLlama3\n\t}\n}\n\nfunc GetAdaptor(model string) utils.AwsAdapter {\n\tadaptorType := adaptors[model]\n\tswitch adaptorType {\n\tcase AwsClaude:\n\t\treturn &claude.Adaptor{}\n\tcase AwsLlama3:\n\t\treturn &llama3.Adaptor{}\n\tdefault:\n\t\treturn nil\n\t}\n}\n"
},
{
"path": "relay/adaptor/aws/utils/adaptor.go",
"content": "package utils\n\nimport (\n\t\"errors\"\n\t\"io\"\n\t\"net/http\"\n\n\t\"github.com/aws/aws-sdk-go-v2/aws\"\n\t\"github.com/aws/aws-sdk-go-v2/credentials\"\n\t\"github.com/aws/aws-sdk-go-v2/service/bedrockruntime\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\ntype AwsAdapter interface {\n\tConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error)\n\tDoResponse(c *gin.Context, awsCli *bedrockruntime.Client, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode)\n}\n\ntype Adaptor struct {\n\tMeta *meta.Meta\n\tAwsClient *bedrockruntime.Client\n}\n\nfunc (a *Adaptor) Init(meta *meta.Meta) {\n\ta.Meta = meta\n\ta.AwsClient = bedrockruntime.New(bedrockruntime.Options{\n\t\tRegion: meta.Config.Region,\n\t\tCredentials: aws.NewCredentialsCache(credentials.NewStaticCredentialsProvider(meta.Config.AK, meta.Config.SK, \"\")),\n\t})\n}\n\nfunc (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {\n\treturn \"\", nil\n}\n\nfunc (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {\n\treturn nil\n}\n\nfunc (a *Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\treturn request, nil\n}\n\nfunc (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {\n\treturn nil, nil\n}\n"
},
{
"path": "relay/adaptor/aws/utils/utils.go",
"content": "package utils\n\nimport (\n\t\"net/http\"\n\n\trelaymodel \"github.com/songquanpeng/one-api/relay/model\"\n)\n\nfunc WrapErr(err error) *relaymodel.ErrorWithStatusCode {\n\treturn &relaymodel.ErrorWithStatusCode{\n\t\tStatusCode: http.StatusInternalServerError,\n\t\tError: relaymodel.Error{\n\t\t\tMessage: err.Error(),\n\t\t},\n\t}\n}\n"
},
{
"path": "relay/adaptor/baichuan/constants.go",
"content": "package baichuan\n\nvar ModelList = []string{\n\t\"Baichuan2-Turbo\",\n\t\"Baichuan2-Turbo-192k\",\n\t\"Baichuan-Text-Embedding\",\n}\n"
},
{
"path": "relay/adaptor/baidu/adaptor.go",
"content": "package baidu\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/relaymode\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\ntype Adaptor struct {\n}\n\nfunc (a *Adaptor) Init(meta *meta.Meta) {\n\n}\n\nfunc (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {\n\t// https://cloud.baidu.com/doc/WENXINWORKSHOP/s/clntwmv7t\n\tsuffix := \"chat/\"\n\tif strings.HasPrefix(meta.ActualModelName, \"Embedding\") {\n\t\tsuffix = \"embeddings/\"\n\t}\n\tif strings.HasPrefix(meta.ActualModelName, \"bge-large\") {\n\t\tsuffix = \"embeddings/\"\n\t}\n\tif strings.HasPrefix(meta.ActualModelName, \"tao-8k\") {\n\t\tsuffix = \"embeddings/\"\n\t}\n\tswitch meta.ActualModelName {\n\tcase \"ERNIE-4.0\":\n\t\tsuffix += \"completions_pro\"\n\tcase \"ERNIE-Bot-4\":\n\t\tsuffix += \"completions_pro\"\n\tcase \"ERNIE-Bot\":\n\t\tsuffix += \"completions\"\n\tcase \"ERNIE-Bot-turbo\":\n\t\tsuffix += \"eb-instant\"\n\tcase \"ERNIE-Speed\":\n\t\tsuffix += \"ernie_speed\"\n\tcase \"ERNIE-4.0-8K\":\n\t\tsuffix += \"completions_pro\"\n\tcase \"ERNIE-3.5-8K\":\n\t\tsuffix += \"completions\"\n\tcase \"ERNIE-3.5-8K-0205\":\n\t\tsuffix += \"ernie-3.5-8k-0205\"\n\tcase \"ERNIE-3.5-8K-1222\":\n\t\tsuffix += \"ernie-3.5-8k-1222\"\n\tcase \"ERNIE-Bot-8K\":\n\t\tsuffix += \"ernie_bot_8k\"\n\tcase \"ERNIE-3.5-4K-0205\":\n\t\tsuffix += \"ernie-3.5-4k-0205\"\n\tcase \"ERNIE-Speed-8K\":\n\t\tsuffix += \"ernie_speed\"\n\tcase \"ERNIE-Speed-128K\":\n\t\tsuffix += \"ernie-speed-128k\"\n\tcase \"ERNIE-Lite-8K-0922\":\n\t\tsuffix += \"eb-instant\"\n\tcase \"ERNIE-Lite-8K-0308\":\n\t\tsuffix += \"ernie-lite-8k\"\n\tcase \"ERNIE-Tiny-8K\":\n\t\tsuffix += \"ernie-tiny-8k\"\n\tcase \"BLOOMZ-7B\":\n\t\tsuffix += \"bloomz_7b1\"\n\tcase \"Embedding-V1\":\n\t\tsuffix += \"embedding-v1\"\n\tcase \"bge-large-zh\":\n\t\tsuffix += \"bge_large_zh\"\n\tcase \"bge-large-en\":\n\t\tsuffix += \"bge_large_en\"\n\tcase \"tao-8k\":\n\t\tsuffix += \"tao_8k\"\n\tdefault:\n\t\tsuffix += strings.ToLower(meta.ActualModelName)\n\t}\n\tfullRequestURL := fmt.Sprintf(\"%s/rpc/2.0/ai_custom/v1/wenxinworkshop/%s\", meta.BaseURL, suffix)\n\tvar accessToken string\n\tvar err error\n\tif accessToken, err = GetAccessToken(meta.APIKey); err != nil {\n\t\treturn \"\", err\n\t}\n\tfullRequestURL += \"?access_token=\" + accessToken\n\treturn fullRequestURL, nil\n}\n\nfunc (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {\n\tadaptor.SetupCommonRequestHeader(c, req, meta)\n\treq.Header.Set(\"Authorization\", \"Bearer \"+meta.APIKey)\n\treturn nil\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\tswitch relayMode {\n\tcase relaymode.Embeddings:\n\t\tbaiduEmbeddingRequest := ConvertEmbeddingRequest(*request)\n\t\treturn baiduEmbeddingRequest, nil\n\tdefault:\n\t\tbaiduRequest := ConvertRequest(*request)\n\t\treturn baiduRequest, nil\n\t}\n}\n\nfunc (a *Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\treturn request, nil\n}\n\nfunc (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {\n\treturn adaptor.DoRequestHelper(a, c, meta, requestBody)\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tif meta.IsStream {\n\t\terr, usage = StreamHandler(c, resp)\n\t} else {\n\t\tswitch meta.Mode {\n\t\tcase relaymode.Embeddings:\n\t\t\terr, usage = EmbeddingHandler(c, resp)\n\t\tdefault:\n\t\t\terr, usage = Handler(c, resp)\n\t\t}\n\t}\n\treturn\n}\n\nfunc (a *Adaptor) GetModelList() []string {\n\treturn ModelList\n}\n\nfunc (a *Adaptor) GetChannelName() string {\n\treturn \"baidu\"\n}\n"
},
{
"path": "relay/adaptor/baidu/constants.go",
"content": "package baidu\n\nvar ModelList = []string{\n\t\"ERNIE-4.0-8K\",\n\t\"ERNIE-3.5-8K\",\n\t\"ERNIE-3.5-8K-0205\",\n\t\"ERNIE-3.5-8K-1222\",\n\t\"ERNIE-Bot-8K\",\n\t\"ERNIE-3.5-4K-0205\",\n\t\"ERNIE-Speed-8K\",\n\t\"ERNIE-Speed-128K\",\n\t\"ERNIE-Lite-8K-0922\",\n\t\"ERNIE-Lite-8K-0308\",\n\t\"ERNIE-Tiny-8K\",\n\t\"BLOOMZ-7B\",\n\t\"Embedding-V1\",\n\t\"bge-large-zh\",\n\t\"bge-large-en\",\n\t\"tao-8k\",\n}\n"
},
{
"path": "relay/adaptor/baidu/main.go",
"content": "package baidu\n\nimport (\n\t\"bufio\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"github.com/songquanpeng/one-api/common/render\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/client\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/constant\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\n// https://cloud.baidu.com/doc/WENXINWORKSHOP/s/flfmc9do2\n\ntype TokenResponse struct {\n\tExpiresIn int `json:\"expires_in\"`\n\tAccessToken string `json:\"access_token\"`\n}\n\ntype Message struct {\n\tRole string `json:\"role\"`\n\tContent string `json:\"content\"`\n}\n\ntype ChatRequest struct {\n\tMessages []Message `json:\"messages\"`\n\tTemperature *float64 `json:\"temperature,omitempty\"`\n\tTopP *float64 `json:\"top_p,omitempty\"`\n\tPenaltyScore *float64 `json:\"penalty_score,omitempty\"`\n\tStream bool `json:\"stream,omitempty\"`\n\tSystem string `json:\"system,omitempty\"`\n\tDisableSearch bool `json:\"disable_search,omitempty\"`\n\tEnableCitation bool `json:\"enable_citation,omitempty\"`\n\tMaxOutputTokens int `json:\"max_output_tokens,omitempty\"`\n\tUserId string `json:\"user_id,omitempty\"`\n}\n\ntype Error struct {\n\tErrorCode int `json:\"error_code\"`\n\tErrorMsg string `json:\"error_msg\"`\n}\n\nvar baiduTokenStore sync.Map\n\nfunc ConvertRequest(request model.GeneralOpenAIRequest) *ChatRequest {\n\tbaiduRequest := ChatRequest{\n\t\tMessages: make([]Message, 0, len(request.Messages)),\n\t\tTemperature: request.Temperature,\n\t\tTopP: request.TopP,\n\t\tPenaltyScore: request.FrequencyPenalty,\n\t\tStream: request.Stream,\n\t\tDisableSearch: false,\n\t\tEnableCitation: false,\n\t\tMaxOutputTokens: request.MaxTokens,\n\t\tUserId: request.User,\n\t}\n\tfor _, message := range request.Messages {\n\t\tif message.Role == \"system\" {\n\t\t\tbaiduRequest.System = message.StringContent()\n\t\t} else {\n\t\t\tbaiduRequest.Messages = append(baiduRequest.Messages, Message{\n\t\t\t\tRole: message.Role,\n\t\t\t\tContent: message.StringContent(),\n\t\t\t})\n\t\t}\n\t}\n\treturn &baiduRequest\n}\n\nfunc responseBaidu2OpenAI(response *ChatResponse) *openai.TextResponse {\n\tchoice := openai.TextResponseChoice{\n\t\tIndex: 0,\n\t\tMessage: model.Message{\n\t\t\tRole: \"assistant\",\n\t\t\tContent: response.Result,\n\t\t},\n\t\tFinishReason: \"stop\",\n\t}\n\tfullTextResponse := openai.TextResponse{\n\t\tId: response.Id,\n\t\tObject: \"chat.completion\",\n\t\tCreated: response.Created,\n\t\tChoices: []openai.TextResponseChoice{choice},\n\t\tUsage: response.Usage,\n\t}\n\treturn &fullTextResponse\n}\n\nfunc streamResponseBaidu2OpenAI(baiduResponse *ChatStreamResponse) *openai.ChatCompletionsStreamResponse {\n\tvar choice openai.ChatCompletionsStreamResponseChoice\n\tchoice.Delta.Content = baiduResponse.Result\n\tif baiduResponse.IsEnd {\n\t\tchoice.FinishReason = &constant.StopFinishReason\n\t}\n\tresponse := openai.ChatCompletionsStreamResponse{\n\t\tId: baiduResponse.Id,\n\t\tObject: \"chat.completion.chunk\",\n\t\tCreated: baiduResponse.Created,\n\t\tModel: \"ernie-bot\",\n\t\tChoices: []openai.ChatCompletionsStreamResponseChoice{choice},\n\t}\n\treturn &response\n}\n\nfunc ConvertEmbeddingRequest(request model.GeneralOpenAIRequest) *EmbeddingRequest {\n\treturn &EmbeddingRequest{\n\t\tInput: request.ParseInput(),\n\t}\n}\n\nfunc embeddingResponseBaidu2OpenAI(response *EmbeddingResponse) *openai.EmbeddingResponse {\n\topenAIEmbeddingResponse := openai.EmbeddingResponse{\n\t\tObject: \"list\",\n\t\tData: make([]openai.EmbeddingResponseItem, 0, len(response.Data)),\n\t\tModel: \"baidu-embedding\",\n\t\tUsage: response.Usage,\n\t}\n\tfor _, item := range response.Data {\n\t\topenAIEmbeddingResponse.Data = append(openAIEmbeddingResponse.Data, openai.EmbeddingResponseItem{\n\t\t\tObject: item.Object,\n\t\t\tIndex: item.Index,\n\t\t\tEmbedding: item.Embedding,\n\t\t})\n\t}\n\treturn &openAIEmbeddingResponse\n}\n\nfunc StreamHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tvar usage model.Usage\n\tscanner := bufio.NewScanner(resp.Body)\n\tscanner.Split(bufio.ScanLines)\n\n\tcommon.SetEventStreamHeaders(c)\n\n\tfor scanner.Scan() {\n\t\tdata := scanner.Text()\n\t\tif len(data) < 6 {\n\t\t\tcontinue\n\t\t}\n\t\tdata = data[6:]\n\n\t\tvar baiduResponse ChatStreamResponse\n\t\terr := json.Unmarshal([]byte(data), &baiduResponse)\n\t\tif err != nil {\n\t\t\tlogger.SysError(\"error unmarshalling stream response: \" + err.Error())\n\t\t\tcontinue\n\t\t}\n\t\tif baiduResponse.Usage.TotalTokens != 0 {\n\t\t\tusage.TotalTokens = baiduResponse.Usage.TotalTokens\n\t\t\tusage.PromptTokens = baiduResponse.Usage.PromptTokens\n\t\t\tusage.CompletionTokens = baiduResponse.Usage.TotalTokens - baiduResponse.Usage.PromptTokens\n\t\t}\n\t\tresponse := streamResponseBaidu2OpenAI(&baiduResponse)\n\t\terr = render.ObjectData(c, response)\n\t\tif err != nil {\n\t\t\tlogger.SysError(err.Error())\n\t\t}\n\t}\n\n\tif err := scanner.Err(); err != nil {\n\t\tlogger.SysError(\"error reading stream: \" + err.Error())\n\t}\n\n\trender.Done(c)\n\n\terr := resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\treturn nil, &usage\n}\n\nfunc Handler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tvar baiduResponse ChatResponse\n\tresponseBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = json.Unmarshal(responseBody, &baiduResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tif baiduResponse.ErrorMsg != \"\" {\n\t\treturn &model.ErrorWithStatusCode{\n\t\t\tError: model.Error{\n\t\t\t\tMessage: baiduResponse.ErrorMsg,\n\t\t\t\tType: \"baidu_error\",\n\t\t\t\tParam: \"\",\n\t\t\t\tCode: baiduResponse.ErrorCode,\n\t\t\t},\n\t\t\tStatusCode: resp.StatusCode,\n\t\t}, nil\n\t}\n\tfullTextResponse := responseBaidu2OpenAI(&baiduResponse)\n\tfullTextResponse.Model = \"ernie-bot\"\n\tjsonResponse, err := json.Marshal(fullTextResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tc.Writer.Header().Set(\"Content-Type\", \"application/json\")\n\tc.Writer.WriteHeader(resp.StatusCode)\n\t_, err = c.Writer.Write(jsonResponse)\n\treturn nil, &fullTextResponse.Usage\n}\n\nfunc EmbeddingHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tvar baiduResponse EmbeddingResponse\n\tresponseBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = json.Unmarshal(responseBody, &baiduResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tif baiduResponse.ErrorMsg != \"\" {\n\t\treturn &model.ErrorWithStatusCode{\n\t\t\tError: model.Error{\n\t\t\t\tMessage: baiduResponse.ErrorMsg,\n\t\t\t\tType: \"baidu_error\",\n\t\t\t\tParam: \"\",\n\t\t\t\tCode: baiduResponse.ErrorCode,\n\t\t\t},\n\t\t\tStatusCode: resp.StatusCode,\n\t\t}, nil\n\t}\n\tfullTextResponse := embeddingResponseBaidu2OpenAI(&baiduResponse)\n\tjsonResponse, err := json.Marshal(fullTextResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tc.Writer.Header().Set(\"Content-Type\", \"application/json\")\n\tc.Writer.WriteHeader(resp.StatusCode)\n\t_, err = c.Writer.Write(jsonResponse)\n\treturn nil, &fullTextResponse.Usage\n}\n\nfunc GetAccessToken(apiKey string) (string, error) {\n\tif val, ok := baiduTokenStore.Load(apiKey); ok {\n\t\tvar accessToken AccessToken\n\t\tif accessToken, ok = val.(AccessToken); ok {\n\t\t\t// soon this will expire\n\t\t\tif time.Now().Add(time.Hour).After(accessToken.ExpiresAt) {\n\t\t\t\tgo func() {\n\t\t\t\t\t_, _ = getBaiduAccessTokenHelper(apiKey)\n\t\t\t\t}()\n\t\t\t}\n\t\t\treturn accessToken.AccessToken, nil\n\t\t}\n\t}\n\taccessToken, err := getBaiduAccessTokenHelper(apiKey)\n\tif err != nil {\n\t\treturn \"\", err\n\t}\n\tif accessToken == nil {\n\t\treturn \"\", errors.New(\"GetAccessToken return a nil token\")\n\t}\n\treturn (*accessToken).AccessToken, nil\n}\n\nfunc getBaiduAccessTokenHelper(apiKey string) (*AccessToken, error) {\n\tparts := strings.Split(apiKey, \"|\")\n\tif len(parts) != 2 {\n\t\treturn nil, errors.New(\"invalid baidu apikey\")\n\t}\n\treq, err := http.NewRequest(\"POST\", fmt.Sprintf(\"https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id=%s&client_secret=%s\",\n\t\tparts[0], parts[1]), nil)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treq.Header.Add(\"Content-Type\", \"application/json\")\n\treq.Header.Add(\"Accept\", \"application/json\")\n\tres, err := client.ImpatientHTTPClient.Do(req)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tdefer res.Body.Close()\n\n\tvar accessToken AccessToken\n\terr = json.NewDecoder(res.Body).Decode(&accessToken)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif accessToken.Error != \"\" {\n\t\treturn nil, errors.New(accessToken.Error + \": \" + accessToken.ErrorDescription)\n\t}\n\tif accessToken.AccessToken == \"\" {\n\t\treturn nil, errors.New(\"getBaiduAccessTokenHelper get empty access token\")\n\t}\n\taccessToken.ExpiresAt = time.Now().Add(time.Duration(accessToken.ExpiresIn) * time.Second)\n\tbaiduTokenStore.Store(apiKey, accessToken)\n\treturn &accessToken, nil\n}\n"
},
{
"path": "relay/adaptor/baidu/model.go",
"content": "package baidu\n\nimport (\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"time\"\n)\n\ntype ChatResponse struct {\n\tId string `json:\"id\"`\n\tObject string `json:\"object\"`\n\tCreated int64 `json:\"created\"`\n\tResult string `json:\"result\"`\n\tIsTruncated bool `json:\"is_truncated\"`\n\tNeedClearHistory bool `json:\"need_clear_history\"`\n\tUsage model.Usage `json:\"usage\"`\n\tError\n}\n\ntype ChatStreamResponse struct {\n\tChatResponse\n\tSentenceId int `json:\"sentence_id\"`\n\tIsEnd bool `json:\"is_end\"`\n}\n\ntype EmbeddingRequest struct {\n\tInput []string `json:\"input\"`\n}\n\ntype EmbeddingData struct {\n\tObject string `json:\"object\"`\n\tEmbedding []float64 `json:\"embedding\"`\n\tIndex int `json:\"index\"`\n}\n\ntype EmbeddingResponse struct {\n\tId string `json:\"id\"`\n\tObject string `json:\"object\"`\n\tCreated int64 `json:\"created\"`\n\tData []EmbeddingData `json:\"data\"`\n\tUsage model.Usage `json:\"usage\"`\n\tError\n}\n\ntype AccessToken struct {\n\tAccessToken string `json:\"access_token\"`\n\tError string `json:\"error,omitempty\"`\n\tErrorDescription string `json:\"error_description,omitempty\"`\n\tExpiresIn int64 `json:\"expires_in,omitempty\"`\n\tExpiresAt time.Time `json:\"-\"`\n}\n"
},
{
"path": "relay/adaptor/baiduv2/constants.go",
"content": "package baiduv2\n\n// https://console.bce.baidu.com/support/?_=1692863460488×tamp=1739074632076#/api?product=QIANFAN&project=%E5%8D%83%E5%B8%86ModelBuilder&parent=%E5%AF%B9%E8%AF%9DChat%20V2&api=v2%2Fchat%2Fcompletions&method=post\n// https://cloud.baidu.com/doc/WENXINWORKSHOP/s/Fm2vrveyu#%E6%94%AF%E6%8C%81%E6%A8%A1%E5%9E%8B%E5%88%97%E8%A1%A8\n\nvar ModelList = []string{\n\t\"ernie-4.0-8k-latest\",\n\t\"ernie-4.0-8k-preview\",\n\t\"ernie-4.0-8k\",\n\t\"ernie-4.0-turbo-8k-latest\",\n\t\"ernie-4.0-turbo-8k-preview\",\n\t\"ernie-4.0-turbo-8k\",\n\t\"ernie-4.0-turbo-128k\",\n\t\"ernie-3.5-8k-preview\",\n\t\"ernie-3.5-8k\",\n\t\"ernie-3.5-128k\",\n\t\"ernie-speed-8k\",\n\t\"ernie-speed-128k\",\n\t\"ernie-speed-pro-128k\",\n\t\"ernie-lite-8k\",\n\t\"ernie-lite-pro-128k\",\n\t\"ernie-tiny-8k\",\n\t\"ernie-char-8k\",\n\t\"ernie-char-fiction-8k\",\n\t\"ernie-novel-8k\",\n\t\"deepseek-v3\",\n\t\"deepseek-r1\",\n\t\"deepseek-r1-distill-qwen-32b\",\n\t\"deepseek-r1-distill-qwen-14b\",\n}\n"
},
{
"path": "relay/adaptor/baiduv2/main.go",
"content": "package baiduv2\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/relaymode\"\n)\n\nfunc GetRequestURL(meta *meta.Meta) (string, error) {\n\tswitch meta.Mode {\n\tcase relaymode.ChatCompletions:\n\t\treturn fmt.Sprintf(\"%s/v2/chat/completions\", meta.BaseURL), nil\n\tdefault:\n\t}\n\treturn \"\", fmt.Errorf(\"unsupported relay mode %d for baidu v2\", meta.Mode)\n}\n"
},
{
"path": "relay/adaptor/cloudflare/adaptor.go",
"content": "package cloudflare\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"github.com/songquanpeng/one-api/relay/relaymode\"\n)\n\ntype Adaptor struct {\n\tmeta *meta.Meta\n}\n\n// ConvertImageRequest implements adaptor.Adaptor.\nfunc (*Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {\n\treturn nil, errors.New(\"not implemented\")\n}\n\n// ConvertImageRequest implements adaptor.Adaptor.\n\nfunc (a *Adaptor) Init(meta *meta.Meta) {\n\ta.meta = meta\n}\n\n// WorkerAI cannot be used across accounts with AIGateWay\n// https://developers.cloudflare.com/ai-gateway/providers/workersai/#openai-compatible-endpoints\n// https://gateway.ai.cloudflare.com/v1/{account_id}/{gateway_id}/workers-ai\nfunc (a *Adaptor) isAIGateWay(baseURL string) bool {\n\treturn strings.HasPrefix(baseURL, \"https://gateway.ai.cloudflare.com\") && strings.HasSuffix(baseURL, \"/workers-ai\")\n}\n\nfunc (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {\n\tisAIGateWay := a.isAIGateWay(meta.BaseURL)\n\tvar urlPrefix string\n\tif isAIGateWay {\n\t\turlPrefix = meta.BaseURL\n\t} else {\n\t\turlPrefix = fmt.Sprintf(\"%s/client/v4/accounts/%s/ai\", meta.BaseURL, meta.Config.UserID)\n\t}\n\n\tswitch meta.Mode {\n\tcase relaymode.ChatCompletions:\n\t\treturn fmt.Sprintf(\"%s/v1/chat/completions\", urlPrefix), nil\n\tcase relaymode.Embeddings:\n\t\treturn fmt.Sprintf(\"%s/v1/embeddings\", urlPrefix), nil\n\tdefault:\n\t\tif isAIGateWay {\n\t\t\treturn fmt.Sprintf(\"%s/%s\", urlPrefix, meta.ActualModelName), nil\n\t\t}\n\t\treturn fmt.Sprintf(\"%s/run/%s\", urlPrefix, meta.ActualModelName), nil\n\t}\n}\n\nfunc (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {\n\tadaptor.SetupCommonRequestHeader(c, req, meta)\n\treq.Header.Set(\"Authorization\", \"Bearer \"+meta.APIKey)\n\treturn nil\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\tswitch relayMode {\n\tcase relaymode.Completions:\n\t\treturn ConvertCompletionsRequest(*request), nil\n\tcase relaymode.ChatCompletions, relaymode.Embeddings:\n\t\treturn request, nil\n\tdefault:\n\t\treturn nil, errors.New(\"not implemented\")\n\t}\n}\n\nfunc (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {\n\treturn adaptor.DoRequestHelper(a, c, meta, requestBody)\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tif meta.IsStream {\n\t\terr, usage = StreamHandler(c, resp, meta.PromptTokens, meta.ActualModelName)\n\t} else {\n\t\terr, usage = Handler(c, resp, meta.PromptTokens, meta.ActualModelName)\n\t}\n\treturn\n}\n\nfunc (a *Adaptor) GetModelList() []string {\n\treturn ModelList\n}\n\nfunc (a *Adaptor) GetChannelName() string {\n\treturn \"cloudflare\"\n}\n"
},
{
"path": "relay/adaptor/cloudflare/constant.go",
"content": "package cloudflare\n\nvar ModelList = []string{\n\t\"@cf/meta/llama-3.1-8b-instruct\",\n\t\"@cf/meta/llama-2-7b-chat-fp16\",\n\t\"@cf/meta/llama-2-7b-chat-int8\",\n\t\"@cf/mistral/mistral-7b-instruct-v0.1\",\n\t\"@hf/thebloke/deepseek-coder-6.7b-base-awq\",\n\t\"@hf/thebloke/deepseek-coder-6.7b-instruct-awq\",\n\t\"@cf/deepseek-ai/deepseek-math-7b-base\",\n\t\"@cf/deepseek-ai/deepseek-math-7b-instruct\",\n\t\"@cf/thebloke/discolm-german-7b-v1-awq\",\n\t\"@cf/tiiuae/falcon-7b-instruct\",\n\t\"@cf/google/gemma-2b-it-lora\",\n\t\"@hf/google/gemma-7b-it\",\n\t\"@cf/google/gemma-7b-it-lora\",\n\t\"@hf/nousresearch/hermes-2-pro-mistral-7b\",\n\t\"@hf/thebloke/llama-2-13b-chat-awq\",\n\t\"@cf/meta-llama/llama-2-7b-chat-hf-lora\",\n\t\"@cf/meta/llama-3-8b-instruct\",\n\t\"@hf/thebloke/llamaguard-7b-awq\",\n\t\"@hf/thebloke/mistral-7b-instruct-v0.1-awq\",\n\t\"@hf/mistralai/mistral-7b-instruct-v0.2\",\n\t\"@cf/mistral/mistral-7b-instruct-v0.2-lora\",\n\t\"@hf/thebloke/neural-chat-7b-v3-1-awq\",\n\t\"@cf/openchat/openchat-3.5-0106\",\n\t\"@hf/thebloke/openhermes-2.5-mistral-7b-awq\",\n\t\"@cf/microsoft/phi-2\",\n\t\"@cf/qwen/qwen1.5-0.5b-chat\",\n\t\"@cf/qwen/qwen1.5-1.8b-chat\",\n\t\"@cf/qwen/qwen1.5-14b-chat-awq\",\n\t\"@cf/qwen/qwen1.5-7b-chat-awq\",\n\t\"@cf/defog/sqlcoder-7b-2\",\n\t\"@hf/nexusflow/starling-lm-7b-beta\",\n\t\"@cf/tinyllama/tinyllama-1.1b-chat-v1.0\",\n\t\"@hf/thebloke/zephyr-7b-beta-awq\",\n}\n"
},
{
"path": "relay/adaptor/cloudflare/main.go",
"content": "package cloudflare\n\nimport (\n\t\"bufio\"\n\t\"encoding/json\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n\n\t\"github.com/songquanpeng/one-api/common/ctxkey\"\n\t\"github.com/songquanpeng/one-api/common/render\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\nfunc ConvertCompletionsRequest(textRequest model.GeneralOpenAIRequest) *Request {\n\tp, _ := textRequest.Prompt.(string)\n\treturn &Request{\n\t\tPrompt: p,\n\t\tMaxTokens: textRequest.MaxTokens,\n\t\tStream: textRequest.Stream,\n\t\tTemperature: textRequest.Temperature,\n\t}\n}\n\nfunc StreamHandler(c *gin.Context, resp *http.Response, promptTokens int, modelName string) (*model.ErrorWithStatusCode, *model.Usage) {\n\tscanner := bufio.NewScanner(resp.Body)\n\tscanner.Split(bufio.ScanLines)\n\n\tcommon.SetEventStreamHeaders(c)\n\tid := helper.GetResponseID(c)\n\tresponseModel := c.GetString(ctxkey.OriginalModel)\n\tvar responseText string\n\n\tfor scanner.Scan() {\n\t\tdata := scanner.Text()\n\t\tif len(data) < len(\"data: \") {\n\t\t\tcontinue\n\t\t}\n\t\tdata = strings.TrimPrefix(data, \"data: \")\n\t\tdata = strings.TrimSuffix(data, \"\\r\")\n\n\t\tif data == \"[DONE]\" {\n\t\t\tbreak\n\t\t}\n\n\t\tvar response openai.ChatCompletionsStreamResponse\n\t\terr := json.Unmarshal([]byte(data), &response)\n\t\tif err != nil {\n\t\t\tlogger.SysError(\"error unmarshalling stream response: \" + err.Error())\n\t\t\tcontinue\n\t\t}\n\t\tfor _, v := range response.Choices {\n\t\t\tv.Delta.Role = \"assistant\"\n\t\t\tresponseText += v.Delta.StringContent()\n\t\t}\n\t\tresponse.Id = id\n\t\tresponse.Model = modelName\n\t\terr = render.ObjectData(c, response)\n\t\tif err != nil {\n\t\t\tlogger.SysError(err.Error())\n\t\t}\n\t}\n\n\tif err := scanner.Err(); err != nil {\n\t\tlogger.SysError(\"error reading stream: \" + err.Error())\n\t}\n\n\trender.Done(c)\n\n\terr := resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\n\tusage := openai.ResponseText2Usage(responseText, responseModel, promptTokens)\n\treturn nil, usage\n}\n\nfunc Handler(c *gin.Context, resp *http.Response, promptTokens int, modelName string) (*model.ErrorWithStatusCode, *model.Usage) {\n\tresponseBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tvar response openai.TextResponse\n\terr = json.Unmarshal(responseBody, &response)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tresponse.Model = modelName\n\tvar responseText string\n\tfor _, v := range response.Choices {\n\t\tresponseText += v.Message.Content.(string)\n\t}\n\tusage := openai.ResponseText2Usage(responseText, modelName, promptTokens)\n\tresponse.Usage = *usage\n\tresponse.Id = helper.GetResponseID(c)\n\tjsonResponse, err := json.Marshal(response)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tc.Writer.Header().Set(\"Content-Type\", \"application/json\")\n\tc.Writer.WriteHeader(resp.StatusCode)\n\t_, _ = c.Writer.Write(jsonResponse)\n\treturn nil, usage\n}\n"
},
{
"path": "relay/adaptor/cloudflare/model.go",
"content": "package cloudflare\n\nimport \"github.com/songquanpeng/one-api/relay/model\"\n\ntype Request struct {\n\tMessages []model.Message `json:\"messages,omitempty\"`\n\tLora string `json:\"lora,omitempty\"`\n\tMaxTokens int `json:\"max_tokens,omitempty\"`\n\tPrompt string `json:\"prompt,omitempty\"`\n\tRaw bool `json:\"raw,omitempty\"`\n\tStream bool `json:\"stream,omitempty\"`\n\tTemperature *float64 `json:\"temperature,omitempty\"`\n}\n"
},
{
"path": "relay/adaptor/cohere/adaptor.go",
"content": "package cohere\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"io\"\n\t\"net/http\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\ntype Adaptor struct{}\n\n// ConvertImageRequest implements adaptor.Adaptor.\nfunc (*Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {\n\treturn nil, errors.New(\"not implemented\")\n}\n\n// ConvertImageRequest implements adaptor.Adaptor.\n\nfunc (a *Adaptor) Init(meta *meta.Meta) {\n\n}\n\nfunc (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {\n\treturn fmt.Sprintf(\"%s/v1/chat\", meta.BaseURL), nil\n}\n\nfunc (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {\n\tadaptor.SetupCommonRequestHeader(c, req, meta)\n\treq.Header.Set(\"Authorization\", \"Bearer \"+meta.APIKey)\n\treturn nil\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\treturn ConvertRequest(*request), nil\n}\n\nfunc (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {\n\treturn adaptor.DoRequestHelper(a, c, meta, requestBody)\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tif meta.IsStream {\n\t\terr, usage = StreamHandler(c, resp)\n\t} else {\n\t\terr, usage = Handler(c, resp, meta.PromptTokens, meta.ActualModelName)\n\t}\n\treturn\n}\n\nfunc (a *Adaptor) GetModelList() []string {\n\treturn ModelList\n}\n\nfunc (a *Adaptor) GetChannelName() string {\n\treturn \"Cohere\"\n}\n"
},
{
"path": "relay/adaptor/cohere/constant.go",
"content": "package cohere\n\nvar ModelList = []string{\n\t\"command\", \"command-nightly\",\n\t\"command-light\", \"command-light-nightly\",\n\t\"command-r\", \"command-r-plus\",\n}\n\nfunc init() {\n\tnum := len(ModelList)\n\tfor i := 0; i < num; i++ {\n\t\tModelList = append(ModelList, ModelList[i]+\"-internet\")\n\t}\n}\n"
},
{
"path": "relay/adaptor/cohere/main.go",
"content": "package cohere\n\nimport (\n\t\"bufio\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"github.com/songquanpeng/one-api/common/render\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\nvar (\n\tWebSearchConnector = Connector{ID: \"web-search\"}\n)\n\nfunc stopReasonCohere2OpenAI(reason *string) string {\n\tif reason == nil {\n\t\treturn \"\"\n\t}\n\tswitch *reason {\n\tcase \"COMPLETE\":\n\t\treturn \"stop\"\n\tdefault:\n\t\treturn *reason\n\t}\n}\n\nfunc ConvertRequest(textRequest model.GeneralOpenAIRequest) *Request {\n\tcohereRequest := Request{\n\t\tModel: textRequest.Model,\n\t\tMessage: \"\",\n\t\tMaxTokens: textRequest.MaxTokens,\n\t\tTemperature: textRequest.Temperature,\n\t\tP: textRequest.TopP,\n\t\tK: textRequest.TopK,\n\t\tStream: textRequest.Stream,\n\t\tFrequencyPenalty: textRequest.FrequencyPenalty,\n\t\tPresencePenalty: textRequest.PresencePenalty,\n\t\tSeed: int(textRequest.Seed),\n\t}\n\tif cohereRequest.Model == \"\" {\n\t\tcohereRequest.Model = \"command-r\"\n\t}\n\tif strings.HasSuffix(cohereRequest.Model, \"-internet\") {\n\t\tcohereRequest.Model = strings.TrimSuffix(cohereRequest.Model, \"-internet\")\n\t\tcohereRequest.Connectors = append(cohereRequest.Connectors, WebSearchConnector)\n\t}\n\tfor _, message := range textRequest.Messages {\n\t\tif message.Role == \"user\" {\n\t\t\tcohereRequest.Message = message.Content.(string)\n\t\t} else {\n\t\t\tvar role string\n\t\t\tif message.Role == \"assistant\" {\n\t\t\t\trole = \"CHATBOT\"\n\t\t\t} else if message.Role == \"system\" {\n\t\t\t\trole = \"SYSTEM\"\n\t\t\t} else {\n\t\t\t\trole = \"USER\"\n\t\t\t}\n\t\t\tcohereRequest.ChatHistory = append(cohereRequest.ChatHistory, ChatMessage{\n\t\t\t\tRole: role,\n\t\t\t\tMessage: message.Content.(string),\n\t\t\t})\n\t\t}\n\t}\n\treturn &cohereRequest\n}\n\nfunc StreamResponseCohere2OpenAI(cohereResponse *StreamResponse) (*openai.ChatCompletionsStreamResponse, *Response) {\n\tvar response *Response\n\tvar responseText string\n\tvar finishReason string\n\n\tswitch cohereResponse.EventType {\n\tcase \"stream-start\":\n\t\treturn nil, nil\n\tcase \"text-generation\":\n\t\tresponseText += cohereResponse.Text\n\tcase \"stream-end\":\n\t\tusage := cohereResponse.Response.Meta.Tokens\n\t\tresponse = &Response{\n\t\t\tMeta: Meta{\n\t\t\t\tTokens: Usage{\n\t\t\t\t\tInputTokens: usage.InputTokens,\n\t\t\t\t\tOutputTokens: usage.OutputTokens,\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\t\tfinishReason = *cohereResponse.Response.FinishReason\n\tdefault:\n\t\treturn nil, nil\n\t}\n\n\tvar choice openai.ChatCompletionsStreamResponseChoice\n\tchoice.Delta.Content = responseText\n\tchoice.Delta.Role = \"assistant\"\n\tif finishReason != \"\" {\n\t\tchoice.FinishReason = &finishReason\n\t}\n\tvar openaiResponse openai.ChatCompletionsStreamResponse\n\topenaiResponse.Object = \"chat.completion.chunk\"\n\topenaiResponse.Choices = []openai.ChatCompletionsStreamResponseChoice{choice}\n\treturn &openaiResponse, response\n}\n\nfunc ResponseCohere2OpenAI(cohereResponse *Response) *openai.TextResponse {\n\tchoice := openai.TextResponseChoice{\n\t\tIndex: 0,\n\t\tMessage: model.Message{\n\t\t\tRole: \"assistant\",\n\t\t\tContent: cohereResponse.Text,\n\t\t\tName: nil,\n\t\t},\n\t\tFinishReason: stopReasonCohere2OpenAI(cohereResponse.FinishReason),\n\t}\n\tfullTextResponse := openai.TextResponse{\n\t\tId: fmt.Sprintf(\"chatcmpl-%s\", cohereResponse.ResponseID),\n\t\tModel: \"model\",\n\t\tObject: \"chat.completion\",\n\t\tCreated: helper.GetTimestamp(),\n\t\tChoices: []openai.TextResponseChoice{choice},\n\t}\n\treturn &fullTextResponse\n}\n\nfunc StreamHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tcreatedTime := helper.GetTimestamp()\n\tscanner := bufio.NewScanner(resp.Body)\n\tscanner.Split(bufio.ScanLines)\n\n\tcommon.SetEventStreamHeaders(c)\n\tvar usage model.Usage\n\n\tfor scanner.Scan() {\n\t\tdata := scanner.Text()\n\t\tdata = strings.TrimSuffix(data, \"\\r\")\n\n\t\tvar cohereResponse StreamResponse\n\t\terr := json.Unmarshal([]byte(data), &cohereResponse)\n\t\tif err != nil {\n\t\t\tlogger.SysError(\"error unmarshalling stream response: \" + err.Error())\n\t\t\tcontinue\n\t\t}\n\n\t\tresponse, meta := StreamResponseCohere2OpenAI(&cohereResponse)\n\t\tif meta != nil {\n\t\t\tusage.PromptTokens += meta.Meta.Tokens.InputTokens\n\t\t\tusage.CompletionTokens += meta.Meta.Tokens.OutputTokens\n\t\t\tcontinue\n\t\t}\n\t\tif response == nil {\n\t\t\tcontinue\n\t\t}\n\n\t\tresponse.Id = fmt.Sprintf(\"chatcmpl-%d\", createdTime)\n\t\tresponse.Model = c.GetString(\"original_model\")\n\t\tresponse.Created = createdTime\n\n\t\terr = render.ObjectData(c, response)\n\t\tif err != nil {\n\t\t\tlogger.SysError(err.Error())\n\t\t}\n\t}\n\n\tif err := scanner.Err(); err != nil {\n\t\tlogger.SysError(\"error reading stream: \" + err.Error())\n\t}\n\n\trender.Done(c)\n\n\terr := resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\n\treturn nil, &usage\n}\n\nfunc Handler(c *gin.Context, resp *http.Response, promptTokens int, modelName string) (*model.ErrorWithStatusCode, *model.Usage) {\n\tresponseBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tvar cohereResponse Response\n\terr = json.Unmarshal(responseBody, &cohereResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tif cohereResponse.ResponseID == \"\" {\n\t\treturn &model.ErrorWithStatusCode{\n\t\t\tError: model.Error{\n\t\t\t\tMessage: cohereResponse.Message,\n\t\t\t\tType: cohereResponse.Message,\n\t\t\t\tParam: \"\",\n\t\t\t\tCode: resp.StatusCode,\n\t\t\t},\n\t\t\tStatusCode: resp.StatusCode,\n\t\t}, nil\n\t}\n\tfullTextResponse := ResponseCohere2OpenAI(&cohereResponse)\n\tfullTextResponse.Model = modelName\n\tusage := model.Usage{\n\t\tPromptTokens: cohereResponse.Meta.Tokens.InputTokens,\n\t\tCompletionTokens: cohereResponse.Meta.Tokens.OutputTokens,\n\t\tTotalTokens: cohereResponse.Meta.Tokens.InputTokens + cohereResponse.Meta.Tokens.OutputTokens,\n\t}\n\tfullTextResponse.Usage = usage\n\tjsonResponse, err := json.Marshal(fullTextResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tc.Writer.Header().Set(\"Content-Type\", \"application/json\")\n\tc.Writer.WriteHeader(resp.StatusCode)\n\t_, err = c.Writer.Write(jsonResponse)\n\treturn nil, &usage\n}\n"
},
{
"path": "relay/adaptor/cohere/model.go",
"content": "package cohere\n\ntype Request struct {\n\tMessage string `json:\"message\" required:\"true\"`\n\tModel string `json:\"model,omitempty\"` // 默认值为\"command-r\"\n\tStream bool `json:\"stream,omitempty\"` // 默认值为false\n\tPreamble string `json:\"preamble,omitempty\"`\n\tChatHistory []ChatMessage `json:\"chat_history,omitempty\"`\n\tConversationID string `json:\"conversation_id,omitempty\"`\n\tPromptTruncation string `json:\"prompt_truncation,omitempty\"` // 默认值为\"AUTO\"\n\tConnectors []Connector `json:\"connectors,omitempty\"`\n\tDocuments []Document `json:\"documents,omitempty\"`\n\tTemperature *float64 `json:\"temperature,omitempty\"` // 默认值为0.3\n\tMaxTokens int `json:\"max_tokens,omitempty\"`\n\tMaxInputTokens int `json:\"max_input_tokens,omitempty\"`\n\tK int `json:\"k,omitempty\"` // 默认值为0\n\tP *float64 `json:\"p,omitempty\"` // 默认值为0.75\n\tSeed int `json:\"seed,omitempty\"`\n\tStopSequences []string `json:\"stop_sequences,omitempty\"`\n\tFrequencyPenalty *float64 `json:\"frequency_penalty,omitempty\"` // 默认值为0.0\n\tPresencePenalty *float64 `json:\"presence_penalty,omitempty\"` // 默认值为0.0\n\tTools []Tool `json:\"tools,omitempty\"`\n\tToolResults []ToolResult `json:\"tool_results,omitempty\"`\n}\n\ntype ChatMessage struct {\n\tRole string `json:\"role\" required:\"true\"`\n\tMessage string `json:\"message\" required:\"true\"`\n}\n\ntype Tool struct {\n\tName string `json:\"name\" required:\"true\"`\n\tDescription string `json:\"description\" required:\"true\"`\n\tParameterDefinitions map[string]ParameterSpec `json:\"parameter_definitions\"`\n}\n\ntype ParameterSpec struct {\n\tDescription string `json:\"description\"`\n\tType string `json:\"type\" required:\"true\"`\n\tRequired bool `json:\"required\"`\n}\n\ntype ToolResult struct {\n\tCall ToolCall `json:\"call\"`\n\tOutputs []map[string]interface{} `json:\"outputs\"`\n}\n\ntype ToolCall struct {\n\tName string `json:\"name\" required:\"true\"`\n\tParameters map[string]interface{} `json:\"parameters\" required:\"true\"`\n}\n\ntype StreamResponse struct {\n\tIsFinished bool `json:\"is_finished\"`\n\tEventType string `json:\"event_type\"`\n\tGenerationID string `json:\"generation_id,omitempty\"`\n\tSearchQueries []*SearchQuery `json:\"search_queries,omitempty\"`\n\tSearchResults []*SearchResult `json:\"search_results,omitempty\"`\n\tDocuments []*Document `json:\"documents,omitempty\"`\n\tText string `json:\"text,omitempty\"`\n\tCitations []*Citation `json:\"citations,omitempty\"`\n\tResponse *Response `json:\"response,omitempty\"`\n\tFinishReason string `json:\"finish_reason,omitempty\"`\n}\n\ntype SearchQuery struct {\n\tText string `json:\"text\"`\n\tGenerationID string `json:\"generation_id\"`\n}\n\ntype SearchResult struct {\n\tSearchQuery *SearchQuery `json:\"search_query\"`\n\tDocumentIDs []string `json:\"document_ids\"`\n\tConnector *Connector `json:\"connector\"`\n}\n\ntype Connector struct {\n\tID string `json:\"id\"`\n}\n\ntype Document struct {\n\tID string `json:\"id\"`\n\tSnippet string `json:\"snippet\"`\n\tTimestamp string `json:\"timestamp\"`\n\tTitle string `json:\"title\"`\n\tURL string `json:\"url\"`\n}\n\ntype Citation struct {\n\tStart int `json:\"start\"`\n\tEnd int `json:\"end\"`\n\tText string `json:\"text\"`\n\tDocumentIDs []string `json:\"document_ids\"`\n}\n\ntype Response struct {\n\tResponseID string `json:\"response_id\"`\n\tText string `json:\"text\"`\n\tGenerationID string `json:\"generation_id\"`\n\tChatHistory []*Message `json:\"chat_history\"`\n\tFinishReason *string `json:\"finish_reason\"`\n\tMeta Meta `json:\"meta\"`\n\tCitations []*Citation `json:\"citations\"`\n\tDocuments []*Document `json:\"documents\"`\n\tSearchResults []*SearchResult `json:\"search_results\"`\n\tSearchQueries []*SearchQuery `json:\"search_queries\"`\n\tMessage string `json:\"message\"`\n}\n\ntype Message struct {\n\tRole string `json:\"role\"`\n\tMessage string `json:\"message\"`\n}\n\ntype Version struct {\n\tVersion string `json:\"version\"`\n}\n\ntype Units struct {\n\tInputTokens int `json:\"input_tokens\"`\n\tOutputTokens int `json:\"output_tokens\"`\n}\n\ntype ChatEntry struct {\n\tRole string `json:\"role\"`\n\tMessage string `json:\"message\"`\n}\n\ntype Meta struct {\n\tAPIVersion APIVersion `json:\"api_version\"`\n\tBilledUnits BilledUnits `json:\"billed_units\"`\n\tTokens Usage `json:\"tokens\"`\n}\n\ntype APIVersion struct {\n\tVersion string `json:\"version\"`\n}\n\ntype BilledUnits struct {\n\tInputTokens int `json:\"input_tokens\"`\n\tOutputTokens int `json:\"output_tokens\"`\n}\n\ntype Usage struct {\n\tInputTokens int `json:\"input_tokens\"`\n\tOutputTokens int `json:\"output_tokens\"`\n}\n"
},
{
"path": "relay/adaptor/common.go",
"content": "package adaptor\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common/client\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"io\"\n\t\"net/http\"\n)\n\nfunc SetupCommonRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) {\n\treq.Header.Set(\"Content-Type\", c.Request.Header.Get(\"Content-Type\"))\n\treq.Header.Set(\"Accept\", c.Request.Header.Get(\"Accept\"))\n\tif meta.IsStream && c.Request.Header.Get(\"Accept\") == \"\" {\n\t\treq.Header.Set(\"Accept\", \"text/event-stream\")\n\t}\n}\n\nfunc DoRequestHelper(a Adaptor, c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {\n\tfullRequestURL, err := a.GetRequestURL(meta)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"get request url failed: %w\", err)\n\t}\n\treq, err := http.NewRequest(c.Request.Method, fullRequestURL, requestBody)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"new request failed: %w\", err)\n\t}\n\terr = a.SetupRequestHeader(c, req, meta)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"setup request header failed: %w\", err)\n\t}\n\tresp, err := DoRequest(c, req)\n\tif err != nil {\n\t\treturn nil, fmt.Errorf(\"do request failed: %w\", err)\n\t}\n\treturn resp, nil\n}\n\nfunc DoRequest(c *gin.Context, req *http.Request) (*http.Response, error) {\n\tresp, err := client.HTTPClient.Do(req)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif resp == nil {\n\t\treturn nil, errors.New(\"resp is nil\")\n\t}\n\t_ = req.Body.Close()\n\t_ = c.Request.Body.Close()\n\treturn resp, nil\n}\n"
},
{
"path": "relay/adaptor/coze/adaptor.go",
"content": "package coze\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"io\"\n\t\"net/http\"\n)\n\ntype Adaptor struct {\n\tmeta *meta.Meta\n}\n\nfunc (a *Adaptor) Init(meta *meta.Meta) {\n\ta.meta = meta\n}\n\nfunc (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {\n\treturn fmt.Sprintf(\"%s/open_api/v2/chat\", meta.BaseURL), nil\n}\n\nfunc (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {\n\tadaptor.SetupCommonRequestHeader(c, req, meta)\n\treq.Header.Set(\"Authorization\", \"Bearer \"+meta.APIKey)\n\treturn nil\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\trequest.User = a.meta.Config.UserID\n\treturn ConvertRequest(*request), nil\n}\n\nfunc (a *Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\treturn request, nil\n}\n\nfunc (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {\n\treturn adaptor.DoRequestHelper(a, c, meta, requestBody)\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tvar responseText *string\n\tif meta.IsStream {\n\t\terr, responseText = StreamHandler(c, resp)\n\t} else {\n\t\terr, responseText = Handler(c, resp, meta.PromptTokens, meta.ActualModelName)\n\t}\n\tif responseText != nil {\n\t\tusage = openai.ResponseText2Usage(*responseText, meta.ActualModelName, meta.PromptTokens)\n\t} else {\n\t\tusage = &model.Usage{}\n\t}\n\tusage.PromptTokens = meta.PromptTokens\n\tusage.TotalTokens = usage.PromptTokens + usage.CompletionTokens\n\treturn\n}\n\nfunc (a *Adaptor) GetModelList() []string {\n\treturn ModelList\n}\n\nfunc (a *Adaptor) GetChannelName() string {\n\treturn \"coze\"\n}\n"
},
{
"path": "relay/adaptor/coze/constant/contenttype/define.go",
"content": "package contenttype\n\nconst (\n\tText = \"text\"\n)\n"
},
{
"path": "relay/adaptor/coze/constant/event/define.go",
"content": "package event\n\nconst (\n\tMessage = \"message\"\n\tDone = \"done\"\n\tError = \"error\"\n)\n"
},
{
"path": "relay/adaptor/coze/constant/messagetype/define.go",
"content": "package messagetype\n\nconst (\n\tAnswer = \"answer\"\n\tFollowUp = \"follow_up\"\n)\n"
},
{
"path": "relay/adaptor/coze/constants.go",
"content": "package coze\n\nvar ModelList = []string{}\n"
},
{
"path": "relay/adaptor/coze/helper.go",
"content": "package coze\n\nimport \"github.com/songquanpeng/one-api/relay/adaptor/coze/constant/event\"\n\nfunc event2StopReason(e *string) string {\n\tif e == nil || *e == event.Message {\n\t\treturn \"\"\n\t}\n\treturn \"stop\"\n}\n"
},
{
"path": "relay/adaptor/coze/main.go",
"content": "package coze\n\nimport (\n\t\"bufio\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"github.com/songquanpeng/one-api/common/render\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/conv\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/coze/constant/messagetype\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\n// https://www.coze.com/open\n\nfunc stopReasonCoze2OpenAI(reason *string) string {\n\tif reason == nil {\n\t\treturn \"\"\n\t}\n\tswitch *reason {\n\tcase \"end_turn\":\n\t\treturn \"stop\"\n\tcase \"stop_sequence\":\n\t\treturn \"stop\"\n\tcase \"max_tokens\":\n\t\treturn \"length\"\n\tdefault:\n\t\treturn *reason\n\t}\n}\n\nfunc ConvertRequest(textRequest model.GeneralOpenAIRequest) *Request {\n\tcozeRequest := Request{\n\t\tStream: textRequest.Stream,\n\t\tUser: textRequest.User,\n\t\tBotId: strings.TrimPrefix(textRequest.Model, \"bot-\"),\n\t}\n\tfor i, message := range textRequest.Messages {\n\t\tif i == len(textRequest.Messages)-1 {\n\t\t\tcozeRequest.Query = message.StringContent()\n\t\t\tcontinue\n\t\t}\n\t\tcozeMessage := Message{\n\t\t\tRole: message.Role,\n\t\t\tContent: message.StringContent(),\n\t\t}\n\t\tcozeRequest.ChatHistory = append(cozeRequest.ChatHistory, cozeMessage)\n\t}\n\treturn &cozeRequest\n}\n\nfunc StreamResponseCoze2OpenAI(cozeResponse *StreamResponse) (*openai.ChatCompletionsStreamResponse, *Response) {\n\tvar response *Response\n\tvar stopReason string\n\tvar choice openai.ChatCompletionsStreamResponseChoice\n\n\tif cozeResponse.Message != nil {\n\t\tif cozeResponse.Message.Type != messagetype.Answer {\n\t\t\treturn nil, nil\n\t\t}\n\t\tchoice.Delta.Content = cozeResponse.Message.Content\n\t}\n\tchoice.Delta.Role = \"assistant\"\n\tfinishReason := stopReasonCoze2OpenAI(&stopReason)\n\tif finishReason != \"null\" {\n\t\tchoice.FinishReason = &finishReason\n\t}\n\tvar openaiResponse openai.ChatCompletionsStreamResponse\n\topenaiResponse.Object = \"chat.completion.chunk\"\n\topenaiResponse.Choices = []openai.ChatCompletionsStreamResponseChoice{choice}\n\topenaiResponse.Id = cozeResponse.ConversationId\n\treturn &openaiResponse, response\n}\n\nfunc ResponseCoze2OpenAI(cozeResponse *Response) *openai.TextResponse {\n\tvar responseText string\n\tfor _, message := range cozeResponse.Messages {\n\t\tif message.Type == messagetype.Answer {\n\t\t\tresponseText = message.Content\n\t\t\tbreak\n\t\t}\n\t}\n\tchoice := openai.TextResponseChoice{\n\t\tIndex: 0,\n\t\tMessage: model.Message{\n\t\t\tRole: \"assistant\",\n\t\t\tContent: responseText,\n\t\t\tName: nil,\n\t\t},\n\t\tFinishReason: \"stop\",\n\t}\n\tfullTextResponse := openai.TextResponse{\n\t\tId: fmt.Sprintf(\"chatcmpl-%s\", cozeResponse.ConversationId),\n\t\tModel: \"coze-bot\",\n\t\tObject: \"chat.completion\",\n\t\tCreated: helper.GetTimestamp(),\n\t\tChoices: []openai.TextResponseChoice{choice},\n\t}\n\treturn &fullTextResponse\n}\n\nfunc StreamHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *string) {\n\tvar responseText string\n\tcreatedTime := helper.GetTimestamp()\n\tscanner := bufio.NewScanner(resp.Body)\n\tscanner.Split(bufio.ScanLines)\n\n\tcommon.SetEventStreamHeaders(c)\n\tvar modelName string\n\n\tfor scanner.Scan() {\n\t\tdata := scanner.Text()\n\t\tif len(data) < 5 || !strings.HasPrefix(data, \"data:\") {\n\t\t\tcontinue\n\t\t}\n\t\tdata = strings.TrimPrefix(data, \"data:\")\n\t\tdata = strings.TrimSuffix(data, \"\\r\")\n\n\t\tvar cozeResponse StreamResponse\n\t\terr := json.Unmarshal([]byte(data), &cozeResponse)\n\t\tif err != nil {\n\t\t\tlogger.SysError(\"error unmarshalling stream response: \" + err.Error())\n\t\t\tcontinue\n\t\t}\n\n\t\tresponse, _ := StreamResponseCoze2OpenAI(&cozeResponse)\n\t\tif response == nil {\n\t\t\tcontinue\n\t\t}\n\n\t\tfor _, choice := range response.Choices {\n\t\t\tresponseText += conv.AsString(choice.Delta.Content)\n\t\t}\n\t\tresponse.Model = modelName\n\t\tresponse.Created = createdTime\n\n\t\terr = render.ObjectData(c, response)\n\t\tif err != nil {\n\t\t\tlogger.SysError(err.Error())\n\t\t}\n\t}\n\n\tif err := scanner.Err(); err != nil {\n\t\tlogger.SysError(\"error reading stream: \" + err.Error())\n\t}\n\n\trender.Done(c)\n\n\terr := resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\n\treturn nil, &responseText\n}\n\nfunc Handler(c *gin.Context, resp *http.Response, promptTokens int, modelName string) (*model.ErrorWithStatusCode, *string) {\n\tresponseBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tvar cozeResponse Response\n\terr = json.Unmarshal(responseBody, &cozeResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tif cozeResponse.Code != 0 {\n\t\treturn &model.ErrorWithStatusCode{\n\t\t\tError: model.Error{\n\t\t\t\tMessage: cozeResponse.Msg,\n\t\t\t\tCode: cozeResponse.Code,\n\t\t\t},\n\t\t\tStatusCode: resp.StatusCode,\n\t\t}, nil\n\t}\n\tfullTextResponse := ResponseCoze2OpenAI(&cozeResponse)\n\tfullTextResponse.Model = modelName\n\tjsonResponse, err := json.Marshal(fullTextResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tc.Writer.Header().Set(\"Content-Type\", \"application/json\")\n\tc.Writer.WriteHeader(resp.StatusCode)\n\t_, err = c.Writer.Write(jsonResponse)\n\tvar responseText string\n\tif len(fullTextResponse.Choices) > 0 {\n\t\tresponseText = fullTextResponse.Choices[0].Message.StringContent()\n\t}\n\treturn nil, &responseText\n}\n"
},
{
"path": "relay/adaptor/coze/model.go",
"content": "package coze\n\ntype Message struct {\n\tRole string `json:\"role\"`\n\tType string `json:\"type\"`\n\tContent string `json:\"content\"`\n\tContentType string `json:\"content_type\"`\n}\n\ntype ErrorInformation struct {\n\tCode int `json:\"code\"`\n\tMsg string `json:\"msg\"`\n}\n\ntype Request struct {\n\tConversationId string `json:\"conversation_id,omitempty\"`\n\tBotId string `json:\"bot_id\"`\n\tUser string `json:\"user\"`\n\tQuery string `json:\"query\"`\n\tChatHistory []Message `json:\"chat_history,omitempty\"`\n\tStream bool `json:\"stream\"`\n}\n\ntype Response struct {\n\tConversationId string `json:\"conversation_id,omitempty\"`\n\tMessages []Message `json:\"messages,omitempty\"`\n\tCode int `json:\"code,omitempty\"`\n\tMsg string `json:\"msg,omitempty\"`\n}\n\ntype StreamResponse struct {\n\tEvent string `json:\"event,omitempty\"`\n\tMessage *Message `json:\"message,omitempty\"`\n\tIsFinish bool `json:\"is_finish,omitempty\"`\n\tIndex int `json:\"index,omitempty\"`\n\tConversationId string `json:\"conversation_id,omitempty\"`\n\tErrorInformation *ErrorInformation `json:\"error_information,omitempty\"`\n}\n"
},
{
"path": "relay/adaptor/deepl/adaptor.go",
"content": "package deepl\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"io\"\n\t\"net/http\"\n)\n\ntype Adaptor struct {\n\tmeta *meta.Meta\n\tpromptText string\n}\n\nfunc (a *Adaptor) Init(meta *meta.Meta) {\n\ta.meta = meta\n}\n\nfunc (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {\n\treturn fmt.Sprintf(\"%s/v2/translate\", meta.BaseURL), nil\n}\n\nfunc (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {\n\tadaptor.SetupCommonRequestHeader(c, req, meta)\n\treq.Header.Set(\"Authorization\", \"DeepL-Auth-Key \"+meta.APIKey)\n\treturn nil\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\tconvertedRequest, text := ConvertRequest(*request)\n\ta.promptText = text\n\treturn convertedRequest, nil\n}\n\nfunc (a *Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\treturn request, nil\n}\n\nfunc (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {\n\treturn adaptor.DoRequestHelper(a, c, meta, requestBody)\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tif meta.IsStream {\n\t\terr = StreamHandler(c, resp, meta.ActualModelName)\n\t} else {\n\t\terr = Handler(c, resp, meta.ActualModelName)\n\t}\n\tpromptTokens := len(a.promptText)\n\tusage = &model.Usage{\n\t\tPromptTokens: promptTokens,\n\t\tTotalTokens: promptTokens,\n\t}\n\treturn\n}\n\nfunc (a *Adaptor) GetModelList() []string {\n\treturn ModelList\n}\n\nfunc (a *Adaptor) GetChannelName() string {\n\treturn \"deepl\"\n}\n"
},
{
"path": "relay/adaptor/deepl/constants.go",
"content": "package deepl\n\n// https://developers.deepl.com/docs/api-reference/glossaries\n\nvar ModelList = []string{\n\t\"deepl-zh\",\n\t\"deepl-en\",\n\t\"deepl-ja\",\n}\n"
},
{
"path": "relay/adaptor/deepl/helper.go",
"content": "package deepl\n\nimport \"strings\"\n\nfunc parseLangFromModelName(modelName string) string {\n\tparts := strings.Split(modelName, \"-\")\n\tif len(parts) == 1 {\n\t\treturn \"ZH\"\n\t}\n\treturn parts[1]\n}\n"
},
{
"path": "relay/adaptor/deepl/main.go",
"content": "package deepl\n\nimport (\n\t\"encoding/json\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/constant\"\n\t\"github.com/songquanpeng/one-api/relay/constant/finishreason\"\n\t\"github.com/songquanpeng/one-api/relay/constant/role\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"io\"\n\t\"net/http\"\n)\n\n// https://developers.deepl.com/docs/getting-started/your-first-api-request\n\nfunc ConvertRequest(textRequest model.GeneralOpenAIRequest) (*Request, string) {\n\tvar text string\n\tif len(textRequest.Messages) != 0 {\n\t\ttext = textRequest.Messages[len(textRequest.Messages)-1].StringContent()\n\t}\n\tdeeplRequest := Request{\n\t\tTargetLang: parseLangFromModelName(textRequest.Model),\n\t\tText: []string{text},\n\t}\n\treturn &deeplRequest, text\n}\n\nfunc StreamResponseDeepL2OpenAI(deeplResponse *Response) *openai.ChatCompletionsStreamResponse {\n\tvar choice openai.ChatCompletionsStreamResponseChoice\n\tif len(deeplResponse.Translations) != 0 {\n\t\tchoice.Delta.Content = deeplResponse.Translations[0].Text\n\t}\n\tchoice.Delta.Role = role.Assistant\n\tchoice.FinishReason = &constant.StopFinishReason\n\topenaiResponse := openai.ChatCompletionsStreamResponse{\n\t\tObject: constant.StreamObject,\n\t\tCreated: helper.GetTimestamp(),\n\t\tChoices: []openai.ChatCompletionsStreamResponseChoice{choice},\n\t}\n\treturn &openaiResponse\n}\n\nfunc ResponseDeepL2OpenAI(deeplResponse *Response) *openai.TextResponse {\n\tvar responseText string\n\tif len(deeplResponse.Translations) != 0 {\n\t\tresponseText = deeplResponse.Translations[0].Text\n\t}\n\tchoice := openai.TextResponseChoice{\n\t\tIndex: 0,\n\t\tMessage: model.Message{\n\t\t\tRole: role.Assistant,\n\t\t\tContent: responseText,\n\t\t\tName: nil,\n\t\t},\n\t\tFinishReason: finishreason.Stop,\n\t}\n\tfullTextResponse := openai.TextResponse{\n\t\tObject: constant.NonStreamObject,\n\t\tCreated: helper.GetTimestamp(),\n\t\tChoices: []openai.TextResponseChoice{choice},\n\t}\n\treturn &fullTextResponse\n}\n\nfunc StreamHandler(c *gin.Context, resp *http.Response, modelName string) *model.ErrorWithStatusCode {\n\tresponseBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError)\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError)\n\t}\n\tvar deeplResponse Response\n\terr = json.Unmarshal(responseBody, &deeplResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError)\n\t}\n\tfullTextResponse := StreamResponseDeepL2OpenAI(&deeplResponse)\n\tfullTextResponse.Model = modelName\n\tfullTextResponse.Id = helper.GetResponseID(c)\n\tjsonData, err := json.Marshal(fullTextResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError)\n\t}\n\tcommon.SetEventStreamHeaders(c)\n\tc.Stream(func(w io.Writer) bool {\n\t\tif jsonData != nil {\n\t\t\tc.Render(-1, common.CustomEvent{Data: \"data: \" + string(jsonData)})\n\t\t\tjsonData = nil\n\t\t\treturn true\n\t\t}\n\t\tc.Render(-1, common.CustomEvent{Data: \"data: [DONE]\"})\n\t\treturn false\n\t})\n\t_ = resp.Body.Close()\n\treturn nil\n}\n\nfunc Handler(c *gin.Context, resp *http.Response, modelName string) *model.ErrorWithStatusCode {\n\tresponseBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError)\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError)\n\t}\n\tvar deeplResponse Response\n\terr = json.Unmarshal(responseBody, &deeplResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError)\n\t}\n\tif deeplResponse.Message != \"\" {\n\t\treturn &model.ErrorWithStatusCode{\n\t\t\tError: model.Error{\n\t\t\t\tMessage: deeplResponse.Message,\n\t\t\t\tCode: \"deepl_error\",\n\t\t\t},\n\t\t\tStatusCode: resp.StatusCode,\n\t\t}\n\t}\n\tfullTextResponse := ResponseDeepL2OpenAI(&deeplResponse)\n\tfullTextResponse.Model = modelName\n\tfullTextResponse.Id = helper.GetResponseID(c)\n\tjsonResponse, err := json.Marshal(fullTextResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError)\n\t}\n\tc.Writer.Header().Set(\"Content-Type\", \"application/json\")\n\tc.Writer.WriteHeader(resp.StatusCode)\n\t_, err = c.Writer.Write(jsonResponse)\n\treturn nil\n}\n"
},
{
"path": "relay/adaptor/deepl/model.go",
"content": "package deepl\n\ntype Request struct {\n\tText []string `json:\"text\"`\n\tTargetLang string `json:\"target_lang\"`\n}\n\ntype Translation struct {\n\tDetectedSourceLanguage string `json:\"detected_source_language,omitempty\"`\n\tText string `json:\"text,omitempty\"`\n}\n\ntype Response struct {\n\tTranslations []Translation `json:\"translations,omitempty\"`\n\tMessage string `json:\"message,omitempty\"`\n}\n"
},
{
"path": "relay/adaptor/deepseek/constants.go",
"content": "package deepseek\n\nvar ModelList = []string{\n\t\"deepseek-chat\",\n\t\"deepseek-reasoner\",\n}\n"
},
{
"path": "relay/adaptor/doubao/constants.go",
"content": "package doubao\n\n// https://console.volcengine.com/ark/region:ark+cn-beijing/model\n\nvar ModelList = []string{\n\t\"Doubao-pro-128k\",\n\t\"Doubao-pro-32k\",\n\t\"Doubao-pro-4k\",\n\t\"Doubao-lite-128k\",\n\t\"Doubao-lite-32k\",\n\t\"Doubao-lite-4k\",\n\t\"Doubao-embedding\",\n}\n"
},
{
"path": "relay/adaptor/doubao/main.go",
"content": "package doubao\n\nimport (\n\t\"fmt\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/relaymode\"\n)\n\nfunc GetRequestURL(meta *meta.Meta) (string, error) {\n\tswitch meta.Mode {\n\tcase relaymode.ChatCompletions:\n\t\treturn fmt.Sprintf(\"%s/api/v3/chat/completions\", meta.BaseURL), nil\n\tcase relaymode.Embeddings:\n\t\treturn fmt.Sprintf(\"%s/api/v3/embeddings\", meta.BaseURL), nil\n\tdefault:\n\t}\n\treturn \"\", fmt.Errorf(\"unsupported relay mode %d for doubao\", meta.Mode)\n}\n"
},
{
"path": "relay/adaptor/gemini/adaptor.go",
"content": "package gemini\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\tchannelhelper \"github.com/songquanpeng/one-api/relay/adaptor\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"github.com/songquanpeng/one-api/relay/relaymode\"\n)\n\ntype Adaptor struct {\n}\n\nfunc (a *Adaptor) Init(meta *meta.Meta) {\n}\n\nfunc (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {\n\tdefaultVersion := config.GeminiVersion\n\tif strings.Contains(meta.ActualModelName, \"gemini-2.0\") ||\n\t\tstrings.Contains(meta.ActualModelName, \"gemini-1.5\") {\n\t\tdefaultVersion = \"v1beta\"\n\t}\n\n\tversion := helper.AssignOrDefault(meta.Config.APIVersion, defaultVersion)\n\taction := \"\"\n\tswitch meta.Mode {\n\tcase relaymode.Embeddings:\n\t\taction = \"batchEmbedContents\"\n\tdefault:\n\t\taction = \"generateContent\"\n\t}\n\n\tif meta.IsStream {\n\t\taction = \"streamGenerateContent?alt=sse\"\n\t}\n\n\treturn fmt.Sprintf(\"%s/%s/models/%s:%s\", meta.BaseURL, version, meta.ActualModelName, action), nil\n}\n\nfunc (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {\n\tchannelhelper.SetupCommonRequestHeader(c, req, meta)\n\treq.Header.Set(\"x-goog-api-key\", meta.APIKey)\n\treturn nil\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\tswitch relayMode {\n\tcase relaymode.Embeddings:\n\t\tgeminiEmbeddingRequest := ConvertEmbeddingRequest(*request)\n\t\treturn geminiEmbeddingRequest, nil\n\tdefault:\n\t\tgeminiRequest := ConvertRequest(*request)\n\t\treturn geminiRequest, nil\n\t}\n}\n\nfunc (a *Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\treturn request, nil\n}\n\nfunc (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {\n\treturn channelhelper.DoRequestHelper(a, c, meta, requestBody)\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tif meta.IsStream {\n\t\tvar responseText string\n\t\terr, responseText = StreamHandler(c, resp)\n\t\tusage = openai.ResponseText2Usage(responseText, meta.ActualModelName, meta.PromptTokens)\n\t} else {\n\t\tswitch meta.Mode {\n\t\tcase relaymode.Embeddings:\n\t\t\terr, usage = EmbeddingHandler(c, resp)\n\t\tdefault:\n\t\t\terr, usage = Handler(c, resp, meta.PromptTokens, meta.ActualModelName)\n\t\t}\n\t}\n\treturn\n}\n\nfunc (a *Adaptor) GetModelList() []string {\n\treturn ModelList\n}\n\nfunc (a *Adaptor) GetChannelName() string {\n\treturn \"google gemini\"\n}\n"
},
{
"path": "relay/adaptor/gemini/constants.go",
"content": "package gemini\n\nimport (\n\t\"github.com/songquanpeng/one-api/relay/adaptor/geminiv2\"\n)\n\n// https://ai.google.dev/models/gemini\n\nvar ModelList = geminiv2.ModelList\n\n// ModelsSupportSystemInstruction is the list of models that support system instruction.\n//\n// https://cloud.google.com/vertex-ai/generative-ai/docs/learn/prompts/system-instructions\nvar ModelsSupportSystemInstruction = []string{\n\t// \"gemini-1.0-pro-002\",\n\t// \"gemini-1.5-flash\", \"gemini-1.5-flash-001\", \"gemini-1.5-flash-002\",\n\t// \"gemini-1.5-flash-8b\",\n\t// \"gemini-1.5-pro\", \"gemini-1.5-pro-001\", \"gemini-1.5-pro-002\",\n\t// \"gemini-1.5-pro-experimental\",\n\t\"gemini-2.0-flash\", \"gemini-2.0-flash-exp\",\n\t\"gemini-2.0-flash-thinking-exp-01-21\",\n}\n\n// IsModelSupportSystemInstruction check if the model support system instruction.\n//\n// Because the main version of Go is 1.20, slice.Contains cannot be used\nfunc IsModelSupportSystemInstruction(model string) bool {\n\tfor _, m := range ModelsSupportSystemInstruction {\n\t\tif m == model {\n\t\t\treturn true\n\t\t}\n\t}\n\n\treturn false\n}\n"
},
{
"path": "relay/adaptor/gemini/main.go",
"content": "package gemini\n\nimport (\n\t\"bufio\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n\n\t\"github.com/songquanpeng/one-api/common/render\"\n\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/image\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/common/random\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/constant\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\n\t\"github.com/gin-gonic/gin\"\n)\n\n// https://ai.google.dev/docs/gemini_api_overview?hl=zh-cn\n\nconst (\n\tVisionMaxImageNum = 16\n)\n\nvar mimeTypeMap = map[string]string{\n\t\"json_object\": \"application/json\",\n\t\"text\": \"text/plain\",\n}\n\n// Setting safety to the lowest possible values since Gemini is already powerless enough\nfunc ConvertRequest(textRequest model.GeneralOpenAIRequest) *ChatRequest {\n\tgeminiRequest := ChatRequest{\n\t\tContents: make([]ChatContent, 0, len(textRequest.Messages)),\n\t\tSafetySettings: []ChatSafetySettings{\n\t\t\t{\n\t\t\t\tCategory: \"HARM_CATEGORY_HARASSMENT\",\n\t\t\t\tThreshold: config.GeminiSafetySetting,\n\t\t\t},\n\t\t\t{\n\t\t\t\tCategory: \"HARM_CATEGORY_HATE_SPEECH\",\n\t\t\t\tThreshold: config.GeminiSafetySetting,\n\t\t\t},\n\t\t\t{\n\t\t\t\tCategory: \"HARM_CATEGORY_SEXUALLY_EXPLICIT\",\n\t\t\t\tThreshold: config.GeminiSafetySetting,\n\t\t\t},\n\t\t\t{\n\t\t\t\tCategory: \"HARM_CATEGORY_DANGEROUS_CONTENT\",\n\t\t\t\tThreshold: config.GeminiSafetySetting,\n\t\t\t},\n\t\t\t{\n\t\t\t\tCategory: \"HARM_CATEGORY_CIVIC_INTEGRITY\",\n\t\t\t\tThreshold: config.GeminiSafetySetting,\n\t\t\t},\n\t\t},\n\t\tGenerationConfig: ChatGenerationConfig{\n\t\t\tTemperature: textRequest.Temperature,\n\t\t\tTopP: textRequest.TopP,\n\t\t\tMaxOutputTokens: textRequest.MaxTokens,\n\t\t},\n\t}\n\tif textRequest.ResponseFormat != nil {\n\t\tif mimeType, ok := mimeTypeMap[textRequest.ResponseFormat.Type]; ok {\n\t\t\tgeminiRequest.GenerationConfig.ResponseMimeType = mimeType\n\t\t}\n\t\tif textRequest.ResponseFormat.JsonSchema != nil {\n\t\t\tgeminiRequest.GenerationConfig.ResponseSchema = textRequest.ResponseFormat.JsonSchema.Schema\n\t\t\tgeminiRequest.GenerationConfig.ResponseMimeType = mimeTypeMap[\"json_object\"]\n\t\t}\n\t}\n\tif textRequest.Tools != nil {\n\t\tfunctions := make([]model.Function, 0, len(textRequest.Tools))\n\t\tfor _, tool := range textRequest.Tools {\n\t\t\tfunctions = append(functions, tool.Function)\n\t\t}\n\t\tgeminiRequest.Tools = []ChatTools{\n\t\t\t{\n\t\t\t\tFunctionDeclarations: functions,\n\t\t\t},\n\t\t}\n\t} else if textRequest.Functions != nil {\n\t\tgeminiRequest.Tools = []ChatTools{\n\t\t\t{\n\t\t\t\tFunctionDeclarations: textRequest.Functions,\n\t\t\t},\n\t\t}\n\t}\n\tshouldAddDummyModelMessage := false\n\tfor _, message := range textRequest.Messages {\n\t\tcontent := ChatContent{\n\t\t\tRole: message.Role,\n\t\t\tParts: []Part{\n\t\t\t\t{\n\t\t\t\t\tText: message.StringContent(),\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\t\topenaiContent := message.ParseContent()\n\t\tvar parts []Part\n\t\timageNum := 0\n\t\tfor _, part := range openaiContent {\n\t\t\tif part.Type == model.ContentTypeText {\n\t\t\t\tparts = append(parts, Part{\n\t\t\t\t\tText: part.Text,\n\t\t\t\t})\n\t\t\t} else if part.Type == model.ContentTypeImageURL {\n\t\t\t\timageNum += 1\n\t\t\t\tif imageNum > VisionMaxImageNum {\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\tmimeType, data, _ := image.GetImageFromUrl(part.ImageURL.Url)\n\t\t\t\tparts = append(parts, Part{\n\t\t\t\t\tInlineData: &InlineData{\n\t\t\t\t\t\tMimeType: mimeType,\n\t\t\t\t\t\tData: data,\n\t\t\t\t\t},\n\t\t\t\t})\n\t\t\t}\n\t\t}\n\t\tcontent.Parts = parts\n\n\t\t// there's no assistant role in gemini and API shall vomit if Role is not user or model\n\t\tif content.Role == \"assistant\" {\n\t\t\tcontent.Role = \"model\"\n\t\t}\n\t\t// Converting system prompt to prompt from user for the same reason\n\t\tif content.Role == \"system\" {\n\t\t\tshouldAddDummyModelMessage = true\n\t\t\tif IsModelSupportSystemInstruction(textRequest.Model) {\n\t\t\t\tgeminiRequest.SystemInstruction = &content\n\t\t\t\tgeminiRequest.SystemInstruction.Role = \"\"\n\t\t\t\tcontinue\n\t\t\t} else {\n\t\t\t\tcontent.Role = \"user\"\n\t\t\t}\n\t\t}\n\n\t\tgeminiRequest.Contents = append(geminiRequest.Contents, content)\n\n\t\t// If a system message is the last message, we need to add a dummy model message to make gemini happy\n\t\tif shouldAddDummyModelMessage {\n\t\t\tgeminiRequest.Contents = append(geminiRequest.Contents, ChatContent{\n\t\t\t\tRole: \"model\",\n\t\t\t\tParts: []Part{\n\t\t\t\t\t{\n\t\t\t\t\t\tText: \"Okay\",\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t})\n\t\t\tshouldAddDummyModelMessage = false\n\t\t}\n\t}\n\n\treturn &geminiRequest\n}\n\nfunc ConvertEmbeddingRequest(request model.GeneralOpenAIRequest) *BatchEmbeddingRequest {\n\tinputs := request.ParseInput()\n\trequests := make([]EmbeddingRequest, len(inputs))\n\tmodel := fmt.Sprintf(\"models/%s\", request.Model)\n\n\tfor i, input := range inputs {\n\t\trequests[i] = EmbeddingRequest{\n\t\t\tModel: model,\n\t\t\tContent: ChatContent{\n\t\t\t\tParts: []Part{\n\t\t\t\t\t{\n\t\t\t\t\t\tText: input,\n\t\t\t\t\t},\n\t\t\t\t},\n\t\t\t},\n\t\t}\n\t}\n\n\treturn &BatchEmbeddingRequest{\n\t\tRequests: requests,\n\t}\n}\n\ntype ChatResponse struct {\n\tCandidates []ChatCandidate `json:\"candidates\"`\n\tPromptFeedback ChatPromptFeedback `json:\"promptFeedback\"`\n}\n\nfunc (g *ChatResponse) GetResponseText() string {\n\tif g == nil {\n\t\treturn \"\"\n\t}\n\tif len(g.Candidates) > 0 && len(g.Candidates[0].Content.Parts) > 0 {\n\t\treturn g.Candidates[0].Content.Parts[0].Text\n\t}\n\treturn \"\"\n}\n\ntype ChatCandidate struct {\n\tContent ChatContent `json:\"content\"`\n\tFinishReason string `json:\"finishReason\"`\n\tIndex int64 `json:\"index\"`\n\tSafetyRatings []ChatSafetyRating `json:\"safetyRatings\"`\n}\n\ntype ChatSafetyRating struct {\n\tCategory string `json:\"category\"`\n\tProbability string `json:\"probability\"`\n}\n\ntype ChatPromptFeedback struct {\n\tSafetyRatings []ChatSafetyRating `json:\"safetyRatings\"`\n}\n\nfunc getToolCalls(candidate *ChatCandidate) []model.Tool {\n\tvar toolCalls []model.Tool\n\n\titem := candidate.Content.Parts[0]\n\tif item.FunctionCall == nil {\n\t\treturn toolCalls\n\t}\n\targsBytes, err := json.Marshal(item.FunctionCall.Arguments)\n\tif err != nil {\n\t\tlogger.FatalLog(\"getToolCalls failed: \" + err.Error())\n\t\treturn toolCalls\n\t}\n\ttoolCall := model.Tool{\n\t\tId: fmt.Sprintf(\"call_%s\", random.GetUUID()),\n\t\tType: \"function\",\n\t\tFunction: model.Function{\n\t\t\tArguments: string(argsBytes),\n\t\t\tName: item.FunctionCall.FunctionName,\n\t\t},\n\t}\n\ttoolCalls = append(toolCalls, toolCall)\n\treturn toolCalls\n}\n\nfunc responseGeminiChat2OpenAI(response *ChatResponse) *openai.TextResponse {\n\tfullTextResponse := openai.TextResponse{\n\t\tId: fmt.Sprintf(\"chatcmpl-%s\", random.GetUUID()),\n\t\tObject: \"chat.completion\",\n\t\tCreated: helper.GetTimestamp(),\n\t\tChoices: make([]openai.TextResponseChoice, 0, len(response.Candidates)),\n\t}\n\tfor i, candidate := range response.Candidates {\n\t\tchoice := openai.TextResponseChoice{\n\t\t\tIndex: i,\n\t\t\tMessage: model.Message{\n\t\t\t\tRole: \"assistant\",\n\t\t\t},\n\t\t\tFinishReason: constant.StopFinishReason,\n\t\t}\n\t\tif len(candidate.Content.Parts) > 0 {\n\t\t\tif candidate.Content.Parts[0].FunctionCall != nil {\n\t\t\t\tchoice.Message.ToolCalls = getToolCalls(&candidate)\n\t\t\t} else {\n\t\t\t\tvar builder strings.Builder\n\t\t\t\tfor _, part := range candidate.Content.Parts {\n\t\t\t\t\tif i > 0 {\n\t\t\t\t\t\tbuilder.WriteString(\"\\n\")\n\t\t\t\t\t}\n\t\t\t\t\tbuilder.WriteString(part.Text)\n\t\t\t\t}\n\t\t\t\tchoice.Message.Content = builder.String()\n\t\t\t}\n\t\t} else {\n\t\t\tchoice.Message.Content = \"\"\n\t\t\tchoice.FinishReason = candidate.FinishReason\n\t\t}\n\t\tfullTextResponse.Choices = append(fullTextResponse.Choices, choice)\n\t}\n\treturn &fullTextResponse\n}\n\nfunc streamResponseGeminiChat2OpenAI(geminiResponse *ChatResponse) *openai.ChatCompletionsStreamResponse {\n\tvar choice openai.ChatCompletionsStreamResponseChoice\n\tchoice.Delta.Content = geminiResponse.GetResponseText()\n\t//choice.FinishReason = &constant.StopFinishReason\n\tvar response openai.ChatCompletionsStreamResponse\n\tresponse.Id = fmt.Sprintf(\"chatcmpl-%s\", random.GetUUID())\n\tresponse.Created = helper.GetTimestamp()\n\tresponse.Object = \"chat.completion.chunk\"\n\tresponse.Model = \"gemini\"\n\tresponse.Choices = []openai.ChatCompletionsStreamResponseChoice{choice}\n\treturn &response\n}\n\nfunc embeddingResponseGemini2OpenAI(response *EmbeddingResponse) *openai.EmbeddingResponse {\n\topenAIEmbeddingResponse := openai.EmbeddingResponse{\n\t\tObject: \"list\",\n\t\tData: make([]openai.EmbeddingResponseItem, 0, len(response.Embeddings)),\n\t\tModel: \"gemini-embedding\",\n\t\tUsage: model.Usage{TotalTokens: 0},\n\t}\n\tfor _, item := range response.Embeddings {\n\t\topenAIEmbeddingResponse.Data = append(openAIEmbeddingResponse.Data, openai.EmbeddingResponseItem{\n\t\t\tObject: `embedding`,\n\t\t\tIndex: 0,\n\t\t\tEmbedding: item.Values,\n\t\t})\n\t}\n\treturn &openAIEmbeddingResponse\n}\n\nfunc StreamHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, string) {\n\tresponseText := \"\"\n\tscanner := bufio.NewScanner(resp.Body)\n\tscanner.Split(bufio.ScanLines)\n\n\tcommon.SetEventStreamHeaders(c)\n\n\tfor scanner.Scan() {\n\t\tdata := scanner.Text()\n\t\tdata = strings.TrimSpace(data)\n\t\tif !strings.HasPrefix(data, \"data: \") {\n\t\t\tcontinue\n\t\t}\n\t\tdata = strings.TrimPrefix(data, \"data: \")\n\t\tdata = strings.TrimSuffix(data, \"\\\"\")\n\n\t\tvar geminiResponse ChatResponse\n\t\terr := json.Unmarshal([]byte(data), &geminiResponse)\n\t\tif err != nil {\n\t\t\tlogger.SysError(\"error unmarshalling stream response: \" + err.Error())\n\t\t\tcontinue\n\t\t}\n\n\t\tresponse := streamResponseGeminiChat2OpenAI(&geminiResponse)\n\t\tif response == nil {\n\t\t\tcontinue\n\t\t}\n\n\t\tresponseText += response.Choices[0].Delta.StringContent()\n\n\t\terr = render.ObjectData(c, response)\n\t\tif err != nil {\n\t\t\tlogger.SysError(err.Error())\n\t\t}\n\t}\n\n\tif err := scanner.Err(); err != nil {\n\t\tlogger.SysError(\"error reading stream: \" + err.Error())\n\t}\n\n\trender.Done(c)\n\n\terr := resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), \"\"\n\t}\n\n\treturn nil, responseText\n}\n\nfunc Handler(c *gin.Context, resp *http.Response, promptTokens int, modelName string) (*model.ErrorWithStatusCode, *model.Usage) {\n\tresponseBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tvar geminiResponse ChatResponse\n\terr = json.Unmarshal(responseBody, &geminiResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tif len(geminiResponse.Candidates) == 0 {\n\t\treturn &model.ErrorWithStatusCode{\n\t\t\tError: model.Error{\n\t\t\t\tMessage: \"No candidates returned\",\n\t\t\t\tType: \"server_error\",\n\t\t\t\tParam: \"\",\n\t\t\t\tCode: 500,\n\t\t\t},\n\t\t\tStatusCode: resp.StatusCode,\n\t\t}, nil\n\t}\n\tfullTextResponse := responseGeminiChat2OpenAI(&geminiResponse)\n\tfullTextResponse.Model = modelName\n\tcompletionTokens := openai.CountTokenText(geminiResponse.GetResponseText(), modelName)\n\tusage := model.Usage{\n\t\tPromptTokens: promptTokens,\n\t\tCompletionTokens: completionTokens,\n\t\tTotalTokens: promptTokens + completionTokens,\n\t}\n\tfullTextResponse.Usage = usage\n\tjsonResponse, err := json.Marshal(fullTextResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tc.Writer.Header().Set(\"Content-Type\", \"application/json\")\n\tc.Writer.WriteHeader(resp.StatusCode)\n\t_, err = c.Writer.Write(jsonResponse)\n\treturn nil, &usage\n}\n\nfunc EmbeddingHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tvar geminiEmbeddingResponse EmbeddingResponse\n\tresponseBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = json.Unmarshal(responseBody, &geminiEmbeddingResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tif geminiEmbeddingResponse.Error != nil {\n\t\treturn &model.ErrorWithStatusCode{\n\t\t\tError: model.Error{\n\t\t\t\tMessage: geminiEmbeddingResponse.Error.Message,\n\t\t\t\tType: \"gemini_error\",\n\t\t\t\tParam: \"\",\n\t\t\t\tCode: geminiEmbeddingResponse.Error.Code,\n\t\t\t},\n\t\t\tStatusCode: resp.StatusCode,\n\t\t}, nil\n\t}\n\tfullTextResponse := embeddingResponseGemini2OpenAI(&geminiEmbeddingResponse)\n\tjsonResponse, err := json.Marshal(fullTextResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tc.Writer.Header().Set(\"Content-Type\", \"application/json\")\n\tc.Writer.WriteHeader(resp.StatusCode)\n\t_, err = c.Writer.Write(jsonResponse)\n\treturn nil, &fullTextResponse.Usage\n}\n"
},
{
"path": "relay/adaptor/gemini/model.go",
"content": "package gemini\n\ntype ChatRequest struct {\n\tContents []ChatContent `json:\"contents\"`\n\tSafetySettings []ChatSafetySettings `json:\"safety_settings,omitempty\"`\n\tGenerationConfig ChatGenerationConfig `json:\"generation_config,omitempty\"`\n\tTools []ChatTools `json:\"tools,omitempty\"`\n\tSystemInstruction *ChatContent `json:\"system_instruction,omitempty\"`\n}\n\ntype EmbeddingRequest struct {\n\tModel string `json:\"model\"`\n\tContent ChatContent `json:\"content\"`\n\tTaskType string `json:\"taskType,omitempty\"`\n\tTitle string `json:\"title,omitempty\"`\n\tOutputDimensionality int `json:\"outputDimensionality,omitempty\"`\n}\n\ntype BatchEmbeddingRequest struct {\n\tRequests []EmbeddingRequest `json:\"requests\"`\n}\n\ntype EmbeddingData struct {\n\tValues []float64 `json:\"values\"`\n}\n\ntype EmbeddingResponse struct {\n\tEmbeddings []EmbeddingData `json:\"embeddings\"`\n\tError *Error `json:\"error,omitempty\"`\n}\n\ntype Error struct {\n\tCode int `json:\"code,omitempty\"`\n\tMessage string `json:\"message,omitempty\"`\n\tStatus string `json:\"status,omitempty\"`\n}\n\ntype InlineData struct {\n\tMimeType string `json:\"mimeType\"`\n\tData string `json:\"data\"`\n}\n\ntype FunctionCall struct {\n\tFunctionName string `json:\"name\"`\n\tArguments any `json:\"args\"`\n}\n\ntype Part struct {\n\tText string `json:\"text,omitempty\"`\n\tInlineData *InlineData `json:\"inlineData,omitempty\"`\n\tFunctionCall *FunctionCall `json:\"functionCall,omitempty\"`\n}\n\ntype ChatContent struct {\n\tRole string `json:\"role,omitempty\"`\n\tParts []Part `json:\"parts\"`\n}\n\ntype ChatSafetySettings struct {\n\tCategory string `json:\"category\"`\n\tThreshold string `json:\"threshold\"`\n}\n\ntype ChatTools struct {\n\tFunctionDeclarations any `json:\"function_declarations,omitempty\"`\n}\n\ntype ChatGenerationConfig struct {\n\tResponseMimeType string `json:\"responseMimeType,omitempty\"`\n\tResponseSchema any `json:\"responseSchema,omitempty\"`\n\tTemperature *float64 `json:\"temperature,omitempty\"`\n\tTopP *float64 `json:\"topP,omitempty\"`\n\tTopK float64 `json:\"topK,omitempty\"`\n\tMaxOutputTokens int `json:\"maxOutputTokens,omitempty\"`\n\tCandidateCount int `json:\"candidateCount,omitempty\"`\n\tStopSequences []string `json:\"stopSequences,omitempty\"`\n}\n"
},
{
"path": "relay/adaptor/geminiv2/constants.go",
"content": "package geminiv2\n\n// https://ai.google.dev/models/gemini\n\nvar ModelList = []string{\n\t\"gemini-pro\", \"gemini-1.0-pro\",\n\t// \"gemma-2-2b-it\", \"gemma-2-9b-it\", \"gemma-2-27b-it\",\n\t\"gemini-1.5-flash\", \"gemini-1.5-flash-8b\",\n\t\"gemini-1.5-pro\", \"gemini-1.5-pro-experimental\",\n\t\"text-embedding-004\", \"aqa\",\n\t\"gemini-2.0-flash\", \"gemini-2.0-flash-exp\",\n\t\"gemini-2.0-flash-lite-preview-02-05\",\n\t\"gemini-2.0-flash-thinking-exp-01-21\",\n\t\"gemini-2.0-pro-exp-02-05\",\n}\n"
},
{
"path": "relay/adaptor/geminiv2/main.go",
"content": "package geminiv2\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n)\n\nfunc GetRequestURL(meta *meta.Meta) (string, error) {\n\tbaseURL := strings.TrimSuffix(meta.BaseURL, \"/\")\n\trequestPath := strings.TrimPrefix(meta.RequestURLPath, \"/v1\")\n\treturn fmt.Sprintf(\"%s%s\", baseURL, requestPath), nil\n}\n"
},
{
"path": "relay/adaptor/groq/constants.go",
"content": "package groq\n\n// https://console.groq.com/docs/models\n\nvar ModelList = []string{\n\t\"gemma2-9b-it\",\n\t\"llama-3.1-70b-versatile\",\n\t\"llama-3.1-8b-instant\",\n\t\"llama-3.2-11b-text-preview\",\n\t\"llama-3.2-11b-vision-preview\",\n\t\"llama-3.2-1b-preview\",\n\t\"llama-3.2-3b-preview\",\n\t\"llama-3.2-90b-text-preview\",\n\t\"llama-3.2-90b-vision-preview\",\n\t\"llama-guard-3-8b\",\n\t\"llama3-70b-8192\",\n\t\"llama3-8b-8192\",\n\t\"llama3-groq-70b-8192-tool-use-preview\",\n\t\"llama3-groq-8b-8192-tool-use-preview\",\n\t\"llava-v1.5-7b-4096-preview\",\n\t\"mixtral-8x7b-32768\",\n\t\"distil-whisper-large-v3-en\",\n\t\"whisper-large-v3\",\n\t\"whisper-large-v3-turbo\",\n\t\"deepseek-r1-distill-llama-70b-specdec\",\n\t\"deepseek-r1-distill-llama-70b\",\n}\n"
},
{
"path": "relay/adaptor/interface.go",
"content": "package adaptor\n\nimport (\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"io\"\n\t\"net/http\"\n)\n\ntype Adaptor interface {\n\tInit(meta *meta.Meta)\n\tGetRequestURL(meta *meta.Meta) (string, error)\n\tSetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error\n\tConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error)\n\tConvertImageRequest(request *model.ImageRequest) (any, error)\n\tDoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error)\n\tDoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode)\n\tGetModelList() []string\n\tGetChannelName() string\n}\n"
},
{
"path": "relay/adaptor/lingyiwanwu/constants.go",
"content": "package lingyiwanwu\n\n// https://platform.lingyiwanwu.com/docs\n\nvar ModelList = []string{\n\t\"yi-34b-chat-0205\",\n\t\"yi-34b-chat-200k\",\n\t\"yi-vl-plus\",\n}\n"
},
{
"path": "relay/adaptor/minimax/constants.go",
"content": "package minimax\n\n// https://www.minimaxi.com/document/guides/chat-model/V2?id=65e0736ab2845de20908e2dd\n\nvar ModelList = []string{\n\t\"abab6.5-chat\",\n\t\"abab6.5s-chat\",\n\t\"abab6-chat\",\n\t\"abab5.5-chat\",\n\t\"abab5.5s-chat\",\n\t\"MiniMax-VL-01\",\n\t\"MiniMax-Text-01\",\n}\n"
},
{
"path": "relay/adaptor/minimax/main.go",
"content": "package minimax\n\nimport (\n\t\"fmt\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/relaymode\"\n)\n\nfunc GetRequestURL(meta *meta.Meta) (string, error) {\n\tif meta.Mode == relaymode.ChatCompletions {\n\t\treturn fmt.Sprintf(\"%s/v1/text/chatcompletion_v2\", meta.BaseURL), nil\n\t}\n\treturn \"\", fmt.Errorf(\"unsupported relay mode %d for minimax\", meta.Mode)\n}\n"
},
{
"path": "relay/adaptor/mistral/constants.go",
"content": "package mistral\n\nvar ModelList = []string{\n\t\"open-mistral-7b\",\n\t\"open-mixtral-8x7b\",\n\t\"mistral-small-latest\",\n\t\"mistral-medium-latest\",\n\t\"mistral-large-latest\",\n\t\"mistral-embed\",\n}\n"
},
{
"path": "relay/adaptor/moonshot/constants.go",
"content": "package moonshot\n\nvar ModelList = []string{\n\t\"moonshot-v1-8k\",\n\t\"moonshot-v1-32k\",\n\t\"moonshot-v1-128k\",\n}\n"
},
{
"path": "relay/adaptor/novita/constants.go",
"content": "package novita\n\n// https://novita.ai/llm-api\n\nvar ModelList = []string{\n\t\"meta-llama/llama-3-8b-instruct\",\n\t\"meta-llama/llama-3-70b-instruct\",\n\t\"nousresearch/hermes-2-pro-llama-3-8b\",\n\t\"nousresearch/nous-hermes-llama2-13b\",\n\t\"mistralai/mistral-7b-instruct\",\n\t\"cognitivecomputations/dolphin-mixtral-8x22b\",\n\t\"sao10k/l3-70b-euryale-v2.1\",\n\t\"sophosympatheia/midnight-rose-70b\",\n\t\"gryphe/mythomax-l2-13b\",\n\t\"Nous-Hermes-2-Mixtral-8x7B-DPO\",\n\t\"lzlv_70b\",\n\t\"teknium/openhermes-2.5-mistral-7b\",\n\t\"microsoft/wizardlm-2-8x22b\",\n}\n"
},
{
"path": "relay/adaptor/novita/main.go",
"content": "package novita\n\nimport (\n\t\"fmt\"\n\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/relaymode\"\n)\n\nfunc GetRequestURL(meta *meta.Meta) (string, error) {\n\tif meta.Mode == relaymode.ChatCompletions {\n\t\treturn fmt.Sprintf(\"%s/chat/completions\", meta.BaseURL), nil\n\t}\n\treturn \"\", fmt.Errorf(\"unsupported relay mode %d for novita\", meta.Mode)\n}\n"
},
{
"path": "relay/adaptor/ollama/adaptor.go",
"content": "package ollama\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/relaymode\"\n\t\"io\"\n\t\"net/http\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\ntype Adaptor struct {\n}\n\nfunc (a *Adaptor) Init(meta *meta.Meta) {\n\n}\n\nfunc (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {\n\t// https://github.com/ollama/ollama/blob/main/docs/api.md\n\tfullRequestURL := fmt.Sprintf(\"%s/api/chat\", meta.BaseURL)\n\tif meta.Mode == relaymode.Embeddings {\n\t\tfullRequestURL = fmt.Sprintf(\"%s/api/embed\", meta.BaseURL)\n\t}\n\treturn fullRequestURL, nil\n}\n\nfunc (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {\n\tadaptor.SetupCommonRequestHeader(c, req, meta)\n\treq.Header.Set(\"Authorization\", \"Bearer \"+meta.APIKey)\n\treturn nil\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\tswitch relayMode {\n\tcase relaymode.Embeddings:\n\t\tollamaEmbeddingRequest := ConvertEmbeddingRequest(*request)\n\t\treturn ollamaEmbeddingRequest, nil\n\tdefault:\n\t\treturn ConvertRequest(*request), nil\n\t}\n}\n\nfunc (a *Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\treturn request, nil\n}\n\nfunc (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {\n\treturn adaptor.DoRequestHelper(a, c, meta, requestBody)\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tif meta.IsStream {\n\t\terr, usage = StreamHandler(c, resp)\n\t} else {\n\t\tswitch meta.Mode {\n\t\tcase relaymode.Embeddings:\n\t\t\terr, usage = EmbeddingHandler(c, resp)\n\t\tdefault:\n\t\t\terr, usage = Handler(c, resp)\n\t\t}\n\t}\n\treturn\n}\n\nfunc (a *Adaptor) GetModelList() []string {\n\treturn ModelList\n}\n\nfunc (a *Adaptor) GetChannelName() string {\n\treturn \"ollama\"\n}\n"
},
{
"path": "relay/adaptor/ollama/constants.go",
"content": "package ollama\n\nvar ModelList = []string{\n\t\"codellama:7b-instruct\",\n\t\"llama2:7b\",\n\t\"llama2:latest\",\n\t\"llama3:latest\",\n\t\"phi3:latest\",\n\t\"qwen:0.5b-chat\",\n\t\"qwen:7b\",\n}\n"
},
{
"path": "relay/adaptor/ollama/main.go",
"content": "package ollama\n\nimport (\n\t\"bufio\"\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"github.com/songquanpeng/one-api/common/render\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/random\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/image\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/constant\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\nfunc ConvertRequest(request model.GeneralOpenAIRequest) *ChatRequest {\n\tollamaRequest := ChatRequest{\n\t\tModel: request.Model,\n\t\tOptions: &Options{\n\t\t\tSeed: int(request.Seed),\n\t\t\tTemperature: request.Temperature,\n\t\t\tTopP: request.TopP,\n\t\t\tFrequencyPenalty: request.FrequencyPenalty,\n\t\t\tPresencePenalty: request.PresencePenalty,\n\t\t\tNumPredict: request.MaxTokens,\n\t\t\tNumCtx: request.NumCtx,\n\t\t},\n\t\tStream: request.Stream,\n\t}\n\tfor _, message := range request.Messages {\n\t\topenaiContent := message.ParseContent()\n\t\tvar imageUrls []string\n\t\tvar contentText string\n\t\tfor _, part := range openaiContent {\n\t\t\tswitch part.Type {\n\t\t\tcase model.ContentTypeText:\n\t\t\t\tcontentText = part.Text\n\t\t\tcase model.ContentTypeImageURL:\n\t\t\t\t_, data, _ := image.GetImageFromUrl(part.ImageURL.Url)\n\t\t\t\timageUrls = append(imageUrls, data)\n\t\t\t}\n\t\t}\n\t\tollamaRequest.Messages = append(ollamaRequest.Messages, Message{\n\t\t\tRole: message.Role,\n\t\t\tContent: contentText,\n\t\t\tImages: imageUrls,\n\t\t})\n\t}\n\treturn &ollamaRequest\n}\n\nfunc responseOllama2OpenAI(response *ChatResponse) *openai.TextResponse {\n\tchoice := openai.TextResponseChoice{\n\t\tIndex: 0,\n\t\tMessage: model.Message{\n\t\t\tRole: response.Message.Role,\n\t\t\tContent: response.Message.Content,\n\t\t},\n\t}\n\tif response.Done {\n\t\tchoice.FinishReason = \"stop\"\n\t}\n\tfullTextResponse := openai.TextResponse{\n\t\tId: fmt.Sprintf(\"chatcmpl-%s\", random.GetUUID()),\n\t\tModel: response.Model,\n\t\tObject: \"chat.completion\",\n\t\tCreated: helper.GetTimestamp(),\n\t\tChoices: []openai.TextResponseChoice{choice},\n\t\tUsage: model.Usage{\n\t\t\tPromptTokens: response.PromptEvalCount,\n\t\t\tCompletionTokens: response.EvalCount,\n\t\t\tTotalTokens: response.PromptEvalCount + response.EvalCount,\n\t\t},\n\t}\n\treturn &fullTextResponse\n}\n\nfunc streamResponseOllama2OpenAI(ollamaResponse *ChatResponse) *openai.ChatCompletionsStreamResponse {\n\tvar choice openai.ChatCompletionsStreamResponseChoice\n\tchoice.Delta.Role = ollamaResponse.Message.Role\n\tchoice.Delta.Content = ollamaResponse.Message.Content\n\tif ollamaResponse.Done {\n\t\tchoice.FinishReason = &constant.StopFinishReason\n\t}\n\tresponse := openai.ChatCompletionsStreamResponse{\n\t\tId: fmt.Sprintf(\"chatcmpl-%s\", random.GetUUID()),\n\t\tObject: \"chat.completion.chunk\",\n\t\tCreated: helper.GetTimestamp(),\n\t\tModel: ollamaResponse.Model,\n\t\tChoices: []openai.ChatCompletionsStreamResponseChoice{choice},\n\t}\n\treturn &response\n}\n\nfunc StreamHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tvar usage model.Usage\n\tscanner := bufio.NewScanner(resp.Body)\n\tscanner.Split(func(data []byte, atEOF bool) (advance int, token []byte, err error) {\n\t\tif atEOF && len(data) == 0 {\n\t\t\treturn 0, nil, nil\n\t\t}\n\t\tif i := strings.Index(string(data), \"}\\n\"); i >= 0 {\n\t\t\treturn i + 2, data[0 : i+1], nil\n\t\t}\n\t\tif atEOF {\n\t\t\treturn len(data), data, nil\n\t\t}\n\t\treturn 0, nil, nil\n\t})\n\n\tcommon.SetEventStreamHeaders(c)\n\n\tfor scanner.Scan() {\n\t\tdata := scanner.Text()\n\t\tif strings.HasPrefix(data, \"}\") {\n\t\t\tdata = strings.TrimPrefix(data, \"}\") + \"}\"\n\t\t}\n\n\t\tvar ollamaResponse ChatResponse\n\t\terr := json.Unmarshal([]byte(data), &ollamaResponse)\n\t\tif err != nil {\n\t\t\tlogger.SysError(\"error unmarshalling stream response: \" + err.Error())\n\t\t\tcontinue\n\t\t}\n\n\t\tif ollamaResponse.EvalCount != 0 {\n\t\t\tusage.PromptTokens = ollamaResponse.PromptEvalCount\n\t\t\tusage.CompletionTokens = ollamaResponse.EvalCount\n\t\t\tusage.TotalTokens = ollamaResponse.PromptEvalCount + ollamaResponse.EvalCount\n\t\t}\n\n\t\tresponse := streamResponseOllama2OpenAI(&ollamaResponse)\n\t\terr = render.ObjectData(c, response)\n\t\tif err != nil {\n\t\t\tlogger.SysError(err.Error())\n\t\t}\n\t}\n\n\tif err := scanner.Err(); err != nil {\n\t\tlogger.SysError(\"error reading stream: \" + err.Error())\n\t}\n\n\trender.Done(c)\n\n\terr := resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\n\treturn nil, &usage\n}\n\nfunc ConvertEmbeddingRequest(request model.GeneralOpenAIRequest) *EmbeddingRequest {\n\treturn &EmbeddingRequest{\n\t\tModel: request.Model,\n\t\tInput: request.ParseInput(),\n\t\tOptions: &Options{\n\t\t\tSeed: int(request.Seed),\n\t\t\tTemperature: request.Temperature,\n\t\t\tTopP: request.TopP,\n\t\t\tFrequencyPenalty: request.FrequencyPenalty,\n\t\t\tPresencePenalty: request.PresencePenalty,\n\t\t},\n\t}\n}\n\nfunc EmbeddingHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tvar ollamaResponse EmbeddingResponse\n\terr := json.NewDecoder(resp.Body).Decode(&ollamaResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\n\tif ollamaResponse.Error != \"\" {\n\t\treturn &model.ErrorWithStatusCode{\n\t\t\tError: model.Error{\n\t\t\t\tMessage: ollamaResponse.Error,\n\t\t\t\tType: \"ollama_error\",\n\t\t\t\tParam: \"\",\n\t\t\t\tCode: \"ollama_error\",\n\t\t\t},\n\t\t\tStatusCode: resp.StatusCode,\n\t\t}, nil\n\t}\n\n\tfullTextResponse := embeddingResponseOllama2OpenAI(&ollamaResponse)\n\tjsonResponse, err := json.Marshal(fullTextResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tc.Writer.Header().Set(\"Content-Type\", \"application/json\")\n\tc.Writer.WriteHeader(resp.StatusCode)\n\t_, err = c.Writer.Write(jsonResponse)\n\treturn nil, &fullTextResponse.Usage\n}\n\nfunc embeddingResponseOllama2OpenAI(response *EmbeddingResponse) *openai.EmbeddingResponse {\n\topenAIEmbeddingResponse := openai.EmbeddingResponse{\n\t\tObject: \"list\",\n\t\tData: make([]openai.EmbeddingResponseItem, 0, 1),\n\t\tModel: response.Model,\n\t\tUsage: model.Usage{TotalTokens: 0},\n\t}\n\n\tfor i, embedding := range response.Embeddings {\n\t\topenAIEmbeddingResponse.Data = append(openAIEmbeddingResponse.Data, openai.EmbeddingResponseItem{\n\t\t\tObject: `embedding`,\n\t\t\tIndex: i,\n\t\t\tEmbedding: embedding,\n\t\t})\n\t}\n\treturn &openAIEmbeddingResponse\n}\n\nfunc Handler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tctx := context.TODO()\n\tvar ollamaResponse ChatResponse\n\tresponseBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tlogger.Debugf(ctx, \"ollama response: %s\", string(responseBody))\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = json.Unmarshal(responseBody, &ollamaResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tif ollamaResponse.Error != \"\" {\n\t\treturn &model.ErrorWithStatusCode{\n\t\t\tError: model.Error{\n\t\t\t\tMessage: ollamaResponse.Error,\n\t\t\t\tType: \"ollama_error\",\n\t\t\t\tParam: \"\",\n\t\t\t\tCode: \"ollama_error\",\n\t\t\t},\n\t\t\tStatusCode: resp.StatusCode,\n\t\t}, nil\n\t}\n\tfullTextResponse := responseOllama2OpenAI(&ollamaResponse)\n\tjsonResponse, err := json.Marshal(fullTextResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tc.Writer.Header().Set(\"Content-Type\", \"application/json\")\n\tc.Writer.WriteHeader(resp.StatusCode)\n\t_, err = c.Writer.Write(jsonResponse)\n\treturn nil, &fullTextResponse.Usage\n}\n"
},
{
"path": "relay/adaptor/ollama/model.go",
"content": "package ollama\n\ntype Options struct {\n\tSeed int `json:\"seed,omitempty\"`\n\tTemperature *float64 `json:\"temperature,omitempty\"`\n\tTopK int `json:\"top_k,omitempty\"`\n\tTopP *float64 `json:\"top_p,omitempty\"`\n\tFrequencyPenalty *float64 `json:\"frequency_penalty,omitempty\"`\n\tPresencePenalty *float64 `json:\"presence_penalty,omitempty\"`\n\tNumPredict int `json:\"num_predict,omitempty\"`\n\tNumCtx int `json:\"num_ctx,omitempty\"`\n}\n\ntype Message struct {\n\tRole string `json:\"role,omitempty\"`\n\tContent string `json:\"content,omitempty\"`\n\tImages []string `json:\"images,omitempty\"`\n}\n\ntype ChatRequest struct {\n\tModel string `json:\"model,omitempty\"`\n\tMessages []Message `json:\"messages,omitempty\"`\n\tStream bool `json:\"stream\"`\n\tOptions *Options `json:\"options,omitempty\"`\n}\n\ntype ChatResponse struct {\n\tModel string `json:\"model,omitempty\"`\n\tCreatedAt string `json:\"created_at,omitempty\"`\n\tMessage Message `json:\"message,omitempty\"`\n\tResponse string `json:\"response,omitempty\"` // for stream response\n\tDone bool `json:\"done,omitempty\"`\n\tTotalDuration int `json:\"total_duration,omitempty\"`\n\tLoadDuration int `json:\"load_duration,omitempty\"`\n\tPromptEvalCount int `json:\"prompt_eval_count,omitempty\"`\n\tEvalCount int `json:\"eval_count,omitempty\"`\n\tEvalDuration int `json:\"eval_duration,omitempty\"`\n\tError string `json:\"error,omitempty\"`\n}\n\ntype EmbeddingRequest struct {\n\tModel string `json:\"model\"`\n\tInput []string `json:\"input\"`\n\t// Truncate bool `json:\"truncate,omitempty\"`\n\tOptions *Options `json:\"options,omitempty\"`\n\t// KeepAlive string `json:\"keep_alive,omitempty\"`\n}\n\ntype EmbeddingResponse struct {\n\tError string `json:\"error,omitempty\"`\n\tModel string `json:\"model\"`\n\tEmbeddings [][]float64 `json:\"embeddings\"`\n}\n"
},
{
"path": "relay/adaptor/openai/adaptor.go",
"content": "package openai\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n\n\t\"github.com/gin-gonic/gin\"\n\n\t\"github.com/songquanpeng/one-api/relay/adaptor\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/alibailian\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/baiduv2\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/doubao\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/geminiv2\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/minimax\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/novita\"\n\t\"github.com/songquanpeng/one-api/relay/channeltype\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"github.com/songquanpeng/one-api/relay/relaymode\"\n)\n\ntype Adaptor struct {\n\tChannelType int\n}\n\nfunc (a *Adaptor) Init(meta *meta.Meta) {\n\ta.ChannelType = meta.ChannelType\n}\n\nfunc (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {\n\tswitch meta.ChannelType {\n\tcase channeltype.Azure:\n\t\tif meta.Mode == relaymode.ImagesGenerations {\n\t\t\t// https://learn.microsoft.com/en-us/azure/ai-services/openai/dall-e-quickstart?tabs=dalle3%2Ccommand-line&pivots=rest-api\n\t\t\t// https://{resource_name}.openai.azure.com/openai/deployments/dall-e-3/images/generations?api-version=2024-03-01-preview\n\t\t\tfullRequestURL := fmt.Sprintf(\"%s/openai/deployments/%s/images/generations?api-version=%s\", meta.BaseURL, meta.ActualModelName, meta.Config.APIVersion)\n\t\t\treturn fullRequestURL, nil\n\t\t}\n\n\t\t// https://learn.microsoft.com/en-us/azure/cognitive-services/openai/chatgpt-quickstart?pivots=rest-api&tabs=command-line#rest-api\n\t\trequestURL := strings.Split(meta.RequestURLPath, \"?\")[0]\n\t\trequestURL = fmt.Sprintf(\"%s?api-version=%s\", requestURL, meta.Config.APIVersion)\n\t\ttask := strings.TrimPrefix(requestURL, \"/v1/\")\n\t\tmodel_ := meta.ActualModelName\n\t\tmodel_ = strings.Replace(model_, \".\", \"\", -1)\n\t\t//https://github.com/songquanpeng/one-api/issues/1191\n\t\t// {your endpoint}/openai/deployments/{your azure_model}/chat/completions?api-version={api_version}\n\t\trequestURL = fmt.Sprintf(\"/openai/deployments/%s/%s\", model_, task)\n\t\treturn GetFullRequestURL(meta.BaseURL, requestURL, meta.ChannelType), nil\n\tcase channeltype.Minimax:\n\t\treturn minimax.GetRequestURL(meta)\n\tcase channeltype.Doubao:\n\t\treturn doubao.GetRequestURL(meta)\n\tcase channeltype.Novita:\n\t\treturn novita.GetRequestURL(meta)\n\tcase channeltype.BaiduV2:\n\t\treturn baiduv2.GetRequestURL(meta)\n\tcase channeltype.AliBailian:\n\t\treturn alibailian.GetRequestURL(meta)\n\tcase channeltype.GeminiOpenAICompatible:\n\t\treturn geminiv2.GetRequestURL(meta)\n\tdefault:\n\t\treturn GetFullRequestURL(meta.BaseURL, meta.RequestURLPath, meta.ChannelType), nil\n\t}\n}\n\nfunc (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {\n\tadaptor.SetupCommonRequestHeader(c, req, meta)\n\tif meta.ChannelType == channeltype.Azure {\n\t\treq.Header.Set(\"api-key\", meta.APIKey)\n\t\treturn nil\n\t}\n\treq.Header.Set(\"Authorization\", \"Bearer \"+meta.APIKey)\n\tif meta.ChannelType == channeltype.OpenRouter {\n\t\treq.Header.Set(\"HTTP-Referer\", \"https://github.com/songquanpeng/one-api\")\n\t\treq.Header.Set(\"X-Title\", \"One API\")\n\t}\n\treturn nil\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\tif request.Stream {\n\t\t// always return usage in stream mode\n\t\tif request.StreamOptions == nil {\n\t\t\trequest.StreamOptions = &model.StreamOptions{}\n\t\t}\n\t\trequest.StreamOptions.IncludeUsage = true\n\t}\n\treturn request, nil\n}\n\nfunc (a *Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\treturn request, nil\n}\n\nfunc (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {\n\treturn adaptor.DoRequestHelper(a, c, meta, requestBody)\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tif meta.IsStream {\n\t\tvar responseText string\n\t\terr, responseText, usage = StreamHandler(c, resp, meta.Mode)\n\t\tif usage == nil || usage.TotalTokens == 0 {\n\t\t\tusage = ResponseText2Usage(responseText, meta.ActualModelName, meta.PromptTokens)\n\t\t}\n\t\tif usage.TotalTokens != 0 && usage.PromptTokens == 0 { // some channels don't return prompt tokens & completion tokens\n\t\t\tusage.PromptTokens = meta.PromptTokens\n\t\t\tusage.CompletionTokens = usage.TotalTokens - meta.PromptTokens\n\t\t}\n\t} else {\n\t\tswitch meta.Mode {\n\t\tcase relaymode.ImagesGenerations:\n\t\t\terr, _ = ImageHandler(c, resp)\n\t\tdefault:\n\t\t\terr, usage = Handler(c, resp, meta.PromptTokens, meta.ActualModelName)\n\t\t}\n\t}\n\treturn\n}\n\nfunc (a *Adaptor) GetModelList() []string {\n\t_, modelList := GetCompatibleChannelMeta(a.ChannelType)\n\treturn modelList\n}\n\nfunc (a *Adaptor) GetChannelName() string {\n\tchannelName, _ := GetCompatibleChannelMeta(a.ChannelType)\n\treturn channelName\n}\n"
},
{
"path": "relay/adaptor/openai/compatible.go",
"content": "package openai\n\nimport (\n\t\"github.com/songquanpeng/one-api/relay/adaptor/ai360\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/alibailian\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/baichuan\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/baiduv2\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/deepseek\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/doubao\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/geminiv2\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/groq\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/lingyiwanwu\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/minimax\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/mistral\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/moonshot\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/novita\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openrouter\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/siliconflow\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/stepfun\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/togetherai\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/xai\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/xunfeiv2\"\n\t\"github.com/songquanpeng/one-api/relay/channeltype\"\n)\n\nvar CompatibleChannels = []int{\n\tchanneltype.Azure,\n\tchanneltype.AI360,\n\tchanneltype.Moonshot,\n\tchanneltype.Baichuan,\n\tchanneltype.Minimax,\n\tchanneltype.Doubao,\n\tchanneltype.Mistral,\n\tchanneltype.Groq,\n\tchanneltype.LingYiWanWu,\n\tchanneltype.StepFun,\n\tchanneltype.DeepSeek,\n\tchanneltype.TogetherAI,\n\tchanneltype.Novita,\n\tchanneltype.SiliconFlow,\n\tchanneltype.XAI,\n\tchanneltype.BaiduV2,\n\tchanneltype.XunfeiV2,\n}\n\nfunc GetCompatibleChannelMeta(channelType int) (string, []string) {\n\tswitch channelType {\n\tcase channeltype.Azure:\n\t\treturn \"azure\", ModelList\n\tcase channeltype.AI360:\n\t\treturn \"360\", ai360.ModelList\n\tcase channeltype.Moonshot:\n\t\treturn \"moonshot\", moonshot.ModelList\n\tcase channeltype.Baichuan:\n\t\treturn \"baichuan\", baichuan.ModelList\n\tcase channeltype.Minimax:\n\t\treturn \"minimax\", minimax.ModelList\n\tcase channeltype.Mistral:\n\t\treturn \"mistralai\", mistral.ModelList\n\tcase channeltype.Groq:\n\t\treturn \"groq\", groq.ModelList\n\tcase channeltype.LingYiWanWu:\n\t\treturn \"lingyiwanwu\", lingyiwanwu.ModelList\n\tcase channeltype.StepFun:\n\t\treturn \"stepfun\", stepfun.ModelList\n\tcase channeltype.DeepSeek:\n\t\treturn \"deepseek\", deepseek.ModelList\n\tcase channeltype.TogetherAI:\n\t\treturn \"together.ai\", togetherai.ModelList\n\tcase channeltype.Doubao:\n\t\treturn \"doubao\", doubao.ModelList\n\tcase channeltype.Novita:\n\t\treturn \"novita\", novita.ModelList\n\tcase channeltype.SiliconFlow:\n\t\treturn \"siliconflow\", siliconflow.ModelList\n\tcase channeltype.XAI:\n\t\treturn \"xai\", xai.ModelList\n\tcase channeltype.BaiduV2:\n\t\treturn \"baiduv2\", baiduv2.ModelList\n\tcase channeltype.XunfeiV2:\n\t\treturn \"xunfeiv2\", xunfeiv2.ModelList\n\tcase channeltype.OpenRouter:\n\t\treturn \"openrouter\", openrouter.ModelList\n\tcase channeltype.AliBailian:\n\t\treturn \"alibailian\", alibailian.ModelList\n\tcase channeltype.GeminiOpenAICompatible:\n\t\treturn \"geminiv2\", geminiv2.ModelList\n\tdefault:\n\t\treturn \"openai\", ModelList\n\t}\n}\n"
},
{
"path": "relay/adaptor/openai/constants.go",
"content": "package openai\n\nvar ModelList = []string{\n\t\"gpt-3.5-turbo\", \"gpt-3.5-turbo-0301\", \"gpt-3.5-turbo-0613\", \"gpt-3.5-turbo-1106\", \"gpt-3.5-turbo-0125\",\n\t\"gpt-3.5-turbo-16k\", \"gpt-3.5-turbo-16k-0613\",\n\t\"gpt-3.5-turbo-instruct\",\n\t\"gpt-4\", \"gpt-4-0314\", \"gpt-4-0613\", \"gpt-4-1106-preview\", \"gpt-4-0125-preview\",\n\t\"gpt-4-32k\", \"gpt-4-32k-0314\", \"gpt-4-32k-0613\",\n\t\"gpt-4-turbo-preview\", \"gpt-4-turbo\", \"gpt-4-turbo-2024-04-09\",\n\t\"gpt-4o\", \"gpt-4o-2024-05-13\",\n\t\"gpt-4o-2024-08-06\",\n\t\"gpt-4o-2024-11-20\",\n\t\"chatgpt-4o-latest\",\n\t\"gpt-4o-mini\", \"gpt-4o-mini-2024-07-18\",\n\t\"gpt-4-vision-preview\",\n\t\"text-embedding-ada-002\", \"text-embedding-3-small\", \"text-embedding-3-large\",\n\t\"text-curie-001\", \"text-babbage-001\", \"text-ada-001\", \"text-davinci-002\", \"text-davinci-003\",\n\t\"text-moderation-latest\", \"text-moderation-stable\",\n\t\"text-davinci-edit-001\",\n\t\"davinci-002\", \"babbage-002\",\n\t\"dall-e-2\", \"dall-e-3\",\n\t\"whisper-1\",\n\t\"tts-1\", \"tts-1-1106\", \"tts-1-hd\", \"tts-1-hd-1106\",\n\t\"o1\", \"o1-2024-12-17\",\n\t\"o1-preview\", \"o1-preview-2024-09-12\",\n\t\"o1-mini\", \"o1-mini-2024-09-12\",\n}\n"
},
{
"path": "relay/adaptor/openai/helper.go",
"content": "package openai\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n\n\t\"github.com/songquanpeng/one-api/relay/channeltype\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\nfunc ResponseText2Usage(responseText string, modelName string, promptTokens int) *model.Usage {\n\tusage := &model.Usage{}\n\tusage.PromptTokens = promptTokens\n\tusage.CompletionTokens = CountTokenText(responseText, modelName)\n\tusage.TotalTokens = usage.PromptTokens + usage.CompletionTokens\n\treturn usage\n}\n\nfunc GetFullRequestURL(baseURL string, requestURL string, channelType int) string {\n\tif channelType == channeltype.OpenAICompatible {\n\t\treturn fmt.Sprintf(\"%s%s\", strings.TrimSuffix(baseURL, \"/\"), strings.TrimPrefix(requestURL, \"/v1\"))\n\t}\n\tfullRequestURL := fmt.Sprintf(\"%s%s\", baseURL, requestURL)\n\n\tif strings.HasPrefix(baseURL, \"https://gateway.ai.cloudflare.com\") {\n\t\tswitch channelType {\n\t\tcase channeltype.OpenAI:\n\t\t\tfullRequestURL = fmt.Sprintf(\"%s%s\", baseURL, strings.TrimPrefix(requestURL, \"/v1\"))\n\t\tcase channeltype.Azure:\n\t\t\tfullRequestURL = fmt.Sprintf(\"%s%s\", baseURL, strings.TrimPrefix(requestURL, \"/openai/deployments\"))\n\t\t}\n\t}\n\treturn fullRequestURL\n}\n"
},
{
"path": "relay/adaptor/openai/image.go",
"content": "package openai\n\nimport (\n\t\"bytes\"\n\t\"encoding/json\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"io\"\n\t\"net/http\"\n)\n\nfunc ImageHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tvar imageResponse ImageResponse\n\tresponseBody, err := io.ReadAll(resp.Body)\n\n\tif err != nil {\n\t\treturn ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = json.Unmarshal(responseBody, &imageResponse)\n\tif err != nil {\n\t\treturn ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\n\tresp.Body = io.NopCloser(bytes.NewBuffer(responseBody))\n\n\tfor k, v := range resp.Header {\n\t\tc.Writer.Header().Set(k, v[0])\n\t}\n\tc.Writer.WriteHeader(resp.StatusCode)\n\n\t_, err = io.Copy(c.Writer, resp.Body)\n\tif err != nil {\n\t\treturn ErrorWrapper(err, \"copy_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\treturn nil, nil\n}\n"
},
{
"path": "relay/adaptor/openai/main.go",
"content": "package openai\n\nimport (\n\t\"bufio\"\n\t\"bytes\"\n\t\"encoding/json\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n\n\t\"github.com/songquanpeng/one-api/common/render\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/conv\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"github.com/songquanpeng/one-api/relay/relaymode\"\n)\n\nconst (\n\tdataPrefix = \"data: \"\n\tdone = \"[DONE]\"\n\tdataPrefixLength = len(dataPrefix)\n)\n\nfunc StreamHandler(c *gin.Context, resp *http.Response, relayMode int) (*model.ErrorWithStatusCode, string, *model.Usage) {\n\tresponseText := \"\"\n\tscanner := bufio.NewScanner(resp.Body)\n\tscanner.Split(bufio.ScanLines)\n\tvar usage *model.Usage\n\n\tcommon.SetEventStreamHeaders(c)\n\n\tdoneRendered := false\n\tfor scanner.Scan() {\n\t\tdata := scanner.Text()\n\t\tif len(data) < dataPrefixLength { // ignore blank line or wrong format\n\t\t\tcontinue\n\t\t}\n\t\tif data[:dataPrefixLength] != dataPrefix && data[:dataPrefixLength] != done {\n\t\t\tcontinue\n\t\t}\n\t\tif strings.HasPrefix(data[dataPrefixLength:], done) {\n\t\t\trender.StringData(c, data)\n\t\t\tdoneRendered = true\n\t\t\tcontinue\n\t\t}\n\t\tswitch relayMode {\n\t\tcase relaymode.ChatCompletions:\n\t\t\tvar streamResponse ChatCompletionsStreamResponse\n\t\t\terr := json.Unmarshal([]byte(data[dataPrefixLength:]), &streamResponse)\n\t\t\tif err != nil {\n\t\t\t\tlogger.SysError(\"error unmarshalling stream response: \" + err.Error())\n\t\t\t\trender.StringData(c, data) // if error happened, pass the data to client\n\t\t\t\tcontinue // just ignore the error\n\t\t\t}\n\t\t\tif len(streamResponse.Choices) == 0 && streamResponse.Usage == nil {\n\t\t\t\t// but for empty choice and no usage, we should not pass it to client, this is for azure\n\t\t\t\tcontinue // just ignore empty choice\n\t\t\t}\n\t\t\trender.StringData(c, data)\n\t\t\tfor _, choice := range streamResponse.Choices {\n\t\t\t\tresponseText += conv.AsString(choice.Delta.Content)\n\t\t\t}\n\t\t\tif streamResponse.Usage != nil {\n\t\t\t\tusage = streamResponse.Usage\n\t\t\t}\n\t\tcase relaymode.Completions:\n\t\t\trender.StringData(c, data)\n\t\t\tvar streamResponse CompletionsStreamResponse\n\t\t\terr := json.Unmarshal([]byte(data[dataPrefixLength:]), &streamResponse)\n\t\t\tif err != nil {\n\t\t\t\tlogger.SysError(\"error unmarshalling stream response: \" + err.Error())\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tfor _, choice := range streamResponse.Choices {\n\t\t\t\tresponseText += choice.Text\n\t\t\t}\n\t\t}\n\t}\n\n\tif err := scanner.Err(); err != nil {\n\t\tlogger.SysError(\"error reading stream: \" + err.Error())\n\t}\n\n\tif !doneRendered {\n\t\trender.Done(c)\n\t}\n\n\terr := resp.Body.Close()\n\tif err != nil {\n\t\treturn ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), \"\", nil\n\t}\n\n\treturn nil, responseText, usage\n}\n\nfunc Handler(c *gin.Context, resp *http.Response, promptTokens int, modelName string) (*model.ErrorWithStatusCode, *model.Usage) {\n\tvar textResponse SlimTextResponse\n\tresponseBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = json.Unmarshal(responseBody, &textResponse)\n\tif err != nil {\n\t\treturn ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tif textResponse.Error.Type != \"\" {\n\t\treturn &model.ErrorWithStatusCode{\n\t\t\tError: textResponse.Error,\n\t\t\tStatusCode: resp.StatusCode,\n\t\t}, nil\n\t}\n\t// Reset response body\n\tresp.Body = io.NopCloser(bytes.NewBuffer(responseBody))\n\n\t// We shouldn't set the header before we parse the response body, because the parse part may fail.\n\t// And then we will have to send an error response, but in this case, the header has already been set.\n\t// So the HTTPClient will be confused by the response.\n\t// For example, Postman will report error, and we cannot check the response at all.\n\tfor k, v := range resp.Header {\n\t\tc.Writer.Header().Set(k, v[0])\n\t}\n\tc.Writer.WriteHeader(resp.StatusCode)\n\t_, err = io.Copy(c.Writer, resp.Body)\n\tif err != nil {\n\t\treturn ErrorWrapper(err, \"copy_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\n\tif textResponse.Usage.TotalTokens == 0 || (textResponse.Usage.PromptTokens == 0 && textResponse.Usage.CompletionTokens == 0) {\n\t\tcompletionTokens := 0\n\t\tfor _, choice := range textResponse.Choices {\n\t\t\tcompletionTokens += CountTokenText(choice.Message.StringContent(), modelName)\n\t\t}\n\t\ttextResponse.Usage = model.Usage{\n\t\t\tPromptTokens: promptTokens,\n\t\t\tCompletionTokens: completionTokens,\n\t\t\tTotalTokens: promptTokens + completionTokens,\n\t\t}\n\t}\n\treturn nil, &textResponse.Usage\n}\n"
},
{
"path": "relay/adaptor/openai/model.go",
"content": "package openai\n\nimport \"github.com/songquanpeng/one-api/relay/model\"\n\ntype TextContent struct {\n\tType string `json:\"type,omitempty\"`\n\tText string `json:\"text,omitempty\"`\n}\n\ntype ImageContent struct {\n\tType string `json:\"type,omitempty\"`\n\tImageURL *model.ImageURL `json:\"image_url,omitempty\"`\n}\n\ntype ChatRequest struct {\n\tModel string `json:\"model\"`\n\tMessages []model.Message `json:\"messages\"`\n\tMaxTokens int `json:\"max_tokens\"`\n}\n\ntype TextRequest struct {\n\tModel string `json:\"model\"`\n\tMessages []model.Message `json:\"messages\"`\n\tPrompt string `json:\"prompt\"`\n\tMaxTokens int `json:\"max_tokens\"`\n\t//Stream bool `json:\"stream\"`\n}\n\n// ImageRequest docs: https://platform.openai.com/docs/api-reference/images/create\ntype ImageRequest struct {\n\tModel string `json:\"model\"`\n\tPrompt string `json:\"prompt\" binding:\"required\"`\n\tN int `json:\"n,omitempty\"`\n\tSize string `json:\"size,omitempty\"`\n\tQuality string `json:\"quality,omitempty\"`\n\tResponseFormat string `json:\"response_format,omitempty\"`\n\tStyle string `json:\"style,omitempty\"`\n\tUser string `json:\"user,omitempty\"`\n}\n\ntype WhisperJSONResponse struct {\n\tText string `json:\"text,omitempty\"`\n}\n\ntype WhisperVerboseJSONResponse struct {\n\tTask string `json:\"task,omitempty\"`\n\tLanguage string `json:\"language,omitempty\"`\n\tDuration float64 `json:\"duration,omitempty\"`\n\tText string `json:\"text,omitempty\"`\n\tSegments []Segment `json:\"segments,omitempty\"`\n}\n\ntype Segment struct {\n\tId int `json:\"id\"`\n\tSeek int `json:\"seek\"`\n\tStart float64 `json:\"start\"`\n\tEnd float64 `json:\"end\"`\n\tText string `json:\"text\"`\n\tTokens []int `json:\"tokens\"`\n\tTemperature float64 `json:\"temperature\"`\n\tAvgLogprob float64 `json:\"avg_logprob\"`\n\tCompressionRatio float64 `json:\"compression_ratio\"`\n\tNoSpeechProb float64 `json:\"no_speech_prob\"`\n}\n\ntype TextToSpeechRequest struct {\n\tModel string `json:\"model\" binding:\"required\"`\n\tInput string `json:\"input\" binding:\"required\"`\n\tVoice string `json:\"voice\" binding:\"required\"`\n\tSpeed float64 `json:\"speed\"`\n\tResponseFormat string `json:\"response_format\"`\n}\n\ntype UsageOrResponseText struct {\n\t*model.Usage\n\tResponseText string\n}\n\ntype SlimTextResponse struct {\n\tChoices []TextResponseChoice `json:\"choices\"`\n\tmodel.Usage `json:\"usage\"`\n\tError model.Error `json:\"error\"`\n}\n\ntype TextResponseChoice struct {\n\tIndex int `json:\"index\"`\n\tmodel.Message `json:\"message\"`\n\tFinishReason string `json:\"finish_reason\"`\n}\n\ntype TextResponse struct {\n\tId string `json:\"id\"`\n\tModel string `json:\"model,omitempty\"`\n\tObject string `json:\"object\"`\n\tCreated int64 `json:\"created\"`\n\tChoices []TextResponseChoice `json:\"choices\"`\n\tmodel.Usage `json:\"usage\"`\n}\n\ntype EmbeddingResponseItem struct {\n\tObject string `json:\"object\"`\n\tIndex int `json:\"index\"`\n\tEmbedding []float64 `json:\"embedding\"`\n}\n\ntype EmbeddingResponse struct {\n\tObject string `json:\"object\"`\n\tData []EmbeddingResponseItem `json:\"data\"`\n\tModel string `json:\"model\"`\n\tmodel.Usage `json:\"usage\"`\n}\n\ntype ImageData struct {\n\tUrl string `json:\"url,omitempty\"`\n\tB64Json string `json:\"b64_json,omitempty\"`\n\tRevisedPrompt string `json:\"revised_prompt,omitempty\"`\n}\n\ntype ImageResponse struct {\n\tCreated int64 `json:\"created\"`\n\tData []ImageData `json:\"data\"`\n\t//model.Usage `json:\"usage\"`\n}\n\ntype ChatCompletionsStreamResponseChoice struct {\n\tIndex int `json:\"index\"`\n\tDelta model.Message `json:\"delta\"`\n\tFinishReason *string `json:\"finish_reason,omitempty\"`\n}\n\ntype ChatCompletionsStreamResponse struct {\n\tId string `json:\"id\"`\n\tObject string `json:\"object\"`\n\tCreated int64 `json:\"created\"`\n\tModel string `json:\"model\"`\n\tChoices []ChatCompletionsStreamResponseChoice `json:\"choices\"`\n\tUsage *model.Usage `json:\"usage,omitempty\"`\n}\n\ntype CompletionsStreamResponse struct {\n\tChoices []struct {\n\t\tText string `json:\"text\"`\n\t\tFinishReason string `json:\"finish_reason\"`\n\t} `json:\"choices\"`\n}\n"
},
{
"path": "relay/adaptor/openai/token.go",
"content": "package openai\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"math\"\n\t\"strings\"\n\n\t\"github.com/pkoukk/tiktoken-go\"\n\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/image\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\tbillingratio \"github.com/songquanpeng/one-api/relay/billing/ratio\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\n// tokenEncoderMap won't grow after initialization\nvar tokenEncoderMap = map[string]*tiktoken.Tiktoken{}\nvar defaultTokenEncoder *tiktoken.Tiktoken\n\nfunc InitTokenEncoders() {\n\tlogger.SysLog(\"initializing token encoders\")\n\tgpt35TokenEncoder, err := tiktoken.EncodingForModel(\"gpt-3.5-turbo\")\n\tif err != nil {\n\t\tlogger.FatalLog(fmt.Sprintf(\"failed to get gpt-3.5-turbo token encoder: %s, \"+\n\t\t\t\"if you are using in offline environment, please set TIKTOKEN_CACHE_DIR to use exsited files, check this link for more information: https://stackoverflow.com/questions/76106366/how-to-use-tiktoken-in-offline-mode-computer \", err.Error()))\n\t}\n\tdefaultTokenEncoder = gpt35TokenEncoder\n\tgpt4oTokenEncoder, err := tiktoken.EncodingForModel(\"gpt-4o\")\n\tif err != nil {\n\t\tlogger.FatalLog(fmt.Sprintf(\"failed to get gpt-4o token encoder: %s\", err.Error()))\n\t}\n\tgpt4TokenEncoder, err := tiktoken.EncodingForModel(\"gpt-4\")\n\tif err != nil {\n\t\tlogger.FatalLog(fmt.Sprintf(\"failed to get gpt-4 token encoder: %s\", err.Error()))\n\t}\n\tfor model := range billingratio.ModelRatio {\n\t\tif strings.HasPrefix(model, \"gpt-3.5\") {\n\t\t\ttokenEncoderMap[model] = gpt35TokenEncoder\n\t\t} else if strings.HasPrefix(model, \"gpt-4o\") {\n\t\t\ttokenEncoderMap[model] = gpt4oTokenEncoder\n\t\t} else if strings.HasPrefix(model, \"gpt-4\") {\n\t\t\ttokenEncoderMap[model] = gpt4TokenEncoder\n\t\t} else {\n\t\t\ttokenEncoderMap[model] = nil\n\t\t}\n\t}\n\tlogger.SysLog(\"token encoders initialized\")\n}\n\nfunc getTokenEncoder(model string) *tiktoken.Tiktoken {\n\ttokenEncoder, ok := tokenEncoderMap[model]\n\tif ok && tokenEncoder != nil {\n\t\treturn tokenEncoder\n\t}\n\tif ok {\n\t\ttokenEncoder, err := tiktoken.EncodingForModel(model)\n\t\tif err != nil {\n\t\t\tlogger.SysError(fmt.Sprintf(\"failed to get token encoder for model %s: %s, using encoder for gpt-3.5-turbo\", model, err.Error()))\n\t\t\ttokenEncoder = defaultTokenEncoder\n\t\t}\n\t\ttokenEncoderMap[model] = tokenEncoder\n\t\treturn tokenEncoder\n\t}\n\treturn defaultTokenEncoder\n}\n\nfunc getTokenNum(tokenEncoder *tiktoken.Tiktoken, text string) int {\n\tif config.ApproximateTokenEnabled {\n\t\treturn int(float64(len(text)) * 0.38)\n\t}\n\treturn len(tokenEncoder.Encode(text, nil, nil))\n}\n\nfunc CountTokenMessages(messages []model.Message, model string) int {\n\ttokenEncoder := getTokenEncoder(model)\n\t// Reference:\n\t// https://github.com/openai/openai-cookbook/blob/main/examples/How_to_count_tokens_with_tiktoken.ipynb\n\t// https://github.com/pkoukk/tiktoken-go/issues/6\n\t//\n\t// Every message follows <|start|>{role/name}\\n{content}<|end|>\\n\n\tvar tokensPerMessage int\n\tvar tokensPerName int\n\tif model == \"gpt-3.5-turbo-0301\" {\n\t\ttokensPerMessage = 4\n\t\ttokensPerName = -1 // If there's a name, the role is omitted\n\t} else {\n\t\ttokensPerMessage = 3\n\t\ttokensPerName = 1\n\t}\n\ttokenNum := 0\n\tfor _, message := range messages {\n\t\ttokenNum += tokensPerMessage\n\t\tswitch v := message.Content.(type) {\n\t\tcase string:\n\t\t\ttokenNum += getTokenNum(tokenEncoder, v)\n\t\tcase []any:\n\t\t\tfor _, it := range v {\n\t\t\t\tm := it.(map[string]any)\n\t\t\t\tswitch m[\"type\"] {\n\t\t\t\tcase \"text\":\n\t\t\t\t\tif textValue, ok := m[\"text\"]; ok {\n\t\t\t\t\t\tif textString, ok := textValue.(string); ok {\n\t\t\t\t\t\t\ttokenNum += getTokenNum(tokenEncoder, textString)\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\tcase \"image_url\":\n\t\t\t\t\timageUrl, ok := m[\"image_url\"].(map[string]any)\n\t\t\t\t\tif ok {\n\t\t\t\t\t\turl := imageUrl[\"url\"].(string)\n\t\t\t\t\t\tdetail := \"\"\n\t\t\t\t\t\tif imageUrl[\"detail\"] != nil {\n\t\t\t\t\t\t\tdetail = imageUrl[\"detail\"].(string)\n\t\t\t\t\t\t}\n\t\t\t\t\t\timageTokens, err := countImageTokens(url, detail, model)\n\t\t\t\t\t\tif err != nil {\n\t\t\t\t\t\t\tlogger.SysError(\"error counting image tokens: \" + err.Error())\n\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\ttokenNum += imageTokens\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t\ttokenNum += getTokenNum(tokenEncoder, message.Role)\n\t\tif message.Name != nil {\n\t\t\ttokenNum += tokensPerName\n\t\t\ttokenNum += getTokenNum(tokenEncoder, *message.Name)\n\t\t}\n\t}\n\ttokenNum += 3 // Every reply is primed with <|start|>assistant<|message|>\n\treturn tokenNum\n}\n\nconst (\n\tlowDetailCost = 85\n\thighDetailCostPerTile = 170\n\tadditionalCost = 85\n\t// gpt-4o-mini cost higher than other model\n\tgpt4oMiniLowDetailCost = 2833\n\tgpt4oMiniHighDetailCost = 5667\n\tgpt4oMiniAdditionalCost = 2833\n)\n\n// https://platform.openai.com/docs/guides/vision/calculating-costs\n// https://github.com/openai/openai-cookbook/blob/05e3f9be4c7a2ae7ecf029a7c32065b024730ebe/examples/How_to_count_tokens_with_tiktoken.ipynb\nfunc countImageTokens(url string, detail string, model string) (_ int, err error) {\n\tvar fetchSize = true\n\tvar width, height int\n\t// Reference: https://platform.openai.com/docs/guides/vision/low-or-high-fidelity-image-understanding\n\t// detail == \"auto\" is undocumented on how it works, it just said the model will use the auto setting which will look at the image input size and decide if it should use the low or high setting.\n\t// According to the official guide, \"low\" disable the high-res model,\n\t// and only receive low-res 512px x 512px version of the image, indicating\n\t// that image is treated as low-res when size is smaller than 512px x 512px,\n\t// then we can assume that image size larger than 512px x 512px is treated\n\t// as high-res. Then we have the following logic:\n\t// if detail == \"\" || detail == \"auto\" {\n\t// \twidth, height, err = image.GetImageSize(url)\n\t// \tif err != nil {\n\t// \t\treturn 0, err\n\t// \t}\n\t// \tfetchSize = false\n\t// \t// not sure if this is correct\n\t// \tif width > 512 || height > 512 {\n\t// \t\tdetail = \"high\"\n\t// \t} else {\n\t// \t\tdetail = \"low\"\n\t// \t}\n\t// }\n\n\t// However, in my test, it seems to be always the same as \"high\".\n\t// The following image, which is 125x50, is still treated as high-res, taken\n\t// 255 tokens in the response of non-stream chat completion api.\n\t// https://upload.wikimedia.org/wikipedia/commons/1/10/18_Infantry_Division_Messina.jpg\n\tif detail == \"\" || detail == \"auto\" {\n\t\t// assume by test, not sure if this is correct\n\t\tdetail = \"high\"\n\t}\n\tswitch detail {\n\tcase \"low\":\n\t\tif strings.HasPrefix(model, \"gpt-4o-mini\") {\n\t\t\treturn gpt4oMiniLowDetailCost, nil\n\t\t}\n\t\treturn lowDetailCost, nil\n\tcase \"high\":\n\t\tif fetchSize {\n\t\t\twidth, height, err = image.GetImageSize(url)\n\t\t\tif err != nil {\n\t\t\t\treturn 0, err\n\t\t\t}\n\t\t}\n\t\tif width > 2048 || height > 2048 { // max(width, height) > 2048\n\t\t\tratio := float64(2048) / math.Max(float64(width), float64(height))\n\t\t\twidth = int(float64(width) * ratio)\n\t\t\theight = int(float64(height) * ratio)\n\t\t}\n\t\tif width > 768 && height > 768 { // min(width, height) > 768\n\t\t\tratio := float64(768) / math.Min(float64(width), float64(height))\n\t\t\twidth = int(float64(width) * ratio)\n\t\t\theight = int(float64(height) * ratio)\n\t\t}\n\t\tnumSquares := int(math.Ceil(float64(width)/512) * math.Ceil(float64(height)/512))\n\t\tif strings.HasPrefix(model, \"gpt-4o-mini\") {\n\t\t\treturn numSquares*gpt4oMiniHighDetailCost + gpt4oMiniAdditionalCost, nil\n\t\t}\n\t\tresult := numSquares*highDetailCostPerTile + additionalCost\n\t\treturn result, nil\n\tdefault:\n\t\treturn 0, errors.New(\"invalid detail option\")\n\t}\n}\n\nfunc CountTokenInput(input any, model string) int {\n\tswitch v := input.(type) {\n\tcase string:\n\t\treturn CountTokenText(v, model)\n\tcase []string:\n\t\ttext := \"\"\n\t\tfor _, s := range v {\n\t\t\ttext += s\n\t\t}\n\t\treturn CountTokenText(text, model)\n\t}\n\treturn 0\n}\n\nfunc CountTokenText(text string, model string) int {\n\ttokenEncoder := getTokenEncoder(model)\n\treturn getTokenNum(tokenEncoder, text)\n}\n\nfunc CountToken(text string) int {\n\treturn CountTokenInput(text, \"gpt-3.5-turbo\")\n}\n"
},
{
"path": "relay/adaptor/openai/util.go",
"content": "package openai\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\nfunc ErrorWrapper(err error, code string, statusCode int) *model.ErrorWithStatusCode {\n\tlogger.Error(context.TODO(), fmt.Sprintf(\"[%s]%+v\", code, err))\n\n\tError := model.Error{\n\t\tMessage: err.Error(),\n\t\tType: \"one_api_error\",\n\t\tCode: code,\n\t}\n\treturn &model.ErrorWithStatusCode{\n\t\tError: Error,\n\t\tStatusCode: statusCode,\n\t}\n}\n"
},
{
"path": "relay/adaptor/openrouter/constants.go",
"content": "package openrouter\n\nvar ModelList = []string{\n\t\"01-ai/yi-large\",\n\t\"aetherwiing/mn-starcannon-12b\",\n\t\"ai21/jamba-1-5-large\",\n\t\"ai21/jamba-1-5-mini\",\n\t\"ai21/jamba-instruct\",\n\t\"aion-labs/aion-1.0\",\n\t\"aion-labs/aion-1.0-mini\",\n\t\"aion-labs/aion-rp-llama-3.1-8b\",\n\t\"allenai/llama-3.1-tulu-3-405b\",\n\t\"alpindale/goliath-120b\",\n\t\"alpindale/magnum-72b\",\n\t\"amazon/nova-lite-v1\",\n\t\"amazon/nova-micro-v1\",\n\t\"amazon/nova-pro-v1\",\n\t\"anthracite-org/magnum-v2-72b\",\n\t\"anthracite-org/magnum-v4-72b\",\n\t\"anthropic/claude-2\",\n\t\"anthropic/claude-2.0\",\n\t\"anthropic/claude-2.0:beta\",\n\t\"anthropic/claude-2.1\",\n\t\"anthropic/claude-2.1:beta\",\n\t\"anthropic/claude-2:beta\",\n\t\"anthropic/claude-3-haiku\",\n\t\"anthropic/claude-3-haiku:beta\",\n\t\"anthropic/claude-3-opus\",\n\t\"anthropic/claude-3-opus:beta\",\n\t\"anthropic/claude-3-sonnet\",\n\t\"anthropic/claude-3-sonnet:beta\",\n\t\"anthropic/claude-3.5-haiku\",\n\t\"anthropic/claude-3.5-haiku-20241022\",\n\t\"anthropic/claude-3.5-haiku-20241022:beta\",\n\t\"anthropic/claude-3.5-haiku:beta\",\n\t\"anthropic/claude-3.5-sonnet\",\n\t\"anthropic/claude-3.5-sonnet-20240620\",\n\t\"anthropic/claude-3.5-sonnet-20240620:beta\",\n\t\"anthropic/claude-3.5-sonnet:beta\",\n\t\"cognitivecomputations/dolphin-mixtral-8x22b\",\n\t\"cognitivecomputations/dolphin-mixtral-8x7b\",\n\t\"cohere/command\",\n\t\"cohere/command-r\",\n\t\"cohere/command-r-03-2024\",\n\t\"cohere/command-r-08-2024\",\n\t\"cohere/command-r-plus\",\n\t\"cohere/command-r-plus-04-2024\",\n\t\"cohere/command-r-plus-08-2024\",\n\t\"cohere/command-r7b-12-2024\",\n\t\"databricks/dbrx-instruct\",\n\t\"deepseek/deepseek-chat\",\n\t\"deepseek/deepseek-chat-v2.5\",\n\t\"deepseek/deepseek-chat:free\",\n\t\"deepseek/deepseek-r1\",\n\t\"deepseek/deepseek-r1-distill-llama-70b\",\n\t\"deepseek/deepseek-r1-distill-llama-70b:free\",\n\t\"deepseek/deepseek-r1-distill-llama-8b\",\n\t\"deepseek/deepseek-r1-distill-qwen-1.5b\",\n\t\"deepseek/deepseek-r1-distill-qwen-14b\",\n\t\"deepseek/deepseek-r1-distill-qwen-32b\",\n\t\"deepseek/deepseek-r1:free\",\n\t\"eva-unit-01/eva-llama-3.33-70b\",\n\t\"eva-unit-01/eva-qwen-2.5-32b\",\n\t\"eva-unit-01/eva-qwen-2.5-72b\",\n\t\"google/gemini-2.0-flash-001\",\n\t\"google/gemini-2.0-flash-exp:free\",\n\t\"google/gemini-2.0-flash-lite-preview-02-05:free\",\n\t\"google/gemini-2.0-flash-thinking-exp-1219:free\",\n\t\"google/gemini-2.0-flash-thinking-exp:free\",\n\t\"google/gemini-2.0-pro-exp-02-05:free\",\n\t\"google/gemini-exp-1206:free\",\n\t\"google/gemini-flash-1.5\",\n\t\"google/gemini-flash-1.5-8b\",\n\t\"google/gemini-flash-1.5-8b-exp\",\n\t\"google/gemini-pro\",\n\t\"google/gemini-pro-1.5\",\n\t\"google/gemini-pro-vision\",\n\t\"google/gemma-2-27b-it\",\n\t\"google/gemma-2-9b-it\",\n\t\"google/gemma-2-9b-it:free\",\n\t\"google/gemma-7b-it\",\n\t\"google/learnlm-1.5-pro-experimental:free\",\n\t\"google/palm-2-chat-bison\",\n\t\"google/palm-2-chat-bison-32k\",\n\t\"google/palm-2-codechat-bison\",\n\t\"google/palm-2-codechat-bison-32k\",\n\t\"gryphe/mythomax-l2-13b\",\n\t\"gryphe/mythomax-l2-13b:free\",\n\t\"huggingfaceh4/zephyr-7b-beta:free\",\n\t\"infermatic/mn-inferor-12b\",\n\t\"inflection/inflection-3-pi\",\n\t\"inflection/inflection-3-productivity\",\n\t\"jondurbin/airoboros-l2-70b\",\n\t\"liquid/lfm-3b\",\n\t\"liquid/lfm-40b\",\n\t\"liquid/lfm-7b\",\n\t\"mancer/weaver\",\n\t\"meta-llama/llama-2-13b-chat\",\n\t\"meta-llama/llama-2-70b-chat\",\n\t\"meta-llama/llama-3-70b-instruct\",\n\t\"meta-llama/llama-3-8b-instruct\",\n\t\"meta-llama/llama-3-8b-instruct:free\",\n\t\"meta-llama/llama-3.1-405b\",\n\t\"meta-llama/llama-3.1-405b-instruct\",\n\t\"meta-llama/llama-3.1-70b-instruct\",\n\t\"meta-llama/llama-3.1-8b-instruct\",\n\t\"meta-llama/llama-3.2-11b-vision-instruct\",\n\t\"meta-llama/llama-3.2-11b-vision-instruct:free\",\n\t\"meta-llama/llama-3.2-1b-instruct\",\n\t\"meta-llama/llama-3.2-3b-instruct\",\n\t\"meta-llama/llama-3.2-90b-vision-instruct\",\n\t\"meta-llama/llama-3.3-70b-instruct\",\n\t\"meta-llama/llama-3.3-70b-instruct:free\",\n\t\"meta-llama/llama-guard-2-8b\",\n\t\"microsoft/phi-3-medium-128k-instruct\",\n\t\"microsoft/phi-3-medium-128k-instruct:free\",\n\t\"microsoft/phi-3-mini-128k-instruct\",\n\t\"microsoft/phi-3-mini-128k-instruct:free\",\n\t\"microsoft/phi-3.5-mini-128k-instruct\",\n\t\"microsoft/phi-4\",\n\t\"microsoft/wizardlm-2-7b\",\n\t\"microsoft/wizardlm-2-8x22b\",\n\t\"minimax/minimax-01\",\n\t\"mistralai/codestral-2501\",\n\t\"mistralai/codestral-mamba\",\n\t\"mistralai/ministral-3b\",\n\t\"mistralai/ministral-8b\",\n\t\"mistralai/mistral-7b-instruct\",\n\t\"mistralai/mistral-7b-instruct-v0.1\",\n\t\"mistralai/mistral-7b-instruct-v0.3\",\n\t\"mistralai/mistral-7b-instruct:free\",\n\t\"mistralai/mistral-large\",\n\t\"mistralai/mistral-large-2407\",\n\t\"mistralai/mistral-large-2411\",\n\t\"mistralai/mistral-medium\",\n\t\"mistralai/mistral-nemo\",\n\t\"mistralai/mistral-nemo:free\",\n\t\"mistralai/mistral-small\",\n\t\"mistralai/mistral-small-24b-instruct-2501\",\n\t\"mistralai/mistral-small-24b-instruct-2501:free\",\n\t\"mistralai/mistral-tiny\",\n\t\"mistralai/mixtral-8x22b-instruct\",\n\t\"mistralai/mixtral-8x7b\",\n\t\"mistralai/mixtral-8x7b-instruct\",\n\t\"mistralai/pixtral-12b\",\n\t\"mistralai/pixtral-large-2411\",\n\t\"neversleep/llama-3-lumimaid-70b\",\n\t\"neversleep/llama-3-lumimaid-8b\",\n\t\"neversleep/llama-3-lumimaid-8b:extended\",\n\t\"neversleep/llama-3.1-lumimaid-70b\",\n\t\"neversleep/llama-3.1-lumimaid-8b\",\n\t\"neversleep/noromaid-20b\",\n\t\"nothingiisreal/mn-celeste-12b\",\n\t\"nousresearch/hermes-2-pro-llama-3-8b\",\n\t\"nousresearch/hermes-3-llama-3.1-405b\",\n\t\"nousresearch/hermes-3-llama-3.1-70b\",\n\t\"nousresearch/nous-hermes-2-mixtral-8x7b-dpo\",\n\t\"nousresearch/nous-hermes-llama2-13b\",\n\t\"nvidia/llama-3.1-nemotron-70b-instruct\",\n\t\"nvidia/llama-3.1-nemotron-70b-instruct:free\",\n\t\"openai/chatgpt-4o-latest\",\n\t\"openai/gpt-3.5-turbo\",\n\t\"openai/gpt-3.5-turbo-0125\",\n\t\"openai/gpt-3.5-turbo-0613\",\n\t\"openai/gpt-3.5-turbo-1106\",\n\t\"openai/gpt-3.5-turbo-16k\",\n\t\"openai/gpt-3.5-turbo-instruct\",\n\t\"openai/gpt-4\",\n\t\"openai/gpt-4-0314\",\n\t\"openai/gpt-4-1106-preview\",\n\t\"openai/gpt-4-32k\",\n\t\"openai/gpt-4-32k-0314\",\n\t\"openai/gpt-4-turbo\",\n\t\"openai/gpt-4-turbo-preview\",\n\t\"openai/gpt-4o\",\n\t\"openai/gpt-4o-2024-05-13\",\n\t\"openai/gpt-4o-2024-08-06\",\n\t\"openai/gpt-4o-2024-11-20\",\n\t\"openai/gpt-4o-mini\",\n\t\"openai/gpt-4o-mini-2024-07-18\",\n\t\"openai/gpt-4o:extended\",\n\t\"openai/o1\",\n\t\"openai/o1-mini\",\n\t\"openai/o1-mini-2024-09-12\",\n\t\"openai/o1-preview\",\n\t\"openai/o1-preview-2024-09-12\",\n\t\"openai/o3-mini\",\n\t\"openai/o3-mini-high\",\n\t\"openchat/openchat-7b\",\n\t\"openchat/openchat-7b:free\",\n\t\"openrouter/auto\",\n\t\"perplexity/llama-3.1-sonar-huge-128k-online\",\n\t\"perplexity/llama-3.1-sonar-large-128k-chat\",\n\t\"perplexity/llama-3.1-sonar-large-128k-online\",\n\t\"perplexity/llama-3.1-sonar-small-128k-chat\",\n\t\"perplexity/llama-3.1-sonar-small-128k-online\",\n\t\"perplexity/sonar\",\n\t\"perplexity/sonar-reasoning\",\n\t\"pygmalionai/mythalion-13b\",\n\t\"qwen/qvq-72b-preview\",\n\t\"qwen/qwen-2-72b-instruct\",\n\t\"qwen/qwen-2-7b-instruct\",\n\t\"qwen/qwen-2-7b-instruct:free\",\n\t\"qwen/qwen-2-vl-72b-instruct\",\n\t\"qwen/qwen-2-vl-7b-instruct\",\n\t\"qwen/qwen-2.5-72b-instruct\",\n\t\"qwen/qwen-2.5-7b-instruct\",\n\t\"qwen/qwen-2.5-coder-32b-instruct\",\n\t\"qwen/qwen-max\",\n\t\"qwen/qwen-plus\",\n\t\"qwen/qwen-turbo\",\n\t\"qwen/qwen-vl-plus:free\",\n\t\"qwen/qwen2.5-vl-72b-instruct:free\",\n\t\"qwen/qwq-32b-preview\",\n\t\"raifle/sorcererlm-8x22b\",\n\t\"sao10k/fimbulvetr-11b-v2\",\n\t\"sao10k/l3-euryale-70b\",\n\t\"sao10k/l3-lunaris-8b\",\n\t\"sao10k/l3.1-70b-hanami-x1\",\n\t\"sao10k/l3.1-euryale-70b\",\n\t\"sao10k/l3.3-euryale-70b\",\n\t\"sophosympatheia/midnight-rose-70b\",\n\t\"sophosympatheia/rogue-rose-103b-v0.2:free\",\n\t\"teknium/openhermes-2.5-mistral-7b\",\n\t\"thedrummer/rocinante-12b\",\n\t\"thedrummer/unslopnemo-12b\",\n\t\"undi95/remm-slerp-l2-13b\",\n\t\"undi95/toppy-m-7b\",\n\t\"undi95/toppy-m-7b:free\",\n\t\"x-ai/grok-2-1212\",\n\t\"x-ai/grok-2-vision-1212\",\n\t\"x-ai/grok-beta\",\n\t\"x-ai/grok-vision-beta\",\n\t\"xwin-lm/xwin-lm-70b\",\n}\n"
},
{
"path": "relay/adaptor/palm/adaptor.go",
"content": "package palm\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"io\"\n\t\"net/http\"\n)\n\ntype Adaptor struct {\n}\n\nfunc (a *Adaptor) Init(meta *meta.Meta) {\n\n}\n\nfunc (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {\n\treturn fmt.Sprintf(\"%s/v1beta2/models/chat-bison-001:generateMessage\", meta.BaseURL), nil\n}\n\nfunc (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {\n\tadaptor.SetupCommonRequestHeader(c, req, meta)\n\treq.Header.Set(\"x-goog-api-key\", meta.APIKey)\n\treturn nil\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\treturn ConvertRequest(*request), nil\n}\n\nfunc (a *Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\treturn request, nil\n}\n\nfunc (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {\n\treturn adaptor.DoRequestHelper(a, c, meta, requestBody)\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tif meta.IsStream {\n\t\tvar responseText string\n\t\terr, responseText = StreamHandler(c, resp)\n\t\tusage = openai.ResponseText2Usage(responseText, meta.ActualModelName, meta.PromptTokens)\n\t} else {\n\t\terr, usage = Handler(c, resp, meta.PromptTokens, meta.ActualModelName)\n\t}\n\treturn\n}\n\nfunc (a *Adaptor) GetModelList() []string {\n\treturn ModelList\n}\n\nfunc (a *Adaptor) GetChannelName() string {\n\treturn \"google palm\"\n}\n"
},
{
"path": "relay/adaptor/palm/constants.go",
"content": "package palm\n\nvar ModelList = []string{\n\t\"PaLM-2\",\n}\n"
},
{
"path": "relay/adaptor/palm/model.go",
"content": "package palm\n\nimport (\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\ntype ChatMessage struct {\n\tAuthor string `json:\"author\"`\n\tContent string `json:\"content\"`\n}\n\ntype Filter struct {\n\tReason string `json:\"reason\"`\n\tMessage string `json:\"message\"`\n}\n\ntype Prompt struct {\n\tMessages []ChatMessage `json:\"messages\"`\n}\n\ntype ChatRequest struct {\n\tPrompt Prompt `json:\"prompt\"`\n\tTemperature *float64 `json:\"temperature,omitempty\"`\n\tCandidateCount int `json:\"candidateCount,omitempty\"`\n\tTopP *float64 `json:\"topP,omitempty\"`\n\tTopK int `json:\"topK,omitempty\"`\n}\n\ntype Error struct {\n\tCode int `json:\"code\"`\n\tMessage string `json:\"message\"`\n\tStatus string `json:\"status\"`\n}\n\ntype ChatResponse struct {\n\tCandidates []ChatMessage `json:\"candidates\"`\n\tMessages []model.Message `json:\"messages\"`\n\tFilters []Filter `json:\"filters\"`\n\tError Error `json:\"error\"`\n}\n"
},
{
"path": "relay/adaptor/palm/palm.go",
"content": "package palm\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"github.com/songquanpeng/one-api/common/render\"\n\t\"io\"\n\t\"net/http\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/common/random\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/constant\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\n// https://developers.generativeai.google/api/rest/generativelanguage/models/generateMessage#request-body\n// https://developers.generativeai.google/api/rest/generativelanguage/models/generateMessage#response-body\n\nfunc ConvertRequest(textRequest model.GeneralOpenAIRequest) *ChatRequest {\n\tpalmRequest := ChatRequest{\n\t\tPrompt: Prompt{\n\t\t\tMessages: make([]ChatMessage, 0, len(textRequest.Messages)),\n\t\t},\n\t\tTemperature: textRequest.Temperature,\n\t\tCandidateCount: textRequest.N,\n\t\tTopP: textRequest.TopP,\n\t\tTopK: textRequest.MaxTokens,\n\t}\n\tfor _, message := range textRequest.Messages {\n\t\tpalmMessage := ChatMessage{\n\t\t\tContent: message.StringContent(),\n\t\t}\n\t\tif message.Role == \"user\" {\n\t\t\tpalmMessage.Author = \"0\"\n\t\t} else {\n\t\t\tpalmMessage.Author = \"1\"\n\t\t}\n\t\tpalmRequest.Prompt.Messages = append(palmRequest.Prompt.Messages, palmMessage)\n\t}\n\treturn &palmRequest\n}\n\nfunc responsePaLM2OpenAI(response *ChatResponse) *openai.TextResponse {\n\tfullTextResponse := openai.TextResponse{\n\t\tChoices: make([]openai.TextResponseChoice, 0, len(response.Candidates)),\n\t}\n\tfor i, candidate := range response.Candidates {\n\t\tchoice := openai.TextResponseChoice{\n\t\t\tIndex: i,\n\t\t\tMessage: model.Message{\n\t\t\t\tRole: \"assistant\",\n\t\t\t\tContent: candidate.Content,\n\t\t\t},\n\t\t\tFinishReason: \"stop\",\n\t\t}\n\t\tfullTextResponse.Choices = append(fullTextResponse.Choices, choice)\n\t}\n\treturn &fullTextResponse\n}\n\nfunc streamResponsePaLM2OpenAI(palmResponse *ChatResponse) *openai.ChatCompletionsStreamResponse {\n\tvar choice openai.ChatCompletionsStreamResponseChoice\n\tif len(palmResponse.Candidates) > 0 {\n\t\tchoice.Delta.Content = palmResponse.Candidates[0].Content\n\t}\n\tchoice.FinishReason = &constant.StopFinishReason\n\tvar response openai.ChatCompletionsStreamResponse\n\tresponse.Object = \"chat.completion.chunk\"\n\tresponse.Model = \"palm2\"\n\tresponse.Choices = []openai.ChatCompletionsStreamResponseChoice{choice}\n\treturn &response\n}\n\nfunc StreamHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, string) {\n\tresponseText := \"\"\n\tresponseId := fmt.Sprintf(\"chatcmpl-%s\", random.GetUUID())\n\tcreatedTime := helper.GetTimestamp()\n\n\tcommon.SetEventStreamHeaders(c)\n\n\tresponseBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\tlogger.SysError(\"error reading stream response: \" + err.Error())\n\t\terr := resp.Body.Close()\n\t\tif err != nil {\n\t\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), \"\"\n\t\t}\n\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError), \"\"\n\t}\n\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), \"\"\n\t}\n\n\tvar palmResponse ChatResponse\n\terr = json.Unmarshal(responseBody, &palmResponse)\n\tif err != nil {\n\t\tlogger.SysError(\"error unmarshalling stream response: \" + err.Error())\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), \"\"\n\t}\n\n\tfullTextResponse := streamResponsePaLM2OpenAI(&palmResponse)\n\tfullTextResponse.Id = responseId\n\tfullTextResponse.Created = createdTime\n\tif len(palmResponse.Candidates) > 0 {\n\t\tresponseText = palmResponse.Candidates[0].Content\n\t}\n\n\tjsonResponse, err := json.Marshal(fullTextResponse)\n\tif err != nil {\n\t\tlogger.SysError(\"error marshalling stream response: \" + err.Error())\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError), \"\"\n\t}\n\n\terr = render.ObjectData(c, string(jsonResponse))\n\tif err != nil {\n\t\tlogger.SysError(err.Error())\n\t}\n\n\trender.Done(c)\n\n\treturn nil, responseText\n}\n\nfunc Handler(c *gin.Context, resp *http.Response, promptTokens int, modelName string) (*model.ErrorWithStatusCode, *model.Usage) {\n\tresponseBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tvar palmResponse ChatResponse\n\terr = json.Unmarshal(responseBody, &palmResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tif palmResponse.Error.Code != 0 || len(palmResponse.Candidates) == 0 {\n\t\treturn &model.ErrorWithStatusCode{\n\t\t\tError: model.Error{\n\t\t\t\tMessage: palmResponse.Error.Message,\n\t\t\t\tType: palmResponse.Error.Status,\n\t\t\t\tParam: \"\",\n\t\t\t\tCode: palmResponse.Error.Code,\n\t\t\t},\n\t\t\tStatusCode: resp.StatusCode,\n\t\t}, nil\n\t}\n\tfullTextResponse := responsePaLM2OpenAI(&palmResponse)\n\tfullTextResponse.Model = modelName\n\tcompletionTokens := openai.CountTokenText(palmResponse.Candidates[0].Content, modelName)\n\tusage := model.Usage{\n\t\tPromptTokens: promptTokens,\n\t\tCompletionTokens: completionTokens,\n\t\tTotalTokens: promptTokens + completionTokens,\n\t}\n\tfullTextResponse.Usage = usage\n\tjsonResponse, err := json.Marshal(fullTextResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tc.Writer.Header().Set(\"Content-Type\", \"application/json\")\n\tc.Writer.WriteHeader(resp.StatusCode)\n\t_, err = c.Writer.Write(jsonResponse)\n\treturn nil, &usage\n}\n"
},
{
"path": "relay/adaptor/proxy/adaptor.go",
"content": "package proxy\n\nimport (\n\t\"fmt\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/pkg/errors\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor\"\n\tchannelhelper \"github.com/songquanpeng/one-api/relay/adaptor\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\trelaymodel \"github.com/songquanpeng/one-api/relay/model\"\n)\n\nvar _ adaptor.Adaptor = new(Adaptor)\n\nconst channelName = \"proxy\"\n\ntype Adaptor struct{}\n\nfunc (a *Adaptor) Init(meta *meta.Meta) {\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\treturn nil, errors.New(\"notimplement\")\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tfor k, v := range resp.Header {\n\t\tfor _, vv := range v {\n\t\t\tc.Writer.Header().Set(k, vv)\n\t\t}\n\t}\n\n\tc.Writer.WriteHeader(resp.StatusCode)\n\tif _, gerr := io.Copy(c.Writer, resp.Body); gerr != nil {\n\t\treturn nil, &relaymodel.ErrorWithStatusCode{\n\t\t\tStatusCode: http.StatusInternalServerError,\n\t\t\tError: relaymodel.Error{\n\t\t\t\tMessage: gerr.Error(),\n\t\t\t},\n\t\t}\n\t}\n\n\treturn nil, nil\n}\n\nfunc (a *Adaptor) GetModelList() (models []string) {\n\treturn nil\n}\n\nfunc (a *Adaptor) GetChannelName() string {\n\treturn channelName\n}\n\n// GetRequestURL remove static prefix, and return the real request url to the upstream service\nfunc (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {\n\tprefix := fmt.Sprintf(\"/v1/oneapi/proxy/%d\", meta.ChannelId)\n\treturn meta.BaseURL + strings.TrimPrefix(meta.RequestURLPath, prefix), nil\n\n}\n\nfunc (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {\n\tfor k, v := range c.Request.Header {\n\t\treq.Header.Set(k, v[0])\n\t}\n\n\t// remove unnecessary headers\n\treq.Header.Del(\"Host\")\n\treq.Header.Del(\"Content-Length\")\n\treq.Header.Del(\"Accept-Encoding\")\n\treq.Header.Del(\"Connection\")\n\n\t// set authorization header\n\treq.Header.Set(\"Authorization\", meta.APIKey)\n\n\treturn nil\n}\n\nfunc (a *Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {\n\treturn nil, errors.Errorf(\"not implement\")\n}\n\nfunc (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {\n\treturn channelhelper.DoRequestHelper(a, c, meta, requestBody)\n}\n"
},
{
"path": "relay/adaptor/replicate/adaptor.go",
"content": "package replicate\n\nimport (\n\t\"fmt\"\n\t\"io\"\n\t\"net/http\"\n\t\"slices\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/pkg/errors\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"github.com/songquanpeng/one-api/relay/relaymode\"\n)\n\ntype Adaptor struct {\n\tmeta *meta.Meta\n}\n\n// ConvertImageRequest implements adaptor.Adaptor.\nfunc (*Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {\n\treturn DrawImageRequest{\n\t\tInput: ImageInput{\n\t\t\tSteps: 25,\n\t\t\tPrompt: request.Prompt,\n\t\t\tGuidance: 3,\n\t\t\tSeed: int(time.Now().UnixNano()),\n\t\t\tSafetyTolerance: 5,\n\t\t\tNImages: 1, // replicate will always return 1 image\n\t\t\tWidth: 1440,\n\t\t\tHeight: 1440,\n\t\t\tAspectRatio: \"1:1\",\n\t\t},\n\t}, nil\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif !request.Stream {\n\t\t// TODO: support non-stream mode\n\t\treturn nil, errors.Errorf(\"replicate models only support stream mode now, please set stream=true\")\n\t}\n\n\t// Build the prompt from OpenAI messages\n\tvar promptBuilder strings.Builder\n\tfor _, message := range request.Messages {\n\t\tswitch msgCnt := message.Content.(type) {\n\t\tcase string:\n\t\t\tpromptBuilder.WriteString(message.Role)\n\t\t\tpromptBuilder.WriteString(\": \")\n\t\t\tpromptBuilder.WriteString(msgCnt)\n\t\t\tpromptBuilder.WriteString(\"\\n\")\n\t\tdefault:\n\t\t}\n\t}\n\n\treplicateRequest := ReplicateChatRequest{\n\t\tInput: ChatInput{\n\t\t\tPrompt: promptBuilder.String(),\n\t\t\tMaxTokens: request.MaxTokens,\n\t\t\tTemperature: 1.0,\n\t\t\tTopP: 1.0,\n\t\t\tPresencePenalty: 0.0,\n\t\t\tFrequencyPenalty: 0.0,\n\t\t},\n\t}\n\n\t// Map optional fields\n\tif request.Temperature != nil {\n\t\treplicateRequest.Input.Temperature = *request.Temperature\n\t}\n\tif request.TopP != nil {\n\t\treplicateRequest.Input.TopP = *request.TopP\n\t}\n\tif request.PresencePenalty != nil {\n\t\treplicateRequest.Input.PresencePenalty = *request.PresencePenalty\n\t}\n\tif request.FrequencyPenalty != nil {\n\t\treplicateRequest.Input.FrequencyPenalty = *request.FrequencyPenalty\n\t}\n\tif request.MaxTokens > 0 {\n\t\treplicateRequest.Input.MaxTokens = request.MaxTokens\n\t} else if request.MaxTokens == 0 {\n\t\treplicateRequest.Input.MaxTokens = 500\n\t}\n\n\treturn replicateRequest, nil\n}\n\nfunc (a *Adaptor) Init(meta *meta.Meta) {\n\ta.meta = meta\n}\n\nfunc (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {\n\tif !slices.Contains(ModelList, meta.OriginModelName) {\n\t\treturn \"\", errors.Errorf(\"model %s not supported\", meta.OriginModelName)\n\t}\n\n\treturn fmt.Sprintf(\"https://api.replicate.com/v1/models/%s/predictions\", meta.OriginModelName), nil\n}\n\nfunc (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {\n\tadaptor.SetupCommonRequestHeader(c, req, meta)\n\treq.Header.Set(\"Authorization\", \"Bearer \"+meta.APIKey)\n\treturn nil\n}\n\nfunc (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {\n\tlogger.Info(c, \"send request to replicate\")\n\treturn adaptor.DoRequestHelper(a, c, meta, requestBody)\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tswitch meta.Mode {\n\tcase relaymode.ImagesGenerations:\n\t\terr, usage = ImageHandler(c, resp)\n\tcase relaymode.ChatCompletions:\n\t\terr, usage = ChatHandler(c, resp)\n\tdefault:\n\t\terr = openai.ErrorWrapper(errors.New(\"not implemented\"), \"not_implemented\", http.StatusInternalServerError)\n\t}\n\n\treturn\n}\n\nfunc (a *Adaptor) GetModelList() []string {\n\treturn ModelList\n}\n\nfunc (a *Adaptor) GetChannelName() string {\n\treturn \"replicate\"\n}\n"
},
{
"path": "relay/adaptor/replicate/chat.go",
"content": "package replicate\n\nimport (\n\t\"bufio\"\n\t\"encoding/json\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/pkg/errors\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/render\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\nfunc ChatHandler(c *gin.Context, resp *http.Response) (\n\tsrvErr *model.ErrorWithStatusCode, usage *model.Usage) {\n\tif resp.StatusCode != http.StatusCreated {\n\t\tpayload, _ := io.ReadAll(resp.Body)\n\t\treturn openai.ErrorWrapper(\n\t\t\t\terrors.Errorf(\"bad_status_code [%d]%s\", resp.StatusCode, string(payload)),\n\t\t\t\t\"bad_status_code\", http.StatusInternalServerError),\n\t\t\tnil\n\t}\n\n\trespBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\n\trespData := new(ChatResponse)\n\tif err = json.Unmarshal(respBody, respData); err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\n\tfor {\n\t\terr = func() error {\n\t\t\t// get task\n\t\t\ttaskReq, err := http.NewRequestWithContext(c.Request.Context(),\n\t\t\t\thttp.MethodGet, respData.URLs.Get, nil)\n\t\t\tif err != nil {\n\t\t\t\treturn errors.Wrap(err, \"new request\")\n\t\t\t}\n\n\t\t\ttaskReq.Header.Set(\"Authorization\", \"Bearer \"+meta.GetByContext(c).APIKey)\n\t\t\ttaskResp, err := http.DefaultClient.Do(taskReq)\n\t\t\tif err != nil {\n\t\t\t\treturn errors.Wrap(err, \"get task\")\n\t\t\t}\n\t\t\tdefer taskResp.Body.Close()\n\n\t\t\tif taskResp.StatusCode != http.StatusOK {\n\t\t\t\tpayload, _ := io.ReadAll(taskResp.Body)\n\t\t\t\treturn errors.Errorf(\"bad status code [%d]%s\",\n\t\t\t\t\ttaskResp.StatusCode, string(payload))\n\t\t\t}\n\n\t\t\ttaskBody, err := io.ReadAll(taskResp.Body)\n\t\t\tif err != nil {\n\t\t\t\treturn errors.Wrap(err, \"read task response\")\n\t\t\t}\n\n\t\t\ttaskData := new(ChatResponse)\n\t\t\tif err = json.Unmarshal(taskBody, taskData); err != nil {\n\t\t\t\treturn errors.Wrap(err, \"decode task response\")\n\t\t\t}\n\n\t\t\tswitch taskData.Status {\n\t\t\tcase \"succeeded\":\n\t\t\tcase \"failed\", \"canceled\":\n\t\t\t\treturn errors.Errorf(\"task failed, [%s]%s\", taskData.Status, taskData.Error)\n\t\t\tdefault:\n\t\t\t\ttime.Sleep(time.Second * 3)\n\t\t\t\treturn errNextLoop\n\t\t\t}\n\n\t\t\tif taskData.URLs.Stream == \"\" {\n\t\t\t\treturn errors.New(\"stream url is empty\")\n\t\t\t}\n\n\t\t\t// request stream url\n\t\t\tresponseText, err := chatStreamHandler(c, taskData.URLs.Stream)\n\t\t\tif err != nil {\n\t\t\t\treturn errors.Wrap(err, \"chat stream handler\")\n\t\t\t}\n\n\t\t\tctxMeta := meta.GetByContext(c)\n\t\t\tusage = openai.ResponseText2Usage(responseText,\n\t\t\t\tctxMeta.ActualModelName, ctxMeta.PromptTokens)\n\t\t\treturn nil\n\t\t}()\n\t\tif err != nil {\n\t\t\tif errors.Is(err, errNextLoop) {\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\treturn openai.ErrorWrapper(err, \"chat_task_failed\", http.StatusInternalServerError), nil\n\t\t}\n\n\t\tbreak\n\t}\n\n\treturn nil, usage\n}\n\nconst (\n\teventPrefix = \"event: \"\n\tdataPrefix = \"data: \"\n\tdone = \"[DONE]\"\n)\n\nfunc chatStreamHandler(c *gin.Context, streamUrl string) (responseText string, err error) {\n\t// request stream endpoint\n\tstreamReq, err := http.NewRequestWithContext(c.Request.Context(), http.MethodGet, streamUrl, nil)\n\tif err != nil {\n\t\treturn \"\", errors.Wrap(err, \"new request to stream\")\n\t}\n\n\tstreamReq.Header.Set(\"Authorization\", \"Bearer \"+meta.GetByContext(c).APIKey)\n\tstreamReq.Header.Set(\"Accept\", \"text/event-stream\")\n\tstreamReq.Header.Set(\"Cache-Control\", \"no-store\")\n\n\tresp, err := http.DefaultClient.Do(streamReq)\n\tif err != nil {\n\t\treturn \"\", errors.Wrap(err, \"do request to stream\")\n\t}\n\tdefer resp.Body.Close()\n\n\tif resp.StatusCode != http.StatusOK {\n\t\tpayload, _ := io.ReadAll(resp.Body)\n\t\treturn \"\", errors.Errorf(\"bad status code [%d]%s\", resp.StatusCode, string(payload))\n\t}\n\n\tscanner := bufio.NewScanner(resp.Body)\n\tscanner.Split(bufio.ScanLines)\n\n\tcommon.SetEventStreamHeaders(c)\n\tdoneRendered := false\n\tfor scanner.Scan() {\n\t\tline := strings.TrimSpace(scanner.Text())\n\t\tif line == \"\" {\n\t\t\tcontinue\n\t\t}\n\n\t\t// Handle comments starting with ':'\n\t\tif strings.HasPrefix(line, \":\") {\n\t\t\tcontinue\n\t\t}\n\n\t\t// Parse SSE fields\n\t\tif strings.HasPrefix(line, eventPrefix) {\n\t\t\tevent := strings.TrimSpace(line[len(eventPrefix):])\n\t\t\tvar data string\n\t\t\t// Read the following lines to get data and id\n\t\t\tfor scanner.Scan() {\n\t\t\t\tnextLine := scanner.Text()\n\t\t\t\tif nextLine == \"\" {\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t\tif strings.HasPrefix(nextLine, dataPrefix) {\n\t\t\t\t\tdata = nextLine[len(dataPrefix):]\n\t\t\t\t} else if strings.HasPrefix(nextLine, \"id:\") {\n\t\t\t\t\t// id = strings.TrimSpace(nextLine[len(\"id:\"):])\n\t\t\t\t}\n\t\t\t}\n\n\t\t\tif event == \"output\" {\n\t\t\t\trender.StringData(c, data)\n\t\t\t\tresponseText += data\n\t\t\t} else if event == \"done\" {\n\t\t\t\trender.Done(c)\n\t\t\t\tdoneRendered = true\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t}\n\n\tif err := scanner.Err(); err != nil {\n\t\treturn \"\", errors.Wrap(err, \"scan stream\")\n\t}\n\n\tif !doneRendered {\n\t\trender.Done(c)\n\t}\n\n\treturn responseText, nil\n}\n"
},
{
"path": "relay/adaptor/replicate/constant.go",
"content": "package replicate\n\n// ModelList is a list of models that can be used with Replicate.\n//\n// https://replicate.com/pricing\nvar ModelList = []string{\n\t// -------------------------------------\n\t// image model\n\t// -------------------------------------\n\t\"black-forest-labs/flux-1.1-pro\",\n\t\"black-forest-labs/flux-1.1-pro-ultra\",\n\t\"black-forest-labs/flux-canny-dev\",\n\t\"black-forest-labs/flux-canny-pro\",\n\t\"black-forest-labs/flux-depth-dev\",\n\t\"black-forest-labs/flux-depth-pro\",\n\t\"black-forest-labs/flux-dev\",\n\t\"black-forest-labs/flux-dev-lora\",\n\t\"black-forest-labs/flux-fill-dev\",\n\t\"black-forest-labs/flux-fill-pro\",\n\t\"black-forest-labs/flux-pro\",\n\t\"black-forest-labs/flux-redux-dev\",\n\t\"black-forest-labs/flux-redux-schnell\",\n\t\"black-forest-labs/flux-schnell\",\n\t\"black-forest-labs/flux-schnell-lora\",\n\t\"ideogram-ai/ideogram-v2\",\n\t\"ideogram-ai/ideogram-v2-turbo\",\n\t\"recraft-ai/recraft-v3\",\n\t\"recraft-ai/recraft-v3-svg\",\n\t\"stability-ai/stable-diffusion-3\",\n\t\"stability-ai/stable-diffusion-3.5-large\",\n\t\"stability-ai/stable-diffusion-3.5-large-turbo\",\n\t\"stability-ai/stable-diffusion-3.5-medium\",\n\t// -------------------------------------\n\t// language model\n\t// -------------------------------------\n\t\"ibm-granite/granite-20b-code-instruct-8k\",\n\t\"ibm-granite/granite-3.0-2b-instruct\",\n\t\"ibm-granite/granite-3.0-8b-instruct\",\n\t\"ibm-granite/granite-8b-code-instruct-128k\",\n\t\"meta/llama-2-13b\",\n\t\"meta/llama-2-13b-chat\",\n\t\"meta/llama-2-70b\",\n\t\"meta/llama-2-70b-chat\",\n\t\"meta/llama-2-7b\",\n\t\"meta/llama-2-7b-chat\",\n\t\"meta/meta-llama-3.1-405b-instruct\",\n\t\"meta/meta-llama-3-70b\",\n\t\"meta/meta-llama-3-70b-instruct\",\n\t\"meta/meta-llama-3-8b\",\n\t\"meta/meta-llama-3-8b-instruct\",\n\t\"mistralai/mistral-7b-instruct-v0.2\",\n\t\"mistralai/mistral-7b-v0.1\",\n\t\"mistralai/mixtral-8x7b-instruct-v0.1\",\n\t// -------------------------------------\n\t// video model\n\t// -------------------------------------\n\t// \"minimax/video-01\", // TODO: implement the adaptor\n}\n"
},
{
"path": "relay/adaptor/replicate/image.go",
"content": "package replicate\n\nimport (\n\t\"bytes\"\n\t\"encoding/base64\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"image\"\n\t\"image/png\"\n\t\"io\"\n\t\"net/http\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/pkg/errors\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"golang.org/x/image/webp\"\n\t\"golang.org/x/sync/errgroup\"\n)\n\n// ImagesEditsHandler just copy response body to client\n//\n// https://replicate.com/black-forest-labs/flux-fill-pro\n// func ImagesEditsHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n// \tc.Writer.WriteHeader(resp.StatusCode)\n// \tfor k, v := range resp.Header {\n// \t\tc.Writer.Header().Set(k, v[0])\n// \t}\n\n// \tif _, err := io.Copy(c.Writer, resp.Body); err != nil {\n// \t\treturn ErrorWrapper(err, \"copy_response_body_failed\", http.StatusInternalServerError), nil\n// \t}\n// \tdefer resp.Body.Close()\n\n// \treturn nil, nil\n// }\n\nvar errNextLoop = errors.New(\"next_loop\")\n\nfunc ImageHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tif resp.StatusCode != http.StatusCreated {\n\t\tpayload, _ := io.ReadAll(resp.Body)\n\t\treturn openai.ErrorWrapper(\n\t\t\t\terrors.Errorf(\"bad_status_code [%d]%s\", resp.StatusCode, string(payload)),\n\t\t\t\t\"bad_status_code\", http.StatusInternalServerError),\n\t\t\tnil\n\t}\n\n\trespBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\n\trespData := new(ImageResponse)\n\tif err = json.Unmarshal(respBody, respData); err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\n\tfor {\n\t\terr = func() error {\n\t\t\t// get task\n\t\t\ttaskReq, err := http.NewRequestWithContext(c.Request.Context(),\n\t\t\t\thttp.MethodGet, respData.URLs.Get, nil)\n\t\t\tif err != nil {\n\t\t\t\treturn errors.Wrap(err, \"new request\")\n\t\t\t}\n\n\t\t\ttaskReq.Header.Set(\"Authorization\", \"Bearer \"+meta.GetByContext(c).APIKey)\n\t\t\ttaskResp, err := http.DefaultClient.Do(taskReq)\n\t\t\tif err != nil {\n\t\t\t\treturn errors.Wrap(err, \"get task\")\n\t\t\t}\n\t\t\tdefer taskResp.Body.Close()\n\n\t\t\tif taskResp.StatusCode != http.StatusOK {\n\t\t\t\tpayload, _ := io.ReadAll(taskResp.Body)\n\t\t\t\treturn errors.Errorf(\"bad status code [%d]%s\",\n\t\t\t\t\ttaskResp.StatusCode, string(payload))\n\t\t\t}\n\n\t\t\ttaskBody, err := io.ReadAll(taskResp.Body)\n\t\t\tif err != nil {\n\t\t\t\treturn errors.Wrap(err, \"read task response\")\n\t\t\t}\n\n\t\t\ttaskData := new(ImageResponse)\n\t\t\tif err = json.Unmarshal(taskBody, taskData); err != nil {\n\t\t\t\treturn errors.Wrap(err, \"decode task response\")\n\t\t\t}\n\n\t\t\tswitch taskData.Status {\n\t\t\tcase \"succeeded\":\n\t\t\tcase \"failed\", \"canceled\":\n\t\t\t\treturn errors.Errorf(\"task failed: %s\", taskData.Status)\n\t\t\tdefault:\n\t\t\t\ttime.Sleep(time.Second * 3)\n\t\t\t\treturn errNextLoop\n\t\t\t}\n\n\t\t\toutput, err := taskData.GetOutput()\n\t\t\tif err != nil {\n\t\t\t\treturn errors.Wrap(err, \"get output\")\n\t\t\t}\n\t\t\tif len(output) == 0 {\n\t\t\t\treturn errors.New(\"response output is empty\")\n\t\t\t}\n\n\t\t\tvar mu sync.Mutex\n\t\t\tvar pool errgroup.Group\n\t\t\trespBody := &openai.ImageResponse{\n\t\t\t\tCreated: taskData.CompletedAt.Unix(),\n\t\t\t\tData: []openai.ImageData{},\n\t\t\t}\n\n\t\t\tfor _, imgOut := range output {\n\t\t\t\timgOut := imgOut\n\t\t\t\tpool.Go(func() error {\n\t\t\t\t\t// download image\n\t\t\t\t\tdownloadReq, err := http.NewRequestWithContext(c.Request.Context(),\n\t\t\t\t\t\thttp.MethodGet, imgOut, nil)\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn errors.Wrap(err, \"new request\")\n\t\t\t\t\t}\n\n\t\t\t\t\timgResp, err := http.DefaultClient.Do(downloadReq)\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn errors.Wrap(err, \"download image\")\n\t\t\t\t\t}\n\t\t\t\t\tdefer imgResp.Body.Close()\n\n\t\t\t\t\tif imgResp.StatusCode != http.StatusOK {\n\t\t\t\t\t\tpayload, _ := io.ReadAll(imgResp.Body)\n\t\t\t\t\t\treturn errors.Errorf(\"bad status code [%d]%s\",\n\t\t\t\t\t\t\timgResp.StatusCode, string(payload))\n\t\t\t\t\t}\n\n\t\t\t\t\timgData, err := io.ReadAll(imgResp.Body)\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn errors.Wrap(err, \"read image\")\n\t\t\t\t\t}\n\n\t\t\t\t\timgData, err = ConvertImageToPNG(imgData)\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\treturn errors.Wrap(err, \"convert image\")\n\t\t\t\t\t}\n\n\t\t\t\t\tmu.Lock()\n\t\t\t\t\trespBody.Data = append(respBody.Data, openai.ImageData{\n\t\t\t\t\t\tB64Json: fmt.Sprintf(\"data:image/png;base64,%s\",\n\t\t\t\t\t\t\tbase64.StdEncoding.EncodeToString(imgData)),\n\t\t\t\t\t})\n\t\t\t\t\tmu.Unlock()\n\n\t\t\t\t\treturn nil\n\t\t\t\t})\n\t\t\t}\n\n\t\t\tif err := pool.Wait(); err != nil {\n\t\t\t\tif len(respBody.Data) == 0 {\n\t\t\t\t\treturn errors.WithStack(err)\n\t\t\t\t}\n\n\t\t\t\tlogger.Error(c, fmt.Sprintf(\"some images failed to download: %+v\", err))\n\t\t\t}\n\n\t\t\tc.JSON(http.StatusOK, respBody)\n\t\t\treturn nil\n\t\t}()\n\t\tif err != nil {\n\t\t\tif errors.Is(err, errNextLoop) {\n\t\t\t\tcontinue\n\t\t\t}\n\n\t\t\treturn openai.ErrorWrapper(err, \"image_task_failed\", http.StatusInternalServerError), nil\n\t\t}\n\n\t\tbreak\n\t}\n\n\treturn nil, nil\n}\n\n// ConvertImageToPNG converts a WebP image to PNG format\nfunc ConvertImageToPNG(webpData []byte) ([]byte, error) {\n\t// bypass if it's already a PNG image\n\tif bytes.HasPrefix(webpData, []byte(\"\\x89PNG\")) {\n\t\treturn webpData, nil\n\t}\n\n\t// check if is jpeg, convert to png\n\tif bytes.HasPrefix(webpData, []byte(\"\\xff\\xd8\\xff\")) {\n\t\timg, _, err := image.Decode(bytes.NewReader(webpData))\n\t\tif err != nil {\n\t\t\treturn nil, errors.Wrap(err, \"decode jpeg\")\n\t\t}\n\n\t\tvar pngBuffer bytes.Buffer\n\t\tif err := png.Encode(&pngBuffer, img); err != nil {\n\t\t\treturn nil, errors.Wrap(err, \"encode png\")\n\t\t}\n\n\t\treturn pngBuffer.Bytes(), nil\n\t}\n\n\t// Decode the WebP image\n\timg, err := webp.Decode(bytes.NewReader(webpData))\n\tif err != nil {\n\t\treturn nil, errors.Wrap(err, \"decode webp\")\n\t}\n\n\t// Encode the image as PNG\n\tvar pngBuffer bytes.Buffer\n\tif err := png.Encode(&pngBuffer, img); err != nil {\n\t\treturn nil, errors.Wrap(err, \"encode png\")\n\t}\n\n\treturn pngBuffer.Bytes(), nil\n}\n"
},
{
"path": "relay/adaptor/replicate/model.go",
"content": "package replicate\n\nimport (\n\t\"time\"\n\n\t\"github.com/pkg/errors\"\n)\n\n// DrawImageRequest draw image by fluxpro\n//\n// https://replicate.com/black-forest-labs/flux-pro?prediction=kg1krwsdf9rg80ch1sgsrgq7h8&output=json\ntype DrawImageRequest struct {\n\tInput ImageInput `json:\"input\"`\n}\n\n// ImageInput is input of DrawImageByFluxProRequest\n//\n// https://replicate.com/black-forest-labs/flux-1.1-pro/api/schema\ntype ImageInput struct {\n\tSteps int `json:\"steps\" binding:\"required,min=1\"`\n\tPrompt string `json:\"prompt\" binding:\"required,min=5\"`\n\tImagePrompt string `json:\"image_prompt\"`\n\tGuidance int `json:\"guidance\" binding:\"required,min=2,max=5\"`\n\tInterval int `json:\"interval\" binding:\"required,min=1,max=4\"`\n\tAspectRatio string `json:\"aspect_ratio\" binding:\"required,oneof=1:1 16:9 2:3 3:2 4:5 5:4 9:16\"`\n\tSafetyTolerance int `json:\"safety_tolerance\" binding:\"required,min=1,max=5\"`\n\tSeed int `json:\"seed\"`\n\tNImages int `json:\"n_images\" binding:\"required,min=1,max=8\"`\n\tWidth int `json:\"width\" binding:\"required,min=256,max=1440\"`\n\tHeight int `json:\"height\" binding:\"required,min=256,max=1440\"`\n}\n\n// InpaintingImageByFlusReplicateRequest is request to inpainting image by flux pro\n//\n// https://replicate.com/black-forest-labs/flux-fill-pro/api/schema\ntype InpaintingImageByFlusReplicateRequest struct {\n\tInput FluxInpaintingInput `json:\"input\"`\n}\n\n// FluxInpaintingInput is input of DrawImageByFluxProRequest\n//\n// https://replicate.com/black-forest-labs/flux-fill-pro/api/schema\ntype FluxInpaintingInput struct {\n\tMask string `json:\"mask\" binding:\"required\"`\n\tImage string `json:\"image\" binding:\"required\"`\n\tSeed int `json:\"seed\"`\n\tSteps int `json:\"steps\" binding:\"required,min=1\"`\n\tPrompt string `json:\"prompt\" binding:\"required,min=5\"`\n\tGuidance int `json:\"guidance\" binding:\"required,min=2,max=5\"`\n\tOutputFormat string `json:\"output_format\"`\n\tSafetyTolerance int `json:\"safety_tolerance\" binding:\"required,min=1,max=5\"`\n\tPromptUnsampling bool `json:\"prompt_unsampling\"`\n}\n\n// ImageResponse is response of DrawImageByFluxProRequest\n//\n// https://replicate.com/black-forest-labs/flux-pro?prediction=kg1krwsdf9rg80ch1sgsrgq7h8&output=json\ntype ImageResponse struct {\n\tCompletedAt time.Time `json:\"completed_at\"`\n\tCreatedAt time.Time `json:\"created_at\"`\n\tDataRemoved bool `json:\"data_removed\"`\n\tError string `json:\"error\"`\n\tID string `json:\"id\"`\n\tInput DrawImageRequest `json:\"input\"`\n\tLogs string `json:\"logs\"`\n\tMetrics FluxMetrics `json:\"metrics\"`\n\t// Output could be `string` or `[]string`\n\tOutput any `json:\"output\"`\n\tStartedAt time.Time `json:\"started_at\"`\n\tStatus string `json:\"status\"`\n\tURLs FluxURLs `json:\"urls\"`\n\tVersion string `json:\"version\"`\n}\n\nfunc (r *ImageResponse) GetOutput() ([]string, error) {\n\tswitch v := r.Output.(type) {\n\tcase string:\n\t\treturn []string{v}, nil\n\tcase []string:\n\t\treturn v, nil\n\tcase nil:\n\t\treturn nil, nil\n\tcase []interface{}:\n\t\t// convert []interface{} to []string\n\t\tret := make([]string, len(v))\n\t\tfor idx, vv := range v {\n\t\t\tif vvv, ok := vv.(string); ok {\n\t\t\t\tret[idx] = vvv\n\t\t\t} else {\n\t\t\t\treturn nil, errors.Errorf(\"unknown output type: [%T]%v\", vv, vv)\n\t\t\t}\n\t\t}\n\n\t\treturn ret, nil\n\tdefault:\n\t\treturn nil, errors.Errorf(\"unknown output type: [%T]%v\", r.Output, r.Output)\n\t}\n}\n\n// FluxMetrics is metrics of ImageResponse\ntype FluxMetrics struct {\n\tImageCount int `json:\"image_count\"`\n\tPredictTime float64 `json:\"predict_time\"`\n\tTotalTime float64 `json:\"total_time\"`\n}\n\n// FluxURLs is urls of ImageResponse\ntype FluxURLs struct {\n\tGet string `json:\"get\"`\n\tCancel string `json:\"cancel\"`\n}\n\ntype ReplicateChatRequest struct {\n\tInput ChatInput `json:\"input\" form:\"input\" binding:\"required\"`\n}\n\n// ChatInput is input of ChatByReplicateRequest\n//\n// https://replicate.com/meta/meta-llama-3.1-405b-instruct/api/schema\ntype ChatInput struct {\n\tTopK int `json:\"top_k\"`\n\tTopP float64 `json:\"top_p\"`\n\tPrompt string `json:\"prompt\"`\n\tMaxTokens int `json:\"max_tokens\"`\n\tMinTokens int `json:\"min_tokens\"`\n\tTemperature float64 `json:\"temperature\"`\n\tSystemPrompt string `json:\"system_prompt\"`\n\tStopSequences string `json:\"stop_sequences\"`\n\tPromptTemplate string `json:\"prompt_template\"`\n\tPresencePenalty float64 `json:\"presence_penalty\"`\n\tFrequencyPenalty float64 `json:\"frequency_penalty\"`\n}\n\n// ChatResponse is response of ChatByReplicateRequest\n//\n// https://replicate.com/meta/meta-llama-3.1-405b-instruct/examples?input=http&output=json\ntype ChatResponse struct {\n\tCompletedAt time.Time `json:\"completed_at\"`\n\tCreatedAt time.Time `json:\"created_at\"`\n\tDataRemoved bool `json:\"data_removed\"`\n\tError string `json:\"error\"`\n\tID string `json:\"id\"`\n\tInput ChatInput `json:\"input\"`\n\tLogs string `json:\"logs\"`\n\tMetrics FluxMetrics `json:\"metrics\"`\n\t// Output could be `string` or `[]string`\n\tOutput []string `json:\"output\"`\n\tStartedAt time.Time `json:\"started_at\"`\n\tStatus string `json:\"status\"`\n\tURLs ChatResponseUrl `json:\"urls\"`\n\tVersion string `json:\"version\"`\n}\n\n// ChatResponseUrl is task urls of ChatResponse\ntype ChatResponseUrl struct {\n\tStream string `json:\"stream\"`\n\tGet string `json:\"get\"`\n\tCancel string `json:\"cancel\"`\n}\n"
},
{
"path": "relay/adaptor/siliconflow/constants.go",
"content": "package siliconflow\n\n// https://docs.siliconflow.cn/docs/getting-started\n\nvar ModelList = []string{\n\t\"deepseek-ai/deepseek-llm-67b-chat\",\n\t\"Qwen/Qwen1.5-14B-Chat\",\n\t\"Qwen/Qwen1.5-7B-Chat\",\n\t\"Qwen/Qwen1.5-110B-Chat\",\n\t\"Qwen/Qwen1.5-32B-Chat\",\n\t\"01-ai/Yi-1.5-6B-Chat\",\n\t\"01-ai/Yi-1.5-9B-Chat-16K\",\n\t\"01-ai/Yi-1.5-34B-Chat-16K\",\n\t\"THUDM/chatglm3-6b\",\n\t\"deepseek-ai/DeepSeek-V2-Chat\",\n\t\"THUDM/glm-4-9b-chat\",\n\t\"Qwen/Qwen2-72B-Instruct\",\n\t\"Qwen/Qwen2-7B-Instruct\",\n\t\"Qwen/Qwen2-57B-A14B-Instruct\",\n\t\"deepseek-ai/DeepSeek-Coder-V2-Instruct\",\n\t\"Qwen/Qwen2-1.5B-Instruct\",\n\t\"internlm/internlm2_5-7b-chat\",\n\t\"BAAI/bge-large-en-v1.5\",\n\t\"BAAI/bge-large-zh-v1.5\",\n\t\"Pro/Qwen/Qwen2-7B-Instruct\",\n\t\"Pro/Qwen/Qwen2-1.5B-Instruct\",\n\t\"Pro/Qwen/Qwen1.5-7B-Chat\",\n\t\"Pro/THUDM/glm-4-9b-chat\",\n\t\"Pro/THUDM/chatglm3-6b\",\n\t\"Pro/01-ai/Yi-1.5-9B-Chat-16K\",\n\t\"Pro/01-ai/Yi-1.5-6B-Chat\",\n\t\"Pro/google/gemma-2-9b-it\",\n\t\"Pro/internlm/internlm2_5-7b-chat\",\n\t\"Pro/meta-llama/Meta-Llama-3-8B-Instruct\",\n\t\"Pro/mistralai/Mistral-7B-Instruct-v0.2\",\n}\n"
},
{
"path": "relay/adaptor/stepfun/constants.go",
"content": "package stepfun\n\nvar ModelList = []string{\n\t\"step-1-8k\",\n\t\"step-1-32k\",\n\t\"step-1-128k\",\n\t\"step-1-256k\",\n\t\"step-1-flash\",\n\t\"step-2-16k\",\n\t\"step-1v-8k\",\n\t\"step-1v-32k\",\n\t\"step-1x-medium\",\n}\n"
},
{
"path": "relay/adaptor/tencent/adaptor.go",
"content": "package tencent\n\nimport (\n\t\"errors\"\n\t\"io\"\n\t\"net/http\"\n\t\"strconv\"\n\t\"strings\"\n\n\t\"github.com/gin-gonic/gin\"\n\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"github.com/songquanpeng/one-api/relay/relaymode\"\n)\n\n// https://cloud.tencent.com/document/api/1729/101837\n\ntype Adaptor struct {\n\tSign string\n\tAction string\n\tVersion string\n\tTimestamp int64\n}\n\nfunc (a *Adaptor) Init(meta *meta.Meta) {\n\ta.Action = \"ChatCompletions\"\n\ta.Version = \"2023-09-01\"\n\ta.Timestamp = helper.GetTimestamp()\n}\n\nfunc (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {\n\treturn meta.BaseURL + \"/\", nil\n}\n\nfunc (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {\n\tadaptor.SetupCommonRequestHeader(c, req, meta)\n\treq.Header.Set(\"Authorization\", a.Sign)\n\treq.Header.Set(\"X-TC-Action\", a.Action)\n\treq.Header.Set(\"X-TC-Version\", a.Version)\n\treq.Header.Set(\"X-TC-Timestamp\", strconv.FormatInt(a.Timestamp, 10))\n\treturn nil\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\tapiKey := c.Request.Header.Get(\"Authorization\")\n\tapiKey = strings.TrimPrefix(apiKey, \"Bearer \")\n\t_, secretId, secretKey, err := ParseConfig(apiKey)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tvar convertedRequest any\n\tswitch relayMode {\n\tcase relaymode.Embeddings:\n\t\ta.Action = \"GetEmbedding\"\n\t\tconvertedRequest = ConvertEmbeddingRequest(*request)\n\tdefault:\n\t\ta.Action = \"ChatCompletions\"\n\t\tconvertedRequest = ConvertRequest(*request)\n\t}\n\t// we have to calculate the sign here\n\ta.Sign = GetSign(convertedRequest, a, secretId, secretKey)\n\treturn convertedRequest, nil\n}\n\nfunc (a *Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\treturn request, nil\n}\n\nfunc (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {\n\treturn adaptor.DoRequestHelper(a, c, meta, requestBody)\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tif meta.IsStream {\n\t\tvar responseText string\n\t\terr, responseText = StreamHandler(c, resp)\n\t\tusage = openai.ResponseText2Usage(responseText, meta.ActualModelName, meta.PromptTokens)\n\t} else {\n\t\tswitch meta.Mode {\n\t\tcase relaymode.Embeddings:\n\t\t\terr, usage = EmbeddingHandler(c, resp)\n\t\tdefault:\n\t\t\terr, usage = Handler(c, resp)\n\t\t}\n\t}\n\treturn\n}\n\nfunc (a *Adaptor) GetModelList() []string {\n\treturn ModelList\n}\n\nfunc (a *Adaptor) GetChannelName() string {\n\treturn \"tencent\"\n}\n"
},
{
"path": "relay/adaptor/tencent/constants.go",
"content": "package tencent\n\nvar ModelList = []string{\n\t\"hunyuan-lite\",\n\t\"hunyuan-standard\",\n\t\"hunyuan-standard-256K\",\n\t\"hunyuan-pro\",\n\t\"hunyuan-vision\",\n\t\"hunyuan-embedding\",\n}\n"
},
{
"path": "relay/adaptor/tencent/main.go",
"content": "package tencent\n\nimport (\n\t\"bufio\"\n\t\"crypto/hmac\"\n\t\"crypto/sha256\"\n\t\"encoding/hex\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"io\"\n\t\"net/http\"\n\t\"strconv\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/gin-gonic/gin\"\n\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/conv\"\n\t\"github.com/songquanpeng/one-api/common/ctxkey\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/common/random\"\n\t\"github.com/songquanpeng/one-api/common/render\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/constant\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\nfunc ConvertRequest(request model.GeneralOpenAIRequest) *ChatRequest {\n\tmessages := make([]*Message, 0, len(request.Messages))\n\tfor i := 0; i < len(request.Messages); i++ {\n\t\tmessage := request.Messages[i]\n\t\tmessages = append(messages, &Message{\n\t\t\tContent: message.StringContent(),\n\t\t\tRole: message.Role,\n\t\t})\n\t}\n\treturn &ChatRequest{\n\t\tModel: &request.Model,\n\t\tStream: &request.Stream,\n\t\tMessages: messages,\n\t\tTopP: request.TopP,\n\t\tTemperature: request.Temperature,\n\t}\n}\n\nfunc ConvertEmbeddingRequest(request model.GeneralOpenAIRequest) *EmbeddingRequest {\n\treturn &EmbeddingRequest{\n\t\tInputList: request.ParseInput(),\n\t}\n}\n\nfunc EmbeddingHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tvar tencentResponseP EmbeddingResponseP\n\terr := json.NewDecoder(resp.Body).Decode(&tencentResponseP)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\n\ttencentResponse := tencentResponseP.Response\n\tif tencentResponse.Error.Code != \"\" {\n\t\treturn &model.ErrorWithStatusCode{\n\t\t\tError: model.Error{\n\t\t\t\tMessage: tencentResponse.Error.Message,\n\t\t\t\tCode: tencentResponse.Error.Code,\n\t\t\t},\n\t\t\tStatusCode: resp.StatusCode,\n\t\t}, nil\n\t}\n\trequestModel := c.GetString(ctxkey.RequestModel)\n\tfullTextResponse := embeddingResponseTencent2OpenAI(&tencentResponse)\n\tfullTextResponse.Model = requestModel\n\tjsonResponse, err := json.Marshal(fullTextResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tc.Writer.Header().Set(\"Content-Type\", \"application/json\")\n\tc.Writer.WriteHeader(resp.StatusCode)\n\t_, err = c.Writer.Write(jsonResponse)\n\treturn nil, &fullTextResponse.Usage\n}\n\nfunc embeddingResponseTencent2OpenAI(response *EmbeddingResponse) *openai.EmbeddingResponse {\n\topenAIEmbeddingResponse := openai.EmbeddingResponse{\n\t\tObject: \"list\",\n\t\tData: make([]openai.EmbeddingResponseItem, 0, len(response.Data)),\n\t\tModel: \"hunyuan-embedding\",\n\t\tUsage: model.Usage{TotalTokens: response.EmbeddingUsage.TotalTokens},\n\t}\n\n\tfor _, item := range response.Data {\n\t\topenAIEmbeddingResponse.Data = append(openAIEmbeddingResponse.Data, openai.EmbeddingResponseItem{\n\t\t\tObject: item.Object,\n\t\t\tIndex: item.Index,\n\t\t\tEmbedding: item.Embedding,\n\t\t})\n\t}\n\treturn &openAIEmbeddingResponse\n}\n\nfunc responseTencent2OpenAI(response *ChatResponse) *openai.TextResponse {\n\tfullTextResponse := openai.TextResponse{\n\t\tId: response.ReqID,\n\t\tObject: \"chat.completion\",\n\t\tCreated: helper.GetTimestamp(),\n\t\tUsage: model.Usage{\n\t\t\tPromptTokens: response.Usage.PromptTokens,\n\t\t\tCompletionTokens: response.Usage.CompletionTokens,\n\t\t\tTotalTokens: response.Usage.TotalTokens,\n\t\t},\n\t}\n\tif len(response.Choices) > 0 {\n\t\tchoice := openai.TextResponseChoice{\n\t\t\tIndex: 0,\n\t\t\tMessage: model.Message{\n\t\t\t\tRole: \"assistant\",\n\t\t\t\tContent: response.Choices[0].Messages.Content,\n\t\t\t},\n\t\t\tFinishReason: response.Choices[0].FinishReason,\n\t\t}\n\t\tfullTextResponse.Choices = append(fullTextResponse.Choices, choice)\n\t}\n\treturn &fullTextResponse\n}\n\nfunc streamResponseTencent2OpenAI(TencentResponse *ChatResponse) *openai.ChatCompletionsStreamResponse {\n\tresponse := openai.ChatCompletionsStreamResponse{\n\t\tId: fmt.Sprintf(\"chatcmpl-%s\", random.GetUUID()),\n\t\tObject: \"chat.completion.chunk\",\n\t\tCreated: helper.GetTimestamp(),\n\t\tModel: \"tencent-hunyuan\",\n\t}\n\tif len(TencentResponse.Choices) > 0 {\n\t\tvar choice openai.ChatCompletionsStreamResponseChoice\n\t\tchoice.Delta.Content = TencentResponse.Choices[0].Delta.Content\n\t\tif TencentResponse.Choices[0].FinishReason == \"stop\" {\n\t\t\tchoice.FinishReason = &constant.StopFinishReason\n\t\t}\n\t\tresponse.Choices = append(response.Choices, choice)\n\t}\n\treturn &response\n}\n\nfunc StreamHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, string) {\n\tvar responseText string\n\tscanner := bufio.NewScanner(resp.Body)\n\tscanner.Split(bufio.ScanLines)\n\n\tcommon.SetEventStreamHeaders(c)\n\n\tfor scanner.Scan() {\n\t\tdata := scanner.Text()\n\t\tif len(data) < 5 || !strings.HasPrefix(data, \"data:\") {\n\t\t\tcontinue\n\t\t}\n\t\tdata = strings.TrimPrefix(data, \"data:\")\n\n\t\tvar tencentResponse ChatResponse\n\t\terr := json.Unmarshal([]byte(data), &tencentResponse)\n\t\tif err != nil {\n\t\t\tlogger.SysError(\"error unmarshalling stream response: \" + err.Error())\n\t\t\tcontinue\n\t\t}\n\n\t\tresponse := streamResponseTencent2OpenAI(&tencentResponse)\n\t\tif len(response.Choices) != 0 {\n\t\t\tresponseText += conv.AsString(response.Choices[0].Delta.Content)\n\t\t}\n\n\t\terr = render.ObjectData(c, response)\n\t\tif err != nil {\n\t\t\tlogger.SysError(err.Error())\n\t\t}\n\t}\n\n\tif err := scanner.Err(); err != nil {\n\t\tlogger.SysError(\"error reading stream: \" + err.Error())\n\t}\n\n\trender.Done(c)\n\n\terr := resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), \"\"\n\t}\n\n\treturn nil, responseText\n}\n\nfunc Handler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tvar TencentResponse ChatResponse\n\tvar responseP ChatResponseP\n\tresponseBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = json.Unmarshal(responseBody, &responseP)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tTencentResponse = responseP.Response\n\tif TencentResponse.Error.Code != \"\" {\n\t\treturn &model.ErrorWithStatusCode{\n\t\t\tError: model.Error{\n\t\t\t\tMessage: TencentResponse.Error.Message,\n\t\t\t\tCode: TencentResponse.Error.Code,\n\t\t\t},\n\t\t\tStatusCode: resp.StatusCode,\n\t\t}, nil\n\t}\n\tfullTextResponse := responseTencent2OpenAI(&TencentResponse)\n\tfullTextResponse.Model = \"hunyuan\"\n\tjsonResponse, err := json.Marshal(fullTextResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tc.Writer.Header().Set(\"Content-Type\", \"application/json\")\n\tc.Writer.WriteHeader(resp.StatusCode)\n\t_, err = c.Writer.Write(jsonResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"write_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\treturn nil, &fullTextResponse.Usage\n}\n\nfunc ParseConfig(config string) (appId int64, secretId string, secretKey string, err error) {\n\tparts := strings.Split(config, \"|\")\n\tif len(parts) != 3 {\n\t\terr = errors.New(\"invalid tencent config\")\n\t\treturn\n\t}\n\tappId, err = strconv.ParseInt(parts[0], 10, 64)\n\tsecretId = parts[1]\n\tsecretKey = parts[2]\n\treturn\n}\n\nfunc sha256hex(s string) string {\n\tb := sha256.Sum256([]byte(s))\n\treturn hex.EncodeToString(b[:])\n}\n\nfunc hmacSha256(s, key string) string {\n\thashed := hmac.New(sha256.New, []byte(key))\n\thashed.Write([]byte(s))\n\treturn string(hashed.Sum(nil))\n}\n\nfunc GetSign(req any, adaptor *Adaptor, secId, secKey string) string {\n\t// build canonical request string\n\thost := \"hunyuan.tencentcloudapi.com\"\n\thttpRequestMethod := \"POST\"\n\tcanonicalURI := \"/\"\n\tcanonicalQueryString := \"\"\n\tcanonicalHeaders := fmt.Sprintf(\"content-type:%s\\nhost:%s\\nx-tc-action:%s\\n\",\n\t\t\"application/json\", host, strings.ToLower(adaptor.Action))\n\tsignedHeaders := \"content-type;host;x-tc-action\"\n\tpayload, _ := json.Marshal(req)\n\thashedRequestPayload := sha256hex(string(payload))\n\tcanonicalRequest := fmt.Sprintf(\"%s\\n%s\\n%s\\n%s\\n%s\\n%s\",\n\t\thttpRequestMethod,\n\t\tcanonicalURI,\n\t\tcanonicalQueryString,\n\t\tcanonicalHeaders,\n\t\tsignedHeaders,\n\t\thashedRequestPayload)\n\t// build string to sign\n\talgorithm := \"TC3-HMAC-SHA256\"\n\trequestTimestamp := strconv.FormatInt(adaptor.Timestamp, 10)\n\ttimestamp, _ := strconv.ParseInt(requestTimestamp, 10, 64)\n\tt := time.Unix(timestamp, 0).UTC()\n\t// must be the format 2006-01-02, ref to package time for more info\n\tdate := t.Format(\"2006-01-02\")\n\tcredentialScope := fmt.Sprintf(\"%s/%s/tc3_request\", date, \"hunyuan\")\n\thashedCanonicalRequest := sha256hex(canonicalRequest)\n\tstring2sign := fmt.Sprintf(\"%s\\n%s\\n%s\\n%s\",\n\t\talgorithm,\n\t\trequestTimestamp,\n\t\tcredentialScope,\n\t\thashedCanonicalRequest)\n\n\t// sign string\n\tsecretDate := hmacSha256(date, \"TC3\"+secKey)\n\tsecretService := hmacSha256(\"hunyuan\", secretDate)\n\tsecretKey := hmacSha256(\"tc3_request\", secretService)\n\tsignature := hex.EncodeToString([]byte(hmacSha256(string2sign, secretKey)))\n\n\t// build authorization\n\tauthorization := fmt.Sprintf(\"%s Credential=%s/%s, SignedHeaders=%s, Signature=%s\",\n\t\talgorithm,\n\t\tsecId,\n\t\tcredentialScope,\n\t\tsignedHeaders,\n\t\tsignature)\n\treturn authorization\n}\n"
},
{
"path": "relay/adaptor/tencent/model.go",
"content": "package tencent\n\ntype Message struct {\n\tRole string `json:\"Role\"`\n\tContent string `json:\"Content\"`\n}\n\ntype ChatRequest struct {\n\t// 模型名称，可选值包括 hunyuan-lite、hunyuan-standard、hunyuan-standard-256K、hunyuan-pro。\n\t// 各模型介绍请阅读 [产品概述](https://cloud.tencent.com/document/product/1729/104753) 中的说明。\n\t//\n\t// 注意：\n\t// 不同的模型计费不同，请根据 [购买指南](https://cloud.tencent.com/document/product/1729/97731) 按需调用。\n\tModel *string `json:\"Model\"`\n\t// 聊天上下文信息。\n\t// 说明：\n\t// 1. 长度最多为 40，按对话时间从旧到新在数组中排列。\n\t// 2. Message.Role 可选值：system、user、assistant。\n\t// 其中，system 角色可选，如存在则必须位于列表的最开始。user 和 assistant 需交替出现（一问一答），以 user 提问开始和结束，且 Content 不能为空。Role 的顺序示例：[system（可选） user assistant user assistant user ...]。\n\t// 3. Messages 中 Content 总长度不能超过模型输入长度上限（可参考 [产品概述](https://cloud.tencent.com/document/product/1729/104753) 文档），超过则会截断最前面的内容，只保留尾部内容。\n\tMessages []*Message `json:\"Messages\"`\n\t// 流式调用开关。\n\t// 说明：\n\t// 1. 未传值时默认为非流式调用（false）。\n\t// 2. 流式调用时以 SSE 协议增量返回结果（返回值取 Choices[n].Delta 中的值，需要拼接增量数据才能获得完整结果）。\n\t// 3. 非流式调用时：\n\t// 调用方式与普通 HTTP 请求无异。\n\t// 接口响应耗时较长，**如需更低时延建议设置为 true**。\n\t// 只返回一次最终结果（返回值取 Choices[n].Message 中的值）。\n\t//\n\t// 注意：\n\t// 通过 SDK 调用时，流式和非流式调用需用**不同的方式**获取返回值，具体参考 SDK 中的注释或示例（在各语言 SDK 代码仓库的 examples/hunyuan/v20230901/ 目录中）。\n\tStream *bool `json:\"Stream\"`\n\t// 说明：\n\t// 1. 影响输出文本的多样性，取值越大，生成文本的多样性越强。\n\t// 2. 取值区间为 [0.0, 1.0]，未传值时使用各模型推荐值。\n\t// 3. 非必要不建议使用，不合理的取值会影响效果。\n\tTopP *float64 `json:\"TopP,omitempty\"`\n\t// 说明：\n\t// 1. 较高的数值会使输出更加随机，而较低的数值会使其更加集中和确定。\n\t// 2. 取值区间为 [0.0, 2.0]，未传值时使用各模型推荐值。\n\t// 3. 非必要不建议使用，不合理的取值会影响效果。\n\tTemperature *float64 `json:\"Temperature,omitempty\"`\n}\n\ntype Error struct {\n\tCode string `json:\"Code\"`\n\tMessage string `json:\"Message\"`\n}\n\ntype Usage struct {\n\tPromptTokens int `json:\"PromptTokens\"`\n\tCompletionTokens int `json:\"CompletionTokens\"`\n\tTotalTokens int `json:\"TotalTokens\"`\n}\n\ntype ResponseChoices struct {\n\tFinishReason string `json:\"FinishReason,omitempty\"` // 流式结束标志位，为 stop 则表示尾包\n\tMessages Message `json:\"Message,omitempty\"` // 内容，同步模式返回内容，流模式为 null 输出 content 内容总数最多支持 1024token。\n\tDelta Message `json:\"Delta,omitempty\"` // 内容，流模式返回内容，同步模式为 null 输出 content 内容总数最多支持 1024token。\n}\n\ntype ChatResponse struct {\n\tChoices []ResponseChoices `json:\"Choices,omitempty\"` // 结果\n\tCreated int64 `json:\"Created,omitempty\"` // unix 时间戳的字符串\n\tId string `json:\"Id,omitempty\"` // 会话 id\n\tUsage Usage `json:\"Usage,omitempty\"` // token 数量\n\tError Error `json:\"Error,omitempty\"` // 错误信息 注意：此字段可能返回 null，表示取不到有效值\n\tNote string `json:\"Note,omitempty\"` // 注释\n\tReqID string `json:\"RequestId,omitempty\"` // 唯一请求 Id，每次请求都会返回。用于反馈接口入参\n}\n\ntype ChatResponseP struct {\n\tResponse ChatResponse `json:\"Response,omitempty\"`\n}\n\ntype EmbeddingRequest struct {\n\tInputList []string `json:\"InputList\"`\n}\n\ntype EmbeddingData struct {\n\tEmbedding []float64 `json:\"Embedding\"`\n\tIndex int `json:\"Index\"`\n\tObject string `json:\"Object\"`\n}\n\ntype EmbeddingUsage struct {\n\tPromptTokens int `json:\"PromptTokens\"`\n\tTotalTokens int `json:\"TotalTokens\"`\n}\n\ntype EmbeddingResponse struct {\n\tData []EmbeddingData `json:\"Data\"`\n\tEmbeddingUsage EmbeddingUsage `json:\"Usage,omitempty\"`\n\tRequestId string `json:\"RequestId,omitempty\"`\n\tError Error `json:\"Error,omitempty\"`\n}\n\ntype EmbeddingResponseP struct {\n\tResponse EmbeddingResponse `json:\"Response,omitempty\"`\n}\n"
},
{
"path": "relay/adaptor/togetherai/constants.go",
"content": "package togetherai\n\n// https://docs.together.ai/docs/inference-models\n\nvar ModelList = []string{\n\t\"meta-llama/Llama-3-70b-chat-hf\",\n\t\"deepseek-ai/deepseek-coder-33b-instruct\",\n\t\"mistralai/Mixtral-8x22B-Instruct-v0.1\",\n\t\"Qwen/Qwen1.5-72B-Chat\",\n}\n"
},
{
"path": "relay/adaptor/vertexai/adaptor.go",
"content": "package vertexai\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor\"\n\tchannelhelper \"github.com/songquanpeng/one-api/relay/adaptor\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\trelaymodel \"github.com/songquanpeng/one-api/relay/model\"\n)\n\nvar _ adaptor.Adaptor = new(Adaptor)\n\nconst channelName = \"vertexai\"\n\ntype Adaptor struct{}\n\nfunc (a *Adaptor) Init(meta *meta.Meta) {\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\n\tadaptor := GetAdaptor(request.Model)\n\tif adaptor == nil {\n\t\treturn nil, errors.New(\"adaptor not found\")\n\t}\n\n\treturn adaptor.ConvertRequest(c, relayMode, request)\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tadaptor := GetAdaptor(meta.ActualModelName)\n\tif adaptor == nil {\n\t\treturn nil, &relaymodel.ErrorWithStatusCode{\n\t\t\tStatusCode: http.StatusInternalServerError,\n\t\t\tError: relaymodel.Error{\n\t\t\t\tMessage: \"adaptor not found\",\n\t\t\t},\n\t\t}\n\t}\n\treturn adaptor.DoResponse(c, resp, meta)\n}\n\nfunc (a *Adaptor) GetModelList() (models []string) {\n\tmodels = modelList\n\treturn\n}\n\nfunc (a *Adaptor) GetChannelName() string {\n\treturn channelName\n}\n\nfunc (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {\n\tsuffix := \"\"\n\tif strings.HasPrefix(meta.ActualModelName, \"gemini\") {\n\t\tif meta.IsStream {\n\t\t\tsuffix = \"streamGenerateContent?alt=sse\"\n\t\t} else {\n\t\t\tsuffix = \"generateContent\"\n\t\t}\n\t} else {\n\t\tif meta.IsStream {\n\t\t\tsuffix = \"streamRawPredict?alt=sse\"\n\t\t} else {\n\t\t\tsuffix = \"rawPredict\"\n\t\t}\n\t}\n\n\tif meta.BaseURL != \"\" {\n\t\treturn fmt.Sprintf(\n\t\t\t\"%s/v1/projects/%s/locations/%s/publishers/google/models/%s:%s\",\n\t\t\tmeta.BaseURL,\n\t\t\tmeta.Config.VertexAIProjectID,\n\t\t\tmeta.Config.Region,\n\t\t\tmeta.ActualModelName,\n\t\t\tsuffix,\n\t\t), nil\n\t}\n\treturn fmt.Sprintf(\n\t\t\"https://%s-aiplatform.googleapis.com/v1/projects/%s/locations/%s/publishers/google/models/%s:%s\",\n\t\tmeta.Config.Region,\n\t\tmeta.Config.VertexAIProjectID,\n\t\tmeta.Config.Region,\n\t\tmeta.ActualModelName,\n\t\tsuffix,\n\t), nil\n}\n\nfunc (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {\n\tadaptor.SetupCommonRequestHeader(c, req, meta)\n\ttoken, err := getToken(c, meta.ChannelId, meta.Config.VertexAIADC)\n\tif err != nil {\n\t\treturn err\n\t}\n\treq.Header.Set(\"Authorization\", \"Bearer \"+token)\n\treturn nil\n}\n\nfunc (a *Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\treturn request, nil\n}\n\nfunc (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {\n\treturn channelhelper.DoRequestHelper(a, c, meta, requestBody)\n}\n"
},
{
"path": "relay/adaptor/vertexai/claude/adapter.go",
"content": "package vertexai\n\nimport (\n\t\"net/http\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/pkg/errors\"\n\t\"github.com/songquanpeng/one-api/common/ctxkey\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/anthropic\"\n\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\nvar ModelList = []string{\n\t\"claude-3-haiku@20240307\",\n\t\"claude-3-sonnet@20240229\",\n\t\"claude-3-opus@20240229\",\n\t\"claude-3-5-sonnet@20240620\",\n\t\"claude-3-5-sonnet-v2@20241022\",\n\t\"claude-3-5-haiku@20241022\",\n}\n\nconst anthropicVersion = \"vertex-2023-10-16\"\n\ntype Adaptor struct {\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\n\tclaudeReq := anthropic.ConvertRequest(*request)\n\treq := Request{\n\t\tAnthropicVersion: anthropicVersion,\n\t\t// Model: claudeReq.Model,\n\t\tMessages: claudeReq.Messages,\n\t\tSystem: claudeReq.System,\n\t\tMaxTokens: claudeReq.MaxTokens,\n\t\tTemperature: claudeReq.Temperature,\n\t\tTopP: claudeReq.TopP,\n\t\tTopK: claudeReq.TopK,\n\t\tStream: claudeReq.Stream,\n\t\tTools: claudeReq.Tools,\n\t}\n\n\tc.Set(ctxkey.RequestModel, request.Model)\n\tc.Set(ctxkey.ConvertedRequest, req)\n\treturn req, nil\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tif meta.IsStream {\n\t\terr, usage = anthropic.StreamHandler(c, resp)\n\t} else {\n\t\terr, usage = anthropic.Handler(c, resp, meta.PromptTokens, meta.ActualModelName)\n\t}\n\treturn\n}\n"
},
{
"path": "relay/adaptor/vertexai/claude/model.go",
"content": "package vertexai\n\nimport \"github.com/songquanpeng/one-api/relay/adaptor/anthropic\"\n\ntype Request struct {\n\t// AnthropicVersion must be \"vertex-2023-10-16\"\n\tAnthropicVersion string `json:\"anthropic_version\"`\n\t// Model string `json:\"model\"`\n\tMessages []anthropic.Message `json:\"messages\"`\n\tSystem string `json:\"system,omitempty\"`\n\tMaxTokens int `json:\"max_tokens,omitempty\"`\n\tStopSequences []string `json:\"stop_sequences,omitempty\"`\n\tStream bool `json:\"stream,omitempty\"`\n\tTemperature *float64 `json:\"temperature,omitempty\"`\n\tTopP *float64 `json:\"top_p,omitempty\"`\n\tTopK int `json:\"top_k,omitempty\"`\n\tTools []anthropic.Tool `json:\"tools,omitempty\"`\n\tToolChoice any `json:\"tool_choice,omitempty\"`\n}\n"
},
{
"path": "relay/adaptor/vertexai/gemini/adapter.go",
"content": "package vertexai\n\nimport (\n\t\"net/http\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/pkg/errors\"\n\t\"github.com/songquanpeng/one-api/common/ctxkey\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/gemini\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/relaymode\"\n\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\nvar ModelList = []string{\n\t\"gemini-pro\", \"gemini-pro-vision\",\n\t\"gemini-exp-1206\",\n\t\"gemini-1.5-pro-001\", \"gemini-1.5-pro-002\",\n\t\"gemini-1.5-flash-001\", \"gemini-1.5-flash-002\",\n\t\"gemini-2.0-flash-exp\", \"gemini-2.0-flash-001\",\n\t\"gemini-2.0-flash-lite-preview-02-05\",\n\t\"gemini-2.0-flash-thinking-exp-01-21\",\n}\n\ntype Adaptor struct {\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\n\tgeminiRequest := gemini.ConvertRequest(*request)\n\tc.Set(ctxkey.RequestModel, request.Model)\n\tc.Set(ctxkey.ConvertedRequest, geminiRequest)\n\treturn geminiRequest, nil\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tif meta.IsStream {\n\t\tvar responseText string\n\t\terr, responseText = gemini.StreamHandler(c, resp)\n\t\tusage = openai.ResponseText2Usage(responseText, meta.ActualModelName, meta.PromptTokens)\n\t} else {\n\t\tswitch meta.Mode {\n\t\tcase relaymode.Embeddings:\n\t\t\terr, usage = gemini.EmbeddingHandler(c, resp)\n\t\tdefault:\n\t\t\terr, usage = gemini.Handler(c, resp, meta.PromptTokens, meta.ActualModelName)\n\t\t}\n\t}\n\treturn\n}\n"
},
{
"path": "relay/adaptor/vertexai/registry.go",
"content": "package vertexai\n\nimport (\n\t\"net/http\"\n\n\t\"github.com/gin-gonic/gin\"\n\tclaude \"github.com/songquanpeng/one-api/relay/adaptor/vertexai/claude\"\n\tgemini \"github.com/songquanpeng/one-api/relay/adaptor/vertexai/gemini\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\ntype VertexAIModelType int\n\nconst (\n\tVerterAIClaude VertexAIModelType = iota + 1\n\tVerterAIGemini\n)\n\nvar modelMapping = map[string]VertexAIModelType{}\nvar modelList = []string{}\n\nfunc init() {\n\tmodelList = append(modelList, claude.ModelList...)\n\tfor _, model := range claude.ModelList {\n\t\tmodelMapping[model] = VerterAIClaude\n\t}\n\n\tmodelList = append(modelList, gemini.ModelList...)\n\tfor _, model := range gemini.ModelList {\n\t\tmodelMapping[model] = VerterAIGemini\n\t}\n}\n\ntype innerAIAdapter interface {\n\tConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error)\n\tDoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode)\n}\n\nfunc GetAdaptor(model string) innerAIAdapter {\n\tadaptorType := modelMapping[model]\n\tswitch adaptorType {\n\tcase VerterAIClaude:\n\t\treturn &claude.Adaptor{}\n\tcase VerterAIGemini:\n\t\treturn &gemini.Adaptor{}\n\tdefault:\n\t\treturn nil\n\t}\n}\n"
},
{
"path": "relay/adaptor/vertexai/token.go",
"content": "package vertexai\n\nimport (\n\t\"context\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"time\"\n\n\tcredentials \"cloud.google.com/go/iam/credentials/apiv1\"\n\t\"cloud.google.com/go/iam/credentials/apiv1/credentialspb\"\n\t\"github.com/patrickmn/go-cache\"\n\t\"google.golang.org/api/option\"\n)\n\ntype ApplicationDefaultCredentials struct {\n\tType string `json:\"type\"`\n\tProjectID string `json:\"project_id\"`\n\tPrivateKeyID string `json:\"private_key_id\"`\n\tPrivateKey string `json:\"private_key\"`\n\tClientEmail string `json:\"client_email\"`\n\tClientID string `json:\"client_id\"`\n\tAuthURI string `json:\"auth_uri\"`\n\tTokenURI string `json:\"token_uri\"`\n\tAuthProviderX509CertURL string `json:\"auth_provider_x509_cert_url\"`\n\tClientX509CertURL string `json:\"client_x509_cert_url\"`\n\tUniverseDomain string `json:\"universe_domain\"`\n}\n\nvar Cache = cache.New(50*time.Minute, 55*time.Minute)\n\nconst defaultScope = \"https://www.googleapis.com/auth/cloud-platform\"\n\nfunc getToken(ctx context.Context, channelId int, adcJson string) (string, error) {\n\tcacheKey := fmt.Sprintf(\"vertexai-token-%d\", channelId)\n\tif token, found := Cache.Get(cacheKey); found {\n\t\treturn token.(string), nil\n\t}\n\tadc := &ApplicationDefaultCredentials{}\n\tif err := json.Unmarshal([]byte(adcJson), adc); err != nil {\n\t\treturn \"\", fmt.Errorf(\"Failed to decode credentials file: %w\", err)\n\t}\n\n\tc, err := credentials.NewIamCredentialsClient(ctx, option.WithCredentialsJSON([]byte(adcJson)))\n\tif err != nil {\n\t\treturn \"\", fmt.Errorf(\"Failed to create client: %w\", err)\n\t}\n\tdefer c.Close()\n\n\treq := &credentialspb.GenerateAccessTokenRequest{\n\t\t// See https://pkg.go.dev/cloud.google.com/go/iam/credentials/apiv1/credentialspb#GenerateAccessTokenRequest.\n\t\tName: fmt.Sprintf(\"projects/-/serviceAccounts/%s\", adc.ClientEmail),\n\t\tScope: []string{defaultScope},\n\t}\n\tresp, err := c.GenerateAccessToken(ctx, req)\n\tif err != nil {\n\t\treturn \"\", fmt.Errorf(\"Failed to generate access token: %w\", err)\n\t}\n\t_ = resp\n\n\tCache.Set(cacheKey, resp.AccessToken, cache.DefaultExpiration)\n\treturn resp.AccessToken, nil\n}\n"
},
{
"path": "relay/adaptor/xai/constants.go",
"content": "package xai\n\n//https://console.x.ai/\n\nvar ModelList = []string{\n\t\"grok-2\",\n\t\"grok-vision-beta\",\n\t\"grok-2-vision-1212\",\n\t\"grok-2-vision\",\n\t\"grok-2-vision-latest\",\n\t\"grok-2-1212\",\n\t\"grok-2-latest\",\n\t\"grok-beta\",\n}\n"
},
{
"path": "relay/adaptor/xunfei/adaptor.go",
"content": "package xunfei\n\nimport (\n\t\"errors\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n)\n\ntype Adaptor struct {\n\trequest *model.GeneralOpenAIRequest\n\tmeta *meta.Meta\n}\n\nfunc (a *Adaptor) Init(meta *meta.Meta) {\n\ta.meta = meta\n}\n\nfunc (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {\n\treturn \"\", nil\n}\n\nfunc (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {\n\tadaptor.SetupCommonRequestHeader(c, req, meta)\n\t// check DoResponse for auth part\n\treturn nil\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\ta.request = request\n\treturn nil, nil\n}\n\nfunc (a *Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\treturn request, nil\n}\n\nfunc (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {\n\t// xunfei's request is not http request, so we don't need to do anything here\n\tdummyResp := &http.Response{}\n\tdummyResp.StatusCode = http.StatusOK\n\treturn dummyResp, nil\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tsplits := strings.Split(meta.APIKey, \"|\")\n\tif len(splits) != 3 {\n\t\treturn nil, openai.ErrorWrapper(errors.New(\"invalid auth\"), \"invalid_auth\", http.StatusBadRequest)\n\t}\n\tif a.request == nil {\n\t\treturn nil, openai.ErrorWrapper(errors.New(\"request is nil\"), \"request_is_nil\", http.StatusBadRequest)\n\t}\n\tversion := parseAPIVersionByModelName(meta.ActualModelName)\n\tif version == \"\" {\n\t\tversion = a.meta.Config.APIVersion\n\t}\n\tif version == \"\" {\n\t\tversion = \"v1.1\"\n\t}\n\ta.meta.Config.APIVersion = version\n\tif meta.IsStream {\n\t\terr, usage = StreamHandler(c, meta, *a.request, splits[0], splits[1], splits[2])\n\t} else {\n\t\terr, usage = Handler(c, meta, *a.request, splits[0], splits[1], splits[2])\n\t}\n\treturn\n}\n\nfunc (a *Adaptor) GetModelList() []string {\n\treturn ModelList\n}\n\nfunc (a *Adaptor) GetChannelName() string {\n\treturn \"xunfei\"\n}\n"
},
{
"path": "relay/adaptor/xunfei/constants.go",
"content": "package xunfei\n\nvar ModelList = []string{\n\t\"Spark-Lite\",\n\t\"Spark-Pro\",\n\t\"Spark-Pro-128K\",\n\t\"Spark-Max\",\n\t\"Spark-Max-32K\",\n\t\"Spark-4.0-Ultra\",\n}\n"
},
{
"path": "relay/adaptor/xunfei/domain.go",
"content": "package xunfei\n\nimport (\n\t\"fmt\"\n\t\"strings\"\n)\n\n// https://www.xfyun.cn/doc/spark/Web.html#_1-%E6%8E%A5%E5%8F%A3%E8%AF%B4%E6%98%8E\n\n//Spark4.0 Ultra 请求地址，对应的domain参数为4.0Ultra：\n//\n//wss://spark-api.xf-yun.com/v4.0/chat\n//Spark Max-32K请求地址，对应的domain参数为max-32k\n//\n//wss://spark-api.xf-yun.com/chat/max-32k\n//Spark Max请求地址，对应的domain参数为generalv3.5\n//\n//wss://spark-api.xf-yun.com/v3.5/chat\n//Spark Pro-128K请求地址，对应的domain参数为pro-128k：\n//\n// wss://spark-api.xf-yun.com/chat/pro-128k\n//Spark Pro请求地址，对应的domain参数为generalv3：\n//\n//wss://spark-api.xf-yun.com/v3.1/chat\n//Spark Lite请求地址，对应的domain参数为lite：\n//\n//wss://spark-api.xf-yun.com/v1.1/chat\n\n// Lite、Pro、Pro-128K、Max、Max-32K和4.0 Ultra\n\nfunc parseAPIVersionByModelName(modelName string) string {\n\tapiVersion := modelName2APIVersion(modelName)\n\tif apiVersion != \"\" {\n\t\treturn apiVersion\n\t}\n\n\tindex := strings.IndexAny(modelName, \"-\")\n\tif index != -1 {\n\t\treturn modelName[index+1:]\n\t}\n\treturn \"\"\n}\n\nfunc modelName2APIVersion(modelName string) string {\n\tswitch modelName {\n\tcase \"Spark-Lite\":\n\t\treturn \"v1.1\"\n\tcase \"Spark-Pro\":\n\t\treturn \"v3.1\"\n\tcase \"Spark-Pro-128K\":\n\t\treturn \"v3.1-128K\"\n\tcase \"Spark-Max\":\n\t\treturn \"v3.5\"\n\tcase \"Spark-Max-32K\":\n\t\treturn \"v3.5-32K\"\n\tcase \"Spark-4.0-Ultra\":\n\t\treturn \"v4.0\"\n\t}\n\treturn \"\"\n}\n\n// https://www.xfyun.cn/doc/spark/Web.html#_1-%E6%8E%A5%E5%8F%A3%E8%AF%B4%E6%98%8E\nfunc apiVersion2domain(apiVersion string) string {\n\tswitch apiVersion {\n\tcase \"v1.1\":\n\t\treturn \"lite\"\n\tcase \"v2.1\":\n\t\treturn \"generalv2\"\n\tcase \"v3.1\":\n\t\treturn \"generalv3\"\n\tcase \"v3.1-128K\":\n\t\treturn \"pro-128k\"\n\tcase \"v3.5\":\n\t\treturn \"generalv3.5\"\n\tcase \"v3.5-32K\":\n\t\treturn \"max-32k\"\n\tcase \"v4.0\":\n\t\treturn \"4.0Ultra\"\n\t}\n\treturn \"general\" + apiVersion\n}\n\nfunc getXunfeiAuthUrl(apiVersion string, apiKey string, apiSecret string) (string, string) {\n\tvar authUrl string\n\tdomain := apiVersion2domain(apiVersion)\n\tswitch apiVersion {\n\tcase \"v3.1-128K\":\n\t\tauthUrl = buildXunfeiAuthUrl(fmt.Sprintf(\"wss://spark-api.xf-yun.com/chat/pro-128k\"), apiKey, apiSecret)\n\t\tbreak\n\tcase \"v3.5-32K\":\n\t\tauthUrl = buildXunfeiAuthUrl(fmt.Sprintf(\"wss://spark-api.xf-yun.com/chat/max-32k\"), apiKey, apiSecret)\n\t\tbreak\n\tdefault:\n\t\tauthUrl = buildXunfeiAuthUrl(fmt.Sprintf(\"wss://spark-api.xf-yun.com/%s/chat\", apiVersion), apiKey, apiSecret)\n\t}\n\treturn domain, authUrl\n}\n"
},
{
"path": "relay/adaptor/xunfei/main.go",
"content": "package xunfei\n\nimport (\n\t\"crypto/hmac\"\n\t\"crypto/sha256\"\n\t\"encoding/base64\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"io\"\n\t\"net/http\"\n\t\"net/url\"\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/gorilla/websocket\"\n\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/common/random\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/constant\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\n// https://console.xfyun.cn/services/cbm\n// https://www.xfyun.cn/doc/spark/Web.html\n\nfunc requestOpenAI2Xunfei(request model.GeneralOpenAIRequest, xunfeiAppId string, domain string) *ChatRequest {\n\tmessages := make([]Message, 0, len(request.Messages))\n\tfor _, message := range request.Messages {\n\t\tmessages = append(messages, Message{\n\t\t\tRole: message.Role,\n\t\t\tContent: message.StringContent(),\n\t\t})\n\t}\n\txunfeiRequest := ChatRequest{}\n\txunfeiRequest.Header.AppId = xunfeiAppId\n\txunfeiRequest.Parameter.Chat.Domain = domain\n\txunfeiRequest.Parameter.Chat.Temperature = request.Temperature\n\txunfeiRequest.Parameter.Chat.TopK = request.N\n\txunfeiRequest.Parameter.Chat.MaxTokens = request.MaxTokens\n\txunfeiRequest.Payload.Message.Text = messages\n\n\tif strings.HasPrefix(domain, \"generalv3\") || domain == \"4.0Ultra\" {\n\t\tfunctions := make([]model.Function, len(request.Tools))\n\t\tfor i, tool := range request.Tools {\n\t\t\tfunctions[i] = tool.Function\n\t\t}\n\t\txunfeiRequest.Payload.Functions = &Functions{\n\t\t\tText: functions,\n\t\t}\n\t}\n\n\treturn &xunfeiRequest\n}\n\nfunc getToolCalls(response *ChatResponse) []model.Tool {\n\tvar toolCalls []model.Tool\n\tif len(response.Payload.Choices.Text) == 0 {\n\t\treturn toolCalls\n\t}\n\titem := response.Payload.Choices.Text[0]\n\tif item.FunctionCall == nil {\n\t\treturn toolCalls\n\t}\n\ttoolCall := model.Tool{\n\t\tId: fmt.Sprintf(\"call_%s\", random.GetUUID()),\n\t\tType: \"function\",\n\t\tFunction: *item.FunctionCall,\n\t}\n\ttoolCalls = append(toolCalls, toolCall)\n\treturn toolCalls\n}\n\nfunc responseXunfei2OpenAI(response *ChatResponse) *openai.TextResponse {\n\tif len(response.Payload.Choices.Text) == 0 {\n\t\tresponse.Payload.Choices.Text = []ChatResponseTextItem{\n\t\t\t{\n\t\t\t\tContent: \"\",\n\t\t\t},\n\t\t}\n\t}\n\tchoice := openai.TextResponseChoice{\n\t\tIndex: 0,\n\t\tMessage: model.Message{\n\t\t\tRole: \"assistant\",\n\t\t\tContent: response.Payload.Choices.Text[0].Content,\n\t\t\tToolCalls: getToolCalls(response),\n\t\t},\n\t\tFinishReason: constant.StopFinishReason,\n\t}\n\tfullTextResponse := openai.TextResponse{\n\t\tId: fmt.Sprintf(\"chatcmpl-%s\", random.GetUUID()),\n\t\tObject: \"chat.completion\",\n\t\tCreated: helper.GetTimestamp(),\n\t\tChoices: []openai.TextResponseChoice{choice},\n\t\tUsage: response.Payload.Usage.Text,\n\t}\n\treturn &fullTextResponse\n}\n\nfunc streamResponseXunfei2OpenAI(xunfeiResponse *ChatResponse) *openai.ChatCompletionsStreamResponse {\n\tif len(xunfeiResponse.Payload.Choices.Text) == 0 {\n\t\txunfeiResponse.Payload.Choices.Text = []ChatResponseTextItem{\n\t\t\t{\n\t\t\t\tContent: \"\",\n\t\t\t},\n\t\t}\n\t}\n\tvar choice openai.ChatCompletionsStreamResponseChoice\n\tchoice.Delta.Content = xunfeiResponse.Payload.Choices.Text[0].Content\n\tchoice.Delta.ToolCalls = getToolCalls(xunfeiResponse)\n\tif xunfeiResponse.Payload.Choices.Status == 2 {\n\t\tchoice.FinishReason = &constant.StopFinishReason\n\t}\n\tresponse := openai.ChatCompletionsStreamResponse{\n\t\tId: fmt.Sprintf(\"chatcmpl-%s\", random.GetUUID()),\n\t\tObject: \"chat.completion.chunk\",\n\t\tCreated: helper.GetTimestamp(),\n\t\tModel: \"SparkDesk\",\n\t\tChoices: []openai.ChatCompletionsStreamResponseChoice{choice},\n\t}\n\treturn &response\n}\n\nfunc buildXunfeiAuthUrl(hostUrl string, apiKey, apiSecret string) string {\n\tHmacWithShaToBase64 := func(algorithm, data, key string) string {\n\t\tmac := hmac.New(sha256.New, []byte(key))\n\t\tmac.Write([]byte(data))\n\t\tencodeData := mac.Sum(nil)\n\t\treturn base64.StdEncoding.EncodeToString(encodeData)\n\t}\n\tul, err := url.Parse(hostUrl)\n\tif err != nil {\n\t\tfmt.Println(err)\n\t}\n\tdate := time.Now().UTC().Format(time.RFC1123)\n\tsignString := []string{\"host: \" + ul.Host, \"date: \" + date, \"GET \" + ul.Path + \" HTTP/1.1\"}\n\tsign := strings.Join(signString, \"\\n\")\n\tsha := HmacWithShaToBase64(\"hmac-sha256\", sign, apiSecret)\n\tauthUrl := fmt.Sprintf(\"hmac username=\\\"%s\\\", algorithm=\\\"%s\\\", headers=\\\"%s\\\", signature=\\\"%s\\\"\", apiKey,\n\t\t\"hmac-sha256\", \"host date request-line\", sha)\n\tauthorization := base64.StdEncoding.EncodeToString([]byte(authUrl))\n\tv := url.Values{}\n\tv.Add(\"host\", ul.Host)\n\tv.Add(\"date\", date)\n\tv.Add(\"authorization\", authorization)\n\tcallUrl := hostUrl + \"?\" + v.Encode()\n\treturn callUrl\n}\n\nfunc StreamHandler(c *gin.Context, meta *meta.Meta, textRequest model.GeneralOpenAIRequest, appId string, apiSecret string, apiKey string) (*model.ErrorWithStatusCode, *model.Usage) {\n\tdomain, authUrl := getXunfeiAuthUrl(meta.Config.APIVersion, apiKey, apiSecret)\n\tdataChan, stopChan, err := xunfeiMakeRequest(textRequest, domain, authUrl, appId)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"xunfei_request_failed\", http.StatusInternalServerError), nil\n\t}\n\tcommon.SetEventStreamHeaders(c)\n\tvar usage model.Usage\n\tc.Stream(func(w io.Writer) bool {\n\t\tselect {\n\t\tcase xunfeiResponse := <-dataChan:\n\t\t\tusage.PromptTokens += xunfeiResponse.Payload.Usage.Text.PromptTokens\n\t\t\tusage.CompletionTokens += xunfeiResponse.Payload.Usage.Text.CompletionTokens\n\t\t\tusage.TotalTokens += xunfeiResponse.Payload.Usage.Text.TotalTokens\n\t\t\tresponse := streamResponseXunfei2OpenAI(&xunfeiResponse)\n\t\t\tjsonResponse, err := json.Marshal(response)\n\t\t\tif err != nil {\n\t\t\t\tlogger.SysError(\"error marshalling stream response: \" + err.Error())\n\t\t\t\treturn true\n\t\t\t}\n\t\t\tc.Render(-1, common.CustomEvent{Data: \"data: \" + string(jsonResponse)})\n\t\t\treturn true\n\t\tcase <-stopChan:\n\t\t\tc.Render(-1, common.CustomEvent{Data: \"data: [DONE]\"})\n\t\t\treturn false\n\t\t}\n\t})\n\treturn nil, &usage\n}\n\nfunc Handler(c *gin.Context, meta *meta.Meta, textRequest model.GeneralOpenAIRequest, appId string, apiSecret string, apiKey string) (*model.ErrorWithStatusCode, *model.Usage) {\n\tdomain, authUrl := getXunfeiAuthUrl(meta.Config.APIVersion, apiKey, apiSecret)\n\tdataChan, stopChan, err := xunfeiMakeRequest(textRequest, domain, authUrl, appId)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"xunfei_request_failed\", http.StatusInternalServerError), nil\n\t}\n\tvar usage model.Usage\n\tvar content string\n\tvar xunfeiResponse ChatResponse\n\tstop := false\n\tfor !stop {\n\t\tselect {\n\t\tcase xunfeiResponse = <-dataChan:\n\t\t\tif len(xunfeiResponse.Payload.Choices.Text) == 0 {\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tcontent += xunfeiResponse.Payload.Choices.Text[0].Content\n\t\t\tusage.PromptTokens += xunfeiResponse.Payload.Usage.Text.PromptTokens\n\t\t\tusage.CompletionTokens += xunfeiResponse.Payload.Usage.Text.CompletionTokens\n\t\t\tusage.TotalTokens += xunfeiResponse.Payload.Usage.Text.TotalTokens\n\t\tcase stop = <-stopChan:\n\t\t}\n\t}\n\tif len(xunfeiResponse.Payload.Choices.Text) == 0 {\n\t\treturn openai.ErrorWrapper(errors.New(\"xunfei empty response detected\"), \"xunfei_empty_response_detected\", http.StatusInternalServerError), nil\n\t}\n\txunfeiResponse.Payload.Choices.Text[0].Content = content\n\n\tresponse := responseXunfei2OpenAI(&xunfeiResponse)\n\tjsonResponse, err := json.Marshal(response)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tc.Writer.Header().Set(\"Content-Type\", \"application/json\")\n\t_, _ = c.Writer.Write(jsonResponse)\n\treturn nil, &usage\n}\n\nfunc xunfeiMakeRequest(textRequest model.GeneralOpenAIRequest, domain, authUrl, appId string) (chan ChatResponse, chan bool, error) {\n\td := websocket.Dialer{\n\t\tHandshakeTimeout: 5 * time.Second,\n\t}\n\tconn, resp, err := d.Dial(authUrl, nil)\n\tif err != nil || resp.StatusCode != 101 {\n\t\treturn nil, nil, err\n\t}\n\tdata := requestOpenAI2Xunfei(textRequest, appId, domain)\n\terr = conn.WriteJSON(data)\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\t_, msg, err := conn.ReadMessage()\n\tif err != nil {\n\t\treturn nil, nil, err\n\t}\n\n\tdataChan := make(chan ChatResponse)\n\tstopChan := make(chan bool)\n\tgo func() {\n\t\tfor {\n\t\t\tif msg == nil {\n\t\t\t\t_, msg, err = conn.ReadMessage()\n\t\t\t\tif err != nil {\n\t\t\t\t\tlogger.SysError(\"error reading stream response: \" + err.Error())\n\t\t\t\t\tbreak\n\t\t\t\t}\n\t\t\t}\n\t\t\tvar response ChatResponse\n\t\t\terr = json.Unmarshal(msg, &response)\n\t\t\tif err != nil {\n\t\t\t\tlogger.SysError(\"error unmarshalling stream response: \" + err.Error())\n\t\t\t\tbreak\n\t\t\t}\n\t\t\tmsg = nil\n\t\t\tdataChan <- response\n\t\t\tif response.Payload.Choices.Status == 2 {\n\t\t\t\terr := conn.Close()\n\t\t\t\tif err != nil {\n\t\t\t\t\tlogger.SysError(\"error closing websocket connection: \" + err.Error())\n\t\t\t\t}\n\t\t\t\tbreak\n\t\t\t}\n\t\t}\n\t\tstopChan <- true\n\t}()\n\n\treturn dataChan, stopChan, nil\n}\n"
},
{
"path": "relay/adaptor/xunfei/model.go",
"content": "package xunfei\n\nimport (\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\ntype Message struct {\n\tRole string `json:\"role\"`\n\tContent string `json:\"content\"`\n}\n\ntype Functions struct {\n\tText []model.Function `json:\"text,omitempty\"`\n}\n\ntype ChatRequest struct {\n\tHeader struct {\n\t\tAppId string `json:\"app_id\"`\n\t} `json:\"header\"`\n\tParameter struct {\n\t\tChat struct {\n\t\t\tDomain string `json:\"domain,omitempty\"`\n\t\t\tTemperature *float64 `json:\"temperature,omitempty\"`\n\t\t\tTopK int `json:\"top_k,omitempty\"`\n\t\t\tMaxTokens int `json:\"max_tokens,omitempty\"`\n\t\t\tAuditing bool `json:\"auditing,omitempty\"`\n\t\t} `json:\"chat\"`\n\t} `json:\"parameter\"`\n\tPayload struct {\n\t\tMessage struct {\n\t\t\tText []Message `json:\"text\"`\n\t\t} `json:\"message\"`\n\t\tFunctions *Functions `json:\"functions,omitempty\"`\n\t} `json:\"payload\"`\n}\n\ntype ChatResponseTextItem struct {\n\tContent string `json:\"content\"`\n\tRole string `json:\"role\"`\n\tIndex int `json:\"index\"`\n\tContentType string `json:\"content_type\"`\n\tFunctionCall *model.Function `json:\"function_call\"`\n}\n\ntype ChatResponse struct {\n\tHeader struct {\n\t\tCode int `json:\"code\"`\n\t\tMessage string `json:\"message\"`\n\t\tSid string `json:\"sid\"`\n\t\tStatus int `json:\"status\"`\n\t} `json:\"header\"`\n\tPayload struct {\n\t\tChoices struct {\n\t\t\tStatus int `json:\"status\"`\n\t\t\tSeq int `json:\"seq\"`\n\t\t\tText []ChatResponseTextItem `json:\"text\"`\n\t\t} `json:\"choices\"`\n\t\tUsage struct {\n\t\t\t//Text struct {\n\t\t\t//\tQuestionTokens string `json:\"question_tokens\"`\n\t\t\t//\tPromptTokens string `json:\"prompt_tokens\"`\n\t\t\t//\tCompletionTokens string `json:\"completion_tokens\"`\n\t\t\t//\tTotalTokens string `json:\"total_tokens\"`\n\t\t\t//} `json:\"text\"`\n\t\t\tText model.Usage `json:\"text\"`\n\t\t} `json:\"usage\"`\n\t} `json:\"payload\"`\n}\n"
},
{
"path": "relay/adaptor/xunfeiv2/constants.go",
"content": "package xunfeiv2\n\n// https://www.xfyun.cn/doc/spark/HTTP%E8%B0%83%E7%94%A8%E6%96%87%E6%A1%A3.html#_3-%E8%AF%B7%E6%B1%82%E8%AF%B4%E6%98%8E\n\nvar ModelList = []string{\n\t\"lite\",\n\t\"generalv3\",\n\t\"pro-128k\",\n\t\"generalv3.5\",\n\t\"max-32k\",\n\t\"4.0Ultra\",\n}\n"
},
{
"path": "relay/adaptor/zhipu/adaptor.go",
"content": "package zhipu\n\nimport (\n\t\"errors\"\n\t\"fmt\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"github.com/songquanpeng/one-api/relay/relaymode\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n)\n\ntype Adaptor struct {\n\tAPIVersion string\n}\n\nfunc (a *Adaptor) Init(meta *meta.Meta) {\n\n}\n\nfunc (a *Adaptor) SetVersionByModeName(modelName string) {\n\tif strings.HasPrefix(modelName, \"glm-\") {\n\t\ta.APIVersion = \"v4\"\n\t} else {\n\t\ta.APIVersion = \"v3\"\n\t}\n}\n\nfunc (a *Adaptor) GetRequestURL(meta *meta.Meta) (string, error) {\n\tswitch meta.Mode {\n\tcase relaymode.ImagesGenerations:\n\t\treturn fmt.Sprintf(\"%s/api/paas/v4/images/generations\", meta.BaseURL), nil\n\tcase relaymode.Embeddings:\n\t\treturn fmt.Sprintf(\"%s/api/paas/v4/embeddings\", meta.BaseURL), nil\n\t}\n\ta.SetVersionByModeName(meta.ActualModelName)\n\tif a.APIVersion == \"v4\" {\n\t\treturn fmt.Sprintf(\"%s/api/paas/v4/chat/completions\", meta.BaseURL), nil\n\t}\n\tmethod := \"invoke\"\n\tif meta.IsStream {\n\t\tmethod = \"sse-invoke\"\n\t}\n\treturn fmt.Sprintf(\"%s/api/paas/v3/model-api/%s/%s\", meta.BaseURL, meta.ActualModelName, method), nil\n}\n\nfunc (a *Adaptor) SetupRequestHeader(c *gin.Context, req *http.Request, meta *meta.Meta) error {\n\tadaptor.SetupCommonRequestHeader(c, req, meta)\n\ttoken := GetToken(meta.APIKey)\n\treq.Header.Set(\"Authorization\", token)\n\treturn nil\n}\n\nfunc (a *Adaptor) ConvertRequest(c *gin.Context, relayMode int, request *model.GeneralOpenAIRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\tswitch relayMode {\n\tcase relaymode.Embeddings:\n\t\tbaiduEmbeddingRequest, err := ConvertEmbeddingRequest(*request)\n\t\treturn baiduEmbeddingRequest, err\n\tdefault:\n\t\t// TopP [0.0, 1.0]\n\t\trequest.TopP = helper.Float64PtrMax(request.TopP, 1)\n\t\trequest.TopP = helper.Float64PtrMin(request.TopP, 0)\n\n\t\t// Temperature [0.0, 1.0]\n\t\trequest.Temperature = helper.Float64PtrMax(request.Temperature, 1)\n\t\trequest.Temperature = helper.Float64PtrMin(request.Temperature, 0)\n\t\ta.SetVersionByModeName(request.Model)\n\t\tif a.APIVersion == \"v4\" {\n\t\t\treturn request, nil\n\t\t}\n\t\treturn ConvertRequest(*request), nil\n\t}\n}\n\nfunc (a *Adaptor) ConvertImageRequest(request *model.ImageRequest) (any, error) {\n\tif request == nil {\n\t\treturn nil, errors.New(\"request is nil\")\n\t}\n\tnewRequest := ImageRequest{\n\t\tModel: request.Model,\n\t\tPrompt: request.Prompt,\n\t\tUserId: request.User,\n\t}\n\treturn newRequest, nil\n}\n\nfunc (a *Adaptor) DoRequest(c *gin.Context, meta *meta.Meta, requestBody io.Reader) (*http.Response, error) {\n\treturn adaptor.DoRequestHelper(a, c, meta, requestBody)\n}\n\nfunc (a *Adaptor) DoResponseV4(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tif meta.IsStream {\n\t\terr, _, usage = openai.StreamHandler(c, resp, meta.Mode)\n\t} else {\n\t\terr, usage = openai.Handler(c, resp, meta.PromptTokens, meta.ActualModelName)\n\t}\n\treturn\n}\n\nfunc (a *Adaptor) DoResponse(c *gin.Context, resp *http.Response, meta *meta.Meta) (usage *model.Usage, err *model.ErrorWithStatusCode) {\n\tswitch meta.Mode {\n\tcase relaymode.Embeddings:\n\t\terr, usage = EmbeddingsHandler(c, resp)\n\t\treturn\n\tcase relaymode.ImagesGenerations:\n\t\terr, usage = openai.ImageHandler(c, resp)\n\t\treturn\n\t}\n\tif a.APIVersion == \"v4\" {\n\t\treturn a.DoResponseV4(c, resp, meta)\n\t}\n\tif meta.IsStream {\n\t\terr, usage = StreamHandler(c, resp)\n\t} else {\n\t\tif meta.Mode == relaymode.Embeddings {\n\t\t\terr, usage = EmbeddingsHandler(c, resp)\n\t\t} else {\n\t\t\terr, usage = Handler(c, resp)\n\t\t}\n\t}\n\treturn\n}\n\nfunc ConvertEmbeddingRequest(request model.GeneralOpenAIRequest) (*EmbeddingRequest, error) {\n\tinputs := request.ParseInput()\n\tif len(inputs) != 1 {\n\t\treturn nil, errors.New(\"invalid input length, zhipu only support one input\")\n\t}\n\treturn &EmbeddingRequest{\n\t\tModel: request.Model,\n\t\tInput: inputs[0],\n\t}, nil\n}\n\nfunc (a *Adaptor) GetModelList() []string {\n\treturn ModelList\n}\n\nfunc (a *Adaptor) GetChannelName() string {\n\treturn \"zhipu\"\n}\n"
},
{
"path": "relay/adaptor/zhipu/constants.go",
"content": "package zhipu\n\n// https://open.bigmodel.cn/pricing\n\nvar ModelList = []string{\n\t\"glm-zero-preview\", \"glm-4-plus\", \"glm-4-0520\", \"glm-4-airx\",\n\t\"glm-4-air\", \"glm-4-long\", \"glm-4-flashx\", \"glm-4-flash\",\n\t\"glm-4\", \"glm-3-turbo\",\n\t\"glm-4v-plus\", \"glm-4v\", \"glm-4v-flash\",\n\t\"cogview-3-plus\", \"cogview-3\", \"cogview-3-flash\",\n\t\"cogviewx\", \"cogviewx-flash\",\n\t\"charglm-4\", \"emohaa\", \"codegeex-4\",\n\t\"embedding-2\", \"embedding-3\",\n}\n"
},
{
"path": "relay/adaptor/zhipu/main.go",
"content": "package zhipu\n\nimport (\n\t\"bufio\"\n\t\"encoding/json\"\n\t\"github.com/songquanpeng/one-api/common/render\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n\t\"sync\"\n\t\"time\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/golang-jwt/jwt\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/constant\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\n// https://open.bigmodel.cn/doc/api#chatglm_std\n// chatglm_std, chatglm_lite\n// https://open.bigmodel.cn/api/paas/v3/model-api/chatglm_std/invoke\n// https://open.bigmodel.cn/api/paas/v3/model-api/chatglm_std/sse-invoke\n\nvar zhipuTokens sync.Map\nvar expSeconds int64 = 24 * 3600\n\nfunc GetToken(apikey string) string {\n\tdata, ok := zhipuTokens.Load(apikey)\n\tif ok {\n\t\ttokenData := data.(tokenData)\n\t\tif time.Now().Before(tokenData.ExpiryTime) {\n\t\t\treturn tokenData.Token\n\t\t}\n\t}\n\n\tsplit := strings.Split(apikey, \".\")\n\tif len(split) != 2 {\n\t\tlogger.SysError(\"invalid zhipu key: \" + apikey)\n\t\treturn \"\"\n\t}\n\n\tid := split[0]\n\tsecret := split[1]\n\n\texpMillis := time.Now().Add(time.Duration(expSeconds)*time.Second).UnixNano() / 1e6\n\texpiryTime := time.Now().Add(time.Duration(expSeconds) * time.Second)\n\n\ttimestamp := time.Now().UnixNano() / 1e6\n\n\tpayload := jwt.MapClaims{\n\t\t\"api_key\": id,\n\t\t\"exp\": expMillis,\n\t\t\"timestamp\": timestamp,\n\t}\n\n\ttoken := jwt.NewWithClaims(jwt.SigningMethodHS256, payload)\n\n\ttoken.Header[\"alg\"] = \"HS256\"\n\ttoken.Header[\"sign_type\"] = \"SIGN\"\n\n\ttokenString, err := token.SignedString([]byte(secret))\n\tif err != nil {\n\t\treturn \"\"\n\t}\n\n\tzhipuTokens.Store(apikey, tokenData{\n\t\tToken: tokenString,\n\t\tExpiryTime: expiryTime,\n\t})\n\n\treturn tokenString\n}\n\nfunc ConvertRequest(request model.GeneralOpenAIRequest) *Request {\n\tmessages := make([]Message, 0, len(request.Messages))\n\tfor _, message := range request.Messages {\n\t\tmessages = append(messages, Message{\n\t\t\tRole: message.Role,\n\t\t\tContent: message.StringContent(),\n\t\t})\n\t}\n\treturn &Request{\n\t\tPrompt: messages,\n\t\tTemperature: request.Temperature,\n\t\tTopP: request.TopP,\n\t\tIncremental: false,\n\t}\n}\n\nfunc responseZhipu2OpenAI(response *Response) *openai.TextResponse {\n\tfullTextResponse := openai.TextResponse{\n\t\tId: response.Data.TaskId,\n\t\tObject: \"chat.completion\",\n\t\tCreated: helper.GetTimestamp(),\n\t\tChoices: make([]openai.TextResponseChoice, 0, len(response.Data.Choices)),\n\t\tUsage: response.Data.Usage,\n\t}\n\tfor i, choice := range response.Data.Choices {\n\t\topenaiChoice := openai.TextResponseChoice{\n\t\t\tIndex: i,\n\t\t\tMessage: model.Message{\n\t\t\t\tRole: choice.Role,\n\t\t\t\tContent: strings.Trim(choice.Content, \"\\\"\"),\n\t\t\t},\n\t\t\tFinishReason: \"\",\n\t\t}\n\t\tif i == len(response.Data.Choices)-1 {\n\t\t\topenaiChoice.FinishReason = \"stop\"\n\t\t}\n\t\tfullTextResponse.Choices = append(fullTextResponse.Choices, openaiChoice)\n\t}\n\treturn &fullTextResponse\n}\n\nfunc streamResponseZhipu2OpenAI(zhipuResponse string) *openai.ChatCompletionsStreamResponse {\n\tvar choice openai.ChatCompletionsStreamResponseChoice\n\tchoice.Delta.Content = zhipuResponse\n\tresponse := openai.ChatCompletionsStreamResponse{\n\t\tObject: \"chat.completion.chunk\",\n\t\tCreated: helper.GetTimestamp(),\n\t\tModel: \"chatglm\",\n\t\tChoices: []openai.ChatCompletionsStreamResponseChoice{choice},\n\t}\n\treturn &response\n}\n\nfunc streamMetaResponseZhipu2OpenAI(zhipuResponse *StreamMetaResponse) (*openai.ChatCompletionsStreamResponse, *model.Usage) {\n\tvar choice openai.ChatCompletionsStreamResponseChoice\n\tchoice.Delta.Content = \"\"\n\tchoice.FinishReason = &constant.StopFinishReason\n\tresponse := openai.ChatCompletionsStreamResponse{\n\t\tId: zhipuResponse.RequestId,\n\t\tObject: \"chat.completion.chunk\",\n\t\tCreated: helper.GetTimestamp(),\n\t\tModel: \"chatglm\",\n\t\tChoices: []openai.ChatCompletionsStreamResponseChoice{choice},\n\t}\n\treturn &response, &zhipuResponse.Usage\n}\n\nfunc StreamHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tvar usage *model.Usage\n\tscanner := bufio.NewScanner(resp.Body)\n\tscanner.Split(func(data []byte, atEOF bool) (advance int, token []byte, err error) {\n\t\tif atEOF && len(data) == 0 {\n\t\t\treturn 0, nil, nil\n\t\t}\n\t\tif i := strings.Index(string(data), \"\\n\\n\"); i >= 0 && strings.Index(string(data), \":\") >= 0 {\n\t\t\treturn i + 2, data[0:i], nil\n\t\t}\n\t\tif atEOF {\n\t\t\treturn len(data), data, nil\n\t\t}\n\t\treturn 0, nil, nil\n\t})\n\n\tcommon.SetEventStreamHeaders(c)\n\n\tfor scanner.Scan() {\n\t\tdata := scanner.Text()\n\t\tlines := strings.Split(data, \"\\n\")\n\t\tfor i, line := range lines {\n\t\t\tif len(line) < 5 {\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tif strings.HasPrefix(line, \"data:\") {\n\t\t\t\tdataSegment := line[5:]\n\t\t\t\tif i != len(lines)-1 {\n\t\t\t\t\tdataSegment += \"\\n\"\n\t\t\t\t}\n\t\t\t\tresponse := streamResponseZhipu2OpenAI(dataSegment)\n\t\t\t\terr := render.ObjectData(c, response)\n\t\t\t\tif err != nil {\n\t\t\t\t\tlogger.SysError(\"error marshalling stream response: \" + err.Error())\n\t\t\t\t}\n\t\t\t} else if strings.HasPrefix(line, \"meta:\") {\n\t\t\t\tmetaSegment := line[5:]\n\t\t\t\tvar zhipuResponse StreamMetaResponse\n\t\t\t\terr := json.Unmarshal([]byte(metaSegment), &zhipuResponse)\n\t\t\t\tif err != nil {\n\t\t\t\t\tlogger.SysError(\"error unmarshalling stream response: \" + err.Error())\n\t\t\t\t\tcontinue\n\t\t\t\t}\n\t\t\t\tresponse, zhipuUsage := streamMetaResponseZhipu2OpenAI(&zhipuResponse)\n\t\t\t\terr = render.ObjectData(c, response)\n\t\t\t\tif err != nil {\n\t\t\t\t\tlogger.SysError(\"error marshalling stream response: \" + err.Error())\n\t\t\t\t}\n\t\t\t\tusage = zhipuUsage\n\t\t\t}\n\t\t}\n\t}\n\n\tif err := scanner.Err(); err != nil {\n\t\tlogger.SysError(\"error reading stream: \" + err.Error())\n\t}\n\n\trender.Done(c)\n\n\terr := resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\n\treturn nil, usage\n}\n\nfunc Handler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tvar zhipuResponse Response\n\tresponseBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = json.Unmarshal(responseBody, &zhipuResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tif !zhipuResponse.Success {\n\t\treturn &model.ErrorWithStatusCode{\n\t\t\tError: model.Error{\n\t\t\t\tMessage: zhipuResponse.Msg,\n\t\t\t\tType: \"zhipu_error\",\n\t\t\t\tParam: \"\",\n\t\t\t\tCode: zhipuResponse.Code,\n\t\t\t},\n\t\t\tStatusCode: resp.StatusCode,\n\t\t}, nil\n\t}\n\tfullTextResponse := responseZhipu2OpenAI(&zhipuResponse)\n\tfullTextResponse.Model = \"chatglm\"\n\tjsonResponse, err := json.Marshal(fullTextResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tc.Writer.Header().Set(\"Content-Type\", \"application/json\")\n\tc.Writer.WriteHeader(resp.StatusCode)\n\t_, err = c.Writer.Write(jsonResponse)\n\treturn nil, &fullTextResponse.Usage\n}\n\nfunc EmbeddingsHandler(c *gin.Context, resp *http.Response) (*model.ErrorWithStatusCode, *model.Usage) {\n\tvar zhipuResponse EmbeddingResponse\n\tresponseBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\terr = json.Unmarshal(responseBody, &zhipuResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"unmarshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tfullTextResponse := embeddingResponseZhipu2OpenAI(&zhipuResponse)\n\tjsonResponse, err := json.Marshal(fullTextResponse)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"marshal_response_body_failed\", http.StatusInternalServerError), nil\n\t}\n\tc.Writer.Header().Set(\"Content-Type\", \"application/json\")\n\tc.Writer.WriteHeader(resp.StatusCode)\n\t_, err = c.Writer.Write(jsonResponse)\n\treturn nil, &fullTextResponse.Usage\n}\n\nfunc embeddingResponseZhipu2OpenAI(response *EmbeddingResponse) *openai.EmbeddingResponse {\n\topenAIEmbeddingResponse := openai.EmbeddingResponse{\n\t\tObject: \"list\",\n\t\tData: make([]openai.EmbeddingResponseItem, 0, len(response.Embeddings)),\n\t\tModel: response.Model,\n\t\tUsage: model.Usage{\n\t\t\tPromptTokens: response.PromptTokens,\n\t\t\tCompletionTokens: response.CompletionTokens,\n\t\t\tTotalTokens: response.Usage.TotalTokens,\n\t\t},\n\t}\n\n\tfor _, item := range response.Embeddings {\n\t\topenAIEmbeddingResponse.Data = append(openAIEmbeddingResponse.Data, openai.EmbeddingResponseItem{\n\t\t\tObject: `embedding`,\n\t\t\tIndex: item.Index,\n\t\t\tEmbedding: item.Embedding,\n\t\t})\n\t}\n\treturn &openAIEmbeddingResponse\n}\n"
},
{
"path": "relay/adaptor/zhipu/model.go",
"content": "package zhipu\n\nimport (\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"time\"\n)\n\ntype Message struct {\n\tRole string `json:\"role\"`\n\tContent string `json:\"content\"`\n}\n\ntype Request struct {\n\tPrompt []Message `json:\"prompt\"`\n\tTemperature *float64 `json:\"temperature,omitempty\"`\n\tTopP *float64 `json:\"top_p,omitempty\"`\n\tRequestId string `json:\"request_id,omitempty\"`\n\tIncremental bool `json:\"incremental,omitempty\"`\n}\n\ntype ResponseData struct {\n\tTaskId string `json:\"task_id\"`\n\tRequestId string `json:\"request_id\"`\n\tTaskStatus string `json:\"task_status\"`\n\tChoices []Message `json:\"choices\"`\n\tmodel.Usage `json:\"usage\"`\n}\n\ntype Response struct {\n\tCode int `json:\"code\"`\n\tMsg string `json:\"msg\"`\n\tSuccess bool `json:\"success\"`\n\tData ResponseData `json:\"data\"`\n}\n\ntype StreamMetaResponse struct {\n\tRequestId string `json:\"request_id\"`\n\tTaskId string `json:\"task_id\"`\n\tTaskStatus string `json:\"task_status\"`\n\tmodel.Usage `json:\"usage\"`\n}\n\ntype tokenData struct {\n\tToken string\n\tExpiryTime time.Time\n}\n\ntype EmbeddingRequest struct {\n\tModel string `json:\"model\"`\n\tInput string `json:\"input\"`\n}\n\ntype EmbeddingResponse struct {\n\tModel string `json:\"model\"`\n\tObject string `json:\"object\"`\n\tEmbeddings []EmbeddingData `json:\"data\"`\n\tmodel.Usage `json:\"usage\"`\n}\n\ntype EmbeddingData struct {\n\tIndex int `json:\"index\"`\n\tObject string `json:\"object\"`\n\tEmbedding []float64 `json:\"embedding\"`\n}\n\ntype ImageRequest struct {\n\tModel string `json:\"model\"`\n\tPrompt string `json:\"prompt\"`\n\tUserId string `json:\"user_id,omitempty\"`\n}\n"
},
{
"path": "relay/adaptor.go",
"content": "package relay\n\nimport (\n\t\"github.com/songquanpeng/one-api/relay/adaptor\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/aiproxy\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/ali\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/anthropic\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/aws\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/baidu\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/cloudflare\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/cohere\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/coze\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/deepl\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/gemini\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/ollama\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/palm\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/proxy\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/replicate\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/tencent\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/vertexai\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/xunfei\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/zhipu\"\n\t\"github.com/songquanpeng/one-api/relay/apitype\"\n)\n\nfunc GetAdaptor(apiType int) adaptor.Adaptor {\n\tswitch apiType {\n\tcase apitype.AIProxyLibrary:\n\t\treturn &aiproxy.Adaptor{}\n\tcase apitype.Ali:\n\t\treturn &ali.Adaptor{}\n\tcase apitype.Anthropic:\n\t\treturn &anthropic.Adaptor{}\n\tcase apitype.AwsClaude:\n\t\treturn &aws.Adaptor{}\n\tcase apitype.Baidu:\n\t\treturn &baidu.Adaptor{}\n\tcase apitype.Gemini:\n\t\treturn &gemini.Adaptor{}\n\tcase apitype.OpenAI:\n\t\treturn &openai.Adaptor{}\n\tcase apitype.PaLM:\n\t\treturn &palm.Adaptor{}\n\tcase apitype.Tencent:\n\t\treturn &tencent.Adaptor{}\n\tcase apitype.Xunfei:\n\t\treturn &xunfei.Adaptor{}\n\tcase apitype.Zhipu:\n\t\treturn &zhipu.Adaptor{}\n\tcase apitype.Ollama:\n\t\treturn &ollama.Adaptor{}\n\tcase apitype.Coze:\n\t\treturn &coze.Adaptor{}\n\tcase apitype.Cohere:\n\t\treturn &cohere.Adaptor{}\n\tcase apitype.Cloudflare:\n\t\treturn &cloudflare.Adaptor{}\n\tcase apitype.DeepL:\n\t\treturn &deepl.Adaptor{}\n\tcase apitype.VertexAI:\n\t\treturn &vertexai.Adaptor{}\n\tcase apitype.Proxy:\n\t\treturn &proxy.Adaptor{}\n\tcase apitype.Replicate:\n\t\treturn &replicate.Adaptor{}\n\t}\n\treturn nil\n}\n"
},
{
"path": "relay/adaptor_test.go",
"content": "package relay\n\nimport (\n\t. \"github.com/smartystreets/goconvey/convey\"\n\t\"github.com/songquanpeng/one-api/relay/apitype\"\n\t\"testing\"\n)\n\nfunc TestGetAdaptor(t *testing.T) {\n\tConvey(\"get adaptor\", t, func() {\n\t\tfor i := 0; i < apitype.Dummy; i++ {\n\t\t\ta := GetAdaptor(i)\n\t\t\tSo(a, ShouldNotBeNil)\n\t\t}\n\t})\n}\n"
},
{
"path": "relay/apitype/define.go",
"content": "package apitype\n\nconst (\n\tOpenAI = iota\n\tAnthropic\n\tPaLM\n\tBaidu\n\tZhipu\n\tAli\n\tXunfei\n\tAIProxyLibrary\n\tTencent\n\tGemini\n\tOllama\n\tAwsClaude\n\tCoze\n\tCohere\n\tCloudflare\n\tDeepL\n\tVertexAI\n\tProxy\n\tReplicate\n\n\tDummy // this one is only for count, do not add any channel after this\n)\n"
},
{
"path": "relay/billing/billing.go",
"content": "package billing\n\nimport (\n\t\"context\"\n\t\"fmt\"\n\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/model\"\n)\n\nfunc ReturnPreConsumedQuota(ctx context.Context, preConsumedQuota int64, tokenId int) {\n\tif preConsumedQuota != 0 {\n\t\tgo func(ctx context.Context) {\n\t\t\t// return pre-consumed quota\n\t\t\terr := model.PostConsumeTokenQuota(tokenId, -preConsumedQuota)\n\t\t\tif err != nil {\n\t\t\t\tlogger.Error(ctx, \"error return pre-consumed quota: \"+err.Error())\n\t\t\t}\n\t\t}(ctx)\n\t}\n}\n\nfunc PostConsumeQuota(ctx context.Context, tokenId int, quotaDelta int64, totalQuota int64, userId int, channelId int, modelRatio float64, groupRatio float64, modelName string, tokenName string) {\n\t// quotaDelta is remaining quota to be consumed\n\terr := model.PostConsumeTokenQuota(tokenId, quotaDelta)\n\tif err != nil {\n\t\tlogger.SysError(\"error consuming token remain quota: \" + err.Error())\n\t}\n\terr = model.CacheUpdateUserQuota(ctx, userId)\n\tif err != nil {\n\t\tlogger.SysError(\"error update user quota cache: \" + err.Error())\n\t}\n\t// totalQuota is total quota consumed\n\tif totalQuota != 0 {\n\t\tlogContent := fmt.Sprintf(\"倍率：%.2f × %.2f\", modelRatio, groupRatio)\n\t\tmodel.RecordConsumeLog(ctx, &model.Log{\n\t\t\tUserId: userId,\n\t\t\tChannelId: channelId,\n\t\t\tPromptTokens: int(totalQuota),\n\t\t\tCompletionTokens: 0,\n\t\t\tModelName: modelName,\n\t\t\tTokenName: tokenName,\n\t\t\tQuota: int(totalQuota),\n\t\t\tContent: logContent,\n\t\t})\n\t\tmodel.UpdateUserUsedQuotaAndRequestCount(userId, totalQuota)\n\t\tmodel.UpdateChannelUsedQuota(channelId, totalQuota)\n\t}\n\tif totalQuota <= 0 {\n\t\tlogger.Error(ctx, fmt.Sprintf(\"totalQuota consumed is %d, something is wrong\", totalQuota))\n\t}\n}\n"
},
{
"path": "relay/billing/ratio/group.go",
"content": "package ratio\n\nimport (\n\t\"encoding/json\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"sync\"\n)\n\nvar groupRatioLock sync.RWMutex\nvar GroupRatio = map[string]float64{\n\t\"default\": 1,\n\t\"vip\": 1,\n\t\"svip\": 1,\n}\n\nfunc GroupRatio2JSONString() string {\n\tjsonBytes, err := json.Marshal(GroupRatio)\n\tif err != nil {\n\t\tlogger.SysError(\"error marshalling model ratio: \" + err.Error())\n\t}\n\treturn string(jsonBytes)\n}\n\nfunc UpdateGroupRatioByJSONString(jsonStr string) error {\n\tgroupRatioLock.Lock()\n\tdefer groupRatioLock.Unlock()\n\tGroupRatio = make(map[string]float64)\n\treturn json.Unmarshal([]byte(jsonStr), &GroupRatio)\n}\n\nfunc GetGroupRatio(name string) float64 {\n\tgroupRatioLock.RLock()\n\tdefer groupRatioLock.RUnlock()\n\tratio, ok := GroupRatio[name]\n\tif !ok {\n\t\tlogger.SysError(\"group ratio not found: \" + name)\n\t\treturn 1\n\t}\n\treturn ratio\n}\n"
},
{
"path": "relay/billing/ratio/image.go",
"content": "package ratio\n\nvar ImageSizeRatios = map[string]map[string]float64{\n\t\"dall-e-2\": {\n\t\t\"256x256\": 1,\n\t\t\"512x512\": 1.125,\n\t\t\"1024x1024\": 1.25,\n\t},\n\t\"dall-e-3\": {\n\t\t\"1024x1024\": 1,\n\t\t\"1024x1792\": 2,\n\t\t\"1792x1024\": 2,\n\t},\n\t\"ali-stable-diffusion-xl\": {\n\t\t\"512x1024\": 1,\n\t\t\"1024x768\": 1,\n\t\t\"1024x1024\": 1,\n\t\t\"576x1024\": 1,\n\t\t\"1024x576\": 1,\n\t},\n\t\"ali-stable-diffusion-v1.5\": {\n\t\t\"512x1024\": 1,\n\t\t\"1024x768\": 1,\n\t\t\"1024x1024\": 1,\n\t\t\"576x1024\": 1,\n\t\t\"1024x576\": 1,\n\t},\n\t\"wanx-v1\": {\n\t\t\"1024x1024\": 1,\n\t\t\"720x1280\": 1,\n\t\t\"1280x720\": 1,\n\t},\n\t\"step-1x-medium\": {\n\t\t\"256x256\": 1,\n\t\t\"512x512\": 1,\n\t\t\"768x768\": 1,\n\t\t\"1024x1024\": 1,\n\t\t\"1280x800\": 1,\n\t\t\"800x1280\": 1,\n\t},\n}\n\nvar ImageGenerationAmounts = map[string][2]int{\n\t\"dall-e-2\": {1, 10},\n\t\"dall-e-3\": {1, 1}, // OpenAI allows n=1 currently.\n\t\"ali-stable-diffusion-xl\": {1, 4}, // Ali\n\t\"ali-stable-diffusion-v1.5\": {1, 4}, // Ali\n\t\"wanx-v1\": {1, 4}, // Ali\n\t\"cogview-3\": {1, 1},\n\t\"step-1x-medium\": {1, 1},\n}\n\nvar ImagePromptLengthLimitations = map[string]int{\n\t\"dall-e-2\": 1000,\n\t\"dall-e-3\": 4000,\n\t\"ali-stable-diffusion-xl\": 4000,\n\t\"ali-stable-diffusion-v1.5\": 4000,\n\t\"wanx-v1\": 4000,\n\t\"cogview-3\": 833,\n\t\"step-1x-medium\": 4000,\n}\n\nvar ImageOriginModelName = map[string]string{\n\t\"ali-stable-diffusion-xl\": \"stable-diffusion-xl\",\n\t\"ali-stable-diffusion-v1.5\": \"stable-diffusion-v1.5\",\n}\n"
},
{
"path": "relay/billing/ratio/model.go",
"content": "package ratio\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"strings\"\n\t\"sync\"\n\n\t\"github.com/songquanpeng/one-api/common/logger\"\n)\n\nconst (\n\tUSD2RMB = 7\n\tUSD = 500 // $0.002 = 1 -> $1 = 500\n\tMILLI_USD = 1.0 / 1000 * USD\n\tRMB = USD / USD2RMB\n)\n\nvar modelRatioLock sync.RWMutex\n\n// ModelRatio\n// https://platform.openai.com/docs/models/model-endpoint-compatibility\n// https://cloud.baidu.com/doc/WENXINWORKSHOP/s/Blfmc9dlf\n// https://openai.com/pricing\n// 1 === $0.002 / 1K tokens\n// 1 === ￥0.014 / 1k tokens\nvar ModelRatio = map[string]float64{\n\t// https://openai.com/pricing\n\t\"gpt-4\": 15,\n\t\"gpt-4-0314\": 15,\n\t\"gpt-4-0613\": 15,\n\t\"gpt-4-32k\": 30,\n\t\"gpt-4-32k-0314\": 30,\n\t\"gpt-4-32k-0613\": 30,\n\t\"gpt-4-1106-preview\": 5, // $0.01 / 1K tokens\n\t\"gpt-4-0125-preview\": 5, // $0.01 / 1K tokens\n\t\"gpt-4-turbo-preview\": 5, // $0.01 / 1K tokens\n\t\"gpt-4-turbo\": 5, // $0.01 / 1K tokens\n\t\"gpt-4-turbo-2024-04-09\": 5, // $0.01 / 1K tokens\n\t\"gpt-4o\": 2.5, // $0.005 / 1K tokens\n\t\"chatgpt-4o-latest\": 2.5, // $0.005 / 1K tokens\n\t\"gpt-4o-2024-05-13\": 2.5, // $0.005 / 1K tokens\n\t\"gpt-4o-2024-08-06\": 1.25, // $0.0025 / 1K tokens\n\t\"gpt-4o-2024-11-20\": 1.25, // $0.0025 / 1K tokens\n\t\"gpt-4o-mini\": 0.075, // $0.00015 / 1K tokens\n\t\"gpt-4o-mini-2024-07-18\": 0.075, // $0.00015 / 1K tokens\n\t\"gpt-4-vision-preview\": 5, // $0.01 / 1K tokens\n\t\"gpt-3.5-turbo\": 0.25, // $0.0005 / 1K tokens\n\t\"gpt-3.5-turbo-0301\": 0.75,\n\t\"gpt-3.5-turbo-0613\": 0.75,\n\t\"gpt-3.5-turbo-16k\": 1.5, // $0.003 / 1K tokens\n\t\"gpt-3.5-turbo-16k-0613\": 1.5,\n\t\"gpt-3.5-turbo-instruct\": 0.75, // $0.0015 / 1K tokens\n\t\"gpt-3.5-turbo-1106\": 0.5, // $0.001 / 1K tokens\n\t\"gpt-3.5-turbo-0125\": 0.25, // $0.0005 / 1K tokens\n\t\"o1\": 7.5, // $15.00 / 1M input tokens\n\t\"o1-2024-12-17\": 7.5,\n\t\"o1-preview\": 7.5, // $15.00 / 1M input tokens\n\t\"o1-preview-2024-09-12\": 7.5,\n\t\"o1-mini\": 1.5, // $3.00 / 1M input tokens\n\t\"o1-mini-2024-09-12\": 1.5,\n\t\"o3-mini\": 1.5, // $3.00 / 1M input tokens\n\t\"o3-mini-2025-01-31\": 1.5,\n\t\"davinci-002\": 1, // $0.002 / 1K tokens\n\t\"babbage-002\": 0.2, // $0.0004 / 1K tokens\n\t\"text-ada-001\": 0.2,\n\t\"text-babbage-001\": 0.25,\n\t\"text-curie-001\": 1,\n\t\"text-davinci-002\": 10,\n\t\"text-davinci-003\": 10,\n\t\"text-davinci-edit-001\": 10,\n\t\"code-davinci-edit-001\": 10,\n\t\"whisper-1\": 15, // $0.006 / minute -> $0.006 / 150 words -> $0.006 / 200 tokens -> $0.03 / 1k tokens\n\t\"tts-1\": 7.5, // $0.015 / 1K characters\n\t\"tts-1-1106\": 7.5,\n\t\"tts-1-hd\": 15, // $0.030 / 1K characters\n\t\"tts-1-hd-1106\": 15,\n\t\"davinci\": 10,\n\t\"curie\": 10,\n\t\"babbage\": 10,\n\t\"ada\": 10,\n\t\"text-embedding-ada-002\": 0.05,\n\t\"text-embedding-3-small\": 0.01,\n\t\"text-embedding-3-large\": 0.065,\n\t\"text-search-ada-doc-001\": 10,\n\t\"text-moderation-stable\": 0.1,\n\t\"text-moderation-latest\": 0.1,\n\t\"dall-e-2\": 0.02 * USD, // $0.016 - $0.020 / image\n\t\"dall-e-3\": 0.04 * USD, // $0.040 - $0.120 / image\n\t// https://docs.anthropic.com/en/docs/about-claude/models\n\t\"claude-instant-1.2\": 0.8 / 1000 * USD,\n\t\"claude-2.0\": 8.0 / 1000 * USD,\n\t\"claude-2.1\": 8.0 / 1000 * USD,\n\t\"claude-3-haiku-20240307\": 0.25 / 1000 * USD,\n\t\"claude-3-5-haiku-20241022\": 1.0 / 1000 * USD,\n\t\"claude-3-5-haiku-latest\": 1.0 / 1000 * USD,\n\t\"claude-3-sonnet-20240229\": 3.0 / 1000 * USD,\n\t\"claude-3-5-sonnet-20240620\": 3.0 / 1000 * USD,\n\t\"claude-3-5-sonnet-20241022\": 3.0 / 1000 * USD,\n\t\"claude-3-5-sonnet-latest\": 3.0 / 1000 * USD,\n\t\"claude-3-opus-20240229\": 15.0 / 1000 * USD,\n\t// https://cloud.baidu.com/doc/WENXINWORKSHOP/s/hlrk4akp7\n\t\"ERNIE-4.0-8K\": 0.120 * RMB,\n\t\"ERNIE-3.5-8K\": 0.012 * RMB,\n\t\"ERNIE-3.5-8K-0205\": 0.024 * RMB,\n\t\"ERNIE-3.5-8K-1222\": 0.012 * RMB,\n\t\"ERNIE-Bot-8K\": 0.024 * RMB,\n\t\"ERNIE-3.5-4K-0205\": 0.012 * RMB,\n\t\"ERNIE-Speed-8K\": 0.004 * RMB,\n\t\"ERNIE-Speed-128K\": 0.004 * RMB,\n\t\"ERNIE-Lite-8K-0922\": 0.008 * RMB,\n\t\"ERNIE-Lite-8K-0308\": 0.003 * RMB,\n\t\"ERNIE-Tiny-8K\": 0.001 * RMB,\n\t\"BLOOMZ-7B\": 0.004 * RMB,\n\t\"Embedding-V1\": 0.002 * RMB,\n\t\"bge-large-zh\": 0.002 * RMB,\n\t\"bge-large-en\": 0.002 * RMB,\n\t\"tao-8k\": 0.002 * RMB,\n\t// https://ai.google.dev/pricing\n\t// https://cloud.google.com/vertex-ai/generative-ai/pricing\n\t// \"gemma-2-2b-it\": 0,\n\t// \"gemma-2-9b-it\": 0,\n\t// \"gemma-2-27b-it\": 0,\n\t\"gemini-pro\": 0.25 * MILLI_USD, // $0.00025 / 1k characters -> $0.001 / 1k tokens\n\t\"gemini-1.0-pro\": 0.125 * MILLI_USD,\n\t\"gemini-1.5-pro\": 1.25 * MILLI_USD,\n\t\"gemini-1.5-pro-001\": 1.25 * MILLI_USD,\n\t\"gemini-1.5-pro-experimental\": 1.25 * MILLI_USD,\n\t\"gemini-1.5-flash\": 0.075 * MILLI_USD,\n\t\"gemini-1.5-flash-001\": 0.075 * MILLI_USD,\n\t\"gemini-1.5-flash-8b\": 0.0375 * MILLI_USD,\n\t\"gemini-2.0-flash-exp\": 0.075 * MILLI_USD,\n\t\"gemini-2.0-flash\": 0.15 * MILLI_USD,\n\t\"gemini-2.0-flash-001\": 0.15 * MILLI_USD,\n\t\"gemini-2.0-flash-lite-preview-02-05\": 0.075 * MILLI_USD,\n\t\"gemini-2.0-flash-thinking-exp-01-21\": 0.075 * MILLI_USD,\n\t\"gemini-2.0-pro-exp-02-05\": 1.25 * MILLI_USD,\n\t\"aqa\": 1,\n\t// https://open.bigmodel.cn/pricing\n\t\"glm-zero-preview\": 0.01 * RMB,\n\t\"glm-4-plus\": 0.05 * RMB,\n\t\"glm-4-0520\": 0.1 * RMB,\n\t\"glm-4-airx\": 0.01 * RMB,\n\t\"glm-4-air\": 0.0005 * RMB,\n\t\"glm-4-long\": 0.001 * RMB,\n\t\"glm-4-flashx\": 0.0001 * RMB,\n\t\"glm-4-flash\": 0,\n\t\"glm-4\": 0.1 * RMB, // deprecated model, available until 2025/06\n\t\"glm-3-turbo\": 0.001 * RMB, // deprecated model, available until 2025/06\n\t\"glm-4v-plus\": 0.004 * RMB,\n\t\"glm-4v\": 0.05 * RMB,\n\t\"glm-4v-flash\": 0,\n\t\"cogview-3-plus\": 0.06 * RMB,\n\t\"cogview-3\": 0.1 * RMB,\n\t\"cogview-3-flash\": 0,\n\t\"cogviewx\": 0.5 * RMB,\n\t\"cogviewx-flash\": 0,\n\t\"charglm-4\": 0.001 * RMB,\n\t\"emohaa\": 0.015 * RMB,\n\t\"codegeex-4\": 0.0001 * RMB,\n\t\"embedding-2\": 0.0005 * RMB,\n\t\"embedding-3\": 0.0005 * RMB,\n\t// https://help.aliyun.com/zh/dashscope/developer-reference/tongyi-thousand-questions-metering-and-billing\n\t\"qwen-turbo\": 0.0003 * RMB,\n\t\"qwen-turbo-latest\": 0.0003 * RMB,\n\t\"qwen-plus\": 0.0008 * RMB,\n\t\"qwen-plus-latest\": 0.0008 * RMB,\n\t\"qwen-max\": 0.0024 * RMB,\n\t\"qwen-max-latest\": 0.0024 * RMB,\n\t\"qwen-max-longcontext\": 0.0005 * RMB,\n\t\"qwen-vl-max\": 0.003 * RMB,\n\t\"qwen-vl-max-latest\": 0.003 * RMB,\n\t\"qwen-vl-plus\": 0.0015 * RMB,\n\t\"qwen-vl-plus-latest\": 0.0015 * RMB,\n\t\"qwen-vl-ocr\": 0.005 * RMB,\n\t\"qwen-vl-ocr-latest\": 0.005 * RMB,\n\t\"qwen-audio-turbo\": 1.4286,\n\t\"qwen-math-plus\": 0.004 * RMB,\n\t\"qwen-math-plus-latest\": 0.004 * RMB,\n\t\"qwen-math-turbo\": 0.002 * RMB,\n\t\"qwen-math-turbo-latest\": 0.002 * RMB,\n\t\"qwen-coder-plus\": 0.0035 * RMB,\n\t\"qwen-coder-plus-latest\": 0.0035 * RMB,\n\t\"qwen-coder-turbo\": 0.002 * RMB,\n\t\"qwen-coder-turbo-latest\": 0.002 * RMB,\n\t\"qwen-mt-plus\": 0.015 * RMB,\n\t\"qwen-mt-turbo\": 0.001 * RMB,\n\t\"qwq-32b-preview\": 0.002 * RMB,\n\t\"qwen2.5-72b-instruct\": 0.004 * RMB,\n\t\"qwen2.5-32b-instruct\": 0.03 * RMB,\n\t\"qwen2.5-14b-instruct\": 0.001 * RMB,\n\t\"qwen2.5-7b-instruct\": 0.0005 * RMB,\n\t\"qwen2.5-3b-instruct\": 0.006 * RMB,\n\t\"qwen2.5-1.5b-instruct\": 0.0003 * RMB,\n\t\"qwen2.5-0.5b-instruct\": 0.0003 * RMB,\n\t\"qwen2-72b-instruct\": 0.004 * RMB,\n\t\"qwen2-57b-a14b-instruct\": 0.0035 * RMB,\n\t\"qwen2-7b-instruct\": 0.001 * RMB,\n\t\"qwen2-1.5b-instruct\": 0.001 * RMB,\n\t\"qwen2-0.5b-instruct\": 0.001 * RMB,\n\t\"qwen1.5-110b-chat\": 0.007 * RMB,\n\t\"qwen1.5-72b-chat\": 0.005 * RMB,\n\t\"qwen1.5-32b-chat\": 0.0035 * RMB,\n\t\"qwen1.5-14b-chat\": 0.002 * RMB,\n\t\"qwen1.5-7b-chat\": 0.001 * RMB,\n\t\"qwen1.5-1.8b-chat\": 0.001 * RMB,\n\t\"qwen1.5-0.5b-chat\": 0.001 * RMB,\n\t\"qwen-72b-chat\": 0.02 * RMB,\n\t\"qwen-14b-chat\": 0.008 * RMB,\n\t\"qwen-7b-chat\": 0.006 * RMB,\n\t\"qwen-1.8b-chat\": 0.006 * RMB,\n\t\"qwen-1.8b-longcontext-chat\": 0.006 * RMB,\n\t\"qvq-72b-preview\": 0.012 * RMB,\n\t\"qwen2.5-vl-72b-instruct\": 0.016 * RMB,\n\t\"qwen2.5-vl-7b-instruct\": 0.002 * RMB,\n\t\"qwen2.5-vl-3b-instruct\": 0.0012 * RMB,\n\t\"qwen2-vl-7b-instruct\": 0.016 * RMB,\n\t\"qwen2-vl-2b-instruct\": 0.002 * RMB,\n\t\"qwen-vl-v1\": 0.002 * RMB,\n\t\"qwen-vl-chat-v1\": 0.002 * RMB,\n\t\"qwen2-audio-instruct\": 0.002 * RMB,\n\t\"qwen-audio-chat\": 0.002 * RMB,\n\t\"qwen2.5-math-72b-instruct\": 0.004 * RMB,\n\t\"qwen2.5-math-7b-instruct\": 0.001 * RMB,\n\t\"qwen2.5-math-1.5b-instruct\": 0.001 * RMB,\n\t\"qwen2-math-72b-instruct\": 0.004 * RMB,\n\t\"qwen2-math-7b-instruct\": 0.001 * RMB,\n\t\"qwen2-math-1.5b-instruct\": 0.001 * RMB,\n\t\"qwen2.5-coder-32b-instruct\": 0.002 * RMB,\n\t\"qwen2.5-coder-14b-instruct\": 0.002 * RMB,\n\t\"qwen2.5-coder-7b-instruct\": 0.001 * RMB,\n\t\"qwen2.5-coder-3b-instruct\": 0.001 * RMB,\n\t\"qwen2.5-coder-1.5b-instruct\": 0.001 * RMB,\n\t\"qwen2.5-coder-0.5b-instruct\": 0.001 * RMB,\n\t\"text-embedding-v1\": 0.0007 * RMB, // ￥0.0007 / 1k tokens\n\t\"text-embedding-v3\": 0.0007 * RMB,\n\t\"text-embedding-v2\": 0.0007 * RMB,\n\t\"text-embedding-async-v2\": 0.0007 * RMB,\n\t\"text-embedding-async-v1\": 0.0007 * RMB,\n\t\"ali-stable-diffusion-xl\": 8.00,\n\t\"ali-stable-diffusion-v1.5\": 8.00,\n\t\"wanx-v1\": 8.00,\n\t\"deepseek-r1\": 0.002 * RMB,\n\t\"deepseek-v3\": 0.001 * RMB,\n\t\"deepseek-r1-distill-qwen-1.5b\": 0.001 * RMB,\n\t\"deepseek-r1-distill-qwen-7b\": 0.0005 * RMB,\n\t\"deepseek-r1-distill-qwen-14b\": 0.001 * RMB,\n\t\"deepseek-r1-distill-qwen-32b\": 0.002 * RMB,\n\t\"deepseek-r1-distill-llama-8b\": 0.0005 * RMB,\n\t\"deepseek-r1-distill-llama-70b\": 0.004 * RMB,\n\t\"SparkDesk\": 1.2858, // ￥0.018 / 1k tokens\n\t\"SparkDesk-v1.1\": 1.2858, // ￥0.018 / 1k tokens\n\t\"SparkDesk-v2.1\": 1.2858, // ￥0.018 / 1k tokens\n\t\"SparkDesk-v3.1\": 1.2858, // ￥0.018 / 1k tokens\n\t\"SparkDesk-v3.1-128K\": 1.2858, // ￥0.018 / 1k tokens\n\t\"SparkDesk-v3.5\": 1.2858, // ￥0.018 / 1k tokens\n\t\"SparkDesk-v3.5-32K\": 1.2858, // ￥0.018 / 1k tokens\n\t\"SparkDesk-v4.0\": 1.2858, // ￥0.018 / 1k tokens\n\t\"360GPT_S2_V9\": 0.8572, // ¥0.012 / 1k tokens\n\t\"embedding-bert-512-v1\": 0.0715, // ¥0.001 / 1k tokens\n\t\"embedding_s1_v1\": 0.0715, // ¥0.001 / 1k tokens\n\t\"semantic_similarity_s1_v1\": 0.0715, // ¥0.001 / 1k tokens\n\t// https://cloud.tencent.com/document/product/1729/97731#e0e6be58-60c8-469f-bdeb-6c264ce3b4d0\n\t\"hunyuan-turbo\": 0.015 * RMB,\n\t\"hunyuan-large\": 0.004 * RMB,\n\t\"hunyuan-large-longcontext\": 0.006 * RMB,\n\t\"hunyuan-standard\": 0.0008 * RMB,\n\t\"hunyuan-standard-256K\": 0.0005 * RMB,\n\t\"hunyuan-translation-lite\": 0.005 * RMB,\n\t\"hunyuan-role\": 0.004 * RMB,\n\t\"hunyuan-functioncall\": 0.004 * RMB,\n\t\"hunyuan-code\": 0.004 * RMB,\n\t\"hunyuan-turbo-vision\": 0.08 * RMB,\n\t\"hunyuan-vision\": 0.018 * RMB,\n\t\"hunyuan-embedding\": 0.0007 * RMB,\n\t// https://platform.moonshot.cn/pricing\n\t\"moonshot-v1-8k\": 0.012 * RMB,\n\t\"moonshot-v1-32k\": 0.024 * RMB,\n\t\"moonshot-v1-128k\": 0.06 * RMB,\n\t// https://platform.baichuan-ai.com/price\n\t\"Baichuan2-Turbo\": 0.008 * RMB,\n\t\"Baichuan2-Turbo-192k\": 0.016 * RMB,\n\t\"Baichuan2-53B\": 0.02 * RMB,\n\t// https://api.minimax.chat/document/price\n\t\"abab6.5-chat\": 0.03 * RMB,\n\t\"abab6.5s-chat\": 0.01 * RMB,\n\t\"abab6-chat\": 0.1 * RMB,\n\t\"abab5.5-chat\": 0.015 * RMB,\n\t\"abab5.5s-chat\": 0.005 * RMB,\n\t// https://docs.mistral.ai/platform/pricing/\n\t\"open-mistral-7b\": 0.25 / 1000 * USD,\n\t\"open-mixtral-8x7b\": 0.7 / 1000 * USD,\n\t\"mistral-small-latest\": 2.0 / 1000 * USD,\n\t\"mistral-medium-latest\": 2.7 / 1000 * USD,\n\t\"mistral-large-latest\": 8.0 / 1000 * USD,\n\t\"mistral-embed\": 0.1 / 1000 * USD,\n\t// https://wow.groq.com/#:~:text=inquiries%C2%A0here.-,Model,-Current%20Speed\n\t\"gemma-7b-it\": 0.07 / 1000000 * USD,\n\t\"gemma2-9b-it\": 0.20 / 1000000 * USD,\n\t\"llama-3.1-70b-versatile\": 0.59 / 1000000 * USD,\n\t\"llama-3.1-8b-instant\": 0.05 / 1000000 * USD,\n\t\"llama-3.2-11b-text-preview\": 0.05 / 1000000 * USD,\n\t\"llama-3.2-11b-vision-preview\": 0.05 / 1000000 * USD,\n\t\"llama-3.2-1b-preview\": 0.05 / 1000000 * USD,\n\t\"llama-3.2-3b-preview\": 0.05 / 1000000 * USD,\n\t\"llama-3.2-90b-text-preview\": 0.59 / 1000000 * USD,\n\t\"llama-guard-3-8b\": 0.05 / 1000000 * USD,\n\t\"llama3-70b-8192\": 0.59 / 1000000 * USD,\n\t\"llama3-8b-8192\": 0.05 / 1000000 * USD,\n\t\"llama3-groq-70b-8192-tool-use-preview\": 0.89 / 1000000 * USD,\n\t\"llama3-groq-8b-8192-tool-use-preview\": 0.19 / 1000000 * USD,\n\t\"mixtral-8x7b-32768\": 0.24 / 1000000 * USD,\n\n\t// https://platform.lingyiwanwu.com/docs#-计费单元\n\t\"yi-34b-chat-0205\": 2.5 / 1000 * RMB,\n\t\"yi-34b-chat-200k\": 12.0 / 1000 * RMB,\n\t\"yi-vl-plus\": 6.0 / 1000 * RMB,\n\t// https://platform.stepfun.com/docs/pricing/details\n\t\"step-1-8k\": 0.005 / 1000 * RMB,\n\t\"step-1-32k\": 0.015 / 1000 * RMB,\n\t\"step-1-128k\": 0.040 / 1000 * RMB,\n\t\"step-1-256k\": 0.095 / 1000 * RMB,\n\t\"step-1-flash\": 0.001 / 1000 * RMB,\n\t\"step-2-16k\": 0.038 / 1000 * RMB,\n\t\"step-1v-8k\": 0.005 / 1000 * RMB,\n\t\"step-1v-32k\": 0.015 / 1000 * RMB,\n\t// aws llama3 https://aws.amazon.com/cn/bedrock/pricing/\n\t\"llama3-8b-8192(33)\": 0.0003 / 0.002, // $0.0003 / 1K tokens\n\t\"llama3-70b-8192(33)\": 0.00265 / 0.002, // $0.00265 / 1K tokens\n\t// https://cohere.com/pricing\n\t\"command\": 0.5,\n\t\"command-nightly\": 0.5,\n\t\"command-light\": 0.5,\n\t\"command-light-nightly\": 0.5,\n\t\"command-r\": 0.5 / 1000 * USD,\n\t\"command-r-plus\": 3.0 / 1000 * USD,\n\t// https://platform.deepseek.com/api-docs/pricing/\n\t\"deepseek-chat\": 0.14 * MILLI_USD,\n\t\"deepseek-reasoner\": 0.55 * MILLI_USD,\n\t// https://www.deepl.com/pro?cta=header-prices\n\t\"deepl-zh\": 25.0 / 1000 * USD,\n\t\"deepl-en\": 25.0 / 1000 * USD,\n\t\"deepl-ja\": 25.0 / 1000 * USD,\n\t// https://console.x.ai/\n\t\"grok-beta\": 5.0 / 1000 * USD,\n\t// replicate charges based on the number of generated images\n\t// https://replicate.com/pricing\n\t\"black-forest-labs/flux-1.1-pro\": 0.04 * USD,\n\t\"black-forest-labs/flux-1.1-pro-ultra\": 0.06 * USD,\n\t\"black-forest-labs/flux-canny-dev\": 0.025 * USD,\n\t\"black-forest-labs/flux-canny-pro\": 0.05 * USD,\n\t\"black-forest-labs/flux-depth-dev\": 0.025 * USD,\n\t\"black-forest-labs/flux-depth-pro\": 0.05 * USD,\n\t\"black-forest-labs/flux-dev\": 0.025 * USD,\n\t\"black-forest-labs/flux-dev-lora\": 0.032 * USD,\n\t\"black-forest-labs/flux-fill-dev\": 0.04 * USD,\n\t\"black-forest-labs/flux-fill-pro\": 0.05 * USD,\n\t\"black-forest-labs/flux-pro\": 0.055 * USD,\n\t\"black-forest-labs/flux-redux-dev\": 0.025 * USD,\n\t\"black-forest-labs/flux-redux-schnell\": 0.003 * USD,\n\t\"black-forest-labs/flux-schnell\": 0.003 * USD,\n\t\"black-forest-labs/flux-schnell-lora\": 0.02 * USD,\n\t\"ideogram-ai/ideogram-v2\": 0.08 * USD,\n\t\"ideogram-ai/ideogram-v2-turbo\": 0.05 * USD,\n\t\"recraft-ai/recraft-v3\": 0.04 * USD,\n\t\"recraft-ai/recraft-v3-svg\": 0.08 * USD,\n\t\"stability-ai/stable-diffusion-3\": 0.035 * USD,\n\t\"stability-ai/stable-diffusion-3.5-large\": 0.065 * USD,\n\t\"stability-ai/stable-diffusion-3.5-large-turbo\": 0.04 * USD,\n\t\"stability-ai/stable-diffusion-3.5-medium\": 0.035 * USD,\n\t// replicate chat models\n\t\"ibm-granite/granite-20b-code-instruct-8k\": 0.100 * USD,\n\t\"ibm-granite/granite-3.0-2b-instruct\": 0.030 * USD,\n\t\"ibm-granite/granite-3.0-8b-instruct\": 0.050 * USD,\n\t\"ibm-granite/granite-8b-code-instruct-128k\": 0.050 * USD,\n\t\"meta/llama-2-13b\": 0.100 * USD,\n\t\"meta/llama-2-13b-chat\": 0.100 * USD,\n\t\"meta/llama-2-70b\": 0.650 * USD,\n\t\"meta/llama-2-70b-chat\": 0.650 * USD,\n\t\"meta/llama-2-7b\": 0.050 * USD,\n\t\"meta/llama-2-7b-chat\": 0.050 * USD,\n\t\"meta/meta-llama-3.1-405b-instruct\": 9.500 * USD,\n\t\"meta/meta-llama-3-70b\": 0.650 * USD,\n\t\"meta/meta-llama-3-70b-instruct\": 0.650 * USD,\n\t\"meta/meta-llama-3-8b\": 0.050 * USD,\n\t\"meta/meta-llama-3-8b-instruct\": 0.050 * USD,\n\t\"mistralai/mistral-7b-instruct-v0.2\": 0.050 * USD,\n\t\"mistralai/mistral-7b-v0.1\": 0.050 * USD,\n\t\"mistralai/mixtral-8x7b-instruct-v0.1\": 0.300 * USD,\n\t//https://openrouter.ai/models\n\t\"01-ai/yi-large\": 1.5,\n\t\"aetherwiing/mn-starcannon-12b\": 0.6,\n\t\"ai21/jamba-1-5-large\": 4.0,\n\t\"ai21/jamba-1-5-mini\": 0.2,\n\t\"ai21/jamba-instruct\": 0.35,\n\t\"aion-labs/aion-1.0\": 6.0,\n\t\"aion-labs/aion-1.0-mini\": 1.2,\n\t\"aion-labs/aion-rp-llama-3.1-8b\": 0.1,\n\t\"allenai/llama-3.1-tulu-3-405b\": 5.0,\n\t\"alpindale/goliath-120b\": 4.6875,\n\t\"alpindale/magnum-72b\": 1.125,\n\t\"amazon/nova-lite-v1\": 0.12,\n\t\"amazon/nova-micro-v1\": 0.07,\n\t\"amazon/nova-pro-v1\": 1.6,\n\t\"anthracite-org/magnum-v2-72b\": 1.5,\n\t\"anthracite-org/magnum-v4-72b\": 1.125,\n\t\"anthropic/claude-2\": 12.0,\n\t\"anthropic/claude-2.0\": 12.0,\n\t\"anthropic/claude-2.0:beta\": 12.0,\n\t\"anthropic/claude-2.1\": 12.0,\n\t\"anthropic/claude-2.1:beta\": 12.0,\n\t\"anthropic/claude-2:beta\": 12.0,\n\t\"anthropic/claude-3-haiku\": 0.625,\n\t\"anthropic/claude-3-haiku:beta\": 0.625,\n\t\"anthropic/claude-3-opus\": 37.5,\n\t\"anthropic/claude-3-opus:beta\": 37.5,\n\t\"anthropic/claude-3-sonnet\": 7.5,\n\t\"anthropic/claude-3-sonnet:beta\": 7.5,\n\t\"anthropic/claude-3.5-haiku\": 2.0,\n\t\"anthropic/claude-3.5-haiku-20241022\": 2.0,\n\t\"anthropic/claude-3.5-haiku-20241022:beta\": 2.0,\n\t\"anthropic/claude-3.5-haiku:beta\": 2.0,\n\t\"anthropic/claude-3.5-sonnet\": 7.5,\n\t\"anthropic/claude-3.5-sonnet-20240620\": 7.5,\n\t\"anthropic/claude-3.5-sonnet-20240620:beta\": 7.5,\n\t\"anthropic/claude-3.5-sonnet:beta\": 7.5,\n\t\"cognitivecomputations/dolphin-mixtral-8x22b\": 0.45,\n\t\"cognitivecomputations/dolphin-mixtral-8x7b\": 0.25,\n\t\"cohere/command\": 0.95,\n\t\"cohere/command-r\": 0.7125,\n\t\"cohere/command-r-03-2024\": 0.7125,\n\t\"cohere/command-r-08-2024\": 0.285,\n\t\"cohere/command-r-plus\": 7.125,\n\t\"cohere/command-r-plus-04-2024\": 7.125,\n\t\"cohere/command-r-plus-08-2024\": 4.75,\n\t\"cohere/command-r7b-12-2024\": 0.075,\n\t\"databricks/dbrx-instruct\": 0.6,\n\t\"deepseek/deepseek-chat\": 0.445,\n\t\"deepseek/deepseek-chat-v2.5\": 1.0,\n\t\"deepseek/deepseek-chat:free\": 0.0,\n\t\"deepseek/deepseek-r1\": 1.2,\n\t\"deepseek/deepseek-r1-distill-llama-70b\": 0.345,\n\t\"deepseek/deepseek-r1-distill-llama-70b:free\": 0.0,\n\t\"deepseek/deepseek-r1-distill-llama-8b\": 0.02,\n\t\"deepseek/deepseek-r1-distill-qwen-1.5b\": 0.09,\n\t\"deepseek/deepseek-r1-distill-qwen-14b\": 0.075,\n\t\"deepseek/deepseek-r1-distill-qwen-32b\": 0.09,\n\t\"deepseek/deepseek-r1:free\": 0.0,\n\t\"eva-unit-01/eva-llama-3.33-70b\": 3.0,\n\t\"eva-unit-01/eva-qwen-2.5-32b\": 1.7,\n\t\"eva-unit-01/eva-qwen-2.5-72b\": 3.0,\n\t\"google/gemini-2.0-flash-001\": 0.2,\n\t\"google/gemini-2.0-flash-exp:free\": 0.0,\n\t\"google/gemini-2.0-flash-lite-preview-02-05:free\": 0.0,\n\t\"google/gemini-2.0-flash-thinking-exp-1219:free\": 0.0,\n\t\"google/gemini-2.0-flash-thinking-exp:free\": 0.0,\n\t\"google/gemini-2.0-pro-exp-02-05:free\": 0.0,\n\t\"google/gemini-exp-1206:free\": 0.0,\n\t\"google/gemini-flash-1.5\": 0.15,\n\t\"google/gemini-flash-1.5-8b\": 0.075,\n\t\"google/gemini-flash-1.5-8b-exp\": 0.0,\n\t\"google/gemini-pro\": 0.75,\n\t\"google/gemini-pro-1.5\": 2.5,\n\t\"google/gemini-pro-vision\": 0.75,\n\t\"google/gemma-2-27b-it\": 0.135,\n\t\"google/gemma-2-9b-it\": 0.03,\n\t\"google/gemma-2-9b-it:free\": 0.0,\n\t\"google/gemma-7b-it\": 0.075,\n\t\"google/learnlm-1.5-pro-experimental:free\": 0.0,\n\t\"google/palm-2-chat-bison\": 1.0,\n\t\"google/palm-2-chat-bison-32k\": 1.0,\n\t\"google/palm-2-codechat-bison\": 1.0,\n\t\"google/palm-2-codechat-bison-32k\": 1.0,\n\t\"gryphe/mythomax-l2-13b\": 0.0325,\n\t\"gryphe/mythomax-l2-13b:free\": 0.0,\n\t\"huggingfaceh4/zephyr-7b-beta:free\": 0.0,\n\t\"infermatic/mn-inferor-12b\": 0.6,\n\t\"inflection/inflection-3-pi\": 5.0,\n\t\"inflection/inflection-3-productivity\": 5.0,\n\t\"jondurbin/airoboros-l2-70b\": 0.25,\n\t\"liquid/lfm-3b\": 0.01,\n\t\"liquid/lfm-40b\": 0.075,\n\t\"liquid/lfm-7b\": 0.005,\n\t\"mancer/weaver\": 1.125,\n\t\"meta-llama/llama-2-13b-chat\": 0.11,\n\t\"meta-llama/llama-2-70b-chat\": 0.45,\n\t\"meta-llama/llama-3-70b-instruct\": 0.2,\n\t\"meta-llama/llama-3-8b-instruct\": 0.03,\n\t\"meta-llama/llama-3-8b-instruct:free\": 0.0,\n\t\"meta-llama/llama-3.1-405b\": 1.0,\n\t\"meta-llama/llama-3.1-405b-instruct\": 0.4,\n\t\"meta-llama/llama-3.1-70b-instruct\": 0.15,\n\t\"meta-llama/llama-3.1-8b-instruct\": 0.025,\n\t\"meta-llama/llama-3.2-11b-vision-instruct\": 0.0275,\n\t\"meta-llama/llama-3.2-11b-vision-instruct:free\": 0.0,\n\t\"meta-llama/llama-3.2-1b-instruct\": 0.005,\n\t\"meta-llama/llama-3.2-3b-instruct\": 0.0125,\n\t\"meta-llama/llama-3.2-90b-vision-instruct\": 0.8,\n\t\"meta-llama/llama-3.3-70b-instruct\": 0.15,\n\t\"meta-llama/llama-3.3-70b-instruct:free\": 0.0,\n\t\"meta-llama/llama-guard-2-8b\": 0.1,\n\t\"microsoft/phi-3-medium-128k-instruct\": 0.5,\n\t\"microsoft/phi-3-medium-128k-instruct:free\": 0.0,\n\t\"microsoft/phi-3-mini-128k-instruct\": 0.05,\n\t\"microsoft/phi-3-mini-128k-instruct:free\": 0.0,\n\t\"microsoft/phi-3.5-mini-128k-instruct\": 0.05,\n\t\"microsoft/phi-4\": 0.07,\n\t\"microsoft/wizardlm-2-7b\": 0.035,\n\t\"microsoft/wizardlm-2-8x22b\": 0.25,\n\t\"minimax/minimax-01\": 0.55,\n\t\"mistralai/codestral-2501\": 0.45,\n\t\"mistralai/codestral-mamba\": 0.125,\n\t\"mistralai/ministral-3b\": 0.02,\n\t\"mistralai/ministral-8b\": 0.05,\n\t\"mistralai/mistral-7b-instruct\": 0.0275,\n\t\"mistralai/mistral-7b-instruct-v0.1\": 0.1,\n\t\"mistralai/mistral-7b-instruct-v0.3\": 0.0275,\n\t\"mistralai/mistral-7b-instruct:free\": 0.0,\n\t\"mistralai/mistral-large\": 3.0,\n\t\"mistralai/mistral-large-2407\": 3.0,\n\t\"mistralai/mistral-large-2411\": 3.0,\n\t\"mistralai/mistral-medium\": 4.05,\n\t\"mistralai/mistral-nemo\": 0.04,\n\t\"mistralai/mistral-nemo:free\": 0.0,\n\t\"mistralai/mistral-small\": 0.3,\n\t\"mistralai/mistral-small-24b-instruct-2501\": 0.07,\n\t\"mistralai/mistral-small-24b-instruct-2501:free\": 0.0,\n\t\"mistralai/mistral-tiny\": 0.125,\n\t\"mistralai/mixtral-8x22b-instruct\": 0.45,\n\t\"mistralai/mixtral-8x7b\": 0.3,\n\t\"mistralai/mixtral-8x7b-instruct\": 0.12,\n\t\"mistralai/pixtral-12b\": 0.05,\n\t\"mistralai/pixtral-large-2411\": 3.0,\n\t\"neversleep/llama-3-lumimaid-70b\": 2.25,\n\t\"neversleep/llama-3-lumimaid-8b\": 0.5625,\n\t\"neversleep/llama-3-lumimaid-8b:extended\": 0.5625,\n\t\"neversleep/llama-3.1-lumimaid-70b\": 2.25,\n\t\"neversleep/llama-3.1-lumimaid-8b\": 0.5625,\n\t\"neversleep/noromaid-20b\": 1.125,\n\t\"nothingiisreal/mn-celeste-12b\": 0.6,\n\t\"nousresearch/hermes-2-pro-llama-3-8b\": 0.02,\n\t\"nousresearch/hermes-3-llama-3.1-405b\": 0.4,\n\t\"nousresearch/hermes-3-llama-3.1-70b\": 0.15,\n\t\"nousresearch/nous-hermes-2-mixtral-8x7b-dpo\": 0.3,\n\t\"nousresearch/nous-hermes-llama2-13b\": 0.085,\n\t\"nvidia/llama-3.1-nemotron-70b-instruct\": 0.15,\n\t\"nvidia/llama-3.1-nemotron-70b-instruct:free\": 0.0,\n\t\"openai/chatgpt-4o-latest\": 7.5,\n\t\"openai/gpt-3.5-turbo\": 0.75,\n\t\"openai/gpt-3.5-turbo-0125\": 0.75,\n\t\"openai/gpt-3.5-turbo-0613\": 1.0,\n\t\"openai/gpt-3.5-turbo-1106\": 1.0,\n\t\"openai/gpt-3.5-turbo-16k\": 2.0,\n\t\"openai/gpt-3.5-turbo-instruct\": 1.0,\n\t\"openai/gpt-4\": 30.0,\n\t\"openai/gpt-4-0314\": 30.0,\n\t\"openai/gpt-4-1106-preview\": 15.0,\n\t\"openai/gpt-4-32k\": 60.0,\n\t\"openai/gpt-4-32k-0314\": 60.0,\n\t\"openai/gpt-4-turbo\": 15.0,\n\t\"openai/gpt-4-turbo-preview\": 15.0,\n\t\"openai/gpt-4o\": 5.0,\n\t\"openai/gpt-4o-2024-05-13\": 7.5,\n\t\"openai/gpt-4o-2024-08-06\": 5.0,\n\t\"openai/gpt-4o-2024-11-20\": 5.0,\n\t\"openai/gpt-4o-mini\": 0.3,\n\t\"openai/gpt-4o-mini-2024-07-18\": 0.3,\n\t\"openai/gpt-4o:extended\": 9.0,\n\t\"openai/o1\": 30.0,\n\t\"openai/o1-mini\": 2.2,\n\t\"openai/o1-mini-2024-09-12\": 2.2,\n\t\"openai/o1-preview\": 30.0,\n\t\"openai/o1-preview-2024-09-12\": 30.0,\n\t\"openai/o3-mini\": 2.2,\n\t\"openai/o3-mini-high\": 2.2,\n\t\"openchat/openchat-7b\": 0.0275,\n\t\"openchat/openchat-7b:free\": 0.0,\n\t\"openrouter/auto\": -500000.0,\n\t\"perplexity/llama-3.1-sonar-huge-128k-online\": 2.5,\n\t\"perplexity/llama-3.1-sonar-large-128k-chat\": 0.5,\n\t\"perplexity/llama-3.1-sonar-large-128k-online\": 0.5,\n\t\"perplexity/llama-3.1-sonar-small-128k-chat\": 0.1,\n\t\"perplexity/llama-3.1-sonar-small-128k-online\": 0.1,\n\t\"perplexity/sonar\": 0.5,\n\t\"perplexity/sonar-reasoning\": 2.5,\n\t\"pygmalionai/mythalion-13b\": 0.6,\n\t\"qwen/qvq-72b-preview\": 0.25,\n\t\"qwen/qwen-2-72b-instruct\": 0.45,\n\t\"qwen/qwen-2-7b-instruct\": 0.027,\n\t\"qwen/qwen-2-7b-instruct:free\": 0.0,\n\t\"qwen/qwen-2-vl-72b-instruct\": 0.2,\n\t\"qwen/qwen-2-vl-7b-instruct\": 0.05,\n\t\"qwen/qwen-2.5-72b-instruct\": 0.2,\n\t\"qwen/qwen-2.5-7b-instruct\": 0.025,\n\t\"qwen/qwen-2.5-coder-32b-instruct\": 0.08,\n\t\"qwen/qwen-max\": 3.2,\n\t\"qwen/qwen-plus\": 0.6,\n\t\"qwen/qwen-turbo\": 0.1,\n\t\"qwen/qwen-vl-plus:free\": 0.0,\n\t\"qwen/qwen2.5-vl-72b-instruct:free\": 0.0,\n\t\"qwen/qwq-32b-preview\": 0.09,\n\t\"raifle/sorcererlm-8x22b\": 2.25,\n\t\"sao10k/fimbulvetr-11b-v2\": 0.6,\n\t\"sao10k/l3-euryale-70b\": 0.4,\n\t\"sao10k/l3-lunaris-8b\": 0.03,\n\t\"sao10k/l3.1-70b-hanami-x1\": 1.5,\n\t\"sao10k/l3.1-euryale-70b\": 0.4,\n\t\"sao10k/l3.3-euryale-70b\": 0.4,\n\t\"sophosympatheia/midnight-rose-70b\": 0.4,\n\t\"sophosympatheia/rogue-rose-103b-v0.2:free\": 0.0,\n\t\"teknium/openhermes-2.5-mistral-7b\": 0.085,\n\t\"thedrummer/rocinante-12b\": 0.25,\n\t\"thedrummer/unslopnemo-12b\": 0.25,\n\t\"undi95/remm-slerp-l2-13b\": 0.6,\n\t\"undi95/toppy-m-7b\": 0.035,\n\t\"undi95/toppy-m-7b:free\": 0.0,\n\t\"x-ai/grok-2-1212\": 5.0,\n\t\"x-ai/grok-2-vision-1212\": 5.0,\n\t\"x-ai/grok-beta\": 7.5,\n\t\"x-ai/grok-vision-beta\": 7.5,\n\t\"xwin-lm/xwin-lm-70b\": 1.875,\n}\n\nvar CompletionRatio = map[string]float64{\n\t// aws llama3\n\t\"llama3-8b-8192(33)\": 0.0006 / 0.0003,\n\t\"llama3-70b-8192(33)\": 0.0035 / 0.00265,\n\t// whisper\n\t\"whisper-1\": 0, // only count input tokens\n\t// deepseek\n\t\"deepseek-chat\": 0.28 / 0.14,\n\t\"deepseek-reasoner\": 2.19 / 0.55,\n}\n\nvar (\n\tDefaultModelRatio map[string]float64\n\tDefaultCompletionRatio map[string]float64\n)\n\nfunc init() {\n\tDefaultModelRatio = make(map[string]float64)\n\tfor k, v := range ModelRatio {\n\t\tDefaultModelRatio[k] = v\n\t}\n\tDefaultCompletionRatio = make(map[string]float64)\n\tfor k, v := range CompletionRatio {\n\t\tDefaultCompletionRatio[k] = v\n\t}\n}\n\nfunc AddNewMissingRatio(oldRatio string) string {\n\tnewRatio := make(map[string]float64)\n\terr := json.Unmarshal([]byte(oldRatio), &newRatio)\n\tif err != nil {\n\t\tlogger.SysError(\"error unmarshalling old ratio: \" + err.Error())\n\t\treturn oldRatio\n\t}\n\tfor k, v := range DefaultModelRatio {\n\t\tif _, ok := newRatio[k]; !ok {\n\t\t\tnewRatio[k] = v\n\t\t}\n\t}\n\tjsonBytes, err := json.Marshal(newRatio)\n\tif err != nil {\n\t\tlogger.SysError(\"error marshalling new ratio: \" + err.Error())\n\t\treturn oldRatio\n\t}\n\treturn string(jsonBytes)\n}\n\nfunc ModelRatio2JSONString() string {\n\tjsonBytes, err := json.Marshal(ModelRatio)\n\tif err != nil {\n\t\tlogger.SysError(\"error marshalling model ratio: \" + err.Error())\n\t}\n\treturn string(jsonBytes)\n}\n\nfunc UpdateModelRatioByJSONString(jsonStr string) error {\n\tmodelRatioLock.Lock()\n\tdefer modelRatioLock.Unlock()\n\tModelRatio = make(map[string]float64)\n\treturn json.Unmarshal([]byte(jsonStr), &ModelRatio)\n}\n\nfunc GetModelRatio(name string, channelType int) float64 {\n\tmodelRatioLock.RLock()\n\tdefer modelRatioLock.RUnlock()\n\tif strings.HasPrefix(name, \"qwen-\") && strings.HasSuffix(name, \"-internet\") {\n\t\tname = strings.TrimSuffix(name, \"-internet\")\n\t}\n\tif strings.HasPrefix(name, \"command-\") && strings.HasSuffix(name, \"-internet\") {\n\t\tname = strings.TrimSuffix(name, \"-internet\")\n\t}\n\tmodel := fmt.Sprintf(\"%s(%d)\", name, channelType)\n\tif ratio, ok := ModelRatio[model]; ok {\n\t\treturn ratio\n\t}\n\tif ratio, ok := DefaultModelRatio[model]; ok {\n\t\treturn ratio\n\t}\n\tif ratio, ok := ModelRatio[name]; ok {\n\t\treturn ratio\n\t}\n\tif ratio, ok := DefaultModelRatio[name]; ok {\n\t\treturn ratio\n\t}\n\tlogger.SysError(\"model ratio not found: \" + name)\n\treturn 30\n}\n\nfunc CompletionRatio2JSONString() string {\n\tjsonBytes, err := json.Marshal(CompletionRatio)\n\tif err != nil {\n\t\tlogger.SysError(\"error marshalling completion ratio: \" + err.Error())\n\t}\n\treturn string(jsonBytes)\n}\n\nfunc UpdateCompletionRatioByJSONString(jsonStr string) error {\n\tCompletionRatio = make(map[string]float64)\n\treturn json.Unmarshal([]byte(jsonStr), &CompletionRatio)\n}\n\nfunc GetCompletionRatio(name string, channelType int) float64 {\n\tif strings.HasPrefix(name, \"qwen-\") && strings.HasSuffix(name, \"-internet\") {\n\t\tname = strings.TrimSuffix(name, \"-internet\")\n\t}\n\tmodel := fmt.Sprintf(\"%s(%d)\", name, channelType)\n\tif ratio, ok := CompletionRatio[model]; ok {\n\t\treturn ratio\n\t}\n\tif ratio, ok := DefaultCompletionRatio[model]; ok {\n\t\treturn ratio\n\t}\n\tif ratio, ok := CompletionRatio[name]; ok {\n\t\treturn ratio\n\t}\n\tif ratio, ok := DefaultCompletionRatio[name]; ok {\n\t\treturn ratio\n\t}\n\tif strings.HasPrefix(name, \"gpt-3.5\") {\n\t\tif name == \"gpt-3.5-turbo\" || strings.HasSuffix(name, \"0125\") {\n\t\t\t// https://openai.com/blog/new-embedding-models-and-api-updates\n\t\t\t// Updated GPT-3.5 Turbo model and lower pricing\n\t\t\treturn 3\n\t\t}\n\t\tif strings.HasSuffix(name, \"1106\") {\n\t\t\treturn 2\n\t\t}\n\t\treturn 4.0 / 3.0\n\t}\n\tif strings.HasPrefix(name, \"gpt-4\") {\n\t\tif strings.HasPrefix(name, \"gpt-4o\") {\n\t\t\tif name == \"gpt-4o-2024-05-13\" {\n\t\t\t\treturn 3\n\t\t\t}\n\t\t\treturn 4\n\t\t}\n\t\tif strings.HasPrefix(name, \"gpt-4-turbo\") ||\n\t\t\tstrings.HasSuffix(name, \"preview\") {\n\t\t\treturn 3\n\t\t}\n\t\treturn 2\n\t}\n\t// including o1, o1-preview, o1-mini\n\tif strings.HasPrefix(name, \"o1\") {\n\t\treturn 4\n\t}\n\tif name == \"chatgpt-4o-latest\" {\n\t\treturn 3\n\t}\n\tif strings.HasPrefix(name, \"claude-3\") {\n\t\treturn 5\n\t}\n\tif strings.HasPrefix(name, \"claude-\") {\n\t\treturn 3\n\t}\n\tif strings.HasPrefix(name, \"mistral-\") {\n\t\treturn 3\n\t}\n\tif strings.HasPrefix(name, \"gemini-\") {\n\t\treturn 3\n\t}\n\tif strings.HasPrefix(name, \"deepseek-\") {\n\t\treturn 2\n\t}\n\n\tswitch name {\n\tcase \"llama2-70b-4096\":\n\t\treturn 0.8 / 0.64\n\tcase \"llama3-8b-8192\":\n\t\treturn 2\n\tcase \"llama3-70b-8192\":\n\t\treturn 0.79 / 0.59\n\tcase \"command\", \"command-light\", \"command-nightly\", \"command-light-nightly\":\n\t\treturn 2\n\tcase \"command-r\":\n\t\treturn 3\n\tcase \"command-r-plus\":\n\t\treturn 5\n\tcase \"grok-beta\":\n\t\treturn 3\n\t// Replicate Models\n\t// https://replicate.com/pricing\n\tcase \"ibm-granite/granite-20b-code-instruct-8k\":\n\t\treturn 5\n\tcase \"ibm-granite/granite-3.0-2b-instruct\":\n\t\treturn 8.333333333333334\n\tcase \"ibm-granite/granite-3.0-8b-instruct\",\n\t\t\"ibm-granite/granite-8b-code-instruct-128k\":\n\t\treturn 5\n\tcase \"meta/llama-2-13b\",\n\t\t\"meta/llama-2-13b-chat\",\n\t\t\"meta/llama-2-7b\",\n\t\t\"meta/llama-2-7b-chat\",\n\t\t\"meta/meta-llama-3-8b\",\n\t\t\"meta/meta-llama-3-8b-instruct\":\n\t\treturn 5\n\tcase \"meta/llama-2-70b\",\n\t\t\"meta/llama-2-70b-chat\",\n\t\t\"meta/meta-llama-3-70b\",\n\t\t\"meta/meta-llama-3-70b-instruct\":\n\t\treturn 2.750 / 0.650 // ≈4.230769\n\tcase \"meta/meta-llama-3.1-405b-instruct\":\n\t\treturn 1\n\tcase \"mistralai/mistral-7b-instruct-v0.2\",\n\t\t\"mistralai/mistral-7b-v0.1\":\n\t\treturn 5\n\tcase \"mistralai/mixtral-8x7b-instruct-v0.1\":\n\t\treturn 1.000 / 0.300 // ≈3.333333\n\t}\n\n\treturn 1\n}\n"
},
{
"path": "relay/channeltype/define.go",
"content": "package channeltype\n\nconst (\n\tUnknown = iota\n\tOpenAI\n\tAPI2D\n\tAzure\n\tCloseAI\n\tOpenAISB\n\tOpenAIMax\n\tOhMyGPT\n\tCustom\n\tAils\n\tAIProxy\n\tPaLM\n\tAPI2GPT\n\tAIGC2D\n\tAnthropic\n\tBaidu\n\tZhipu\n\tAli\n\tXunfei\n\tAI360\n\tOpenRouter\n\tAIProxyLibrary\n\tFastGPT\n\tTencent\n\tGemini\n\tMoonshot\n\tBaichuan\n\tMinimax\n\tMistral\n\tGroq\n\tOllama\n\tLingYiWanWu\n\tStepFun\n\tAwsClaude\n\tCoze\n\tCohere\n\tDeepSeek\n\tCloudflare\n\tDeepL\n\tTogetherAI\n\tDoubao\n\tNovita\n\tVertextAI\n\tProxy\n\tSiliconFlow\n\tXAI\n\tReplicate\n\tBaiduV2\n\tXunfeiV2\n\tAliBailian\n\tOpenAICompatible\n\tGeminiOpenAICompatible\n\tDummy\n)\n"
},
{
"path": "relay/channeltype/helper.go",
"content": "package channeltype\n\nimport \"github.com/songquanpeng/one-api/relay/apitype\"\n\nfunc ToAPIType(channelType int) int {\n\tapiType := apitype.OpenAI\n\tswitch channelType {\n\tcase Anthropic:\n\t\tapiType = apitype.Anthropic\n\tcase Baidu:\n\t\tapiType = apitype.Baidu\n\tcase PaLM:\n\t\tapiType = apitype.PaLM\n\tcase Zhipu:\n\t\tapiType = apitype.Zhipu\n\tcase Ali:\n\t\tapiType = apitype.Ali\n\tcase Xunfei:\n\t\tapiType = apitype.Xunfei\n\tcase AIProxyLibrary:\n\t\tapiType = apitype.AIProxyLibrary\n\tcase Tencent:\n\t\tapiType = apitype.Tencent\n\tcase Gemini:\n\t\tapiType = apitype.Gemini\n\tcase Ollama:\n\t\tapiType = apitype.Ollama\n\tcase AwsClaude:\n\t\tapiType = apitype.AwsClaude\n\tcase Coze:\n\t\tapiType = apitype.Coze\n\tcase Cohere:\n\t\tapiType = apitype.Cohere\n\tcase Cloudflare:\n\t\tapiType = apitype.Cloudflare\n\tcase DeepL:\n\t\tapiType = apitype.DeepL\n\tcase VertextAI:\n\t\tapiType = apitype.VertexAI\n\tcase Replicate:\n\t\tapiType = apitype.Replicate\n\tcase Proxy:\n\t\tapiType = apitype.Proxy\n\t}\n\n\treturn apiType\n}\n"
},
{
"path": "relay/channeltype/url.go",
"content": "package channeltype\n\nvar ChannelBaseURLs = []string{\n\t\"\", // 0\n\t\"https://api.openai.com\", // 1\n\t\"https://oa.api2d.net\", // 2\n\t\"\", // 3\n\t\"https://api.closeai-proxy.xyz\", // 4\n\t\"https://api.openai-sb.com\", // 5\n\t\"https://api.openaimax.com\", // 6\n\t\"https://api.ohmygpt.com\", // 7\n\t\"\", // 8\n\t\"https://api.caipacity.com\", // 9\n\t\"https://api.aiproxy.io\", // 10\n\t\"https://generativelanguage.googleapis.com\", // 11\n\t\"https://api.api2gpt.com\", // 12\n\t\"https://api.aigc2d.com\", // 13\n\t\"https://api.anthropic.com\", // 14\n\t\"https://aip.baidubce.com\", // 15\n\t\"https://open.bigmodel.cn\", // 16\n\t\"https://dashscope.aliyuncs.com\", // 17\n\t\"\", // 18\n\t\"https://ai.360.cn\", // 19\n\t\"https://openrouter.ai/api\", // 20\n\t\"https://api.aiproxy.io\", // 21\n\t\"https://fastgpt.run/api/openapi\", // 22\n\t\"https://hunyuan.tencentcloudapi.com\", // 23\n\t\"https://generativelanguage.googleapis.com\", // 24\n\t\"https://api.moonshot.cn\", // 25\n\t\"https://api.baichuan-ai.com\", // 26\n\t\"https://api.minimax.chat\", // 27\n\t\"https://api.mistral.ai\", // 28\n\t\"https://api.groq.com/openai\", // 29\n\t\"http://localhost:11434\", // 30\n\t\"https://api.lingyiwanwu.com\", // 31\n\t\"https://api.stepfun.com\", // 32\n\t\"\", // 33\n\t\"https://api.coze.com\", // 34\n\t\"https://api.cohere.ai\", // 35\n\t\"https://api.deepseek.com\", // 36\n\t\"https://api.cloudflare.com\", // 37\n\t\"https://api-free.deepl.com\", // 38\n\t\"https://api.together.xyz\", // 39\n\t\"https://ark.cn-beijing.volces.com\", // 40\n\t\"https://api.novita.ai/v3/openai\", // 41\n\t\"\", // 42\n\t\"\", // 43\n\t\"https://api.siliconflow.cn\", // 44\n\t\"https://api.x.ai\", // 45\n\t\"https://api.replicate.com/v1/models/\", // 46\n\t\"https://qianfan.baidubce.com\", // 47\n\t\"https://spark-api-open.xf-yun.com\", // 48\n\t\"https://dashscope.aliyuncs.com\", // 49\n\t\"\", // 50\n\n\t\"https://generativelanguage.googleapis.com/v1beta/openai/\", // 51\n}\n\nfunc init() {\n\tif len(ChannelBaseURLs) != Dummy {\n\t\tpanic(\"channel base urls length not match\")\n\t}\n}\n"
},
{
"path": "relay/channeltype/url_test.go",
"content": "package channeltype\n\nimport (\n\t. \"github.com/smartystreets/goconvey/convey\"\n\t\"testing\"\n)\n\nfunc TestChannelBaseURLs(t *testing.T) {\n\tConvey(\"channel base urls\", t, func() {\n\t\tSo(len(ChannelBaseURLs), ShouldEqual, Dummy)\n\t})\n}\n"
},
{
"path": "relay/constant/common.go",
"content": "package constant\n\nvar StopFinishReason = \"stop\"\nvar StreamObject = \"chat.completion.chunk\"\nvar NonStreamObject = \"chat.completion\"\n"
},
{
"path": "relay/constant/finishreason/define.go",
"content": "package finishreason\n\nconst (\n\tStop = \"stop\"\n)\n"
},
{
"path": "relay/constant/role/define.go",
"content": "package role\n\nconst (\n\tSystem = \"system\"\n\tAssistant = \"assistant\"\n)\n"
},
{
"path": "relay/controller/audio.go",
"content": "package controller\n\nimport (\n\t\"bufio\"\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"io\"\n\t\"net/http\"\n\t\"strings\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/client\"\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/ctxkey\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/model\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/billing\"\n\tbillingratio \"github.com/songquanpeng/one-api/relay/billing/ratio\"\n\t\"github.com/songquanpeng/one-api/relay/channeltype\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\trelaymodel \"github.com/songquanpeng/one-api/relay/model\"\n\t\"github.com/songquanpeng/one-api/relay/relaymode\"\n)\n\nfunc RelayAudioHelper(c *gin.Context, relayMode int) *relaymodel.ErrorWithStatusCode {\n\tctx := c.Request.Context()\n\tmeta := meta.GetByContext(c)\n\taudioModel := \"whisper-1\"\n\n\ttokenId := c.GetInt(ctxkey.TokenId)\n\tchannelType := c.GetInt(ctxkey.Channel)\n\tchannelId := c.GetInt(ctxkey.ChannelId)\n\tuserId := c.GetInt(ctxkey.Id)\n\tgroup := c.GetString(ctxkey.Group)\n\ttokenName := c.GetString(ctxkey.TokenName)\n\n\tvar ttsRequest openai.TextToSpeechRequest\n\tif relayMode == relaymode.AudioSpeech {\n\t\t// Read JSON\n\t\terr := common.UnmarshalBodyReusable(c, &ttsRequest)\n\t\t// Check if JSON is valid\n\t\tif err != nil {\n\t\t\treturn openai.ErrorWrapper(err, \"invalid_json\", http.StatusBadRequest)\n\t\t}\n\t\taudioModel = ttsRequest.Model\n\t\t// Check if text is too long 4096\n\t\tif len(ttsRequest.Input) > 4096 {\n\t\t\treturn openai.ErrorWrapper(errors.New(\"input is too long (over 4096 characters)\"), \"text_too_long\", http.StatusBadRequest)\n\t\t}\n\t}\n\n\tmodelRatio := billingratio.GetModelRatio(audioModel, channelType)\n\tgroupRatio := billingratio.GetGroupRatio(group)\n\tratio := modelRatio * groupRatio\n\tvar quota int64\n\tvar preConsumedQuota int64\n\tswitch relayMode {\n\tcase relaymode.AudioSpeech:\n\t\tpreConsumedQuota = int64(float64(len(ttsRequest.Input)) * ratio)\n\t\tquota = preConsumedQuota\n\tdefault:\n\t\tpreConsumedQuota = int64(float64(config.PreConsumedQuota) * ratio)\n\t}\n\tuserQuota, err := model.CacheGetUserQuota(ctx, userId)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"get_user_quota_failed\", http.StatusInternalServerError)\n\t}\n\n\t// Check if user quota is enough\n\tif userQuota-preConsumedQuota < 0 {\n\t\treturn openai.ErrorWrapper(errors.New(\"user quota is not enough\"), \"insufficient_user_quota\", http.StatusForbidden)\n\t}\n\terr = model.CacheDecreaseUserQuota(userId, preConsumedQuota)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"decrease_user_quota_failed\", http.StatusInternalServerError)\n\t}\n\tif userQuota > 100*preConsumedQuota {\n\t\t// in this case, we do not pre-consume quota\n\t\t// because the user has enough quota\n\t\tpreConsumedQuota = 0\n\t}\n\tif preConsumedQuota > 0 {\n\t\terr := model.PreConsumeTokenQuota(tokenId, preConsumedQuota)\n\t\tif err != nil {\n\t\t\treturn openai.ErrorWrapper(err, \"pre_consume_token_quota_failed\", http.StatusForbidden)\n\t\t}\n\t}\n\tsucceed := false\n\tdefer func() {\n\t\tif succeed {\n\t\t\treturn\n\t\t}\n\t\tif preConsumedQuota > 0 {\n\t\t\t// we need to roll back the pre-consumed quota\n\t\t\tdefer func(ctx context.Context) {\n\t\t\t\tgo func() {\n\t\t\t\t\t// negative means add quota back for token & user\n\t\t\t\t\terr := model.PostConsumeTokenQuota(tokenId, -preConsumedQuota)\n\t\t\t\t\tif err != nil {\n\t\t\t\t\t\tlogger.Error(ctx, fmt.Sprintf(\"error rollback pre-consumed quota: %s\", err.Error()))\n\t\t\t\t\t}\n\t\t\t\t}()\n\t\t\t}(c.Request.Context())\n\t\t}\n\t}()\n\n\t// map model name\n\tmodelMapping := c.GetStringMapString(ctxkey.ModelMapping)\n\tif modelMapping != nil && modelMapping[audioModel] != \"\" {\n\t\taudioModel = modelMapping[audioModel]\n\t}\n\n\tbaseURL := channeltype.ChannelBaseURLs[channelType]\n\trequestURL := c.Request.URL.String()\n\tif c.GetString(ctxkey.BaseURL) != \"\" {\n\t\tbaseURL = c.GetString(ctxkey.BaseURL)\n\t}\n\n\tfullRequestURL := openai.GetFullRequestURL(baseURL, requestURL, channelType)\n\tif channelType == channeltype.Azure {\n\t\tapiVersion := meta.Config.APIVersion\n\t\tif relayMode == relaymode.AudioTranscription {\n\t\t\t// https://learn.microsoft.com/en-us/azure/ai-services/openai/whisper-quickstart?tabs=command-line#rest-api\n\t\t\tfullRequestURL = fmt.Sprintf(\"%s/openai/deployments/%s/audio/transcriptions?api-version=%s\", baseURL, audioModel, apiVersion)\n\t\t} else if relayMode == relaymode.AudioSpeech {\n\t\t\t// https://learn.microsoft.com/en-us/azure/ai-services/openai/text-to-speech-quickstart?tabs=command-line#rest-api\n\t\t\tfullRequestURL = fmt.Sprintf(\"%s/openai/deployments/%s/audio/speech?api-version=%s\", baseURL, audioModel, apiVersion)\n\t\t}\n\t}\n\n\trequestBody := &bytes.Buffer{}\n\t_, err = io.Copy(requestBody, c.Request.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"new_request_body_failed\", http.StatusInternalServerError)\n\t}\n\tc.Request.Body = io.NopCloser(bytes.NewBuffer(requestBody.Bytes()))\n\tresponseFormat := c.DefaultPostForm(\"response_format\", \"json\")\n\n\treq, err := http.NewRequest(c.Request.Method, fullRequestURL, requestBody)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"new_request_failed\", http.StatusInternalServerError)\n\t}\n\n\tif (relayMode == relaymode.AudioTranscription || relayMode == relaymode.AudioSpeech) && channelType == channeltype.Azure {\n\t\t// https://learn.microsoft.com/en-us/azure/ai-services/openai/whisper-quickstart?tabs=command-line#rest-api\n\t\tapiKey := c.Request.Header.Get(\"Authorization\")\n\t\tapiKey = strings.TrimPrefix(apiKey, \"Bearer \")\n\t\treq.Header.Set(\"api-key\", apiKey)\n\t\treq.ContentLength = c.Request.ContentLength\n\t} else {\n\t\treq.Header.Set(\"Authorization\", c.Request.Header.Get(\"Authorization\"))\n\t}\n\treq.Header.Set(\"Content-Type\", c.Request.Header.Get(\"Content-Type\"))\n\treq.Header.Set(\"Accept\", c.Request.Header.Get(\"Accept\"))\n\n\tresp, err := client.HTTPClient.Do(req)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"do_request_failed\", http.StatusInternalServerError)\n\t}\n\n\terr = req.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_request_body_failed\", http.StatusInternalServerError)\n\t}\n\terr = c.Request.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_request_body_failed\", http.StatusInternalServerError)\n\t}\n\n\tif relayMode != relaymode.AudioSpeech {\n\t\tresponseBody, err := io.ReadAll(resp.Body)\n\t\tif err != nil {\n\t\t\treturn openai.ErrorWrapper(err, \"read_response_body_failed\", http.StatusInternalServerError)\n\t\t}\n\t\terr = resp.Body.Close()\n\t\tif err != nil {\n\t\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError)\n\t\t}\n\n\t\tvar openAIErr openai.SlimTextResponse\n\t\tif err = json.Unmarshal(responseBody, &openAIErr); err == nil {\n\t\t\tif openAIErr.Error.Message != \"\" {\n\t\t\t\treturn openai.ErrorWrapper(fmt.Errorf(\"type %s, code %v, message %s\", openAIErr.Error.Type, openAIErr.Error.Code, openAIErr.Error.Message), \"request_error\", http.StatusInternalServerError)\n\t\t\t}\n\t\t}\n\n\t\tvar text string\n\t\tswitch responseFormat {\n\t\tcase \"json\":\n\t\t\ttext, err = getTextFromJSON(responseBody)\n\t\tcase \"text\":\n\t\t\ttext, err = getTextFromText(responseBody)\n\t\tcase \"srt\":\n\t\t\ttext, err = getTextFromSRT(responseBody)\n\t\tcase \"verbose_json\":\n\t\t\ttext, err = getTextFromVerboseJSON(responseBody)\n\t\tcase \"vtt\":\n\t\t\ttext, err = getTextFromVTT(responseBody)\n\t\tdefault:\n\t\t\treturn openai.ErrorWrapper(errors.New(\"unexpected_response_format\"), \"unexpected_response_format\", http.StatusInternalServerError)\n\t\t}\n\t\tif err != nil {\n\t\t\treturn openai.ErrorWrapper(err, \"get_text_from_body_err\", http.StatusInternalServerError)\n\t\t}\n\t\tquota = int64(openai.CountTokenText(text, audioModel))\n\t\tresp.Body = io.NopCloser(bytes.NewBuffer(responseBody))\n\t}\n\tif resp.StatusCode != http.StatusOK {\n\t\treturn RelayErrorHandler(resp)\n\t}\n\tsucceed = true\n\tquotaDelta := quota - preConsumedQuota\n\tdefer func(ctx context.Context) {\n\t\tgo billing.PostConsumeQuota(ctx, tokenId, quotaDelta, quota, userId, channelId, modelRatio, groupRatio, audioModel, tokenName)\n\t}(c.Request.Context())\n\n\tfor k, v := range resp.Header {\n\t\tc.Writer.Header().Set(k, v[0])\n\t}\n\tc.Writer.WriteHeader(resp.StatusCode)\n\n\t_, err = io.Copy(c.Writer, resp.Body)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"copy_response_body_failed\", http.StatusInternalServerError)\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"close_response_body_failed\", http.StatusInternalServerError)\n\t}\n\treturn nil\n}\n\nfunc getTextFromVTT(body []byte) (string, error) {\n\treturn getTextFromSRT(body)\n}\n\nfunc getTextFromVerboseJSON(body []byte) (string, error) {\n\tvar whisperResponse openai.WhisperVerboseJSONResponse\n\tif err := json.Unmarshal(body, &whisperResponse); err != nil {\n\t\treturn \"\", fmt.Errorf(\"unmarshal_response_body_failed err :%w\", err)\n\t}\n\treturn whisperResponse.Text, nil\n}\n\nfunc getTextFromSRT(body []byte) (string, error) {\n\tscanner := bufio.NewScanner(strings.NewReader(string(body)))\n\tvar builder strings.Builder\n\tvar textLine bool\n\tfor scanner.Scan() {\n\t\tline := scanner.Text()\n\t\tif textLine {\n\t\t\tbuilder.WriteString(line)\n\t\t\ttextLine = false\n\t\t\tcontinue\n\t\t} else if strings.Contains(line, \"-->\") {\n\t\t\ttextLine = true\n\t\t\tcontinue\n\t\t}\n\t}\n\tif err := scanner.Err(); err != nil {\n\t\treturn \"\", err\n\t}\n\treturn builder.String(), nil\n}\n\nfunc getTextFromText(body []byte) (string, error) {\n\treturn strings.TrimSuffix(string(body), \"\\n\"), nil\n}\n\nfunc getTextFromJSON(body []byte) (string, error) {\n\tvar whisperResponse openai.WhisperJSONResponse\n\tif err := json.Unmarshal(body, &whisperResponse); err != nil {\n\t\treturn \"\", fmt.Errorf(\"unmarshal_response_body_failed err :%w\", err)\n\t}\n\treturn whisperResponse.Text, nil\n}\n"
},
{
"path": "relay/controller/error.go",
"content": "package controller\n\nimport (\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"io\"\n\t\"net/http\"\n\t\"strconv\"\n)\n\ntype GeneralErrorResponse struct {\n\tError model.Error `json:\"error\"`\n\tMessage string `json:\"message\"`\n\tMsg string `json:\"msg\"`\n\tErr string `json:\"err\"`\n\tErrorMsg string `json:\"error_msg\"`\n\tHeader struct {\n\t\tMessage string `json:\"message\"`\n\t} `json:\"header\"`\n\tResponse struct {\n\t\tError struct {\n\t\t\tMessage string `json:\"message\"`\n\t\t} `json:\"error\"`\n\t} `json:\"response\"`\n}\n\nfunc (e GeneralErrorResponse) ToMessage() string {\n\tif e.Error.Message != \"\" {\n\t\treturn e.Error.Message\n\t}\n\tif e.Message != \"\" {\n\t\treturn e.Message\n\t}\n\tif e.Msg != \"\" {\n\t\treturn e.Msg\n\t}\n\tif e.Err != \"\" {\n\t\treturn e.Err\n\t}\n\tif e.ErrorMsg != \"\" {\n\t\treturn e.ErrorMsg\n\t}\n\tif e.Header.Message != \"\" {\n\t\treturn e.Header.Message\n\t}\n\tif e.Response.Error.Message != \"\" {\n\t\treturn e.Response.Error.Message\n\t}\n\treturn \"\"\n}\n\nfunc RelayErrorHandler(resp *http.Response) (ErrorWithStatusCode *model.ErrorWithStatusCode) {\n\tif resp == nil {\n\t\treturn &model.ErrorWithStatusCode{\n\t\t\tStatusCode: 500,\n\t\t\tError: model.Error{\n\t\t\t\tMessage: \"resp is nil\",\n\t\t\t\tType: \"upstream_error\",\n\t\t\t\tCode: \"bad_response\",\n\t\t\t},\n\t\t}\n\t}\n\tErrorWithStatusCode = &model.ErrorWithStatusCode{\n\t\tStatusCode: resp.StatusCode,\n\t\tError: model.Error{\n\t\t\tMessage: \"\",\n\t\t\tType: \"upstream_error\",\n\t\t\tCode: \"bad_response_status_code\",\n\t\t\tParam: strconv.Itoa(resp.StatusCode),\n\t\t},\n\t}\n\tresponseBody, err := io.ReadAll(resp.Body)\n\tif err != nil {\n\t\treturn\n\t}\n\tif config.DebugEnabled {\n\t\tlogger.SysLog(fmt.Sprintf(\"error happened, status code: %d, response: \\n%s\", resp.StatusCode, string(responseBody)))\n\t}\n\terr = resp.Body.Close()\n\tif err != nil {\n\t\treturn\n\t}\n\tvar errResponse GeneralErrorResponse\n\terr = json.Unmarshal(responseBody, &errResponse)\n\tif err != nil {\n\t\treturn\n\t}\n\tif errResponse.Error.Message != \"\" {\n\t\t// OpenAI format error, so we override the default one\n\t\tErrorWithStatusCode.Error = errResponse.Error\n\t} else {\n\t\tErrorWithStatusCode.Error.Message = errResponse.ToMessage()\n\t}\n\tif ErrorWithStatusCode.Error.Message == \"\" {\n\t\tErrorWithStatusCode.Error.Message = fmt.Sprintf(\"bad response status code %d\", resp.StatusCode)\n\t}\n\treturn\n}\n"
},
{
"path": "relay/controller/helper.go",
"content": "package controller\n\nimport (\n\t\"context\"\n\t\"errors\"\n\t\"fmt\"\n\t\"math\"\n\t\"net/http\"\n\t\"strings\"\n\n\t\"github.com/songquanpeng/one-api/common/helper\"\n\t\"github.com/songquanpeng/one-api/relay/constant/role\"\n\n\t\"github.com/gin-gonic/gin\"\n\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/model\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\tbillingratio \"github.com/songquanpeng/one-api/relay/billing/ratio\"\n\t\"github.com/songquanpeng/one-api/relay/channeltype\"\n\t\"github.com/songquanpeng/one-api/relay/controller/validator\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\trelaymodel \"github.com/songquanpeng/one-api/relay/model\"\n\t\"github.com/songquanpeng/one-api/relay/relaymode\"\n)\n\nfunc getAndValidateTextRequest(c *gin.Context, relayMode int) (*relaymodel.GeneralOpenAIRequest, error) {\n\ttextRequest := &relaymodel.GeneralOpenAIRequest{}\n\terr := common.UnmarshalBodyReusable(c, textRequest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif relayMode == relaymode.Moderations && textRequest.Model == \"\" {\n\t\ttextRequest.Model = \"text-moderation-latest\"\n\t}\n\tif relayMode == relaymode.Embeddings && textRequest.Model == \"\" {\n\t\ttextRequest.Model = c.Param(\"model\")\n\t}\n\terr = validator.ValidateTextRequest(textRequest, relayMode)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\treturn textRequest, nil\n}\n\nfunc getPromptTokens(textRequest *relaymodel.GeneralOpenAIRequest, relayMode int) int {\n\tswitch relayMode {\n\tcase relaymode.ChatCompletions:\n\t\treturn openai.CountTokenMessages(textRequest.Messages, textRequest.Model)\n\tcase relaymode.Completions:\n\t\treturn openai.CountTokenInput(textRequest.Prompt, textRequest.Model)\n\tcase relaymode.Moderations:\n\t\treturn openai.CountTokenInput(textRequest.Input, textRequest.Model)\n\t}\n\treturn 0\n}\n\nfunc getPreConsumedQuota(textRequest *relaymodel.GeneralOpenAIRequest, promptTokens int, ratio float64) int64 {\n\tpreConsumedTokens := config.PreConsumedQuota + int64(promptTokens)\n\tif textRequest.MaxTokens != 0 {\n\t\tpreConsumedTokens += int64(textRequest.MaxTokens)\n\t}\n\treturn int64(float64(preConsumedTokens) * ratio)\n}\n\nfunc preConsumeQuota(ctx context.Context, textRequest *relaymodel.GeneralOpenAIRequest, promptTokens int, ratio float64, meta *meta.Meta) (int64, *relaymodel.ErrorWithStatusCode) {\n\tpreConsumedQuota := getPreConsumedQuota(textRequest, promptTokens, ratio)\n\n\tuserQuota, err := model.CacheGetUserQuota(ctx, meta.UserId)\n\tif err != nil {\n\t\treturn preConsumedQuota, openai.ErrorWrapper(err, \"get_user_quota_failed\", http.StatusInternalServerError)\n\t}\n\tif userQuota-preConsumedQuota < 0 {\n\t\treturn preConsumedQuota, openai.ErrorWrapper(errors.New(\"user quota is not enough\"), \"insufficient_user_quota\", http.StatusForbidden)\n\t}\n\terr = model.CacheDecreaseUserQuota(meta.UserId, preConsumedQuota)\n\tif err != nil {\n\t\treturn preConsumedQuota, openai.ErrorWrapper(err, \"decrease_user_quota_failed\", http.StatusInternalServerError)\n\t}\n\tif userQuota > 100*preConsumedQuota {\n\t\t// in this case, we do not pre-consume quota\n\t\t// because the user has enough quota\n\t\tpreConsumedQuota = 0\n\t\tlogger.Info(ctx, fmt.Sprintf(\"user %d has enough quota %d, trusted and no need to pre-consume\", meta.UserId, userQuota))\n\t}\n\tif preConsumedQuota > 0 {\n\t\terr := model.PreConsumeTokenQuota(meta.TokenId, preConsumedQuota)\n\t\tif err != nil {\n\t\t\treturn preConsumedQuota, openai.ErrorWrapper(err, \"pre_consume_token_quota_failed\", http.StatusForbidden)\n\t\t}\n\t}\n\treturn preConsumedQuota, nil\n}\n\nfunc postConsumeQuota(ctx context.Context, usage *relaymodel.Usage, meta *meta.Meta, textRequest *relaymodel.GeneralOpenAIRequest, ratio float64, preConsumedQuota int64, modelRatio float64, groupRatio float64, systemPromptReset bool) {\n\tif usage == nil {\n\t\tlogger.Error(ctx, \"usage is nil, which is unexpected\")\n\t\treturn\n\t}\n\tvar quota int64\n\tcompletionRatio := billingratio.GetCompletionRatio(textRequest.Model, meta.ChannelType)\n\tpromptTokens := usage.PromptTokens\n\tcompletionTokens := usage.CompletionTokens\n\tquota = int64(math.Ceil((float64(promptTokens) + float64(completionTokens)*completionRatio) * ratio))\n\tif ratio != 0 && quota <= 0 {\n\t\tquota = 1\n\t}\n\ttotalTokens := promptTokens + completionTokens\n\tif totalTokens == 0 {\n\t\t// in this case, must be some error happened\n\t\t// we cannot just return, because we may have to return the pre-consumed quota\n\t\tquota = 0\n\t}\n\tquotaDelta := quota - preConsumedQuota\n\terr := model.PostConsumeTokenQuota(meta.TokenId, quotaDelta)\n\tif err != nil {\n\t\tlogger.Error(ctx, \"error consuming token remain quota: \"+err.Error())\n\t}\n\terr = model.CacheUpdateUserQuota(ctx, meta.UserId)\n\tif err != nil {\n\t\tlogger.Error(ctx, \"error update user quota cache: \"+err.Error())\n\t}\n\tlogContent := fmt.Sprintf(\"倍率：%.2f × %.2f × %.2f\", modelRatio, groupRatio, completionRatio)\n\tmodel.RecordConsumeLog(ctx, &model.Log{\n\t\tUserId: meta.UserId,\n\t\tChannelId: meta.ChannelId,\n\t\tPromptTokens: promptTokens,\n\t\tCompletionTokens: completionTokens,\n\t\tModelName: textRequest.Model,\n\t\tTokenName: meta.TokenName,\n\t\tQuota: int(quota),\n\t\tContent: logContent,\n\t\tIsStream: meta.IsStream,\n\t\tElapsedTime: helper.CalcElapsedTime(meta.StartTime),\n\t\tSystemPromptReset: systemPromptReset,\n\t})\n\tmodel.UpdateUserUsedQuotaAndRequestCount(meta.UserId, quota)\n\tmodel.UpdateChannelUsedQuota(meta.ChannelId, quota)\n}\n\nfunc getMappedModelName(modelName string, mapping map[string]string) (string, bool) {\n\tif mapping == nil {\n\t\treturn modelName, false\n\t}\n\tmappedModelName := mapping[modelName]\n\tif mappedModelName != \"\" {\n\t\treturn mappedModelName, true\n\t}\n\treturn modelName, false\n}\n\nfunc isErrorHappened(meta *meta.Meta, resp *http.Response) bool {\n\tif resp == nil {\n\t\tif meta.ChannelType == channeltype.AwsClaude {\n\t\t\treturn false\n\t\t}\n\t\treturn true\n\t}\n\tif resp.StatusCode != http.StatusOK &&\n\t\t// replicate return 201 to create a task\n\t\tresp.StatusCode != http.StatusCreated {\n\t\treturn true\n\t}\n\tif meta.ChannelType == channeltype.DeepL {\n\t\t// skip stream check for deepl\n\t\treturn false\n\t}\n\n\tif meta.IsStream && strings.HasPrefix(resp.Header.Get(\"Content-Type\"), \"application/json\") &&\n\t\t// Even if stream mode is enabled, replicate will first return a task info in JSON format,\n\t\t// requiring the client to request the stream endpoint in the task info\n\t\tmeta.ChannelType != channeltype.Replicate {\n\t\treturn true\n\t}\n\treturn false\n}\n\nfunc setSystemPrompt(ctx context.Context, request *relaymodel.GeneralOpenAIRequest, prompt string) (reset bool) {\n\tif prompt == \"\" {\n\t\treturn false\n\t}\n\tif len(request.Messages) == 0 {\n\t\treturn false\n\t}\n\tif request.Messages[0].Role == role.System {\n\t\trequest.Messages[0].Content = prompt\n\t\tlogger.Infof(ctx, \"rewrite system prompt\")\n\t\treturn true\n\t}\n\trequest.Messages = append([]relaymodel.Message{{\n\t\tRole: role.System,\n\t\tContent: prompt,\n\t}}, request.Messages...)\n\tlogger.Infof(ctx, \"add system prompt\")\n\treturn true\n}\n"
},
{
"path": "relay/controller/image.go",
"content": "package controller\n\nimport (\n\t\"bytes\"\n\t\"context\"\n\t\"encoding/json\"\n\t\"errors\"\n\t\"fmt\"\n\t\"io\"\n\t\"net/http\"\n\n\t\"github.com/gin-gonic/gin\"\n\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/ctxkey\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/model\"\n\t\"github.com/songquanpeng/one-api/relay\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\tbillingratio \"github.com/songquanpeng/one-api/relay/billing/ratio\"\n\t\"github.com/songquanpeng/one-api/relay/channeltype\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\trelaymodel \"github.com/songquanpeng/one-api/relay/model\"\n)\n\nfunc getImageRequest(c *gin.Context, _ int) (*relaymodel.ImageRequest, error) {\n\timageRequest := &relaymodel.ImageRequest{}\n\terr := common.UnmarshalBodyReusable(c, imageRequest)\n\tif err != nil {\n\t\treturn nil, err\n\t}\n\tif imageRequest.N == 0 {\n\t\timageRequest.N = 1\n\t}\n\tif imageRequest.Size == \"\" {\n\t\timageRequest.Size = \"1024x1024\"\n\t}\n\tif imageRequest.Model == \"\" {\n\t\timageRequest.Model = \"dall-e-2\"\n\t}\n\treturn imageRequest, nil\n}\n\nfunc isValidImageSize(model string, size string) bool {\n\tif model == \"cogview-3\" || billingratio.ImageSizeRatios[model] == nil {\n\t\treturn true\n\t}\n\t_, ok := billingratio.ImageSizeRatios[model][size]\n\treturn ok\n}\n\nfunc isValidImagePromptLength(model string, promptLength int) bool {\n\tmaxPromptLength, ok := billingratio.ImagePromptLengthLimitations[model]\n\treturn !ok || promptLength <= maxPromptLength\n}\n\nfunc isWithinRange(element string, value int) bool {\n\tamounts, ok := billingratio.ImageGenerationAmounts[element]\n\treturn !ok || (value >= amounts[0] && value <= amounts[1])\n}\n\nfunc getImageSizeRatio(model string, size string) float64 {\n\tif ratio, ok := billingratio.ImageSizeRatios[model][size]; ok {\n\t\treturn ratio\n\t}\n\treturn 1\n}\n\nfunc validateImageRequest(imageRequest *relaymodel.ImageRequest, _ *meta.Meta) *relaymodel.ErrorWithStatusCode {\n\t// check prompt length\n\tif imageRequest.Prompt == \"\" {\n\t\treturn openai.ErrorWrapper(errors.New(\"prompt is required\"), \"prompt_missing\", http.StatusBadRequest)\n\t}\n\n\t// model validation\n\tif !isValidImageSize(imageRequest.Model, imageRequest.Size) {\n\t\treturn openai.ErrorWrapper(errors.New(\"size not supported for this image model\"), \"size_not_supported\", http.StatusBadRequest)\n\t}\n\n\tif !isValidImagePromptLength(imageRequest.Model, len(imageRequest.Prompt)) {\n\t\treturn openai.ErrorWrapper(errors.New(\"prompt is too long\"), \"prompt_too_long\", http.StatusBadRequest)\n\t}\n\n\t// Number of generated images validation\n\tif !isWithinRange(imageRequest.Model, imageRequest.N) {\n\t\treturn openai.ErrorWrapper(errors.New(\"invalid value of n\"), \"n_not_within_range\", http.StatusBadRequest)\n\t}\n\treturn nil\n}\n\nfunc getImageCostRatio(imageRequest *relaymodel.ImageRequest) (float64, error) {\n\tif imageRequest == nil {\n\t\treturn 0, errors.New(\"imageRequest is nil\")\n\t}\n\timageCostRatio := getImageSizeRatio(imageRequest.Model, imageRequest.Size)\n\tif imageRequest.Quality == \"hd\" && imageRequest.Model == \"dall-e-3\" {\n\t\tif imageRequest.Size == \"1024x1024\" {\n\t\t\timageCostRatio *= 2\n\t\t} else {\n\t\t\timageCostRatio *= 1.5\n\t\t}\n\t}\n\treturn imageCostRatio, nil\n}\n\nfunc RelayImageHelper(c *gin.Context, relayMode int) *relaymodel.ErrorWithStatusCode {\n\tctx := c.Request.Context()\n\tmeta := meta.GetByContext(c)\n\timageRequest, err := getImageRequest(c, meta.Mode)\n\tif err != nil {\n\t\tlogger.Errorf(ctx, \"getImageRequest failed: %s\", err.Error())\n\t\treturn openai.ErrorWrapper(err, \"invalid_image_request\", http.StatusBadRequest)\n\t}\n\n\t// map model name\n\tvar isModelMapped bool\n\tmeta.OriginModelName = imageRequest.Model\n\timageRequest.Model, isModelMapped = getMappedModelName(imageRequest.Model, meta.ModelMapping)\n\tmeta.ActualModelName = imageRequest.Model\n\n\t// model validation\n\tbizErr := validateImageRequest(imageRequest, meta)\n\tif bizErr != nil {\n\t\treturn bizErr\n\t}\n\n\timageCostRatio, err := getImageCostRatio(imageRequest)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"get_image_cost_ratio_failed\", http.StatusInternalServerError)\n\t}\n\n\timageModel := imageRequest.Model\n\t// Convert the original image model\n\timageRequest.Model, _ = getMappedModelName(imageRequest.Model, billingratio.ImageOriginModelName)\n\tc.Set(\"response_format\", imageRequest.ResponseFormat)\n\n\tvar requestBody io.Reader\n\tif isModelMapped || meta.ChannelType == channeltype.Azure { // make Azure channel request body\n\t\tjsonStr, err := json.Marshal(imageRequest)\n\t\tif err != nil {\n\t\t\treturn openai.ErrorWrapper(err, \"marshal_image_request_failed\", http.StatusInternalServerError)\n\t\t}\n\t\trequestBody = bytes.NewBuffer(jsonStr)\n\t} else {\n\t\trequestBody = c.Request.Body\n\t}\n\n\tadaptor := relay.GetAdaptor(meta.APIType)\n\tif adaptor == nil {\n\t\treturn openai.ErrorWrapper(fmt.Errorf(\"invalid api type: %d\", meta.APIType), \"invalid_api_type\", http.StatusBadRequest)\n\t}\n\tadaptor.Init(meta)\n\n\t// these adaptors need to convert the request\n\tswitch meta.ChannelType {\n\tcase channeltype.Zhipu,\n\t\tchanneltype.Ali,\n\t\tchanneltype.Replicate,\n\t\tchanneltype.Baidu:\n\t\tfinalRequest, err := adaptor.ConvertImageRequest(imageRequest)\n\t\tif err != nil {\n\t\t\treturn openai.ErrorWrapper(err, \"convert_image_request_failed\", http.StatusInternalServerError)\n\t\t}\n\t\tjsonStr, err := json.Marshal(finalRequest)\n\t\tif err != nil {\n\t\t\treturn openai.ErrorWrapper(err, \"marshal_image_request_failed\", http.StatusInternalServerError)\n\t\t}\n\t\trequestBody = bytes.NewBuffer(jsonStr)\n\t}\n\n\tmodelRatio := billingratio.GetModelRatio(imageModel, meta.ChannelType)\n\tgroupRatio := billingratio.GetGroupRatio(meta.Group)\n\tratio := modelRatio * groupRatio\n\tuserQuota, err := model.CacheGetUserQuota(ctx, meta.UserId)\n\n\tvar quota int64\n\tswitch meta.ChannelType {\n\tcase channeltype.Replicate:\n\t\t// replicate always return 1 image\n\t\tquota = int64(ratio * imageCostRatio * 1000)\n\tdefault:\n\t\tquota = int64(ratio*imageCostRatio*1000) * int64(imageRequest.N)\n\t}\n\n\tif userQuota-quota < 0 {\n\t\treturn openai.ErrorWrapper(errors.New(\"user quota is not enough\"), \"insufficient_user_quota\", http.StatusForbidden)\n\t}\n\n\t// do request\n\tresp, err := adaptor.DoRequest(c, meta, requestBody)\n\tif err != nil {\n\t\tlogger.Errorf(ctx, \"DoRequest failed: %s\", err.Error())\n\t\treturn openai.ErrorWrapper(err, \"do_request_failed\", http.StatusInternalServerError)\n\t}\n\n\tdefer func(ctx context.Context) {\n\t\tif resp != nil &&\n\t\t\tresp.StatusCode != http.StatusCreated && // replicate returns 201\n\t\t\tresp.StatusCode != http.StatusOK {\n\t\t\treturn\n\t\t}\n\n\t\terr := model.PostConsumeTokenQuota(meta.TokenId, quota)\n\t\tif err != nil {\n\t\t\tlogger.SysError(\"error consuming token remain quota: \" + err.Error())\n\t\t}\n\t\terr = model.CacheUpdateUserQuota(ctx, meta.UserId)\n\t\tif err != nil {\n\t\t\tlogger.SysError(\"error update user quota cache: \" + err.Error())\n\t\t}\n\t\tif quota != 0 {\n\t\t\ttokenName := c.GetString(ctxkey.TokenName)\n\t\t\tlogContent := fmt.Sprintf(\"倍率：%.2f × %.2f\", modelRatio, groupRatio)\n\t\t\tmodel.RecordConsumeLog(ctx, &model.Log{\n\t\t\t\tUserId: meta.UserId,\n\t\t\t\tChannelId: meta.ChannelId,\n\t\t\t\tPromptTokens: 0,\n\t\t\t\tCompletionTokens: 0,\n\t\t\t\tModelName: imageRequest.Model,\n\t\t\t\tTokenName: tokenName,\n\t\t\t\tQuota: int(quota),\n\t\t\t\tContent: logContent,\n\t\t\t})\n\t\t\tmodel.UpdateUserUsedQuotaAndRequestCount(meta.UserId, quota)\n\t\t\tchannelId := c.GetInt(ctxkey.ChannelId)\n\t\t\tmodel.UpdateChannelUsedQuota(channelId, quota)\n\t\t}\n\t}(c.Request.Context())\n\n\t// do response\n\t_, respErr := adaptor.DoResponse(c, resp, meta)\n\tif respErr != nil {\n\t\tlogger.Errorf(ctx, \"respErr is not nil: %+v\", respErr)\n\t\treturn respErr\n\t}\n\n\treturn nil\n}\n"
},
{
"path": "relay/controller/proxy.go",
"content": "// Package controller is a package for handling the relay controller\npackage controller\n\nimport (\n\t\"fmt\"\n\t\"net/http\"\n\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/relay\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\trelaymodel \"github.com/songquanpeng/one-api/relay/model\"\n)\n\n// RelayProxyHelper is a helper function to proxy the request to the upstream service\nfunc RelayProxyHelper(c *gin.Context, relayMode int) *relaymodel.ErrorWithStatusCode {\n\tctx := c.Request.Context()\n\tmeta := meta.GetByContext(c)\n\n\tadaptor := relay.GetAdaptor(meta.APIType)\n\tif adaptor == nil {\n\t\treturn openai.ErrorWrapper(fmt.Errorf(\"invalid api type: %d\", meta.APIType), \"invalid_api_type\", http.StatusBadRequest)\n\t}\n\tadaptor.Init(meta)\n\n\tresp, err := adaptor.DoRequest(c, meta, c.Request.Body)\n\tif err != nil {\n\t\tlogger.Errorf(ctx, \"DoRequest failed: %s\", err.Error())\n\t\treturn openai.ErrorWrapper(err, \"do_request_failed\", http.StatusInternalServerError)\n\t}\n\n\t// do response\n\t_, respErr := adaptor.DoResponse(c, resp, meta)\n\tif respErr != nil {\n\t\tlogger.Errorf(ctx, \"respErr is not nil: %+v\", respErr)\n\t\treturn respErr\n\t}\n\n\treturn nil\n}\n"
},
{
"path": "relay/controller/text.go",
"content": "package controller\n\nimport (\n\t\"bytes\"\n\t\"encoding/json\"\n\t\"fmt\"\n\t\"io\"\n\t\"net/http\"\n\n\t\"github.com/gin-gonic/gin\"\n\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"github.com/songquanpeng/one-api/relay\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor\"\n\t\"github.com/songquanpeng/one-api/relay/adaptor/openai\"\n\t\"github.com/songquanpeng/one-api/relay/apitype\"\n\t\"github.com/songquanpeng/one-api/relay/billing\"\n\tbillingratio \"github.com/songquanpeng/one-api/relay/billing/ratio\"\n\t\"github.com/songquanpeng/one-api/relay/channeltype\"\n\t\"github.com/songquanpeng/one-api/relay/meta\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n)\n\nfunc RelayTextHelper(c *gin.Context) *model.ErrorWithStatusCode {\n\tctx := c.Request.Context()\n\tmeta := meta.GetByContext(c)\n\t// get & validate textRequest\n\ttextRequest, err := getAndValidateTextRequest(c, meta.Mode)\n\tif err != nil {\n\t\tlogger.Errorf(ctx, \"getAndValidateTextRequest failed: %s\", err.Error())\n\t\treturn openai.ErrorWrapper(err, \"invalid_text_request\", http.StatusBadRequest)\n\t}\n\tmeta.IsStream = textRequest.Stream\n\n\t// map model name\n\tmeta.OriginModelName = textRequest.Model\n\ttextRequest.Model, _ = getMappedModelName(textRequest.Model, meta.ModelMapping)\n\tmeta.ActualModelName = textRequest.Model\n\t// set system prompt if not empty\n\tsystemPromptReset := setSystemPrompt(ctx, textRequest, meta.ForcedSystemPrompt)\n\t// get model ratio & group ratio\n\tmodelRatio := billingratio.GetModelRatio(textRequest.Model, meta.ChannelType)\n\tgroupRatio := billingratio.GetGroupRatio(meta.Group)\n\tratio := modelRatio * groupRatio\n\t// pre-consume quota\n\tpromptTokens := getPromptTokens(textRequest, meta.Mode)\n\tmeta.PromptTokens = promptTokens\n\tpreConsumedQuota, bizErr := preConsumeQuota(ctx, textRequest, promptTokens, ratio, meta)\n\tif bizErr != nil {\n\t\tlogger.Warnf(ctx, \"preConsumeQuota failed: %+v\", *bizErr)\n\t\treturn bizErr\n\t}\n\n\tadaptor := relay.GetAdaptor(meta.APIType)\n\tif adaptor == nil {\n\t\treturn openai.ErrorWrapper(fmt.Errorf(\"invalid api type: %d\", meta.APIType), \"invalid_api_type\", http.StatusBadRequest)\n\t}\n\tadaptor.Init(meta)\n\n\t// get request body\n\trequestBody, err := getRequestBody(c, meta, textRequest, adaptor)\n\tif err != nil {\n\t\treturn openai.ErrorWrapper(err, \"convert_request_failed\", http.StatusInternalServerError)\n\t}\n\n\t// do request\n\tresp, err := adaptor.DoRequest(c, meta, requestBody)\n\tif err != nil {\n\t\tlogger.Errorf(ctx, \"DoRequest failed: %s\", err.Error())\n\t\treturn openai.ErrorWrapper(err, \"do_request_failed\", http.StatusInternalServerError)\n\t}\n\tif isErrorHappened(meta, resp) {\n\t\tbilling.ReturnPreConsumedQuota(ctx, preConsumedQuota, meta.TokenId)\n\t\treturn RelayErrorHandler(resp)\n\t}\n\n\t// do response\n\tusage, respErr := adaptor.DoResponse(c, resp, meta)\n\tif respErr != nil {\n\t\tlogger.Errorf(ctx, \"respErr is not nil: %+v\", respErr)\n\t\tbilling.ReturnPreConsumedQuota(ctx, preConsumedQuota, meta.TokenId)\n\t\treturn respErr\n\t}\n\t// post-consume quota\n\tgo postConsumeQuota(ctx, usage, meta, textRequest, ratio, preConsumedQuota, modelRatio, groupRatio, systemPromptReset)\n\treturn nil\n}\n\nfunc getRequestBody(c *gin.Context, meta *meta.Meta, textRequest *model.GeneralOpenAIRequest, adaptor adaptor.Adaptor) (io.Reader, error) {\n\tif !config.EnforceIncludeUsage &&\n\t\tmeta.APIType == apitype.OpenAI &&\n\t\tmeta.OriginModelName == meta.ActualModelName &&\n\t\tmeta.ChannelType != channeltype.Baichuan &&\n\t\tmeta.ForcedSystemPrompt == \"\" {\n\t\t// no need to convert request for openai\n\t\treturn c.Request.Body, nil\n\t}\n\n\t// get request body\n\tvar requestBody io.Reader\n\tconvertedRequest, err := adaptor.ConvertRequest(c, meta.Mode, textRequest)\n\tif err != nil {\n\t\tlogger.Debugf(c.Request.Context(), \"converted request failed: %s\\n\", err.Error())\n\t\treturn nil, err\n\t}\n\tjsonData, err := json.Marshal(convertedRequest)\n\tif err != nil {\n\t\tlogger.Debugf(c.Request.Context(), \"converted request json_marshal_failed: %s\\n\", err.Error())\n\t\treturn nil, err\n\t}\n\tlogger.Debugf(c.Request.Context(), \"converted request: \\n%s\", string(jsonData))\n\trequestBody = bytes.NewBuffer(jsonData)\n\treturn requestBody, nil\n}\n"
},
{
"path": "relay/controller/validator/validation.go",
"content": "package validator\n\nimport (\n\t\"errors\"\n\t\"github.com/songquanpeng/one-api/relay/model\"\n\t\"github.com/songquanpeng/one-api/relay/relaymode\"\n\t\"math\"\n)\n\nfunc ValidateTextRequest(textRequest *model.GeneralOpenAIRequest, relayMode int) error {\n\tif textRequest.MaxTokens < 0 || textRequest.MaxTokens > math.MaxInt32/2 {\n\t\treturn errors.New(\"max_tokens is invalid\")\n\t}\n\tif textRequest.Model == \"\" {\n\t\treturn errors.New(\"model is required\")\n\t}\n\tswitch relayMode {\n\tcase relaymode.Completions:\n\t\tif textRequest.Prompt == \"\" {\n\t\t\treturn errors.New(\"field prompt is required\")\n\t\t}\n\tcase relaymode.ChatCompletions:\n\t\tif textRequest.Messages == nil || len(textRequest.Messages) == 0 {\n\t\t\treturn errors.New(\"field messages is required\")\n\t\t}\n\tcase relaymode.Embeddings:\n\tcase relaymode.Moderations:\n\t\tif textRequest.Input == \"\" {\n\t\t\treturn errors.New(\"field input is required\")\n\t\t}\n\tcase relaymode.Edits:\n\t\tif textRequest.Instruction == \"\" {\n\t\t\treturn errors.New(\"field instruction is required\")\n\t\t}\n\t}\n\treturn nil\n}\n"
},
{
"path": "relay/meta/relay_meta.go",
"content": "package meta\n\nimport (\n\t\"strings\"\n\t\"time\"\n\n\t\"github.com/gin-gonic/gin\"\n\n\t\"github.com/songquanpeng/one-api/common/ctxkey\"\n\t\"github.com/songquanpeng/one-api/model\"\n\t\"github.com/songquanpeng/one-api/relay/channeltype\"\n\t\"github.com/songquanpeng/one-api/relay/relaymode\"\n)\n\ntype Meta struct {\n\tMode int\n\tChannelType int\n\tChannelId int\n\tTokenId int\n\tTokenName string\n\tUserId int\n\tGroup string\n\tModelMapping map[string]string\n\t// BaseURL is the proxy url set in the channel config\n\tBaseURL string\n\tAPIKey string\n\tAPIType int\n\tConfig model.ChannelConfig\n\tIsStream bool\n\t// OriginModelName is the model name from the raw user request\n\tOriginModelName string\n\t// ActualModelName is the model name after mapping\n\tActualModelName string\n\tRequestURLPath string\n\tPromptTokens int // only for DoResponse\n\tForcedSystemPrompt string\n\tStartTime time.Time\n}\n\nfunc GetByContext(c *gin.Context) *Meta {\n\tmeta := Meta{\n\t\tMode: relaymode.GetByPath(c.Request.URL.Path),\n\t\tChannelType: c.GetInt(ctxkey.Channel),\n\t\tChannelId: c.GetInt(ctxkey.ChannelId),\n\t\tTokenId: c.GetInt(ctxkey.TokenId),\n\t\tTokenName: c.GetString(ctxkey.TokenName),\n\t\tUserId: c.GetInt(ctxkey.Id),\n\t\tGroup: c.GetString(ctxkey.Group),\n\t\tModelMapping: c.GetStringMapString(ctxkey.ModelMapping),\n\t\tOriginModelName: c.GetString(ctxkey.RequestModel),\n\t\tBaseURL: c.GetString(ctxkey.BaseURL),\n\t\tAPIKey: strings.TrimPrefix(c.Request.Header.Get(\"Authorization\"), \"Bearer \"),\n\t\tRequestURLPath: c.Request.URL.String(),\n\t\tForcedSystemPrompt: c.GetString(ctxkey.SystemPrompt),\n\t\tStartTime: time.Now(),\n\t}\n\tcfg, ok := c.Get(ctxkey.Config)\n\tif ok {\n\t\tmeta.Config = cfg.(model.ChannelConfig)\n\t}\n\tif meta.BaseURL == \"\" {\n\t\tmeta.BaseURL = channeltype.ChannelBaseURLs[meta.ChannelType]\n\t}\n\tmeta.APIType = channeltype.ToAPIType(meta.ChannelType)\n\treturn &meta\n}\n"
},
{
"path": "relay/model/constant.go",
"content": "package model\n\nconst (\n\tContentTypeText = \"text\"\n\tContentTypeImageURL = \"image_url\"\n\tContentTypeInputAudio = \"input_audio\"\n)\n"
},
{
"path": "relay/model/general.go",
"content": "package model\n\ntype ResponseFormat struct {\n\tType string `json:\"type,omitempty\"`\n\tJsonSchema *JSONSchema `json:\"json_schema,omitempty\"`\n}\n\ntype JSONSchema struct {\n\tDescription string `json:\"description,omitempty\"`\n\tName string `json:\"name\"`\n\tSchema map[string]interface{} `json:\"schema,omitempty\"`\n\tStrict *bool `json:\"strict,omitempty\"`\n}\n\ntype Audio struct {\n\tVoice string `json:\"voice,omitempty\"`\n\tFormat string `json:\"format,omitempty\"`\n}\n\ntype StreamOptions struct {\n\tIncludeUsage bool `json:\"include_usage,omitempty\"`\n}\n\ntype GeneralOpenAIRequest struct {\n\t// https://platform.openai.com/docs/api-reference/chat/create\n\tMessages []Message `json:\"messages,omitempty\"`\n\tModel string `json:\"model,omitempty\"`\n\tStore *bool `json:\"store,omitempty\"`\n\tReasoningEffort *string `json:\"reasoning_effort,omitempty\"`\n\tMetadata any `json:\"metadata,omitempty\"`\n\tFrequencyPenalty *float64 `json:\"frequency_penalty,omitempty\"`\n\tLogitBias any `json:\"logit_bias,omitempty\"`\n\tLogprobs *bool `json:\"logprobs,omitempty\"`\n\tTopLogprobs *int `json:\"top_logprobs,omitempty\"`\n\tMaxTokens int `json:\"max_tokens,omitempty\"`\n\tMaxCompletionTokens *int `json:\"max_completion_tokens,omitempty\"`\n\tN int `json:\"n,omitempty\"`\n\tModalities []string `json:\"modalities,omitempty\"`\n\tPrediction any `json:\"prediction,omitempty\"`\n\tAudio *Audio `json:\"audio,omitempty\"`\n\tPresencePenalty *float64 `json:\"presence_penalty,omitempty\"`\n\tResponseFormat *ResponseFormat `json:\"response_format,omitempty\"`\n\tSeed float64 `json:\"seed,omitempty\"`\n\tServiceTier *string `json:\"service_tier,omitempty\"`\n\tStop any `json:\"stop,omitempty\"`\n\tStream bool `json:\"stream,omitempty\"`\n\tStreamOptions *StreamOptions `json:\"stream_options,omitempty\"`\n\tTemperature *float64 `json:\"temperature,omitempty\"`\n\tTopP *float64 `json:\"top_p,omitempty\"`\n\tTopK int `json:\"top_k,omitempty\"`\n\tTools []Tool `json:\"tools,omitempty\"`\n\tToolChoice any `json:\"tool_choice,omitempty\"`\n\tParallelTooCalls *bool `json:\"parallel_tool_calls,omitempty\"`\n\tUser string `json:\"user,omitempty\"`\n\tFunctionCall any `json:\"function_call,omitempty\"`\n\tFunctions any `json:\"functions,omitempty\"`\n\t// https://platform.openai.com/docs/api-reference/embeddings/create\n\tInput any `json:\"input,omitempty\"`\n\tEncodingFormat string `json:\"encoding_format,omitempty\"`\n\tDimensions int `json:\"dimensions,omitempty\"`\n\t// https://platform.openai.com/docs/api-reference/images/create\n\tPrompt any `json:\"prompt,omitempty\"`\n\tQuality *string `json:\"quality,omitempty\"`\n\tSize string `json:\"size,omitempty\"`\n\tStyle *string `json:\"style,omitempty\"`\n\t// Others\n\tInstruction string `json:\"instruction,omitempty\"`\n\tNumCtx int `json:\"num_ctx,omitempty\"`\n}\n\nfunc (r GeneralOpenAIRequest) ParseInput() []string {\n\tif r.Input == nil {\n\t\treturn nil\n\t}\n\tvar input []string\n\tswitch r.Input.(type) {\n\tcase string:\n\t\tinput = []string{r.Input.(string)}\n\tcase []any:\n\t\tinput = make([]string, 0, len(r.Input.([]any)))\n\t\tfor _, item := range r.Input.([]any) {\n\t\t\tif str, ok := item.(string); ok {\n\t\t\t\tinput = append(input, str)\n\t\t\t}\n\t\t}\n\t}\n\treturn input\n}\n"
},
{
"path": "relay/model/image.go",
"content": "package model\n\ntype ImageRequest struct {\n\tModel string `json:\"model\"`\n\tPrompt string `json:\"prompt\" binding:\"required\"`\n\tN int `json:\"n,omitempty\"`\n\tSize string `json:\"size,omitempty\"`\n\tQuality string `json:\"quality,omitempty\"`\n\tResponseFormat string `json:\"response_format,omitempty\"`\n\tStyle string `json:\"style,omitempty\"`\n\tUser string `json:\"user,omitempty\"`\n}\n"
},
{
"path": "relay/model/message.go",
"content": "package model\n\ntype Message struct {\n\tRole string `json:\"role,omitempty\"`\n\tContent any `json:\"content,omitempty\"`\n\tReasoningContent any `json:\"reasoning_content,omitempty\"`\n\tName *string `json:\"name,omitempty\"`\n\tToolCalls []Tool `json:\"tool_calls,omitempty\"`\n\tToolCallId string `json:\"tool_call_id,omitempty\"`\n}\n\nfunc (m Message) IsStringContent() bool {\n\t_, ok := m.Content.(string)\n\treturn ok\n}\n\nfunc (m Message) StringContent() string {\n\tcontent, ok := m.Content.(string)\n\tif ok {\n\t\treturn content\n\t}\n\tcontentList, ok := m.Content.([]any)\n\tif ok {\n\t\tvar contentStr string\n\t\tfor _, contentItem := range contentList {\n\t\t\tcontentMap, ok := contentItem.(map[string]any)\n\t\t\tif !ok {\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tif contentMap[\"type\"] == ContentTypeText {\n\t\t\t\tif subStr, ok := contentMap[\"text\"].(string); ok {\n\t\t\t\t\tcontentStr += subStr\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t\treturn contentStr\n\t}\n\treturn \"\"\n}\n\nfunc (m Message) ParseContent() []MessageContent {\n\tvar contentList []MessageContent\n\tcontent, ok := m.Content.(string)\n\tif ok {\n\t\tcontentList = append(contentList, MessageContent{\n\t\t\tType: ContentTypeText,\n\t\t\tText: content,\n\t\t})\n\t\treturn contentList\n\t}\n\tanyList, ok := m.Content.([]any)\n\tif ok {\n\t\tfor _, contentItem := range anyList {\n\t\t\tcontentMap, ok := contentItem.(map[string]any)\n\t\t\tif !ok {\n\t\t\t\tcontinue\n\t\t\t}\n\t\t\tswitch contentMap[\"type\"] {\n\t\t\tcase ContentTypeText:\n\t\t\t\tif subStr, ok := contentMap[\"text\"].(string); ok {\n\t\t\t\t\tcontentList = append(contentList, MessageContent{\n\t\t\t\t\t\tType: ContentTypeText,\n\t\t\t\t\t\tText: subStr,\n\t\t\t\t\t})\n\t\t\t\t}\n\t\t\tcase ContentTypeImageURL:\n\t\t\t\tif subObj, ok := contentMap[\"image_url\"].(map[string]any); ok {\n\t\t\t\t\tcontentList = append(contentList, MessageContent{\n\t\t\t\t\t\tType: ContentTypeImageURL,\n\t\t\t\t\t\tImageURL: &ImageURL{\n\t\t\t\t\t\t\tUrl: subObj[\"url\"].(string),\n\t\t\t\t\t\t},\n\t\t\t\t\t})\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t\treturn contentList\n\t}\n\treturn nil\n}\n\ntype ImageURL struct {\n\tUrl string `json:\"url,omitempty\"`\n\tDetail string `json:\"detail,omitempty\"`\n}\n\ntype MessageContent struct {\n\tType string `json:\"type,omitempty\"`\n\tText string `json:\"text\"`\n\tImageURL *ImageURL `json:\"image_url,omitempty\"`\n}\n"
},
{
"path": "relay/model/misc.go",
"content": "package model\n\ntype Usage struct {\n\tPromptTokens int `json:\"prompt_tokens\"`\n\tCompletionTokens int `json:\"completion_tokens\"`\n\tTotalTokens int `json:\"total_tokens\"`\n\n\tCompletionTokensDetails *CompletionTokensDetails `json:\"completion_tokens_details,omitempty\"`\n}\n\ntype CompletionTokensDetails struct {\n\tReasoningTokens int `json:\"reasoning_tokens\"`\n\tAcceptedPredictionTokens int `json:\"accepted_prediction_tokens\"`\n\tRejectedPredictionTokens int `json:\"rejected_prediction_tokens\"`\n}\n\ntype Error struct {\n\tMessage string `json:\"message\"`\n\tType string `json:\"type\"`\n\tParam string `json:\"param\"`\n\tCode any `json:\"code\"`\n}\n\ntype ErrorWithStatusCode struct {\n\tError\n\tStatusCode int `json:\"status_code\"`\n}\n"
},
{
"path": "relay/model/tool.go",
"content": "package model\n\ntype Tool struct {\n\tId string `json:\"id,omitempty\"`\n\tType string `json:\"type,omitempty\"` // when splicing claude tools stream messages, it is empty\n\tFunction Function `json:\"function\"`\n}\n\ntype Function struct {\n\tDescription string `json:\"description,omitempty\"`\n\tName string `json:\"name,omitempty\"` // when splicing claude tools stream messages, it is empty\n\tParameters any `json:\"parameters,omitempty\"` // request\n\tArguments any `json:\"arguments,omitempty\"` // response\n}\n"
},
{
"path": "relay/relaymode/define.go",
"content": "package relaymode\n\nconst (\n\tUnknown = iota\n\tChatCompletions\n\tCompletions\n\tEmbeddings\n\tModerations\n\tImagesGenerations\n\tEdits\n\tAudioSpeech\n\tAudioTranscription\n\tAudioTranslation\n\t// Proxy is a special relay mode for proxying requests to custom upstream\n\tProxy\n)\n"
},
{
"path": "relay/relaymode/helper.go",
"content": "package relaymode\n\nimport \"strings\"\n\nfunc GetByPath(path string) int {\n\trelayMode := Unknown\n\tif strings.HasPrefix(path, \"/v1/chat/completions\") {\n\t\trelayMode = ChatCompletions\n\t} else if strings.HasPrefix(path, \"/v1/completions\") {\n\t\trelayMode = Completions\n\t} else if strings.HasPrefix(path, \"/v1/embeddings\") {\n\t\trelayMode = Embeddings\n\t} else if strings.HasSuffix(path, \"embeddings\") {\n\t\trelayMode = Embeddings\n\t} else if strings.HasPrefix(path, \"/v1/moderations\") {\n\t\trelayMode = Moderations\n\t} else if strings.HasPrefix(path, \"/v1/images/generations\") {\n\t\trelayMode = ImagesGenerations\n\t} else if strings.HasPrefix(path, \"/v1/edits\") {\n\t\trelayMode = Edits\n\t} else if strings.HasPrefix(path, \"/v1/audio/speech\") {\n\t\trelayMode = AudioSpeech\n\t} else if strings.HasPrefix(path, \"/v1/audio/transcriptions\") {\n\t\trelayMode = AudioTranscription\n\t} else if strings.HasPrefix(path, \"/v1/audio/translations\") {\n\t\trelayMode = AudioTranslation\n\t} else if strings.HasPrefix(path, \"/v1/oneapi/proxy\") {\n\t\trelayMode = Proxy\n\t}\n\treturn relayMode\n}\n"
},
{
"path": "router/api.go",
"content": "package router\n\nimport (\n\t\"github.com/songquanpeng/one-api/controller\"\n\t\"github.com/songquanpeng/one-api/controller/auth\"\n\t\"github.com/songquanpeng/one-api/middleware\"\n\n\t\"github.com/gin-contrib/gzip\"\n\t\"github.com/gin-gonic/gin\"\n)\n\nfunc SetApiRouter(router *gin.Engine) {\n\tapiRouter := router.Group(\"/api\")\n\tapiRouter.Use(gzip.Gzip(gzip.DefaultCompression))\n\tapiRouter.Use(middleware.GlobalAPIRateLimit())\n\t{\n\t\tapiRouter.GET(\"/status\", controller.GetStatus)\n\t\tapiRouter.GET(\"/models\", middleware.UserAuth(), controller.DashboardListModels)\n\t\tapiRouter.GET(\"/notice\", controller.GetNotice)\n\t\tapiRouter.GET(\"/about\", controller.GetAbout)\n\t\tapiRouter.GET(\"/home_page_content\", controller.GetHomePageContent)\n\t\tapiRouter.GET(\"/verification\", middleware.CriticalRateLimit(), middleware.TurnstileCheck(), controller.SendEmailVerification)\n\t\tapiRouter.GET(\"/reset_password\", middleware.CriticalRateLimit(), middleware.TurnstileCheck(), controller.SendPasswordResetEmail)\n\t\tapiRouter.POST(\"/user/reset\", middleware.CriticalRateLimit(), controller.ResetPassword)\n\t\tapiRouter.GET(\"/oauth/github\", middleware.CriticalRateLimit(), auth.GitHubOAuth)\n\t\tapiRouter.GET(\"/oauth/oidc\", middleware.CriticalRateLimit(), auth.OidcAuth)\n\t\tapiRouter.GET(\"/oauth/lark\", middleware.CriticalRateLimit(), auth.LarkOAuth)\n\t\tapiRouter.GET(\"/oauth/state\", middleware.CriticalRateLimit(), auth.GenerateOAuthCode)\n\t\tapiRouter.GET(\"/oauth/wechat\", middleware.CriticalRateLimit(), auth.WeChatAuth)\n\t\tapiRouter.GET(\"/oauth/wechat/bind\", middleware.CriticalRateLimit(), middleware.UserAuth(), auth.WeChatBind)\n\t\tapiRouter.GET(\"/oauth/email/bind\", middleware.CriticalRateLimit(), middleware.UserAuth(), controller.EmailBind)\n\t\tapiRouter.POST(\"/topup\", middleware.AdminAuth(), controller.AdminTopUp)\n\n\t\tuserRoute := apiRouter.Group(\"/user\")\n\t\t{\n\t\t\tuserRoute.POST(\"/register\", middleware.CriticalRateLimit(), middleware.TurnstileCheck(), controller.Register)\n\t\t\tuserRoute.POST(\"/login\", middleware.CriticalRateLimit(), controller.Login)\n\t\t\tuserRoute.GET(\"/logout\", controller.Logout)\n\n\t\t\tselfRoute := userRoute.Group(\"/\")\n\t\t\tselfRoute.Use(middleware.UserAuth())\n\t\t\t{\n\t\t\t\tselfRoute.GET(\"/dashboard\", controller.GetUserDashboard)\n\t\t\t\tselfRoute.GET(\"/self\", controller.GetSelf)\n\t\t\t\tselfRoute.PUT(\"/self\", controller.UpdateSelf)\n\t\t\t\tselfRoute.DELETE(\"/self\", controller.DeleteSelf)\n\t\t\t\tselfRoute.GET(\"/token\", controller.GenerateAccessToken)\n\t\t\t\tselfRoute.GET(\"/aff\", controller.GetAffCode)\n\t\t\t\tselfRoute.POST(\"/topup\", controller.TopUp)\n\t\t\t\tselfRoute.GET(\"/available_models\", controller.GetUserAvailableModels)\n\t\t\t}\n\n\t\t\tadminRoute := userRoute.Group(\"/\")\n\t\t\tadminRoute.Use(middleware.AdminAuth())\n\t\t\t{\n\t\t\t\tadminRoute.GET(\"/\", controller.GetAllUsers)\n\t\t\t\tadminRoute.GET(\"/search\", controller.SearchUsers)\n\t\t\t\tadminRoute.GET(\"/:id\", controller.GetUser)\n\t\t\t\tadminRoute.POST(\"/\", controller.CreateUser)\n\t\t\t\tadminRoute.POST(\"/manage\", controller.ManageUser)\n\t\t\t\tadminRoute.PUT(\"/\", controller.UpdateUser)\n\t\t\t\tadminRoute.DELETE(\"/:id\", controller.DeleteUser)\n\t\t\t}\n\t\t}\n\t\toptionRoute := apiRouter.Group(\"/option\")\n\t\toptionRoute.Use(middleware.RootAuth())\n\t\t{\n\t\t\toptionRoute.GET(\"/\", controller.GetOptions)\n\t\t\toptionRoute.PUT(\"/\", controller.UpdateOption)\n\t\t}\n\t\tchannelRoute := apiRouter.Group(\"/channel\")\n\t\tchannelRoute.Use(middleware.AdminAuth())\n\t\t{\n\t\t\tchannelRoute.GET(\"/\", controller.GetAllChannels)\n\t\t\tchannelRoute.GET(\"/search\", controller.SearchChannels)\n\t\t\tchannelRoute.GET(\"/models\", controller.ListAllModels)\n\t\t\tchannelRoute.GET(\"/:id\", controller.GetChannel)\n\t\t\tchannelRoute.GET(\"/test\", controller.TestChannels)\n\t\t\tchannelRoute.GET(\"/test/:id\", controller.TestChannel)\n\t\t\tchannelRoute.GET(\"/update_balance\", controller.UpdateAllChannelsBalance)\n\t\t\tchannelRoute.GET(\"/update_balance/:id\", controller.UpdateChannelBalance)\n\t\t\tchannelRoute.POST(\"/\", controller.AddChannel)\n\t\t\tchannelRoute.PUT(\"/\", controller.UpdateChannel)\n\t\t\tchannelRoute.DELETE(\"/disabled\", controller.DeleteDisabledChannel)\n\t\t\tchannelRoute.DELETE(\"/:id\", controller.DeleteChannel)\n\t\t}\n\t\ttokenRoute := apiRouter.Group(\"/token\")\n\t\ttokenRoute.Use(middleware.UserAuth())\n\t\t{\n\t\t\ttokenRoute.GET(\"/\", controller.GetAllTokens)\n\t\t\ttokenRoute.GET(\"/search\", controller.SearchTokens)\n\t\t\ttokenRoute.GET(\"/:id\", controller.GetToken)\n\t\t\ttokenRoute.POST(\"/\", controller.AddToken)\n\t\t\ttokenRoute.PUT(\"/\", controller.UpdateToken)\n\t\t\ttokenRoute.DELETE(\"/:id\", controller.DeleteToken)\n\t\t}\n\t\tredemptionRoute := apiRouter.Group(\"/redemption\")\n\t\tredemptionRoute.Use(middleware.AdminAuth())\n\t\t{\n\t\t\tredemptionRoute.GET(\"/\", controller.GetAllRedemptions)\n\t\t\tredemptionRoute.GET(\"/search\", controller.SearchRedemptions)\n\t\t\tredemptionRoute.GET(\"/:id\", controller.GetRedemption)\n\t\t\tredemptionRoute.POST(\"/\", controller.AddRedemption)\n\t\t\tredemptionRoute.PUT(\"/\", controller.UpdateRedemption)\n\t\t\tredemptionRoute.DELETE(\"/:id\", controller.DeleteRedemption)\n\t\t}\n\t\tlogRoute := apiRouter.Group(\"/log\")\n\t\tlogRoute.GET(\"/\", middleware.AdminAuth(), controller.GetAllLogs)\n\t\tlogRoute.DELETE(\"/\", middleware.AdminAuth(), controller.DeleteHistoryLogs)\n\t\tlogRoute.GET(\"/stat\", middleware.AdminAuth(), controller.GetLogsStat)\n\t\tlogRoute.GET(\"/self/stat\", middleware.UserAuth(), controller.GetLogsSelfStat)\n\t\tlogRoute.GET(\"/search\", middleware.AdminAuth(), controller.SearchAllLogs)\n\t\tlogRoute.GET(\"/self\", middleware.UserAuth(), controller.GetUserLogs)\n\t\tlogRoute.GET(\"/self/search\", middleware.UserAuth(), controller.SearchUserLogs)\n\t\tgroupRoute := apiRouter.Group(\"/group\")\n\t\tgroupRoute.Use(middleware.AdminAuth())\n\t\t{\n\t\t\tgroupRoute.GET(\"/\", controller.GetGroups)\n\t\t}\n\t}\n}\n"
},
{
"path": "router/dashboard.go",
"content": "package router\n\nimport (\n\t\"github.com/gin-contrib/gzip\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/controller\"\n\t\"github.com/songquanpeng/one-api/middleware\"\n)\n\nfunc SetDashboardRouter(router *gin.Engine) {\n\tapiRouter := router.Group(\"/\")\n\tapiRouter.Use(middleware.CORS())\n\tapiRouter.Use(gzip.Gzip(gzip.DefaultCompression))\n\tapiRouter.Use(middleware.GlobalAPIRateLimit())\n\tapiRouter.Use(middleware.TokenAuth())\n\t{\n\t\tapiRouter.GET(\"/dashboard/billing/subscription\", controller.GetSubscription)\n\t\tapiRouter.GET(\"/v1/dashboard/billing/subscription\", controller.GetSubscription)\n\t\tapiRouter.GET(\"/dashboard/billing/usage\", controller.GetUsage)\n\t\tapiRouter.GET(\"/v1/dashboard/billing/usage\", controller.GetUsage)\n\t}\n}\n"
},
{
"path": "router/main.go",
"content": "package router\n\nimport (\n\t\"embed\"\n\t\"fmt\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/common/logger\"\n\t\"net/http\"\n\t\"os\"\n\t\"strings\"\n)\n\nfunc SetRouter(router *gin.Engine, buildFS embed.FS) {\n\tSetApiRouter(router)\n\tSetDashboardRouter(router)\n\tSetRelayRouter(router)\n\tfrontendBaseUrl := os.Getenv(\"FRONTEND_BASE_URL\")\n\tif config.IsMasterNode && frontendBaseUrl != \"\" {\n\t\tfrontendBaseUrl = \"\"\n\t\tlogger.SysLog(\"FRONTEND_BASE_URL is ignored on master node\")\n\t}\n\tif frontendBaseUrl == \"\" {\n\t\tSetWebRouter(router, buildFS)\n\t} else {\n\t\tfrontendBaseUrl = strings.TrimSuffix(frontendBaseUrl, \"/\")\n\t\trouter.NoRoute(func(c *gin.Context) {\n\t\t\tc.Redirect(http.StatusMovedPermanently, fmt.Sprintf(\"%s%s\", frontendBaseUrl, c.Request.RequestURI))\n\t\t})\n\t}\n}\n"
},
{
"path": "router/relay.go",
"content": "package router\n\nimport (\n\t\"github.com/songquanpeng/one-api/controller\"\n\t\"github.com/songquanpeng/one-api/middleware\"\n\n\t\"github.com/gin-gonic/gin\"\n)\n\nfunc SetRelayRouter(router *gin.Engine) {\n\trouter.Use(middleware.CORS())\n\trouter.Use(middleware.GzipDecodeMiddleware())\n\t// https://platform.openai.com/docs/api-reference/introduction\n\tmodelsRouter := router.Group(\"/v1/models\")\n\tmodelsRouter.Use(middleware.TokenAuth())\n\t{\n\t\tmodelsRouter.GET(\"\", controller.ListModels)\n\t\tmodelsRouter.GET(\"/:model\", controller.RetrieveModel)\n\t}\n\trelayV1Router := router.Group(\"/v1\")\n\trelayV1Router.Use(middleware.RelayPanicRecover(), middleware.TokenAuth(), middleware.Distribute())\n\t{\n\t\trelayV1Router.Any(\"/oneapi/proxy/:channelid/*target\", controller.Relay)\n\t\trelayV1Router.POST(\"/completions\", controller.Relay)\n\t\trelayV1Router.POST(\"/chat/completions\", controller.Relay)\n\t\trelayV1Router.POST(\"/edits\", controller.Relay)\n\t\trelayV1Router.POST(\"/images/generations\", controller.Relay)\n\t\trelayV1Router.POST(\"/images/edits\", controller.RelayNotImplemented)\n\t\trelayV1Router.POST(\"/images/variations\", controller.RelayNotImplemented)\n\t\trelayV1Router.POST(\"/embeddings\", controller.Relay)\n\t\trelayV1Router.POST(\"/engines/:model/embeddings\", controller.Relay)\n\t\trelayV1Router.POST(\"/audio/transcriptions\", controller.Relay)\n\t\trelayV1Router.POST(\"/audio/translations\", controller.Relay)\n\t\trelayV1Router.POST(\"/audio/speech\", controller.Relay)\n\t\trelayV1Router.GET(\"/files\", controller.RelayNotImplemented)\n\t\trelayV1Router.POST(\"/files\", controller.RelayNotImplemented)\n\t\trelayV1Router.DELETE(\"/files/:id\", controller.RelayNotImplemented)\n\t\trelayV1Router.GET(\"/files/:id\", controller.RelayNotImplemented)\n\t\trelayV1Router.GET(\"/files/:id/content\", controller.RelayNotImplemented)\n\t\trelayV1Router.POST(\"/fine_tuning/jobs\", controller.RelayNotImplemented)\n\t\trelayV1Router.GET(\"/fine_tuning/jobs\", controller.RelayNotImplemented)\n\t\trelayV1Router.GET(\"/fine_tuning/jobs/:id\", controller.RelayNotImplemented)\n\t\trelayV1Router.POST(\"/fine_tuning/jobs/:id/cancel\", controller.RelayNotImplemented)\n\t\trelayV1Router.GET(\"/fine_tuning/jobs/:id/events\", controller.RelayNotImplemented)\n\t\trelayV1Router.DELETE(\"/models/:model\", controller.RelayNotImplemented)\n\t\trelayV1Router.POST(\"/moderations\", controller.Relay)\n\t\trelayV1Router.POST(\"/assistants\", controller.RelayNotImplemented)\n\t\trelayV1Router.GET(\"/assistants/:id\", controller.RelayNotImplemented)\n\t\trelayV1Router.POST(\"/assistants/:id\", controller.RelayNotImplemented)\n\t\trelayV1Router.DELETE(\"/assistants/:id\", controller.RelayNotImplemented)\n\t\trelayV1Router.GET(\"/assistants\", controller.RelayNotImplemented)\n\t\trelayV1Router.POST(\"/assistants/:id/files\", controller.RelayNotImplemented)\n\t\trelayV1Router.GET(\"/assistants/:id/files/:fileId\", controller.RelayNotImplemented)\n\t\trelayV1Router.DELETE(\"/assistants/:id/files/:fileId\", controller.RelayNotImplemented)\n\t\trelayV1Router.GET(\"/assistants/:id/files\", controller.RelayNotImplemented)\n\t\trelayV1Router.POST(\"/threads\", controller.RelayNotImplemented)\n\t\trelayV1Router.GET(\"/threads/:id\", controller.RelayNotImplemented)\n\t\trelayV1Router.POST(\"/threads/:id\", controller.RelayNotImplemented)\n\t\trelayV1Router.DELETE(\"/threads/:id\", controller.RelayNotImplemented)\n\t\trelayV1Router.POST(\"/threads/:id/messages\", controller.RelayNotImplemented)\n\t\trelayV1Router.GET(\"/threads/:id/messages/:messageId\", controller.RelayNotImplemented)\n\t\trelayV1Router.POST(\"/threads/:id/messages/:messageId\", controller.RelayNotImplemented)\n\t\trelayV1Router.GET(\"/threads/:id/messages/:messageId/files/:filesId\", controller.RelayNotImplemented)\n\t\trelayV1Router.GET(\"/threads/:id/messages/:messageId/files\", controller.RelayNotImplemented)\n\t\trelayV1Router.POST(\"/threads/:id/runs\", controller.RelayNotImplemented)\n\t\trelayV1Router.GET(\"/threads/:id/runs/:runsId\", controller.RelayNotImplemented)\n\t\trelayV1Router.POST(\"/threads/:id/runs/:runsId\", controller.RelayNotImplemented)\n\t\trelayV1Router.GET(\"/threads/:id/runs\", controller.RelayNotImplemented)\n\t\trelayV1Router.POST(\"/threads/:id/runs/:runsId/submit_tool_outputs\", controller.RelayNotImplemented)\n\t\trelayV1Router.POST(\"/threads/:id/runs/:runsId/cancel\", controller.RelayNotImplemented)\n\t\trelayV1Router.GET(\"/threads/:id/runs/:runsId/steps/:stepId\", controller.RelayNotImplemented)\n\t\trelayV1Router.GET(\"/threads/:id/runs/:runsId/steps\", controller.RelayNotImplemented)\n\t}\n}\n"
},
{
"path": "router/web.go",
"content": "package router\n\nimport (\n\t\"embed\"\n\t\"fmt\"\n\t\"github.com/gin-contrib/gzip\"\n\t\"github.com/gin-contrib/static\"\n\t\"github.com/gin-gonic/gin\"\n\t\"github.com/songquanpeng/one-api/common\"\n\t\"github.com/songquanpeng/one-api/common/config\"\n\t\"github.com/songquanpeng/one-api/controller\"\n\t\"github.com/songquanpeng/one-api/middleware\"\n\t\"net/http\"\n\t\"strings\"\n)\n\nfunc SetWebRouter(router *gin.Engine, buildFS embed.FS) {\n\tindexPageData, _ := buildFS.ReadFile(fmt.Sprintf(\"web/build/%s/index.html\", config.Theme))\n\trouter.Use(gzip.Gzip(gzip.DefaultCompression))\n\trouter.Use(middleware.GlobalWebRateLimit())\n\trouter.Use(middleware.Cache())\n\trouter.Use(static.Serve(\"/\", common.EmbedFolder(buildFS, fmt.Sprintf(\"web/build/%s\", config.Theme))))\n\trouter.NoRoute(func(c *gin.Context) {\n\t\tif strings.HasPrefix(c.Request.RequestURI, \"/v1\") || strings.HasPrefix(c.Request.RequestURI, \"/api\") {\n\t\t\tcontroller.RelayNotFound(c)\n\t\t\treturn\n\t\t}\n\t\tc.Header(\"Cache-Control\", \"no-cache\")\n\t\tc.Data(http.StatusOK, \"text/html; charset=utf-8\", indexPageData)\n\t})\n}\n"
},
{
"path": "web/README.md",
"content": "# One API 的前端界面\n\n> 每个文件夹代表一个主题，欢迎提交你的主题\n\n> [!WARNING]\n> 不是每一个主题都及时同步了所有功能，由于精力有限，优先更新默认主题，其他主题欢迎 & 期待 PR\n\n## 提交新的主题\n\n> 欢迎在页面底部保留你和 One API 的版权信息以及指向链接\n\n1. 在 `web` 文件夹下新建一个文件夹，文件夹名为主题名。\n2. 把你的主题文件放到这个文件夹下。\n3. 修改你的 `package.json` 文件，把 `build` 命令改为：`\"build\": \"react-scripts build && mv -f build ../build/default\"`，其中 `default` 为你的主题名。\n4. 修改 `common/config/config.go` 中的 `ValidThemes`，把你的主题名称注册进去。\n5. 修改 `web/THEMES` 文件，这里也需要同步修改。\n\n## 主题列表\n\n### 主题：default\n\n默认主题，由 [JustSong](https://github.com/songquanpeng) 开发。\n\n预览：\n|||\n|:---:|:---:|\n\n### 主题：berry\n\n由 [MartialBE](https://github.com/MartialBE) 开发。\n\n预览：\n|||\n|:---:|:---:|\n|||\n|||\n|||\n\n### 主题：air\n由 [Calon](https://github.com/Calcium-Ion) 开发。\n|||\n|:---:|:---:|\n\n\n#### 开发说明\n\n请查看 [web/berry/README.md](https://github.com/songquanpeng/one-api/tree/main/web/berry/README.md)\n"
},
{
"path": "web/THEMES",
"content": "default\nberry\nair\n"
},
{
"path": "web/air/.gitignore",
"content": "# See https://help.github.com/articles/ignoring-files/ for more about ignoring files.\n\n# dependencies\n/node_modules\n/.pnp\n.pnp.js\n\n# testing\n/coverage\n\n# production\n/build\n\n# misc\n.DS_Store\n.env.local\n.env.development.local\n.env.test.local\n.env.production.local\n\nnpm-debug.log*\nyarn-debug.log*\nyarn-error.log*\n.idea\npackage-lock.json\nyarn.lock"
},
{
"path": "web/air/README.md",
"content": "# React Template\n\n## Basic Usages\n\n```shell\n# Runs the app in the development mode\nnpm start\n\n# Builds the app for production to the `build` folder\nnpm run build\n```\n\nIf you want to change the default server, please set `REACT_APP_SERVER` environment variables before build,\nfor example: `REACT_APP_SERVER=http://your.domain.com`.\n\nBefore you start editing, make sure your `Actions on Save` options have `Optimize imports` & `Run Prettier` enabled.\n\n## Reference\n\n1. https://github.com/OIerDb-ng/OIerDb\n2. https://github.com/cornflourblue/react-hooks-redux-registration-login-example"
},
{
"path": "web/air/package.json",
"content": "{\n \"name\": \"react-template\",\n \"version\": \"0.1.0\",\n \"private\": true,\n \"dependencies\": {\n \"@douyinfe/semi-icons\": \"^2.46.1\",\n \"@douyinfe/semi-ui\": \"^2.46.1\",\n \"@visactor/react-vchart\": \"~1.8.8\",\n \"@visactor/vchart\": \"~1.8.8\",\n \"@visactor/vchart-semi-theme\": \"~1.8.8\",\n \"axios\": \"^0.27.2\",\n \"history\": \"^5.3.0\",\n \"marked\": \"^4.1.1\",\n \"react\": \"^18.2.0\",\n \"react-dom\": \"^18.2.0\",\n \"react-dropzone\": \"^14.2.3\",\n \"react-fireworks\": \"^1.0.4\",\n \"react-router-dom\": \"^6.3.0\",\n \"react-scripts\": \"5.0.1\",\n \"react-telegram-login\": \"^1.1.2\",\n \"react-toastify\": \"^9.0.8\",\n \"react-turnstile\": \"^1.0.5\",\n \"semantic-ui-css\": \"^2.5.0\",\n \"semantic-ui-react\": \"^2.1.3\",\n \"usehooks-ts\": \"^2.9.1\"\n },\n \"scripts\": {\n \"start\": \"react-scripts start\",\n \"build\": \"react-scripts build && mv -f build ../build/air\",\n \"test\": \"react-scripts test\",\n \"eject\": \"react-scripts eject\"\n },\n \"eslintConfig\": {\n \"extends\": [\n \"react-app\",\n \"react-app/jest\"\n ]\n },\n \"browserslist\": {\n \"production\": [\n \">0.2%\",\n \"not dead\",\n \"not op_mini all\"\n ],\n \"development\": [\n \"last 1 chrome version\",\n \"last 1 firefox version\",\n \"last 1 safari version\"\n ]\n },\n \"devDependencies\": {\n \"prettier\": \"2.8.8\",\n \"typescript\": \"4.4.2\"\n },\n \"prettier\": {\n \"singleQuote\": true,\n \"jsxSingleQuote\": true\n },\n \"proxy\": \"http://localhost:3000\"\n}\n"
},
{
"path": "web/air/public/index.html",
"content": "\n\n\n One API \n\n\nYou need to enable JavaScript to run this app. \n
\n\n\n"
},
{
"path": "web/air/public/robots.txt",
"content": "# https://www.robotstxt.org/robotstxt.html\nUser-agent: *\nDisallow:\n"
},
{
"path": "web/air/src/App.js",
"content": "import React, { lazy, Suspense, useContext, useEffect } from 'react';\nimport { Route, Routes } from 'react-router-dom';\nimport Loading from './components/Loading';\nimport User from './pages/User';\nimport { PrivateRoute } from './components/PrivateRoute';\nimport RegisterForm from './components/RegisterForm';\nimport LoginForm from './components/LoginForm';\nimport NotFound from './pages/NotFound';\nimport Setting from './pages/Setting';\nimport EditUser from './pages/User/EditUser';\nimport { getLogo, getSystemName } from './helpers';\nimport PasswordResetForm from './components/PasswordResetForm';\nimport GitHubOAuth from './components/GitHubOAuth';\nimport PasswordResetConfirm from './components/PasswordResetConfirm';\nimport { UserContext } from './context/User';\nimport Channel from './pages/Channel';\nimport Token from './pages/Token';\nimport EditChannel from './pages/Channel/EditChannel';\nimport Redemption from './pages/Redemption';\nimport TopUp from './pages/TopUp';\nimport Log from './pages/Log';\nimport Chat from './pages/Chat';\nimport { Layout } from '@douyinfe/semi-ui';\nimport Midjourney from './pages/Midjourney';\nimport Detail from './pages/Detail';\n\nconst Home = lazy(() => import('./pages/Home'));\nconst About = lazy(() => import('./pages/About'));\n\nfunction App() {\n const [userState, userDispatch] = useContext(UserContext);\n // const [statusState, statusDispatch] = useContext(StatusContext);\n\n const loadUser = () => {\n let user = localStorage.getItem('user');\n if (user) {\n let data = JSON.parse(user);\n userDispatch({ type: 'login', payload: data });\n }\n };\n\n useEffect(() => {\n loadUser();\n let systemName = getSystemName();\n if (systemName) {\n document.title = systemName;\n }\n let logo = getLogo();\n if (logo) {\n let linkElement = document.querySelector('link[rel~=\\'icon\\']');\n if (linkElement) {\n linkElement.href = logo;\n }\n }\n }, []);\n\n return (\n \n \n \n }>\n \n }>\n }>\n \n \n \n }>\n }>\n }>\n }>\n }>\n }>\n }>\n \n }>\n \n \n }\n />\n \n }>\n \n \n }\n />\n \n \n \n }>\n }>\n \n \n \n );\n}\n\nexport default App;\n"
},
{
"path": "web/air/src/components/ChannelsTable.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { API, isMobile, shouldShowPrompt, showError, showInfo, showSuccess, timestamp2string } from '../helpers';\n\nimport { CHANNEL_OPTIONS, ITEMS_PER_PAGE } from '../constants';\nimport { renderGroup, renderNumberWithPoint, renderQuota } from '../helpers/render';\nimport {\n Button,\n Dropdown,\n Form,\n InputNumber,\n Popconfirm,\n Space,\n SplitButtonGroup,\n Switch,\n Table,\n Tag,\n Tooltip,\n Typography\n} from '@douyinfe/semi-ui';\nimport EditChannel from '../pages/Channel/EditChannel';\nimport { IconTreeTriangleDown } from '@douyinfe/semi-icons';\n\nfunction renderTimestamp(timestamp) {\n return (\n <>\n {timestamp2string(timestamp)}\n \n );\n}\n\nlet type2label = undefined;\n\nfunction renderType(type) {\n if (!type2label) {\n type2label = new Map();\n for (let i = 0; i < CHANNEL_OPTIONS.length; i++) {\n type2label[CHANNEL_OPTIONS[i].value] = CHANNEL_OPTIONS[i];\n }\n type2label[0] = { value: 0, text: '未知类型', color: 'grey' };\n }\n return {type2label[type]?.text} ;\n}\n\nconst ChannelsTable = () => {\n const columns = [\n // {\n // title: '',\n // dataIndex: 'checkbox',\n // className: 'checkbox',\n // },\n {\n title: 'ID',\n dataIndex: 'id'\n },\n {\n title: '名称',\n dataIndex: 'name'\n },\n // {\n // title: '分组',\n // dataIndex: 'group',\n // render: (text, record, index) => {\n // return (\n // \n // \n // {\n // text.split(',').map((item, index) => {\n // return (renderGroup(item));\n // })\n // }\n // \n //
\n // );\n // }\n // },\n {\n title: '类型',\n dataIndex: 'type',\n render: (text, record, index) => {\n return (\n \n {renderType(text)}\n
\n );\n }\n },\n {\n title: '状态',\n dataIndex: 'status',\n render: (text, record, index) => {\n return (\n \n {renderStatus(text)}\n
\n );\n }\n },\n {\n title: '响应时间',\n dataIndex: 'response_time',\n render: (text, record, index) => {\n return (\n \n {renderResponseTime(text)}\n
\n );\n }\n },\n {\n title: '已用/剩余',\n dataIndex: 'expired_time',\n render: (text, record, index) => {\n return (\n \n \n \n {renderQuota(record.used_quota)} \n \n \n {\n updateChannelBalance(record);\n }}>${renderNumberWithPoint(record.balance)} \n \n \n
\n );\n }\n },\n {\n title: '优先级',\n dataIndex: 'priority',\n render: (text, record, index) => {\n return (\n \n {\n manageChannel(record.id, 'priority', record, e.target.value);\n }}\n keepFocus={true}\n innerButtons\n defaultValue={record.priority}\n min={-999}\n />\n
\n );\n }\n },\n // {\n // title: '权重',\n // dataIndex: 'weight',\n // render: (text, record, index) => {\n // return (\n // \n // {\n // manageChannel(record.id, 'weight', record, e.target.value);\n // }}\n // keepFocus={true}\n // innerButtons\n // defaultValue={record.weight}\n // min={0}\n // />\n //
\n // );\n // }\n // },\n {\n title: '',\n dataIndex: 'operate',\n render: (text, record, index) => (\n \n {/*
\n {\n testChannel(record, '');\n }}>测试 \n \n \n */}\n
testChannel(record)}>测试 \n
{\n manageChannel(record.id, 'delete', record).then(\n () => {\n removeRecord(record.id);\n }\n );\n }}\n >\n 删除 \n \n {\n record.status === 1 ?\n {\n manageChannel(\n record.id,\n 'disable',\n record\n );\n }\n }>禁用 :\n {\n manageChannel(\n record.id,\n 'enable',\n record\n );\n }\n }>启用 \n }\n {\n setEditingChannel(record);\n setShowEdit(true);\n }\n }>编辑 \n 已启用 ;\n case 2:\n return (\n \n 已禁用\n \n );\n case 3:\n return (\n \n 自动禁用\n \n );\n default:\n return (\n \n 未知状态\n \n );\n }\n };\n\n const renderResponseTime = (responseTime) => {\n let time = responseTime / 1000;\n time = time.toFixed(2) + ' 秒';\n if (responseTime === 0) {\n return 未测试 ;\n } else if (responseTime <= 1000) {\n return {time} ;\n } else if (responseTime <= 3000) {\n return {time} ;\n } else if (responseTime <= 5000) {\n return {time} ;\n } else {\n return {time} ;\n }\n };\n\n const searchChannels = async (searchKeyword, searchGroup, searchModel) => {\n if (searchKeyword === '' && searchGroup === '' && searchModel === '') {\n // if keyword is blank, load files instead.\n await loadChannels(0, pageSize, idSort);\n setActivePage(1);\n return;\n }\n setSearching(true);\n const res = await API.get(`/api/channel/search?keyword=${searchKeyword}&group=${searchGroup}&model=${searchModel}`);\n const { success, message, data } = res.data;\n if (success) {\n setChannels(data);\n setActivePage(1);\n } else {\n showError(message);\n }\n setSearching(false);\n };\n\n const testChannel = async (record, model) => {\n const res = await API.get(`/api/channel/test/${record.id}?model=${model}`);\n const { success, message, time } = res.data;\n if (success) {\n record.response_time = time * 1000;\n record.test_time = Date.now() / 1000;\n showInfo(`渠道 ${record.name} 测试成功，耗时 ${time.toFixed(2)} 秒。`);\n } else {\n showError(message);\n }\n };\n\n const testChannels = async (scope) => {\n const res = await API.get(`/api/channel/test?scope=${scope}`);\n const { success, message } = res.data;\n if (success) {\n showInfo('已成功开始测试渠道，请刷新页面查看结果。');\n } else {\n showError(message);\n }\n };\n\n const deleteAllDisabledChannels = async () => {\n const res = await API.delete(`/api/channel/disabled`);\n const { success, message, data } = res.data;\n if (success) {\n showSuccess(`已删除所有禁用渠道，共计 ${data} 个`);\n await refresh();\n } else {\n showError(message);\n }\n };\n\n const updateChannelBalance = async (record) => {\n const res = await API.get(`/api/channel/update_balance/${record.id}/`);\n const { success, message, balance } = res.data;\n if (success) {\n record.balance = balance;\n record.balance_updated_time = Date.now() / 1000;\n showInfo(`渠道 ${record.name} 余额更新成功！`);\n } else {\n showError(message);\n }\n };\n\n const updateAllChannelsBalance = async () => {\n setUpdatingBalance(true);\n const res = await API.get(`/api/channel/update_balance`);\n const { success, message } = res.data;\n if (success) {\n showInfo('已更新完毕所有已启用渠道余额！');\n } else {\n showError(message);\n }\n setUpdatingBalance(false);\n };\n\n const batchDeleteChannels = async () => {\n if (selectedChannels.length === 0) {\n showError('请先选择要删除的渠道！');\n return;\n }\n setLoading(true);\n let ids = [];\n selectedChannels.forEach((channel) => {\n ids.push(channel.id);\n });\n const res = await API.post(`/api/channel/batch`, { ids: ids });\n const { success, message, data } = res.data;\n if (success) {\n showSuccess(`已删除 ${data} 个渠道！`);\n await refresh();\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n const fixChannelsAbilities = async () => {\n const res = await API.post(`/api/channel/fix`);\n const { success, message, data } = res.data;\n if (success) {\n showSuccess(`已修复 ${data} 个渠道！`);\n await refresh();\n } else {\n showError(message);\n }\n };\n\n let pageData = channels.slice((activePage - 1) * pageSize, activePage * pageSize);\n\n const handlePageChange = page => {\n setActivePage(page);\n if (page === Math.ceil(channels.length / pageSize) + 1) {\n // In this case we have to load more data and then append them.\n loadChannels(page - 1, pageSize, idSort).then(r => {\n });\n }\n };\n\n const handlePageSizeChange = async (size) => {\n localStorage.setItem('page-size', size + '');\n setPageSize(size);\n setActivePage(1);\n loadChannels(0, size, idSort)\n .then()\n .catch((reason) => {\n showError(reason);\n });\n };\n\n const fetchGroups = async () => {\n try {\n let res = await API.get(`/api/group/`);\n // add 'all' option\n // res.data.data.unshift('all');\n setGroupOptions(res.data.data.map((group) => ({\n label: group,\n value: group\n })));\n } catch (error) {\n showError(error.message);\n }\n };\n\n const closeEdit = () => {\n setShowEdit(false);\n };\n\n const handleRow = (record, index) => {\n if (record.status !== 1) {\n return {\n style: {\n background: 'var(--semi-color-disabled-border)'\n }\n };\n } else {\n return {};\n }\n };\n\n\n return (\n <>\n \n
\n
\n
\n {\n setEditingChannel({\n id: undefined\n });\n setShowEdit(true);\n }\n }>添加新的渠道 \n { testChannels(\"all\") }}\n position={isMobile() ? 'top' : 'left'}\n >\n 测试所有渠道 \n \n { testChannels(\"disabled\") }}\n position={isMobile() ? 'top' : 'left'}\n >\n 测试禁用渠道 \n \n {/* \n 更新所有已启用渠道余额 \n */}\n \n 删除禁用渠道 \n \n\n 刷新 \n \n {/*
*/}\n\n {/*
*/}\n
\n {/*
\n
\n 开启批量删除 \n {\n setEnableBatchDelete(v);\n }}> \n \n 删除所选渠道 \n \n \n 修复数据库一致性 \n \n \n
\n
\n \n \n 使用ID排序 \n {\n localStorage.setItem('id-sort', v + '');\n setIdSort(v);\n loadChannels(0, pageSize, v)\n .then()\n .catch((reason) => {\n showError(reason);\n });\n }}> \n \n \n
*/}\n
'',\n onPageSizeChange: (size) => {\n handlePageSizeChange(size).then();\n },\n onPageChange: handlePageChange\n }} loading={loading} onRow={handleRow} rowSelection={\n enableBatchDelete ?\n {\n onChange: (selectedRowKeys, selectedRows) => {\n // console.log(`selectedRowKeys: ${selectedRowKeys}`, 'selectedRows: ', selectedRows);\n setSelectedChannels(selectedRows);\n }\n } : null\n } />\n \n );\n};\n\nexport default ChannelsTable;\n"
},
{
"path": "web/air/src/components/Footer.js",
"content": "import React, { useEffect, useState } from 'react';\n\nimport { Container, Segment } from 'semantic-ui-react';\nimport { getFooterHTML, getSystemName } from '../helpers';\n\nconst Footer = () => {\n const systemName = getSystemName();\n const [footer, setFooter] = useState(getFooterHTML());\n let remainCheckTimes = 5;\n\n const loadFooter = () => {\n let footer_html = localStorage.getItem('footer_html');\n if (footer_html) {\n setFooter(footer_html);\n }\n };\n\n useEffect(() => {\n const timer = setInterval(() => {\n if (remainCheckTimes <= 0) {\n clearInterval(timer);\n return;\n }\n remainCheckTimes--;\n loadFooter();\n }, 200);\n return () => clearTimeout(timer);\n }, []);\n\n return (\n \n \n {footer ? (\n \n ) : (\n \n )}\n \n \n );\n};\n\nexport default Footer;\n"
},
{
"path": "web/air/src/components/GitHubOAuth.js",
"content": "import React, { useContext, useEffect, useState } from 'react';\nimport { Dimmer, Loader, Segment } from 'semantic-ui-react';\nimport { useNavigate, useSearchParams } from 'react-router-dom';\nimport { API, showError, showSuccess } from '../helpers';\nimport { UserContext } from '../context/User';\n\nconst GitHubOAuth = () => {\n const [searchParams, setSearchParams] = useSearchParams();\n\n const [userState, userDispatch] = useContext(UserContext);\n const [prompt, setPrompt] = useState('处理中...');\n const [processing, setProcessing] = useState(true);\n\n let navigate = useNavigate();\n\n const sendCode = async (code, state, count) => {\n const res = await API.get(`/api/oauth/github?code=${code}&state=${state}`);\n const { success, message, data } = res.data;\n if (success) {\n if (message === 'bind') {\n showSuccess('绑定成功！');\n navigate('/setting');\n } else {\n userDispatch({ type: 'login', payload: data });\n localStorage.setItem('user', JSON.stringify(data));\n showSuccess('登录成功！');\n navigate('/');\n }\n } else {\n showError(message);\n if (count === 0) {\n setPrompt(`操作失败，重定向至登录界面中...`);\n navigate('/setting'); // in case this is failed to bind GitHub\n return;\n }\n count++;\n setPrompt(`出现错误，第 ${count} 次重试中...`);\n await new Promise((resolve) => setTimeout(resolve, count * 2000));\n await sendCode(code, state, count);\n }\n };\n\n useEffect(() => {\n let code = searchParams.get('code');\n let state = searchParams.get('state');\n sendCode(code, state, 0).then();\n }, []);\n\n return (\n \n \n {prompt} \n \n \n );\n};\n\nexport default GitHubOAuth;\n"
},
{
"path": "web/air/src/components/HeaderBar.js",
"content": "import React, { useContext, useEffect, useState } from 'react';\nimport { Link, useNavigate } from 'react-router-dom';\nimport { UserContext } from '../context/User';\n\nimport { API, getLogo, getSystemName, showSuccess } from '../helpers';\nimport '../index.css';\n\nimport fireworks from 'react-fireworks';\n\nimport { IconHelpCircle, IconKey, IconUser } from '@douyinfe/semi-icons';\nimport { Avatar, Dropdown, Layout, Nav, Switch } from '@douyinfe/semi-ui';\nimport { stringToColor } from '../helpers/render';\n\n// HeaderBar Buttons\nlet headerButtons = [\n {\n text: '关于',\n itemKey: 'about',\n to: '/about',\n icon: \n \n
{\n const routerMap = {\n about: '/about',\n login: '/login',\n register: '/register'\n };\n return (\n \n {itemElement}\n \n );\n }}\n selectedKeys={[]}\n // items={headerButtons}\n onSelect={key => {\n\n }}\n footer={\n <>\n {isNewYear &&\n // happy new year\n \n Happy New Year!!! \n \n }\n >\n \n 退出 \n \n }\n >\n \n {userState.user.username[0]}\n \n {userState.user.username} \n \n \n :\n <>\n \n \n );\n};\n\nexport default HeaderBar;\n"
},
{
"path": "web/air/src/components/Loading.js",
"content": "import React from 'react';\nimport { Dimmer, Loader, Segment } from 'semantic-ui-react';\n\nconst Loading = ({ prompt: name = 'page' }) => {\n return (\n \n \n 加载{name}中... \n \n \n );\n};\n\nexport default Loading;\n"
},
{
"path": "web/air/src/components/LoginForm.js",
"content": "import React, { useContext, useEffect, useState } from 'react';\nimport { Link, useNavigate, useSearchParams } from 'react-router-dom';\nimport { UserContext } from '../context/User';\nimport { API, getLogo, showError, showInfo, showSuccess } from '../helpers';\nimport { onGitHubOAuthClicked } from './utils';\nimport Turnstile from 'react-turnstile';\nimport { Button, Card, Divider, Form, Icon, Layout, Modal } from '@douyinfe/semi-ui';\nimport Title from '@douyinfe/semi-ui/lib/es/typography/title';\nimport Text from '@douyinfe/semi-ui/lib/es/typography/text';\nimport TelegramLoginButton from 'react-telegram-login';\n\nimport { IconGithubLogo } from '@douyinfe/semi-icons';\nimport WeChatIcon from './WeChatIcon';\n\nconst LoginForm = () => {\n const [inputs, setInputs] = useState({\n username: '',\n password: '',\n wechat_verification_code: ''\n });\n const [searchParams, setSearchParams] = useSearchParams();\n const [submitted, setSubmitted] = useState(false);\n const { username, password } = inputs;\n const [userState, userDispatch] = useContext(UserContext);\n const [turnstileEnabled, setTurnstileEnabled] = useState(false);\n const [turnstileSiteKey, setTurnstileSiteKey] = useState('');\n const [turnstileToken, setTurnstileToken] = useState('');\n let navigate = useNavigate();\n const [status, setStatus] = useState({});\n const logo = getLogo();\n\n useEffect(() => {\n if (searchParams.get('expired')) {\n showError('未登录或登录已过期，请重新登录！');\n }\n let status = localStorage.getItem('status');\n if (status) {\n status = JSON.parse(status);\n setStatus(status);\n if (status.turnstile_check) {\n setTurnstileEnabled(true);\n setTurnstileSiteKey(status.turnstile_site_key);\n }\n }\n }, []);\n\n const [showWeChatLoginModal, setShowWeChatLoginModal] = useState(false);\n\n const onWeChatLoginClicked = () => {\n setShowWeChatLoginModal(true);\n };\n\n const onSubmitWeChatVerificationCode = async () => {\n if (turnstileEnabled && turnstileToken === '') {\n showInfo('请稍后几秒重试，Turnstile 正在检查用户环境！');\n return;\n }\n const res = await API.get(\n `/api/oauth/wechat?code=${inputs.wechat_verification_code}`\n );\n const { success, message, data } = res.data;\n if (success) {\n userDispatch({ type: 'login', payload: data });\n localStorage.setItem('user', JSON.stringify(data));\n navigate('/');\n showSuccess('登录成功！');\n setShowWeChatLoginModal(false);\n } else {\n showError(message);\n }\n };\n\n function handleChange(name, value) {\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n }\n\n async function handleSubmit(e) {\n if (turnstileEnabled && turnstileToken === '') {\n showInfo('请稍后几秒重试，Turnstile 正在检查用户环境！');\n return;\n }\n setSubmitted(true);\n if (username && password) {\n const res = await API.post(`/api/user/login?turnstile=${turnstileToken}`, {\n username,\n password\n });\n const { success, message, data } = res.data;\n if (success) {\n userDispatch({ type: 'login', payload: data });\n localStorage.setItem('user', JSON.stringify(data));\n showSuccess('登录成功！');\n if (username === 'root' && password === '123456') {\n Modal.error({ title: '您正在使用默认密码！', content: '请立刻修改默认密码！', centered: true });\n }\n navigate('/token');\n } else {\n showError(message);\n }\n } else {\n showError('请输入用户名和密码！');\n }\n }\n\n // 添加Telegram登录处理函数\n const onTelegramLoginClicked = async (response) => {\n const fields = ['id', 'first_name', 'last_name', 'username', 'photo_url', 'auth_date', 'hash', 'lang'];\n const params = {};\n fields.forEach((field) => {\n if (response[field]) {\n params[field] = response[field];\n }\n });\n const res = await API.get(`/api/oauth/telegram/login`, { params });\n const { success, message, data } = res.data;\n if (success) {\n userDispatch({ type: 'login', payload: data });\n localStorage.setItem('user', JSON.stringify(data));\n showSuccess('登录成功！');\n navigate('/');\n } else {\n showError(message);\n }\n };\n\n return (\n \n
\n \n \n \n \n
\n
\n \n 用户登录\n \n \n \n \n 没有账号请先 \n \n 忘记密码 \n
\n {status.github_oauth || status.wechat_login || status.telegram_oauth ? (\n <>\n \n 第三方登录\n \n \n {status.github_oauth ? (\n
\n \n ) : (\n <>\n )}\n setShowWeChatLoginModal(false)}\n okText={'登录'}\n size={'small'}\n centered={true}\n >\n \n
\n
\n
\n 微信扫码关注公众号，输入「验证码」获取验证码（三分钟内有效）\n
\n
\n {turnstileEnabled ? (\n
\n {\n setTurnstileToken(token);\n }}\n />\n
\n ) : (\n <>\n )}\n
\n
\n \n
充值 ;\n case 2:\n return 消费 ;\n case 3:\n return 管理 ;\n case 4:\n return 系统 ;\n case 5:\n return 测试 ;\n default:\n return 未知 ;\n }\n}\n\nfunction renderIsStream(bool) {\n if (bool) {\n return 流 ;\n } else {\n return 非流 ;\n }\n}\n\nfunction renderUseTime(type) {\n const time = parseInt(type);\n if (time < 101) {\n return {time} s ;\n } else if (time < 300) {\n return {time} s ;\n } else {\n return {time} s ;\n }\n}\n\nconst LogsTable = () => {\n const columns = [{\n title: '时间', dataIndex: 'timestamp2string'\n }, {\n title: '渠道',\n dataIndex: 'channel',\n className: isAdmin() ? 'tableShow' : 'tableHiddle',\n render: (text, record, index) => {\n return (isAdminUser ? record.type === 0 || record.type === 2 ? \n { {text} }\n
: <> : <>);\n }\n }, {\n title: '用户',\n dataIndex: 'username',\n className: isAdmin() ? 'tableShow' : 'tableHiddle',\n render: (text, record, index) => {\n return (isAdminUser ? \n
showUserInfo(record.user_id)}>\n {typeof text === 'string' && text.slice(0, 1)}\n \n {text}\n
\n {\n copyText(text);\n }}> {text} \n
: <>);\n }\n }, {\n title: '类型', dataIndex: 'type', render: (text, record, index) => {\n return (\n {renderType(text)}\n
);\n }\n }, {\n title: '模型', dataIndex: 'model_name', render: (text, record, index) => {\n return (record.type === 0 || record.type === 2 ? \n {\n copyText(text);\n }}> {text} \n
: <>);\n }\n },\n // {\n // title: '用时', dataIndex: 'use_time', render: (text, record, index) => {\n // return (\n // \n // {renderUseTime(text)}\n // {renderIsStream(record.is_stream)}\n // \n //
);\n // }\n // },\n {\n title: '提示', dataIndex: 'prompt_tokens', render: (text, record, index) => {\n return (record.type === 0 || record.type === 2 ? \n { {text} }\n
: <>);\n }\n }, {\n title: '补全', dataIndex: 'completion_tokens', render: (text, record, index) => {\n return (parseInt(text) > 0 && (record.type === 0 || record.type === 2) ? \n { {text} }\n
: <>);\n }\n }, {\n title: '花费', dataIndex: 'quota', render: (text, record, index) => {\n return (record.type === 0 || record.type === 2 ? \n {renderQuota(text, 6)}\n
: <>);\n }\n }, {\n title: '详情', dataIndex: 'content', render: (text, record, index) => {\n return \n {text}\n ;\n }\n }];\n\n const [logs, setLogs] = useState([]);\n const [showStat, setShowStat] = useState(false);\n const [loading, setLoading] = useState(false);\n const [loadingStat, setLoadingStat] = useState(false);\n const [activePage, setActivePage] = useState(1);\n const [logCount, setLogCount] = useState(ITEMS_PER_PAGE);\n const [pageSize, setPageSize] = useState(ITEMS_PER_PAGE);\n const [searchKeyword, setSearchKeyword] = useState('');\n const [searching, setSearching] = useState(false);\n const [logType, setLogType] = useState(0);\n const isAdminUser = isAdmin();\n let now = new Date();\n // 初始化start_timestamp为前一天\n const [inputs, setInputs] = useState({\n username: '',\n token_name: '',\n model_name: '',\n start_timestamp: timestamp2string(now.getTime() / 1000 - 86400),\n end_timestamp: timestamp2string(now.getTime() / 1000 + 3600),\n channel: ''\n });\n const { username, token_name, model_name, start_timestamp, end_timestamp, channel } = inputs;\n\n const [stat, setStat] = useState({\n quota: 0, token: 0\n });\n\n const handleInputChange = (value, name) => {\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n };\n\n const getLogSelfStat = async () => {\n let localStartTimestamp = Date.parse(start_timestamp) / 1000;\n let localEndTimestamp = Date.parse(end_timestamp) / 1000;\n let res = await API.get(`/api/log/self/stat?type=${logType}&token_name=${token_name}&model_name=${model_name}&start_timestamp=${localStartTimestamp}&end_timestamp=${localEndTimestamp}`);\n const { success, message, data } = res.data;\n if (success) {\n setStat(data);\n } else {\n showError(message);\n }\n };\n\n const getLogStat = async () => {\n let localStartTimestamp = Date.parse(start_timestamp) / 1000;\n let localEndTimestamp = Date.parse(end_timestamp) / 1000;\n let res = await API.get(`/api/log/stat?type=${logType}&username=${username}&token_name=${token_name}&model_name=${model_name}&start_timestamp=${localStartTimestamp}&end_timestamp=${localEndTimestamp}&channel=${channel}`);\n const { success, message, data } = res.data;\n if (success) {\n setStat(data);\n } else {\n showError(message);\n }\n };\n\n const handleEyeClick = async () => {\n setLoadingStat(true);\n if (isAdminUser) {\n await getLogStat();\n } else {\n await getLogSelfStat();\n }\n setShowStat(true);\n setLoadingStat(false);\n };\n\n const showUserInfo = async (userId) => {\n if (!isAdminUser) {\n return;\n }\n const res = await API.get(`/api/user/${userId}`);\n const { success, message, data } = res.data;\n if (success) {\n Modal.info({\n title: '用户信息', content: \n
用户名: {data.username}
\n
余额: {renderQuota(data.quota)}
\n
已用额度：{renderQuota(data.used_quota)}
\n
请求次数：{renderNumber(data.request_count)}
\n
\n \n \n 使用明细（总消耗额度：\n {showStat ? renderQuota(stat.quota) : '点击查看'} \n ）\n \n \n \n \n {\n handlePageSizeChange(size).then();\n },\n onPageChange: handlePageChange\n }} />\n {\n setLogType(parseInt(value));\n refresh(parseInt(value)).then();\n }}>\n 全部 \n 充值 \n 消费 \n 管理 \n 系统 \n \n \n );\n};\n\nexport default LogsTable;\n"
},
{
"path": "web/air/src/components/MjLogsTable.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { API, copy, isAdmin, showError, showSuccess, timestamp2string } from '../helpers';\n\nimport { Banner, Button, Form, ImagePreview, Layout, Modal, Progress, Table, Tag, Typography } from '@douyinfe/semi-ui';\nimport { ITEMS_PER_PAGE } from '../constants';\n\n\nconst colors = ['amber', 'blue', 'cyan', 'green', 'grey', 'indigo',\n 'light-blue', 'lime', 'orange', 'pink',\n 'purple', 'red', 'teal', 'violet', 'yellow'\n];\n\nfunction renderType(type) {\n switch (type) {\n case 'IMAGINE':\n return 绘图 ;\n case 'UPSCALE':\n return 放大 ;\n case 'VARIATION':\n return 变换 ;\n case 'HIGH_VARIATION':\n return 强变换 ;\n case 'LOW_VARIATION':\n return 弱变换 ;\n case 'PAN':\n return 平移 ;\n case 'DESCRIBE':\n return 图生文 ;\n case 'BLEND':\n return 图混合 ;\n case 'SHORTEN':\n return 缩词 ;\n case 'REROLL':\n return 重绘 ;\n case 'INPAINT':\n return 局部重绘-提交 ;\n case 'ZOOM':\n return 变焦 ;\n case 'CUSTOM_ZOOM':\n return 自定义变焦-提交 ;\n case 'MODAL':\n return 窗口处理 ;\n case 'SWAP_FACE':\n return 换脸 ;\n default:\n return 未知 ;\n }\n}\n\n\nfunction renderCode(code) {\n switch (code) {\n case 1:\n return 已提交 ;\n case 21:\n return 等待中 ;\n case 22:\n return 重复提交 ;\n case 0:\n return 未提交 ;\n default:\n return 未知 ;\n }\n}\n\n\nfunction renderStatus(type) {\n // Ensure all cases are string literals by adding quotes.\n switch (type) {\n case 'SUCCESS':\n return 成功 ;\n case 'NOT_START':\n return 未启动 ;\n case 'SUBMITTED':\n return 队列中 ;\n case 'IN_PROGRESS':\n return 执行中 ;\n case 'FAILURE':\n return 失败 ;\n case 'MODAL':\n return 窗口等待 ;\n default:\n return 未知 ;\n }\n}\n\nconst renderTimestamp = (timestampInSeconds) => {\n const date = new Date(timestampInSeconds * 1000); // 从秒转换为毫秒\n\n const year = date.getFullYear(); // 获取年份\n const month = ('0' + (date.getMonth() + 1)).slice(-2); // 获取月份，从0开始需要+1，并保证两位数\n const day = ('0' + date.getDate()).slice(-2); // 获取日期，并保证两位数\n const hours = ('0' + date.getHours()).slice(-2); // 获取小时，并保证两位数\n const minutes = ('0' + date.getMinutes()).slice(-2); // 获取分钟，并保证两位数\n const seconds = ('0' + date.getSeconds()).slice(-2); // 获取秒钟，并保证两位数\n\n return `${year}-${month}-${day} ${hours}:${minutes}:${seconds}`; // 格式化输出\n};\n\n\nconst LogsTable = () => {\n const [isModalOpen, setIsModalOpen] = useState(false);\n const [modalContent, setModalContent] = useState('');\n const columns = [\n {\n title: '提交时间',\n dataIndex: 'submit_time',\n render: (text, record, index) => {\n return (\n \n {renderTimestamp(text / 1000)}\n
\n );\n }\n },\n {\n title: '渠道',\n dataIndex: 'channel_id',\n className: isAdmin() ? 'tableShow' : 'tableHiddle',\n render: (text, record, index) => {\n return (\n\n \n {\n copyText(text); // 假设copyText是用于文本复制的函数\n }}> {text} \n
\n\n );\n }\n },\n {\n title: '类型',\n dataIndex: 'action',\n render: (text, record, index) => {\n return (\n \n {renderType(text)}\n
\n );\n }\n },\n {\n title: '任务ID',\n dataIndex: 'mj_id',\n render: (text, record, index) => {\n return (\n \n {text}\n
\n );\n }\n },\n {\n title: '提交结果',\n dataIndex: 'code',\n className: isAdmin() ? 'tableShow' : 'tableHiddle',\n render: (text, record, index) => {\n return (\n \n {renderCode(text)}\n
\n );\n }\n },\n {\n title: '任务状态',\n dataIndex: 'status',\n className: isAdmin() ? 'tableShow' : 'tableHiddle',\n render: (text, record, index) => {\n return (\n \n {renderStatus(text)}\n
\n );\n }\n },\n {\n title: '进度',\n dataIndex: 'progress',\n render: (text, record, index) => {\n return (\n \n {\n // 转换例如100%为数字100，如果text未定义，返回0\n
\n }\n
{\n setModalImageUrl(text); // 更新图片URL状态\n setIsModalOpenurl(true); // 打开模态框\n }}\n >\n 查看图片\n \n );\n }\n },\n {\n title: 'Prompt',\n dataIndex: 'prompt',\n render: (text, record, index) => {\n // 如果text未定义，返回替代文本，例如空字符串''或其他\n if (!text) {\n return '无';\n }\n\n return (\n {\n setModalContent(text);\n setIsModalOpen(true);\n }}\n >\n {text}\n \n );\n }\n },\n {\n title: 'PromptEn',\n dataIndex: 'prompt_en',\n render: (text, record, index) => {\n // 如果text未定义，返回替代文本，例如空字符串''或其他\n if (!text) {\n return '无';\n }\n\n return (\n {\n setModalContent(text);\n setIsModalOpen(true);\n }}\n >\n {text}\n \n );\n }\n },\n {\n title: '失败原因',\n dataIndex: 'fail_reason',\n render: (text, record, index) => {\n // 如果text未定义，返回替代文本，例如空字符串''或其他\n if (!text) {\n return '无';\n }\n\n return (\n {\n setModalContent(text);\n setIsModalOpen(true);\n }}\n >\n {text}\n \n );\n }\n }\n\n ];\n\n const [logs, setLogs] = useState([]);\n const [loading, setLoading] = useState(true);\n const [activePage, setActivePage] = useState(1);\n const [logCount, setLogCount] = useState(ITEMS_PER_PAGE);\n const [logType, setLogType] = useState(0);\n const isAdminUser = isAdmin();\n const [isModalOpenurl, setIsModalOpenurl] = useState(false);\n const [showBanner, setShowBanner] = useState(false);\n\n // 定义模态框图片URL的状态和更新函数\n const [modalImageUrl, setModalImageUrl] = useState('');\n let now = new Date();\n // 初始化start_timestamp为前一天\n const [inputs, setInputs] = useState({\n channel_id: '',\n mj_id: '',\n start_timestamp: timestamp2string(now.getTime() / 1000 - 2592000),\n end_timestamp: timestamp2string(now.getTime() / 1000 + 3600)\n });\n const { channel_id, mj_id, start_timestamp, end_timestamp } = inputs;\n\n const [stat, setStat] = useState({\n quota: 0,\n token: 0\n });\n\n const handleInputChange = (value, name) => {\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n };\n\n\n const setLogsFormat = (logs) => {\n for (let i = 0; i < logs.length; i++) {\n logs[i].timestamp2string = timestamp2string(logs[i].created_at);\n logs[i].key = '' + logs[i].id;\n }\n // data.key = '' + data.id\n setLogs(logs);\n setLogCount(logs.length + ITEMS_PER_PAGE);\n // console.log(logCount);\n };\n\n const loadLogs = async (startIdx) => {\n setLoading(true);\n\n let url = '';\n let localStartTimestamp = Date.parse(start_timestamp);\n let localEndTimestamp = Date.parse(end_timestamp);\n if (isAdminUser) {\n url = `/api/mj/?p=${startIdx}&channel_id=${channel_id}&mj_id=${mj_id}&start_timestamp=${localStartTimestamp}&end_timestamp=${localEndTimestamp}`;\n } else {\n url = `/api/mj/self/?p=${startIdx}&mj_id=${mj_id}&start_timestamp=${localStartTimestamp}&end_timestamp=${localEndTimestamp}`;\n }\n const res = await API.get(url);\n const { success, message, data } = res.data;\n if (success) {\n if (startIdx === 0) {\n setLogsFormat(data);\n } else {\n let newLogs = [...logs];\n newLogs.splice(startIdx * ITEMS_PER_PAGE, data.length, ...data);\n setLogsFormat(newLogs);\n }\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n const pageData = logs.slice((activePage - 1) * ITEMS_PER_PAGE, activePage * ITEMS_PER_PAGE);\n\n const handlePageChange = page => {\n setActivePage(page);\n if (page === Math.ceil(logs.length / ITEMS_PER_PAGE) + 1) {\n // In this case we have to load more data and then append them.\n loadLogs(page - 1).then(r => {\n });\n }\n };\n\n const refresh = async () => {\n // setLoading(true);\n setActivePage(1);\n await loadLogs(0);\n };\n\n const copyText = async (text) => {\n if (await copy(text)) {\n showSuccess('已复制：' + text);\n } else {\n // setSearchKeyword(text);\n Modal.error({ title: '无法复制到剪贴板，请手动复制', content: text });\n }\n };\n\n useEffect(() => {\n refresh().then();\n }, [logType]);\n\n useEffect(() => {\n const mjNotifyEnabled = localStorage.getItem('mj_notify_enabled');\n if (mjNotifyEnabled !== 'true') {\n setShowBanner(true);\n }\n }, []);\n\n return (\n <>\n\n \n {isAdminUser && showBanner ? setIsModalOpen(false)}\n onCancel={() => setIsModalOpen(false)}\n closable={null}\n bodyStyle={{ height: '400px', overflow: 'auto' }} // 设置模态框内容区域样式\n width={800} // 设置模态框宽度\n >\n {modalContent}
\n \n setIsModalOpenurl(visible)}\n />\n\n \n \n );\n};\n\nexport default LogsTable;\n"
},
{
"path": "web/air/src/components/OperationSetting.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { Divider, Form, Grid, Header } from 'semantic-ui-react';\nimport { API, showError, showSuccess, timestamp2string, verifyJSON } from '../helpers';\n\nconst OperationSetting = () => {\n let now = new Date();\n let [inputs, setInputs] = useState({\n QuotaForNewUser: 0,\n QuotaForInviter: 0,\n QuotaForInvitee: 0,\n QuotaRemindThreshold: 0,\n PreConsumedQuota: 0,\n ModelRatio: '',\n CompletionRatio: '',\n GroupRatio: '',\n TopUpLink: '',\n ChatLink: '',\n QuotaPerUnit: 0,\n AutomaticDisableChannelEnabled: '',\n AutomaticEnableChannelEnabled: '',\n ChannelDisableThreshold: 0,\n LogConsumeEnabled: '',\n DisplayInCurrencyEnabled: '',\n DisplayTokenStatEnabled: '',\n ApproximateTokenEnabled: '',\n RetryTimes: 0\n });\n const [originInputs, setOriginInputs] = useState({});\n let [loading, setLoading] = useState(false);\n let [historyTimestamp, setHistoryTimestamp] = useState(timestamp2string(now.getTime() / 1000 - 30 * 24 * 3600)); // a month ago\n\n const getOptions = async () => {\n const res = await API.get('/api/option/');\n const { success, message, data } = res.data;\n if (success) {\n let newInputs = {};\n data.forEach((item) => {\n if (item.key === 'ModelRatio' || item.key === 'GroupRatio' || item.key === 'CompletionRatio') {\n item.value = JSON.stringify(JSON.parse(item.value), null, 2);\n }\n if (item.value === '{}') {\n item.value = '';\n }\n newInputs[item.key] = item.value;\n });\n setInputs(newInputs);\n setOriginInputs(newInputs);\n } else {\n showError(message);\n }\n };\n\n useEffect(() => {\n getOptions().then();\n }, []);\n\n const updateOption = async (key, value) => {\n setLoading(true);\n if (key.endsWith('Enabled')) {\n value = inputs[key] === 'true' ? 'false' : 'true';\n }\n const res = await API.put('/api/option/', {\n key,\n value\n });\n const { success, message } = res.data;\n if (success) {\n setInputs((inputs) => ({ ...inputs, [key]: value }));\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n const handleInputChange = async (e, { name, value }) => {\n if (name.endsWith('Enabled')) {\n await updateOption(name, value);\n } else {\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n }\n };\n\n const submitConfig = async (group) => {\n switch (group) {\n case 'monitor':\n if (originInputs['ChannelDisableThreshold'] !== inputs.ChannelDisableThreshold) {\n await updateOption('ChannelDisableThreshold', inputs.ChannelDisableThreshold);\n }\n if (originInputs['QuotaRemindThreshold'] !== inputs.QuotaRemindThreshold) {\n await updateOption('QuotaRemindThreshold', inputs.QuotaRemindThreshold);\n }\n break;\n case 'ratio':\n if (originInputs['ModelRatio'] !== inputs.ModelRatio) {\n if (!verifyJSON(inputs.ModelRatio)) {\n showError('模型倍率不是合法的 JSON 字符串');\n return;\n }\n await updateOption('ModelRatio', inputs.ModelRatio);\n }\n if (originInputs['GroupRatio'] !== inputs.GroupRatio) {\n if (!verifyJSON(inputs.GroupRatio)) {\n showError('分组倍率不是合法的 JSON 字符串');\n return;\n }\n await updateOption('GroupRatio', inputs.GroupRatio);\n }\n if (originInputs['CompletionRatio'] !== inputs.CompletionRatio) {\n if (!verifyJSON(inputs.CompletionRatio)) {\n showError('补全倍率不是合法的 JSON 字符串');\n return;\n }\n await updateOption('CompletionRatio', inputs.CompletionRatio);\n }\n break;\n case 'quota':\n if (originInputs['QuotaForNewUser'] !== inputs.QuotaForNewUser) {\n await updateOption('QuotaForNewUser', inputs.QuotaForNewUser);\n }\n if (originInputs['QuotaForInvitee'] !== inputs.QuotaForInvitee) {\n await updateOption('QuotaForInvitee', inputs.QuotaForInvitee);\n }\n if (originInputs['QuotaForInviter'] !== inputs.QuotaForInviter) {\n await updateOption('QuotaForInviter', inputs.QuotaForInviter);\n }\n if (originInputs['PreConsumedQuota'] !== inputs.PreConsumedQuota) {\n await updateOption('PreConsumedQuota', inputs.PreConsumedQuota);\n }\n break;\n case 'general':\n if (originInputs['TopUpLink'] !== inputs.TopUpLink) {\n await updateOption('TopUpLink', inputs.TopUpLink);\n }\n if (originInputs['ChatLink'] !== inputs.ChatLink) {\n await updateOption('ChatLink', inputs.ChatLink);\n }\n if (originInputs['QuotaPerUnit'] !== inputs.QuotaPerUnit) {\n await updateOption('QuotaPerUnit', inputs.QuotaPerUnit);\n }\n if (originInputs['RetryTimes'] !== inputs.RetryTimes) {\n await updateOption('RetryTimes', inputs.RetryTimes);\n }\n break;\n }\n };\n\n const deleteHistoryLogs = async () => {\n console.log(inputs);\n const res = await API.delete(`/api/log/?target_timestamp=${Date.parse(historyTimestamp) / 1000}`);\n const { success, message, data } = res.data;\n if (success) {\n showSuccess(`${data} 条日志已清理！`);\n return;\n }\n showError('日志清理失败：' + message);\n };\n\n return (\n \n \n \n \n \n );\n};\n\nexport default OperationSetting;\n"
},
{
"path": "web/air/src/components/OtherSetting.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { Button, Divider, Form, Grid, Header, Message, Modal } from 'semantic-ui-react';\nimport { API, showError, showSuccess } from '../helpers';\nimport { marked } from 'marked';\nimport { Link } from 'react-router-dom';\n\nconst OtherSetting = () => {\n let [inputs, setInputs] = useState({\n Footer: '',\n Notice: '',\n About: '',\n SystemName: '',\n Logo: '',\n HomePageContent: '',\n Theme: ''\n });\n let [loading, setLoading] = useState(false);\n const [showUpdateModal, setShowUpdateModal] = useState(false);\n const [updateData, setUpdateData] = useState({\n tag_name: '',\n content: ''\n });\n\n const getOptions = async () => {\n const res = await API.get('/api/option/');\n const { success, message, data } = res.data;\n if (success) {\n let newInputs = {};\n data.forEach((item) => {\n if (item.key in inputs) {\n newInputs[item.key] = item.value;\n }\n });\n setInputs(newInputs);\n } else {\n showError(message);\n }\n };\n\n useEffect(() => {\n getOptions().then();\n }, []);\n\n const updateOption = async (key, value) => {\n setLoading(true);\n const res = await API.put('/api/option/', {\n key,\n value\n });\n const { success, message } = res.data;\n if (success) {\n setInputs((inputs) => ({ ...inputs, [key]: value }));\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n const handleInputChange = async (e, { name, value }) => {\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n };\n\n const submitNotice = async () => {\n await updateOption('Notice', inputs.Notice);\n };\n\n const submitFooter = async () => {\n await updateOption('Footer', inputs.Footer);\n };\n\n const submitSystemName = async () => {\n await updateOption('SystemName', inputs.SystemName);\n };\n\n const submitTheme = async () => {\n await updateOption('Theme', inputs.Theme);\n };\n\n const submitLogo = async () => {\n await updateOption('Logo', inputs.Logo);\n };\n\n const submitAbout = async () => {\n await updateOption('About', inputs.About);\n };\n\n const submitOption = async (key) => {\n await updateOption(key, inputs[key]);\n };\n\n const openGitHubRelease = () => {\n window.location =\n 'https://github.com/songquanpeng/one-api/releases/latest';\n };\n\n const checkUpdate = async () => {\n const res = await API.get(\n 'https://api.github.com/repos/songquanpeng/one-api/releases/latest'\n );\n const { tag_name, body } = res.data;\n if (tag_name === process.env.REACT_APP_VERSION) {\n showSuccess(`已是最新版本：${tag_name}`);\n } else {\n setUpdateData({\n tag_name: tag_name,\n content: marked.parse(body)\n });\n setShowUpdateModal(true);\n }\n };\n\n return (\n \n \n \n \n setShowUpdateModal(false)}\n onOpen={() => setShowUpdateModal(true)}\n open={showUpdateModal}\n >\n 新版本：{updateData.tag_name} \n \n \n
\n \n \n \n setShowUpdateModal(false)}>关闭 \n {\n setShowUpdateModal(false);\n openGitHubRelease();\n }}\n />\n \n \n \n );\n};\n\nexport default OtherSetting;\n"
},
{
"path": "web/air/src/components/PasswordResetConfirm.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { Button, Form, Grid, Header, Image, Segment } from 'semantic-ui-react';\nimport { API, copy, showError, showNotice } from '../helpers';\nimport { useSearchParams } from 'react-router-dom';\n\nconst PasswordResetConfirm = () => {\n const [inputs, setInputs] = useState({\n email: '',\n token: ''\n });\n const { email, token } = inputs;\n\n const [loading, setLoading] = useState(false);\n\n const [disableButton, setDisableButton] = useState(false);\n const [countdown, setCountdown] = useState(30);\n\n const [newPassword, setNewPassword] = useState('');\n\n const [searchParams, setSearchParams] = useSearchParams();\n useEffect(() => {\n let token = searchParams.get('token');\n let email = searchParams.get('email');\n setInputs({\n token,\n email\n });\n }, []);\n\n useEffect(() => {\n let countdownInterval = null;\n if (disableButton && countdown > 0) {\n countdownInterval = setInterval(() => {\n setCountdown(countdown - 1);\n }, 1000);\n } else if (countdown === 0) {\n setDisableButton(false);\n setCountdown(30);\n }\n return () => clearInterval(countdownInterval);\n }, [disableButton, countdown]);\n\n async function handleSubmit(e) {\n setDisableButton(true);\n if (!email) return;\n setLoading(true);\n const res = await API.post(`/api/user/reset`, {\n email,\n token\n });\n const { success, message } = res.data;\n if (success) {\n let password = res.data.data;\n setNewPassword(password);\n await copy(password);\n showNotice(`新密码已复制到剪贴板：${password}`);\n } else {\n showError(message);\n }\n setLoading(false);\n }\n\n return (\n \n \n \n \n \n \n );\n};\n\nexport default PasswordResetConfirm;\n"
},
{
"path": "web/air/src/components/PasswordResetForm.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { Button, Form, Grid, Header, Image, Segment } from 'semantic-ui-react';\nimport { API, showError, showInfo, showSuccess } from '../helpers';\nimport Turnstile from 'react-turnstile';\n\nconst PasswordResetForm = () => {\n const [inputs, setInputs] = useState({\n email: ''\n });\n const { email } = inputs;\n\n const [loading, setLoading] = useState(false);\n const [turnstileEnabled, setTurnstileEnabled] = useState(false);\n const [turnstileSiteKey, setTurnstileSiteKey] = useState('');\n const [turnstileToken, setTurnstileToken] = useState('');\n const [disableButton, setDisableButton] = useState(false);\n const [countdown, setCountdown] = useState(30);\n\n useEffect(() => {\n let countdownInterval = null;\n if (disableButton && countdown > 0) {\n countdownInterval = setInterval(() => {\n setCountdown(countdown - 1);\n }, 1000);\n } else if (countdown === 0) {\n setDisableButton(false);\n setCountdown(30);\n }\n return () => clearInterval(countdownInterval);\n }, [disableButton, countdown]);\n\n function handleChange(e) {\n const { name, value } = e.target;\n setInputs(inputs => ({ ...inputs, [name]: value }));\n }\n\n async function handleSubmit(e) {\n setDisableButton(true);\n if (!email) return;\n if (turnstileEnabled && turnstileToken === '') {\n showInfo('请稍后几秒重试，Turnstile 正在检查用户环境！');\n return;\n }\n setLoading(true);\n const res = await API.get(\n `/api/reset_password?email=${email}&turnstile=${turnstileToken}`\n );\n const { success, message } = res.data;\n if (success) {\n showSuccess('重置邮件发送成功，请检查邮箱！');\n setInputs({ ...inputs, email: '' });\n } else {\n showError(message);\n }\n setLoading(false);\n }\n\n return (\n \n \n \n \n \n \n );\n};\n\nexport default PasswordResetForm;\n"
},
{
"path": "web/air/src/components/PersonalSetting.js",
"content": "import React, { useContext, useEffect, useState } from 'react';\nimport { useNavigate } from 'react-router-dom';\nimport { API, copy, isRoot, showError, showInfo, showSuccess } from '../helpers';\nimport Turnstile from 'react-turnstile';\nimport { UserContext } from '../context/User';\nimport { onGitHubOAuthClicked } from './utils';\nimport {\n Avatar,\n Banner,\n Button,\n Card,\n Descriptions,\n Image,\n Input,\n InputNumber,\n Layout,\n Modal,\n Space,\n Tag,\n Typography\n} from '@douyinfe/semi-ui';\nimport { getQuotaPerUnit, renderQuota, renderQuotaWithPrompt, stringToColor } from '../helpers/render';\nimport TelegramLoginButton from 'react-telegram-login';\n\nconst PersonalSetting = () => {\n const [userState, userDispatch] = useContext(UserContext);\n let navigate = useNavigate();\n\n const [inputs, setInputs] = useState({\n wechat_verification_code: '',\n email_verification_code: '',\n email: '',\n self_account_deletion_confirmation: '',\n set_new_password: '',\n set_new_password_confirmation: ''\n });\n const [status, setStatus] = useState({});\n const [showChangePasswordModal, setShowChangePasswordModal] = useState(false);\n const [showWeChatBindModal, setShowWeChatBindModal] = useState(false);\n const [showEmailBindModal, setShowEmailBindModal] = useState(false);\n const [showAccountDeleteModal, setShowAccountDeleteModal] = useState(false);\n const [turnstileEnabled, setTurnstileEnabled] = useState(false);\n const [turnstileSiteKey, setTurnstileSiteKey] = useState('');\n const [turnstileToken, setTurnstileToken] = useState('');\n const [loading, setLoading] = useState(false);\n const [disableButton, setDisableButton] = useState(false);\n const [countdown, setCountdown] = useState(30);\n const [affLink, setAffLink] = useState('');\n const [systemToken, setSystemToken] = useState('');\n const [models, setModels] = useState([]);\n const [openTransfer, setOpenTransfer] = useState(false);\n const [transferAmount, setTransferAmount] = useState(0);\n\n useEffect(() => {\n // let user = localStorage.getItem('user');\n // if (user) {\n // userDispatch({ type: 'login', payload: user });\n // }\n // console.log(localStorage.getItem('user'))\n\n let status = localStorage.getItem('status');\n if (status) {\n status = JSON.parse(status);\n setStatus(status);\n if (status.turnstile_check) {\n setTurnstileEnabled(true);\n setTurnstileSiteKey(status.turnstile_site_key);\n }\n }\n getUserData().then(\n (res) => {\n console.log(userState);\n }\n );\n loadModels().then();\n getAffLink().then();\n setTransferAmount(getQuotaPerUnit());\n }, []);\n\n useEffect(() => {\n let countdownInterval = null;\n if (disableButton && countdown > 0) {\n countdownInterval = setInterval(() => {\n setCountdown(countdown - 1);\n }, 1000);\n } else if (countdown === 0) {\n setDisableButton(false);\n setCountdown(30);\n }\n return () => clearInterval(countdownInterval); // Clean up on unmount\n }, [disableButton, countdown]);\n\n const handleInputChange = (name, value) => {\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n };\n\n const generateAccessToken = async () => {\n const res = await API.get('/api/user/token');\n const { success, message, data } = res.data;\n if (success) {\n setSystemToken(data);\n await copy(data);\n showSuccess(`令牌已重置并已复制到剪贴板`);\n } else {\n showError(message);\n }\n };\n\n const getAffLink = async () => {\n const res = await API.get('/api/user/aff');\n const { success, message, data } = res.data;\n if (success) {\n let link = `${window.location.origin}/register?aff=${data}`;\n setAffLink(link);\n } else {\n showError(message);\n }\n };\n\n const getUserData = async () => {\n let res = await API.get(`/api/user/self`);\n const { success, message, data } = res.data;\n if (success) {\n userDispatch({ type: 'login', payload: data });\n } else {\n showError(message);\n }\n };\n\n const loadModels = async () => {\n let res = await API.get(`/api/user/available_models`);\n const { success, message, data } = res.data;\n if (success) {\n setModels(data);\n console.log(data);\n } else {\n showError(message);\n }\n };\n\n const handleAffLinkClick = async (e) => {\n e.target.select();\n await copy(e.target.value);\n showSuccess(`邀请链接已复制到剪切板`);\n };\n\n const handleSystemTokenClick = async (e) => {\n e.target.select();\n await copy(e.target.value);\n showSuccess(`系统令牌已复制到剪切板`);\n };\n\n const deleteAccount = async () => {\n if (inputs.self_account_deletion_confirmation !== userState.user.username) {\n showError('请输入你的账户名以确认删除！');\n return;\n }\n\n const res = await API.delete('/api/user/self');\n const { success, message } = res.data;\n\n if (success) {\n showSuccess('账户已删除！');\n await API.get('/api/user/logout');\n userDispatch({ type: 'logout' });\n localStorage.removeItem('user');\n navigate('/login');\n } else {\n showError(message);\n }\n };\n\n const bindWeChat = async () => {\n if (inputs.wechat_verification_code === '') return;\n const res = await API.get(\n `/api/oauth/wechat/bind?code=${inputs.wechat_verification_code}`\n );\n const { success, message } = res.data;\n if (success) {\n showSuccess('微信账户绑定成功！');\n setShowWeChatBindModal(false);\n } else {\n showError(message);\n }\n };\n\n const changePassword = async () => {\n if (inputs.set_new_password !== inputs.set_new_password_confirmation) {\n showError('两次输入的密码不一致！');\n return;\n }\n const res = await API.put(\n `/api/user/self`,\n {\n password: inputs.set_new_password\n }\n );\n const { success, message } = res.data;\n if (success) {\n showSuccess('密码修改成功！');\n setShowWeChatBindModal(false);\n } else {\n showError(message);\n }\n setShowChangePasswordModal(false);\n };\n\n const transfer = async () => {\n if (transferAmount < getQuotaPerUnit()) {\n showError('划转金额最低为' + renderQuota(getQuotaPerUnit()));\n return;\n }\n const res = await API.post(\n `/api/user/aff_transfer`,\n {\n quota: transferAmount\n }\n );\n const { success, message } = res.data;\n if (success) {\n showSuccess(message);\n setOpenTransfer(false);\n getUserData().then();\n } else {\n showError(message);\n }\n };\n\n const sendVerificationCode = async () => {\n if (inputs.email === '') {\n showError('请输入邮箱！');\n return;\n }\n setDisableButton(true);\n if (turnstileEnabled && turnstileToken === '') {\n showInfo('请稍后几秒重试，Turnstile 正在检查用户环境！');\n return;\n }\n setLoading(true);\n const res = await API.get(\n `/api/verification?email=${inputs.email}&turnstile=${turnstileToken}`\n );\n const { success, message } = res.data;\n if (success) {\n showSuccess('验证码发送成功，请检查邮箱！');\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n const bindEmail = async () => {\n if (inputs.email_verification_code === '') {\n showError('请输入邮箱验证码！');\n return;\n }\n setLoading(true);\n const res = await API.get(\n `/api/oauth/email/bind?email=${inputs.email}&code=${inputs.email_verification_code}`\n );\n const { success, message } = res.data;\n if (success) {\n showSuccess('邮箱账户绑定成功！');\n setShowEmailBindModal(false);\n userState.user.email = inputs.email;\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n const getUsername = () => {\n if (userState.user) {\n return userState.user.username;\n } else {\n return 'null';\n }\n };\n\n const handleCancel = () => {\n setOpenTransfer(false);\n };\n\n const copyText = async (text) => {\n if (await copy(text)) {\n showSuccess('已复制：' + text);\n } else {\n // setSearchKeyword(text);\n Modal.error({ title: '无法复制到剪贴板，请手动复制', content: text });\n }\n };\n\n return (\n \n
\n \n \n \n {`可用额度${renderQuotaWithPrompt(userState?.user?.aff_quota)}`} \n
\n \n
{`划转额度${renderQuotaWithPrompt(transferAmount)} 最低` + renderQuota(getQuotaPerUnit())} \n
\n setTransferAmount(value)} disabled={false}> \n
\n
\n
\n {typeof getUsername() === 'string' && getUsername().slice(0, 1)}\n }\n title={{getUsername()} }\n description={isRoot() ? 管理员 : 普通用户 }\n >\n }\n headerExtraContent={\n <>\n \n {'ID: ' + userState?.user?.id} \n {userState?.user?.group} \n \n \n }\n footer={\n \n {renderQuota(userState?.user?.quota)} \n {renderQuota(userState?.user?.used_quota)} \n {userState.user?.request_count} \n \n }\n >\n 调用信息 \n 可用模型（可点击复制）
\n \n \n {models.map((model) => (\n {\n copyText(model);\n }}>\n {model}\n \n ))}\n \n
\n \n {/* \n 邀请链接 \n 邀请信息 \n \n \n \n \n {\n renderQuota(userState?.user?.aff_quota)\n }\n \n setOpenTransfer(true)} size={'small'}\n style={{ marginLeft: 10 }}>划转 \n \n {renderQuota(userState?.user?.aff_history_quota)}\n {userState?.user?.aff_count} \n \n
\n */}\n \n 邀请链接 \n \n \n 个人信息 \n \n
邮箱 \n
\n
\n \n
\n
\n {\n setShowEmailBindModal(true);\n }}>{\n userState.user && userState.user.email !== '' ? '修改绑定' : '绑定邮箱'\n } \n
\n
\n
\n
微信 \n
\n
\n \n
\n
\n \n {\n status.wechat_login ? '绑定' : '未启用'\n }\n \n
\n
\n
\n
GitHub \n
\n
\n \n
\n
\n {\n onGitHubOAuthClicked(status.github_client_id);\n }}\n disabled={(userState.user && userState.user.github_id !== '') || !status.github_oauth}\n >\n {\n status.github_oauth ? '绑定' : '未启用'\n }\n \n
\n
\n
\n
Telegram \n
\n
\n \n
\n
\n {status.telegram_oauth ?\n userState.user.telegram_id !== '' ? 已绑定 \n : 未启用 \n }\n
\n
\n
\n
\n 生成系统访问令牌 \n {\n setShowChangePasswordModal(true);\n }}>修改密码 \n {\n setShowAccountDeleteModal(true);\n }}>删除个人账户 \n \n\n {systemToken && (\n
\n )}\n {\n status.wechat_login && (\n
{\n setShowWeChatBindModal(true);\n }}\n >\n 绑定微信账号\n \n )\n }\n setShowWeChatBindModal(false)}\n // onOpen={() => setShowWeChatBindModal(true)}\n visible={showWeChatBindModal}\n size={'mini'}\n >\n \n
\n 微信扫码关注公众号，输入「验证码」获取验证码（三分钟内有效）\n
\n
handleInputChange('wechat_verification_code', v)}\n />\n \n 绑定\n \n \n \n setShowEmailBindModal(false)}\n // onOpen={() => setShowEmailBindModal(true)}\n onOk={bindEmail}\n visible={showEmailBindModal}\n size={'small'}\n centered={true}\n maskClosable={false}\n >\n 绑定邮箱地址 \n \n handleInputChange('email', value)}\n name=\"email\"\n type=\"email\"\n />\n \n {disableButton ? `重新发送(${countdown})` : '获取验证码'}\n \n
\n \n handleInputChange('email_verification_code', value)}\n />\n
\n {turnstileEnabled ? (\n {\n setTurnstileToken(token);\n }}\n />\n ) : (\n <>\n )}\n \n setShowAccountDeleteModal(false)}\n visible={showAccountDeleteModal}\n size={'small'}\n centered={true}\n onOk={deleteAccount}\n >\n \n
\n \n handleInputChange('self_account_deletion_confirmation', value)}\n />\n {turnstileEnabled ? (\n {\n setTurnstileToken(token);\n }}\n />\n ) : (\n <>\n )}\n
\n \n setShowChangePasswordModal(false)}\n visible={showChangePasswordModal}\n size={'small'}\n centered={true}\n onOk={changePassword}\n >\n \n handleInputChange('set_new_password', value)}\n />\n handleInputChange('set_new_password_confirmation', value)}\n />\n {turnstileEnabled ? (\n {\n setTurnstileToken(token);\n }}\n />\n ) : (\n <>\n )}\n
\n \n 未使用 ;\n case 2:\n return 已禁用 ;\n case 3:\n return 已使用 ;\n default:\n return 未知状态 ;\n }\n}\n\nconst RedemptionsTable = () => {\n const columns = [\n {\n title: 'ID',\n dataIndex: 'id'\n },\n {\n title: '名称',\n dataIndex: 'name'\n },\n {\n title: '状态',\n dataIndex: 'status',\n key: 'status',\n render: (text, record, index) => {\n return (\n \n {renderStatus(text)}\n
\n );\n }\n },\n {\n title: '额度',\n dataIndex: 'quota',\n render: (text, record, index) => {\n return (\n \n {renderQuota(parseInt(text))}\n
\n );\n }\n },\n {\n title: '创建时间',\n dataIndex: 'created_time',\n render: (text, record, index) => {\n return (\n \n {renderTimestamp(text)}\n
\n );\n }\n },\n // {\n // title: '兑换人ID',\n // dataIndex: 'used_user_id',\n // render: (text, record, index) => {\n // return (\n // \n // {text === 0 ? '无' : text}\n //
\n // );\n // }\n // },\n {\n title: '',\n dataIndex: 'operate',\n render: (text, record, index) => (\n \n
\n 查看 \n \n {\n await copyText(record.key);\n }}\n >复制 \n {\n manageRedemption(record.id, 'delete', record).then(\n () => {\n removeRecord(record.key);\n }\n );\n }}\n >\n 删除 \n \n {\n record.status === 1 ?\n {\n manageRedemption(\n record.id,\n 'disable',\n record\n );\n }\n }>禁用 :\n {\n manageRedemption(\n record.id,\n 'enable',\n record\n );\n }\n } disabled={record.status === 3}>启用 \n }\n {\n setEditingRedemption(record);\n setShowEdit(true);\n }\n } disabled={record.status !== 1}>编辑 \n `第 ${page.currentStart} - ${page.currentEnd} 条，共 ${redemptions.length} 条`,\n // onPageSizeChange: (size) => {\n // setPageSize(size);\n // setActivePage(1);\n // },\n onPageChange: handlePageChange\n }} loading={loading} rowSelection={rowSelection} onRow={handleRow}>\n
\n {\n setEditingRedemption({\n id: undefined\n });\n setShowEdit(true);\n }\n }>添加兑换码 \n {\n if (selectedKeys.length === 0) {\n showError('请至少选择一个兑换码！');\n return;\n }\n let keys = '';\n for (let i = 0; i < selectedKeys.length; i++) {\n keys += selectedKeys[i].name + ' ' + selectedKeys[i].key + '\\n';\n }\n await copyText(keys);\n }\n }>复制所选兑换码到剪贴板 \n \n );\n};\n\nexport default RedemptionsTable;\n"
},
{
"path": "web/air/src/components/RegisterForm.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { Button, Form, Grid, Header, Image, Message, Segment } from 'semantic-ui-react';\nimport { Link, useNavigate } from 'react-router-dom';\nimport { API, getLogo, showError, showInfo, showSuccess } from '../helpers';\nimport Turnstile from 'react-turnstile';\n\nconst RegisterForm = () => {\n const [inputs, setInputs] = useState({\n username: '',\n password: '',\n password2: '',\n email: '',\n verification_code: ''\n });\n const { username, password, password2 } = inputs;\n const [showEmailVerification, setShowEmailVerification] = useState(false);\n const [turnstileEnabled, setTurnstileEnabled] = useState(false);\n const [turnstileSiteKey, setTurnstileSiteKey] = useState('');\n const [turnstileToken, setTurnstileToken] = useState('');\n const [loading, setLoading] = useState(false);\n const logo = getLogo();\n let affCode = new URLSearchParams(window.location.search).get('aff');\n if (affCode) {\n localStorage.setItem('aff', affCode);\n }\n\n useEffect(() => {\n let status = localStorage.getItem('status');\n if (status) {\n status = JSON.parse(status);\n setShowEmailVerification(status.email_verification);\n if (status.turnstile_check) {\n setTurnstileEnabled(true);\n setTurnstileSiteKey(status.turnstile_site_key);\n }\n }\n });\n\n let navigate = useNavigate();\n\n function handleChange(e) {\n const { name, value } = e.target;\n console.log(name, value);\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n }\n\n async function handleSubmit(e) {\n if (password.length < 8) {\n showInfo('密码长度不得小于 8 位！');\n return;\n }\n if (password !== password2) {\n showInfo('两次输入的密码不一致');\n return;\n }\n if (username && password) {\n if (turnstileEnabled && turnstileToken === '') {\n showInfo('请稍后几秒重试，Turnstile 正在检查用户环境！');\n return;\n }\n setLoading(true);\n if (!affCode) {\n affCode = localStorage.getItem('aff');\n }\n inputs.aff_code = affCode;\n const res = await API.post(\n `/api/user/register?turnstile=${turnstileToken}`,\n inputs\n );\n const { success, message } = res.data;\n if (success) {\n navigate('/login');\n showSuccess('注册成功！');\n } else {\n showError(message);\n }\n setLoading(false);\n }\n }\n\n const sendVerificationCode = async () => {\n if (inputs.email === '') return;\n if (turnstileEnabled && turnstileToken === '') {\n showInfo('请稍后几秒重试，Turnstile 正在检查用户环境！');\n return;\n }\n setLoading(true);\n const res = await API.get(\n `/api/verification?email=${inputs.email}&turnstile=${turnstileToken}`\n );\n const { success, message } = res.data;\n if (success) {\n showSuccess('验证码发送成功，请检查你的邮箱！');\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n return (\n \n \n \n \n \n 已有账户？\n \n \n \n );\n};\n\nexport default RegisterForm;\n"
},
{
"path": "web/air/src/components/SiderBar.js",
"content": "import React, { useContext, useEffect, useMemo, useState } from 'react';\nimport { Link, useNavigate } from 'react-router-dom';\nimport { UserContext } from '../context/User';\nimport { StatusContext } from '../context/Status';\n\nimport { API, getLogo, getSystemName, isAdmin, isMobile, showError } from '../helpers';\nimport '../index.css';\n\nimport {\n IconCalendarClock,\n IconComment,\n IconCreditCard,\n IconGift,\n IconHistogram,\n IconHome,\n IconImage,\n IconKey,\n IconLayers,\n IconSetting,\n IconUser\n} from '@douyinfe/semi-icons';\nimport { Layout, Nav } from '@douyinfe/semi-ui';\n\n// HeaderBar Buttons\n\nconst SiderBar = () => {\n const [userState, userDispatch] = useContext(UserContext);\n const [statusState, statusDispatch] = useContext(StatusContext);\n const defaultIsCollapsed = isMobile() || localStorage.getItem('default_collapse_sidebar') === 'true';\n\n let navigate = useNavigate();\n const [selectedKeys, setSelectedKeys] = useState(['home']);\n const systemName = getSystemName();\n const logo = getLogo();\n const [isCollapsed, setIsCollapsed] = useState(defaultIsCollapsed);\n\n const headerButtons = useMemo(() => [\n {\n text: '首页',\n itemKey: 'home',\n to: '/',\n icon: \n \n
{\n setIsCollapsed(collapsed);\n }}\n selectedKeys={selectedKeys}\n renderWrapper={({ itemElement, isSubNav, isInSubNav, props }) => {\n const routerMap = {\n home: '/',\n channel: '/channel',\n token: '/token',\n redemption: '/redemption',\n topup: '/topup',\n user: '/user',\n log: '/log',\n midjourney: '/midjourney',\n setting: '/setting',\n about: '/about',\n chat: '/chat',\n detail: '/detail'\n };\n return (\n \n {itemElement}\n \n );\n }}\n items={headerButtons}\n onSelect={key => {\n setSelectedKeys([key.itemKey]);\n }}\n header={{\n logo: \n \n \n \n \n );\n};\n\nexport default SiderBar;\n"
},
{
"path": "web/air/src/components/SystemSetting.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { Button, Divider, Form, Grid, Header, Modal, Message } from 'semantic-ui-react';\nimport { API, removeTrailingSlash, showError } from '../helpers';\n\nconst SystemSetting = () => {\n let [inputs, setInputs] = useState({\n PasswordLoginEnabled: '',\n PasswordRegisterEnabled: '',\n EmailVerificationEnabled: '',\n GitHubOAuthEnabled: '',\n GitHubClientId: '',\n GitHubClientSecret: '',\n Notice: '',\n SMTPServer: '',\n SMTPPort: '',\n SMTPAccount: '',\n SMTPFrom: '',\n SMTPToken: '',\n ServerAddress: '',\n Footer: '',\n WeChatAuthEnabled: '',\n WeChatServerAddress: '',\n WeChatServerToken: '',\n WeChatAccountQRCodeImageURL: '',\n MessagePusherAddress: '',\n MessagePusherToken: '',\n TurnstileCheckEnabled: '',\n TurnstileSiteKey: '',\n TurnstileSecretKey: '',\n RegisterEnabled: '',\n EmailDomainRestrictionEnabled: '',\n EmailDomainWhitelist: ''\n });\n const [originInputs, setOriginInputs] = useState({});\n let [loading, setLoading] = useState(false);\n const [EmailDomainWhitelist, setEmailDomainWhitelist] = useState([]);\n const [restrictedDomainInput, setRestrictedDomainInput] = useState('');\n const [showPasswordWarningModal, setShowPasswordWarningModal] = useState(false);\n\n const getOptions = async () => {\n const res = await API.get('/api/option/');\n const { success, message, data } = res.data;\n if (success) {\n let newInputs = {};\n data.forEach((item) => {\n newInputs[item.key] = item.value;\n });\n setInputs({\n ...newInputs,\n EmailDomainWhitelist: newInputs.EmailDomainWhitelist.split(',')\n });\n setOriginInputs(newInputs);\n\n setEmailDomainWhitelist(newInputs.EmailDomainWhitelist.split(',').map((item) => {\n return { key: item, text: item, value: item };\n }));\n } else {\n showError(message);\n }\n };\n\n useEffect(() => {\n getOptions().then();\n }, []);\n\n const updateOption = async (key, value) => {\n setLoading(true);\n switch (key) {\n case 'PasswordLoginEnabled':\n case 'PasswordRegisterEnabled':\n case 'EmailVerificationEnabled':\n case 'GitHubOAuthEnabled':\n case 'WeChatAuthEnabled':\n case 'TurnstileCheckEnabled':\n case 'EmailDomainRestrictionEnabled':\n case 'RegisterEnabled':\n value = inputs[key] === 'true' ? 'false' : 'true';\n break;\n default:\n break;\n }\n const res = await API.put('/api/option/', {\n key,\n value\n });\n const { success, message } = res.data;\n if (success) {\n if (key === 'EmailDomainWhitelist') {\n value = value.split(',');\n }\n setInputs((inputs) => ({\n ...inputs, [key]: value\n }));\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n const handleInputChange = async (e, { name, value }) => {\n if (name === 'PasswordLoginEnabled' && inputs[name] === 'true') {\n // block disabling password login\n setShowPasswordWarningModal(true);\n return;\n }\n if (\n name === 'Notice' ||\n name.startsWith('SMTP') ||\n name === 'ServerAddress' ||\n name === 'GitHubClientId' ||\n name === 'GitHubClientSecret' ||\n name === 'WeChatServerAddress' ||\n name === 'WeChatServerToken' ||\n name === 'WeChatAccountQRCodeImageURL' ||\n name === 'TurnstileSiteKey' ||\n name === 'TurnstileSecretKey' ||\n name === 'EmailDomainWhitelist'\n ) {\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n } else {\n await updateOption(name, value);\n }\n };\n\n const submitServerAddress = async () => {\n let ServerAddress = removeTrailingSlash(inputs.ServerAddress);\n await updateOption('ServerAddress', ServerAddress);\n };\n\n const submitSMTP = async () => {\n if (originInputs['SMTPServer'] !== inputs.SMTPServer) {\n await updateOption('SMTPServer', inputs.SMTPServer);\n }\n if (originInputs['SMTPAccount'] !== inputs.SMTPAccount) {\n await updateOption('SMTPAccount', inputs.SMTPAccount);\n }\n if (originInputs['SMTPFrom'] !== inputs.SMTPFrom) {\n await updateOption('SMTPFrom', inputs.SMTPFrom);\n }\n if (\n originInputs['SMTPPort'] !== inputs.SMTPPort &&\n inputs.SMTPPort !== ''\n ) {\n await updateOption('SMTPPort', inputs.SMTPPort);\n }\n if (\n originInputs['SMTPToken'] !== inputs.SMTPToken &&\n inputs.SMTPToken !== ''\n ) {\n await updateOption('SMTPToken', inputs.SMTPToken);\n }\n };\n\n\n const submitEmailDomainWhitelist = async () => {\n if (\n originInputs['EmailDomainWhitelist'] !== inputs.EmailDomainWhitelist.join(',') &&\n inputs.SMTPToken !== ''\n ) {\n await updateOption('EmailDomainWhitelist', inputs.EmailDomainWhitelist.join(','));\n }\n };\n\n const submitWeChat = async () => {\n if (originInputs['WeChatServerAddress'] !== inputs.WeChatServerAddress) {\n await updateOption(\n 'WeChatServerAddress',\n removeTrailingSlash(inputs.WeChatServerAddress)\n );\n }\n if (\n originInputs['WeChatAccountQRCodeImageURL'] !==\n inputs.WeChatAccountQRCodeImageURL\n ) {\n await updateOption(\n 'WeChatAccountQRCodeImageURL',\n inputs.WeChatAccountQRCodeImageURL\n );\n }\n if (\n originInputs['WeChatServerToken'] !== inputs.WeChatServerToken &&\n inputs.WeChatServerToken !== ''\n ) {\n await updateOption('WeChatServerToken', inputs.WeChatServerToken);\n }\n };\n\n const submitMessagePusher = async () => {\n if (originInputs['MessagePusherAddress'] !== inputs.MessagePusherAddress) {\n await updateOption(\n 'MessagePusherAddress',\n removeTrailingSlash(inputs.MessagePusherAddress)\n );\n }\n if (\n originInputs['MessagePusherToken'] !== inputs.MessagePusherToken &&\n inputs.MessagePusherToken !== ''\n ) {\n await updateOption('MessagePusherToken', inputs.MessagePusherToken);\n }\n };\n\n const submitGitHubOAuth = async () => {\n if (originInputs['GitHubClientId'] !== inputs.GitHubClientId) {\n await updateOption('GitHubClientId', inputs.GitHubClientId);\n }\n if (\n originInputs['GitHubClientSecret'] !== inputs.GitHubClientSecret &&\n inputs.GitHubClientSecret !== ''\n ) {\n await updateOption('GitHubClientSecret', inputs.GitHubClientSecret);\n }\n };\n\n const submitTurnstile = async () => {\n if (originInputs['TurnstileSiteKey'] !== inputs.TurnstileSiteKey) {\n await updateOption('TurnstileSiteKey', inputs.TurnstileSiteKey);\n }\n if (\n originInputs['TurnstileSecretKey'] !== inputs.TurnstileSecretKey &&\n inputs.TurnstileSecretKey !== ''\n ) {\n await updateOption('TurnstileSecretKey', inputs.TurnstileSecretKey);\n }\n };\n\n const submitNewRestrictedDomain = () => {\n const localDomainList = inputs.EmailDomainWhitelist;\n if (restrictedDomainInput !== '' && !localDomainList.includes(restrictedDomainInput)) {\n setRestrictedDomainInput('');\n setInputs({\n ...inputs,\n EmailDomainWhitelist: [...localDomainList, restrictedDomainInput],\n });\n setEmailDomainWhitelist([...EmailDomainWhitelist, {\n key: restrictedDomainInput,\n text: restrictedDomainInput,\n value: restrictedDomainInput,\n }]);\n }\n }\n\n return (\n \n \n \n \n \n );\n};\n\nexport default SystemSetting;\n"
},
{
"path": "web/air/src/components/TokensTable.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { API, copy, showError, showSuccess, timestamp2string } from '../helpers';\n\nimport { ITEMS_PER_PAGE } from '../constants';\nimport { renderQuota } from '../helpers/render';\nimport { Button, Dropdown, Form, Modal, Popconfirm, Popover, SplitButtonGroup, Table, Tag } from '@douyinfe/semi-ui';\n\nimport { IconTreeTriangleDown } from '@douyinfe/semi-icons';\nimport EditToken from '../pages/Token/EditToken';\n\nconst COPY_OPTIONS = [\n { key: 'next', text: 'ChatGPT Next Web', value: 'next' },\n { key: 'ama', text: 'ChatGPT Web & Midjourney', value: 'ama' },\n { key: 'opencat', text: 'OpenCat', value: 'opencat' },\n { key: 'lobechat', text: 'LobeChat', value: 'lobechat' },\n];\n\nconst OPEN_LINK_OPTIONS = [\n { key: 'ama', text: 'ChatGPT Web & Midjourney', value: 'ama' },\n { key: 'opencat', text: 'OpenCat', value: 'opencat' },\n { key: 'lobechat', text: 'LobeChat', value: 'lobechat' }\n];\n\nfunction renderTimestamp(timestamp) {\n return (\n <>\n {timestamp2string(timestamp)}\n \n );\n}\n\nfunction renderStatus(status, model_limits_enabled = false) {\n switch (status) {\n case 1:\n if (model_limits_enabled) {\n return 已启用：限制模型 ;\n } else {\n return 已启用 ;\n }\n case 2:\n return 已禁用 ;\n case 3:\n return 已过期 ;\n case 4:\n return 已耗尽 ;\n default:\n return 未知状态 ;\n }\n}\n\nconst TokensTable = () => {\n\n const link_menu = [\n {\n node: 'item', key: 'next', name: 'ChatGPT Next Web', onClick: () => {\n onOpenLink('next');\n }\n },\n { node: 'item', key: 'ama', name: 'AMA 问天', value: 'ama' },\n {\n node: 'item', key: 'next-mj', name: 'ChatGPT Web & Midjourney', value: 'next-mj', onClick: () => {\n onOpenLink('next-mj');\n }\n },\n { node: 'item', key: 'opencat', name: 'OpenCat', value: 'opencat' },\n {\n node: 'item', key: 'lobechat', name: 'LobeChat', onClick: () => {\n onOpenLink('lobechat');\n }\n }\n ];\n\n const columns = [\n {\n title: '名称',\n dataIndex: 'name'\n },\n {\n title: '状态',\n dataIndex: 'status',\n key: 'status',\n render: (text, record, index) => {\n return (\n \n {renderStatus(text, record.model_limits_enabled)}\n
\n );\n }\n },\n {\n title: '已用额度',\n dataIndex: 'used_quota',\n render: (text, record, index) => {\n return (\n \n {renderQuota(parseInt(text))}\n
\n );\n }\n },\n {\n title: '剩余额度',\n dataIndex: 'remain_quota',\n render: (text, record, index) => {\n return (\n \n {record.unlimited_quota ? 无限制 :\n {renderQuota(parseInt(text))} }\n
\n );\n }\n },\n {\n title: '创建时间',\n dataIndex: 'created_time',\n render: (text, record, index) => {\n return (\n \n {renderTimestamp(text)}\n
\n );\n }\n },\n {\n title: '过期时间',\n dataIndex: 'expired_time',\n render: (text, record, index) => {\n return (\n \n {record.expired_time === -1 ? '永不过期' : renderTimestamp(text)}\n
\n );\n }\n },\n {\n title: '',\n dataIndex: 'operate',\n render: (text, record, index) => (\n \n
\n 查看 \n \n {\n await copyText('sk-' + record.key);\n }}\n >复制 \n \n {\n onOpenLink('next', record.key);\n }}>聊天 \n {\n onOpenLink('next', record.key);\n }\n },\n {\n node: 'item',\n key: 'next-mj',\n disabled: !localStorage.getItem('chat_link2'),\n name: 'ChatGPT Web & Midjourney',\n onClick: () => {\n onOpenLink('next-mj', record.key);\n }\n },\n {\n node: 'item', key: 'ama', name: 'AMA 问天（BotGem）', onClick: () => {\n onOpenLink('ama', record.key);\n }\n },\n {\n node: 'item', key: 'opencat', name: 'OpenCat', onClick: () => {\n onOpenLink('opencat', record.key);\n }\n },\n {\n node: 'item', key: 'lobechat', name: 'LobeChat', onClick: () => {\n onOpenLink('lobechat');\n }\n }\n ]\n }\n >\n \n \n {\n manageToken(record.id, 'delete', record).then(\n () => {\n removeRecord(record.key);\n }\n );\n }}\n >\n 删除 \n \n {\n record.status === 1 ?\n {\n manageToken(\n record.id,\n 'disable',\n record\n );\n }\n }>禁用 :\n {\n manageToken(\n record.id,\n 'enable',\n record\n );\n }\n }>启用 \n }\n {\n setEditingToken(record);\n setShowEdit(true);\n }\n }>编辑 \n `第 ${page.currentStart} - ${page.currentEnd} 条，共 ${tokens.length} 条`,\n onPageSizeChange: (size) => {\n setPageSize(size);\n setActivePage(1);\n },\n onPageChange: handlePageChange\n }} loading={loading} rowSelection={rowSelection} onRow={handleRow}>\n
\n {\n setEditingToken({\n id: undefined\n });\n setShowEdit(true);\n }\n }>添加令牌 \n {\n if (selectedKeys.length === 0) {\n showError('请至少选择一个令牌！');\n return;\n }\n let keys = '';\n for (let i = 0; i < selectedKeys.length; i++) {\n keys += selectedKeys[i].name + ' sk-' + selectedKeys[i].key + '\\n';\n }\n await copyText(keys);\n }\n }>复制所选令牌到剪贴板 \n setDropdownVisible(visible)}\n render={\n \n handleOrderByChange('', { value: '' })}>默认排序 \n handleOrderByChange('', { value: 'remain_quota' })}>按剩余额度排序 \n handleOrderByChange('', { value: 'used_quota' })}>按已用额度排序 \n \n }\n >\n {renderSelectedOption(orderBy)} \n \n \n );\n};\n\nexport default TokensTable;\n"
},
{
"path": "web/air/src/components/UsersTable.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { API, showError, showSuccess } from '../helpers';\nimport { Button, Form, Popconfirm, Space, Table, Tag, Tooltip, Dropdown } from '@douyinfe/semi-ui';\nimport { ITEMS_PER_PAGE } from '../constants';\nimport { renderGroup, renderNumber, renderQuota } from '../helpers/render';\nimport AddUser from '../pages/User/AddUser';\nimport EditUser from '../pages/User/EditUser';\n\nfunction renderRole(role) {\n switch (role) {\n case 1:\n return 普通用户 ;\n case 10:\n return 管理员 ;\n case 100:\n return 超级管理员 ;\n default:\n return 未知身份 ;\n }\n}\n\nconst UsersTable = () => {\n const columns = [{\n title: 'ID', dataIndex: 'id'\n }, {\n title: '用户名', dataIndex: 'username'\n }, {\n title: '分组', dataIndex: 'group', render: (text, record, index) => {\n return (\n {renderGroup(text)}\n
);\n }\n }, {\n title: '统计信息', dataIndex: 'info', render: (text, record, index) => {\n return (\n \n \n {renderQuota(record.quota)} \n \n \n {renderQuota(record.used_quota)} \n \n \n {renderNumber(record.request_count)} \n \n \n
);\n }\n },\n // {\n // title: '邀请信息', dataIndex: 'invite', render: (text, record, index) => {\n // return (\n // \n // \n // {renderNumber(record.aff_count)} \n // \n // \n // {renderQuota(record.aff_history_quota)} \n // \n // \n // {record.inviter_id === 0 ? 无 :\n // {record.inviter_id} }\n // \n // \n //
);\n // }\n // },\n {\n title: '角色', dataIndex: 'role', render: (text, record, index) => {\n return (\n {renderRole(text)}\n
);\n }\n },\n {\n title: '状态', dataIndex: 'status', render: (text, record, index) => {\n return (\n {renderStatus(text)}\n
);\n }\n },\n {\n title: '', dataIndex: 'operate', render: (text, record, index) => (\n <>\n
{\n manageUser(record.username, 'promote', record);\n }}\n >\n 提升 \n \n {\n manageUser(record.username, 'demote', record);\n }}\n >\n 降级 \n \n {record.status === 1 ?\n {\n manageUser(record.username, 'disable', record);\n }}>禁用 :\n {\n manageUser(record.username, 'enable', record);\n }} disabled={record.status === 3}>启用 }\n {\n setEditingUser(record);\n setShowEditUser(true);\n }}>编辑 \n \n {\n manageUser(record.username, 'delete', record).then(() => {\n removeRecord(record.id);\n });\n }}\n >\n 删除 \n \n 已激活 ;\n case 2:\n return (\n 已封禁\n );\n default:\n return (\n 未知状态\n );\n }\n };\n\n const searchUsers = async () => {\n if (searchKeyword === '') {\n // if keyword is blank, load files instead.\n await loadUsers(0);\n setActivePage(1);\n setOrderBy('');\n return;\n }\n setSearching(true);\n const res = await API.get(`/api/user/search?keyword=${searchKeyword}`);\n const { success, message, data } = res.data;\n if (success) {\n setUsers(data);\n setActivePage(1);\n } else {\n showError(message);\n }\n setSearching(false);\n };\n\n const handleKeywordChange = async (value) => {\n setSearchKeyword(value.trim());\n };\n\n const sortUser = (key) => {\n if (users.length === 0) return;\n setLoading(true);\n let sortedUsers = [...users];\n sortedUsers.sort((a, b) => {\n return ('' + a[key]).localeCompare(b[key]);\n });\n if (sortedUsers[0].id === users[0].id) {\n sortedUsers.reverse();\n }\n setUsers(sortedUsers);\n setLoading(false);\n };\n\n const handlePageChange = page => {\n setActivePage(page);\n if (page === Math.ceil(users.length / ITEMS_PER_PAGE) + 1) {\n // In this case we have to load more data and then append them.\n loadUsers(page - 1).then(r => {\n });\n }\n };\n\n const pageData = users.slice((activePage - 1) * ITEMS_PER_PAGE, activePage * ITEMS_PER_PAGE);\n\n const closeAddUser = () => {\n setShowAddUser(false);\n };\n\n const closeEditUser = () => {\n setShowEditUser(false);\n setEditingUser({\n id: undefined\n });\n };\n\n const refresh = async () => {\n if (searchKeyword === '') {\n await loadUsers(activePage - 1);\n } else {\n await searchUsers();\n }\n };\n\n const handleOrderByChange = (e, { value }) => {\n setOrderBy(value);\n setActivePage(1);\n setDropdownVisible(false);\n };\n\n const renderSelectedOption = (orderBy) => {\n switch (orderBy) {\n case 'quota':\n return '按剩余额度排序';\n case 'used_quota':\n return '按已用额度排序';\n case 'request_count':\n return '按请求次数排序';\n default:\n return '默认排序';\n }\n };\n\n return (\n <>\n {\n setShowAddUser(true);\n }\n }>添加用户 \n setDropdownVisible(visible)}\n render={\n \n handleOrderByChange('', { value: '' })}>默认排序 \n handleOrderByChange('', { value: 'quota' })}>按剩余额度排序 \n handleOrderByChange('', { value: 'used_quota' })}>按已用额度排序 \n handleOrderByChange('', { value: 'request_count' })}>按请求次数排序 \n \n }\n >\n {renderSelectedOption(orderBy)} \n \n \n );\n};\n\nexport default UsersTable;\n"
},
{
"path": "web/air/src/components/WeChatIcon.js",
"content": "import React from 'react';\nimport { Icon } from '@douyinfe/semi-ui';\n\nconst WeChatIcon = () => {\n function CustomIcon() {\n return \n \n \n ;\n }\n\n return (\n \n
\n );\n};\n\nexport default WeChatIcon;\n"
},
{
"path": "web/air/src/components/utils.js",
"content": "import { API, showError } from '../helpers';\n\nexport async function getOAuthState() {\n const res = await API.get('/api/oauth/state');\n const { success, message, data } = res.data;\n if (success) {\n return data;\n } else {\n showError(message);\n return '';\n }\n}\n\nexport async function onGitHubOAuthClicked(github_client_id) {\n const state = await getOAuthState();\n if (!state) return;\n window.open(\n `https://github.com/login/oauth/authorize?client_id=${github_client_id}&state=${state}&scope=user:email`\n );\n}"
},
{
"path": "web/air/src/constants/channel.constants.js",
"content": "export const CHANNEL_OPTIONS = [\n { key: 1, text: 'OpenAI', value: 1, color: 'green' },\n { key: 14, text: 'Anthropic Claude', value: 14, color: 'black' },\n { key: 33, text: 'AWS', value: 33, color: 'black' },\n { key: 3, text: 'Azure OpenAI', value: 3, color: 'olive' },\n { key: 11, text: 'Google PaLM2', value: 11, color: 'orange' },\n { key: 24, text: 'Google Gemini', value: 24, color: 'orange' },\n { key: 28, text: 'Mistral AI', value: 28, color: 'orange' },\n { key: 41, text: 'Novita', value: 41, color: 'purple' },\n {key: 40, text: '字节火山引擎', value: 40, color: 'blue'},\n { key: 15, text: '百度文心千帆', value: 15, color: 'blue' },\n { key: 17, text: '阿里通义千问', value: 17, color: 'orange' },\n { key: 18, text: '讯飞星火认知', value: 18, color: 'blue' },\n { key: 16, text: '智谱 ChatGLM', value: 16, color: 'violet' },\n { key: 19, text: '360 智脑', value: 19, color: 'blue' },\n { key: 25, text: 'Moonshot AI', value: 25, color: 'black' },\n { key: 23, text: '腾讯混元', value: 23, color: 'teal' },\n { key: 26, text: '百川大模型', value: 26, color: 'orange' },\n { key: 27, text: 'MiniMax', value: 27, color: 'red' },\n { key: 29, text: 'Groq', value: 29, color: 'orange' },\n { key: 30, text: 'Ollama', value: 30, color: 'black' },\n { key: 31, text: '零一万物', value: 31, color: 'green' },\n { key: 32, text: '阶跃星辰', value: 32, color: 'blue' },\n { key: 34, text: 'Coze', value: 34, color: 'blue' },\n { key: 35, text: 'Cohere', value: 35, color: 'blue' },\n { key: 36, text: 'DeepSeek', value: 36, color: 'black' },\n { key: 37, text: 'Cloudflare', value: 37, color: 'orange' },\n { key: 38, text: 'DeepL', value: 38, color: 'black' },\n { key: 39, text: 'together.ai', value: 39, color: 'blue' },\n { key: 42, text: 'VertexAI', value: 42, color: 'blue' },\n { key: 43, text: 'Proxy', value: 43, color: 'blue' },\n { key: 44, text: 'SiliconFlow', value: 44, color: 'blue' },\n { key: 45, text: 'xAI', value: 45, color: 'blue' },\n { key: 46, text: 'Replicate', value: 46, color: 'blue' },\n { key: 8, text: '自定义渠道', value: 8, color: 'pink' },\n { key: 22, text: '知识库：FastGPT', value: 22, color: 'blue' },\n { key: 21, text: '知识库：AI Proxy', value: 21, color: 'purple' },\n {key: 20, text: 'OpenRouter', value: 20, color: 'black'},\n { key: 2, text: '代理：API2D', value: 2, color: 'blue' },\n { key: 5, text: '代理：OpenAI-SB', value: 5, color: 'brown' },\n { key: 7, text: '代理：OhMyGPT', value: 7, color: 'purple' },\n { key: 10, text: '代理：AI Proxy', value: 10, color: 'purple' },\n { key: 4, text: '代理：CloseAI', value: 4, color: 'teal' },\n { key: 6, text: '代理：OpenAI Max', value: 6, color: 'violet' },\n { key: 9, text: '代理：AI.LS', value: 9, color: 'yellow' },\n { key: 12, text: '代理：API2GPT', value: 12, color: 'blue' },\n { key: 13, text: '代理：AIGC2D', value: 13, color: 'purple' }\n];\n\nfor (let i = 0; i < CHANNEL_OPTIONS.length; i++) {\n CHANNEL_OPTIONS[i].label = CHANNEL_OPTIONS[i].text;\n}\n"
},
{
"path": "web/air/src/constants/common.constant.js",
"content": "export const ITEMS_PER_PAGE = 10; // this value must keep same as the one defined in backend!\n"
},
{
"path": "web/air/src/constants/index.js",
"content": "export * from './toast.constants';\nexport * from './user.constants';\nexport * from './common.constant';\nexport * from './channel.constants';"
},
{
"path": "web/air/src/constants/toast.constants.js",
"content": "export const toastConstants = {\n SUCCESS_TIMEOUT: 1500,\n INFO_TIMEOUT: 3000,\n ERROR_TIMEOUT: 5000,\n WARNING_TIMEOUT: 10000,\n NOTICE_TIMEOUT: 20000\n};\n"
},
{
"path": "web/air/src/constants/user.constants.js",
"content": "export const userConstants = {\n REGISTER_REQUEST: 'USERS_REGISTER_REQUEST',\n REGISTER_SUCCESS: 'USERS_REGISTER_SUCCESS',\n REGISTER_FAILURE: 'USERS_REGISTER_FAILURE',\n\n LOGIN_REQUEST: 'USERS_LOGIN_REQUEST',\n LOGIN_SUCCESS: 'USERS_LOGIN_SUCCESS',\n LOGIN_FAILURE: 'USERS_LOGIN_FAILURE',\n \n LOGOUT: 'USERS_LOGOUT',\n\n GETALL_REQUEST: 'USERS_GETALL_REQUEST',\n GETALL_SUCCESS: 'USERS_GETALL_SUCCESS',\n GETALL_FAILURE: 'USERS_GETALL_FAILURE',\n\n DELETE_REQUEST: 'USERS_DELETE_REQUEST',\n DELETE_SUCCESS: 'USERS_DELETE_SUCCESS',\n DELETE_FAILURE: 'USERS_DELETE_FAILURE' \n};\n"
},
{
"path": "web/air/src/context/Status/index.js",
"content": "// contexts/User/index.jsx\n\nimport React from 'react';\nimport { initialState, reducer } from './reducer';\n\nexport const StatusContext = React.createContext({\n state: initialState,\n dispatch: () => null,\n});\n\nexport const StatusProvider = ({ children }) => {\n const [state, dispatch] = React.useReducer(reducer, initialState);\n\n return (\n \n {children}\n \n );\n};"
},
{
"path": "web/air/src/context/Status/reducer.js",
"content": "export const reducer = (state, action) => {\n switch (action.type) {\n case 'set':\n return {\n ...state,\n status: action.payload,\n };\n case 'unset':\n return {\n ...state,\n status: undefined,\n };\n default:\n return state;\n }\n};\n\nexport const initialState = {\n status: undefined,\n};\n"
},
{
"path": "web/air/src/context/User/index.js",
"content": "// contexts/User/index.jsx\n\nimport React from \"react\"\nimport { reducer, initialState } from \"./reducer\"\n\nexport const UserContext = React.createContext({\n state: initialState,\n dispatch: () => null\n})\n\nexport const UserProvider = ({ children }) => {\n const [state, dispatch] = React.useReducer(reducer, initialState)\n\n return (\n \n { children }\n \n )\n}"
},
{
"path": "web/air/src/context/User/reducer.js",
"content": "export const reducer = (state, action) => {\n switch (action.type) {\n case 'login':\n return {\n ...state,\n user: action.payload\n };\n case 'logout':\n return {\n ...state,\n user: undefined\n };\n\n default:\n return state;\n }\n};\n\nexport const initialState = {\n user: undefined\n};"
},
{
"path": "web/air/src/helpers/api.js",
"content": "import { showError } from './utils';\nimport axios from 'axios';\n\nexport const API = axios.create({\n baseURL: process.env.REACT_APP_SERVER ? process.env.REACT_APP_SERVER : '',\n});\n\nAPI.interceptors.response.use(\n (response) => response,\n (error) => {\n showError(error);\n }\n);\n"
},
{
"path": "web/air/src/helpers/auth-header.js",
"content": "export function authHeader() {\n // return authorization header with jwt token\n let user = JSON.parse(localStorage.getItem('user'));\n\n if (user && user.token) {\n return { 'Authorization': 'Bearer ' + user.token };\n } else {\n return {};\n }\n}"
},
{
"path": "web/air/src/helpers/history.js",
"content": "import { createBrowserHistory } from 'history';\n\nexport const history = createBrowserHistory();"
},
{
"path": "web/air/src/helpers/index.js",
"content": "export * from './history';\nexport * from './auth-header';\nexport * from './utils';\nexport * from './api';"
},
{
"path": "web/air/src/helpers/render.js",
"content": "import {Label} from 'semantic-ui-react';\nimport {Tag} from \"@douyinfe/semi-ui\";\n\nexport function renderText(text, limit) {\n if (text.length > limit) {\n return text.slice(0, limit - 3) + '...';\n }\n return text;\n}\n\nexport function renderGroup(group) {\n if (group === '') {\n return default ;\n }\n let groups = group.split(',');\n groups.sort();\n return <>\n {groups.map((group) => {\n if (group === 'vip' || group === 'pro') {\n return {group} ;\n } else if (group === 'svip' || group === 'premium') {\n return {group} ;\n }\n if (group === 'default') {\n return {group} ;\n } else {\n return {group} ;\n }\n })}\n ;\n}\n\nexport function renderNumber(num) {\n if (num >= 1000000000) {\n return (num / 1000000000).toFixed(1) + 'B';\n } else if (num >= 1000000) {\n return (num / 1000000).toFixed(1) + 'M';\n } else if (num >= 10000) {\n return (num / 1000).toFixed(1) + 'k';\n } else {\n return num;\n }\n}\n\nexport function renderQuotaNumberWithDigit(num, digits = 2) {\n let displayInCurrency = localStorage.getItem('display_in_currency');\n num = num.toFixed(digits);\n if (displayInCurrency) {\n return '$' + num;\n }\n return num;\n}\n\nexport function renderNumberWithPoint(num) {\n num = num.toFixed(2);\n if (num >= 100000) {\n // Convert number to string to manipulate it\n let numStr = num.toString();\n // Find the position of the decimal point\n let decimalPointIndex = numStr.indexOf('.');\n\n let wholePart = numStr;\n let decimalPart = '';\n\n // If there is a decimal point, split the number into whole and decimal parts\n if (decimalPointIndex !== -1) {\n wholePart = numStr.slice(0, decimalPointIndex);\n decimalPart = numStr.slice(decimalPointIndex);\n }\n\n // Take the first two and last two digits of the whole number part\n let shortenedWholePart = wholePart.slice(0, 2) + '..' + wholePart.slice(-2);\n\n // Return the formatted number\n return shortenedWholePart + decimalPart;\n }\n\n // If the number is less than 100,000, return it unmodified\n return num;\n}\n\nexport function getQuotaPerUnit() {\n let quotaPerUnit = localStorage.getItem('quota_per_unit');\n quotaPerUnit = parseFloat(quotaPerUnit);\n return quotaPerUnit;\n}\n\nexport function getQuotaWithUnit(quota, digits = 6) {\n let quotaPerUnit = localStorage.getItem('quota_per_unit');\n quotaPerUnit = parseFloat(quotaPerUnit);\n return (quota / quotaPerUnit).toFixed(digits);\n}\n\nexport function renderQuota(quota, digits = 2) {\n let quotaPerUnit = localStorage.getItem('quota_per_unit');\n let displayInCurrency = localStorage.getItem('display_in_currency');\n quotaPerUnit = parseFloat(quotaPerUnit);\n displayInCurrency = displayInCurrency === 'true';\n if (displayInCurrency) {\n return '$' + (quota / quotaPerUnit).toFixed(digits);\n }\n return renderNumber(quota);\n}\n\nexport function renderQuotaWithPrompt(quota, digits) {\n let displayInCurrency = localStorage.getItem('display_in_currency');\n displayInCurrency = displayInCurrency === 'true';\n if (displayInCurrency) {\n return `（等价金额：${renderQuota(quota, digits)}）`;\n }\n return '';\n}\n\nconst colors = ['amber', 'blue', 'cyan', 'green', 'grey', 'indigo',\n 'light-blue', 'lime', 'orange', 'pink',\n 'purple', 'red', 'teal', 'violet', 'yellow'\n]\n\nexport const modelColorMap = {\n 'dall-e': 'rgb(147,112,219)', // 深紫色\n 'dall-e-2': 'rgb(147,112,219)', // 介于紫色和蓝色之间的色调\n 'dall-e-3': 'rgb(153,50,204)', // 介于紫罗兰和洋红之间的色调\n 'midjourney': 'rgb(136,43,180)', // 介于紫罗兰和洋红之间的色调\n 'gpt-3.5-turbo': 'rgb(184,227,167)', // 浅绿色\n 'gpt-3.5-turbo-0301': 'rgb(131,220,131)', // 亮绿色\n 'gpt-3.5-turbo-0613': 'rgb(60,179,113)', // 海洋绿\n 'gpt-3.5-turbo-1106': 'rgb(32,178,170)', // 浅海洋绿\n 'gpt-3.5-turbo-16k': 'rgb(252,200,149)', // 淡橙色\n 'gpt-3.5-turbo-16k-0613': 'rgb(255,181,119)', // 淡桃色\n 'gpt-3.5-turbo-instruct': 'rgb(175,238,238)', // 粉蓝色\n 'gpt-4': 'rgb(135,206,235)', // 天蓝色\n 'gpt-4-0314': 'rgb(70,130,180)', // 钢蓝色\n 'gpt-4-0613': 'rgb(100,149,237)', // 矢车菊蓝\n 'gpt-4-1106-preview': 'rgb(30,144,255)', // 道奇蓝\n 'gpt-4-0125-preview': 'rgb(2,177,236)', // 深天蓝\n 'gpt-4-turbo-preview': 'rgb(2,177,255)', // 深天蓝\n 'gpt-4-32k': 'rgb(104,111,238)', // 中紫色\n 'gpt-4-32k-0314': 'rgb(90,105,205)', // 暗灰蓝色\n 'gpt-4-32k-0613': 'rgb(61,71,139)', // 暗蓝灰色\n 'gpt-4-all': 'rgb(65,105,225)', // 皇家蓝\n 'gpt-4-gizmo-*': 'rgb(0,0,255)', // 纯蓝色\n 'gpt-4-vision-preview': 'rgb(25,25,112)', // 午夜蓝\n 'text-ada-001': 'rgb(255,192,203)', // 粉红色\n 'text-babbage-001': 'rgb(255,160,122)', // 浅珊瑚色\n 'text-curie-001': 'rgb(219,112,147)', // 苍紫罗兰色\n 'text-davinci-002': 'rgb(199,21,133)', // 中紫罗兰红色\n 'text-davinci-003': 'rgb(219,112,147)', // 苍紫罗兰色（与Curie相同，表示同一个系列）\n 'text-davinci-edit-001': 'rgb(255,105,180)', // 热粉色\n 'text-embedding-ada-002': 'rgb(255,182,193)', // 浅粉红\n 'text-embedding-v1': 'rgb(255,174,185)', // 浅粉红色（略有区别）\n 'text-moderation-latest': 'rgb(255,130,171)', // 强粉色\n 'text-moderation-stable': 'rgb(255,160,122)', // 浅珊瑚色（与Babbage相同，表示同一类功能）\n 'tts-1': 'rgb(255,140,0)', // 深橙色\n 'tts-1-1106': 'rgb(255,165,0)', // 橙色\n 'tts-1-hd': 'rgb(255,215,0)', // 金色\n 'tts-1-hd-1106': 'rgb(255,223,0)', // 金黄色（略有区别）\n 'whisper-1': 'rgb(245,245,220)' // 米色\n}\n\nexport function stringToColor(str) {\n let sum = 0;\n // 对字符串中的每个字符进行操作\n for (let i = 0; i < str.length; i++) {\n // 将字符的ASCII值加到sum中\n sum += str.charCodeAt(i);\n }\n // 使用模运算得到个位数\n let i = sum % colors.length;\n return colors[i];\n}"
},
{
"path": "web/air/src/helpers/utils.js",
"content": "import { Toast } from '@douyinfe/semi-ui';\nimport { toastConstants } from '../constants';\nimport React from 'react';\nimport {toast} from \"react-toastify\";\n\nconst HTMLToastContent = ({ htmlContent }) => {\n return
;\n};\nexport default HTMLToastContent;\nexport function isAdmin() {\n let user = localStorage.getItem('user');\n if (!user) return false;\n user = JSON.parse(user);\n return user.role >= 10;\n}\n\nexport function isRoot() {\n let user = localStorage.getItem('user');\n if (!user) return false;\n user = JSON.parse(user);\n return user.role >= 100;\n}\n\nexport function getSystemName() {\n let system_name = localStorage.getItem('system_name');\n if (!system_name) return 'One API';\n return system_name;\n}\n\nexport function getLogo() {\n let logo = localStorage.getItem('logo');\n if (!logo) return '/logo.png';\n return logo\n}\n\nexport function getFooterHTML() {\n return localStorage.getItem('footer_html');\n}\n\nexport async function copy(text) {\n let okay = true;\n try {\n await navigator.clipboard.writeText(text);\n } catch (e) {\n okay = false;\n console.error(e);\n }\n return okay;\n}\n\nexport function isMobile() {\n return window.innerWidth <= 600;\n}\n\nlet showErrorOptions = { autoClose: toastConstants.ERROR_TIMEOUT };\nlet showWarningOptions = { autoClose: toastConstants.WARNING_TIMEOUT };\nlet showSuccessOptions = { autoClose: toastConstants.SUCCESS_TIMEOUT };\nlet showInfoOptions = { autoClose: toastConstants.INFO_TIMEOUT };\nlet showNoticeOptions = { autoClose: false };\n\nif (isMobile()) {\n showErrorOptions.position = 'top-center';\n // showErrorOptions.transition = 'flip';\n\n showSuccessOptions.position = 'top-center';\n // showSuccessOptions.transition = 'flip';\n\n showInfoOptions.position = 'top-center';\n // showInfoOptions.transition = 'flip';\n\n showNoticeOptions.position = 'top-center';\n // showNoticeOptions.transition = 'flip';\n}\n\nexport function showError(error) {\n console.error(error);\n if (error.message) {\n if (error.name === 'AxiosError') {\n switch (error.response.status) {\n case 401:\n // toast.error('错误：未登录或登录已过期，请重新登录！', showErrorOptions);\n window.location.href = '/login?expired=true';\n break;\n case 429:\n Toast.error('错误：请求次数过多，请稍后再试！');\n break;\n case 500:\n Toast.error('错误：服务器内部错误，请联系管理员！');\n break;\n case 405:\n Toast.info('本站仅作演示之用，无服务端！');\n break;\n default:\n Toast.error('错误：' + error.message);\n }\n return;\n }\n Toast.error('错误：' + error.message);\n } else {\n Toast.error('错误：' + error);\n }\n}\n\nexport function showWarning(message) {\n Toast.warning(message);\n}\n\nexport function showSuccess(message) {\n Toast.success(message);\n}\n\nexport function showInfo(message) {\n Toast.info(message);\n}\n\nexport function showNotice(message, isHTML = false) {\n if (isHTML) {\n toast(\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n);\n"
},
{
"path": "web/air/src/pages/About/index.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { Header, Segment } from 'semantic-ui-react';\nimport { API, showError } from '../../helpers';\nimport { marked } from 'marked';\n\nconst About = () => {\n const [about, setAbout] = useState('');\n const [aboutLoaded, setAboutLoaded] = useState(false);\n\n const displayAbout = async () => {\n setAbout(localStorage.getItem('about') || '');\n const res = await API.get('/api/about');\n const { success, message, data } = res.data;\n if (success) {\n let aboutContent = data;\n if (!data.startsWith('https://')) {\n aboutContent = marked.parse(data);\n }\n setAbout(aboutContent);\n localStorage.setItem('about', aboutContent);\n } else {\n showError(message);\n setAbout('加载关于内容失败...');\n }\n setAboutLoaded(true);\n };\n\n useEffect(() => {\n displayAbout().then();\n }, []);\n\n return (\n <>\n {\n aboutLoaded && about === '' ? <>\n \n \n 可在设置页面设置关于内容，支持 HTML & Markdown
\n 项目仓库地址：\n \n https://github.com/songquanpeng/one-api\n \n \n : <>\n {\n about.startsWith('https://') ?
\n }\n \n }\n \n );\n};\n\n\nexport default About;\n"
},
{
"path": "web/air/src/pages/Channel/EditChannel.js",
"content": "import React, {useEffect, useRef, useState} from 'react';\nimport {useNavigate, useParams} from 'react-router-dom';\nimport {API, isMobile, showError, showInfo, showSuccess, verifyJSON} from '../../helpers';\nimport {CHANNEL_OPTIONS} from '../../constants';\nimport Title from \"@douyinfe/semi-ui/lib/es/typography/title\";\nimport {SideSheet, Space, Spin, Button, Input, Typography, Select, TextArea, Checkbox, Banner} from \"@douyinfe/semi-ui\";\n\nconst MODEL_MAPPING_EXAMPLE = {\n 'gpt-3.5-turbo-0301': 'gpt-3.5-turbo',\n 'gpt-4-0314': 'gpt-4',\n 'gpt-4-32k-0314': 'gpt-4-32k'\n};\n\nfunction type2secretPrompt(type) {\n // inputs.type === 15 ? '按照如下格式输入：APIKey|SecretKey' : (inputs.type === 18 ? '按照如下格式输入：APPID|APISecret|APIKey' : '请输入渠道对应的鉴权密钥')\n switch (type) {\n case 15:\n return '按照如下格式输入：APIKey|SecretKey';\n case 18:\n return '按照如下格式输入：APPID|APISecret|APIKey';\n case 22:\n return '按照如下格式输入：APIKey-AppId，例如：fastgpt-0sp2gtvfdgyi4k30jwlgwf1i-64f335d84283f05518e9e041';\n case 23:\n return '按照如下格式输入：AppId|SecretId|SecretKey';\n default:\n return '请输入渠道对应的鉴权密钥';\n }\n}\n\nconst EditChannel = (props) => {\n const navigate = useNavigate();\n const channelId = props.editingChannel.id;\n const isEdit = channelId !== undefined;\n const [loading, setLoading] = useState(isEdit);\n const handleCancel = () => {\n props.handleClose()\n };\n const originInputs = {\n name: '',\n type: 1,\n key: '',\n openai_organization: '',\n base_url: '',\n other: '',\n model_mapping: '',\n system_prompt: '',\n models: [],\n auto_ban: 1,\n groups: ['default']\n };\n const [batch, setBatch] = useState(false);\n const [autoBan, setAutoBan] = useState(true);\n // const [autoBan, setAutoBan] = useState(true);\n const [inputs, setInputs] = useState(originInputs);\n const [originModelOptions, setOriginModelOptions] = useState([]);\n const [modelOptions, setModelOptions] = useState([]);\n const [groupOptions, setGroupOptions] = useState([]);\n const [basicModels, setBasicModels] = useState([]);\n const [fullModels, setFullModels] = useState([]);\n const [customModel, setCustomModel] = useState('');\n const handleInputChange = (name, value) => {\n setInputs((inputs) => ({...inputs, [name]: value}));\n if (name === 'type' && inputs.models.length === 0) {\n let localModels = [];\n switch (value) {\n case 14:\n localModels = [\"claude-instant-1.2\", \"claude-2\", \"claude-2.0\", \"claude-2.1\", \"claude-3-opus-20240229\", \"claude-3-sonnet-20240229\", \"claude-3-haiku-20240307\", \"claude-3-5-haiku-20241022\", \"claude-3-5-sonnet-20240620\", \"claude-3-5-sonnet-20241022\"];\n break;\n case 11:\n localModels = ['PaLM-2'];\n break;\n case 15:\n localModels = ['ERNIE-Bot', 'ERNIE-Bot-turbo', 'ERNIE-Bot-4', 'Embedding-V1'];\n break;\n case 17:\n localModels = [\"qwen-turbo\", \"qwen-plus\", \"qwen-max\", \"qwen-max-longcontext\", 'text-embedding-v1'];\n break;\n case 16:\n localModels = ['chatglm_pro', 'chatglm_std', 'chatglm_lite'];\n break;\n case 18:\n localModels = ['SparkDesk', 'SparkDesk-v1.1', 'SparkDesk-v2.1', 'SparkDesk-v3.1', 'SparkDesk-v3.1-128K', 'SparkDesk-v3.5', 'SparkDesk-v3.5-32K', 'SparkDesk-v4.0'];\n break;\n case 19:\n localModels = ['360GPT_S2_V9', 'embedding-bert-512-v1', 'embedding_s1_v1', 'semantic_similarity_s1_v1'];\n break;\n case 23:\n localModels = ['hunyuan'];\n break;\n case 24:\n localModels = ['gemini-pro', 'gemini-pro-vision'];\n break;\n case 25:\n localModels = ['moonshot-v1-8k', 'moonshot-v1-32k', 'moonshot-v1-128k'];\n break;\n case 26:\n localModels = ['glm-4', 'glm-4v', 'glm-3-turbo'];\n break;\n case 2:\n localModels = ['mj_imagine', 'mj_variation', 'mj_reroll', 'mj_blend', 'mj_upscale', 'mj_describe'];\n break;\n case 5:\n localModels = [\n 'swap_face',\n 'mj_imagine',\n 'mj_variation',\n 'mj_reroll',\n 'mj_blend',\n 'mj_upscale',\n 'mj_describe',\n 'mj_zoom',\n 'mj_shorten',\n 'mj_modal',\n 'mj_inpaint',\n 'mj_custom_zoom',\n 'mj_high_variation',\n 'mj_low_variation',\n 'mj_pan',\n ];\n break;\n }\n setInputs((inputs) => ({...inputs, models: localModels}));\n }\n //setAutoBan\n };\n\n\n const loadChannel = async () => {\n setLoading(true)\n let res = await API.get(`/api/channel/${channelId}`);\n const {success, message, data} = res.data;\n if (success) {\n if (data.models === '') {\n data.models = [];\n } else {\n data.models = data.models.split(',');\n }\n if (data.group === '') {\n data.groups = [];\n } else {\n data.groups = data.group.split(',');\n }\n if (data.model_mapping !== '') {\n data.model_mapping = JSON.stringify(JSON.parse(data.model_mapping), null, 2);\n }\n setInputs(data);\n if (data.auto_ban === 0) {\n setAutoBan(false);\n } else {\n setAutoBan(true);\n }\n // console.log(data);\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n const fetchModels = async () => {\n try {\n let res = await API.get(`/api/channel/models`);\n let localModelOptions = res.data.data.map((model) => ({\n label: model.id,\n value: model.id\n }));\n setOriginModelOptions(localModelOptions);\n setFullModels(res.data.data.map((model) => model.id));\n setBasicModels(res.data.data.filter((model) => {\n return model.id.startsWith('gpt-3') || model.id.startsWith('text-');\n }).map((model) => model.id));\n } catch (error) {\n showError(error.message);\n }\n };\n\n const fetchGroups = async () => {\n try {\n let res = await API.get(`/api/group/`);\n setGroupOptions(res.data.data.map((group) => ({\n label: group,\n value: group\n })));\n } catch (error) {\n showError(error.message);\n }\n };\n\n useEffect(() => {\n let localModelOptions = [...originModelOptions];\n inputs.models.forEach((model) => {\n if (!localModelOptions.find((option) => option.key === model)) {\n localModelOptions.push({\n label: model,\n value: model\n });\n }\n });\n setModelOptions(localModelOptions);\n }, [originModelOptions, inputs.models]);\n\n useEffect(() => {\n fetchModels().then();\n fetchGroups().then();\n if (isEdit) {\n loadChannel().then(\n () => {\n\n }\n );\n } else {\n setInputs(originInputs)\n }\n }, [props.editingChannel.id]);\n\n\n const submit = async () => {\n if (!isEdit && (inputs.name === '' || inputs.key === '')) {\n showInfo('请填写渠道名称和渠道密钥！');\n return;\n }\n if (inputs.models.length === 0) {\n showInfo('请至少选择一个模型！');\n return;\n }\n if (inputs.model_mapping !== '' && !verifyJSON(inputs.model_mapping)) {\n showInfo('模型映射必须是合法的 JSON 格式！');\n return;\n }\n let localInputs = {...inputs};\n if (localInputs.base_url && localInputs.base_url.endsWith('/')) {\n localInputs.base_url = localInputs.base_url.slice(0, localInputs.base_url.length - 1);\n }\n if (localInputs.type === 3 && localInputs.other === '') {\n localInputs.other = '2024-03-01-preview';\n }\n if (localInputs.type === 18 && localInputs.other === '') {\n localInputs.other = 'v2.1';\n }\n let res;\n if (!Array.isArray(localInputs.models)) {\n showError('提交失败，请勿重复提交！');\n handleCancel();\n return;\n }\n localInputs.auto_ban = autoBan ? 1 : 0;\n localInputs.models = localInputs.models.join(',');\n localInputs.group = localInputs.groups.join(',');\n if (isEdit) {\n res = await API.put(`/api/channel/`, {...localInputs, id: parseInt(channelId)});\n } else {\n res = await API.post(`/api/channel/`, localInputs);\n }\n const {success, message} = res.data;\n if (success) {\n if (isEdit) {\n showSuccess('渠道更新成功！');\n } else {\n showSuccess('渠道创建成功！');\n setInputs(originInputs);\n }\n props.refresh();\n props.handleClose();\n } else {\n showError(message);\n }\n };\n\n const addCustomModel = () => {\n if (customModel.trim() === '') return;\n if (inputs.models.includes(customModel)) return showError(\"该模型已存在！\");\n let localModels = [...inputs.models];\n localModels.push(customModel);\n let localModelOptions = [];\n localModelOptions.push({\n key: customModel,\n text: customModel,\n value: customModel\n });\n setModelOptions(modelOptions => {\n return [...modelOptions, ...localModelOptions];\n });\n setCustomModel('');\n handleInputChange('models', localModels);\n };\n\n return (\n <>\n {isEdit ? '更新渠道信息' : '创建新的渠道'}}\n headerStyle={{borderBottom: '1px solid var(--semi-color-border)'}}\n bodyStyle={{borderBottom: '1px solid var(--semi-color-border)'}}\n visible={props.visible}\n footer={\n \n \n 提交 \n 取消 \n \n
\n }\n closeIcon={null}\n onCancel={() => handleCancel()}\n width={isMobile() ? '100%' : 600}\n >\n \n \n 类型： \n
\n handleInputChange('type', value)}\n style={{ width: '50%' }}\n />\n {\n inputs.type === 3 && (\n <>\n \n
\n 注意，模型部署名称必须和模型名称保持一致 ，因为 One API 会把请求体中的\n model\n 参数替换为你的部署名称（模型名称中的点会被剔除），图片演示 。\n \n }>\n \n
\n AZURE_OPENAI_ENDPOINT： \n
\n {\n handleInputChange('base_url', value)\n }}\n value={inputs.base_url}\n autoComplete='new-password'\n />\n \n 默认 API 版本： \n
\n {\n handleInputChange('other', value)\n }}\n value={inputs.other}\n autoComplete='new-password'\n />\n \n )\n }\n {\n inputs.type === 8 && (\n <>\n \n Base URL： \n
\n {\n handleInputChange('base_url', value)\n }}\n value={inputs.base_url}\n autoComplete='new-password'\n />\n \n )\n }\n \n 名称： \n
\n {\n handleInputChange('name', value)\n }}\n value={inputs.name}\n autoComplete='new-password'\n />\n \n 分组： \n
\n {\n handleInputChange('groups', value)\n }}\n value={inputs.groups}\n autoComplete='new-password'\n optionList={groupOptions}\n />\n {\n inputs.type === 18 && (\n <>\n \n 模型版本： \n
\n {\n handleInputChange('other', value)\n }}\n value={inputs.other}\n autoComplete='new-password'\n />\n \n )\n }\n {\n inputs.type === 21 && (\n <>\n \n 知识库 ID： \n
\n {\n handleInputChange('other', value)\n }}\n value={inputs.other}\n autoComplete='new-password'\n />\n \n )\n }\n \n 模型： \n
\n {\n handleInputChange('models', value)\n }}\n value={inputs.models}\n autoComplete='new-password'\n optionList={modelOptions}\n />\n \n \n {\n handleInputChange('models', basicModels);\n }}>填入基础模型 \n {\n handleInputChange('models', fullModels);\n }}>填入所有模型 \n {\n handleInputChange('models', []);\n }}>清除所有模型 \n \n 填入\n }\n placeholder='输入自定义模型名称'\n value={customModel}\n onChange={(value) => {\n setCustomModel(value.trim());\n }}\n />\n
\n \n 模型重定向： \n
\n {\n handleInputChange('model_mapping', value)\n }}\n autosize\n value={inputs.model_mapping}\n autoComplete='new-password'\n />\n \n 系统提示词： \n
\n {\n handleInputChange('system_prompt', value)\n }}\n autosize\n value={inputs.system_prompt}\n autoComplete='new-password'\n />\n {\n handleInputChange('model_mapping', JSON.stringify(MODEL_MAPPING_EXAMPLE, null, 2))\n }\n }>\n 填入模板\n \n \n 密钥： \n
\n {\n batch ?\n {\n handleInputChange('key', value)\n }}\n value={inputs.key}\n style={{ minHeight: 150, fontFamily: 'JetBrains Mono, Consolas' }}\n autoComplete='new-password'\n />\n :\n {\n handleInputChange('key', value)\n }}\n value={inputs.key}\n autoComplete='new-password'\n />\n }\n \n 组织： \n
\n {\n handleInputChange('openai_organization', value)\n }}\n value={inputs.openai_organization}\n />\n \n \n {\n setAutoBan(!autoBan);\n }\n }\n // onChange={handleInputChange}\n />\n 是否自动禁用（仅当自动禁用开启时有效），关闭后不会自动禁用该渠道：\n \n
\n\n {\n !isEdit && (\n \n \n setBatch(!batch)}\n />\n 批量创建 \n \n
\n )\n }\n {\n inputs.type !== 3 && inputs.type !== 8 && inputs.type !== 22 && (\n <>\n \n 代理： \n
\n {\n handleInputChange('base_url', value)\n }}\n value={inputs.base_url}\n autoComplete='new-password'\n />\n \n )\n }\n {\n inputs.type === 22 && (\n <>\n \n 私有部署地址： \n
\n {\n handleInputChange('base_url', value)\n }}\n value={inputs.base_url}\n autoComplete='new-password'\n />\n \n )\n }\n\n \n \n \n );\n};\n\nexport default EditChannel;\n"
},
{
"path": "web/air/src/pages/Channel/index.js",
"content": "import React from 'react';\nimport ChannelsTable from '../../components/ChannelsTable';\nimport {Layout} from \"@douyinfe/semi-ui\";\n\nconst File = () => (\n <>\n \n \n 管理渠道 \n \n \n \n \n \n);\n\nexport default File;\n"
},
{
"path": "web/air/src/pages/Chat/index.js",
"content": "import React from 'react';\n\nconst Chat = () => {\n const chatLink = localStorage.getItem('chat_link');\n\n return (\n \n \n 数据看板 \n \n \n \n \n \n \n \n \n \n \n );\n};\n\n\nexport default Detail;\n"
},
{
"path": "web/air/src/pages/Home/index.js",
"content": "import React, { useContext, useEffect, useState } from 'react';\nimport { Card, Col, Row } from '@douyinfe/semi-ui';\nimport { API, showError, showNotice, timestamp2string } from '../../helpers';\nimport { StatusContext } from '../../context/Status';\nimport { marked } from 'marked';\n\nconst Home = () => {\n const [statusState] = useContext(StatusContext);\n const [homePageContentLoaded, setHomePageContentLoaded] = useState(false);\n const [homePageContent, setHomePageContent] = useState('');\n\n const displayNotice = async () => {\n const res = await API.get('/api/notice');\n const { success, message, data } = res.data;\n if (success) {\n let oldNotice = localStorage.getItem('notice');\n if (data !== oldNotice && data !== '') {\n const htmlNotice = marked(data);\n showNotice(htmlNotice, true);\n localStorage.setItem('notice', data);\n }\n } else {\n showError(message);\n }\n };\n\n const displayHomePageContent = async () => {\n setHomePageContent(localStorage.getItem('home_page_content') || '');\n const res = await API.get('/api/home_page_content');\n const { success, message, data } = res.data;\n if (success) {\n let content = data;\n if (!data.startsWith('https://')) {\n content = marked.parse(data);\n }\n setHomePageContent(content);\n localStorage.setItem('home_page_content', content);\n } else {\n showError(message);\n setHomePageContent('加载首页内容失败...');\n }\n setHomePageContentLoaded(true);\n };\n\n const getStartTimeString = () => {\n const timestamp = statusState?.status?.start_time;\n return statusState.status ? timestamp2string(timestamp) : '';\n };\n\n useEffect(() => {\n displayNotice().then();\n displayHomePageContent().then();\n }, []);\n return (\n <>\n {\n homePageContentLoaded && homePageContent === '' ?\n <>\n \n \n 系统信息总览}>\n 名称：{statusState?.status?.system_name}
\n 版本：{statusState?.status?.version ? statusState?.status?.version : 'unknown'}
\n \n 源码：\n \n https://github.com/songquanpeng/one-api\n \n
\n 启动时间：{getStartTimeString()}
\n \n \n 系统配置总览}>\n \n 邮箱验证：\n {statusState?.status?.email_verification === true ? '已启用' : '未启用'}\n
\n \n GitHub 身份验证：\n {statusState?.status?.github_oauth === true ? '已启用' : '未启用'}\n
\n \n 微信身份验证：\n {statusState?.status?.wechat_login === true ? '已启用' : '未启用'}\n
\n \n Turnstile 用户校验：\n {statusState?.status?.turnstile_check === true ? '已启用' : '未启用'}\n
\n {/**/}\n {/* Telegram 身份验证：*/}\n {/* {statusState?.status?.telegram_oauth === true*/}\n {/* ? '已启用' : '未启用'}*/}\n {/*
*/}\n \n \n
\n \n\n \n : <>\n {\n homePageContent.startsWith('https://') ?\n :\n
\n }\n \n }\n\n \n );\n};\n\nexport default Home;"
},
{
"path": "web/air/src/pages/Log/index.js",
"content": "import React from 'react';\nimport LogsTable from '../../components/LogsTable';\n\nconst Token = () => (\n <>\n \n 页面不存在 \n 请检查你的浏览器地址是否正确
\n \n \n);\n\nexport default NotFound;\n"
},
{
"path": "web/air/src/pages/Redemption/EditRedemption.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { useNavigate, useParams } from 'react-router-dom';\nimport { API, downloadTextAsFile, isMobile, showError, showSuccess } from '../../helpers';\nimport { renderQuotaWithPrompt } from '../../helpers/render';\nimport { AutoComplete, Button, Input, Modal, SideSheet, Space, Spin, Typography } from '@douyinfe/semi-ui';\nimport Title from '@douyinfe/semi-ui/lib/es/typography/title';\nimport { Divider } from 'semantic-ui-react';\n\nconst EditRedemption = (props) => {\n const isEdit = props.editingRedemption.id !== undefined;\n const [loading, setLoading] = useState(isEdit);\n\n const params = useParams();\n const navigate = useNavigate();\n const originInputs = {\n name: '',\n quota: 100000,\n count: 1\n };\n const [inputs, setInputs] = useState(originInputs);\n const { name, quota, count } = inputs;\n\n const handleCancel = () => {\n props.handleClose();\n };\n\n const handleInputChange = (name, value) => {\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n };\n\n const loadRedemption = async () => {\n setLoading(true);\n let res = await API.get(`/api/redemption/${props.editingRedemption.id}`);\n const { success, message, data } = res.data;\n if (success) {\n setInputs(data);\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n useEffect(() => {\n if (isEdit) {\n loadRedemption().then(\n () => {\n // console.log(inputs);\n }\n );\n } else {\n setInputs(originInputs);\n }\n }, [props.editingRedemption.id]);\n\n const submit = async () => {\n if (!isEdit && inputs.name === '') return;\n setLoading(true);\n let localInputs = inputs;\n localInputs.count = parseInt(localInputs.count);\n localInputs.quota = parseInt(localInputs.quota);\n let res;\n if (isEdit) {\n res = await API.put(`/api/redemption/`, { ...localInputs, id: parseInt(props.editingRedemption.id) });\n } else {\n res = await API.post(`/api/redemption/`, {\n ...localInputs\n });\n }\n const { success, message, data } = res.data;\n if (success) {\n if (isEdit) {\n showSuccess('兑换码更新成功！');\n props.refresh();\n props.handleClose();\n } else {\n showSuccess('兑换码创建成功！');\n setInputs(originInputs);\n props.refresh();\n props.handleClose();\n }\n } else {\n showError(message);\n }\n if (!isEdit && data) {\n let text = '';\n for (let i = 0; i < data.length; i++) {\n text += data[i] + '\\n';\n }\n // downloadTextAsFile(text, `${inputs.name}.txt`);\n Modal.confirm({\n title: '兑换码创建成功',\n content: (\n \n
兑换码创建成功，是否下载兑换码？
\n
兑换码将以文本文件的形式下载，文件名为兑换码的名称。
\n
{isEdit ? '更新兑换码信息' : '创建新的兑换码'}}\n headerStyle={{ borderBottom: '1px solid var(--semi-color-border)' }}\n bodyStyle={{ borderBottom: '1px solid var(--semi-color-border)' }}\n visible={props.visiable}\n footer={\n \n \n 提交 \n 取消 \n \n
\n }\n closeIcon={null}\n onCancel={() => handleCancel()}\n width={isMobile() ? '100%' : 600}\n >\n \n handleInputChange('name', value)}\n value={name}\n autoComplete=\"new-password\"\n required={!isEdit}\n />\n \n {`额度${renderQuotaWithPrompt(quota)}`} \n
\n handleInputChange('quota', value)}\n value={quota}\n autoComplete=\"new-password\"\n type=\"number\"\n position={'bottom'}\n data={[\n { value: 500000, label: '1$' },\n { value: 5000000, label: '10$' },\n { value: 25000000, label: '50$' },\n { value: 50000000, label: '100$' },\n { value: 250000000, label: '500$' },\n { value: 500000000, label: '1000$' }\n ]}\n />\n {\n !isEdit && <>\n 生成数量 \n handleInputChange('count', value)}\n value={count}\n autoComplete=\"new-password\"\n type=\"number\"\n />\n \n }\n \n \n \n );\n};\n\nexport default EditRedemption;\n"
},
{
"path": "web/air/src/pages/Redemption/index.js",
"content": "import React from 'react';\nimport RedemptionsTable from '../../components/RedemptionsTable';\nimport {Layout} from \"@douyinfe/semi-ui\";\n\nconst Redemption = () => (\n <>\n \n \n 管理兑换码 \n \n \n \n \n \n);\n\nexport default Redemption;\n"
},
{
"path": "web/air/src/pages/Setting/index.js",
"content": "import React from 'react';\nimport SystemSetting from '../../components/SystemSetting';\nimport {isRoot} from '../../helpers';\nimport OtherSetting from '../../components/OtherSetting';\nimport PersonalSetting from '../../components/PersonalSetting';\nimport OperationSetting from '../../components/OperationSetting';\nimport {Layout, TabPane, Tabs} from \"@douyinfe/semi-ui\";\n\nconst Setting = () => {\n let panes = [\n {\n tab: '个人设置',\n content: \n \n \n \n {panes.map(pane => (\n \n {pane.content}\n \n ))}\n \n \n \n
\n );\n};\n\nexport default Setting;\n"
},
{
"path": "web/air/src/pages/Token/EditToken.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { useNavigate } from 'react-router-dom';\nimport { API, isMobile, showError, showSuccess, timestamp2string } from '../../helpers';\nimport { renderQuotaWithPrompt } from '../../helpers/render';\nimport {\n AutoComplete,\n Banner,\n Button,\n Checkbox,\n DatePicker,\n Input,\n Select,\n SideSheet,\n Space,\n Spin,\n Typography\n} from '@douyinfe/semi-ui';\nimport Title from '@douyinfe/semi-ui/lib/es/typography/title';\nimport { Divider } from 'semantic-ui-react';\n\nconst EditToken = (props) => {\n const [isEdit, setIsEdit] = useState(false);\n const [loading, setLoading] = useState(isEdit);\n const originInputs = {\n name: '',\n remain_quota: isEdit ? 0 : 500000,\n expired_time: -1,\n unlimited_quota: false,\n model_limits_enabled: false,\n model_limits: []\n };\n const [inputs, setInputs] = useState(originInputs);\n const { name, remain_quota, expired_time, unlimited_quota, model_limits_enabled, model_limits } = inputs;\n // const [visible, setVisible] = useState(false);\n const [models, setModels] = useState({});\n const navigate = useNavigate();\n const handleInputChange = (name, value) => {\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n };\n const handleCancel = () => {\n props.handleClose();\n };\n const setExpiredTime = (month, day, hour, minute) => {\n let now = new Date();\n let timestamp = now.getTime() / 1000;\n let seconds = month * 30 * 24 * 60 * 60;\n seconds += day * 24 * 60 * 60;\n seconds += hour * 60 * 60;\n seconds += minute * 60;\n if (seconds !== 0) {\n timestamp += seconds;\n setInputs({ ...inputs, expired_time: timestamp2string(timestamp) });\n } else {\n setInputs({ ...inputs, expired_time: -1 });\n }\n };\n\n const setUnlimitedQuota = () => {\n setInputs({ ...inputs, unlimited_quota: !unlimited_quota });\n };\n\n // const loadModels = async () => {\n // let res = await API.get(`/api/user/models`);\n // const { success, message, data } = res.data;\n // if (success) {\n // let localModelOptions = data.map((model) => ({\n // label: model,\n // value: model\n // }));\n // setModels(localModelOptions);\n // } else {\n // showError(message);\n // }\n // };\n\n const loadToken = async () => {\n setLoading(true);\n let res = await API.get(`/api/token/${props.editingToken.id}`);\n const { success, message, data } = res.data;\n if (success) {\n if (data.expired_time !== -1) {\n data.expired_time = timestamp2string(data.expired_time);\n }\n // if (data.model_limits !== '') {\n // data.model_limits = data.model_limits.split(',');\n // } else {\n // data.model_limits = [];\n // }\n setInputs(data);\n } else {\n showError(message);\n }\n setLoading(false);\n };\n useEffect(() => {\n setIsEdit(props.editingToken.id !== undefined);\n }, [props.editingToken.id]);\n\n useEffect(() => {\n if (!isEdit) {\n setInputs(originInputs);\n } else {\n loadToken().then(\n () => {\n // console.log(inputs);\n }\n );\n }\n // loadModels();\n }, [isEdit]);\n\n // 新增 state 变量 tokenCount 来记录用户想要创建的令牌数量，默认为 1\n const [tokenCount, setTokenCount] = useState(1);\n\n // 新增处理 tokenCount 变化的函数\n const handleTokenCountChange = (value) => {\n // 确保用户输入的是正整数\n const count = parseInt(value, 10);\n if (!isNaN(count) && count > 0) {\n setTokenCount(count);\n }\n };\n\n // 生成一个随机的四位字母数字字符串\n const generateRandomSuffix = () => {\n const characters = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789';\n let result = '';\n for (let i = 0; i < 6; i++) {\n result += characters.charAt(Math.floor(Math.random() * characters.length));\n }\n return result;\n };\n\n const submit = async () => {\n setLoading(true);\n if (isEdit) {\n // 编辑令牌的逻辑保持不变\n let localInputs = { ...inputs };\n localInputs.remain_quota = parseInt(localInputs.remain_quota);\n if (localInputs.expired_time !== -1) {\n let time = Date.parse(localInputs.expired_time);\n if (isNaN(time)) {\n showError('过期时间格式错误！');\n setLoading(false);\n return;\n }\n localInputs.expired_time = Math.ceil(time / 1000);\n }\n // localInputs.model_limits = localInputs.model_limits.join(',');\n let res = await API.put(`/api/token/`, { ...localInputs, id: parseInt(props.editingToken.id) });\n const { success, message } = res.data;\n if (success) {\n showSuccess('令牌更新成功！');\n props.refresh();\n props.handleClose();\n } else {\n showError(message);\n }\n } else {\n // 处理新增多个令牌的情况\n let successCount = 0; // 记录成功创建的令牌数量\n for (let i = 0; i < tokenCount; i++) {\n let localInputs = { ...inputs };\n if (i !== 0) {\n // 如果用户想要创建多个令牌，则给每个令牌一个序号后缀\n localInputs.name = `${inputs.name}-${generateRandomSuffix()}`;\n }\n localInputs.remain_quota = parseInt(localInputs.remain_quota);\n\n if (localInputs.expired_time !== -1) {\n let time = Date.parse(localInputs.expired_time);\n if (isNaN(time)) {\n showError('过期时间格式错误！');\n setLoading(false);\n break;\n }\n localInputs.expired_time = Math.ceil(time / 1000);\n }\n // localInputs.model_limits = localInputs.model_limits.join(',');\n let res = await API.post(`/api/token/`, localInputs);\n const { success, message } = res.data;\n\n if (success) {\n successCount++;\n } else {\n showError(message);\n break; // 如果创建失败，终止循环\n }\n }\n\n if (successCount > 0) {\n showSuccess(`${successCount}个令牌创建成功，请在列表页面点击复制获取令牌！`);\n props.refresh();\n props.handleClose();\n }\n }\n setLoading(false);\n setInputs(originInputs); // 重置表单\n setTokenCount(1); // 重置数量为默认值\n };\n\n\n return (\n <>\n {isEdit ? '更新令牌信息' : '创建新的令牌'}}\n headerStyle={{ borderBottom: '1px solid var(--semi-color-border)' }}\n bodyStyle={{ borderBottom: '1px solid var(--semi-color-border)' }}\n visible={props.visiable}\n footer={\n \n \n 提交 \n 取消 \n \n
\n }\n closeIcon={null}\n onCancel={() => handleCancel()}\n width={isMobile() ? '100%' : 600}\n >\n \n handleInputChange('name', value)}\n value={name}\n autoComplete=\"new-password\"\n required={!isEdit}\n />\n handleInputChange('expired_time', value)}\n value={expired_time}\n autoComplete=\"new-password\"\n type=\"dateTime\"\n />\n \n \n {\n setExpiredTime(0, 0, 0, 0);\n }}>永不过期 \n {\n setExpiredTime(0, 0, 1, 0);\n }}>一小时 \n {\n setExpiredTime(1, 0, 0, 0);\n }}>一个月 \n {\n setExpiredTime(0, 1, 0, 0);\n }}>一天 \n \n
\n\n \n {`额度${renderQuotaWithPrompt(remain_quota)}`} \n
\n handleInputChange('remain_quota', value)}\n value={remain_quota}\n autoComplete=\"new-password\"\n type=\"number\"\n // position={'top'}\n data={[\n { value: 500000, label: '1$' },\n { value: 5000000, label: '10$' },\n { value: 25000000, label: '50$' },\n { value: 50000000, label: '100$' },\n { value: 250000000, label: '500$' },\n { value: 500000000, label: '1000$' }\n ]}\n disabled={unlimited_quota}\n />\n\n {!isEdit && (\n <>\n \n 新建数量 \n
\n handleTokenCountChange(value)}\n onSelect={(value) => handleTokenCountChange(value)}\n value={tokenCount.toString()}\n autoComplete=\"off\"\n type=\"number\"\n data={[\n { value: 10, label: '10个' },\n { value: 20, label: '20个' },\n { value: 30, label: '30个' },\n { value: 100, label: '100个' }\n ]}\n disabled={unlimited_quota}\n />\n \n )}\n\n \n {\n setUnlimitedQuota();\n }}>{unlimited_quota ? '取消无限额度' : '设为无限额度'} \n
\n {/* \n \n handleInputChange('model_limits_enabled', e.target.checked)}\n >\n \n 启用模型限制（非必要，不建议启用） \n \n
\n\n {\n handleInputChange('model_limits', value);\n }}\n value={inputs.model_limits}\n autoComplete=\"new-password\"\n optionList={models}\n disabled={!model_limits_enabled}\n /> */}\n \n \n \n );\n};\n\nexport default EditToken;\n"
},
{
"path": "web/air/src/pages/Token/index.js",
"content": "import React from 'react';\nimport TokensTable from '../../components/TokensTable';\nimport {Layout} from \"@douyinfe/semi-ui\";\nconst Token = () => (\n <>\n \n \n 我的令牌 \n \n \n \n \n \n);\n\nexport default Token;\n"
},
{
"path": "web/air/src/pages/TopUp/index.js",
"content": "import React, {useEffect, useState} from 'react';\nimport {API, isMobile, showError, showInfo, showSuccess} from '../../helpers';\nimport {renderNumber, renderQuota} from '../../helpers/render';\nimport {Col, Layout, Row, Typography, Card, Button, Form, Divider, Space, Modal} from \"@douyinfe/semi-ui\";\nimport Title from \"@douyinfe/semi-ui/lib/es/typography/title\";\nimport Text from '@douyinfe/semi-ui/lib/es/typography/text';\nimport { Link } from 'react-router-dom';\n\nconst TopUp = () => {\n const [redemptionCode, setRedemptionCode] = useState('');\n const [topUpCode, setTopUpCode] = useState('');\n const [topUpCount, setTopUpCount] = useState(10);\n const [minTopupCount, setMinTopUpCount] = useState(1);\n const [amount, setAmount] = useState(0.0);\n const [minTopUp, setMinTopUp] = useState(1);\n const [topUpLink, setTopUpLink] = useState('');\n const [enableOnlineTopUp, setEnableOnlineTopUp] = useState(false);\n const [userQuota, setUserQuota] = useState(0);\n const [isSubmitting, setIsSubmitting] = useState(false);\n const [open, setOpen] = useState(false);\n const [payWay, setPayWay] = useState('');\n\n const topUp = async () => {\n if (redemptionCode === '') {\n showInfo('请输入兑换码！')\n return;\n }\n setIsSubmitting(true);\n try {\n const res = await API.post('/api/user/topup', {\n key: redemptionCode\n });\n const {success, message, data} = res.data;\n if (success) {\n showSuccess('兑换成功！');\n Modal.success({title: '兑换成功！', content: '成功兑换额度：' + renderQuota(data), centered: true});\n setUserQuota((quota) => {\n return quota + data;\n });\n setRedemptionCode('');\n } else {\n showError(message);\n }\n } catch (err) {\n showError('请求失败');\n } finally {\n setIsSubmitting(false);\n }\n };\n\n const openTopUpLink = () => {\n if (!topUpLink) {\n showError('超级管理员未设置充值链接！');\n return;\n }\n window.open(topUpLink, '_blank');\n };\n\n const preTopUp = async (payment) => {\n if (!enableOnlineTopUp) {\n showError('管理员未开启在线充值！');\n return;\n }\n if (amount === 0) {\n await getAmount();\n }\n if (topUpCount < minTopUp) {\n showInfo('充值数量不能小于' + minTopUp);\n return;\n }\n setPayWay(payment)\n setOpen(true);\n }\n\n const onlineTopUp = async () => {\n if (amount === 0) {\n await getAmount();\n }\n if (topUpCount < minTopUp) {\n showInfo('充值数量不能小于' + minTopUp);\n return;\n }\n setOpen(false);\n try {\n const res = await API.post('/api/user/pay', {\n amount: parseInt(topUpCount),\n top_up_code: topUpCode,\n payment_method: payWay\n });\n if (res !== undefined) {\n const {message, data} = res.data;\n // showInfo(message);\n if (message === 'success') {\n\n let params = data\n let url = res.data.url\n let form = document.createElement('form')\n form.action = url\n form.method = 'POST'\n // 判断是否为safari浏览器\n let isSafari = navigator.userAgent.indexOf(\"Safari\") > -1 && navigator.userAgent.indexOf(\"Chrome\") < 1;\n if (!isSafari) {\n form.target = '_blank'\n }\n for (let key in params) {\n let input = document.createElement('input')\n input.type = 'hidden'\n input.name = key\n input.value = params[key]\n form.appendChild(input)\n }\n document.body.appendChild(form)\n form.submit()\n document.body.removeChild(form)\n } else {\n showError(data);\n // setTopUpCount(parseInt(res.data.count));\n // setAmount(parseInt(data));\n }\n } else {\n showError(res);\n }\n } catch (err) {\n console.log(err);\n } finally {\n }\n }\n\n const getUserQuota = async () => {\n let res = await API.get(`/api/user/self`);\n const {success, message, data} = res.data;\n if (success) {\n setUserQuota(data.quota);\n } else {\n showError(message);\n }\n }\n\n useEffect(() => {\n let status = localStorage.getItem('status');\n if (status) {\n status = JSON.parse(status);\n if (status.top_up_link) {\n setTopUpLink(status.top_up_link);\n }\n if (status.min_topup) {\n setMinTopUp(status.min_topup);\n }\n if (status.enable_online_topup) {\n setEnableOnlineTopUp(status.enable_online_topup);\n }\n }\n getUserQuota().then();\n }, []);\n\n const renderAmount = () => {\n // console.log(amount);\n return amount + '元';\n }\n\n const getAmount = async (value) => {\n if (value === undefined) {\n value = topUpCount;\n }\n try {\n const res = await API.post('/api/user/amount', {\n amount: parseFloat(value),\n top_up_code: topUpCode\n });\n if (res !== undefined) {\n const {message, data} = res.data;\n // showInfo(message);\n if (message === 'success') {\n setAmount(parseFloat(data));\n } else {\n showError(data);\n // setTopUpCount(parseInt(res.data.count));\n // setAmount(parseInt(data));\n }\n } else {\n showError(res);\n }\n } catch (err) {\n console.log(err);\n } finally {\n }\n }\n\n const handleCancel = () => {\n setOpen(false);\n }\n\n return (\n \n
\n \n 充值额度 \n \n \n \n 充值数量：{topUpCount}$
\n 实付金额：{renderAmount()}
\n 是否确认充值？
\n \n \n
\n 余额 {renderQuota(userQuota)} \n \n {/* */}\n {/**/}\n {/* */}\n {/* */}\n {/*
*/}\n \n \n \n
{'添加用户'}}\n headerStyle={{ borderBottom: '1px solid var(--semi-color-border)' }}\n bodyStyle={{ borderBottom: '1px solid var(--semi-color-border)' }}\n visible={props.visible}\n footer={\n \n \n 提交 \n 取消 \n \n
\n }\n closeIcon={null}\n onCancel={() => handleCancel()}\n width={isMobile() ? '100%' : 600}\n >\n \n handleInputChange('username', value)}\n value={username}\n autoComplete=\"off\"\n />\n handleInputChange('display_name', value)}\n value={display_name}\n />\n handleInputChange('password', value)}\n value={password}\n autoComplete=\"off\"\n />\n \n \n \n );\n};\n\nexport default AddUser;\n"
},
{
"path": "web/air/src/pages/User/EditUser.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { useNavigate } from 'react-router-dom';\nimport { API, isMobile, showError, showSuccess } from '../../helpers';\nimport { renderQuotaWithPrompt } from '../../helpers/render';\nimport Title from '@douyinfe/semi-ui/lib/es/typography/title';\nimport { Button, Divider, Input, Select, SideSheet, Space, Spin, Typography } from '@douyinfe/semi-ui';\n\nconst EditUser = (props) => {\n const userId = props.editingUser.id;\n const [loading, setLoading] = useState(true);\n const [inputs, setInputs] = useState({\n username: '',\n display_name: '',\n password: '',\n github_id: '',\n wechat_id: '',\n email: '',\n quota: 0,\n group: 'default'\n });\n const [groupOptions, setGroupOptions] = useState([]);\n const { username, display_name, password, github_id, wechat_id, telegram_id, email, quota, group } =\n inputs;\n const handleInputChange = (name, value) => {\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n };\n const fetchGroups = async () => {\n try {\n let res = await API.get(`/api/group/`);\n setGroupOptions(res.data.data.map((group) => ({\n label: group,\n value: group\n })));\n } catch (error) {\n showError(error.message);\n }\n };\n const navigate = useNavigate();\n const handleCancel = () => {\n props.handleClose();\n };\n const loadUser = async () => {\n setLoading(true);\n let res = undefined;\n if (userId) {\n res = await API.get(`/api/user/${userId}`);\n } else {\n res = await API.get(`/api/user/self`);\n }\n const { success, message, data } = res.data;\n if (success) {\n data.password = '';\n setInputs(data);\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n useEffect(() => {\n loadUser().then();\n if (userId) {\n fetchGroups().then();\n }\n }, [props.editingUser.id]);\n\n const submit = async () => {\n setLoading(true);\n let res = undefined;\n if (userId) {\n let data = { ...inputs, id: parseInt(userId) };\n if (typeof data.quota === 'string') {\n data.quota = parseInt(data.quota);\n }\n res = await API.put(`/api/user/`, data);\n } else {\n res = await API.put(`/api/user/self`, inputs);\n }\n const { success, message } = res.data;\n if (success) {\n showSuccess('用户信息更新成功！');\n props.refresh();\n props.handleClose();\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n return (\n <>\n {'编辑用户'}}\n headerStyle={{ borderBottom: '1px solid var(--semi-color-border)' }}\n bodyStyle={{ borderBottom: '1px solid var(--semi-color-border)' }}\n visible={props.visible}\n footer={\n \n \n 提交 \n 取消 \n \n
\n }\n closeIcon={null}\n onCancel={() => handleCancel()}\n width={isMobile() ? '100%' : 600}\n >\n \n \n 用户名 \n
\n handleInputChange('username', value)}\n value={username}\n autoComplete=\"new-password\"\n />\n \n 密码 \n
\n handleInputChange('password', value)}\n value={password}\n autoComplete=\"new-password\"\n />\n \n 显示名称 \n
\n handleInputChange('display_name', value)}\n value={display_name}\n autoComplete=\"new-password\"\n />\n {\n userId && <>\n \n 分组 \n
\n handleInputChange('group', value)}\n value={inputs.group}\n autoComplete=\"new-password\"\n optionList={groupOptions}\n />\n \n {`剩余额度${renderQuotaWithPrompt(quota)}`} \n
\n handleInputChange('quota', value)}\n value={quota}\n type={'number'}\n autoComplete=\"new-password\"\n />\n \n }\n 以下信息不可修改 \n \n 已绑定的 GitHub 账户 \n
\n \n 已绑定的微信账户 \n
\n \n 已绑定的邮箱账户 \n
\n \n \n \n );\n};\n\nexport default EditUser;\n"
},
{
"path": "web/air/src/pages/User/index.js",
"content": "import React from 'react';\nimport UsersTable from '../../components/UsersTable';\nimport {Layout} from \"@douyinfe/semi-ui\";\n\nconst User = () => (\n <>\n \n \n 管理用户 \n \n \n \n \n \n);\n\nexport default User;\n"
},
{
"path": "web/air/vercel.json",
"content": "{\n \"github\": {\n \"silent\": true\n }\n}\n"
},
{
"path": "web/berry/.gitignore",
"content": "# See https://help.github.com/articles/ignoring-files/ for more about ignoring files.\n\n# dependencies\n/node_modules\n/.pnp\n.pnp.js\n\n# testing\n/coverage\n\n# production\n/build\n\n# misc\n.DS_Store\n.env.local\n.env.development.local\n.env.test.local\n.env.production.local\n\nnpm-debug.log*\nyarn-debug.log*\nyarn-error.log*\n.idea\npackage-lock.json\nyarn.lock"
},
{
"path": "web/berry/.prettierrc",
"content": "{\n \"bracketSpacing\": true,\n \"printWidth\": 140,\n \"singleQuote\": true,\n \"trailingComma\": \"none\",\n \"tabWidth\": 2,\n \"useTabs\": false\n}\n"
},
{
"path": "web/berry/README.md",
"content": "# One API 前端界面\n\n这个项目是 One API 的前端界面，它基于 [Berry Free React Admin Template](https://github.com/codedthemes/berry-free-react-admin-template) 进行开发。\n\n## 使用的开源项目\n\n使用了以下开源项目作为我们项目的一部分：\n\n- [Berry Free React Admin Template](https://github.com/codedthemes/berry-free-react-admin-template)\n- [minimal-ui-kit](minimal-ui-kit)\n\n## 开发说明\n\n当添加新的渠道时，需要修改以下地方：\n\n1. `web/berry/src/constants/ChannelConstants.js`\n\n在该文件中的 `CHANNEL_OPTIONS` 添加新的渠道\n\n```js\nexport const CHANNEL_OPTIONS = {\n //key 为渠道ID\n 1: {\n key: 1, // 渠道ID\n text: \"OpenAI\", // 渠道名称\n value: 1, // 渠道ID\n color: \"primary\", // 渠道列表显示的颜色\n },\n};\n```\n\n2. `web/berry/src/views/Channel/type/Config.js`\n\n在该文件中的`typeConfig`添加新的渠道配置， 如果无需配置，可以不添加\n\n```js\nconst typeConfig = {\n // key 为渠道ID\n 3: {\n inputLabel: {\n // 输入框名称 配置\n // 对应的字段名称\n base_url: \"AZURE_OPENAI_ENDPOINT\",\n other: \"默认 API 版本\",\n },\n prompt: {\n // 输入框提示 配置\n // 对应的字段名称\n base_url: \"请填写AZURE_OPENAI_ENDPOINT\",\n\n // 注意：通过判断 `other` 是否有值来判断是否需要显示 `other` 输入框， 默认是没有值的\n other: \"请输入默认API版本，例如：2024-03-01-preview\",\n },\n modelGroup: \"openai\", // 模型组名称,这个值是给 填入渠道支持模型 按钮使用的。 填入渠道支持模型 按钮会根据这个值来获取模型组，如果填写默认是 openai\n },\n};\n```\n\n## 许可证\n\n本项目中使用的代码遵循 MIT 许可证。\n"
},
{
"path": "web/berry/jsconfig.json",
"content": "{\n \"compilerOptions\": {\n \"target\": \"esnext\",\n \"module\": \"commonjs\",\n \"baseUrl\": \"src\"\n },\n \"include\": [\"src/**/*\"],\n \"exclude\": [\"node_modules\"]\n}\n"
},
{
"path": "web/berry/package.json",
"content": "{\n \"name\": \"one_api_web\",\n \"version\": \"1.0.0\",\n \"proxy\": \"http://127.0.0.1:3000\",\n \"private\": true,\n \"homepage\": \"\",\n \"dependencies\": {\n \"@emotion/cache\": \"^11.9.3\",\n \"@emotion/react\": \"^11.9.3\",\n \"@emotion/styled\": \"^11.9.3\",\n \"@mui/icons-material\": \"^5.8.4\",\n \"@mui/lab\": \"^5.0.0-alpha.88\",\n \"@mui/material\": \"^5.8.6\",\n \"@mui/system\": \"^5.8.6\",\n \"@mui/utils\": \"^5.8.6\",\n \"@mui/x-date-pickers\": \"^6.18.5\",\n \"@tabler/icons-react\": \"^2.44.0\",\n \"apexcharts\": \"3.35.3\",\n \"axios\": \"^0.27.2\",\n \"dayjs\": \"^1.11.10\",\n \"formik\": \"^2.2.9\",\n \"framer-motion\": \"^6.3.16\",\n \"history\": \"^5.3.0\",\n \"marked\": \"^4.1.1\",\n \"material-ui-popup-state\": \"^4.0.1\",\n \"notistack\": \"^3.0.1\",\n \"prop-types\": \"^15.8.1\",\n \"react\": \"^18.2.0\",\n \"react-apexcharts\": \"1.4.0\",\n \"react-device-detect\": \"^2.2.2\",\n \"react-dom\": \"^18.2.0\",\n \"react-perfect-scrollbar\": \"^1.5.8\",\n \"react-redux\": \"^8.0.2\",\n \"react-router\": \"6.3.0\",\n \"react-router-dom\": \"6.3.0\",\n \"react-scripts\": \"^5.0.1\",\n \"react-turnstile\": \"^1.1.2\",\n \"redux\": \"^4.2.0\",\n \"yup\": \"^0.32.11\"\n },\n \"scripts\": {\n \"start\": \"react-scripts start\",\n \"build\": \"react-scripts build && mv -f build ../build/berry\",\n \"test\": \"react-scripts test\",\n \"eject\": \"react-scripts eject\"\n },\n \"eslintConfig\": {\n \"extends\": [\n \"react-app\"\n ]\n },\n \"babel\": {\n \"presets\": [\n \"@babel/preset-react\"\n ]\n },\n \"browserslist\": {\n \"production\": [\n \"defaults\",\n \"not IE 11\"\n ],\n \"development\": [\n \"last 1 chrome version\",\n \"last 1 firefox version\",\n \"last 1 safari version\"\n ]\n },\n \"devDependencies\": {\n \"@babel/core\": \"^7.21.4\",\n \"@babel/eslint-parser\": \"^7.21.3\",\n \"eslint\": \"^8.38.0\",\n \"eslint-config-prettier\": \"^8.8.0\",\n \"eslint-config-react-app\": \"^7.0.1\",\n \"eslint-plugin-flowtype\": \"^8.0.3\",\n \"eslint-plugin-import\": \"^2.27.5\",\n \"eslint-plugin-jsx-a11y\": \"^6.7.1\",\n \"eslint-plugin-prettier\": \"^4.2.1\",\n \"eslint-plugin-react\": \"^7.32.2\",\n \"eslint-plugin-react-hooks\": \"^4.6.0\",\n \"immutable\": \"^4.3.0\",\n \"prettier\": \"^2.8.7\",\n \"sass\": \"^1.53.0\"\n }\n}\n"
},
{
"path": "web/berry/public/index.html",
"content": "\n\n \n One API \n You need to enable JavaScript to run this app. \n
\n\n \n\n"
},
{
"path": "web/berry/src/App.js",
"content": "import { useEffect } from 'react';\nimport { useSelector, useDispatch } from 'react-redux';\n\nimport { ThemeProvider } from '@mui/material/styles';\nimport { CssBaseline, StyledEngineProvider } from '@mui/material';\nimport { SET_THEME } from 'store/actions';\n// routing\nimport Routes from 'routes';\n\n// defaultTheme\nimport themes from 'themes';\n\n// project imports\nimport NavigationScroll from 'layout/NavigationScroll';\n\n// auth\nimport UserProvider from 'contexts/UserContext';\nimport StatusProvider from 'contexts/StatusContext';\nimport { SnackbarProvider } from 'notistack';\n\n// ==============================|| APP ||============================== //\n\nconst App = () => {\n const dispatch = useDispatch();\n const customization = useSelector((state) => state.customization);\n\n useEffect(() => {\n const storedTheme = localStorage.getItem('theme');\n if (storedTheme) {\n dispatch({ type: SET_THEME, theme: storedTheme });\n }\n }, [dispatch]);\n\n return (\n \n \n \n \n \n \n \n \n \n \n \n \n );\n};\n\nexport default App;\n"
},
{
"path": "web/berry/src/assets/scss/_themes-vars.module.scss",
"content": "// paper & background\n$paper: #ffffff;\n\n// primary\n$primaryLight: #eef2f6;\n$primaryMain: #2196f3;\n$primaryDark: #1e88e5;\n$primary200: #90caf9;\n$primary800: #1565c0;\n\n// secondary\n$secondaryLight: #ede7f6;\n$secondaryMain: #673ab7;\n$secondaryDark: #5e35b1;\n$secondary200: #b39ddb;\n$secondary800: #4527a0;\n\n// success Colors\n$successLight: #b9f6ca;\n$success200: #69f0ae;\n$successMain: #00e676;\n$successDark: #00c853;\n\n// error\n$errorLight: #ef9a9a;\n$errorMain: #f44336;\n$errorDark: #c62828;\n\n// orange\n$orangeLight: #fbe9e7;\n$orangeMain: #ffab91;\n$orangeDark: #d84315;\n\n// warning\n$warningLight: #fff8e1;\n$warningMain: #ffe57f;\n$warningDark: #ffc107;\n\n// grey\n$grey50: #f8fafc;\n$grey100: #eef2f6;\n$grey200: #e3e8ef;\n$grey300: #cdd5df;\n$grey500: #697586;\n$grey600: #4b5565;\n$grey700: #364152;\n$grey900: #121926;\n\n$tableBackground: #f4f6f8;\n$tableBorderBottom: #f1f3f4;\n\n// ==============================|| DARK THEME VARIANTS ||============================== //\n\n// paper & background\n$darkBackground: #1a223f; // level 3\n$darkPaper: #111936; // level 4\n$darkDivider: rgba(227, 232, 239, 0.2);\n$darkSelectedBack : rgba(124, 77, 255, 0.15);\n\n// dark 800 & 900\n$darkLevel1: #29314f; // level 1\n$darkLevel2: #212946; // level 2\n\n// primary dark\n$darkPrimaryLight: #eef2f6;\n$darkPrimaryMain: #2196f3;\n$darkPrimaryDark: #1e88e5;\n$darkPrimary200: #90caf9;\n$darkPrimary800: #1565c0;\n\n// secondary dark\n$darkSecondaryLight: #d1c4e9;\n$darkSecondaryMain: #7c4dff;\n$darkSecondaryDark: #651fff;\n$darkSecondary200: #b39ddb;\n$darkSecondary800: #6200ea;\n\n// text variants\n$darkTextTitle: #d7dcec;\n$darkTextPrimary: #bdc8f0;\n$darkTextSecondary: #8492c4;\n\n// ==============================|| JAVASCRIPT ||============================== //\n\n:export {\n // paper & background\n paper: $paper;\n\n // primary\n primaryLight: $primaryLight;\n primary200: $primary200;\n primaryMain: $primaryMain;\n primaryDark: $primaryDark;\n primary800: $primary800;\n\n // secondary\n secondaryLight: $secondaryLight;\n secondary200: $secondary200;\n secondaryMain: $secondaryMain;\n secondaryDark: $secondaryDark;\n secondary800: $secondary800;\n\n // success\n successLight: $successLight;\n success200: $success200;\n successMain: $successMain;\n successDark: $successDark;\n\n // error\n errorLight: $errorLight;\n errorMain: $errorMain;\n errorDark: $errorDark;\n\n // orange\n orangeLight: $orangeLight;\n orangeMain: $orangeMain;\n orangeDark: $orangeDark;\n\n // warning\n warningLight: $warningLight;\n warningMain: $warningMain;\n warningDark: $warningDark;\n\n // grey\n grey50: $grey50;\n grey100: $grey100;\n grey200: $grey200;\n grey300: $grey300;\n grey500: $grey500;\n grey600: $grey600;\n grey700: $grey700;\n grey900: $grey900;\n\n // ==============================|| DARK THEME VARIANTS ||============================== //\n\n // paper & background\n darkPaper: $darkPaper;\n darkBackground: $darkBackground;\n\n // dark 800 & 900\n darkLevel1: $darkLevel1;\n darkLevel2: $darkLevel2;\n\n // text variants\n darkTextTitle: $darkTextTitle;\n darkTextPrimary: $darkTextPrimary;\n darkTextSecondary: $darkTextSecondary;\n\n // primary dark\n darkPrimaryLight: $darkPrimaryLight;\n darkPrimaryMain: $darkPrimaryMain;\n darkPrimaryDark: $darkPrimaryDark;\n darkPrimary200: $darkPrimary200;\n darkPrimary800: $darkPrimary800;\n\n // secondary dark\n darkSecondaryLight: $darkSecondaryLight;\n darkSecondaryMain: $darkSecondaryMain;\n darkSecondaryDark: $darkSecondaryDark;\n darkSecondary200: $darkSecondary200;\n darkSecondary800: $darkSecondary800;\n\n darkDivider: $darkDivider;\n darkSelectedBack: $darkSelectedBack;\n tableBackground: $tableBackground;\n tableBorderBottom: $tableBorderBottom;\n}\n"
},
{
"path": "web/berry/src/assets/scss/fonts.scss",
"content": "\n/* roboto-regular */\n@font-face {\n font-family: 'Roboto';\n font-style: normal;\n font-weight: 400;\n font-display: swap;\n src: local('Roboto'), url('../fonts/roboto-regular.woff2') format('woff2');\n unicode-range: U+0000-00FF, U+0131, U+0152-0153, U+02BB-02BC, U+02C6, U+02DA, U+02DC, U+0304, U+0308, U+0329, U+2000-206F, U+2074, U+20AC, U+2122, U+2191, U+2193, U+2212, U+2215, U+FEFF, U+FFFD;\n }\n\n /* roboto-500 */\n@font-face {\n font-family: 'Roboto';\n font-style: normal;\n font-weight: 500;\n font-display: swap;\n src: local('Roboto'), url('../fonts/roboto-500.woff2') format('woff2');\n unicode-range: U+0000-00FF, U+0131, U+0152-0153, U+02BB-02BC, U+02C6, U+02DA, U+02DC, U+0304, U+0308, U+0329, U+2000-206F, U+2074, U+20AC, U+2122, U+2191, U+2193, U+2212, U+2215, U+FEFF, U+FFFD;\n}\n \n \n/* roboto-700 */\n@font-face {\n font-family: 'Roboto';\n font-style: normal;\n font-weight: 700;\n font-display: swap;\n src: local('Roboto'), url('../fonts/roboto-700.woff2') format('woff2');\n unicode-range: U+0000-00FF, U+0131, U+0152-0153, U+02BB-02BC, U+02C6, U+02DA, U+02DC, U+0304, U+0308, U+0329, U+2000-206F, U+2074, U+20AC, U+2122, U+2191, U+2193, U+2212, U+2215, U+FEFF, U+FFFD;\n}\n "
},
{
"path": "web/berry/src/assets/scss/style.scss",
"content": "@import 'fonts.scss';\n// color variants\n@import 'themes-vars.module.scss';\n\n// third-party\n@import '~react-perfect-scrollbar/dist/css/styles.css';\n\n// ==============================|| LIGHT BOX ||============================== //\n.fullscreen .react-images__blanket {\n z-index: 1200;\n}\n\n// ==============================|| APEXCHART ||============================== //\n\n.apexcharts-legend-series .apexcharts-legend-marker {\n margin-right: 8px;\n}\n\n// ==============================|| PERFECT SCROLLBAR ||============================== //\n\n.scrollbar-container {\n .ps__rail-y {\n &:hover > .ps__thumb-y,\n &:focus > .ps__thumb-y,\n &.ps--clicking .ps__thumb-y {\n background-color: $grey500;\n width: 5px;\n }\n }\n .ps__thumb-y {\n background-color: $grey500;\n border-radius: 6px;\n width: 5px;\n right: 0;\n }\n}\n\n.scrollbar-container.ps,\n.scrollbar-container > .ps {\n &.ps--active-y > .ps__rail-y {\n width: 5px;\n background-color: transparent !important;\n z-index: 999;\n &:hover,\n &.ps--clicking {\n width: 5px;\n background-color: transparent;\n }\n }\n &.ps--scrolling-y > .ps__rail-y,\n &.ps--scrolling-x > .ps__rail-x {\n opacity: 0.4;\n background-color: transparent;\n }\n}\n\n// ==============================|| ANIMATION KEYFRAMES ||============================== //\n\n@keyframes wings {\n 50% {\n transform: translateY(-40px);\n }\n 100% {\n transform: translateY(0px);\n }\n}\n\n@keyframes blink {\n 50% {\n opacity: 0;\n }\n 100% {\n opacity: 1;\n }\n}\n\n@keyframes bounce {\n 0%,\n 20%,\n 53%,\n to {\n animation-timing-function: cubic-bezier(0.215, 0.61, 0.355, 1);\n transform: translateZ(0);\n }\n 40%,\n 43% {\n animation-timing-function: cubic-bezier(0.755, 0.05, 0.855, 0.06);\n transform: translate3d(0, -5px, 0);\n }\n 70% {\n animation-timing-function: cubic-bezier(0.755, 0.05, 0.855, 0.06);\n transform: translate3d(0, -7px, 0);\n }\n 80% {\n transition-timing-function: cubic-bezier(0.215, 0.61, 0.355, 1);\n transform: translateZ(0);\n }\n 90% {\n transform: translate3d(0, -2px, 0);\n }\n}\n\n@keyframes slideY {\n 0%,\n 50%,\n 100% {\n transform: translateY(0px);\n }\n 25% {\n transform: translateY(-10px);\n }\n 75% {\n transform: translateY(10px);\n }\n}\n\n@keyframes slideX {\n 0%,\n 50%,\n 100% {\n transform: translateX(0px);\n }\n 25% {\n transform: translateX(-10px);\n }\n 75% {\n transform: translateX(10px);\n }\n}\n"
},
{
"path": "web/berry/src/config.js",
"content": "const config = {\n // basename: only at build time to set, and Don't add '/' at end off BASENAME for breadcrumbs, also Don't put only '/' use blank('') instead,\n // like '/berry-material-react/react/default'\n basename: '/',\n defaultPath: '/panel/dashboard',\n fontFamily: `'Roboto', sans-serif, Helvetica, Arial, sans-serif`,\n borderRadius: 12,\n siteInfo: {\n chat_link: '',\n display_in_currency: true,\n email_verification: false,\n footer_html: '',\n github_client_id: '',\n github_oauth: false,\n logo: '',\n quota_per_unit: 500000,\n server_address: '',\n start_time: 0,\n system_name: 'One API',\n top_up_link: '',\n turnstile_check: false,\n turnstile_site_key: '',\n version: '',\n wechat_login: false,\n wechat_qrcode: '',\n oidc: false,\n oidc_client_id: '',\n oidc_authorization_endpoint: '',\n oidc_token_endpoint: '',\n oidc_userinfo_endpoint: '',\n }\n};\n\nexport default config;\n"
},
{
"path": "web/berry/src/constants/ChannelConstants.js",
"content": "export const CHANNEL_OPTIONS = {\n 1: {\n key: 1,\n text: 'OpenAI',\n value: 1,\n color: 'success'\n },\n 14: {\n key: 14,\n text: 'Anthropic Claude',\n value: 14,\n color: 'primary'\n },\n 33: {\n key: 33,\n text: 'AWS',\n value: 33,\n color: 'primary'\n },\n 37: {\n key: 37,\n text: 'Cloudflare',\n value: 37,\n color: 'success'\n },\n 3: {\n key: 3,\n text: 'Azure OpenAI',\n value: 3,\n color: 'success'\n },\n 11: {\n key: 11,\n text: 'Google PaLM2',\n value: 11,\n color: 'warning'\n },\n 24: {\n key: 24,\n text: 'Google Gemini',\n value: 24,\n color: 'warning'\n },\n 28: {\n key: 28,\n text: 'Mistral AI',\n value: 28,\n color: 'warning'\n },\n 40: {\n key: 40,\n text: '字节火山引擎',\n value: 40,\n color: 'primary'\n },\n 15: {\n key: 15,\n text: '百度文心千帆',\n value: 15,\n color: 'primary'\n },\n 17: {\n key: 17,\n text: '阿里通义千问',\n value: 17,\n color: 'primary'\n },\n 18: {\n key: 18,\n text: '讯飞星火认知',\n value: 18,\n color: 'primary'\n },\n 16: {\n key: 16,\n text: '智谱 ChatGLM',\n value: 16,\n color: 'primary'\n },\n 19: {\n key: 19,\n text: '360 智脑',\n value: 19,\n color: 'primary'\n },\n 25: {\n key: 25,\n text: 'Moonshot AI',\n value: 25,\n color: 'primary'\n },\n 23: {\n key: 23,\n text: '腾讯混元',\n value: 23,\n color: 'primary'\n },\n 26: {\n key: 26,\n text: '百川大模型',\n value: 26,\n color: 'primary'\n },\n 27: {\n key: 27,\n text: 'MiniMax',\n value: 27,\n color: 'primary'\n },\n 29: {\n key: 29,\n text: 'Groq',\n value: 29,\n color: 'primary'\n },\n 30: {\n key: 30,\n text: 'Ollama',\n value: 30,\n color: 'primary'\n },\n 31: {\n key: 31,\n text: '零一万物',\n value: 31,\n color: 'primary'\n },\n 32: {\n key: 32,\n text: '阶跃星辰',\n value: 32,\n color: 'primary'\n },\n 34: {\n key: 34,\n text: 'Coze',\n value: 34,\n color: 'primary'\n },\n 35: {\n key: 35,\n text: 'Cohere',\n value: 35,\n color: 'primary'\n },\n 36: {\n key: 36,\n text: 'DeepSeek',\n value: 36,\n color: 'primary'\n },\n 38: {\n key: 38,\n text: 'DeepL',\n value: 38,\n color: 'primary'\n },\n 39: {\n key: 39,\n text: 'together.ai',\n value: 39,\n color: 'primary'\n },\n 42: {\n key: 42,\n text: 'VertexAI',\n value: 42,\n color: 'primary'\n },\n 43: {\n key: 43,\n text: 'Proxy',\n value: 43,\n color: 'primary'\n },\n 44: {\n key: 44,\n text: 'SiliconFlow',\n value: 44,\n color: 'primary'\n },\n 45: {\n key: 45,\n text: 'xAI',\n value: 45,\n color: 'primary'\n },\n 45: {\n key: 46,\n text: 'Replicate',\n value: 46,\n color: 'primary'\n },\n 41: {\n key: 41,\n text: 'Novita',\n value: 41,\n color: 'purple'\n },\n 8: {\n key: 8,\n text: '自定义渠道',\n value: 8,\n color: 'error'\n },\n 22: {\n key: 22,\n text: '知识库：FastGPT',\n value: 22,\n color: 'success'\n },\n 21: {\n key: 21,\n text: '知识库：AI Proxy',\n value: 21,\n color: 'success'\n },\n 20: {\n key: 20,\n text: 'OpenRouter',\n value: 20,\n color: 'success'\n },\n 2: {\n key: 2,\n text: '代理：API2D',\n value: 2,\n color: 'success'\n },\n 5: {\n key: 5,\n text: '代理：OpenAI-SB',\n value: 5,\n color: 'success'\n },\n 7: {\n key: 7,\n text: '代理：OhMyGPT',\n value: 7,\n color: 'success'\n },\n 10: {\n key: 10,\n text: '代理：AI Proxy',\n value: 10,\n color: 'success'\n },\n 4: {\n key: 4,\n text: '代理：CloseAI',\n value: 4,\n color: 'success'\n },\n 6: {\n key: 6,\n text: '代理：OpenAI Max',\n value: 6,\n color: 'success'\n },\n 9: {\n key: 9,\n text: '代理：AI.LS',\n value: 9,\n color: 'success'\n },\n 12: {\n key: 12,\n text: '代理：API2GPT',\n value: 12,\n color: 'success'\n },\n 13: {\n key: 13,\n text: '代理：AIGC2D',\n value: 13,\n color: 'success'\n }\n};\n"
},
{
"path": "web/berry/src/constants/CommonConstants.js",
"content": "export const ITEMS_PER_PAGE = 10; // this value must keep same as the one defined in backend!\n"
},
{
"path": "web/berry/src/constants/SnackbarConstants.js",
"content": "import { closeSnackbar } from 'notistack';\nimport { IconX } from '@tabler/icons-react';\nimport { IconButton } from '@mui/material';\nconst action = (snackbarId) => (\n <>\n {\n closeSnackbar(snackbarId);\n }}\n >\n \n {\" \"}\n {children}{\" \"}\n \n );\n};\n\nexport default StatusProvider;\n"
},
{
"path": "web/berry/src/contexts/UserContext.js",
"content": "// contexts/User/index.jsx\nimport React, { useEffect, useCallback, createContext, useState } from 'react';\nimport { LOGIN } from 'store/actions';\nimport { useDispatch } from 'react-redux';\n\nexport const UserContext = createContext();\n\n// eslint-disable-next-line\nconst UserProvider = ({ children }) => {\n const dispatch = useDispatch();\n const [isUserLoaded, setIsUserLoaded] = useState(false);\n\n const loadUser = useCallback(() => {\n let user = localStorage.getItem('user');\n if (user) {\n let data = JSON.parse(user);\n dispatch({ type: LOGIN, payload: data });\n }\n setIsUserLoaded(true);\n }, [dispatch]);\n\n useEffect(() => {\n loadUser();\n }, [loadUser]);\n\n return {children} ;\n};\n\nexport default UserProvider;\n"
},
{
"path": "web/berry/src/hooks/useAuth.js",
"content": "import { isAdmin } from 'utils/common';\nimport { useNavigate } from 'react-router-dom';\nconst navigate = useNavigate();\n\nconst useAuth = () => {\n const userIsAdmin = isAdmin();\n\n if (!userIsAdmin) {\n navigate('/panel/404');\n }\n};\n\nexport default useAuth;\n"
},
{
"path": "web/berry/src/hooks/useLogin.js",
"content": "import { API } from 'utils/api';\nimport { useDispatch } from 'react-redux';\nimport { LOGIN } from 'store/actions';\nimport { useNavigate } from 'react-router';\nimport { showSuccess } from 'utils/common';\n\nconst useLogin = () => {\n const dispatch = useDispatch();\n const navigate = useNavigate();\n const login = async (username, password) => {\n try {\n const res = await API.post(`/api/user/login`, {\n username,\n password\n });\n const { success, message, data } = res.data;\n if (success) {\n localStorage.setItem('user', JSON.stringify(data));\n dispatch({ type: LOGIN, payload: data });\n navigate('/panel');\n }\n return { success, message };\n } catch (err) {\n // 请求失败，设置错误信息\n return { success: false, message: '' };\n }\n };\n\n const githubLogin = async (code, state) => {\n try {\n const res = await API.get(`/api/oauth/github?code=${code}&state=${state}`);\n const { success, message, data } = res.data;\n if (success) {\n if (message === 'bind') {\n showSuccess('绑定成功！');\n navigate('/panel');\n } else {\n dispatch({ type: LOGIN, payload: data });\n localStorage.setItem('user', JSON.stringify(data));\n showSuccess('登录成功！');\n navigate('/panel');\n }\n }\n return { success, message };\n } catch (err) {\n // 请求失败，设置错误信息\n return { success: false, message: '' };\n }\n };\n\n const larkLogin = async (code, state) => {\n try {\n const res = await API.get(`/api/oauth/lark?code=${code}&state=${state}`);\n const { success, message, data } = res.data;\n if (success) {\n if (message === 'bind') {\n showSuccess('绑定成功！');\n navigate('/panel');\n } else {\n dispatch({ type: LOGIN, payload: data });\n localStorage.setItem('user', JSON.stringify(data));\n showSuccess('登录成功！');\n navigate('/panel');\n }\n }\n return { success, message };\n } catch (err) {\n // 请求失败，设置错误信息\n return { success: false, message: '' };\n }\n };\n\n const oidcLogin = async (code, state) => {\n try {\n const res = await API.get(`/api/oauth/oidc?code=${code}&state=${state}`);\n const { success, message, data } = res.data;\n if (success) {\n if (message === 'bind') {\n showSuccess('绑定成功！');\n navigate('/panel');\n } else {\n dispatch({ type: LOGIN, payload: data });\n localStorage.setItem('user', JSON.stringify(data));\n showSuccess('登录成功！');\n navigate('/panel');\n }\n }\n return { success, message };\n } catch (err) {\n // 请求失败，设置错误信息\n return { success: false, message: '' };\n }\n }\n\n const wechatLogin = async (code) => {\n try {\n const res = await API.get(`/api/oauth/wechat?code=${code}`);\n const { success, message, data } = res.data;\n if (success) {\n dispatch({ type: LOGIN, payload: data });\n localStorage.setItem('user', JSON.stringify(data));\n showSuccess('登录成功！');\n navigate('/panel');\n }\n return { success, message };\n } catch (err) {\n // 请求失败，设置错误信息\n return { success: false, message: '' };\n }\n };\n\n const logout = async () => {\n await API.get('/api/user/logout');\n localStorage.removeItem('user');\n dispatch({ type: LOGIN, payload: null });\n navigate('/');\n };\n\n return { login, logout, githubLogin, wechatLogin, larkLogin,oidcLogin };\n};\n\nexport default useLogin;\n"
},
{
"path": "web/berry/src/hooks/useRegister.js",
"content": "import { API } from 'utils/api';\nimport { useNavigate } from 'react-router';\nimport { showSuccess } from 'utils/common';\n\nconst useRegister = () => {\n const navigate = useNavigate();\n const register = async (input, turnstile) => {\n try {\n let affCode = localStorage.getItem('aff');\n if (affCode) {\n input = { ...input, aff_code: affCode };\n }\n const res = await API.post(`/api/user/register?turnstile=${turnstile}`, input);\n const { success, message } = res.data;\n if (success) {\n showSuccess('注册成功！');\n navigate('/login');\n }\n return { success, message };\n } catch (err) {\n // 请求失败，设置错误信息\n return { success: false, message: '' };\n }\n };\n\n const sendVerificationCode = async (email, turnstile) => {\n try {\n const res = await API.get(`/api/verification?email=${email}&turnstile=${turnstile}`);\n const { success, message } = res.data;\n if (success) {\n showSuccess('验证码发送成功，请检查你的邮箱！');\n }\n return { success, message };\n } catch (err) {\n // 请求失败，设置错误信息\n return { success: false, message: '' };\n }\n };\n\n return { register, sendVerificationCode };\n};\n\nexport default useRegister;\n"
},
{
"path": "web/berry/src/hooks/useScriptRef.js",
"content": "import { useEffect, useRef } from 'react';\n\n// ==============================|| ELEMENT REFERENCE HOOKS ||============================== //\n\nconst useScriptRef = () => {\n const scripted = useRef(true);\n\n useEffect(\n () => () => {\n scripted.current = true;\n },\n []\n );\n\n return scripted;\n};\n\nexport default useScriptRef;\n"
},
{
"path": "web/berry/src/index.js",
"content": "import { createRoot } from 'react-dom/client';\n\n// third party\nimport { BrowserRouter } from 'react-router-dom';\nimport { Provider } from 'react-redux';\n\n// project imports\nimport * as serviceWorker from 'serviceWorker';\nimport App from 'App';\nimport { store } from 'store';\n\n// style + assets\nimport 'assets/scss/style.scss';\nimport config from './config';\n\n// ==============================|| REACT DOM RENDER ||============================== //\n\nconst container = document.getElementById('root');\nconst root = createRoot(container); // createRoot(container!) if you use TypeScript\nroot.render(\n \n \n \n \n);\n\n// If you want your app to work offline and load faster, you can change\n// unregister() to register() below. Note this comes with some pitfalls.\n// Learn more about service workers: https://bit.ly/CRA-PWA\nserviceWorker.register();\n"
},
{
"path": "web/berry/src/layout/MainLayout/Header/ProfileSection/index.js",
"content": "import { useState, useRef, useEffect } from 'react';\n\nimport { useSelector } from 'react-redux';\nimport { useNavigate } from 'react-router-dom';\n// material-ui\nimport { useTheme } from '@mui/material/styles';\nimport {\n Avatar,\n Chip,\n ClickAwayListener,\n List,\n ListItemButton,\n ListItemIcon,\n ListItemText,\n Paper,\n Popper,\n Typography\n} from '@mui/material';\n\n// project imports\nimport MainCard from 'ui-component/cards/MainCard';\nimport Transitions from 'ui-component/extended/Transitions';\nimport User1 from 'assets/images/users/user-round.svg';\nimport useLogin from 'hooks/useLogin';\n\n// assets\nimport { IconLogout, IconSettings, IconUserScan } from '@tabler/icons-react';\n\n// ==============================|| PROFILE MENU ||============================== //\n\nconst ProfileSection = () => {\n const theme = useTheme();\n const navigate = useNavigate();\n const customization = useSelector((state) => state.customization);\n const { logout } = useLogin();\n\n const [open, setOpen] = useState(false);\n /**\n * anchorRef is used on different componets and specifying one type leads to other components throwing an error\n * */\n const anchorRef = useRef(null);\n const handleLogout = async () => {\n logout();\n };\n\n const handleClose = (event) => {\n if (anchorRef.current && anchorRef.current.contains(event.target)) {\n return;\n }\n setOpen(false);\n };\n\n const handleToggle = () => {\n setOpen((prevOpen) => !prevOpen);\n };\n\n const prevOpen = useRef(open);\n useEffect(() => {\n if (prevOpen.current === true && open === false) {\n anchorRef.current.focus();\n }\n\n prevOpen.current = open;\n }, [open]);\n\n return (\n <>\n \n {({ TransitionProps }) => (\n \n \n \n \n \n navigate('/panel/profile')}>\n \n \n 设置} />\n \n\n \n \n \n 登出} />\n \n \n \n \n \n \n )}\n \n \n );\n};\n\nexport default ProfileSection;\n"
},
{
"path": "web/berry/src/layout/MainLayout/Header/index.js",
"content": "import PropTypes from 'prop-types';\n\n// material-ui\nimport { useTheme } from '@mui/material/styles';\nimport { Avatar, Box, ButtonBase } from '@mui/material';\n\n// project imports\nimport LogoSection from '../LogoSection';\nimport ProfileSection from './ProfileSection';\nimport ThemeButton from 'ui-component/ThemeButton';\n\n// assets\nimport { IconMenu2 } from '@tabler/icons-react';\n\n// ==============================|| MAIN NAVBAR / HEADER ||============================== //\n\nconst Header = ({ handleLeftDrawerToggle }) => {\n const theme = useTheme();\n\n return (\n <>\n {/* logo & toggler button */}\n \n \n \n \n \n \n \n\n dispatch({ type: MENU_OPEN, id: defaultId })} component={Link} to=\"/\">\n \n );\n};\n\nexport default LogoSection;\n"
},
{
"path": "web/berry/src/layout/MainLayout/Sidebar/MenuCard/index.js",
"content": "// import PropTypes from 'prop-types';\nimport { useSelector } from 'react-redux';\n\n// material-ui\nimport { styled, useTheme } from '@mui/material/styles';\nimport {\n Avatar,\n Card,\n CardContent,\n // Grid,\n // LinearProgress,\n List,\n ListItem,\n ListItemAvatar,\n ListItemText,\n Typography\n // linearProgressClasses\n} from '@mui/material';\nimport User1 from 'assets/images/users/user-round.svg';\nimport { useNavigate } from 'react-router-dom';\n\n// assets\n// import TableChartOutlinedIcon from '@mui/icons-material/TableChartOutlined';\n\n// styles\n// const BorderLinearProgress = styled(LinearProgress)(({ theme }) => ({\n// height: 10,\n// borderRadius: 30,\n// [`&.${linearProgressClasses.colorPrimary}`]: {\n// backgroundColor: '#fff'\n// },\n// [`& .${linearProgressClasses.bar}`]: {\n// borderRadius: 5,\n// backgroundColor: theme.palette.primary.main\n// }\n// }));\n\nconst CardStyle = styled(Card)(({ theme }) => ({\n background: theme.typography.menuChip.background,\n marginBottom: '22px',\n overflow: 'hidden',\n position: 'relative',\n '&:after': {\n content: '\"\"',\n position: 'absolute',\n width: '157px',\n height: '157px',\n background: theme.palette.primary[200],\n borderRadius: '50%',\n top: '-105px',\n right: '-96px'\n }\n}));\n\n// ==============================|| PROGRESS BAR WITH LABEL ||============================== //\n\n// function LinearProgressWithLabel({ value, ...others }) {\n// const theme = useTheme();\n\n// return (\n// \n// \n// \n// \n// \n// Progress\n// \n// \n// \n// {`${Math.round(value)}%`} \n// \n// \n// \n// \n// \n// \n// );\n// }\n\n// LinearProgressWithLabel.propTypes = {\n// value: PropTypes.number\n// };\n\n// ==============================|| SIDEBAR MENU Card ||============================== //\n\nconst MenuCard = () => {\n const theme = useTheme();\n const account = useSelector((state) => state.account);\n const navigate = useNavigate();\n\n return (\n \n \n \n \n \n navigate('/panel/profile')}\n >\n \n \n {account.user?.username}\n \n }\n secondary={ 欢迎回来 }\n />\n \n
\n \n \n );\n};\n\nexport default MenuCard;\n"
},
{
"path": "web/berry/src/layout/MainLayout/Sidebar/MenuList/NavCollapse/index.js",
"content": "import PropTypes from 'prop-types';\nimport { useEffect, useState } from 'react';\nimport { useSelector } from 'react-redux';\nimport { useLocation, useNavigate } from 'react-router';\n\n// material-ui\nimport { useTheme } from '@mui/material/styles';\nimport { Collapse, List, ListItemButton, ListItemIcon, ListItemText, Typography } from '@mui/material';\n\n// project imports\nimport NavItem from '../NavItem';\n\n// assets\nimport FiberManualRecordIcon from '@mui/icons-material/FiberManualRecord';\nimport { IconChevronDown, IconChevronUp } from '@tabler/icons-react';\n\n// ==============================|| SIDEBAR MENU LIST COLLAPSE ITEMS ||============================== //\n\nconst NavCollapse = ({ menu, level }) => {\n const theme = useTheme();\n const customization = useSelector((state) => state.customization);\n const navigate = useNavigate();\n\n const [open, setOpen] = useState(false);\n const [selected, setSelected] = useState(null);\n\n const handleClick = () => {\n setOpen(!open);\n setSelected(!selected ? menu.id : null);\n if (menu?.id !== 'authentication') {\n navigate(menu.children[0]?.url);\n }\n };\n\n const { pathname } = useLocation();\n const checkOpenForParent = (child, id) => {\n child.forEach((item) => {\n if (item.url === pathname) {\n setOpen(true);\n setSelected(id);\n }\n });\n };\n\n // menu collapse for sub-levels\n useEffect(() => {\n setOpen(false);\n setSelected(null);\n if (menu.children) {\n menu.children.forEach((item) => {\n if (item.children?.length) {\n checkOpenForParent(item.children, menu.id);\n }\n if (item.url === pathname) {\n setSelected(menu.id);\n setOpen(true);\n }\n });\n }\n\n // eslint-disable-next-line react-hooks/exhaustive-deps\n }, [pathname, menu.children]);\n\n // menu collapse & item\n const menus = menu.children?.map((item) => {\n switch (item.type) {\n case 'collapse':\n return \n Menu Items Error\n \n );\n }\n });\n\n const Icon = menu.icon;\n const menuIcon = menu.icon ? (\n 0 ? 'inherit' : 'medium'}\n />\n );\n\n return (\n <>\n 1 ? 'transparent !important' : 'inherit',\n py: level > 1 ? 1 : 1.25,\n pl: `${level * 24}px`\n }}\n selected={selected === menu.id}\n onClick={handleClick}\n >\n {menuIcon} \n \n {menu.title}\n \n }\n secondary={\n menu.caption && (\n \n {menu.caption}\n \n )\n }\n />\n {open ? (\n \n \n {menus}\n \n \n \n );\n};\n\nNavCollapse.propTypes = {\n menu: PropTypes.object,\n level: PropTypes.number\n};\n\nexport default NavCollapse;\n"
},
{
"path": "web/berry/src/layout/MainLayout/Sidebar/MenuList/NavGroup/index.js",
"content": "import PropTypes from 'prop-types';\n\n// material-ui\nimport { useTheme } from '@mui/material/styles';\nimport { Divider, List, Typography } from '@mui/material';\n\n// project imports\nimport NavItem from '../NavItem';\nimport NavCollapse from '../NavCollapse';\n\n// ==============================|| SIDEBAR MENU LIST GROUP ||============================== //\n\nconst NavGroup = ({ item }) => {\n const theme = useTheme();\n\n // menu list collapse & items\n const items = item.children?.map((menu) => {\n switch (menu.type) {\n case 'collapse':\n return \n Menu Items Error\n \n );\n }\n });\n\n return (\n <>\n \n {item.title}\n {item.caption && (\n \n {item.caption}\n \n )}\n \n )\n }\n >\n {items}\n \n\n {/* group divider */}\n id === item?.id) > -1 ? 8 : 6,\n height: customization.isOpen.findIndex((id) => id === item?.id) > -1 ? 8 : 6\n }}\n fontSize={level > 0 ? 'inherit' : 'medium'}\n />\n );\n\n let itemTarget = '_self';\n if (item.target) {\n itemTarget = '_blank';\n }\n\n let listItemProps = {\n component: forwardRef((props, ref) => 1 ? 'transparent !important' : 'inherit',\n py: level > 1 ? 1 : 1.25,\n pl: `${level * 24}px`\n }}\n selected={customization.isOpen.findIndex((id) => id === item.id) > -1}\n onClick={() => itemHandler(item.id)}\n >\n {itemIcon} \n id === item.id) > -1 ? 'h5' : 'body1'} color=\"inherit\">\n {item.title}\n \n }\n secondary={\n item.caption && (\n \n {item.caption}\n \n )\n }\n />\n {item.chip && (\n {item.chip.avatar}}\n />\n )}\n \n );\n};\n\nNavItem.propTypes = {\n item: PropTypes.object,\n level: PropTypes.number\n};\n\nexport default NavItem;\n"
},
{
"path": "web/berry/src/layout/MainLayout/Sidebar/MenuList/index.js",
"content": "// material-ui\nimport { Typography } from '@mui/material';\n\n// project imports\nimport NavGroup from './NavGroup';\nimport menuItem from 'menu-items';\nimport { isAdmin } from 'utils/common';\n\n// ==============================|| SIDEBAR MENU LIST ||============================== //\nconst MenuList = () => {\n const userIsAdmin = isAdmin();\n\n return (\n <>\n {menuItem.items.map((item) => {\n if (item.type !== 'group') {\n return (\n \n Menu Items Error\n \n );\n }\n\n const filteredChildren = item.children.filter((child) => !child.isAdmin || userIsAdmin);\n\n if (filteredChildren.length === 0) {\n return null;\n }\n\n return \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n );\n\n const container = window !== undefined ? () => window.document.body : undefined;\n\n return (\n \n \n {drawer}\n \n \n );\n};\n\nSidebar.propTypes = {\n drawerOpen: PropTypes.bool,\n drawerToggle: PropTypes.func,\n window: PropTypes.object\n};\n\nexport default Sidebar;\n"
},
{
"path": "web/berry/src/layout/MainLayout/index.js",
"content": "import { useDispatch, useSelector } from 'react-redux';\nimport { Outlet } from 'react-router-dom';\nimport AuthGuard from 'utils/route-guard/AuthGuard';\n\n// material-ui\nimport { styled, useTheme } from '@mui/material/styles';\nimport { AppBar, Box, CssBaseline, Toolbar, useMediaQuery } from '@mui/material';\nimport AdminContainer from 'ui-component/AdminContainer';\n\n// project imports\nimport Breadcrumbs from 'ui-component/extended/Breadcrumbs';\nimport Header from './Header';\nimport Sidebar from './Sidebar';\nimport navigation from 'menu-items';\nimport { drawerWidth } from 'store/constant';\nimport { SET_MENU } from 'store/actions';\n\n// assets\nimport { IconChevronRight } from '@tabler/icons-react';\n\n// styles\nconst Main = styled('main', { shouldForwardProp: (prop) => prop !== 'open' })(({ theme, open }) => ({\n ...theme.typography.mainContent,\n borderBottomLeftRadius: 0,\n borderBottomRightRadius: 0,\n transition: theme.transitions.create(\n 'margin',\n open\n ? {\n easing: theme.transitions.easing.easeOut,\n duration: theme.transitions.duration.enteringScreen\n }\n : {\n easing: theme.transitions.easing.sharp,\n duration: theme.transitions.duration.leavingScreen\n }\n ),\n [theme.breakpoints.up('md')]: {\n marginLeft: open ? 0 : -(drawerWidth - 20),\n width: `calc(100% - ${drawerWidth}px)`\n },\n [theme.breakpoints.down('md')]: {\n marginLeft: '20px',\n width: `calc(100% - ${drawerWidth}px)`,\n padding: '16px'\n },\n [theme.breakpoints.down('sm')]: {\n marginLeft: '10px',\n width: `calc(100% - ${drawerWidth}px)`,\n padding: '16px',\n marginRight: '10px'\n }\n}));\n\n// ==============================|| MAIN LAYOUT ||============================== //\n\nconst MainLayout = () => {\n const theme = useTheme();\n const matchDownMd = useMediaQuery(theme.breakpoints.down('md'));\n // Handle left drawer\n const leftDrawerOpened = useSelector((state) => state.customization.opened);\n const dispatch = useDispatch();\n const handleLeftDrawerToggle = () => {\n dispatch({ type: SET_MENU, opened: !leftDrawerOpened });\n };\n\n return (\n \n \n \n \n \n \n\n {/* drawer */}\n \n {/* breadcrumb */}\n \n \n \n \n \n \n );\n};\n\nexport default MainLayout;\n"
},
{
"path": "web/berry/src/layout/MinimalLayout/Header/index.js",
"content": "// material-ui\nimport { useState } from 'react';\nimport { useTheme } from '@mui/material/styles';\nimport {\n Box,\n Button,\n Stack,\n Popper,\n IconButton,\n List,\n ListItemButton,\n Paper,\n ListItemText,\n Typography,\n Divider,\n ClickAwayListener\n} from '@mui/material';\nimport LogoSection from 'layout/MainLayout/LogoSection';\nimport { Link } from 'react-router-dom';\nimport { useLocation } from 'react-router-dom';\nimport { useSelector } from 'react-redux';\nimport ThemeButton from 'ui-component/ThemeButton';\nimport ProfileSection from 'layout/MainLayout/Header/ProfileSection';\nimport { IconMenu2 } from '@tabler/icons-react';\nimport Transitions from 'ui-component/extended/Transitions';\nimport MainCard from 'ui-component/cards/MainCard';\nimport { useMediaQuery } from '@mui/material';\n\n// ==============================|| MAIN NAVBAR / HEADER ||============================== //\n\nconst Header = () => {\n const theme = useTheme();\n const { pathname } = useLocation();\n const account = useSelector((state) => state.account);\n const [open, setOpen] = useState(null);\n const isMobile = useMediaQuery(theme.breakpoints.down('sm'));\n\n const handleOpenMenu = (event) => {\n setOpen(open ? null : event.currentTarget);\n };\n\n const handleCloseMenu = () => {\n setOpen(null);\n };\n\n return (\n <>\n \n \n \n \n\n \n {isMobile ? (\n <>\n \n \n \n ) : (\n <>\n \n 首页\n \n \n 关于\n \n \n 控制台\n \n \n 登录\n \n )}\n \n )}\n \n\n \n {({ TransitionProps }) => (\n \n \n \n \n \n \n 首页} />\n \n\n \n 关于} />\n \n \n 控制台\n \n ) : (\n \n 登录\n \n )}\n \n \n \n \n \n )}\n \n \n );\n};\n\nexport default Header;\n"
},
{
"path": "web/berry/src/layout/MinimalLayout/index.js",
"content": "import { Outlet } from 'react-router-dom';\nimport { useTheme } from '@mui/material/styles';\nimport { AppBar, Box, CssBaseline, Toolbar, Container } from '@mui/material';\nimport Header from './Header';\nimport Footer from 'ui-component/Footer';\n\n// ==============================|| MINIMAL LAYOUT ||============================== //\n\nconst MinimalLayout = () => {\n const theme = useTheme();\n\n return (\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n );\n};\n\nexport default MinimalLayout;\n"
},
{
"path": "web/berry/src/layout/NavMotion.js",
"content": "import PropTypes from 'prop-types';\nimport { motion } from 'framer-motion';\n\n// ==============================|| ANIMATION FOR CONTENT ||============================== //\n\nconst NavMotion = ({ children }) => {\n const motionVariants = {\n initial: {\n opacity: 0,\n scale: 0.99\n },\n in: {\n opacity: 1,\n scale: 1\n },\n out: {\n opacity: 0,\n scale: 1.01\n }\n };\n\n const motionTransition = {\n type: 'tween',\n ease: 'anticipate',\n duration: 0.4\n };\n\n return (\n \n {children}\n \n );\n};\n\nNavMotion.propTypes = {\n children: PropTypes.node\n};\n\nexport default NavMotion;\n"
},
{
"path": "web/berry/src/layout/NavigationScroll.js",
"content": "import PropTypes from 'prop-types';\nimport { useEffect } from 'react';\nimport { useLocation } from 'react-router-dom';\n\n// ==============================|| NAVIGATION SCROLL TO TOP ||============================== //\n\nconst NavigationScroll = ({ children }) => {\n const location = useLocation();\n const { pathname } = location;\n\n useEffect(() => {\n window.scrollTo({\n top: 0,\n left: 0,\n behavior: 'smooth'\n });\n }, [pathname]);\n\n return children || null;\n};\n\nNavigationScroll.propTypes = {\n children: PropTypes.node\n};\n\nexport default NavigationScroll;\n"
},
{
"path": "web/berry/src/menu-items/index.js",
"content": "import panel from './panel';\n\n// ==============================|| MENU ITEMS ||============================== //\n\nconst menuItems = {\n items: [panel],\n urlMap: {}\n};\n\n// Initialize urlMap\nmenuItems.urlMap = menuItems.items.reduce((map, item) => {\n item.children.forEach((child) => {\n map[child.url] = child;\n });\n return map;\n}, {});\n\nexport default menuItems;\n"
},
{
"path": "web/berry/src/menu-items/panel.js",
"content": "// assets\nimport {\n IconDashboard,\n IconSitemap,\n IconArticle,\n IconCoin,\n IconAdjustments,\n IconKey,\n IconGardenCart,\n IconUser,\n IconUserScan\n} from '@tabler/icons-react';\n\n// constant\nconst icons = { IconDashboard, IconSitemap, IconArticle, IconCoin, IconAdjustments, IconKey, IconGardenCart, IconUser, IconUserScan };\n\n// ==============================|| DASHBOARD MENU ITEMS ||============================== //\n\nconst panel = {\n id: 'panel',\n type: 'group',\n children: [\n {\n id: 'dashboard',\n title: '总览',\n type: 'item',\n url: '/panel/dashboard',\n icon: icons.IconDashboard,\n breadcrumbs: false,\n isAdmin: false\n },\n {\n id: 'channel',\n title: '渠道',\n type: 'item',\n url: '/panel/channel',\n icon: icons.IconSitemap,\n breadcrumbs: false,\n isAdmin: true\n },\n {\n id: 'token',\n title: '令牌',\n type: 'item',\n url: '/panel/token',\n icon: icons.IconKey,\n breadcrumbs: false\n },\n {\n id: 'log',\n title: '日志',\n type: 'item',\n url: '/panel/log',\n icon: icons.IconArticle,\n breadcrumbs: false\n },\n {\n id: 'redemption',\n title: '兑换',\n type: 'item',\n url: '/panel/redemption',\n icon: icons.IconCoin,\n breadcrumbs: false,\n isAdmin: true\n },\n {\n id: 'topup',\n title: '充值',\n type: 'item',\n url: '/panel/topup',\n icon: icons.IconGardenCart,\n breadcrumbs: false\n },\n {\n id: 'user',\n title: '用户',\n type: 'item',\n url: '/panel/user',\n icon: icons.IconUser,\n breadcrumbs: false,\n isAdmin: true\n },\n {\n id: 'profile',\n title: '我的',\n type: 'item',\n url: '/panel/profile',\n icon: icons.IconUserScan,\n breadcrumbs: false,\n isAdmin: false\n },\n {\n id: 'setting',\n title: '设置',\n type: 'item',\n url: '/panel/setting',\n icon: icons.IconAdjustments,\n breadcrumbs: false,\n isAdmin: true\n }\n ]\n};\n\nexport default panel;\n"
},
{
"path": "web/berry/src/routes/MainRoutes.js",
"content": "import { lazy } from 'react';\n\n// project imports\nimport MainLayout from 'layout/MainLayout';\nimport Loadable from 'ui-component/Loadable';\n\nconst Channel = Loadable(lazy(() => import('views/Channel')));\nconst Log = Loadable(lazy(() => import('views/Log')));\nconst Redemption = Loadable(lazy(() => import('views/Redemption')));\nconst Setting = Loadable(lazy(() => import('views/Setting')));\nconst Token = Loadable(lazy(() => import('views/Token')));\nconst Topup = Loadable(lazy(() => import('views/Topup')));\nconst User = Loadable(lazy(() => import('views/User')));\nconst Profile = Loadable(lazy(() => import('views/Profile')));\nconst NotFoundView = Loadable(lazy(() => import('views/Error')));\n\n// dashboard routing\nconst Dashboard = Loadable(lazy(() => import('views/Dashboard')));\n\n// ==============================|| MAIN ROUTING ||============================== //\n\nconst MainRoutes = {\n path: '/panel',\n element: \n \n {siteInfo.footer_html ? (\n
\n ) : (\n <>\n \n \n );\n};\n\nexport default Footer;\n"
},
{
"path": "web/berry/src/ui-component/Label.js",
"content": "/*\n * Label.js\n *\n * This file uses code from the Minimal UI project, available at\n * https://github.com/minimal-ui-kit/material-kit-react/blob/main/src/components/label/label.jsx\n *\n * Minimal UI is licensed under the MIT License. A copy of the license is included below:\n *\n * MIT License\n *\n * Copyright (c) 2021 Minimal UI (https://minimals.cc/)\n *\n * Permission is hereby granted, free of charge, to any person obtaining a copy\n * of this software and associated documentation files (the \"Software\"), to deal\n * in the Software without restriction, including without limitation the rights\n * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n * copies of the Software, and to permit persons to whom the Software is\n * furnished to do so, subject to the following conditions:\n *\n * The above copyright notice and this permission notice shall be included in all\n * copies or substantial portions of the Software.\n *\n * THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n * SOFTWARE.\n */\nimport PropTypes from 'prop-types';\nimport { forwardRef } from 'react';\n\nimport Box from '@mui/material/Box';\nimport { useTheme } from '@mui/material/styles';\nimport { alpha, styled } from '@mui/material/styles';\n\n// ----------------------------------------------------------------------\n\nconst Label = forwardRef(({ children, color = 'default', variant = 'soft', startIcon, endIcon, sx, ...other }, ref) => {\n const theme = useTheme();\n\n const iconStyles = {\n width: 16,\n height: 16,\n '& svg, img': { width: 1, height: 1, objectFit: 'cover' }\n };\n\n return (\n \n {startIcon && {startIcon} }\n\n {children}\n\n {endIcon && {endIcon} }\n \n );\n});\n\nLabel.propTypes = {\n children: PropTypes.node,\n endIcon: PropTypes.object,\n startIcon: PropTypes.object,\n sx: PropTypes.object,\n variant: PropTypes.oneOf(['filled', 'outlined', 'ghost', 'soft']),\n color: PropTypes.oneOf(['default', 'primary', 'secondary', 'info', 'success', 'warning', 'orange', 'error'])\n};\n\nexport default Label;\n\nconst StyledLabel = styled(Box)(({ theme, ownerState }) => {\n // const lightMode = theme.palette.mode === 'light';\n\n const filledVariant = ownerState.variant === 'filled';\n\n const outlinedVariant = ownerState.variant === 'outlined';\n\n const softVariant = ownerState.variant === 'soft';\n\n const ghostVariant = ownerState.variant === 'ghost';\n\n const defaultStyle = {\n ...(ownerState.color === 'default' && {\n // FILLED\n ...(filledVariant && {\n color: theme.palette.grey[300],\n backgroundColor: theme.palette.text.primary\n }),\n // OUTLINED\n ...(outlinedVariant && {\n color: theme.palette.grey[500],\n border: `2px solid ${theme.palette.grey[500]}`\n }),\n // SOFT\n ...(softVariant && {\n color: theme.palette.text.secondary,\n backgroundColor: alpha(theme.palette.grey[500], 0.16)\n })\n })\n };\n\n const colorStyle = {\n ...(ownerState.color !== 'default' && {\n // FILLED\n ...(filledVariant && {\n color: theme.palette.background.paper,\n backgroundColor: theme.palette[ownerState.color]?.main\n }),\n // OUTLINED\n ...(outlinedVariant && {\n backgroundColor: 'transparent',\n color: theme.palette[ownerState.color]?.main,\n border: `2px solid ${theme.palette[ownerState.color]?.main}`\n }),\n // SOFT\n ...(softVariant && {\n color: theme.palette[ownerState.color]['dark'],\n backgroundColor: alpha(theme.palette[ownerState.color]?.main, 0.16)\n }),\n // GHOST\n ...(ghostVariant && {\n color: theme.palette[ownerState.color]?.main\n })\n })\n };\n\n return {\n height: 24,\n minWidth: 24,\n lineHeight: 0,\n borderRadius: 6,\n cursor: 'default',\n alignItems: 'center',\n whiteSpace: 'nowrap',\n display: 'inline-flex',\n justifyContent: 'center',\n // textTransform: 'capitalize',\n padding: theme.spacing(0, 0.75),\n fontSize: theme.typography.pxToRem(12),\n fontWeight: theme.typography.fontWeightBold,\n transition: theme.transitions.create('all', {\n duration: theme.transitions.duration.shorter\n }),\n ...defaultStyle,\n ...colorStyle\n };\n});\n"
},
{
"path": "web/berry/src/ui-component/Loadable.js",
"content": "import { Suspense } from 'react';\n\n// project imports\nimport Loader from './Loader';\n\n// ==============================|| LOADABLE - LAZY LOADING ||============================== //\n\nconst Loadable = (Component) => (props) =>\n (\n \n \n);\n\nexport default Loader;\n"
},
{
"path": "web/berry/src/ui-component/Logo.js",
"content": "// material-ui\nimport logoLight from 'assets/images/logo.svg';\nimport logoDark from 'assets/images/logo-white.svg';\nimport { useSelector } from 'react-redux';\nimport { useTheme } from '@mui/material/styles';\n\n/**\n * if you want to use image instead of uncomment following.\n *\n * import logoDark from 'assets/images/logo-dark.svg';\n * import logo from 'assets/images/logo.svg';\n *\n */\n\n// ==============================|| LOGO SVG ||============================== //\n\nconst Logo = () => {\n const siteInfo = useSelector((state) => state.siteInfo);\n const theme = useTheme();\n const logo = theme.palette.mode === 'light' ? logoLight : logoDark;\n\n return theme.spacing(0, 1, 0, 3)\n }}\n >\n \n \n \n {\n let theme = defaultTheme === 'light' ? 'dark' : 'light';\n dispatch({ type: SET_THEME, theme: theme });\n localStorage.setItem('theme', theme);\n }}\n color=\"inherit\"\n >\n {defaultTheme === 'light' ? \n \n );\n}\n"
},
{
"path": "web/berry/src/ui-component/cards/CardSecondaryAction.js",
"content": "import PropTypes from 'prop-types';\nimport { useTheme } from '@mui/material/styles';\nimport { ButtonBase, Link, Tooltip } from '@mui/material';\n\n// project imports\nimport Avatar from '../extended/Avatar';\n\n// ==============================|| CARD SECONDARY ACTION ||============================== //\n\nconst CardSecondaryAction = ({ title, link, icon }) => {\n const theme = useTheme();\n\n return (\n \n \n {!icon && (\n \n \n \n \n \n \n \n \n \n \n )}\n {icon && (\n \n {icon}\n \n )}\n \n \n );\n};\n\nCardSecondaryAction.propTypes = {\n icon: PropTypes.node,\n link: PropTypes.string,\n title: PropTypes.string\n};\n\nexport default CardSecondaryAction;\n"
},
{
"path": "web/berry/src/ui-component/cards/MainCard.js",
"content": "import PropTypes from 'prop-types';\nimport { forwardRef } from 'react';\n\n// material-ui\nimport { useTheme } from '@mui/material/styles';\nimport { Card, CardContent, CardHeader, Divider, Typography } from '@mui/material';\n\n// constant\nconst headerSX = {\n '& .MuiCardHeader-action': { mr: 0 }\n};\n\n// ==============================|| CUSTOM MAIN CARD ||============================== //\n\nconst MainCard = forwardRef(\n (\n {\n border = false,\n boxShadow,\n children,\n content = true,\n contentClass = '',\n contentSX = {},\n darkTitle,\n secondary,\n shadow,\n sx = {},\n title,\n ...others\n },\n ref\n ) => {\n const theme = useTheme();\n\n return (\n \n {/* card header and action */}\n {title && {title} : title} action={secondary} />}\n\n {/* content & header divider */}\n {title && \n {children}\n \n )}\n {!content && children}\n \n );\n }\n);\n\nMainCard.propTypes = {\n border: PropTypes.bool,\n boxShadow: PropTypes.bool,\n children: PropTypes.node,\n content: PropTypes.bool,\n contentClass: PropTypes.string,\n contentSX: PropTypes.object,\n darkTitle: PropTypes.bool,\n secondary: PropTypes.oneOfType([PropTypes.node, PropTypes.string, PropTypes.object]),\n shadow: PropTypes.string,\n sx: PropTypes.object,\n title: PropTypes.oneOfType([PropTypes.node, PropTypes.string, PropTypes.object])\n};\n\nexport default MainCard;\n"
},
{
"path": "web/berry/src/ui-component/cards/Skeleton/EarningCard.js",
"content": "// material-ui\nimport { Card, CardContent, Grid } from '@mui/material';\nimport Skeleton from '@mui/material/Skeleton';\n\n// ==============================|| SKELETON - EARNING CARD ||============================== //\n\nconst EarningCard = () => (\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n);\n\nexport default EarningCard;\n"
},
{
"path": "web/berry/src/ui-component/cards/Skeleton/ImagePlaceholder.js",
"content": "// material-ui\nimport Skeleton from '@mui/material/Skeleton';\n\n// ==============================|| SKELETON IMAGE CARD ||============================== //\n\nconst ImagePlaceholder = ({ ...others }) => \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n);\n\nexport default PopularCard;\n"
},
{
"path": "web/berry/src/ui-component/cards/Skeleton/ProductPlaceholder.js",
"content": "// material-ui\nimport { CardContent, Grid, Skeleton, Stack } from '@mui/material';\n\n// project import\nimport MainCard from '../MainCard';\n\n// ===========================|| SKELETON TOTAL GROWTH BAR CHART ||=========================== //\n\nconst ProductPlaceholder = () => (\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n);\n\nexport default ProductPlaceholder;\n"
},
{
"path": "web/berry/src/ui-component/cards/Skeleton/TotalGrowthBarChart.js",
"content": "// material-ui\nimport { Card, CardContent, Grid } from '@mui/material';\nimport Skeleton from '@mui/material/Skeleton';\n\n// project imports\nimport { gridSpacing } from 'store/constant';\n\n// ==============================|| SKELETON TOTAL GROWTH BAR CHART ||============================== //\n\nconst TotalGrowthBarChart = () => (\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n);\n\nexport default TotalGrowthBarChart;\n"
},
{
"path": "web/berry/src/ui-component/cards/Skeleton/TotalIncomeCard.js",
"content": "// material-ui\nimport { Card, List, ListItem, ListItemAvatar, ListItemText, Skeleton } from '@mui/material';\n\n// ==============================|| SKELETON - TOTAL INCOME DARK/LIGHT CARD ||============================== //\n\nconst TotalIncomeCard = () => (\n \n \n \n \n \n \n
\n \n);\n\nexport default TotalIncomeCard;\n"
},
{
"path": "web/berry/src/ui-component/cards/SubCard.js",
"content": "import PropTypes from 'prop-types';\nimport { forwardRef } from 'react';\n\n// material-ui\nimport { useTheme } from '@mui/material/styles';\nimport { Card, CardContent, CardHeader, Divider, Typography } from '@mui/material';\n\n// ==============================|| CUSTOM SUB CARD ||============================== //\n\nconst SubCard = forwardRef(\n ({ children, content, contentClass, darkTitle, secondary, sx = {}, contentSX = {}, title, subTitle, ...others }, ref) => {\n const theme = useTheme();\n\n return (\n \n {/* card header and action */}\n {!darkTitle && title && (\n {title}} action={secondary} subheader={subTitle} />\n )}\n {darkTitle && title && (\n {title}} action={secondary} subheader={subTitle} />\n )}\n\n {/* content & header divider */}\n {title && (\n \n {children}\n \n )}\n {!content && children}\n \n );\n }\n);\n\nSubCard.propTypes = {\n children: PropTypes.node,\n content: PropTypes.bool,\n contentClass: PropTypes.string,\n darkTitle: PropTypes.bool,\n secondary: PropTypes.oneOfType([PropTypes.node, PropTypes.string, PropTypes.object]),\n sx: PropTypes.object,\n contentSX: PropTypes.object,\n title: PropTypes.oneOfType([PropTypes.node, PropTypes.string, PropTypes.object]),\n subTitle: PropTypes.oneOfType([PropTypes.node, PropTypes.string, PropTypes.object])\n};\n\nSubCard.defaultProps = {\n content: true\n};\n\nexport default SubCard;\n"
},
{
"path": "web/berry/src/ui-component/cards/UserCard.js",
"content": "/*\n * UserCard.js\n *\n * This file uses code from the Minimal UI project, available at\n * https://github.com/minimal-ui-kit/material-kit-react/blob/main/src/sections/blog/post-card.jsx\n *\n * Minimal UI is licensed under the MIT License. A copy of the license is included below:\n *\n * MIT License\n *\n * Copyright (c) 2021 Minimal UI (https://minimals.cc/)\n *\n * Permission is hereby granted, free of charge, to any person obtaining a copy\n * of this software and associated documentation files (the \"Software\"), to deal\n * in the Software without restriction, including without limitation the rights\n * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n * copies of the Software, and to permit persons to whom the Software is\n * furnished to do so, subject to the following conditions:\n *\n * The above copyright notice and this permission notice shall be included in all\n * copies or substantial portions of the Software.\n *\n * THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n * SOFTWARE.\n */\nimport { Box, Avatar } from '@mui/material';\nimport { alpha } from '@mui/material/styles';\nimport Card from '@mui/material/Card';\nimport shapeAvatar from 'assets/images/icons/shape-avatar.svg';\nimport coverAvatar from 'assets/images/invite/cover.jpg';\nimport userAvatar from 'assets/images/users/user-round.svg';\nimport SvgColor from 'ui-component/SvgColor';\n\nimport React from 'react';\n\nexport default function UserCard({ children }) {\n const renderShape = (\n theme.spacing(-4)\n }}\n />\n );\n\n const renderCover = (\n \n alpha(theme.palette.primary.main, 0.42)\n },\n pt: {\n xs: 'calc(100% / 3)',\n sm: 'calc(100% / 4.66)'\n }\n }}\n >\n {renderShape}\n {renderAvatar}\n {renderCover}\n \n theme.spacing(4, 3, 3, 3)\n }}\n >\n {children}\n \n \n );\n}\n"
},
{
"path": "web/berry/src/ui-component/extended/AnimateButton.js",
"content": "import PropTypes from 'prop-types';\nimport { forwardRef } from 'react';\n// third-party\nimport { motion, useCycle } from 'framer-motion';\n\n// ==============================|| ANIMATION BUTTON ||============================== //\n\nconst AnimateButton = forwardRef(({ children, type, direction, offset, scale }, ref) => {\n let offset1;\n let offset2;\n switch (direction) {\n case 'up':\n case 'left':\n offset1 = offset;\n offset2 = 0;\n break;\n case 'right':\n case 'down':\n default:\n offset1 = 0;\n offset2 = offset;\n break;\n }\n\n const [x, cycleX] = useCycle(offset1, offset2);\n const [y, cycleY] = useCycle(offset1, offset2);\n\n switch (type) {\n case 'rotate':\n return (\n \n {children}\n \n );\n case 'slide':\n if (direction === 'up' || direction === 'down') {\n return (\n cycleY()} onHoverStart={() => cycleY()}>\n {children}\n \n );\n }\n return (\n cycleX()} onHoverStart={() => cycleX()}>\n {children}\n \n );\n\n case 'scale':\n default:\n if (typeof scale === 'number') {\n scale = {\n hover: scale,\n tap: scale\n };\n }\n return (\n \n {children}\n \n );\n }\n});\n\nAnimateButton.propTypes = {\n children: PropTypes.node,\n offset: PropTypes.number,\n type: PropTypes.oneOf(['slide', 'scale', 'rotate']),\n direction: PropTypes.oneOf(['up', 'down', 'left', 'right']),\n scale: PropTypes.oneOfType([PropTypes.number, PropTypes.object])\n};\n\nAnimateButton.defaultProps = {\n type: 'scale',\n offset: 10,\n direction: 'right',\n scale: {\n hover: 1,\n tap: 0.9\n }\n};\n\nexport default AnimateButton;\n"
},
{
"path": "web/berry/src/ui-component/extended/Avatar.js",
"content": "import PropTypes from 'prop-types';\n\n// material-ui\nimport { useTheme } from '@mui/material/styles';\nimport MuiAvatar from '@mui/material/Avatar';\n\n// ==============================|| AVATAR ||============================== //\n\nconst Avatar = ({ color, outline, size, sx, ...others }) => {\n const theme = useTheme();\n\n const colorSX = color && !outline && { color: theme.palette.background.paper, bgcolor: `${color}.main` };\n const outlineSX = outline && {\n color: color ? `${color}.main` : `primary.main`,\n bgcolor: theme.palette.background.paper,\n border: '2px solid',\n borderColor: color ? `${color}.main` : `primary.main`\n };\n let sizeSX = {};\n switch (size) {\n case 'badge':\n sizeSX = {\n width: theme.spacing(3.5),\n height: theme.spacing(3.5)\n };\n break;\n case 'xs':\n sizeSX = {\n width: theme.spacing(4.25),\n height: theme.spacing(4.25)\n };\n break;\n case 'sm':\n sizeSX = {\n width: theme.spacing(5),\n height: theme.spacing(5)\n };\n break;\n case 'lg':\n sizeSX = {\n width: theme.spacing(9),\n height: theme.spacing(9)\n };\n break;\n case 'xl':\n sizeSX = {\n width: theme.spacing(10.25),\n height: theme.spacing(10.25)\n };\n break;\n case 'md':\n sizeSX = {\n width: theme.spacing(7.5),\n height: theme.spacing(7.5)\n };\n break;\n default:\n sizeSX = {};\n }\n\n return \n {icons && \n );\n }\n\n // items\n if (item && item.type === 'item') {\n itemTitle = item.title;\n\n ItemIcon = item.icon ? item.icon : AccountTreeTwoToneIcon;\n itemContent = (\n \n {icons && \n \n \n {title && !titleBottom && (\n \n \n {item.title}\n \n \n )}\n \n \n \n {icons && \n {mainContent}\n {itemContent}\n \n \n {title && titleBottom && (\n \n \n {item.title}\n \n \n )}\n \n \n {card === false && divider !== false && \n {type === 'grow' && (\n \n {children} \n \n )}\n {type === 'collapse' && (\n \n {children}\n \n )}\n {type === 'fade' && (\n \n {children} \n \n )}\n {type === 'slide' && (\n \n {children} \n \n )}\n {type === 'zoom' && (\n \n {children} \n \n )}\n \n );\n});\n\nTransitions.propTypes = {\n children: PropTypes.node,\n type: PropTypes.oneOf(['grow', 'fade', 'collapse', 'slide', 'zoom']),\n position: PropTypes.oneOf(['top-left', 'top-right', 'top', 'bottom-left', 'bottom-right', 'bottom']),\n direction: PropTypes.oneOf(['up', 'down', 'left', 'right'])\n};\n\nTransitions.defaultProps = {\n type: 'grow',\n position: 'top-left',\n direction: 'up'\n};\n\nexport default Transitions;\n"
},
{
"path": "web/berry/src/utils/api.js",
"content": "import { showError } from './common';\nimport axios from 'axios';\nimport { store } from 'store/index';\nimport { LOGIN } from 'store/actions';\nimport config from 'config';\n\nexport const API = axios.create({\n baseURL: process.env.REACT_APP_SERVER ? process.env.REACT_APP_SERVER : '/'\n});\n\nAPI.interceptors.response.use(\n (response) => response,\n (error) => {\n if (error.response?.status === 401) {\n localStorage.removeItem('user');\n store.dispatch({ type: LOGIN, payload: null });\n window.location.href = config.basename + 'login';\n }\n\n if (error.response?.data?.message) {\n error.message = error.response.data.message;\n }\n\n showError(error);\n }\n);\n"
},
{
"path": "web/berry/src/utils/chart.js",
"content": "export function getLastSevenDays() {\n const dates = [];\n for (let i = 6; i >= 0; i--) {\n const d = new Date();\n d.setDate(d.getDate() - i);\n const month = '' + (d.getMonth() + 1);\n const day = '' + d.getDate();\n const year = d.getFullYear();\n\n const formattedDate = [year, month.padStart(2, '0'), day.padStart(2, '0')].join('-');\n dates.push(formattedDate);\n }\n return dates;\n}\n\nexport function getTodayDay() {\n let today = new Date();\n return today.toISOString().slice(0, 10);\n}\n\nexport function generateChartOptions(data, unit) {\n const dates = data.map((item) => item.date);\n const values = data.map((item) => item.value);\n\n const minDate = dates[0];\n const maxDate = dates[dates.length - 1];\n\n const minValue = Math.min(...values);\n const maxValue = Math.max(...values);\n\n return {\n series: [\n {\n data: values\n }\n ],\n type: 'line',\n height: 90,\n options: {\n chart: {\n sparkline: {\n enabled: true\n },\n background: 'transparent'\n },\n dataLabels: {\n enabled: false\n },\n colors: ['#fff'],\n fill: {\n type: 'solid',\n opacity: 1\n },\n stroke: {\n curve: 'smooth',\n width: 3\n },\n xaxis: {\n categories: dates,\n labels: {\n show: false\n },\n min: minDate,\n max: maxDate\n },\n yaxis: {\n min: minValue,\n max: maxValue,\n labels: {\n show: false\n }\n },\n tooltip: {\n theme: 'dark',\n fixed: {\n enabled: false\n },\n x: {\n format: 'yyyy-MM-dd'\n },\n y: {\n formatter: function (val) {\n return val + ` ${unit}`;\n },\n title: {\n formatter: function () {\n return '';\n }\n }\n },\n marker: {\n show: false\n }\n }\n }\n };\n}\n"
},
{
"path": "web/berry/src/utils/common.js",
"content": "import {enqueueSnackbar} from 'notistack';\nimport {snackbarConstants} from 'constants/SnackbarConstants';\nimport {API} from './api';\n\nexport function getSystemName() {\n let system_name = localStorage.getItem('system_name');\n if (!system_name) return 'One API';\n return system_name;\n}\n\nexport function isMobile() {\n return window.innerWidth <= 600;\n}\n\n// eslint-disable-next-line\nexport function SnackbarHTMLContent({htmlContent}) {\n return
;\n}\n\nexport function getSnackbarOptions(variant) {\n let options = snackbarConstants.Common[variant];\n if (isMobile()) {\n // 合并 options 和 snackbarConstants.Mobile\n options = {...options, ...snackbarConstants.Mobile};\n }\n return options;\n}\n\nexport function showError(error) {\n if (error.message) {\n if (error.name === 'AxiosError') {\n switch (error.response.status) {\n case 429:\n enqueueSnackbar('错误：请求次数过多，请稍后再试！', getSnackbarOptions('ERROR'));\n break;\n case 500:\n enqueueSnackbar('错误：服务器内部错误，请联系管理员！', getSnackbarOptions('ERROR'));\n break;\n case 405:\n enqueueSnackbar('本站仅作演示之用，无服务端！', getSnackbarOptions('INFO'));\n break;\n default:\n enqueueSnackbar('错误：' + error.message, getSnackbarOptions('ERROR'));\n }\n return;\n }\n } else {\n enqueueSnackbar('错误：' + error, getSnackbarOptions('ERROR'));\n }\n}\n\nexport function showNotice(message, isHTML = false) {\n if (isHTML) {\n enqueueSnackbar(, getSnackbarOptions('NOTICE'));\n } else {\n enqueueSnackbar(message, getSnackbarOptions('NOTICE'));\n }\n}\n\nexport function showWarning(message) {\n enqueueSnackbar(message, getSnackbarOptions('WARNING'));\n}\n\nexport function showSuccess(message) {\n enqueueSnackbar(message, getSnackbarOptions('SUCCESS'));\n}\n\nexport function showInfo(message) {\n enqueueSnackbar(message, getSnackbarOptions('INFO'));\n}\n\nexport async function getOAuthState() {\n const res = await API.get('/api/oauth/state');\n const {success, message, data} = res.data;\n if (success) {\n return data;\n } else {\n showError(message);\n return '';\n }\n}\n\nexport async function onGitHubOAuthClicked(github_client_id, openInNewTab = false) {\n const state = await getOAuthState();\n if (!state) return;\n let url = `https://github.com/login/oauth/authorize?client_id=${github_client_id}&state=${state}&scope=user:email`;\n if (openInNewTab) {\n window.open(url);\n } else {\n window.location.href = url;\n }\n}\n\nexport async function onLarkOAuthClicked(lark_client_id) {\n const state = await getOAuthState();\n if (!state) return;\n let redirect_uri = `${window.location.origin}/oauth/lark`;\n window.open(`https://accounts.feishu.cn/open-apis/authen/v1/authorize?redirect_uri=${redirect_uri}&client_id=${lark_client_id}&state=${state}`);\n}\n\nexport async function onOidcClicked(auth_url, client_id, openInNewTab = false) {\n const state = await getOAuthState();\n if (!state) return;\n const redirect_uri = `${window.location.origin}/oauth/oidc`;\n const response_type = \"code\";\n const scope = \"openid profile email\";\n const url = `${auth_url}?client_id=${client_id}&redirect_uri=${redirect_uri}&response_type=${response_type}&scope=${scope}&state=${state}`;\n if (openInNewTab) {\n window.open(url);\n } else {\n window.location.href = url;\n }\n}\n\nexport function isAdmin() {\n let user = localStorage.getItem('user');\n if (!user) return false;\n user = JSON.parse(user);\n return user.role >= 10;\n}\n\nexport function timestamp2string(timestamp) {\n let date = new Date(timestamp * 1000);\n let year = date.getFullYear().toString();\n let month = (date.getMonth() + 1).toString();\n let day = date.getDate().toString();\n let hour = date.getHours().toString();\n let minute = date.getMinutes().toString();\n let second = date.getSeconds().toString();\n if (month.length === 1) {\n month = '0' + month;\n }\n if (day.length === 1) {\n day = '0' + day;\n }\n if (hour.length === 1) {\n hour = '0' + hour;\n }\n if (minute.length === 1) {\n minute = '0' + minute;\n }\n if (second.length === 1) {\n second = '0' + second;\n }\n return year + '-' + month + '-' + day + ' ' + hour + ':' + minute + ':' + second;\n}\n\nexport function calculateQuota(quota, digits = 2) {\n let quotaPerUnit = localStorage.getItem('quota_per_unit');\n quotaPerUnit = parseFloat(quotaPerUnit);\n\n return (quota / quotaPerUnit).toFixed(digits);\n}\n\nexport function renderQuota(quota, digits = 2) {\n let displayInCurrency = localStorage.getItem('display_in_currency');\n displayInCurrency = displayInCurrency === 'true';\n if (displayInCurrency) {\n return '$' + calculateQuota(quota, digits);\n }\n return renderNumber(quota);\n}\n\nexport const verifyJSON = (str) => {\n try {\n JSON.parse(str);\n } catch (e) {\n return false;\n }\n return true;\n};\n\nexport function renderNumber(num) {\n if (num >= 1000000000) {\n return (num / 1000000000).toFixed(1) + 'B';\n } else if (num >= 1000000) {\n return (num / 1000000).toFixed(1) + 'M';\n } else if (num >= 10000) {\n return (num / 1000).toFixed(1) + 'k';\n } else {\n return num;\n }\n}\n\nexport function renderQuotaWithPrompt(quota, digits) {\n let displayInCurrency = localStorage.getItem('display_in_currency');\n displayInCurrency = displayInCurrency === 'true';\n if (displayInCurrency) {\n return `（等价金额：${renderQuota(quota, digits)}）`;\n }\n return '';\n}\n\nexport function downloadTextAsFile(text, filename) {\n let blob = new Blob([text], {type: 'text/plain;charset=utf-8'});\n let url = URL.createObjectURL(blob);\n let a = document.createElement('a');\n a.href = url;\n a.download = filename;\n a.click();\n}\n\nexport function removeTrailingSlash(url) {\n if (url.endsWith('/')) {\n return url.slice(0, -1);\n } else {\n return url;\n }\n}\n\nlet channelModels = undefined;\n\nexport async function loadChannelModels() {\n const res = await API.get('/api/models');\n const {success, data} = res.data;\n if (!success) {\n return;\n }\n channelModels = data;\n localStorage.setItem('channel_models', JSON.stringify(data));\n}\n\nexport function getChannelModels(type) {\n if (channelModels !== undefined && type in channelModels) {\n return channelModels[type];\n }\n let models = localStorage.getItem('channel_models');\n if (!models) {\n return [];\n }\n channelModels = JSON.parse(models);\n if (type in channelModels) {\n return channelModels[type];\n }\n return [];\n}\n\nexport function copy(text, name = '') {\n if (navigator.clipboard && navigator.clipboard.writeText) {\n navigator.clipboard.writeText(text).then(() => {\n showNotice(`复制${name}成功！`, true);\n }, () => {\n text = `复制${name}失败，请手动复制：, getSnackbarOptions('COPY'));\n });\n } else {\n const textArea = document.createElement(\"textarea\");\n textArea.value = text;\n document.body.appendChild(textArea);\n textArea.select();\n try {\n document.execCommand('copy');\n showNotice(`复制${name}成功！`, true);\n } catch (err) {\n text = `复制${name}失败，请手动复制：, getSnackbarOptions('COPY'));\n }\n document.body.removeChild(textArea);\n }\n}\n"
},
{
"path": "web/berry/src/utils/password-strength.js",
"content": "/**\n * Password validator for login pages\n */\nimport value from 'assets/scss/_themes-vars.module.scss';\n\n// has number\nconst hasNumber = (number) => new RegExp(/[0-9]/).test(number);\n\n// has mix of small and capitals\nconst hasMixed = (number) => new RegExp(/[a-z]/).test(number) && new RegExp(/[A-Z]/).test(number);\n\n// has special chars\nconst hasSpecial = (number) => new RegExp(/[!#@$%^&*)(+=._-]/).test(number);\n\n// set color based on password strength\nexport const strengthColor = (count) => {\n if (count < 2) return { label: 'Poor', color: value.errorMain };\n if (count < 3) return { label: 'Weak', color: value.warningDark };\n if (count < 4) return { label: 'Normal', color: value.orangeMain };\n if (count < 5) return { label: 'Good', color: value.successMain };\n if (count < 6) return { label: 'Strong', color: value.successDark };\n return { label: 'Poor', color: value.errorMain };\n};\n\n// password strength indicator\nexport const strengthIndicator = (number) => {\n let strengths = 0;\n if (number.length > 5) strengths += 1;\n if (number.length > 7) strengths += 1;\n if (hasNumber(number)) strengths += 1;\n if (hasSpecial(number)) strengths += 1;\n if (hasMixed(number)) strengths += 1;\n return strengths;\n};\n"
},
{
"path": "web/berry/src/utils/route-guard/AuthGuard.js",
"content": "import { useSelector } from 'react-redux';\nimport { useEffect, useContext } from 'react';\nimport { UserContext } from 'contexts/UserContext';\nimport { useNavigate } from 'react-router-dom';\n\nconst AuthGuard = ({ children }) => {\n const account = useSelector((state) => state.account);\n const { isUserLoaded } = useContext(UserContext);\n const navigate = useNavigate();\n useEffect(() => {\n if (isUserLoaded && !account.user) {\n navigate('/login');\n return;\n }\n }, [account, navigate, isUserLoaded]);\n\n return children;\n};\n\nexport default AuthGuard;\n"
},
{
"path": "web/berry/src/views/About/index.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { API } from 'utils/api';\nimport { showError } from 'utils/common';\nimport { marked } from 'marked';\nimport { Box, Container, Typography } from '@mui/material';\nimport MainCard from 'ui-component/cards/MainCard';\n\nconst About = () => {\n const [about, setAbout] = useState('');\n const [aboutLoaded, setAboutLoaded] = useState(false);\n\n const displayAbout = async () => {\n setAbout(localStorage.getItem('about') || '');\n const res = await API.get('/api/about');\n const { success, message, data } = res.data;\n if (success) {\n let aboutContent = data;\n if (!data.startsWith('https://')) {\n aboutContent = marked.parse(data);\n }\n setAbout(aboutContent);\n localStorage.setItem('about', aboutContent);\n } else {\n showError(message);\n setAbout('加载关于内容失败...');\n }\n setAboutLoaded(true);\n };\n\n useEffect(() => {\n displayAbout().then();\n }, []);\n\n return (\n <>\n {aboutLoaded && about === '' ? (\n <>\n \n \n \n \n 可在设置页面设置关于内容，支持 HTML & Markdown https://github.com/songquanpeng/one-api \n \n \n \n \n \n ) : (\n <>\n \n {about.startsWith('https://') ? (\n \n ) : (\n <>\n \n
\n \n \n )}\n \n \n )}\n \n );\n};\n\nexport default About;\n"
},
{
"path": "web/berry/src/views/Authentication/Auth/ForgetPassword.js",
"content": "import { Link } from 'react-router-dom';\n\n// material-ui\nimport { useTheme } from '@mui/material/styles';\nimport { Divider, Grid, Stack, Typography, useMediaQuery } from '@mui/material';\n\n// project imports\nimport AuthWrapper from '../AuthWrapper';\nimport AuthCardWrapper from '../AuthCardWrapper';\nimport ForgetPasswordForm from '../AuthForms/ForgetPasswordForm';\nimport Logo from 'ui-component/Logo';\n\n// assets\n\n// ================================|| AUTH3 - LOGIN ||================================ //\n\nconst ForgetPassword = () => {\n const theme = useTheme();\n const matchDownSM = useMediaQuery(theme.breakpoints.down('md'));\n\n return (\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n 密码重置\n \n \n \n \n \n \n \n \n \n \n \n \n 登录\n \n \n \n \n \n \n \n \n \n \n );\n};\n\nexport default ForgetPassword;\n"
},
{
"path": "web/berry/src/views/Authentication/Auth/GitHubOAuth.js",
"content": "import { Link, useNavigate, useSearchParams } from 'react-router-dom';\nimport React, { useEffect, useState } from 'react';\nimport { showError } from 'utils/common';\nimport useLogin from 'hooks/useLogin';\n\n// material-ui\nimport { useTheme } from '@mui/material/styles';\nimport { Grid, Stack, Typography, useMediaQuery, CircularProgress } from '@mui/material';\n\n// project imports\nimport AuthWrapper from '../AuthWrapper';\nimport AuthCardWrapper from '../AuthCardWrapper';\nimport Logo from 'ui-component/Logo';\n\n// assets\n\n// ================================|| AUTH3 - LOGIN ||================================ //\n\nconst GitHubOAuth = () => {\n const theme = useTheme();\n const matchDownSM = useMediaQuery(theme.breakpoints.down('md'));\n\n const [searchParams] = useSearchParams();\n const [prompt, setPrompt] = useState('处理中...');\n const { githubLogin } = useLogin();\n\n let navigate = useNavigate();\n\n const sendCode = async (code, state, count) => {\n const { success, message } = await githubLogin(code, state);\n if (!success) {\n if (message) {\n showError(message);\n }\n if (count === 0) {\n setPrompt(`操作失败，重定向至登录界面中...`);\n await new Promise((resolve) => setTimeout(resolve, 2000));\n navigate('/login');\n return;\n }\n count++;\n setPrompt(`出现错误，第 ${count} 次重试中...`);\n await new Promise((resolve) => setTimeout(resolve, 2000));\n await sendCode(code, state, count);\n }\n };\n\n useEffect(() => {\n let code = searchParams.get('code');\n let state = searchParams.get('state');\n sendCode(code, state, 0).then();\n }, []);\n\n return (\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n GitHub 登录\n \n \n \n \n \n \n \n {prompt}\n \n \n \n \n \n \n \n \n \n );\n};\n\nexport default GitHubOAuth;\n"
},
{
"path": "web/berry/src/views/Authentication/Auth/LarkOAuth.js",
"content": "import { Link, useNavigate, useSearchParams } from 'react-router-dom';\nimport React, { useEffect, useState } from 'react';\nimport { showError } from 'utils/common';\nimport useLogin from 'hooks/useLogin';\n\n// material-ui\nimport { useTheme } from '@mui/material/styles';\nimport { Grid, Stack, Typography, useMediaQuery, CircularProgress } from '@mui/material';\n\n// project imports\nimport AuthWrapper from '../AuthWrapper';\nimport AuthCardWrapper from '../AuthCardWrapper';\nimport Logo from 'ui-component/Logo';\n\n// assets\n\n// ================================|| AUTH3 - LOGIN ||================================ //\n\nconst LarkOAuth = () => {\n const theme = useTheme();\n const matchDownSM = useMediaQuery(theme.breakpoints.down('md'));\n\n const [searchParams] = useSearchParams();\n const [prompt, setPrompt] = useState('处理中...');\n const { larkLogin } = useLogin();\n\n let navigate = useNavigate();\n\n const sendCode = async (code, state, count) => {\n const { success, message } = await larkLogin(code, state);\n if (!success) {\n if (message) {\n showError(message);\n }\n if (count === 0) {\n setPrompt(`操作失败，重定向至登录界面中...`);\n await new Promise((resolve) => setTimeout(resolve, 2000));\n navigate('/login');\n return;\n }\n count++;\n setPrompt(`出现错误，第 ${count} 次重试中...`);\n await new Promise((resolve) => setTimeout(resolve, 2000));\n await sendCode(code, state, count);\n }\n };\n\n useEffect(() => {\n let code = searchParams.get('code');\n let state = searchParams.get('state');\n sendCode(code, state, 0).then();\n }, []);\n\n return (\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n 飞书 登录\n \n \n \n \n \n \n \n {prompt}\n \n \n \n \n \n \n \n \n \n );\n};\n\nexport default LarkOAuth;\n"
},
{
"path": "web/berry/src/views/Authentication/Auth/Login.js",
"content": "import { Link } from 'react-router-dom';\n\n// material-ui\nimport { useTheme } from '@mui/material/styles';\nimport { Divider, Grid, Stack, Typography, useMediaQuery } from '@mui/material';\n\n// project imports\nimport AuthWrapper from '../AuthWrapper';\nimport AuthCardWrapper from '../AuthCardWrapper';\nimport AuthLogin from '../AuthForms/AuthLogin';\nimport Logo from 'ui-component/Logo';\n\n// ================================|| AUTH3 - LOGIN ||================================ //\n\nconst Login = () => {\n const theme = useTheme();\n const matchDownSM = useMediaQuery(theme.breakpoints.down('md'));\n\n return (\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n 登录\n \n \n \n \n \n \n \n \n \n \n \n \n 注册\n \n \n \n \n \n \n \n \n \n \n );\n};\n\nexport default Login;\n"
},
{
"path": "web/berry/src/views/Authentication/Auth/OidcOAuth.js",
"content": "import { Link, useNavigate, useSearchParams } from 'react-router-dom';\nimport React, { useEffect, useState } from 'react';\nimport { showError } from 'utils/common';\nimport useLogin from 'hooks/useLogin';\n\n// material-ui\nimport { useTheme } from '@mui/material/styles';\nimport { Grid, Stack, Typography, useMediaQuery, CircularProgress } from '@mui/material';\n\n// project imports\nimport AuthWrapper from '../AuthWrapper';\nimport AuthCardWrapper from '../AuthCardWrapper';\nimport Logo from 'ui-component/Logo';\n\n// assets\n\n// ================================|| AUTH3 - LOGIN ||================================ //\n\nconst OidcOAuth = () => {\n const theme = useTheme();\n const matchDownSM = useMediaQuery(theme.breakpoints.down('md'));\n\n const [searchParams] = useSearchParams();\n const [prompt, setPrompt] = useState('处理中...');\n const { oidcLogin } = useLogin();\n\n let navigate = useNavigate();\n\n const sendCode = async (code, state, count) => {\n const { success, message } = await oidcLogin(code, state);\n if (!success) {\n if (message) {\n showError(message);\n }\n if (count === 0) {\n setPrompt(`操作失败，重定向至登录界面中...`);\n await new Promise((resolve) => setTimeout(resolve, 2000));\n navigate('/login');\n return;\n }\n count++;\n setPrompt(`出现错误，第 ${count} 次重试中...`);\n await new Promise((resolve) => setTimeout(resolve, 2000));\n await sendCode(code, state, count);\n }\n };\n\n useEffect(() => {\n let code = searchParams.get('code');\n let state = searchParams.get('state');\n sendCode(code, state, 0).then();\n }, []);\n\n return (\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n OIDC 登录\n \n \n \n \n \n \n \n {prompt}\n \n \n \n \n \n \n \n \n \n );\n};\n\nexport default OidcOAuth;\n"
},
{
"path": "web/berry/src/views/Authentication/Auth/Register.js",
"content": "import { Link } from 'react-router-dom';\n\n// material-ui\nimport { useTheme } from '@mui/material/styles';\nimport { Divider, Grid, Stack, Typography, useMediaQuery } from '@mui/material';\n\n// project imports\nimport AuthWrapper from '../AuthWrapper';\nimport AuthCardWrapper from '../AuthCardWrapper';\nimport Logo from 'ui-component/Logo';\nimport AuthRegister from '../AuthForms/AuthRegister';\n\n// assets\n\n// ===============================|| AUTH3 - REGISTER ||=============================== //\n\nconst Register = () => {\n const theme = useTheme();\n const matchDownSM = useMediaQuery(theme.breakpoints.down('md'));\n\n return (\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n 注册\n \n \n \n \n \n \n \n \n \n \n \n \n 已经有帐号了？点击登录\n \n \n \n \n \n \n \n \n {/* \n */}\n \n \n );\n};\n\nexport default Register;\n"
},
{
"path": "web/berry/src/views/Authentication/Auth/ResetPassword.js",
"content": "import { Link } from 'react-router-dom';\n\n// material-ui\nimport { useTheme } from '@mui/material/styles';\nimport { Divider, Grid, Stack, Typography, useMediaQuery } from '@mui/material';\n\n// project imports\nimport AuthWrapper from '../AuthWrapper';\nimport AuthCardWrapper from '../AuthCardWrapper';\nimport ResetPasswordForm from '../AuthForms/ResetPasswordForm';\nimport Logo from 'ui-component/Logo';\n\n// ================================|| AUTH3 - LOGIN ||================================ //\n\nconst ResetPassword = () => {\n const theme = useTheme();\n const matchDownSM = useMediaQuery(theme.breakpoints.down('md'));\n\n return (\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n 密码重置确认\n \n \n \n \n \n \n \n \n \n \n \n \n 登录\n \n \n \n \n \n \n \n \n \n \n );\n};\n\nexport default ResetPassword;\n"
},
{
"path": "web/berry/src/views/Authentication/AuthCardWrapper.js",
"content": "import PropTypes from 'prop-types';\n\n// material-ui\nimport { Box } from '@mui/material';\n\n// project import\nimport MainCard from 'ui-component/cards/MainCard';\n\n// ==============================|| AUTHENTICATION CARD WRAPPER ||============================== //\n\nconst AuthCardWrapper = ({ children, ...other }) => (\n *': {\n flexGrow: 1,\n flexBasis: '50%'\n }\n }}\n content={false}\n {...other}\n >\n {children} \n \n);\n\nAuthCardWrapper.propTypes = {\n children: PropTypes.node\n};\n\nexport default AuthCardWrapper;\n"
},
{
"path": "web/berry/src/views/Authentication/AuthForms/AuthLogin.js",
"content": "import { useState } from 'react';\nimport { useSelector } from 'react-redux';\nimport { Link } from 'react-router-dom';\n\n// material-ui\nimport { useTheme } from '@mui/material/styles';\nimport {\n Box,\n Button,\n Divider,\n FormControl,\n FormHelperText,\n Grid,\n IconButton,\n InputAdornment,\n InputLabel,\n OutlinedInput,\n Stack,\n Typography,\n useMediaQuery\n} from '@mui/material';\n\n// third party\nimport * as Yup from 'yup';\nimport { Formik } from 'formik';\n\n// project imports\nimport useLogin from 'hooks/useLogin';\nimport AnimateButton from 'ui-component/extended/AnimateButton';\nimport WechatModal from 'views/Authentication/AuthForms/WechatModal';\n\n// assets\nimport Visibility from '@mui/icons-material/Visibility';\nimport VisibilityOff from '@mui/icons-material/VisibilityOff';\n\nimport Github from 'assets/images/icons/github.svg';\nimport Wechat from 'assets/images/icons/wechat.svg';\nimport Lark from 'assets/images/icons/lark.svg';\nimport OIDC from 'assets/images/icons/oidc.svg';\nimport { onGitHubOAuthClicked, onLarkOAuthClicked, onOidcClicked } from 'utils/common';\n\n// ============================|| FIREBASE - LOGIN ||============================ //\n\nconst LoginForm = ({ ...others }) => {\n const theme = useTheme();\n const { login, wechatLogin } = useLogin();\n const [openWechat, setOpenWechat] = useState(false);\n const matchDownSM = useMediaQuery(theme.breakpoints.down('md'));\n const customization = useSelector((state) => state.customization);\n const siteInfo = useSelector((state) => state.siteInfo);\n // const [checked, setChecked] = useState(true);\n\n let tripartiteLogin = false;\n if (siteInfo.github_oauth || siteInfo.wechat_login || siteInfo.lark_client_id || siteInfo.oidc) {\n tripartiteLogin = true;\n }\n\n const handleWechatOpen = () => {\n setOpenWechat(true);\n };\n\n const handleWechatClose = () => {\n setOpenWechat(false);\n };\n\n const [showPassword, setShowPassword] = useState(false);\n const handleClickShowPassword = () => {\n setShowPassword(!showPassword);\n };\n\n const handleMouseDownPassword = (event) => {\n event.preventDefault();\n };\n\n return (\n <>\n {tripartiteLogin && (\n \n {siteInfo.github_oauth && (\n \n \n onGitHubOAuthClicked(siteInfo.github_client_id)}\n size=\"large\"\n variant=\"outlined\"\n sx={{\n color: 'grey.700',\n backgroundColor: theme.palette.grey[50],\n borderColor: theme.palette.grey[100]\n }}\n >\n \n \n 使用 GitHub 登录\n \n \n \n )}\n {siteInfo.wechat_login && (\n \n \n \n \n \n 使用微信登录\n \n \n \n )}\n {siteInfo.lark_client_id && (\n \n \n onLarkOAuthClicked(siteInfo.lark_client_id)}\n size=\"large\"\n variant=\"outlined\"\n sx={{\n color: 'grey.700',\n backgroundColor: theme.palette.grey[50],\n borderColor: theme.palette.grey[100]\n }}\n >\n \n \n 使用飞书登录\n \n \n \n )}\n {siteInfo.oidc && (\n \n \n onOidcClicked(siteInfo.oidc_authorization_endpoint,siteInfo.oidc_client_id)}\n size=\"large\"\n variant=\"outlined\"\n sx={{\n color: 'grey.700',\n backgroundColor: theme.palette.grey[50],\n borderColor: theme.palette.grey[100]\n }}\n >\n \n \n 使用 OIDC 登录\n \n \n \n )}\n \n \n \n OR\n \n\n \n \n )}\n\n {\n const { success, message } = await login(values.username, values.password);\n if (success) {\n setStatus({ success: true });\n } else {\n setStatus({ success: false });\n if (message) {\n setErrors({ submit: message });\n }\n }\n setSubmitting(false);\n }}\n >\n {({ errors, handleBlur, handleChange, handleSubmit, isSubmitting, touched, values }) => (\n \n )}\n \n \n );\n};\n\nexport default LoginForm;\n"
},
{
"path": "web/berry/src/views/Authentication/AuthForms/AuthRegister.js",
"content": "import { useState, useEffect } from 'react';\nimport { useSelector } from 'react-redux';\nimport useRegister from 'hooks/useRegister';\nimport Turnstile from 'react-turnstile';\nimport { useSearchParams } from 'react-router-dom';\n// import { useSelector } from 'react-redux';\n\n// material-ui\nimport { useTheme } from '@mui/material/styles';\nimport {\n Box,\n Button,\n FormControl,\n FormHelperText,\n Grid,\n IconButton,\n InputAdornment,\n InputLabel,\n OutlinedInput,\n Typography\n} from '@mui/material';\n\n// third party\nimport * as Yup from 'yup';\nimport { Formik } from 'formik';\n\n// project imports\nimport AnimateButton from 'ui-component/extended/AnimateButton';\nimport { strengthColor, strengthIndicator } from 'utils/password-strength';\n\n// assets\nimport Visibility from '@mui/icons-material/Visibility';\nimport VisibilityOff from '@mui/icons-material/VisibilityOff';\nimport { showError, showInfo } from 'utils/common';\n\n// ===========================|| FIREBASE - REGISTER ||=========================== //\n\nconst RegisterForm = ({ ...others }) => {\n const theme = useTheme();\n const { register, sendVerificationCode } = useRegister();\n const siteInfo = useSelector((state) => state.siteInfo);\n const [showPassword, setShowPassword] = useState(false);\n const [searchParams] = useSearchParams();\n\n const [showEmailVerification, setShowEmailVerification] = useState(false);\n const [turnstileEnabled, setTurnstileEnabled] = useState(false);\n const [turnstileSiteKey, setTurnstileSiteKey] = useState('');\n const [turnstileToken, setTurnstileToken] = useState('');\n\n const [strength, setStrength] = useState(0);\n const [level, setLevel] = useState();\n\n const handleClickShowPassword = () => {\n setShowPassword(!showPassword);\n };\n\n const handleMouseDownPassword = (event) => {\n event.preventDefault();\n };\n\n const changePassword = (value) => {\n const temp = strengthIndicator(value);\n setStrength(temp);\n setLevel(strengthColor(temp));\n };\n\n const handleSendCode = async (email) => {\n if (email === '') {\n showError('请输入邮箱');\n return;\n }\n if (turnstileEnabled && turnstileToken === '') {\n showError('请稍后几秒重试，Turnstile 正在检查用户环境！');\n return;\n }\n\n const { success, message } = await sendVerificationCode(email, turnstileToken);\n if (!success) {\n showError(message);\n return;\n }\n };\n\n useEffect(() => {\n let affCode = searchParams.get('aff');\n if (affCode) {\n localStorage.setItem('aff', affCode);\n }\n\n setShowEmailVerification(siteInfo.email_verification);\n if (siteInfo.turnstile_check) {\n setTurnstileEnabled(true);\n setTurnstileSiteKey(siteInfo.turnstile_site_key);\n }\n }, [siteInfo]);\n\n return (\n <>\n {\n if (turnstileEnabled && turnstileToken === '') {\n showInfo('请稍后几秒重试，Turnstile 正在检查用户环境！');\n setSubmitting(false);\n return;\n }\n\n const { success, message } = await register(values, turnstileToken);\n if (success) {\n setStatus({ success: true });\n } else {\n setStatus({ success: false });\n if (message) {\n setErrors({ submit: message });\n }\n }\n }}\n >\n {({ errors, handleBlur, handleChange, handleSubmit, isSubmitting, touched, values }) => (\n \n )}\n \n \n );\n};\n\nexport default RegisterForm;\n"
},
{
"path": "web/berry/src/views/Authentication/AuthForms/ForgetPasswordForm.js",
"content": "import { useState, useEffect } from \"react\";\nimport { useSelector } from \"react-redux\";\nimport Turnstile from \"react-turnstile\";\nimport { API } from \"utils/api\";\n\n// material-ui\nimport { useTheme } from \"@mui/material/styles\";\nimport {\n Box,\n Button,\n FormControl,\n FormHelperText,\n InputLabel,\n OutlinedInput,\n Typography,\n} from \"@mui/material\";\n\n// third party\nimport * as Yup from \"yup\";\nimport { Formik } from \"formik\";\n\n// project imports\nimport AnimateButton from \"ui-component/extended/AnimateButton\";\n\n// assets\nimport { showError, showInfo, showSuccess } from \"utils/common\";\n\n// ===========================|| FIREBASE - REGISTER ||=========================== //\n\nconst ForgetPasswordForm = ({ ...others }) => {\n const theme = useTheme();\n const siteInfo = useSelector((state) => state.siteInfo);\n\n const [sendEmail, setSendEmail] = useState(false);\n const [turnstileEnabled, setTurnstileEnabled] = useState(false);\n const [turnstileSiteKey, setTurnstileSiteKey] = useState(\"\");\n const [turnstileToken, setTurnstileToken] = useState(\"\");\n const [disableButton, setDisableButton] = useState(false);\n const [countdown, setCountdown] = useState(30);\n\n const submit = async (values, { setSubmitting }) => {\n setDisableButton(true);\n setSubmitting(true);\n if (turnstileEnabled && turnstileToken === \"\") {\n showInfo(\"请稍后几秒重试，Turnstile 正在检查用户环境！\");\n setSubmitting(false);\n return;\n }\n const res = await API.get(\n `/api/reset_password?email=${values.email}&turnstile=${turnstileToken}`\n );\n const { success, message } = res.data;\n if (success) {\n showSuccess(\"重置邮件发送成功，请检查邮箱！\");\n setSendEmail(true);\n } else {\n showError(message);\n setDisableButton(false);\n setCountdown(30);\n }\n setSubmitting(false);\n };\n\n useEffect(() => {\n let countdownInterval = null;\n if (disableButton && countdown > 0) {\n countdownInterval = setInterval(() => {\n setCountdown(countdown - 1);\n }, 1000);\n } else if (countdown === 0) {\n setDisableButton(false);\n setCountdown(30);\n }\n return () => clearInterval(countdownInterval);\n }, [disableButton, countdown]);\n\n useEffect(() => {\n if (siteInfo.turnstile_check) {\n setTurnstileEnabled(true);\n setTurnstileSiteKey(siteInfo.turnstile_site_key);\n }\n }, [siteInfo]);\n\n return (\n <>\n {sendEmail ? (\n \n 重置邮件发送成功，请检查邮箱！\n \n ) : (\n \n {({\n errors,\n handleBlur,\n handleChange,\n handleSubmit,\n isSubmitting,\n touched,\n values,\n }) => (\n \n )}\n \n )}\n \n );\n};\n\nexport default ForgetPasswordForm;\n"
},
{
"path": "web/berry/src/views/Authentication/AuthForms/ResetPasswordForm.js",
"content": "import { useState, useEffect } from 'react';\nimport { useSearchParams } from 'react-router-dom';\n\n// material-ui\nimport { Button, Stack, Typography, Alert } from '@mui/material';\n\n// assets\nimport { showError, copy } from 'utils/common';\nimport { API } from 'utils/api';\n\n// ===========================|| FIREBASE - REGISTER ||=========================== //\n\nconst ResetPasswordForm = () => {\n const [searchParams] = useSearchParams();\n const [inputs, setInputs] = useState({\n email: '',\n token: ''\n });\n const [newPassword, setNewPassword] = useState('');\n\n const submit = async () => {\n const res = await API.post(`/api/user/reset`, inputs);\n const { success, message } = res.data;\n if (success) {\n let password = res.data.data;\n setNewPassword(password);\n copy(password, '新密码');\n } else {\n showError(message);\n }\n };\n\n useEffect(() => {\n let email = searchParams.get('email');\n let token = searchParams.get('token');\n setInputs({\n token,\n email\n });\n }, []);\n\n return (\n \n {!inputs.email || !inputs.token ? (\n \n 无效的链接\n \n ) : newPassword ? (\n \n 你的新密码是: {newPassword} \n ) : (\n \n 点击重置密码\n \n )}\n \n );\n};\n\nexport default ResetPasswordForm;\n"
},
{
"path": "web/berry/src/views/Authentication/AuthForms/WechatModal.js",
"content": "// WechatModal.js\nimport PropTypes from 'prop-types';\nimport React from 'react';\nimport { Dialog, DialogTitle, DialogContent, TextField, Button, Typography, Grid } from '@mui/material';\nimport { Formik, Form, Field } from 'formik';\nimport { showError } from 'utils/common';\nimport * as Yup from 'yup';\n\nconst validationSchema = Yup.object().shape({\n code: Yup.string().required('验证码不能为空')\n});\n\nconst WechatModal = ({ open, handleClose, wechatLogin, qrCode }) => {\n const handleSubmit = (values) => {\n const { success, message } = wechatLogin(values.code);\n if (success) {\n handleClose();\n } else {\n showError(message || '未知错误');\n }\n };\n\n return (\n \n 微信验证码登录 \n \n \n \n 请使用微信扫描二维码关注公众号，输入「验证码」获取验证码（三分钟内有效）\n \n \n {({ errors, touched }) => (\n \n )}\n \n \n \n \n );\n};\n\nexport default WechatModal;\n\nWechatModal.propTypes = {\n open: PropTypes.bool,\n handleClose: PropTypes.func,\n wechatLogin: PropTypes.func,\n qrCode: PropTypes.string\n};\n"
},
{
"path": "web/berry/src/views/Authentication/AuthWrapper.js",
"content": "// material-ui\nimport { styled } from '@mui/material/styles';\nimport { useSelector } from 'react-redux';\nimport { useNavigate } from 'react-router';\nimport { useEffect, useContext } from 'react';\nimport { UserContext } from 'contexts/UserContext';\n\n// ==============================|| AUTHENTICATION 1 WRAPPER ||============================== //\n\nconst AuthStyle = styled('div')(({ theme }) => ({\n backgroundColor: theme.palette.background.default\n}));\n\n// eslint-disable-next-line\nconst AuthWrapper = ({ children }) => {\n const account = useSelector((state) => state.account);\n const { isUserLoaded } = useContext(UserContext);\n const navigate = useNavigate();\n useEffect(() => {\n if (isUserLoaded && account.user) {\n navigate('/panel');\n }\n }, [account, navigate, isUserLoaded]);\n\n return {children} ;\n};\n\nexport default AuthWrapper;\n"
},
{
"path": "web/berry/src/views/Channel/component/EditModal.js",
"content": "import PropTypes from 'prop-types';\nimport { useState, useEffect } from 'react';\nimport { CHANNEL_OPTIONS } from 'constants/ChannelConstants';\nimport { useTheme } from '@mui/material/styles';\nimport { API } from 'utils/api';\nimport { showError, showSuccess, getChannelModels } from 'utils/common';\nimport {\n Dialog,\n DialogTitle,\n DialogContent,\n DialogActions,\n TextField,\n Button,\n Divider,\n Select,\n MenuItem,\n FormControl,\n InputLabel,\n OutlinedInput,\n ButtonGroup,\n Container,\n Autocomplete,\n FormHelperText,\n Switch,\n Checkbox\n} from '@mui/material';\n\nimport { Formik } from 'formik';\nimport * as Yup from 'yup';\nimport { defaultConfig, typeConfig } from '../type/Config'; //typeConfig\nimport { createFilterOptions } from '@mui/material/Autocomplete';\nimport CheckBoxOutlineBlankIcon from '@mui/icons-material/CheckBoxOutlineBlank';\nimport CheckBoxIcon from '@mui/icons-material/CheckBox';\n\nconst icon = \n \n {channelId ? '编辑渠道' : '新建渠道'}\n \n \n \n {({ errors, handleBlur, handleChange, handleSubmit, isSubmitting, touched, values, setFieldValue }) => (\n \n )}\n \n \n \n );\n};\n\nexport default EditModal;\n\nEditModal.propTypes = {\n open: PropTypes.bool,\n channelId: PropTypes.number,\n onCancel: PropTypes.func,\n onOk: PropTypes.func\n};\n"
},
{
"path": "web/berry/src/views/Channel/component/GroupLabel.js",
"content": "import PropTypes from \"prop-types\";\nimport Label from \"ui-component/Label\";\nimport Stack from \"@mui/material/Stack\";\nimport Divider from \"@mui/material/Divider\";\n\nfunction name2color(name) {\n switch (name) {\n case \"default\":\n return \"info\";\n case \"vip\":\n return \"warning\"\n case \"svip\":\n return \"error\"\n default:\n return \"info\"\n }\n}\n\nconst GroupLabel = ({ group }) => {\n let groups = [];\n if (group === \"\") {\n groups = [\"default\"];\n } else {\n groups = group.split(\",\");\n groups.sort();\n }\n return (\n {group} ;\n })}\n \n );\n};\n\nGroupLabel.propTypes = {\n group: PropTypes.string,\n};\n\nexport default GroupLabel;\n"
},
{
"path": "web/berry/src/views/Channel/component/NameLabel.js",
"content": "import PropTypes from 'prop-types';\nimport { Tooltip, Stack, Container } from '@mui/material';\nimport Label from 'ui-component/Label';\nimport { styled } from '@mui/material/styles';\nimport { showSuccess, copy } from 'utils/common';\n\nconst TooltipContainer = styled(Container)({\n maxHeight: '250px',\n overflow: 'auto',\n '&::-webkit-scrollbar': {\n width: '0px' // Set the width to 0 to hide the scrollbar\n }\n});\n\nconst NameLabel = ({ name, models }) => {\n let modelMap = [];\n modelMap = models.split(',');\n modelMap.sort();\n\n return (\n \n \n {modelMap.map((item, index) => {\n return (\n {\n copy(item, '模型名称');\n }}\n >\n {item}\n \n );\n })}\n \n \n }\n placement=\"top\"\n >\n {name} \n \n );\n};\n\nNameLabel.propTypes = {\n name: PropTypes.string,\n models: PropTypes.string\n};\n\nexport default NameLabel;\n"
},
{
"path": "web/berry/src/views/Channel/component/ResponseTimeLabel.js",
"content": "import PropTypes from 'prop-types';\nimport Label from 'ui-component/Label';\nimport Tooltip from '@mui/material/Tooltip';\nimport { timestamp2string } from 'utils/common';\n\nconst ResponseTimeLabel = ({ test_time, response_time, handle_action }) => {\n let color = 'default';\n let time = response_time / 1000;\n time = time.toFixed(2) + ' 秒';\n\n if (response_time === 0) {\n color = 'default';\n } else if (response_time <= 1000) {\n color = 'success';\n } else if (response_time <= 3000) {\n color = 'primary';\n } else if (response_time <= 5000) {\n color = 'secondary';\n } else {\n color = 'error';\n }\n let title = (\n <>\n 点击测速\n \n {response_time == 0 ? '未测试' : time} \n \n );\n};\n\nResponseTimeLabel.propTypes = {\n test_time: PropTypes.number,\n response_time: PropTypes.number,\n handle_action: PropTypes.func\n};\n\nexport default ResponseTimeLabel;\n"
},
{
"path": "web/berry/src/views/Channel/component/TableHead.js",
"content": "import { TableCell, TableHead, TableRow } from '@mui/material';\n\nconst ChannelTableHead = () => {\n return (\n \n \n ID \n 名称 \n 分组 \n 类型 \n 状态 \n 响应时间 \n 已消耗 \n 余额 \n 优先级 \n 操作 \n \n \n );\n};\n\nexport default ChannelTableHead;\n"
},
{
"path": "web/berry/src/views/Channel/component/TableRow.js",
"content": "import PropTypes from \"prop-types\";\nimport { useState } from \"react\";\n\nimport { showInfo, showError, renderNumber } from \"utils/common\";\nimport { API } from \"utils/api\";\nimport { CHANNEL_OPTIONS } from \"constants/ChannelConstants\";\n\nimport {\n Popover,\n TableRow,\n MenuItem,\n TableCell,\n IconButton,\n TextField,\n Dialog,\n DialogActions,\n DialogContent,\n DialogContentText,\n DialogTitle,\n Tooltip,\n Button,\n} from \"@mui/material\";\n\nimport Label from \"ui-component/Label\";\nimport TableSwitch from \"ui-component/Switch\";\n\nimport ResponseTimeLabel from \"./ResponseTimeLabel\";\nimport GroupLabel from \"./GroupLabel\";\nimport NameLabel from \"./NameLabel\";\n\nimport { IconDotsVertical, IconEdit, IconTrash } from \"@tabler/icons-react\";\n\nexport default function ChannelTableRow({\n item,\n manageChannel,\n handleOpenModal,\n setModalChannelId,\n}) {\n const [open, setOpen] = useState(null);\n const [openDelete, setOpenDelete] = useState(false);\n const [statusSwitch, setStatusSwitch] = useState(item.status);\n const [priorityValve, setPriority] = useState(item.priority);\n const [responseTimeData, setResponseTimeData] = useState({\n test_time: item.test_time,\n response_time: item.response_time,\n });\n const [itemBalance, setItemBalance] = useState(item.balance);\n\n const handleDeleteOpen = () => {\n handleCloseMenu();\n setOpenDelete(true);\n };\n\n const handleDeleteClose = () => {\n setOpenDelete(false);\n };\n\n const handleOpenMenu = (event) => {\n setOpen(event.currentTarget);\n };\n\n const handleCloseMenu = () => {\n setOpen(null);\n };\n\n const handleStatus = async () => {\n const switchVlue = statusSwitch === 1 ? 2 : 1;\n const { success } = await manageChannel(item.id, \"status\", switchVlue);\n if (success) {\n setStatusSwitch(switchVlue);\n }\n };\n\n const handlePriority = async (event) => {\n const currentValue = parseInt(event.target.value);\n if (isNaN(currentValue) || currentValue === priorityValve) {\n return;\n }\n\n if (currentValue < 0) {\n showError(\"优先级不能小于 0\");\n return;\n }\n\n await manageChannel(item.id, \"priority\", currentValue);\n setPriority(currentValue);\n };\n\n const handleResponseTime = async () => {\n const { success, time } = await manageChannel(item.id, \"test\", \"\");\n if (success) {\n setResponseTimeData({\n test_time: Date.now() / 1000,\n response_time: time * 1000,\n });\n showInfo(`渠道 ${item.name} 测试成功，耗时 ${time.toFixed(2)} 秒。`);\n }\n };\n\n const updateChannelBalance = async () => {\n const res = await API.get(`/api/channel/update_balance/${item.id}`);\n const { success, message, balance } = res.data;\n if (success) {\n setItemBalance(balance);\n\n showInfo(`余额更新成功！`);\n } else {\n showError(message);\n }\n };\n\n const handleDelete = async () => {\n handleCloseMenu();\n await manageChannel(item.id, \"delete\", \"\");\n };\n\n return (\n <>\n \n {item.id} \n\n \n \n\n \n \n\n \n {!CHANNEL_OPTIONS[item.type] ? (\n \n 未知\n \n ) : (\n \n {CHANNEL_OPTIONS[item.type].text}\n \n )}\n \n\n \n {\n switch (statusSwitch) {\n case 1:\n return \"已启用\";\n case 2:\n return \"本渠道被手动禁用\";\n case 3:\n return \"本渠道被程序自动禁用\";\n default:\n return \"未知\";\n }\n })()}\n placement=\"top\"\n >\n \n\n \n \n {renderNumber(item.used_quota)} \n \n \n {renderBalance(item.type, itemBalance)}\n \n \n \n \n\n \n \n \n \n\n \n {\n handleCloseMenu();\n handleOpenModal();\n setModalChannelId(item.id);\n }}\n >\n \n \n \n\n \n 删除渠道 \n \n 是否删除渠道 {item.name}？ \n \n \n 关闭 \n \n 删除\n \n \n \n \n );\n}\n\nChannelTableRow.propTypes = {\n item: PropTypes.object,\n manageChannel: PropTypes.func,\n handleOpenModal: PropTypes.func,\n setModalChannelId: PropTypes.func,\n};\n\nfunction renderBalance(type, balance) {\n switch (type) {\n case 1: // OpenAI\n return ${balance.toFixed(2)} ;\n case 4: // CloseAI\n return ¥{balance.toFixed(2)} ;\n case 8: // 自定义\n return ${balance.toFixed(2)} ;\n case 5: // OpenAI-SB\n return ¥{(balance / 10000).toFixed(2)} ;\n case 10: // AI Proxy\n return {renderNumber(balance)} ;\n case 12: // API2GPT\n return ¥{balance.toFixed(2)} ;\n case 13: // AIGC2D\n return {renderNumber(balance)} ;\n case 36: // DeepSeek\n return ¥{balance.toFixed(2)} ;\n case 44: // SiliconFlow\n return ¥{balance.toFixed(2)} ;\n default:\n return 不支持 ;\n }\n}\n"
},
{
"path": "web/berry/src/views/Channel/index.js",
"content": "import { useState, useEffect } from 'react';\nimport { showError, showSuccess, showInfo, loadChannelModels } from 'utils/common';\n\nimport { useTheme } from '@mui/material/styles';\nimport Table from '@mui/material/Table';\nimport TableBody from '@mui/material/TableBody';\nimport TableContainer from '@mui/material/TableContainer';\nimport PerfectScrollbar from 'react-perfect-scrollbar';\nimport TablePagination from '@mui/material/TablePagination';\nimport LinearProgress from '@mui/material/LinearProgress';\nimport ButtonGroup from '@mui/material/ButtonGroup';\nimport Toolbar from '@mui/material/Toolbar';\nimport useMediaQuery from '@mui/material/useMediaQuery';\n\nimport { Button, IconButton, Card, Box, Stack, Container, Typography, Divider } from '@mui/material';\nimport ChannelTableRow from './component/TableRow';\nimport ChannelTableHead from './component/TableHead';\nimport TableToolBar from 'ui-component/TableToolBar';\nimport { API } from 'utils/api';\nimport { ITEMS_PER_PAGE } from 'constants';\nimport { IconRefresh, IconHttpDelete, IconPlus, IconBrandSpeedtest, IconCoinYuan } from '@tabler/icons-react';\nimport EditeModal from './component/EditModal';\n\n// ----------------------------------------------------------------------\n// CHANNEL_OPTIONS,\nexport default function ChannelPage() {\n const [channels, setChannels] = useState([]);\n const [activePage, setActivePage] = useState(0);\n const [searching, setSearching] = useState(false);\n const [searchKeyword, setSearchKeyword] = useState('');\n const theme = useTheme();\n const matchUpMd = useMediaQuery(theme.breakpoints.up('sm'));\n const [openModal, setOpenModal] = useState(false);\n const [editChannelId, setEditChannelId] = useState(0);\n\n const loadChannels = async (startIdx) => {\n setSearching(true);\n const res = await API.get(`/api/channel/?p=${startIdx}`);\n const { success, message, data } = res.data;\n if (success) {\n if (startIdx === 0) {\n setChannels(data);\n } else {\n let newChannels = [...channels];\n newChannels.splice(startIdx * ITEMS_PER_PAGE, data.length, ...data);\n setChannels(newChannels);\n }\n } else {\n showError(message);\n }\n setSearching(false);\n };\n\n const onPaginationChange = (event, activePage) => {\n (async () => {\n if (activePage === Math.ceil(channels.length / ITEMS_PER_PAGE)) {\n // In this case we have to load more data and then append them.\n await loadChannels(activePage);\n }\n setActivePage(activePage);\n })();\n };\n\n const searchChannels = async (event) => {\n event.preventDefault();\n if (searchKeyword === '') {\n await loadChannels(0);\n setActivePage(0);\n return;\n }\n setSearching(true);\n const res = await API.get(`/api/channel/search?keyword=${searchKeyword}`);\n const { success, message, data } = res.data;\n if (success) {\n setChannels(data);\n setActivePage(0);\n } else {\n showError(message);\n }\n setSearching(false);\n };\n\n const handleSearchKeyword = (event) => {\n setSearchKeyword(event.target.value);\n };\n\n const manageChannel = async (id, action, value) => {\n const url = '/api/channel/';\n let data = { id };\n let res;\n switch (action) {\n case 'delete':\n res = await API.delete(url + id);\n break;\n case 'status':\n res = await API.put(url, {\n ...data,\n status: value\n });\n break;\n case 'priority':\n if (value === '') {\n return;\n }\n res = await API.put(url, {\n ...data,\n priority: parseInt(value)\n });\n break;\n case 'test':\n res = await API.get(url + `test/${id}`);\n break;\n }\n const { success, message } = res.data;\n if (success) {\n showSuccess('操作成功完成！');\n if (action === 'delete') {\n await handleRefresh();\n }\n } else {\n showError(message);\n }\n\n return res.data;\n };\n\n // 处理刷新\n const handleRefresh = async () => {\n await loadChannels(activePage);\n };\n\n // 处理测试所有启用渠道\n const testAllChannels = async () => {\n const res = await API.get(`/api/channel/test`);\n const { success, message } = res.data;\n if (success) {\n showInfo('已成功开始测试所有渠道，请刷新页面查看结果。');\n } else {\n showError(message);\n }\n };\n\n // 处理删除所有禁用渠道\n const deleteAllDisabledChannels = async () => {\n const res = await API.delete(`/api/channel/disabled`);\n const { success, message, data } = res.data;\n if (success) {\n showSuccess(`已删除所有禁用渠道，共计 ${data} 个`);\n await handleRefresh();\n } else {\n showError(message);\n }\n };\n\n // 处理更新所有启用渠道余额\n const updateAllChannelsBalance = async () => {\n setSearching(true);\n const res = await API.get(`/api/channel/update_balance`);\n const { success, message } = res.data;\n if (success) {\n showInfo('已更新完毕所有已启用渠道余额！');\n } else {\n showError(message);\n }\n setSearching(false);\n };\n\n const handleOpenModal = (channelId) => {\n setEditChannelId(channelId);\n setOpenModal(true);\n };\n\n const handleCloseModal = () => {\n setOpenModal(false);\n setEditChannelId(0);\n };\n\n const handleOkModal = (status) => {\n if (status === true) {\n handleCloseModal();\n handleRefresh();\n }\n };\n\n useEffect(() => {\n loadChannels(0)\n .then()\n .catch((reason) => {\n showError(reason);\n });\n loadChannelModels().then();\n }, []);\n\n return (\n <>\n \n 渠道 \n \n \n \n \n theme.spacing(0, 1, 0, 3)\n }}\n >\n \n {matchUpMd ? (\n \n \n ) : (\n \n \n \n \n \n \n \n \n \n )}\n \n \n {searching && \n \n \n \n {channels.slice(activePage * ITEMS_PER_PAGE, (activePage + 1) * ITEMS_PER_PAGE).map((row) => (\n \n
\n \n \n \n \n \n \n \n \n 统计 \n \n \n \n \n {chartData.series ? (\n \n \n 暂无数据\n \n \n )}\n \n \n \n )}\n \n );\n};\n\nStatisticalBarChart.propTypes = {\n isLoading: PropTypes.bool\n};\n\nexport default StatisticalBarChart;\n\nconst chartData = {\n height: 480,\n type: 'bar',\n options: {\n colors: [\n '#008FFB',\n '#00E396',\n '#FEB019',\n '#FF4560',\n '#775DD0',\n '#55efc4',\n '#81ecec',\n '#74b9ff',\n '#a29bfe',\n '#00b894',\n '#00cec9',\n '#0984e3',\n '#6c5ce7',\n '#ffeaa7',\n '#fab1a0',\n '#ff7675',\n '#fd79a8',\n '#fdcb6e',\n '#e17055',\n '#d63031',\n '#e84393'\n ],\n chart: {\n id: 'bar-chart',\n stacked: true,\n toolbar: {\n show: true\n },\n zoom: {\n enabled: true\n }\n },\n responsive: [\n {\n breakpoint: 480,\n options: {\n legend: {\n position: 'bottom',\n offsetX: -10,\n offsetY: 0\n }\n }\n }\n ],\n plotOptions: {\n bar: {\n horizontal: false,\n columnWidth: '50%'\n }\n },\n xaxis: {\n type: 'category',\n categories: []\n },\n legend: {\n show: true,\n fontSize: '14px',\n fontFamily: `'Roboto', sans-serif`,\n position: 'bottom',\n offsetX: 20,\n labels: {\n useSeriesColors: false\n },\n markers: {\n width: 16,\n height: 16,\n radius: 5\n },\n itemMargin: {\n horizontal: 15,\n vertical: 8\n }\n },\n fill: {\n type: 'solid'\n },\n dataLabels: {\n enabled: false\n },\n grid: {\n show: true\n },\n tooltip: {\n theme: 'dark',\n fixed: {\n enabled: false\n },\n y: {\n formatter: function (val) {\n return '$' + val;\n }\n },\n marker: {\n show: false\n }\n }\n },\n series: []\n};\n"
},
{
"path": "web/berry/src/views/Dashboard/component/StatisticalCard.js",
"content": "import PropTypes from 'prop-types';\n\n// material-ui\nimport { styled, useTheme } from '@mui/material/styles';\nimport { Avatar, Box, List, ListItem, ListItemAvatar, ListItemText, Typography } from '@mui/material';\n\n// project imports\nimport MainCard from 'ui-component/cards/MainCard';\nimport TotalIncomeCard from 'ui-component/cards/Skeleton/TotalIncomeCard';\n\n// assets\nimport TableChartOutlinedIcon from '@mui/icons-material/TableChartOutlined';\n\n// styles\nconst CardWrapper = styled(MainCard)(({ theme }) => ({\n backgroundColor: theme.palette.primary.dark,\n color: theme.palette.primary.light,\n overflow: 'hidden',\n position: 'relative',\n '&:after': {\n content: '\"\"',\n position: 'absolute',\n width: 210,\n height: 210,\n background: `linear-gradient(210.04deg, ${theme.palette.primary[200]} -50.94%, rgba(144, 202, 249, 0) 83.49%)`,\n borderRadius: '50%',\n top: -30,\n right: -180\n },\n '&:before': {\n content: '\"\"',\n position: 'absolute',\n width: 210,\n height: 210,\n background: `linear-gradient(140.9deg, ${theme.palette.primary[200]} -14.02%, rgba(144, 202, 249, 0) 77.58%)`,\n borderRadius: '50%',\n top: -160,\n right: -130\n }\n}));\n\n// ==============================|| DASHBOARD - TOTAL INCOME DARK CARD ||============================== //\n\nconst StatisticalCard = ({ isLoading }) => {\n const theme = useTheme();\n\n return (\n <>\n {isLoading ? (\n \n \n \n \n \n \n \n \n $203k\n \n }\n secondary={\n \n Total Income\n \n }\n />\n \n
\n \n \n )}\n \n );\n};\n\nStatisticalCard.propTypes = {\n isLoading: PropTypes.bool\n};\n\nexport default StatisticalCard;\n"
},
{
"path": "web/berry/src/views/Dashboard/component/StatisticalLineChartCard.js",
"content": "import PropTypes from 'prop-types';\n\n// material-ui\nimport { useTheme, styled } from '@mui/material/styles';\nimport { Box, Grid, Typography } from '@mui/material';\n\n// third-party\nimport Chart from 'react-apexcharts';\n\n// project imports\nimport MainCard from 'ui-component/cards/MainCard';\nimport SkeletonTotalOrderCard from 'ui-component/cards/Skeleton/EarningCard';\n\nconst CardWrapper = styled(MainCard)(({ theme }) => ({\n ...theme.typography.CardWrapper,\n color: '#fff',\n overflow: 'hidden',\n position: 'relative',\n '&>div': {\n position: 'relative',\n zIndex: 5\n },\n '&:after': {\n content: '\"\"',\n position: 'absolute',\n width: 210,\n height: 210,\n background: theme.palette.primary[800],\n borderRadius: '50%',\n zIndex: 1,\n top: -85,\n right: -95,\n [theme.breakpoints.down('sm')]: {\n top: -105,\n right: -140\n }\n },\n '&:before': {\n content: '\"\"',\n position: 'absolute',\n zIndex: 1,\n width: 210,\n height: 210,\n background: theme.palette.primary[800],\n borderRadius: '50%',\n top: -125,\n right: -15,\n opacity: 0.5,\n [theme.breakpoints.down('sm')]: {\n top: -155,\n right: -70\n }\n }\n}));\n\n// ==============================|| DASHBOARD - TOTAL ORDER LINE CHART CARD ||============================== //\n\nconst StatisticalLineChartCard = ({ isLoading, title, chartData, todayValue }) => {\n const theme = useTheme();\n\n return (\n <>\n {isLoading ? (\n \n \n \n \n \n \n \n \n \n {todayValue || '0'}\n \n \n \n \n {title}\n \n \n \n \n \n {chartData ? (\n \n 无数据\n \n )}\n \n \n \n \n \n \n )}\n \n );\n};\n\nStatisticalLineChartCard.propTypes = {\n isLoading: PropTypes.bool,\n title: PropTypes.string\n};\n\nexport default StatisticalLineChartCard;\n"
},
{
"path": "web/berry/src/views/Dashboard/index.js",
"content": "import { useEffect, useState } from 'react';\nimport { Grid, Typography } from '@mui/material';\nimport { gridSpacing } from 'store/constant';\nimport StatisticalLineChartCard from './component/StatisticalLineChartCard';\nimport StatisticalBarChart from './component/StatisticalBarChart';\nimport { generateChartOptions, getLastSevenDays } from 'utils/chart';\nimport { API } from 'utils/api';\nimport { showError, calculateQuota, renderNumber } from 'utils/common';\nimport UserCard from 'ui-component/cards/UserCard';\n\nconst Dashboard = () => {\n const [isLoading, setLoading] = useState(true);\n const [statisticalData, setStatisticalData] = useState([]);\n const [requestChart, setRequestChart] = useState(null);\n const [quotaChart, setQuotaChart] = useState(null);\n const [tokenChart, setTokenChart] = useState(null);\n const [users, setUsers] = useState([]);\n\n const userDashboard = async () => {\n const res = await API.get('/api/user/dashboard');\n const { success, message, data } = res.data;\n if (success) {\n if (data) {\n let lineData = getLineDataGroup(data);\n setRequestChart(getLineCardOption(lineData, 'RequestCount'));\n setQuotaChart(getLineCardOption(lineData, 'Quota'));\n setTokenChart(getLineCardOption(lineData, 'PromptTokens'));\n setStatisticalData(getBarDataGroup(data));\n }\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n const loadUser = async () => {\n let res = await API.get(`/api/user/self`);\n const { success, message, data } = res.data;\n if (success) {\n setUsers(data);\n } else {\n showError(message);\n }\n };\n\n useEffect(() => {\n userDashboard();\n loadUser();\n }, []);\n\n return (\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n 余额： \n \n \n {users?.quota ? '$' + calculateQuota(users.quota) : '未知'} \n \n \n 已使用： \n \n \n {users?.used_quota ? '$' + calculateQuota(users.used_quota) : '未知'} \n \n \n 调用次数： \n \n \n {users?.request_count || '未知'} \n \n \n \n \n \n \n \n );\n};\nexport default Dashboard;\n\nfunction getLineDataGroup(statisticalData) {\n let groupedData = statisticalData.reduce((acc, cur) => {\n if (!acc[cur.Day]) {\n acc[cur.Day] = {\n date: cur.Day,\n RequestCount: 0,\n Quota: 0,\n PromptTokens: 0,\n CompletionTokens: 0\n };\n }\n acc[cur.Day].RequestCount += cur.RequestCount;\n acc[cur.Day].Quota += cur.Quota;\n acc[cur.Day].PromptTokens += cur.PromptTokens;\n acc[cur.Day].CompletionTokens += cur.CompletionTokens;\n return acc;\n }, {});\n let lastSevenDays = getLastSevenDays();\n return lastSevenDays.map((day) => {\n if (!groupedData[day]) {\n return {\n date: day,\n RequestCount: 0,\n Quota: 0,\n PromptTokens: 0,\n CompletionTokens: 0\n };\n } else {\n return groupedData[day];\n }\n });\n}\n\nfunction getBarDataGroup(data) {\n const lastSevenDays = getLastSevenDays();\n const result = [];\n const map = new Map();\n\n for (const item of data) {\n if (!map.has(item.ModelName)) {\n const newData = { name: item.ModelName, data: new Array(7) };\n map.set(item.ModelName, newData);\n result.push(newData);\n }\n const index = lastSevenDays.indexOf(item.Day);\n if (index !== -1) {\n map.get(item.ModelName).data[index] = calculateQuota(item.Quota, 3);\n }\n }\n\n for (const item of result) {\n for (let i = 0; i < 7; i++) {\n if (item.data[i] === undefined) {\n item.data[i] = 0;\n }\n }\n }\n\n return { data: result, xaxis: lastSevenDays };\n}\n\nfunction getLineCardOption(lineDataGroup, field) {\n let todayValue = 0;\n let chartData = null;\n const lastItem = lineDataGroup.length - 1;\n let lineData = lineDataGroup.map((item, index) => {\n let tmp = {\n date: item.date,\n value: item[field]\n };\n switch (field) {\n case 'Quota':\n tmp.value = calculateQuota(item.Quota, 3);\n break;\n case 'PromptTokens':\n tmp.value += item.CompletionTokens;\n break;\n }\n\n if (index == lastItem) {\n todayValue = tmp.value;\n }\n return tmp;\n });\n\n switch (field) {\n case 'RequestCount':\n chartData = generateChartOptions(lineData, '次');\n todayValue = renderNumber(todayValue);\n break;\n case 'Quota':\n chartData = generateChartOptions(lineData, '美元');\n todayValue = '$' + renderNumber(todayValue);\n break;\n case 'PromptTokens':\n chartData = generateChartOptions(lineData, '');\n todayValue = renderNumber(todayValue);\n break;\n }\n\n return { chartData: chartData, todayValue: todayValue };\n}\n"
},
{
"path": "web/berry/src/views/Error/index.js",
"content": "import Box from '@mui/material/Box';\nimport Button from '@mui/material/Button';\nimport Container from '@mui/material/Container';\nimport NotFound from 'assets/images/404.svg';\nimport { useNavigate } from 'react-router';\n\n// ----------------------------------------------------------------------\n\nexport default function NotFoundView() {\n const navigate = useNavigate();\n const goBack = () => {\n navigate(-1);\n };\n return (\n <>\n \n \n \n 返回\n \n \n \n \n );\n}\n"
},
{
"path": "web/berry/src/views/Home/baseIndex.js",
"content": "import { Box, Typography, Button, Container, Stack } from '@mui/material';\nimport Grid from '@mui/material/Unstable_Grid2';\nimport { GitHub } from '@mui/icons-material';\n\nconst BaseIndex = () => (\n \n \n \n \n \n \n One API\n \n \n All in one 的 OpenAI 接口 \n \n \n \n \n \n);\n\nexport default BaseIndex;\n"
},
{
"path": "web/berry/src/views/Home/index.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { showError, showNotice } from 'utils/common';\nimport { API } from 'utils/api';\nimport { marked } from 'marked';\nimport BaseIndex from './baseIndex';\nimport { Box, Container } from '@mui/material';\n\nconst Home = () => {\n const [homePageContentLoaded, setHomePageContentLoaded] = useState(false);\n const [homePageContent, setHomePageContent] = useState('');\n const displayNotice = async () => {\n const res = await API.get('/api/notice');\n const { success, message, data } = res.data;\n if (success) {\n let oldNotice = localStorage.getItem('notice');\n if (data !== oldNotice && data !== '') {\n const htmlNotice = marked(data);\n showNotice(htmlNotice, true);\n localStorage.setItem('notice', data);\n }\n } else {\n showError(message);\n }\n };\n\n const displayHomePageContent = async () => {\n setHomePageContent(localStorage.getItem('home_page_content') || '');\n const res = await API.get('/api/home_page_content');\n const { success, message, data } = res.data;\n if (success) {\n let content = data;\n if (!data.startsWith('https://')) {\n content = marked.parse(data);\n }\n setHomePageContent(content);\n localStorage.setItem('home_page_content', content);\n } else {\n showError(message);\n setHomePageContent('加载首页内容失败...');\n }\n setHomePageContentLoaded(true);\n };\n\n useEffect(() => {\n displayNotice().then();\n displayHomePageContent().then();\n }, []);\n\n return (\n <>\n {homePageContentLoaded && homePageContent === '' ? (\n \n {homePageContent.startsWith('https://') ? (\n \n ) : (\n <>\n \n
\n \n \n )}\n \n \n )}\n \n );\n};\n\nexport default Home;\n"
},
{
"path": "web/berry/src/views/Log/component/TableHead.js",
"content": "import PropTypes from 'prop-types';\nimport { TableCell, TableHead, TableRow } from '@mui/material';\n\nconst LogTableHead = ({ userIsAdmin }) => {\n return (\n \n \n 时间 \n {userIsAdmin && 渠道 }\n {userIsAdmin && 用户 }\n 令牌 \n 类型 \n 模型 \n 提示 \n 补全 \n 额度 \n 详情 \n \n \n );\n};\n\nexport default LogTableHead;\n\nLogTableHead.propTypes = {\n userIsAdmin: PropTypes.bool\n};\n"
},
{
"path": "web/berry/src/views/Log/component/TableRow.js",
"content": "import PropTypes from 'prop-types';\n\nimport { TableRow, TableCell } from '@mui/material';\n\nimport { timestamp2string, renderQuota } from 'utils/common';\nimport Label from 'ui-component/Label';\nimport LogType from '../type/LogType';\n\nfunction renderType(type) {\n const typeOption = LogType[type];\n if (typeOption) {\n return (\n \n {' '}\n {typeOption.text}{' '}\n \n );\n } else {\n return (\n \n {' '}\n 未知{' '}\n \n );\n }\n}\n\nexport default function LogTableRow({ item, userIsAdmin }) {\n return (\n <>\n \n {timestamp2string(item.created_at)} \n\n {userIsAdmin && {item.channel || ''} }\n {userIsAdmin && (\n \n \n {item.username}\n \n \n )}\n \n {item.token_name && (\n \n {item.token_name}\n \n )}\n \n {renderType(item.type)} \n \n {item.model_name && (\n \n {item.model_name}\n \n )}\n \n {item.prompt_tokens || ''} \n {item.completion_tokens || ''} \n {item.quota ? renderQuota(item.quota, 6) : ''} \n {item.content} \n \n \n );\n}\n\nLogTableRow.propTypes = {\n item: PropTypes.object,\n userIsAdmin: PropTypes.bool\n};\n"
},
{
"path": "web/berry/src/views/Log/component/TableToolBar.js",
"content": "import PropTypes from \"prop-types\";\nimport { useTheme } from \"@mui/material/styles\";\nimport {\n IconUser,\n IconKey,\n IconBrandGithubCopilot,\n IconSitemap,\n} from \"@tabler/icons-react\";\nimport {\n InputAdornment,\n OutlinedInput,\n Stack,\n FormControl,\n InputLabel,\n Select,\n MenuItem,\n} from \"@mui/material\";\nimport { LocalizationProvider, DateTimePicker } from \"@mui/x-date-pickers\";\nimport { AdapterDayjs } from \"@mui/x-date-pickers/AdapterDayjs\";\nimport dayjs from \"dayjs\";\nimport LogType from \"../type/LogType\";\nrequire(\"dayjs/locale/zh-cn\");\n// ----------------------------------------------------------------------\n\nexport default function TableToolBar({\n filterName,\n handleFilterName,\n userIsAdmin,\n}) {\n const theme = useTheme();\n const grey500 = theme.palette.grey[500];\n\n return (\n <>\n \n \n 令牌名称 \n \n \n \n 模型名称 \n \n \n\n \n \n {\n if (value === null) {\n handleFilterName({\n target: { name: \"start_timestamp\", value: 0 },\n });\n return;\n }\n handleFilterName({\n target: { name: \"start_timestamp\", value: value.unix() },\n });\n }}\n slotProps={{\n actionBar: {\n actions: [\"clear\", \"today\", \"accept\"],\n },\n }}\n />\n \n \n\n \n \n {\n if (value === null) {\n handleFilterName({\n target: { name: \"end_timestamp\", value: 0 },\n });\n return;\n }\n handleFilterName({\n target: { name: \"end_timestamp\", value: value.unix() },\n });\n }}\n slotProps={{\n actionBar: {\n actions: [\"clear\", \"today\", \"accept\"],\n },\n }}\n />\n \n \n \n\n \n {userIsAdmin && (\n \n 渠道ID \n \n \n )}\n\n {userIsAdmin && (\n \n 用户名称 \n \n \n )}\n\n \n 类型 \n \n {Object.values(LogType).map((option) => {\n return (\n \n {option.text}\n \n );\n })}\n \n \n \n \n );\n}\n\nTableToolBar.propTypes = {\n filterName: PropTypes.object,\n handleFilterName: PropTypes.func,\n userIsAdmin: PropTypes.bool,\n};\n"
},
{
"path": "web/berry/src/views/Log/index.js",
"content": "import { useState, useEffect } from 'react';\nimport { showError } from 'utils/common';\n\nimport Table from '@mui/material/Table';\nimport TableBody from '@mui/material/TableBody';\nimport TableContainer from '@mui/material/TableContainer';\nimport PerfectScrollbar from 'react-perfect-scrollbar';\nimport TablePagination from '@mui/material/TablePagination';\nimport LinearProgress from '@mui/material/LinearProgress';\nimport ButtonGroup from '@mui/material/ButtonGroup';\nimport Toolbar from '@mui/material/Toolbar';\n\nimport { Button, Card, Stack, Container, Typography, Box } from '@mui/material';\nimport LogTableRow from './component/TableRow';\nimport LogTableHead from './component/TableHead';\nimport TableToolBar from './component/TableToolBar';\nimport { API } from 'utils/api';\nimport { isAdmin } from 'utils/common';\nimport { ITEMS_PER_PAGE } from 'constants';\nimport { IconRefresh, IconSearch } from '@tabler/icons-react';\n\nexport default function Log() {\n const originalKeyword = {\n p: 0,\n username: '',\n token_name: '',\n model_name: '',\n start_timestamp: 0,\n end_timestamp: new Date().getTime() / 1000 + 3600,\n type: 0,\n channel: ''\n };\n const [logs, setLogs] = useState([]);\n const [activePage, setActivePage] = useState(0);\n const [searching, setSearching] = useState(false);\n const [searchKeyword, setSearchKeyword] = useState(originalKeyword);\n const [initPage, setInitPage] = useState(true);\n const userIsAdmin = isAdmin();\n\n const loadLogs = async (startIdx) => {\n setSearching(true);\n const url = userIsAdmin ? '/api/log/' : '/api/log/self/';\n const query = searchKeyword;\n\n query.p = startIdx;\n if (!userIsAdmin) {\n delete query.username;\n delete query.channel;\n }\n const res = await API.get(url, { params: query });\n const { success, message, data } = res.data;\n if (success) {\n if (startIdx === 0) {\n setLogs(data);\n } else {\n let newLogs = [...logs];\n newLogs.splice(startIdx * ITEMS_PER_PAGE, data.length, ...data);\n setLogs(newLogs);\n }\n } else {\n showError(message);\n }\n setSearching(false);\n };\n\n const onPaginationChange = (event, activePage) => {\n (async () => {\n if (activePage === Math.ceil(logs.length / ITEMS_PER_PAGE)) {\n // In this case we have to load more data and then append them.\n await loadLogs(activePage);\n }\n setActivePage(activePage);\n })();\n };\n\n const searchLogs = async (event) => {\n event.preventDefault();\n await loadLogs(0);\n setActivePage(0);\n return;\n };\n\n const handleSearchKeyword = (event) => {\n setSearchKeyword({ ...searchKeyword, [event.target.name]: event.target.value });\n };\n\n // 处理刷新\n const handleRefresh = () => {\n setInitPage(true);\n };\n\n useEffect(() => {\n setSearchKeyword(originalKeyword);\n setActivePage(0);\n loadLogs(0)\n .then()\n .catch((reason) => {\n showError(reason);\n });\n setInitPage(false);\n }, [initPage]);\n\n return (\n <>\n \n 日志 \n \n \n \n \n theme.spacing(0, 1, 0, 3)\n }}\n >\n \n \n \n \n \n {searching && \n \n \n \n {logs.slice(activePage * ITEMS_PER_PAGE, (activePage + 1) * ITEMS_PER_PAGE).map((row, index) => (\n \n
\n \n \n \n \n );\n}\n"
},
{
"path": "web/berry/src/views/Log/type/LogType.js",
"content": "const LOG_TYPE = {\n 0: { value: '0', text: '全部', color: '' },\n 1: { value: '1', text: '充值', color: 'primary' },\n 2: { value: '2', text: '消费', color: 'orange' },\n 3: { value: '3', text: '管理', color: 'default' },\n 4: { value: '4', text: '系统', color: 'secondary' },\n 5: { value: '5', text: '测试', color: 'secondary' },\n};\n\nexport default LOG_TYPE;\n"
},
{
"path": "web/berry/src/views/Profile/component/EmailModal.js",
"content": "import { useState, useEffect } from \"react\";\nimport PropTypes from \"prop-types\";\nimport React from \"react\";\nimport {\n Dialog,\n DialogTitle,\n DialogContent,\n DialogActions,\n OutlinedInput,\n Button,\n InputLabel,\n Grid,\n InputAdornment,\n FormControl,\n FormHelperText,\n} from \"@mui/material\";\nimport { Formik } from \"formik\";\nimport { showError, showSuccess } from \"utils/common\";\nimport { useTheme } from \"@mui/material/styles\";\nimport * as Yup from \"yup\";\nimport useRegister from \"hooks/useRegister\";\nimport { API } from \"utils/api\";\n\nconst validationSchema = Yup.object().shape({\n email: Yup.string().email(\"请输入正确的邮箱地址\").required(\"邮箱不能为空\"),\n email_verification_code: Yup.string().required(\"验证码不能为空\"),\n});\n\nconst EmailModal = ({ open, handleClose, turnstileToken }) => {\n const theme = useTheme();\n const [countdown, setCountdown] = useState(30);\n const [disableButton, setDisableButton] = useState(false);\n const { sendVerificationCode } = useRegister();\n const [loading, setLoading] = useState(false);\n\n const submit = async (values, { setErrors, setStatus, setSubmitting }) => {\n setLoading(true);\n setSubmitting(true);\n const res = await API.get(\n `/api/oauth/email/bind?email=${values.email}&code=${values.email_verification_code}`\n );\n const { success, message } = res.data;\n if (success) {\n showSuccess(\"邮箱账户绑定成功！\");\n setSubmitting(false);\n setStatus({ success: true });\n handleClose();\n } else {\n showError(message);\n setErrors({ submit: message });\n }\n setLoading(false);\n };\n\n useEffect(() => {\n let countdownInterval = null;\n if (disableButton && countdown > 0) {\n countdownInterval = setInterval(() => {\n setCountdown(countdown - 1);\n }, 1000);\n } else if (countdown === 0) {\n setDisableButton(false);\n setCountdown(30);\n }\n return () => clearInterval(countdownInterval); // Clean up on unmount\n }, [disableButton, countdown]);\n\n const handleSendCode = async (email) => {\n setDisableButton(true);\n if (email === \"\") {\n showError(\"请输入邮箱\");\n return;\n }\n if (turnstileToken === \"\") {\n showError(\"请稍后几秒重试，Turnstile 正在检查用户环境！\");\n return;\n }\n setLoading(true);\n const { success, message } = await sendVerificationCode(\n email,\n turnstileToken\n );\n setLoading(false);\n if (!success) {\n showError(message);\n return;\n }\n };\n\n return (\n \n 绑定邮箱 \n \n \n \n {({\n errors,\n touched,\n handleBlur,\n handleChange,\n handleSubmit,\n values,\n }) => (\n \n )}\n \n \n \n \n );\n};\n\nexport default EmailModal;\n\nEmailModal.propTypes = {\n open: PropTypes.bool,\n handleClose: PropTypes.func,\n};\n"
},
{
"path": "web/berry/src/views/Profile/index.js",
"content": "import { useState, useEffect } from 'react';\nimport UserCard from 'ui-component/cards/UserCard';\nimport {\n Card,\n Button,\n InputLabel,\n FormControl,\n OutlinedInput,\n Stack,\n Alert,\n Dialog,\n DialogTitle,\n DialogContent,\n DialogActions,\n Divider,\n SvgIcon\n} from '@mui/material';\nimport Grid from '@mui/material/Unstable_Grid2';\nimport SubCard from 'ui-component/cards/SubCard';\nimport { IconBrandWechat, IconBrandGithub, IconMail } from '@tabler/icons-react';\nimport Label from 'ui-component/Label';\nimport { API } from 'utils/api';\nimport { onOidcClicked, showError, showSuccess } from 'utils/common';\nimport { onGitHubOAuthClicked, onLarkOAuthClicked, copy } from 'utils/common';\nimport * as Yup from 'yup';\nimport WechatModal from 'views/Authentication/AuthForms/WechatModal';\nimport { useSelector } from 'react-redux';\nimport EmailModal from './component/EmailModal';\nimport Turnstile from 'react-turnstile';\nimport { ReactComponent as Lark } from 'assets/images/icons/lark.svg';\nimport { ReactComponent as OIDC } from 'assets/images/icons/oidc.svg';\n\nconst validationSchema = Yup.object().shape({\n username: Yup.string().required('用户名 不能为空').min(3, '用户名 不能小于 3 个字符'),\n display_name: Yup.string(),\n password: Yup.string().test('password', '密码不能小于 8 个字符', (val) => {\n return !val || val.length >= 8;\n })\n});\n\nexport default function Profile() {\n const [inputs, setInputs] = useState([]);\n const [showAccountDeleteModal, setShowAccountDeleteModal] = useState(false);\n const [turnstileEnabled, setTurnstileEnabled] = useState(false);\n const [turnstileSiteKey, setTurnstileSiteKey] = useState('');\n const [turnstileToken, setTurnstileToken] = useState('');\n const [openWechat, setOpenWechat] = useState(false);\n const [openEmail, setOpenEmail] = useState(false);\n const status = useSelector((state) => state.siteInfo);\n\n const handleWechatOpen = () => {\n setOpenWechat(true);\n };\n\n const handleWechatClose = () => {\n setOpenWechat(false);\n };\n\n const handleInputChange = (event) => {\n let { name, value } = event.target;\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n };\n\n const loadUser = async () => {\n let res = await API.get(`/api/user/self`);\n const { success, message, data } = res.data;\n if (success) {\n setInputs(data);\n } else {\n showError(message);\n }\n };\n\n const bindWeChat = async (code) => {\n if (code === '') return;\n try {\n const res = await API.get(`/api/oauth/wechat/bind?code=${code}`);\n const { success, message } = res.data;\n if (success) {\n showSuccess('微信账户绑定成功！');\n }\n return { success, message };\n } catch (err) {\n // 请求失败，设置错误信息\n return { success: false, message: '' };\n }\n };\n\n const generateAccessToken = async () => {\n const res = await API.get('/api/user/token');\n const { success, message, data } = res.data;\n if (success) {\n setInputs((inputs) => ({ ...inputs, access_token: data }));\n copy(data, '访问令牌');\n } else {\n showError(message);\n }\n\n console.log(turnstileEnabled, turnstileSiteKey, status);\n };\n\n const submit = async () => {\n try {\n await validationSchema.validate(inputs);\n const res = await API.put(`/api/user/self`, inputs);\n const { success, message } = res.data;\n if (success) {\n showSuccess('用户信息更新成功！');\n } else {\n showError(message);\n }\n } catch (err) {\n showError(err.message);\n }\n };\n\n useEffect(() => {\n if (status) {\n if (status.turnstile_check) {\n setTurnstileEnabled(true);\n setTurnstileSiteKey(status.turnstile_site_key);\n }\n }\n loadUser().then();\n }, [status]);\n\n function getOidcId(){\n if (!inputs.oidc_id) return '';\n let oidc_id = inputs.oidc_id;\n if (inputs.oidc_id.length > 8) {\n oidc_id = inputs.oidc_id.slice(0, 6) + '...' + inputs.oidc_id.slice(-6);\n }\n return oidc_id;\n }\n\n return (\n <>\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n 用户名 \n \n \n \n \n 密码 \n \n \n \n \n 显示名称 \n \n \n \n \n 提交\n \n \n \n \n \n \n {status.wechat_login && !inputs.wechat_id && (\n \n \n 绑定微信账号\n \n \n )}\n {status.github_oauth && !inputs.github_id && (\n \n onGitHubOAuthClicked(status.github_client_id, true)}>\n 绑定 GitHub 账号\n \n \n )}\n {status.lark_client_id && !inputs.lark_id && (\n \n onLarkOAuthClicked(status.lark_client_id)}>\n 绑定 飞书 账号\n \n \n )}\n {status.oidc && !inputs.oidc_id && (\n \n onOidcClicked(status.oidc_authorization_endpoint,status.oidc_client_id,true)}>\n 绑定 OIDC 账号\n \n \n )}\n \n {\n setOpenEmail(true);\n }}\n >\n {inputs.email ? '更换邮箱' : '绑定邮箱'}\n \n {turnstileEnabled ? (\n {\n setTurnstileToken(token);\n }}\n />\n ) : (\n <>\n )}\n \n \n \n \n \n \n 注意，此处生成的令牌用于系统管理，而非用于请求 OpenAI 相关的服务，请知悉。 \n \n {inputs.access_token && (\n \n \n 你的访问令牌是: {inputs.access_token} \n \n )}\n \n \n {inputs.access_token ? '重置访问令牌' : '生成访问令牌'}\n \n \n\n \n {\n setShowAccountDeleteModal(true);\n }}\n >\n 删除帐号\n \n \n \n \n \n \n \n setShowAccountDeleteModal(false)} maxWidth={'md'}>\n \n 危险操作\n \n 您正在删除自己的帐户，将清空所有数据且不可恢复 \n \n setShowAccountDeleteModal(false)}>取消 \n {\n setShowAccountDeleteModal(false);\n }}\n >\n 确定\n \n \n \n {\n setOpenEmail(false);\n }}\n />\n \n );\n}\n"
},
{
"path": "web/berry/src/views/Redemption/component/EditModal.js",
"content": "import PropTypes from 'prop-types';\nimport * as Yup from 'yup';\nimport { Formik } from 'formik';\nimport { useTheme } from '@mui/material/styles';\nimport { useState, useEffect } from 'react';\nimport {\n Dialog,\n DialogTitle,\n DialogContent,\n DialogActions,\n Button,\n Divider,\n FormControl,\n InputLabel,\n OutlinedInput,\n InputAdornment,\n FormHelperText\n} from '@mui/material';\n\nimport { renderQuotaWithPrompt, showSuccess, showError, downloadTextAsFile } from 'utils/common';\nimport { API } from 'utils/api';\n\nconst validationSchema = Yup.object().shape({\n is_edit: Yup.boolean(),\n name: Yup.string().required('名称 不能为空'),\n quota: Yup.number().min(0, '必须大于等于0'),\n count: Yup.number().when('is_edit', {\n is: false,\n then: Yup.number().min(1, '必须大于等于1'),\n otherwise: Yup.number()\n })\n});\n\nconst originInputs = {\n is_edit: false,\n name: '',\n quota: 100000,\n count: 1\n};\n\nconst EditModal = ({ open, redemptiondId, onCancel, onOk }) => {\n const theme = useTheme();\n const [inputs, setInputs] = useState(originInputs);\n\n const submit = async (values, { setErrors, setStatus, setSubmitting }) => {\n setSubmitting(true);\n\n let res;\n if (values.is_edit) {\n res = await API.put(`/api/redemption/`, { ...values, id: parseInt(redemptiondId) });\n } else {\n res = await API.post(`/api/redemption/`, values);\n }\n const { success, message, data } = res.data;\n if (success) {\n if (values.is_edit) {\n showSuccess('兑换码更新成功！');\n } else {\n showSuccess('兑换码创建成功！');\n if (data.length > 1) {\n let text = '';\n for (let i = 0; i < data.length; i++) {\n text += data[i] + '\\n';\n }\n downloadTextAsFile(text, `${values.name}.txt`);\n }\n }\n setSubmitting(false);\n setStatus({ success: true });\n onOk(true);\n } else {\n showError(message);\n setErrors({ submit: message });\n }\n };\n\n const loadRedemptiond = async () => {\n let res = await API.get(`/api/redemption/${redemptiondId}`);\n const { success, message, data } = res.data;\n if (success) {\n data.is_edit = true;\n setInputs(data);\n } else {\n showError(message);\n }\n };\n\n useEffect(() => {\n if (redemptiondId) {\n loadRedemptiond().then();\n } else {\n setInputs(originInputs);\n }\n }, [redemptiondId]);\n\n return (\n \n \n {redemptiondId ? '编辑兑换码' : '新建兑换码'}\n \n \n \n {({ errors, handleBlur, handleChange, handleSubmit, touched, values, isSubmitting }) => (\n \n )}\n \n \n \n );\n};\n\nexport default EditModal;\n\nEditModal.propTypes = {\n open: PropTypes.bool,\n redemptiondId: PropTypes.number,\n onCancel: PropTypes.func,\n onOk: PropTypes.func\n};\n"
},
{
"path": "web/berry/src/views/Redemption/component/TableHead.js",
"content": "import { TableCell, TableHead, TableRow } from '@mui/material';\n\nconst RedemptionTableHead = () => {\n return (\n \n \n ID \n 名称 \n 状态 \n 额度 \n 创建时间 \n 兑换时间 \n 操作 \n \n \n );\n};\n\nexport default RedemptionTableHead;\n"
},
{
"path": "web/berry/src/views/Redemption/component/TableRow.js",
"content": "import PropTypes from 'prop-types';\nimport { useState } from 'react';\n\nimport {\n Popover,\n TableRow,\n MenuItem,\n TableCell,\n IconButton,\n Dialog,\n DialogActions,\n DialogContent,\n DialogContentText,\n DialogTitle,\n Button,\n Stack\n} from '@mui/material';\n\nimport Label from 'ui-component/Label';\nimport TableSwitch from 'ui-component/Switch';\nimport { timestamp2string, renderQuota, copy } from 'utils/common';\n\nimport { IconDotsVertical, IconEdit, IconTrash } from '@tabler/icons-react';\n\nexport default function RedemptionTableRow({ item, manageRedemption, handleOpenModal, setModalRedemptionId }) {\n const [open, setOpen] = useState(null);\n const [openDelete, setOpenDelete] = useState(false);\n const [statusSwitch, setStatusSwitch] = useState(item.status);\n\n const handleDeleteOpen = () => {\n handleCloseMenu();\n setOpenDelete(true);\n };\n\n const handleDeleteClose = () => {\n setOpenDelete(false);\n };\n\n const handleOpenMenu = (event) => {\n setOpen(event.currentTarget);\n };\n\n const handleCloseMenu = () => {\n setOpen(null);\n };\n\n const handleStatus = async () => {\n const switchVlue = statusSwitch === 1 ? 2 : 1;\n const { success } = await manageRedemption(item.id, 'status', switchVlue);\n if (success) {\n setStatusSwitch(switchVlue);\n }\n };\n\n const handleDelete = async () => {\n handleCloseMenu();\n await manageRedemption(item.id, 'delete', '');\n };\n\n return (\n <>\n \n {item.id} \n\n {item.name} \n\n \n {item.status !== 1 && item.status !== 2 ? (\n \n {item.status === 3 ? '已使用' : '未知'}\n \n ) : (\n \n\n {renderQuota(item.quota)} \n {timestamp2string(item.created_time)} \n {item.redeemed_time ? timestamp2string(item.redeemed_time) : '尚未兑换'} \n \n \n {\n copy(item.key, '兑换码');\n }}\n >\n 复制\n \n \n \n \n \n \n\n \n {\n handleCloseMenu();\n handleOpenModal();\n setModalRedemptionId(item.id);\n }}\n >\n \n \n \n\n \n 删除兑换码 \n \n 是否删除兑换码 {item.name}？ \n \n \n 关闭 \n \n 删除\n \n \n \n \n );\n}\n\nRedemptionTableRow.propTypes = {\n item: PropTypes.object,\n manageRedemption: PropTypes.func,\n handleOpenModal: PropTypes.func,\n setModalRedemptionId: PropTypes.func\n};\n"
},
{
"path": "web/berry/src/views/Redemption/index.js",
"content": "import { useState, useEffect } from 'react';\nimport { showError, showSuccess } from 'utils/common';\n\nimport Table from '@mui/material/Table';\nimport TableBody from '@mui/material/TableBody';\nimport TableContainer from '@mui/material/TableContainer';\nimport PerfectScrollbar from 'react-perfect-scrollbar';\nimport TablePagination from '@mui/material/TablePagination';\nimport LinearProgress from '@mui/material/LinearProgress';\nimport ButtonGroup from '@mui/material/ButtonGroup';\nimport Toolbar from '@mui/material/Toolbar';\n\nimport { Button, Card, Box, Stack, Container, Typography } from '@mui/material';\nimport RedemptionTableRow from './component/TableRow';\nimport RedemptionTableHead from './component/TableHead';\nimport TableToolBar from 'ui-component/TableToolBar';\nimport { API } from 'utils/api';\nimport { ITEMS_PER_PAGE } from 'constants';\nimport { IconRefresh, IconPlus } from '@tabler/icons-react';\nimport EditeModal from './component/EditModal';\n\n// ----------------------------------------------------------------------\nexport default function Redemption() {\n const [redemptions, setRedemptions] = useState([]);\n const [activePage, setActivePage] = useState(0);\n const [searching, setSearching] = useState(false);\n const [searchKeyword, setSearchKeyword] = useState('');\n const [openModal, setOpenModal] = useState(false);\n const [editRedemptionId, setEditRedemptionId] = useState(0);\n\n const loadRedemptions = async (startIdx) => {\n setSearching(true);\n const res = await API.get(`/api/redemption/?p=${startIdx}`);\n const { success, message, data } = res.data;\n if (success) {\n if (startIdx === 0) {\n setRedemptions(data);\n } else {\n let newRedemptions = [...redemptions];\n newRedemptions.splice(startIdx * ITEMS_PER_PAGE, data.length, ...data);\n setRedemptions(newRedemptions);\n }\n } else {\n showError(message);\n }\n setSearching(false);\n };\n\n const onPaginationChange = (event, activePage) => {\n (async () => {\n if (activePage === Math.ceil(redemptions.length / ITEMS_PER_PAGE)) {\n // In this case we have to load more data and then append them.\n await loadRedemptions(activePage);\n }\n setActivePage(activePage);\n })();\n };\n\n const searchRedemptions = async (event) => {\n event.preventDefault();\n if (searchKeyword === '') {\n await loadRedemptions(0);\n setActivePage(0);\n return;\n }\n setSearching(true);\n const res = await API.get(`/api/redemption/search?keyword=${searchKeyword}`);\n const { success, message, data } = res.data;\n if (success) {\n setRedemptions(data);\n setActivePage(0);\n } else {\n showError(message);\n }\n setSearching(false);\n };\n\n const handleSearchKeyword = (event) => {\n setSearchKeyword(event.target.value);\n };\n\n const manageRedemptions = async (id, action, value) => {\n const url = '/api/redemption/';\n let data = { id };\n let res;\n switch (action) {\n case 'delete':\n res = await API.delete(url + id);\n break;\n case 'status':\n res = await API.put(url + '?status_only=true', {\n ...data,\n status: value\n });\n break;\n }\n const { success, message } = res.data;\n if (success) {\n showSuccess('操作成功完成！');\n if (action === 'delete') {\n await loadRedemptions(0);\n }\n } else {\n showError(message);\n }\n\n return res.data;\n };\n\n // 处理刷新\n const handleRefresh = async () => {\n await loadRedemptions(0);\n setActivePage(0);\n setSearchKeyword('');\n };\n\n const handleOpenModal = (redemptionId) => {\n setEditRedemptionId(redemptionId);\n setOpenModal(true);\n };\n\n const handleCloseModal = () => {\n setOpenModal(false);\n setEditRedemptionId(0);\n };\n\n const handleOkModal = (status) => {\n if (status === true) {\n handleCloseModal();\n handleRefresh();\n }\n };\n\n useEffect(() => {\n loadRedemptions(0)\n .then()\n .catch((reason) => {\n showError(reason);\n });\n }, []);\n\n return (\n <>\n \n 兑换 \n\n \n \n \n \n theme.spacing(0, 1, 0, 3)\n }}\n >\n \n \n \n \n \n {searching && \n \n \n \n {redemptions.slice(activePage * ITEMS_PER_PAGE, (activePage + 1) * ITEMS_PER_PAGE).map((row) => (\n \n
\n \n \n \n \n \n \n \n \n 充值链接 \n \n \n 聊天链接 \n \n \n 单位额度 \n \n \n 重试次数 \n \n \n \n \n {\n submitConfig(\"general\").then();\n }}\n >\n 保存通用设置\n \n \n \n \n \n \n \n {\n setHistoryTimestamp(\n newValue === null ? null : newValue.unix()\n );\n }}\n slotProps={{\n actionBar: {\n actions: [\"today\", \"clear\", \"accept\"],\n },\n }}\n />\n \n \n {\n deleteHistoryLogs().then();\n }}\n >\n 清理历史日志\n \n \n \n \n \n \n \n \n 最长响应时间\n \n \n \n \n 额度提醒阈值\n \n \n \n {\n submitConfig(\"monitor\").then();\n }}\n >\n 保存监控设置\n \n \n \n \n \n \n \n 新用户初始额度 \n \n \n 请求预扣费额度 \n \n \n \n 邀请新用户奖励额度\n \n \n \n \n 新用户使用邀请码奖励额度\n \n \n \n {\n submitConfig(\"quota\").then();\n }}\n >\n 保存额度设置\n \n \n \n \n \n \n \n \n \n \n \n {\n submitConfig(\"ratio\").then();\n }}\n >\n 保存倍率设置\n \n \n \n \n );\n};\n\nexport default OperationSetting;\n"
},
{
"path": "web/berry/src/views/Setting/component/OtherSetting.js",
"content": "import { useState, useEffect } from 'react';\nimport SubCard from 'ui-component/cards/SubCard';\nimport {\n Stack,\n FormControl,\n InputLabel,\n OutlinedInput,\n Button,\n Alert,\n TextField,\n Dialog,\n DialogTitle,\n DialogActions,\n DialogContent,\n Divider, Link\n} from '@mui/material';\nimport Grid from '@mui/material/Unstable_Grid2';\nimport { showError, showSuccess } from 'utils/common'; //,\nimport { API } from 'utils/api';\nimport { marked } from 'marked';\n\nconst OtherSetting = () => {\n let [inputs, setInputs] = useState({\n Footer: '',\n Notice: '',\n About: '',\n SystemName: '',\n Logo: '',\n HomePageContent: '',\n Theme: '',\n });\n let [loading, setLoading] = useState(false);\n const [showUpdateModal, setShowUpdateModal] = useState(false);\n const [updateData, setUpdateData] = useState({\n tag_name: '',\n content: ''\n });\n\n const getOptions = async () => {\n const res = await API.get('/api/option/');\n const { success, message, data } = res.data;\n if (success) {\n let newInputs = {};\n data.forEach((item) => {\n if (item.key in inputs) {\n newInputs[item.key] = item.value;\n }\n });\n setInputs(newInputs);\n } else {\n showError(message);\n }\n };\n\n useEffect(() => {\n getOptions().then();\n }, []);\n\n const updateOption = async (key, value) => {\n setLoading(true);\n const res = await API.put('/api/option/', {\n key,\n value\n });\n const { success, message } = res.data;\n if (success) {\n setInputs((inputs) => ({ ...inputs, [key]: value }));\n showSuccess('保存成功');\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n const handleInputChange = async (event) => {\n let { name, value } = event.target;\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n };\n\n const submitNotice = async () => {\n await updateOption('Notice', inputs.Notice);\n };\n\n const submitFooter = async () => {\n await updateOption('Footer', inputs.Footer);\n };\n\n const submitSystemName = async () => {\n await updateOption('SystemName', inputs.SystemName);\n };\n\n const submitTheme = async () => {\n await updateOption('Theme', inputs.Theme);\n };\n\n const submitLogo = async () => {\n await updateOption('Logo', inputs.Logo);\n };\n\n const submitAbout = async () => {\n await updateOption('About', inputs.About);\n };\n\n const submitOption = async (key) => {\n await updateOption(key, inputs[key]);\n };\n\n const openGitHubRelease = () => {\n window.location = 'https://github.com/songquanpeng/one-api/releases/latest';\n };\n\n const checkUpdate = async () => {\n const res = await API.get('https://api.github.com/repos/songquanpeng/one-api/releases/latest');\n const { tag_name, body } = res.data;\n if (tag_name === process.env.REACT_APP_VERSION) {\n showSuccess(`已是最新版本：${tag_name}`);\n } else {\n setUpdateData({\n tag_name: tag_name,\n content: marked.parse(body)\n });\n setShowUpdateModal(true);\n }\n };\n\n return (\n <>\n \n \n \n \n \n 检查更新\n \n \n \n \n \n \n \n \n 保存公告\n \n \n \n \n \n \n \n \n 系统名称 \n \n \n \n \n 设置系统名称\n \n \n \n \n 主题名称 \n \n \n \n \n 设置主题（重启生效）\n \n \n \n \n Logo 图片地址 \n \n \n \n \n 设置 Logo\n \n \n \n \n \n \n \n submitOption('HomePageContent')}>\n 保存首页内容\n \n \n \n \n \n \n \n \n 保存关于\n \n \n \n \n 移除 One API 的版权标识必须首先获得授权，项目维护需要花费大量精力，如果本项目对你有意义，请主动支持本项目。\n \n \n \n \n \n \n \n \n \n 设置页脚\n \n \n \n \n \n setShowUpdateModal(false)} fullWidth maxWidth={'md'}>\n \n 新版本：{updateData.tag_name}\n \n \n {' '}\n
\n \n \n setShowUpdateModal(false)}>关闭 \n {\n setShowUpdateModal(false);\n openGitHubRelease();\n }}\n >\n 去GitHub查看\n \n \n \n \n );\n};\n\nexport default OtherSetting;\n"
},
{
"path": "web/berry/src/views/Setting/component/SystemSetting.js",
"content": "import { useState, useEffect } from 'react';\nimport SubCard from 'ui-component/cards/SubCard';\nimport {\n Stack,\n FormControl,\n InputLabel,\n OutlinedInput,\n Checkbox,\n Button,\n FormControlLabel,\n Dialog,\n DialogTitle,\n DialogContent,\n DialogActions,\n Divider,\n Alert,\n Autocomplete,\n TextField\n} from '@mui/material';\nimport Grid from '@mui/material/Unstable_Grid2';\nimport { showError, showSuccess, removeTrailingSlash } from 'utils/common'; //,\nimport { API } from 'utils/api';\nimport { createFilterOptions } from '@mui/material/Autocomplete';\n\nconst filter = createFilterOptions();\nconst SystemSetting = () => {\n let [inputs, setInputs] = useState({\n PasswordLoginEnabled: '',\n PasswordRegisterEnabled: '',\n EmailVerificationEnabled: '',\n GitHubOAuthEnabled: '',\n GitHubClientId: '',\n GitHubClientSecret: '',\n LarkClientId: '',\n LarkClientSecret: '',\n OidcEnabled: '',\n OidcWellKnown: '',\n OidcClientId: '',\n OidcClientSecret: '',\n OidcAuthorizationEndpoint: '',\n OidcTokenEndpoint: '',\n OidcUserinfoEndpoint: '',\n Notice: '',\n SMTPServer: '',\n SMTPPort: '',\n SMTPAccount: '',\n SMTPFrom: '',\n SMTPToken: '',\n ServerAddress: '',\n Footer: '',\n WeChatAuthEnabled: '',\n WeChatServerAddress: '',\n WeChatServerToken: '',\n WeChatAccountQRCodeImageURL: '',\n TurnstileCheckEnabled: '',\n TurnstileSiteKey: '',\n TurnstileSecretKey: '',\n RegisterEnabled: '',\n EmailDomainRestrictionEnabled: '',\n EmailDomainWhitelist: [],\n MessagePusherAddress: '',\n MessagePusherToken: ''\n });\n const [originInputs, setOriginInputs] = useState({});\n let [loading, setLoading] = useState(false);\n const [EmailDomainWhitelist, setEmailDomainWhitelist] = useState([]);\n const [showPasswordWarningModal, setShowPasswordWarningModal] = useState(false);\n\n const getOptions = async () => {\n const res = await API.get('/api/option/');\n const { success, message, data } = res.data;\n if (success) {\n let newInputs = {};\n data.forEach((item) => {\n newInputs[item.key] = item.value;\n });\n setInputs({\n ...newInputs,\n EmailDomainWhitelist: newInputs.EmailDomainWhitelist.split(',')\n });\n setOriginInputs(newInputs);\n\n setEmailDomainWhitelist(newInputs.EmailDomainWhitelist.split(','));\n } else {\n showError(message);\n }\n };\n\n useEffect(() => {\n getOptions().then();\n }, []);\n\n const updateOption = async (key, value) => {\n setLoading(true);\n switch (key) {\n case 'PasswordLoginEnabled':\n case 'PasswordRegisterEnabled':\n case 'EmailVerificationEnabled':\n case 'GitHubOAuthEnabled':\n case 'WeChatAuthEnabled':\n case 'TurnstileCheckEnabled':\n case 'EmailDomainRestrictionEnabled':\n case 'RegisterEnabled':\n case 'OidcEnabled':\n value = inputs[key] === 'true' ? 'false' : 'true';\n break;\n default:\n break;\n }\n const res = await API.put('/api/option/', {\n key,\n value\n });\n const { success, message } = res.data;\n if (success) {\n if (key === 'EmailDomainWhitelist') {\n value = value.split(',');\n }\n setInputs((inputs) => ({\n ...inputs,\n [key]: value\n }));\n showSuccess('设置成功！');\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n const handleInputChange = async (event) => {\n let { name, value } = event.target;\n\n if (name === 'PasswordLoginEnabled' && inputs[name] === 'true') {\n // block disabling password login\n setShowPasswordWarningModal(true);\n return;\n }\n if (\n name === 'Notice' ||\n name.startsWith('SMTP') ||\n name === 'ServerAddress' ||\n name === 'GitHubClientId' ||\n name === 'GitHubClientSecret' ||\n name === 'WeChatServerAddress' ||\n name === 'WeChatServerToken' ||\n name === 'WeChatAccountQRCodeImageURL' ||\n name === 'TurnstileSiteKey' ||\n name === 'TurnstileSecretKey' ||\n name === 'EmailDomainWhitelist' ||\n name === 'MessagePusherAddress' ||\n name === 'MessagePusherToken' ||\n name === 'LarkClientId' ||\n name === 'LarkClientSecret' ||\n name === 'OidcClientId' ||\n name === 'OidcClientSecret' ||\n name === 'OidcWellKnown' ||\n name === 'OidcAuthorizationEndpoint' ||\n name === 'OidcTokenEndpoint' ||\n name === 'OidcUserinfoEndpoint'\n )\n {\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n } else {\n await updateOption(name, value);\n }\n };\n\n const submitServerAddress = async () => {\n let ServerAddress = removeTrailingSlash(inputs.ServerAddress);\n await updateOption('ServerAddress', ServerAddress);\n };\n\n const submitSMTP = async () => {\n if (originInputs['SMTPServer'] !== inputs.SMTPServer) {\n await updateOption('SMTPServer', inputs.SMTPServer);\n }\n if (originInputs['SMTPAccount'] !== inputs.SMTPAccount) {\n await updateOption('SMTPAccount', inputs.SMTPAccount);\n }\n if (originInputs['SMTPFrom'] !== inputs.SMTPFrom) {\n await updateOption('SMTPFrom', inputs.SMTPFrom);\n }\n if (originInputs['SMTPPort'] !== inputs.SMTPPort && inputs.SMTPPort !== '') {\n await updateOption('SMTPPort', inputs.SMTPPort);\n }\n if (originInputs['SMTPToken'] !== inputs.SMTPToken && inputs.SMTPToken !== '') {\n await updateOption('SMTPToken', inputs.SMTPToken);\n }\n };\n\n const submitEmailDomainWhitelist = async () => {\n await updateOption('EmailDomainWhitelist', inputs.EmailDomainWhitelist.join(','));\n };\n\n const submitWeChat = async () => {\n if (originInputs['WeChatServerAddress'] !== inputs.WeChatServerAddress) {\n await updateOption('WeChatServerAddress', removeTrailingSlash(inputs.WeChatServerAddress));\n }\n if (originInputs['WeChatAccountQRCodeImageURL'] !== inputs.WeChatAccountQRCodeImageURL) {\n await updateOption('WeChatAccountQRCodeImageURL', inputs.WeChatAccountQRCodeImageURL);\n }\n if (originInputs['WeChatServerToken'] !== inputs.WeChatServerToken && inputs.WeChatServerToken !== '') {\n await updateOption('WeChatServerToken', inputs.WeChatServerToken);\n }\n };\n\n const submitGitHubOAuth = async () => {\n if (originInputs['GitHubClientId'] !== inputs.GitHubClientId) {\n await updateOption('GitHubClientId', inputs.GitHubClientId);\n }\n if (originInputs['GitHubClientSecret'] !== inputs.GitHubClientSecret && inputs.GitHubClientSecret !== '') {\n await updateOption('GitHubClientSecret', inputs.GitHubClientSecret);\n }\n };\n\n const submitTurnstile = async () => {\n if (originInputs['TurnstileSiteKey'] !== inputs.TurnstileSiteKey) {\n await updateOption('TurnstileSiteKey', inputs.TurnstileSiteKey);\n }\n if (originInputs['TurnstileSecretKey'] !== inputs.TurnstileSecretKey && inputs.TurnstileSecretKey !== '') {\n await updateOption('TurnstileSecretKey', inputs.TurnstileSecretKey);\n }\n };\n\n const submitMessagePusher = async () => {\n if (originInputs['MessagePusherAddress'] !== inputs.MessagePusherAddress) {\n await updateOption('MessagePusherAddress', removeTrailingSlash(inputs.MessagePusherAddress));\n }\n if (originInputs['MessagePusherToken'] !== inputs.MessagePusherToken && inputs.MessagePusherToken !== '') {\n await updateOption('MessagePusherToken', inputs.MessagePusherToken);\n }\n };\n\n const submitLarkOAuth = async () => {\n if (originInputs['LarkClientId'] !== inputs.LarkClientId) {\n await updateOption('LarkClientId', inputs.LarkClientId);\n }\n if (originInputs['LarkClientSecret'] !== inputs.LarkClientSecret && inputs.LarkClientSecret !== '') {\n await updateOption('LarkClientSecret', inputs.LarkClientSecret);\n }\n };\n\n const submitOidc = async () => {\n if (inputs.OidcWellKnown !== '') {\n if (!inputs.OidcWellKnown.startsWith('http://') && !inputs.OidcWellKnown.startsWith('https://')) {\n showError('Well-Known URL 必须以 http:// 或 https:// 开头');\n return;\n }\n try {\n const res = await API.get(inputs.OidcWellKnown);\n inputs.OidcAuthorizationEndpoint = res.data['authorization_endpoint'];\n inputs.OidcTokenEndpoint = res.data['token_endpoint'];\n inputs.OidcUserinfoEndpoint = res.data['userinfo_endpoint'];\n showSuccess('获取 OIDC 配置成功！');\n } catch (err) {\n showError(\"获取 OIDC 配置失败，请检查网络状况和 Well-Known URL 是否正确\");\n }\n }\n\n if (originInputs['OidcWellKnown'] !== inputs.OidcWellKnown) {\n await updateOption('OidcWellKnown', inputs.OidcWellKnown);\n }\n if (originInputs['OidcClientId'] !== inputs.OidcClientId) {\n await updateOption('OidcClientId', inputs.OidcClientId);\n }\n if (originInputs['OidcClientSecret'] !== inputs.OidcClientSecret && inputs.OidcClientSecret !== '') {\n await updateOption('OidcClientSecret', inputs.OidcClientSecret);\n }\n if (originInputs['OidcAuthorizationEndpoint'] !== inputs.OidcAuthorizationEndpoint) {\n await updateOption('OidcAuthorizationEndpoint', inputs.OidcAuthorizationEndpoint);\n }\n if (originInputs['OidcTokenEndpoint'] !== inputs.OidcTokenEndpoint) {\n await updateOption('OidcTokenEndpoint', inputs.OidcTokenEndpoint);\n }\n if (originInputs['OidcUserinfoEndpoint'] !== inputs.OidcUserinfoEndpoint) {\n await updateOption('OidcUserinfoEndpoint', inputs.OidcUserinfoEndpoint);\n }\n };\n\n return (\n <>\n \n \n \n \n \n 服务器地址 \n \n \n \n \n 更新服务器地址\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n {\n const event = {\n target: {\n name: 'EmailDomainWhitelist',\n value: value\n }\n };\n handleInputChange(event);\n }}\n filterSelectedOptions\n renderInput={(params) => \n \n \n \n 保存邮箱域名白名单设置\n \n \n \n \n \n \n \n \n SMTP 服务器地址 \n \n \n \n \n SMTP 端口 \n \n \n \n \n SMTP 账户 \n \n \n \n \n SMTP 发送者邮箱 \n \n \n \n \n SMTP 访问凭证 \n \n \n \n \n 保存 SMTP 设置\n \n \n \n \n \n {' '}\n 用以支持通过 GitHub 进行登录注册，\n \n 点击此处\n \n 管理你的 GitHub OAuth App\n \n }\n >\n \n \n \n Homepage URL 填 {inputs.ServerAddress} \n ，Authorization callback URL 填 {`${inputs.ServerAddress}/oauth/github`} \n \n \n \n \n GitHub Client ID \n \n \n \n \n GitHub Client Secret \n \n \n \n \n 保存 GitHub OAuth 设置\n \n \n \n \n \n {' '}\n 用以支持通过飞书进行登录注册，\n \n 点击此处\n \n 管理你的飞书应用\n \n }\n >\n \n \n \n 主页链接填 {inputs.ServerAddress}\n ，重定向 URL 填 {`${inputs.ServerAddress}/oauth/lark`}\n \n \n \n \n App ID \n \n \n \n \n App Secret \n \n \n \n \n 保存飞书 OAuth 设置\n \n \n \n \n \n 用以支持通过微信进行登录注册，\n \n 点击此处\n \n 了解 WeChat Server\n \n }\n >\n \n \n \n WeChat Server 服务器地址 \n \n \n \n \n WeChat Server 访问凭证 \n \n \n \n \n 微信公众号二维码图片链接 \n \n \n \n \n 保存 WeChat Server 设置\n \n \n \n \n\n \n 用以支持通过 OIDC 登录，例如 Okta、Auth0 等兼容 OIDC 协议的 IdP\n \n }\n >\n \n \n \n 主页链接填 { inputs.ServerAddress }\n ，重定向 URL 填 { `${ inputs.ServerAddress }/oauth/oidc` }\n \n 若你的 OIDC Provider 支持 Discovery Endpoint，你可以仅填写 OIDC Well-Known URL，系统会自动获取 OIDC 配置\n \n \n \n \n Client ID \n \n \n \n \n Client Secret \n \n \n \n \n Well-Known URL \n \n \n \n \n Authorization Endpoint \n \n \n \n \n Token Endpoint \n \n \n \n \n Userinfo Endpoint \n \n \n \n \n 保存 OIDC 设置\n \n \n \n \n\n \n 用以推送报警信息，\n \n 点击此处\n \n 了解 Message Pusher\n \n }\n >\n \n \n \n Message Pusher 推送地址 \n \n \n \n \n Message Pusher 访问凭证 \n \n \n \n \n 保存 Message Pusher 设置\n \n \n \n \n \n 用以支持用户校验，\n \n 点击此处\n \n 管理你的 Turnstile Sites，推荐选择 Invisible Widget Type\n \n }\n >\n \n \n \n Turnstile Site Key \n \n \n \n \n Turnstile Secret Key \n \n \n \n \n 保存 Turnstile 设置\n \n \n \n \n \n setShowPasswordWarningModal(false)} maxWidth={'md'}>\n \n 警告\n \n 取消密码登录将导致所有未绑定其他登录方式的用户（包括管理员）无法通过密码登录，确认取消？ \n \n setShowPasswordWarningModal(false)}>取消 \n {\n setShowPasswordWarningModal(false);\n await updateOption('PasswordLoginEnabled', 'false');\n }}\n >\n 确定\n \n \n \n \n );\n};\n\nexport default SystemSetting;\n"
},
{
"path": "web/berry/src/views/Setting/index.js",
"content": "import { useState, useEffect } from 'react';\nimport PropTypes from 'prop-types';\nimport { Tabs, Tab, Box, Card } from '@mui/material';\nimport { IconSettings2, IconActivity, IconSettings } from '@tabler/icons-react';\nimport OperationSetting from './component/OperationSetting';\nimport SystemSetting from './component/SystemSetting';\nimport OtherSetting from './component/OtherSetting';\nimport AdminContainer from 'ui-component/AdminContainer';\nimport { useLocation, useNavigate } from 'react-router-dom';\n\nfunction CustomTabPanel(props) {\n const { children, value, index, ...other } = props;\n\n return (\n \n {value === index && {children} }\n
\n );\n}\n\nCustomTabPanel.propTypes = {\n children: PropTypes.node,\n index: PropTypes.number.isRequired,\n value: PropTypes.number.isRequired\n};\n\nfunction a11yProps(index) {\n return {\n id: `setting-tab-${index}`,\n 'aria-controls': `setting-tabpanel-${index}`\n };\n}\n\nconst Setting = () => {\n const location = useLocation();\n const navigate = useNavigate();\n const hash = location.hash.replace('#', '');\n const tabMap = {\n operation: 0,\n system: 1,\n other: 2\n };\n const [value, setValue] = useState(tabMap[hash] || 0);\n\n const handleChange = (event, newValue) => {\n setValue(newValue);\n const hashArray = Object.keys(tabMap);\n navigate(`#${hashArray[newValue]}`);\n };\n\n useEffect(() => {\n const handleHashChange = () => {\n const hash = location.hash.replace('#', '');\n setValue(tabMap[hash] || 0);\n };\n window.addEventListener('hashchange', handleHashChange);\n return () => {\n window.removeEventListener('hashchange', handleHashChange);\n };\n }, [location, tabMap]);\n\n return (\n <>\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n );\n};\n\nexport default Setting;\n"
},
{
"path": "web/berry/src/views/Token/component/EditModal.js",
"content": "import PropTypes from 'prop-types';\nimport * as Yup from 'yup';\nimport { Formik } from 'formik';\nimport { useTheme } from '@mui/material/styles';\nimport { useState, useEffect } from 'react';\nimport dayjs from 'dayjs';\nimport {\n Dialog,\n DialogTitle,\n DialogContent,\n DialogActions,\n Button,\n Divider,\n Alert,\n FormControl,\n InputLabel,\n OutlinedInput,\n InputAdornment,\n Autocomplete,\n Checkbox,\n TextField,\n Switch,\n FormHelperText\n} from '@mui/material';\n\nimport { AdapterDayjs } from '@mui/x-date-pickers/AdapterDayjs';\nimport { LocalizationProvider } from '@mui/x-date-pickers/LocalizationProvider';\nimport { DateTimePicker } from '@mui/x-date-pickers/DateTimePicker';\nimport { renderQuotaWithPrompt, showSuccess, showError } from 'utils/common';\nimport { API } from 'utils/api';\nimport CheckBoxOutlineBlankIcon from '@mui/icons-material/CheckBoxOutlineBlank';\nimport CheckBoxIcon from '@mui/icons-material/CheckBox';\nimport { createFilterOptions } from '@mui/material/Autocomplete';\nrequire('dayjs/locale/zh-cn');\nconst icon = \n \n {tokenId ? '编辑令牌' : '新建令牌'}\n \n \n 注意，令牌的额度仅用于限制令牌本身的最大额度使用量，实际的使用受到账户的剩余额度限制。 \n \n {({ errors, handleBlur, handleChange, handleSubmit, touched, values, setFieldError, setFieldValue, isSubmitting }) => (\n \n )}\n \n \n \n );\n};\n\nexport default EditModal;\n\nEditModal.propTypes = {\n open: PropTypes.bool,\n tokenId: PropTypes.number,\n onCancel: PropTypes.func,\n onOk: PropTypes.func\n};\n"
},
{
"path": "web/berry/src/views/Token/component/TableHead.js",
"content": "import { TableCell, TableHead, TableRow } from '@mui/material';\n\nconst TokenTableHead = () => {\n return (\n \n \n 名称 \n 状态 \n 已用额度 \n 剩余额度 \n 创建时间 \n 过期时间 \n 操作 \n \n \n );\n};\n\nexport default TokenTableHead;\n"
},
{
"path": "web/berry/src/views/Token/component/TableRow.js",
"content": "import PropTypes from 'prop-types';\nimport { useState } from 'react';\nimport { useSelector } from 'react-redux';\n\nimport {\n Popover,\n TableRow,\n MenuItem,\n TableCell,\n IconButton,\n Dialog,\n DialogActions,\n DialogContent,\n DialogContentText,\n DialogTitle,\n Button,\n Tooltip,\n Stack,\n ButtonGroup\n} from '@mui/material';\n\nimport TableSwitch from 'ui-component/Switch';\nimport { renderQuota, timestamp2string, copy } from 'utils/common';\n\nimport { IconDotsVertical, IconEdit, IconTrash, IconCaretDownFilled } from '@tabler/icons-react';\n\nconst COPY_OPTIONS = [\n {\n key: 'next',\n text: 'ChatGPT Next',\n url: 'https://app.nextchat.dev/#/?settings={\"key\":\"sk-{key}\",\"url\":\"{serverAddress}\"}',\n encode: false\n },\n { key: 'ama', text: 'BotGem', url: 'ama://set-api-key?server={serverAddress}&key=sk-{key}', encode: true },\n { key: 'opencat', text: 'OpenCat', url: 'opencat://team/join?domain={serverAddress}&token=sk-{key}', encode: true },\n { key: 'lobechat', text: 'LobeChat', url: 'https://lobehub.com/?settings={\"keyVaults\":{\"openai\":{\"apiKey\":\"sk-{key}\",\"baseURL\":\"{serverAddress}\"}}}', encode: true }\n];\n\nfunction replacePlaceholders(text, key, serverAddress) {\n return text.replace('{key}', key).replace('{serverAddress}', serverAddress);\n}\n\nfunction createMenu(menuItems) {\n return (\n <>\n {menuItems.map((menuItem, index) => (\n \n {menuItem.icon}\n {menuItem.text}\n \n ))}\n \n );\n}\n\nexport default function TokensTableRow({ item, manageToken, handleOpenModal, setModalTokenId }) {\n const [open, setOpen] = useState(null);\n const [menuItems, setMenuItems] = useState(null);\n const [openDelete, setOpenDelete] = useState(false);\n const [statusSwitch, setStatusSwitch] = useState(item.status);\n const siteInfo = useSelector((state) => state.siteInfo);\n\n const handleDeleteOpen = () => {\n handleCloseMenu();\n setOpenDelete(true);\n };\n\n const handleDeleteClose = () => {\n setOpenDelete(false);\n };\n\n const handleOpenMenu = (event, type) => {\n switch (type) {\n case 'copy':\n setMenuItems(copyItems);\n break;\n case 'link':\n setMenuItems(linkItems);\n break;\n default:\n setMenuItems(actionItems);\n }\n setOpen(event.currentTarget);\n };\n\n const handleCloseMenu = () => {\n setOpen(null);\n };\n\n const handleStatus = async () => {\n const switchVlue = statusSwitch === 1 ? 2 : 1;\n const { success } = await manageToken(item.id, 'status', switchVlue);\n if (success) {\n setStatusSwitch(switchVlue);\n }\n };\n\n const handleDelete = async () => {\n handleCloseMenu();\n await manageToken(item.id, 'delete', '');\n };\n\n const actionItems = createMenu([\n {\n text: '编辑',\n icon: \n {item.name} \n\n \n {\n switch (statusSwitch) {\n case 1:\n return '已启用';\n case 2:\n return '已禁用';\n case 3:\n return '已过期';\n case 4:\n return '已耗尽';\n default:\n return '未知';\n }\n })()}\n placement=\"top\"\n >\n \n\n {renderQuota(item.used_quota)} \n\n {item.unlimited_quota ? '无限制' : renderQuota(item.remain_quota, 2)} \n\n {timestamp2string(item.created_time)} \n\n {item.expired_time === -1 ? '永不过期' : timestamp2string(item.expired_time)} \n\n \n \n \n {\n copy(`sk-${item.key}`);\n }}\n >\n 复制\n \n handleOpenMenu(e, 'copy')}>\n \n \n \n handleCopy(COPY_OPTIONS[0], 'link')}>\n 聊天\n \n handleOpenMenu(e, 'link')}>\n \n \n handleOpenMenu(e, 'action')} sx={{ color: 'rgb(99, 115, 129)' }}>\n \n \n \n \n \n {menuItems}\n \n\n \n 删除Token \n \n 是否删除Token {item.name}？ \n \n \n 关闭 \n \n 删除\n \n \n \n \n );\n}\n\nTokensTableRow.propTypes = {\n item: PropTypes.object,\n manageToken: PropTypes.func,\n handleOpenModal: PropTypes.func,\n setModalTokenId: PropTypes.func\n};\n"
},
{
"path": "web/berry/src/views/Token/index.js",
"content": "import { useState, useEffect } from 'react';\nimport { showError, showSuccess } from 'utils/common';\n\nimport Table from '@mui/material/Table';\nimport TableBody from '@mui/material/TableBody';\nimport TableContainer from '@mui/material/TableContainer';\nimport PerfectScrollbar from 'react-perfect-scrollbar';\nimport TablePagination from '@mui/material/TablePagination';\nimport LinearProgress from '@mui/material/LinearProgress';\nimport Alert from '@mui/material/Alert';\nimport ButtonGroup from '@mui/material/ButtonGroup';\nimport Toolbar from '@mui/material/Toolbar';\n\nimport { Button, Card, Box, Stack, Container, Typography } from '@mui/material';\nimport TokensTableRow from './component/TableRow';\nimport TokenTableHead from './component/TableHead';\nimport TableToolBar from 'ui-component/TableToolBar';\nimport { API } from 'utils/api';\nimport { ITEMS_PER_PAGE } from 'constants';\nimport { IconRefresh, IconPlus } from '@tabler/icons-react';\nimport EditeModal from './component/EditModal';\nimport { useSelector } from 'react-redux';\n\nexport default function Token() {\n const [tokens, setTokens] = useState([]);\n const [activePage, setActivePage] = useState(0);\n const [searching, setSearching] = useState(false);\n const [searchKeyword, setSearchKeyword] = useState('');\n const [openModal, setOpenModal] = useState(false);\n const [editTokenId, setEditTokenId] = useState(0);\n const siteInfo = useSelector((state) => state.siteInfo);\n\n const loadTokens = async (startIdx) => {\n setSearching(true);\n const res = await API.get(`/api/token/?p=${startIdx}`);\n const { success, message, data } = res.data;\n if (success) {\n if (startIdx === 0) {\n setTokens(data);\n } else {\n let newTokens = [...tokens];\n newTokens.splice(startIdx * ITEMS_PER_PAGE, data.length, ...data);\n setTokens(newTokens);\n }\n } else {\n showError(message);\n }\n setSearching(false);\n };\n\n useEffect(() => {\n loadTokens(0)\n .then()\n .catch((reason) => {\n showError(reason);\n });\n }, []);\n\n const onPaginationChange = (event, activePage) => {\n (async () => {\n if (activePage === Math.ceil(tokens.length / ITEMS_PER_PAGE)) {\n // In this case we have to load more data and then append them.\n await loadTokens(activePage);\n }\n setActivePage(activePage);\n })();\n };\n\n const searchTokens = async (event) => {\n event.preventDefault();\n if (searchKeyword === '') {\n await loadTokens(0);\n setActivePage(0);\n return;\n }\n setSearching(true);\n const res = await API.get(`/api/token/search?keyword=${searchKeyword}`);\n const { success, message, data } = res.data;\n if (success) {\n setTokens(data);\n setActivePage(0);\n } else {\n showError(message);\n }\n setSearching(false);\n };\n\n const handleSearchKeyword = (event) => {\n setSearchKeyword(event.target.value);\n };\n\n const manageToken = async (id, action, value) => {\n const url = '/api/token/';\n let data = { id };\n let res;\n switch (action) {\n case 'delete':\n res = await API.delete(url + id);\n break;\n case 'status':\n res = await API.put(url + `?status_only=true`, {\n ...data,\n status: value\n });\n break;\n }\n const { success, message } = res.data;\n if (success) {\n showSuccess('操作成功完成！');\n if (action === 'delete') {\n await handleRefresh();\n }\n } else {\n showError(message);\n }\n\n return res.data;\n };\n\n // 处理刷新\n const handleRefresh = async () => {\n await loadTokens(activePage);\n };\n\n const handleOpenModal = (tokenId) => {\n setEditTokenId(tokenId);\n setOpenModal(true);\n };\n\n const handleCloseModal = () => {\n setOpenModal(false);\n setEditTokenId(0);\n };\n\n const handleOkModal = (status) => {\n if (status === true) {\n handleCloseModal();\n handleRefresh();\n }\n };\n\n return (\n <>\n \n 令牌 \n {\n handleOpenModal(0);\n }}\n startIcon={ \n \n \n 将 OpenAI API 基础地址 https://api.openai.com 替换为 {siteInfo.server_address} ，复制下面的密钥即可使用\n \n \n \n \n \n theme.spacing(0, 1, 0, 3)\n }}\n >\n \n \n \n \n \n {searching && \n \n \n \n {tokens.slice(activePage * ITEMS_PER_PAGE, (activePage + 1) * ITEMS_PER_PAGE).map((row) => (\n \n
\n \n \n \n \n \n \n \n \n \n \n \n \n \n 邀请奖励\n \n \n 分享您的邀请链接，邀请好友注册，即可获得奖励！\n \n\n \n \n {inviteUl ? '复制' : '生成'}\n \n \n }\n aria-describedby=\"helper-text-channel-quota-label\"\n disabled={true}\n />\n \n \n \n );\n};\n\nexport default InviteCard;\n"
},
{
"path": "web/berry/src/views/Topup/component/TopupCard.js",
"content": "import { Typography, Stack, OutlinedInput, InputAdornment, Button, InputLabel, FormControl } from '@mui/material';\nimport { IconWallet } from '@tabler/icons-react';\nimport { useTheme } from '@mui/material/styles';\nimport SubCard from 'ui-component/cards/SubCard';\nimport UserCard from 'ui-component/cards/UserCard';\n\nimport { API } from 'utils/api';\nimport React, { useEffect, useState } from 'react';\nimport { showError, showInfo, showSuccess, renderQuota } from 'utils/common';\n\nconst TopupCard = () => {\n const theme = useTheme();\n const [redemptionCode, setRedemptionCode] = useState('');\n const [topUpLink, setTopUpLink] = useState('');\n const [userQuota, setUserQuota] = useState(0);\n const [isSubmitting, setIsSubmitting] = useState(false);\n\n const topUp = async () => {\n if (redemptionCode === '') {\n showInfo('请输入充值码！');\n return;\n }\n setIsSubmitting(true);\n try {\n const res = await API.post('/api/user/topup', {\n key: redemptionCode\n });\n const { success, message, data } = res.data;\n if (success) {\n showSuccess('充值成功！');\n setUserQuota((quota) => {\n return quota + data;\n });\n setRedemptionCode('');\n } else {\n showError(message);\n }\n } catch (err) {\n showError('请求失败');\n } finally {\n setIsSubmitting(false);\n }\n };\n\n const openTopUpLink = () => {\n if (!topUpLink) {\n showError('超级管理员未设置充值链接！');\n return;\n }\n window.open(topUpLink, '_blank');\n };\n\n const getUserQuota = async () => {\n let res = await API.get(`/api/user/self`);\n const { success, message, data } = res.data;\n if (success) {\n setUserQuota(data.quota);\n } else {\n showError(message);\n }\n };\n\n useEffect(() => {\n let status = localStorage.getItem('siteInfo');\n if (status) {\n status = JSON.parse(status);\n if (status.top_up_link) {\n setTopUpLink(status.top_up_link);\n }\n }\n getUserQuota().then();\n }, []);\n\n return (\n \n \n 当前额度: \n {renderQuota(userQuota)} \n \n \n \n 兑换码 \n {\n setRedemptionCode(e.target.value);\n }}\n name=\"key\"\n placeholder=\"请输入兑换码\"\n endAdornment={\n \n \n {isSubmitting ? '兑换中...' : '兑换'}\n \n \n }\n aria-describedby=\"helper-text-channel-quota-label\"\n />\n \n\n \n \n 还没有兑换码？ 点击获取兑换码：\n \n \n 获取兑换码\n \n \n \n \n );\n};\n\nexport default TopupCard;\n"
},
{
"path": "web/berry/src/views/Topup/index.js",
"content": "import { Stack, Alert } from '@mui/material';\nimport Grid from '@mui/material/Unstable_Grid2';\nimport TopupCard from './component/TopupCard';\nimport InviteCard from './component/InviteCard';\n\nconst Topup = () => {\n return (\n \n \n \n 充值记录以及邀请记录请在日志中查询。充值记录请在日志中选择类型【充值】查询；邀请记录请在日志中选择【系统】查询{' '}\n \n \n \n \n \n \n \n \n \n );\n};\n\nexport default Topup;\n"
},
{
"path": "web/berry/src/views/User/component/EditModal.js",
"content": "import PropTypes from 'prop-types';\nimport * as Yup from 'yup';\nimport { Formik } from 'formik';\nimport { useTheme } from '@mui/material/styles';\nimport { useState, useEffect } from 'react';\nimport {\n Dialog,\n DialogTitle,\n DialogContent,\n DialogActions,\n Button,\n Divider,\n FormControl,\n InputLabel,\n OutlinedInput,\n InputAdornment,\n Select,\n MenuItem,\n IconButton,\n FormHelperText\n} from '@mui/material';\n\nimport Visibility from '@mui/icons-material/Visibility';\nimport VisibilityOff from '@mui/icons-material/VisibilityOff';\n\nimport { renderQuotaWithPrompt, showSuccess, showError } from 'utils/common';\nimport { API } from 'utils/api';\n\nconst validationSchema = Yup.object().shape({\n is_edit: Yup.boolean(),\n username: Yup.string().required('用户名 不能为空'),\n display_name: Yup.string(),\n password: Yup.string().when('is_edit', {\n is: false,\n then: Yup.string().required('密码 不能为空'),\n otherwise: Yup.string()\n }),\n group: Yup.string().when('is_edit', {\n is: false,\n then: Yup.string().required('用户组 不能为空'),\n otherwise: Yup.string()\n }),\n quota: Yup.number().when('is_edit', {\n is: false,\n then: Yup.number().min(0, '额度 不能小于 0'),\n otherwise: Yup.number()\n })\n});\n\nconst originInputs = {\n is_edit: false,\n username: '',\n display_name: '',\n password: '',\n group: 'default',\n quota: 0\n};\n\nconst EditModal = ({ open, userId, onCancel, onOk }) => {\n const theme = useTheme();\n const [inputs, setInputs] = useState(originInputs);\n const [groupOptions, setGroupOptions] = useState([]);\n const [showPassword, setShowPassword] = useState(false);\n\n const submit = async (values, { setErrors, setStatus, setSubmitting }) => {\n setSubmitting(true);\n\n let res;\n if (values.is_edit) {\n res = await API.put(`/api/user/`, { ...values, id: parseInt(userId) });\n } else {\n res = await API.post(`/api/user/`, values);\n }\n const { success, message } = res.data;\n if (success) {\n if (values.is_edit) {\n showSuccess('用户更新成功！');\n } else {\n showSuccess('用户创建成功！');\n }\n setSubmitting(false);\n setStatus({ success: true });\n onOk(true);\n } else {\n showError(message);\n setErrors({ submit: message });\n }\n };\n\n const handleClickShowPassword = () => {\n setShowPassword(!showPassword);\n };\n\n const handleMouseDownPassword = (event) => {\n event.preventDefault();\n };\n\n const loadUser = async () => {\n let res = await API.get(`/api/user/${userId}`);\n const { success, message, data } = res.data;\n if (success) {\n data.is_edit = true;\n setInputs(data);\n } else {\n showError(message);\n }\n };\n\n const fetchGroups = async () => {\n try {\n let res = await API.get(`/api/group/`);\n setGroupOptions(res.data.data);\n } catch (error) {\n showError(error.message);\n }\n };\n\n useEffect(() => {\n fetchGroups().then();\n if (userId) {\n loadUser().then();\n } else {\n setInputs(originInputs);\n }\n }, [userId]);\n\n return (\n \n \n {userId ? '编辑用户' : '新建用户'}\n \n \n \n {({ errors, handleBlur, handleChange, handleSubmit, touched, values, isSubmitting }) => (\n \n )}\n \n \n \n );\n};\n\nexport default EditModal;\n\nEditModal.propTypes = {\n open: PropTypes.bool,\n userId: PropTypes.number,\n onCancel: PropTypes.func,\n onOk: PropTypes.func\n};\n"
},
{
"path": "web/berry/src/views/User/component/TableHead.js",
"content": "import { TableCell, TableHead, TableRow } from '@mui/material';\n\nconst UsersTableHead = () => {\n return (\n \n \n ID \n 用户名 \n 分组 \n 统计信息 \n 用户角色 \n 绑定 \n 状态 \n 操作 \n \n \n );\n};\n\nexport default UsersTableHead;\n"
},
{
"path": "web/berry/src/views/User/component/TableRow.js",
"content": "import PropTypes from 'prop-types';\nimport { useState } from 'react';\n\nimport {\n Popover,\n TableRow,\n MenuItem,\n TableCell,\n IconButton,\n Dialog,\n DialogActions,\n DialogContent,\n DialogContentText,\n DialogTitle,\n Button,\n Tooltip,\n Stack\n} from '@mui/material';\n\nimport Label from 'ui-component/Label';\nimport TableSwitch from 'ui-component/Switch';\nimport { renderQuota, renderNumber } from 'utils/common';\nimport { IconDotsVertical, IconEdit, IconTrash, IconUser, IconBrandWechat, IconBrandGithub, IconMail } from '@tabler/icons-react';\nimport { useTheme } from '@mui/material/styles';\n\nfunction renderRole(role) {\n switch (role) {\n case 1:\n return 普通用户 ;\n case 10:\n return 管理员 ;\n case 100:\n return 超级管理员 ;\n default:\n return 未知身份 ;\n }\n}\n\nexport default function UsersTableRow({ item, manageUser, handleOpenModal, setModalUserId }) {\n const theme = useTheme();\n const [open, setOpen] = useState(null);\n const [openDelete, setOpenDelete] = useState(false);\n const [statusSwitch, setStatusSwitch] = useState(item.status);\n\n const handleDeleteOpen = () => {\n handleCloseMenu();\n setOpenDelete(true);\n };\n\n const handleDeleteClose = () => {\n setOpenDelete(false);\n };\n\n const handleOpenMenu = (event) => {\n setOpen(event.currentTarget);\n };\n\n const handleCloseMenu = () => {\n setOpen(null);\n };\n\n const handleStatus = async () => {\n const switchVlue = statusSwitch === 1 ? 2 : 1;\n const { success } = await manageUser(item.username, 'status', switchVlue);\n if (success) {\n setStatusSwitch(switchVlue);\n }\n };\n\n const handleDelete = async () => {\n handleCloseMenu();\n await manageUser(item.username, 'delete', '');\n };\n\n return (\n <>\n \n {item.id} \n\n {item.username} \n\n \n {item.group} \n \n\n \n \n \n \n {' '}\n {renderQuota(item.quota)}{' '}\n \n \n \n \n {' '}\n {renderQuota(item.used_quota)}{' '}\n \n \n \n \n {' '}\n {renderNumber(item.request_count)}{' '}\n \n \n \n \n {renderRole(item.role)} \n \n \n \n \n \n \n \n \n \n \n\n \n {' '}\n \n \n \n \n \n \n\n \n {item.role !== 100 && (\n {\n handleCloseMenu();\n manageUser(item.username, 'role', item.role === 1 ? true : false);\n }}\n >\n {\n handleCloseMenu();\n handleOpenModal();\n setModalUserId(item.id);\n }}\n >\n \n \n \n\n \n 删除用户 \n \n 是否删除用户 {item.name}？ \n \n \n 关闭 \n \n 删除\n \n \n \n \n );\n}\n\nUsersTableRow.propTypes = {\n item: PropTypes.object,\n manageUser: PropTypes.func,\n handleOpenModal: PropTypes.func,\n setModalUserId: PropTypes.func\n};\n"
},
{
"path": "web/berry/src/views/User/index.js",
"content": "import { useState, useEffect } from 'react';\nimport { showError, showSuccess } from 'utils/common';\n\nimport Table from '@mui/material/Table';\nimport TableBody from '@mui/material/TableBody';\nimport TableContainer from '@mui/material/TableContainer';\nimport PerfectScrollbar from 'react-perfect-scrollbar';\nimport TablePagination from '@mui/material/TablePagination';\nimport LinearProgress from '@mui/material/LinearProgress';\nimport ButtonGroup from '@mui/material/ButtonGroup';\nimport Toolbar from '@mui/material/Toolbar';\n\nimport { Button, Card, Box, Stack, Container, Typography } from '@mui/material';\nimport UsersTableRow from './component/TableRow';\nimport UsersTableHead from './component/TableHead';\nimport TableToolBar from 'ui-component/TableToolBar';\nimport { API } from 'utils/api';\nimport { ITEMS_PER_PAGE } from 'constants';\nimport { IconRefresh, IconPlus } from '@tabler/icons-react';\nimport EditeModal from './component/EditModal';\n\n// ----------------------------------------------------------------------\nexport default function Users() {\n const [users, setUsers] = useState([]);\n const [activePage, setActivePage] = useState(0);\n const [searching, setSearching] = useState(false);\n const [searchKeyword, setSearchKeyword] = useState('');\n const [openModal, setOpenModal] = useState(false);\n const [editUserId, setEditUserId] = useState(0);\n\n const loadUsers = async (startIdx) => {\n setSearching(true);\n const res = await API.get(`/api/user/?p=${startIdx}`);\n const { success, message, data } = res.data;\n if (success) {\n if (startIdx === 0) {\n setUsers(data);\n } else {\n let newUsers = [...users];\n newUsers.splice(startIdx * ITEMS_PER_PAGE, data.length, ...data);\n setUsers(newUsers);\n }\n } else {\n showError(message);\n }\n setSearching(false);\n };\n\n const onPaginationChange = (event, activePage) => {\n (async () => {\n if (activePage === Math.ceil(users.length / ITEMS_PER_PAGE)) {\n // In this case we have to load more data and then append them.\n await loadUsers(activePage);\n }\n setActivePage(activePage);\n })();\n };\n\n const searchUsers = async (event) => {\n event.preventDefault();\n if (searchKeyword === '') {\n await loadUsers(0);\n setActivePage(0);\n return;\n }\n setSearching(true);\n const res = await API.get(`/api/user/search?keyword=${searchKeyword}`);\n const { success, message, data } = res.data;\n if (success) {\n setUsers(data);\n setActivePage(0);\n } else {\n showError(message);\n }\n setSearching(false);\n };\n\n const handleSearchKeyword = (event) => {\n setSearchKeyword(event.target.value);\n };\n\n const manageUser = async (username, action, value) => {\n const url = '/api/user/manage';\n let data = { username: username, action: '' };\n let res;\n switch (action) {\n case 'delete':\n data.action = 'delete';\n break;\n case 'status':\n data.action = value === 1 ? 'enable' : 'disable';\n break;\n case 'role':\n data.action = value === true ? 'promote' : 'demote';\n break;\n }\n\n res = await API.post(url, data);\n const { success, message } = res.data;\n if (success) {\n showSuccess('操作成功完成！');\n await loadUsers(activePage);\n } else {\n showError(message);\n }\n\n return res.data;\n };\n\n // 处理刷新\n const handleRefresh = async () => {\n await loadUsers(activePage);\n };\n\n const handleOpenModal = (userId) => {\n setEditUserId(userId);\n setOpenModal(true);\n };\n\n const handleCloseModal = () => {\n setOpenModal(false);\n setEditUserId(0);\n };\n\n const handleOkModal = (status) => {\n if (status === true) {\n handleCloseModal();\n handleRefresh();\n }\n };\n\n useEffect(() => {\n loadUsers(0)\n .then()\n .catch((reason) => {\n showError(reason);\n });\n }, []);\n\n return (\n <>\n \n 用户 \n\n \n \n \n \n theme.spacing(0, 1, 0, 3)\n }}\n >\n \n \n \n \n \n {searching && \n \n \n \n {users.slice(activePage * ITEMS_PER_PAGE, (activePage + 1) * ITEMS_PER_PAGE).map((row) => (\n \n
\n \n \n \n One API \n \n \n You need to enable JavaScript to run this app. \n
\n \n\n"
},
{
"path": "web/default/public/robots.txt",
"content": "# https://www.robotstxt.org/robotstxt.html\nUser-agent: *\nDisallow:\n"
},
{
"path": "web/default/src/App.js",
"content": "import React, { lazy, Suspense, useContext, useEffect } from 'react';\nimport { Route, Routes } from 'react-router-dom';\nimport Loading from './components/Loading';\nimport User from './pages/User';\nimport { PrivateRoute } from './components/PrivateRoute';\nimport RegisterForm from './components/RegisterForm';\nimport LoginForm from './components/LoginForm';\nimport NotFound from './pages/NotFound';\nimport Setting from './pages/Setting';\nimport EditUser from './pages/User/EditUser';\nimport AddUser from './pages/User/AddUser';\nimport { API, getLogo, getSystemName, showError, showNotice } from './helpers';\nimport PasswordResetForm from './components/PasswordResetForm';\nimport GitHubOAuth from './components/GitHubOAuth';\nimport PasswordResetConfirm from './components/PasswordResetConfirm';\nimport { UserContext } from './context/User';\nimport { StatusContext } from './context/Status';\nimport Channel from './pages/Channel';\nimport Token from './pages/Token';\nimport EditToken from './pages/Token/EditToken';\nimport EditChannel from './pages/Channel/EditChannel';\nimport Redemption from './pages/Redemption';\nimport EditRedemption from './pages/Redemption/EditRedemption';\nimport TopUp from './pages/TopUp';\nimport Log from './pages/Log';\nimport Chat from './pages/Chat';\nimport LarkOAuth from './components/LarkOAuth';\nimport Dashboard from './pages/Dashboard';\n\nconst Home = lazy(() => import('./pages/Home'));\nconst About = lazy(() => import('./pages/About'));\n\nfunction App() {\n const [userState, userDispatch] = useContext(UserContext);\n const [statusState, statusDispatch] = useContext(StatusContext);\n\n const loadUser = () => {\n let user = localStorage.getItem('user');\n if (user) {\n let data = JSON.parse(user);\n userDispatch({ type: 'login', payload: data });\n }\n };\n const loadStatus = async () => {\n try {\n const res = await API.get('/api/status');\n const { success, message, data } = res.data || {}; // Add default empty object\n if (success && data) {\n // Check data exists\n localStorage.setItem('status', JSON.stringify(data));\n statusDispatch({ type: 'set', payload: data });\n localStorage.setItem('system_name', data.system_name);\n localStorage.setItem('logo', data.logo);\n localStorage.setItem('footer_html', data.footer_html);\n localStorage.setItem('quota_per_unit', data.quota_per_unit);\n localStorage.setItem('display_in_currency', data.display_in_currency);\n if (data.chat_link) {\n localStorage.setItem('chat_link', data.chat_link);\n } else {\n localStorage.removeItem('chat_link');\n }\n if (\n data.version !== process.env.REACT_APP_VERSION &&\n data.version !== 'v0.0.0' &&\n process.env.REACT_APP_VERSION !== ''\n ) {\n showNotice(\n `新版本可用：${data.version}，请使用快捷键 Shift + F5 刷新页面`\n );\n }\n } else {\n showError(message || '无法正常连接至服务器！');\n }\n } catch (error) {\n showError(error.message || '无法正常连接至服务器！');\n }\n };\n\n useEffect(() => {\n loadUser();\n loadStatus().then();\n let systemName = getSystemName();\n if (systemName) {\n document.title = systemName;\n }\n let logo = getLogo();\n if (logo) {\n let linkElement = document.querySelector(\"link[rel~='icon']\");\n if (linkElement) {\n linkElement.href = logo;\n }\n }\n }, []);\n\n return (\n \n }>\n \n }>\n }>\n \n }>\n }>\n \n }>\n }>\n \n }>\n }>\n }>\n }>\n }>\n }>\n }>\n }>\n }>\n \n }>\n \n \n }\n />\n \n }>\n \n \n }\n />\n \n }>\n }>\n \n \n );\n}\n\nexport default App;\n"
},
{
"path": "web/default/src/components/ChannelsTable.js",
"content": "import React, {useEffect, useState} from 'react';\nimport {useTranslation} from 'react-i18next';\nimport {Button, Dropdown, Form, Input, Label, Message, Pagination, Popup, Table,} from 'semantic-ui-react';\nimport {Link} from 'react-router-dom';\nimport {\n API,\n loadChannelModels,\n setPromptShown,\n shouldShowPrompt,\n showError,\n showInfo,\n showSuccess,\n timestamp2string,\n} from '../helpers';\n\nimport {CHANNEL_OPTIONS, ITEMS_PER_PAGE} from '../constants';\nimport {renderGroup, renderNumber} from '../helpers/render';\n\nfunction renderTimestamp(timestamp) {\n return <>{timestamp2string(timestamp)};\n}\n\nlet type2label = undefined;\n\nfunction renderType(type, t) {\n if (!type2label) {\n type2label = new Map();\n for (let i = 0; i < CHANNEL_OPTIONS.length; i++) {\n type2label[CHANNEL_OPTIONS[i].value] = CHANNEL_OPTIONS[i];\n }\n type2label[0] = {\n value: 0,\n text: t('channel.table.status_unknown'),\n color: 'grey',\n };\n }\n return (\n \n {type2label[type] ? type2label[type].text : type}\n \n );\n}\n\nfunction renderBalance(type, balance, t) {\n switch (type) {\n case 1: // OpenAI\n if (balance === 0) {\n return {t('channel.table.balance_not_supported')} ;\n }\n return ${balance.toFixed(2)} ;\n case 4: // CloseAI\n return ¥{balance.toFixed(2)} ;\n case 8: // 自定义\n return ${balance.toFixed(2)} ;\n case 5: // OpenAI-SB\n return ¥{(balance / 10000).toFixed(2)} ;\n case 10: // AI Proxy\n return {renderNumber(balance)} ;\n case 12: // API2GPT\n return ¥{balance.toFixed(2)} ;\n case 13: // AIGC2D\n return {renderNumber(balance)} ;\n case 20: // OpenRouter\n return ${balance.toFixed(2)} ;\n case 36: // DeepSeek\n return ¥{balance.toFixed(2)} ;\n case 44: // SiliconFlow\n return ¥{balance.toFixed(2)} ;\n default:\n return {t('channel.table.balance_not_supported')} ;\n }\n}\n\nfunction isShowDetail() {\n return localStorage.getItem('show_detail') === 'true';\n}\n\nconst promptID = 'detail';\n\nconst ChannelsTable = () => {\n const { t } = useTranslation();\n const [channels, setChannels] = useState([]);\n const [loading, setLoading] = useState(true);\n const [activePage, setActivePage] = useState(1);\n const [searchKeyword, setSearchKeyword] = useState('');\n const [searching, setSearching] = useState(false);\n const [updatingBalance, setUpdatingBalance] = useState(false);\n const [showPrompt, setShowPrompt] = useState(shouldShowPrompt(promptID));\n const [showDetail, setShowDetail] = useState(isShowDetail());\n\n const processChannelData = (channel) => {\n if (channel.models === '') {\n channel.models = [];\n channel.test_model = '';\n } else {\n channel.models = channel.models.split(',');\n if (channel.models.length > 0) {\n channel.test_model = channel.models[0];\n }\n channel.model_options = channel.models.map((model) => {\n return {\n key: model,\n text: model,\n value: model,\n };\n });\n console.log('channel', channel);\n }\n return channel;\n };\n\n const loadChannels = async (startIdx) => {\n const res = await API.get(`/api/channel/?p=${startIdx}`);\n const { success, message, data } = res.data;\n if (success) {\n let localChannels = data.map(processChannelData);\n if (startIdx === 0) {\n setChannels(localChannels);\n } else {\n let newChannels = [...channels];\n newChannels.splice(\n startIdx * ITEMS_PER_PAGE,\n data.length,\n ...localChannels\n );\n setChannels(newChannels);\n }\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n const onPaginationChange = (e, { activePage }) => {\n (async () => {\n if (activePage === Math.ceil(channels.length / ITEMS_PER_PAGE) + 1) {\n // In this case we have to load more data and then append them.\n await loadChannels(activePage - 1);\n }\n setActivePage(activePage);\n })();\n };\n\n const refresh = async () => {\n setLoading(true);\n await loadChannels(activePage - 1);\n };\n\n const toggleShowDetail = () => {\n setShowDetail(!showDetail);\n localStorage.setItem('show_detail', (!showDetail).toString());\n };\n\n useEffect(() => {\n loadChannels(0)\n .then()\n .catch((reason) => {\n showError(reason);\n });\n loadChannelModels().then();\n }, []);\n\n const manageChannel = async (id, action, idx, value) => {\n let data = { id };\n let res;\n switch (action) {\n case 'delete':\n res = await API.delete(`/api/channel/${id}/`);\n break;\n case 'enable':\n data.status = 1;\n res = await API.put('/api/channel/', data);\n break;\n case 'disable':\n data.status = 2;\n res = await API.put('/api/channel/', data);\n break;\n case 'priority':\n if (value === '') {\n return;\n }\n data.priority = parseInt(value);\n res = await API.put('/api/channel/', data);\n break;\n case 'weight':\n if (value === '') {\n return;\n }\n data.weight = parseInt(value);\n if (data.weight < 0) {\n data.weight = 0;\n }\n res = await API.put('/api/channel/', data);\n break;\n }\n const { success, message } = res.data;\n if (success) {\n showSuccess(t('channel.messages.operation_success'));\n let channel = res.data.data;\n let newChannels = [...channels];\n let realIdx = (activePage - 1) * ITEMS_PER_PAGE + idx;\n if (action === 'delete') {\n newChannels[realIdx].deleted = true;\n } else {\n newChannels[realIdx].status = channel.status;\n }\n setChannels(newChannels);\n } else {\n showError(message);\n }\n };\n\n const renderStatus = (status, t) => {\n switch (status) {\n case 1:\n return (\n \n {t('channel.table.status_enabled')}\n \n );\n case 2:\n return (\n \n {t('channel.table.status_disabled')}\n \n }\n content={t('channel.table.status_disabled_tip')}\n basic\n />\n );\n case 3:\n return (\n \n {t('channel.table.status_auto_disabled')}\n \n }\n content={t('channel.table.status_auto_disabled_tip')}\n basic\n />\n );\n default:\n return (\n \n {t('channel.table.status_unknown')}\n \n );\n }\n };\n\n const renderResponseTime = (responseTime, t) => {\n let time = responseTime / 1000;\n time = time.toFixed(2) + 's';\n if (responseTime === 0) {\n return (\n \n {t('channel.table.not_tested')}\n \n );\n } else if (responseTime <= 1000) {\n return (\n \n {time}\n \n );\n } else if (responseTime <= 3000) {\n return (\n \n {time}\n \n );\n } else if (responseTime <= 5000) {\n return (\n \n {time}\n \n );\n } else {\n return (\n \n {time}\n \n );\n }\n };\n\n const searchChannels = async () => {\n if (searchKeyword === '') {\n // if keyword is blank, load files instead.\n await loadChannels(0);\n setActivePage(1);\n return;\n }\n setSearching(true);\n const res = await API.get(`/api/channel/search?keyword=${searchKeyword}`);\n const { success, message, data } = res.data;\n if (success) {\n let localChannels = data.map(processChannelData);\n setChannels(localChannels);\n setActivePage(1);\n } else {\n showError(message);\n }\n setSearching(false);\n };\n\n const switchTestModel = async (idx, model) => {\n let newChannels = [...channels];\n let realIdx = (activePage - 1) * ITEMS_PER_PAGE + idx;\n newChannels[realIdx].test_model = model;\n setChannels(newChannels);\n };\n\n const testChannel = async (id, name, idx, m) => {\n const res = await API.get(`/api/channel/test/${id}?model=${m}`);\n const { success, message, time, model } = res.data;\n if (success) {\n let newChannels = [...channels];\n let realIdx = (activePage - 1) * ITEMS_PER_PAGE + idx;\n newChannels[realIdx].response_time = time * 1000;\n newChannels[realIdx].test_time = Date.now() / 1000;\n setChannels(newChannels);\n showSuccess(\n t('channel.messages.test_success', { name, model, time, message })\n );\n } else {\n showError(message);\n }\n let newChannels = [...channels];\n let realIdx = (activePage - 1) * ITEMS_PER_PAGE + idx;\n newChannels[realIdx].response_time = time * 1000;\n newChannels[realIdx].test_time = Date.now() / 1000;\n setChannels(newChannels);\n };\n\n const testChannels = async (scope) => {\n const res = await API.get(`/api/channel/test?scope=${scope}`);\n const { success, message } = res.data;\n if (success) {\n showInfo(t('channel.messages.test_all_started'));\n } else {\n showError(message);\n }\n };\n\n const deleteAllDisabledChannels = async () => {\n const res = await API.delete(`/api/channel/disabled`);\n const { success, message, data } = res.data;\n if (success) {\n showSuccess(\n t('channel.messages.delete_disabled_success', { count: data })\n );\n await refresh();\n } else {\n showError(message);\n }\n };\n\n const updateChannelBalance = async (id, name, idx) => {\n const res = await API.get(`/api/channel/update_balance/${id}/`);\n const { success, message, balance } = res.data;\n if (success) {\n let newChannels = [...channels];\n let realIdx = (activePage - 1) * ITEMS_PER_PAGE + idx;\n newChannels[realIdx].balance = balance;\n newChannels[realIdx].balance_updated_time = Date.now() / 1000;\n setChannels(newChannels);\n showSuccess(t('channel.messages.balance_update_success', { name }));\n } else {\n showError(message);\n }\n };\n\n const updateAllChannelsBalance = async () => {\n setUpdatingBalance(true);\n const res = await API.get(`/api/channel/update_balance`);\n const { success, message } = res.data;\n if (success) {\n showInfo(t('channel.messages.all_balance_updated'));\n } else {\n showError(message);\n }\n setUpdatingBalance(false);\n };\n\n const handleKeywordChange = async (e, { value }) => {\n setSearchKeyword(value.trim());\n };\n\n const sortChannel = (key) => {\n if (channels.length === 0) return;\n setLoading(true);\n let sortedChannels = [...channels];\n sortedChannels.sort((a, b) => {\n if (!isNaN(a[key])) {\n // If the value is numeric, subtract to sort\n return a[key] - b[key];\n } else {\n // If the value is not numeric, sort as strings\n return ('' + a[key]).localeCompare(b[key]);\n }\n });\n if (sortedChannels[0].id === channels[0].id) {\n sortedChannels.reverse();\n }\n setChannels(sortedChannels);\n setLoading(false);\n };\n\n return (\n <>\n \n {showPrompt && (\n {\n setShowPrompt(false);\n setPromptShown(promptID);\n }}\n >\n {t('channel.balance_notice')}\n \n \n \n {\n sortChannel('id');\n }}\n >\n {t('channel.table.id')}\n \n {\n sortChannel('name');\n }}\n >\n {t('channel.table.name')}\n \n {\n sortChannel('group');\n }}\n >\n {t('channel.table.group')}\n \n {\n sortChannel('type');\n }}\n >\n {t('channel.table.type')}\n \n {\n sortChannel('status');\n }}\n >\n {t('channel.table.status')}\n \n {\n sortChannel('response_time');\n }}\n >\n {t('channel.table.response_time')}\n \n {\n sortChannel('balance');\n }}\n >\n {t('channel.table.balance')}\n \n {\n sortChannel('priority');\n }}\n hidden={!showDetail}\n >\n {t('channel.table.priority')}\n \n \n {t('channel.table.test_model')}\n \n {t('channel.table.actions')} \n \n \n\n \n {channels\n .slice(\n (activePage - 1) * ITEMS_PER_PAGE,\n activePage * ITEMS_PER_PAGE\n )\n .map((channel, idx) => {\n if (channel.deleted) return <>;\n return (\n \n {channel.id} \n \n {channel.name ? channel.name : t('channel.table.no_name')}\n \n {renderGroup(channel.group)} \n {renderType(channel.type, t)} \n {renderStatus(channel.status, t)} \n \n \n \n {\n updateChannelBalance(channel.id, channel.name, idx);\n }}\n style={{ cursor: 'pointer' }}\n >\n {renderBalance(channel.type, channel.balance, t)}\n \n }\n content={t('channel.table.click_to_update')}\n basic\n />\n \n \n {\n manageChannel(\n channel.id,\n 'priority',\n idx,\n event.target.value\n );\n }}\n >\n \n \n {\n switchTestModel(idx, data.value);\n }}\n />\n \n \n \n {\n testChannel(\n channel.id,\n channel.name,\n idx,\n channel.test_model\n );\n }}\n >\n {t('channel.buttons.test')}\n \n \n {t('channel.buttons.delete')}\n \n }\n on='click'\n flowing\n hoverable\n >\n {\n manageChannel(channel.id, 'delete', idx);\n }}\n >\n {t('channel.buttons.confirm_delete')} {channel.name}\n \n \n {\n manageChannel(\n channel.id,\n channel.status === 1 ? 'disable' : 'enable',\n idx\n );\n }}\n >\n {channel.status === 1\n ? t('channel.buttons.disable')\n : t('channel.buttons.enable')}\n \n \n {t('channel.buttons.edit')}\n \n \n \n \n );\n })}\n \n\n \n \n \n \n {t('channel.buttons.add')}\n \n {\n testChannels('all');\n }}\n >\n {t('channel.buttons.test_all')}\n \n {\n testChannels('disabled');\n }}\n >\n {t('channel.buttons.test_disabled')}\n \n \n {t('channel.buttons.delete_disabled')}\n \n }\n on='click'\n flowing\n hoverable\n >\n \n {t('channel.buttons.confirm_delete_disabled')}\n \n \n \n {t('channel.buttons.refresh')}\n \n \n {showDetail\n ? t('channel.buttons.hide_detail')\n : t('channel.buttons.show_detail')}\n \n \n \n \n
\n \n );\n};\n\nexport default ChannelsTable;\n"
},
{
"path": "web/default/src/components/Footer.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { useTranslation } from 'react-i18next';\nimport { Container, Segment } from 'semantic-ui-react';\nimport { getFooterHTML, getSystemName } from '../helpers';\n\nconst Footer = () => {\n const { t } = useTranslation();\n const systemName = getSystemName();\n const [footer, setFooter] = useState(getFooterHTML());\n let remainCheckTimes = 5;\n\n const loadFooter = () => {\n let footer_html = localStorage.getItem('footer_html');\n if (footer_html) {\n setFooter(footer_html);\n }\n };\n\n useEffect(() => {\n const timer = setInterval(() => {\n if (remainCheckTimes <= 0) {\n clearInterval(timer);\n return;\n }\n remainCheckTimes--;\n loadFooter();\n }, 200);\n return () => clearTimeout(timer);\n }, []);\n\n return (\n \n \n {footer ? (\n \n ) : (\n \n )}\n \n \n );\n};\n\nexport default Footer;\n"
},
{
"path": "web/default/src/components/GitHubOAuth.js",
"content": "import React, { useContext, useEffect, useState } from 'react';\nimport { Dimmer, Loader, Segment } from 'semantic-ui-react';\nimport { useNavigate, useSearchParams } from 'react-router-dom';\nimport { API, showError, showSuccess } from '../helpers';\nimport { UserContext } from '../context/User';\n\nconst GitHubOAuth = () => {\n const [searchParams, setSearchParams] = useSearchParams();\n\n const [userState, userDispatch] = useContext(UserContext);\n const [prompt, setPrompt] = useState('处理中...');\n const [processing, setProcessing] = useState(true);\n\n let navigate = useNavigate();\n\n const sendCode = async (code, state, count) => {\n const res = await API.get(`/api/oauth/github?code=${code}&state=${state}`);\n const { success, message, data } = res.data;\n if (success) {\n if (message === 'bind') {\n showSuccess('绑定成功！');\n navigate('/setting');\n } else {\n userDispatch({ type: 'login', payload: data });\n localStorage.setItem('user', JSON.stringify(data));\n showSuccess('登录成功！');\n navigate('/');\n }\n } else {\n showError(message);\n if (count === 0) {\n setPrompt(`操作失败，重定向至登录界面中...`);\n navigate('/setting'); // in case this is failed to bind GitHub\n return;\n }\n count++;\n setPrompt(`出现错误，第 ${count} 次重试中...`);\n await new Promise((resolve) => setTimeout(resolve, count * 2000));\n await sendCode(code, state, count);\n }\n };\n\n useEffect(() => {\n let code = searchParams.get('code');\n let state = searchParams.get('state');\n sendCode(code, state, 0).then();\n }, []);\n\n return (\n \n \n {prompt} \n \n \n );\n};\n\nexport default GitHubOAuth;\n"
},
{
"path": "web/default/src/components/Header.js",
"content": "import React, { useContext, useState } from 'react';\nimport { Link, useNavigate } from 'react-router-dom';\nimport { UserContext } from '../context/User';\nimport { useTranslation } from 'react-i18next';\n\nimport {\n Button,\n Container,\n Dropdown,\n Icon,\n Menu,\n Segment,\n} from 'semantic-ui-react';\nimport {\n API,\n getLogo,\n getSystemName,\n isAdmin,\n isMobile,\n showSuccess,\n} from '../helpers';\nimport '../index.css';\n\n// Header Buttons\nlet headerButtons = [\n {\n name: 'header.channel',\n to: '/channel',\n icon: 'sitemap',\n admin: true,\n },\n {\n name: 'header.token',\n to: '/token',\n icon: 'key',\n },\n {\n name: 'header.redemption',\n to: '/redemption',\n icon: 'dollar sign',\n admin: true,\n },\n {\n name: 'header.topup',\n to: '/topup',\n icon: 'cart',\n },\n {\n name: 'header.user',\n to: '/user',\n icon: 'user',\n admin: true,\n },\n {\n name: 'header.dashboard',\n to: '/dashboard',\n icon: 'chart bar',\n },\n {\n name: 'header.log',\n to: '/log',\n icon: 'book',\n },\n {\n name: 'header.setting',\n to: '/setting',\n icon: 'setting',\n },\n {\n name: 'header.about',\n to: '/about',\n icon: 'info circle',\n },\n];\n\nif (localStorage.getItem('chat_link')) {\n headerButtons.splice(1, 0, {\n name: 'header.chat',\n to: '/chat',\n icon: 'comments',\n });\n}\n\nconst Header = () => {\n const { t, i18n } = useTranslation();\n const [userState, userDispatch] = useContext(UserContext);\n let navigate = useNavigate();\n\n const [showSidebar, setShowSidebar] = useState(false);\n const systemName = getSystemName();\n const logo = getLogo();\n\n async function logout() {\n setShowSidebar(false);\n await API.get('/api/user/logout');\n showSuccess('注销成功!');\n userDispatch({ type: 'logout' });\n localStorage.removeItem('user');\n navigate('/login');\n }\n\n const toggleSidebar = () => {\n setShowSidebar(!showSidebar);\n };\n\n const renderButtons = (isMobile) => {\n return headerButtons.map((button) => {\n if (button.admin && !isAdmin()) return <>;\n if (isMobile) {\n return (\n {\n navigate(button.to);\n setShowSidebar(false);\n }}\n style={{ fontSize: '15px' }}\n >\n {t(button.name)}\n \n );\n }\n return (\n \n \n \n \n \n {systemName} \n
\n \n \n \n \n \n \n \n {showSidebar ? (\n \n \n {renderButtons(true)}\n \n \n \n {userState.user ? (\n \n {t('header.logout')}\n \n ) : (\n <>\n {\n setShowSidebar(false);\n navigate('/login');\n }}\n >\n {t('header.login')}\n \n {\n setShowSidebar(false);\n navigate('/register');\n }}\n >\n {t('header.register')}\n \n \n )}\n \n \n \n ) : (\n <>\n )}\n \n );\n }\n\n return (\n <>\n \n \n \n \n {systemName}\n \n \n {renderButtons(false)}\n \n \n \n \n {t('header.logout')}\n \n \n \n ) : (\n \n \n \n \n );\n};\n\nexport default Header;\n"
},
{
"path": "web/default/src/components/LarkOAuth.js",
"content": "import React, { useContext, useEffect, useState } from 'react';\nimport { Dimmer, Loader, Segment } from 'semantic-ui-react';\nimport { useNavigate, useSearchParams } from 'react-router-dom';\nimport { API, showError, showSuccess } from '../helpers';\nimport { UserContext } from '../context/User';\n\nconst LarkOAuth = () => {\n const [searchParams, setSearchParams] = useSearchParams();\n\n const [userState, userDispatch] = useContext(UserContext);\n const [prompt, setPrompt] = useState('处理中...');\n const [processing, setProcessing] = useState(true);\n\n let navigate = useNavigate();\n\n const sendCode = async (code, state, count) => {\n const res = await API.get(`/api/oauth/lark?code=${code}&state=${state}`);\n const { success, message, data } = res.data;\n if (success) {\n if (message === 'bind') {\n showSuccess('绑定成功！');\n navigate('/setting');\n } else {\n userDispatch({ type: 'login', payload: data });\n localStorage.setItem('user', JSON.stringify(data));\n showSuccess('登录成功！');\n navigate('/');\n }\n } else {\n showError(message);\n if (count === 0) {\n setPrompt(`操作失败，重定向至登录界面中...`);\n navigate('/setting'); // in case this is failed to bind lark\n return;\n }\n count++;\n setPrompt(`出现错误，第 ${count} 次重试中...`);\n await new Promise((resolve) => setTimeout(resolve, count * 2000));\n await sendCode(code, state, count);\n }\n };\n\n useEffect(() => {\n let code = searchParams.get('code');\n let state = searchParams.get('state');\n sendCode(code, state, 0).then();\n }, []);\n\n return (\n \n \n {prompt} \n \n \n );\n};\n\nexport default LarkOAuth;\n"
},
{
"path": "web/default/src/components/Loading.js",
"content": "import React from 'react';\nimport { Segment, Dimmer, Loader } from 'semantic-ui-react';\n\nconst Loading = ({ prompt: name = 'page' }) => {\n return (\n \n \n 加载{name}中... \n \n \n );\n};\n\nexport default Loading;\n"
},
{
"path": "web/default/src/components/LoginForm.js",
"content": "import React, { useContext, useEffect, useState } from 'react';\nimport {\n Button,\n Divider,\n Form,\n Grid,\n Header,\n Image,\n Message,\n Modal,\n Segment,\n Card,\n} from 'semantic-ui-react';\nimport { Link, useNavigate, useSearchParams } from 'react-router-dom';\nimport { useTranslation } from 'react-i18next';\nimport { UserContext } from '../context/User';\nimport { API, getLogo, showError, showSuccess, showWarning } from '../helpers';\nimport { onGitHubOAuthClicked, onLarkOAuthClicked } from './utils';\nimport larkIcon from '../images/lark.svg';\n\nconst LoginForm = () => {\n const { t } = useTranslation();\n const [inputs, setInputs] = useState({\n username: '',\n password: '',\n wechat_verification_code: '',\n });\n const [searchParams, setSearchParams] = useSearchParams();\n const [submitted, setSubmitted] = useState(false);\n const { username, password } = inputs;\n const [userState, userDispatch] = useContext(UserContext);\n let navigate = useNavigate();\n const [status, setStatus] = useState({});\n const logo = getLogo();\n\n useEffect(() => {\n if (searchParams.get('expired')) {\n showError(t('messages.error.login_expired'));\n }\n let status = localStorage.getItem('status');\n if (status) {\n status = JSON.parse(status);\n setStatus(status);\n }\n }, []);\n\n const [showWeChatLoginModal, setShowWeChatLoginModal] = useState(false);\n\n const onWeChatLoginClicked = () => {\n setShowWeChatLoginModal(true);\n };\n\n const onSubmitWeChatVerificationCode = async () => {\n const res = await API.get(\n `/api/oauth/wechat?code=${inputs.wechat_verification_code}`\n );\n const { success, message, data } = res.data;\n if (success) {\n userDispatch({ type: 'login', payload: data });\n localStorage.setItem('user', JSON.stringify(data));\n navigate('/');\n showSuccess(t('messages.success.login'));\n setShowWeChatLoginModal(false);\n } else {\n showError(message);\n }\n };\n\n function handleChange(e) {\n const { name, value } = e.target;\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n }\n\n async function handleSubmit(e) {\n setSubmitted(true);\n if (username && password) {\n const res = await API.post(`/api/user/login`, {\n username,\n password,\n });\n const { success, message, data } = res.data;\n if (success) {\n userDispatch({ type: 'login', payload: data });\n localStorage.setItem('user', JSON.stringify(data));\n if (username === 'root' && password === '123456') {\n navigate('/user/edit');\n showSuccess(t('messages.success.login'));\n showWarning(t('messages.error.root_password'));\n } else {\n navigate('/token');\n showSuccess(t('messages.success.login'));\n }\n } else {\n showError(message);\n }\n }\n }\n\n return (\n \n \n \n \n \n \n {t('auth.login.title')} \n \n \n \n\n \n \n \n {t('auth.login.forgot_password')}\n \n {t('auth.login.reset_password')}\n \n
\n \n {t('auth.login.no_account')}\n \n {t('auth.login.register')}\n \n
\n \n \n\n {(status.github_oauth ||\n status.wechat_login ||\n status.lark_client_id) && (\n <>\n \n {t('auth.login.other_methods')}\n \n \n {status.github_oauth && (\n \n onGitHubOAuthClicked(status.github_client_id)\n }\n />\n )}\n {status.wechat_login && (\n onLarkOAuthClicked(status.lark_client_id)}\n >\n \n \n setShowWeChatLoginModal(false)}\n onOpen={() => setShowWeChatLoginModal(true)}\n open={showWeChatLoginModal}\n size={'mini'}\n >\n \n \n \n
{t('auth.login.wechat.scan_tip')}
\n
\n \n \n \n \n );\n};\n\nexport default LoginForm;\n"
},
{
"path": "web/default/src/components/LogsTable.js",
"content": "import React, { useEffect, useState } from 'react';\nimport {\n Button,\n Form,\n Header,\n Label,\n Pagination,\n Segment,\n Select,\n Table,\n Popup,\n} from 'semantic-ui-react';\nimport {\n API,\n copy,\n isAdmin,\n showError,\n showSuccess,\n showWarning,\n timestamp2string,\n} from '../helpers';\nimport { useTranslation } from 'react-i18next';\n\nimport { ITEMS_PER_PAGE } from '../constants';\nimport { renderColorLabel, renderQuota } from '../helpers/render';\nimport { Link } from 'react-router-dom';\n\nfunction renderTimestamp(timestamp, request_id) {\n return (\n {\n if (await copy(request_id)) {\n showSuccess(`已复制请求 ID：${request_id}`);\n } else {\n showWarning(`请求 ID 复制失败：${request_id}`);\n }\n }}\n style={{ cursor: 'pointer' }}\n >\n {timestamp2string(timestamp)}\n \n );\n}\n\nconst MODE_OPTIONS = [\n { key: 'all', text: '全部用户', value: 'all' },\n { key: 'self', text: '当前用户', value: 'self' },\n];\n\nfunction renderType(type) {\n switch (type) {\n case 1:\n return (\n \n 充值\n \n );\n case 2:\n return (\n \n 消费\n \n );\n case 3:\n return (\n \n 管理\n \n );\n case 4:\n return (\n \n 系统\n \n );\n case 5:\n return (\n \n 测试\n \n );\n default:\n return (\n \n 未知\n \n );\n }\n}\n\nfunction getColorByElapsedTime(elapsedTime) {\n if (elapsedTime === undefined || 0) return 'black';\n if (elapsedTime < 1000) return 'green';\n if (elapsedTime < 3000) return 'olive';\n if (elapsedTime < 5000) return 'yellow';\n if (elapsedTime < 10000) return 'orange';\n return 'red';\n}\n\nfunction renderDetail(log) {\n return (\n <>\n {log.content}\n \n {log.elapsed_time} ms\n \n )}\n {log.is_stream && (\n <>\n \n Stream\n \n \n )}\n {log.system_prompt_reset && (\n <>\n \n System Prompt Reset\n \n \n )}\n \n );\n}\n\nconst LogsTable = () => {\n const { t } = useTranslation();\n const [logs, setLogs] = useState([]);\n const [showStat, setShowStat] = useState(false);\n const [loading, setLoading] = useState(true);\n const [activePage, setActivePage] = useState(1);\n const [searchKeyword, setSearchKeyword] = useState('');\n const [searching, setSearching] = useState(false);\n const [logType, setLogType] = useState(0);\n const isAdminUser = isAdmin();\n let now = new Date();\n const [inputs, setInputs] = useState({\n username: '',\n token_name: '',\n model_name: '',\n start_timestamp: timestamp2string(0),\n end_timestamp: timestamp2string(now.getTime() / 1000 + 3600),\n channel: '',\n });\n const {\n username,\n token_name,\n model_name,\n start_timestamp,\n end_timestamp,\n channel,\n } = inputs;\n\n const [stat, setStat] = useState({\n quota: 0,\n token: 0,\n });\n\n const LOG_OPTIONS = [\n { key: '0', text: t('log.type.all'), value: 0 },\n { key: '1', text: t('log.type.topup'), value: 1 },\n { key: '2', text: t('log.type.usage'), value: 2 },\n { key: '3', text: t('log.type.admin'), value: 3 },\n { key: '4', text: t('log.type.system'), value: 4 },\n { key: '5', text: t('log.type.test'), value: 5 },\n ];\n\n const handleInputChange = (e, { name, value }) => {\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n };\n\n const getLogSelfStat = async () => {\n let localStartTimestamp = Date.parse(start_timestamp) / 1000;\n let localEndTimestamp = Date.parse(end_timestamp) / 1000;\n let res = await API.get(\n `/api/log/self/stat?type=${logType}&token_name=${token_name}&model_name=${model_name}&start_timestamp=${localStartTimestamp}&end_timestamp=${localEndTimestamp}`\n );\n const { success, message, data } = res.data;\n if (success) {\n setStat(data);\n } else {\n showError(message);\n }\n };\n\n const getLogStat = async () => {\n let localStartTimestamp = Date.parse(start_timestamp) / 1000;\n let localEndTimestamp = Date.parse(end_timestamp) / 1000;\n let res = await API.get(\n `/api/log/stat?type=${logType}&username=${username}&token_name=${token_name}&model_name=${model_name}&start_timestamp=${localStartTimestamp}&end_timestamp=${localEndTimestamp}&channel=${channel}`\n );\n const { success, message, data } = res.data;\n if (success) {\n setStat(data);\n } else {\n showError(message);\n }\n };\n\n const handleEyeClick = async () => {\n if (!showStat) {\n if (isAdminUser) {\n await getLogStat();\n } else {\n await getLogSelfStat();\n }\n }\n setShowStat(!showStat);\n };\n\n const showUserTokenQuota = () => {\n return logType !== 5;\n };\n\n const loadLogs = async (startIdx) => {\n let url = '';\n let localStartTimestamp = Date.parse(start_timestamp) / 1000;\n let localEndTimestamp = Date.parse(end_timestamp) / 1000;\n if (isAdminUser) {\n url = `/api/log/?p=${startIdx}&type=${logType}&username=${username}&token_name=${token_name}&model_name=${model_name}&start_timestamp=${localStartTimestamp}&end_timestamp=${localEndTimestamp}&channel=${channel}`;\n } else {\n url = `/api/log/self/?p=${startIdx}&type=${logType}&token_name=${token_name}&model_name=${model_name}&start_timestamp=${localStartTimestamp}&end_timestamp=${localEndTimestamp}`;\n }\n const res = await API.get(url);\n const { success, message, data } = res.data;\n if (success) {\n if (startIdx === 0) {\n setLogs(data);\n } else {\n let newLogs = [...logs];\n newLogs.splice(startIdx * ITEMS_PER_PAGE, data.length, ...data);\n setLogs(newLogs);\n }\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n const onPaginationChange = (e, { activePage }) => {\n (async () => {\n if (activePage === Math.ceil(logs.length / ITEMS_PER_PAGE) + 1) {\n // In this case we have to load more data and then append them.\n await loadLogs(activePage - 1);\n }\n setActivePage(activePage);\n })();\n };\n\n const refresh = async () => {\n setLoading(true);\n setActivePage(1);\n await loadLogs(0);\n };\n\n useEffect(() => {\n refresh().then();\n }, [logType]);\n\n const searchLogs = async () => {\n if (searchKeyword === '') {\n // if keyword is blank, load files instead.\n await loadLogs(0);\n setActivePage(1);\n return;\n }\n setSearching(true);\n const res = await API.get(`/api/log/self/search?keyword=${searchKeyword}`);\n const { success, message, data } = res.data;\n if (success) {\n setLogs(data);\n setActivePage(1);\n } else {\n showError(message);\n }\n setSearching(false);\n };\n\n const handleKeywordChange = async (e, { value }) => {\n setSearchKeyword(value.trim());\n };\n\n const sortLog = (key) => {\n if (logs.length === 0) return;\n setLoading(true);\n let sortedLogs = [...logs];\n if (typeof sortedLogs[0][key] === 'string') {\n sortedLogs.sort((a, b) => {\n return ('' + a[key]).localeCompare(b[key]);\n });\n } else {\n sortedLogs.sort((a, b) => {\n if (a[key] === b[key]) return 0;\n if (a[key] > b[key]) return -1;\n if (a[key] < b[key]) return 1;\n });\n }\n if (sortedLogs[0].id === logs[0].id) {\n sortedLogs.reverse();\n }\n setLogs(sortedLogs);\n setLoading(false);\n };\n\n return (\n <>\n \n {t('log.usage_details')}（{t('log.total_quota')}：\n {showStat && renderQuota(stat.quota, t)}\n {!showStat && (\n \n {t('log.click_to_view')}\n \n )}\n ）\n \n \n \n \n \n {\n sortLog('created_time');\n }}\n width={3}\n >\n {t('log.table.time')}\n \n {isAdminUser && (\n {\n sortLog('channel');\n }}\n width={1}\n >\n {t('log.table.channel')}\n \n )}\n {\n sortLog('type');\n }}\n width={1}\n >\n {t('log.table.type')}\n \n {\n sortLog('model_name');\n }}\n width={2}\n >\n {t('log.table.model')}\n \n {showUserTokenQuota() && (\n <>\n {isAdminUser && (\n {\n sortLog('username');\n }}\n width={2}\n >\n {t('log.table.username')}\n \n )}\n {\n sortLog('token_name');\n }}\n width={2}\n >\n {t('log.table.token_name')}\n \n {\n sortLog('prompt_tokens');\n }}\n width={1}\n >\n {t('log.table.prompt_tokens')}\n \n {\n sortLog('completion_tokens');\n }}\n width={1}\n >\n {t('log.table.completion_tokens')}\n \n {\n sortLog('quota');\n }}\n width={1}\n >\n {t('log.table.quota')}\n \n \n )}\n {t('log.table.detail')} \n \n \n\n \n {logs\n .slice(\n (activePage - 1) * ITEMS_PER_PAGE,\n activePage * ITEMS_PER_PAGE\n )\n .map((log, idx) => {\n if (log.deleted) return <>;\n return (\n \n \n {renderTimestamp(log.created_at, log.request_id)}\n \n {isAdminUser && (\n \n {log.channel ? (\n \n {log.channel}\n \n ) : (\n ''\n )}\n \n )}\n {renderType(log.type)} \n \n {log.model_name ? renderColorLabel(log.model_name) : ''}\n \n {showUserTokenQuota() && (\n <>\n {isAdminUser && (\n \n {log.username ? (\n \n {log.username}\n \n ) : (\n ''\n )}\n \n )}\n \n {log.token_name ? renderColorLabel(log.token_name) : ''}\n \n\n \n {log.prompt_tokens ? log.prompt_tokens : ''}\n \n \n {log.completion_tokens ? log.completion_tokens : ''}\n \n \n {log.quota ? renderQuota(log.quota, t, 6) : ''}\n \n \n )}\n\n {renderDetail(log)} \n \n );\n })}\n \n\n \n \n \n {\n setLogType(value);\n }}\n />\n \n {t('log.buttons.refresh')}\n \n \n \n \n
\n \n );\n};\n\nexport default LogsTable;\n"
},
{
"path": "web/default/src/components/OperationSetting.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { useTranslation } from 'react-i18next';\nimport { Divider, Form, Grid, Header } from 'semantic-ui-react';\nimport {\n API,\n showError,\n showSuccess,\n timestamp2string,\n verifyJSON,\n} from '../helpers';\n\nconst OperationSetting = () => {\n const { t } = useTranslation();\n let now = new Date();\n let [inputs, setInputs] = useState({\n QuotaForNewUser: 0,\n QuotaForInviter: 0,\n QuotaForInvitee: 0,\n QuotaRemindThreshold: 0,\n PreConsumedQuota: 0,\n ModelRatio: '',\n CompletionRatio: '',\n GroupRatio: '',\n TopUpLink: '',\n ChatLink: '',\n QuotaPerUnit: 0,\n AutomaticDisableChannelEnabled: '',\n AutomaticEnableChannelEnabled: '',\n ChannelDisableThreshold: 0,\n LogConsumeEnabled: '',\n DisplayInCurrencyEnabled: '',\n DisplayTokenStatEnabled: '',\n ApproximateTokenEnabled: '',\n RetryTimes: 0,\n });\n const [originInputs, setOriginInputs] = useState({});\n let [loading, setLoading] = useState(false);\n let [historyTimestamp, setHistoryTimestamp] = useState(\n timestamp2string(now.getTime() / 1000 - 30 * 24 * 3600)\n ); // a month ago\n\n const getOptions = async () => {\n const res = await API.get('/api/option/');\n const { success, message, data } = res.data;\n if (success) {\n let newInputs = {};\n data.forEach((item) => {\n if (\n item.key === 'ModelRatio' ||\n item.key === 'GroupRatio' ||\n item.key === 'CompletionRatio'\n ) {\n item.value = JSON.stringify(JSON.parse(item.value), null, 2);\n }\n if (item.value === '{}') {\n item.value = '';\n }\n newInputs[item.key] = item.value;\n });\n setInputs(newInputs);\n setOriginInputs(newInputs);\n } else {\n showError(message);\n }\n };\n\n useEffect(() => {\n getOptions().then();\n }, []);\n\n const updateOption = async (key, value) => {\n setLoading(true);\n if (key.endsWith('Enabled')) {\n value = inputs[key] === 'true' ? 'false' : 'true';\n }\n const res = await API.put('/api/option/', {\n key,\n value,\n });\n const { success, message } = res.data;\n if (success) {\n setInputs((inputs) => ({ ...inputs, [key]: value }));\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n const handleInputChange = async (e, { name, value }) => {\n if (name.endsWith('Enabled')) {\n await updateOption(name, value);\n } else {\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n }\n };\n\n const submitConfig = async (group) => {\n switch (group) {\n case 'monitor':\n if (\n originInputs['ChannelDisableThreshold'] !==\n inputs.ChannelDisableThreshold\n ) {\n await updateOption(\n 'ChannelDisableThreshold',\n inputs.ChannelDisableThreshold\n );\n }\n if (\n originInputs['QuotaRemindThreshold'] !== inputs.QuotaRemindThreshold\n ) {\n await updateOption(\n 'QuotaRemindThreshold',\n inputs.QuotaRemindThreshold\n );\n }\n break;\n case 'ratio':\n if (originInputs['ModelRatio'] !== inputs.ModelRatio) {\n if (!verifyJSON(inputs.ModelRatio)) {\n showError('模型倍率不是合法的 JSON 字符串');\n return;\n }\n await updateOption('ModelRatio', inputs.ModelRatio);\n }\n if (originInputs['GroupRatio'] !== inputs.GroupRatio) {\n if (!verifyJSON(inputs.GroupRatio)) {\n showError('分组倍率不是合法的 JSON 字符串');\n return;\n }\n await updateOption('GroupRatio', inputs.GroupRatio);\n }\n if (originInputs['CompletionRatio'] !== inputs.CompletionRatio) {\n if (!verifyJSON(inputs.CompletionRatio)) {\n showError('补全倍率不是合法的 JSON 字符串');\n return;\n }\n await updateOption('CompletionRatio', inputs.CompletionRatio);\n }\n break;\n case 'quota':\n if (originInputs['QuotaForNewUser'] !== inputs.QuotaForNewUser) {\n await updateOption('QuotaForNewUser', inputs.QuotaForNewUser);\n }\n if (originInputs['QuotaForInvitee'] !== inputs.QuotaForInvitee) {\n await updateOption('QuotaForInvitee', inputs.QuotaForInvitee);\n }\n if (originInputs['QuotaForInviter'] !== inputs.QuotaForInviter) {\n await updateOption('QuotaForInviter', inputs.QuotaForInviter);\n }\n if (originInputs['PreConsumedQuota'] !== inputs.PreConsumedQuota) {\n await updateOption('PreConsumedQuota', inputs.PreConsumedQuota);\n }\n break;\n case 'general':\n if (originInputs['TopUpLink'] !== inputs.TopUpLink) {\n await updateOption('TopUpLink', inputs.TopUpLink);\n }\n if (originInputs['ChatLink'] !== inputs.ChatLink) {\n await updateOption('ChatLink', inputs.ChatLink);\n }\n if (originInputs['QuotaPerUnit'] !== inputs.QuotaPerUnit) {\n await updateOption('QuotaPerUnit', inputs.QuotaPerUnit);\n }\n if (originInputs['RetryTimes'] !== inputs.RetryTimes) {\n await updateOption('RetryTimes', inputs.RetryTimes);\n }\n break;\n }\n };\n\n const deleteHistoryLogs = async () => {\n console.log(inputs);\n const res = await API.delete(\n `/api/log/?target_timestamp=${Date.parse(historyTimestamp) / 1000}`\n );\n const { success, message, data } = res.data;\n if (success) {\n showSuccess(`${data} 条日志已清理！`);\n return;\n }\n showError('日志清理失败：' + message);\n };\n\n return (\n \n \n \n \n \n );\n};\n\nexport default OperationSetting;\n"
},
{
"path": "web/default/src/components/OtherSetting.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { useTranslation } from 'react-i18next';\nimport {\n Button,\n Divider,\n Form,\n Grid,\n Header,\n Message,\n Modal,\n} from 'semantic-ui-react';\nimport { Link } from 'react-router-dom';\nimport { API, showError, showSuccess, verifyJSON } from '../helpers';\nimport { marked } from 'marked';\n\nconst OtherSetting = () => {\n const { t } = useTranslation();\n let [inputs, setInputs] = useState({\n Footer: '',\n Notice: '',\n About: '',\n SystemName: '',\n Logo: '',\n HomePageContent: '',\n Theme: '',\n });\n let [loading, setLoading] = useState(false);\n const [showUpdateModal, setShowUpdateModal] = useState(false);\n const [updateData, setUpdateData] = useState({\n tag_name: '',\n content: '',\n });\n\n const getOptions = async () => {\n const res = await API.get('/api/option/');\n const { success, message, data } = res.data;\n if (success) {\n let newInputs = {};\n data.forEach((item) => {\n if (item.key in inputs) {\n newInputs[item.key] = item.value;\n }\n });\n setInputs(newInputs);\n } else {\n showError(message);\n }\n };\n\n useEffect(() => {\n getOptions().then();\n }, []);\n\n const updateOption = async (key, value) => {\n setLoading(true);\n const res = await API.put('/api/option/', {\n key,\n value,\n });\n const { success, message } = res.data;\n if (success) {\n setInputs((inputs) => ({ ...inputs, [key]: value }));\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n const handleInputChange = async (e, { name, value }) => {\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n };\n\n const submitNotice = async () => {\n await updateOption('Notice', inputs.Notice);\n };\n\n const submitSystemName = async () => {\n await updateOption('SystemName', inputs.SystemName);\n };\n\n const submitTheme = async () => {\n await updateOption('Theme', inputs.Theme);\n };\n\n const submitLogo = async () => {\n await updateOption('Logo', inputs.Logo);\n };\n\n const submitAbout = async () => {\n await updateOption('About', inputs.About);\n };\n\n const submitOption = async (key) => {\n await updateOption(key, inputs[key]);\n };\n\n const openGitHubRelease = () => {\n window.location = 'https://github.com/songquanpeng/one-api/releases/latest';\n };\n\n const checkUpdate = async () => {\n const res = await API.get(\n 'https://api.github.com/repos/songquanpeng/one-api/releases/latest'\n );\n const { tag_name, body } = res.data;\n if (tag_name === process.env.REACT_APP_VERSION) {\n showSuccess(`已是最新版本：${tag_name}`);\n } else {\n setUpdateData({\n tag_name: tag_name,\n content: marked.parse(body),\n });\n setShowUpdateModal(true);\n }\n };\n\n return (\n \n \n \n \n setShowUpdateModal(false)}\n onOpen={() => setShowUpdateModal(true)}\n open={showUpdateModal}\n >\n 新版本：{updateData.tag_name} \n \n \n
\n \n \n \n setShowUpdateModal(false)}>关闭 \n {\n setShowUpdateModal(false);\n openGitHubRelease();\n }}\n />\n \n \n \n );\n};\n\nexport default OtherSetting;\n"
},
{
"path": "web/default/src/components/PasswordResetConfirm.js",
"content": "import React, { useEffect, useState } from 'react';\nimport {\n Button,\n Form,\n Grid,\n Header,\n Image,\n Card,\n Message,\n} from 'semantic-ui-react';\nimport { useTranslation } from 'react-i18next';\nimport { API, copy, getLogo, showError, showNotice } from '../helpers';\nimport { useSearchParams } from 'react-router-dom';\n\nconst PasswordResetConfirm = () => {\n const { t } = useTranslation();\n const [inputs, setInputs] = useState({\n email: '',\n token: '',\n });\n const { email, token } = inputs;\n const [loading, setLoading] = useState(false);\n const [disableButton, setDisableButton] = useState(false);\n const [newPassword, setNewPassword] = useState('');\n const logo = getLogo();\n\n const [countdown, setCountdown] = useState(30);\n\n const [searchParams, setSearchParams] = useSearchParams();\n useEffect(() => {\n let token = searchParams.get('token');\n let email = searchParams.get('email');\n setInputs({\n token,\n email,\n });\n }, []);\n\n useEffect(() => {\n let countdownInterval = null;\n if (disableButton && countdown > 0) {\n countdownInterval = setInterval(() => {\n setCountdown(countdown - 1);\n }, 1000);\n } else if (countdown === 0) {\n setDisableButton(false);\n setCountdown(30);\n }\n return () => clearInterval(countdownInterval);\n }, [disableButton, countdown]);\n\n async function handleSubmit(e) {\n setDisableButton(true);\n if (!email) return;\n setLoading(true);\n const res = await API.post(`/api/user/reset`, {\n email,\n token,\n });\n const { success, message } = res.data;\n if (success) {\n let password = res.data.data;\n setNewPassword(password);\n await copy(password);\n showNotice(t('messages.notice.password_copied', { password }));\n } else {\n showError(message);\n }\n setLoading(false);\n }\n\n return (\n \n \n \n \n \n \n {t('auth.reset.confirm.title')} \n \n \n \n {newPassword && (\n \n \n {t('auth.reset.confirm.notice')}\n
\n \n )}\n \n \n \n \n );\n};\n\nexport default PasswordResetConfirm;\n"
},
{
"path": "web/default/src/components/PasswordResetForm.js",
"content": "import React, { useEffect, useState } from 'react';\nimport {\n Button,\n Form,\n Grid,\n Header,\n Image,\n Card,\n Message,\n} from 'semantic-ui-react';\nimport { useTranslation } from 'react-i18next';\nimport { API, getLogo, showError, showInfo, showSuccess } from '../helpers';\nimport Turnstile from 'react-turnstile';\n\nconst PasswordResetForm = () => {\n const { t } = useTranslation();\n const [inputs, setInputs] = useState({\n email: '',\n });\n const { email } = inputs;\n const [loading, setLoading] = useState(false);\n const [turnstileEnabled, setTurnstileEnabled] = useState(false);\n const [turnstileSiteKey, setTurnstileSiteKey] = useState('');\n const [turnstileToken, setTurnstileToken] = useState('');\n const [disableButton, setDisableButton] = useState(false);\n const [countdown, setCountdown] = useState(30);\n const logo = getLogo();\n\n useEffect(() => {\n let status = localStorage.getItem('status');\n if (status) {\n status = JSON.parse(status);\n if (status.turnstile_check) {\n setTurnstileEnabled(true);\n setTurnstileSiteKey(status.turnstile_site_key);\n }\n }\n }, []);\n\n useEffect(() => {\n let countdownInterval = null;\n if (disableButton && countdown > 0) {\n countdownInterval = setInterval(() => {\n setCountdown(countdown - 1);\n }, 1000);\n } else if (countdown === 0) {\n setDisableButton(false);\n setCountdown(30);\n }\n return () => clearInterval(countdownInterval);\n }, [disableButton, countdown]);\n\n function handleChange(e) {\n const { name, value } = e.target;\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n }\n\n async function handleSubmit(e) {\n setDisableButton(true);\n if (!email) return;\n if (turnstileEnabled && turnstileToken === '') {\n showInfo('请稍后几秒重试，Turnstile 正在检查用户环境！');\n return;\n }\n setLoading(true);\n const res = await API.get(\n `/api/reset_password?email=${email}&turnstile=${turnstileToken}`\n );\n const { success, message } = res.data;\n if (success) {\n showSuccess(t('auth.reset.notice'));\n setInputs({ ...inputs, email: '' });\n } else {\n showError(message);\n setDisableButton(false);\n setCountdown(30);\n }\n setLoading(false);\n }\n\n return (\n \n \n \n \n \n \n {t('auth.reset.title')} \n \n \n \n \n \n {t('auth.reset.notice')}\n
\n \n \n \n \n \n );\n};\n\nexport default PasswordResetForm;\n"
},
{
"path": "web/default/src/components/PersonalSetting.js",
"content": "import React, { useContext, useEffect, useState } from 'react';\nimport { useTranslation } from 'react-i18next';\nimport {\n Button,\n Divider,\n Form,\n Header,\n Image,\n Message,\n Modal,\n} from 'semantic-ui-react';\nimport { Link, useNavigate } from 'react-router-dom';\nimport {\n API,\n copy,\n showError,\n showInfo,\n showNotice,\n showSuccess,\n} from '../helpers';\nimport Turnstile from 'react-turnstile';\nimport { UserContext } from '../context/User';\nimport { onGitHubOAuthClicked, onLarkOAuthClicked } from './utils';\n\nconst PersonalSetting = () => {\n const { t } = useTranslation();\n const [userState, userDispatch] = useContext(UserContext);\n let navigate = useNavigate();\n\n const [inputs, setInputs] = useState({\n wechat_verification_code: '',\n email_verification_code: '',\n email: '',\n self_account_deletion_confirmation: '',\n });\n const [status, setStatus] = useState({});\n const [showWeChatBindModal, setShowWeChatBindModal] = useState(false);\n const [showEmailBindModal, setShowEmailBindModal] = useState(false);\n const [showAccountDeleteModal, setShowAccountDeleteModal] = useState(false);\n const [turnstileEnabled, setTurnstileEnabled] = useState(false);\n const [turnstileSiteKey, setTurnstileSiteKey] = useState('');\n const [turnstileToken, setTurnstileToken] = useState('');\n const [loading, setLoading] = useState(false);\n const [disableButton, setDisableButton] = useState(false);\n const [countdown, setCountdown] = useState(30);\n const [affLink, setAffLink] = useState('');\n const [systemToken, setSystemToken] = useState('');\n\n useEffect(() => {\n let status = localStorage.getItem('status');\n if (status) {\n status = JSON.parse(status);\n setStatus(status);\n if (status.turnstile_check) {\n setTurnstileEnabled(true);\n setTurnstileSiteKey(status.turnstile_site_key);\n }\n }\n }, []);\n\n useEffect(() => {\n let countdownInterval = null;\n if (disableButton && countdown > 0) {\n countdownInterval = setInterval(() => {\n setCountdown(countdown - 1);\n }, 1000);\n } else if (countdown === 0) {\n setDisableButton(false);\n setCountdown(30);\n }\n return () => clearInterval(countdownInterval); // Clean up on unmount\n }, [disableButton, countdown]);\n\n const handleInputChange = (e, { name, value }) => {\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n };\n\n const generateAccessToken = async () => {\n const res = await API.get('/api/user/token');\n const { success, message, data } = res.data;\n if (success) {\n setSystemToken(data);\n setAffLink('');\n await copy(data);\n showSuccess(`令牌已重置并已复制到剪贴板`);\n } else {\n showError(message);\n }\n };\n\n const getAffLink = async () => {\n const res = await API.get('/api/user/aff');\n const { success, message, data } = res.data;\n if (success) {\n let link = `${window.location.origin}/register?aff=${data}`;\n setAffLink(link);\n setSystemToken('');\n await copy(link);\n showSuccess(`邀请链接已复制到剪切板`);\n } else {\n showError(message);\n }\n };\n\n const handleAffLinkClick = async (e) => {\n e.target.select();\n await copy(e.target.value);\n showSuccess(`邀请链接已复制到剪切板`);\n };\n\n const handleSystemTokenClick = async (e) => {\n e.target.select();\n await copy(e.target.value);\n showSuccess(`系统令牌已复制到剪切板`);\n };\n\n const deleteAccount = async () => {\n if (inputs.self_account_deletion_confirmation !== userState.user.username) {\n showError('请输入你的账户名以确认删除！');\n return;\n }\n\n const res = await API.delete('/api/user/self');\n const { success, message } = res.data;\n\n if (success) {\n showSuccess('账户已删除！');\n await API.get('/api/user/logout');\n userDispatch({ type: 'logout' });\n localStorage.removeItem('user');\n navigate('/login');\n } else {\n showError(message);\n }\n };\n\n const bindWeChat = async () => {\n if (inputs.wechat_verification_code === '') return;\n const res = await API.get(\n `/api/oauth/wechat/bind?code=${inputs.wechat_verification_code}`\n );\n const { success, message } = res.data;\n if (success) {\n showSuccess('微信账户绑定成功！');\n setShowWeChatBindModal(false);\n } else {\n showError(message);\n }\n };\n\n const sendVerificationCode = async () => {\n setDisableButton(true);\n if (inputs.email === '') return;\n if (turnstileEnabled && turnstileToken === '') {\n showInfo('请稍后几秒重试，Turnstile 正在检查用户环境！');\n return;\n }\n setLoading(true);\n const res = await API.get(\n `/api/verification?email=${inputs.email}&turnstile=${turnstileToken}`\n );\n const { success, message } = res.data;\n if (success) {\n showSuccess('验证码发送成功，请检查邮箱！');\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n const bindEmail = async () => {\n if (inputs.email_verification_code === '') return;\n setLoading(true);\n const res = await API.get(\n `/api/oauth/email/bind?email=${inputs.email}&code=${inputs.email_verification_code}`\n );\n const { success, message } = res.data;\n if (success) {\n showSuccess('邮箱账户绑定成功！');\n setShowEmailBindModal(false);\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n return (\n \n
{t('setting.personal.general.title')} \n
{t('setting.personal.general.system_token_notice')} \n
\n {t('setting.personal.general.buttons.update_profile')}\n \n
\n {t('setting.personal.general.buttons.generate_token')}\n \n
\n {t('setting.personal.general.buttons.copy_invite')}\n \n
{\n setShowAccountDeleteModal(true);\n }}\n >\n {t('setting.personal.general.buttons.delete_account')}\n \n\n {systemToken && (\n {t('setting.personal.binding.title')} \n {status.wechat_login && (\n setShowWeChatBindModal(true)}>\n {t('setting.personal.binding.buttons.bind_wechat')}\n \n )}\n setShowWeChatBindModal(false)}\n onOpen={() => setShowWeChatBindModal(true)}\n open={showWeChatBindModal}\n size={'mini'}\n >\n \n \n \n
{t('setting.personal.binding.wechat.description')}
\n
\n \n \n {status.github_oauth && (\n onGitHubOAuthClicked(status.github_client_id)}>\n {t('setting.personal.binding.buttons.bind_github')}\n \n )}\n {status.lark_client_id && (\n onLarkOAuthClicked(status.lark_client_id)}>\n {t('setting.personal.binding.buttons.bind_lark')}\n \n )}\n setShowEmailBindModal(true)}>\n {t('setting.personal.binding.buttons.bind_email')}\n \n setShowEmailBindModal(false)}\n onOpen={() => setShowEmailBindModal(true)}\n open={showEmailBindModal}\n size={'tiny'}\n style={{ maxWidth: '450px' }}\n >\n {t('setting.personal.binding.email.title')} \n \n \n setShowAccountDeleteModal(false)}\n onOpen={() => setShowAccountDeleteModal(true)}\n open={showAccountDeleteModal}\n size={'tiny'}\n style={{ maxWidth: '450px' }}\n >\n \n {t('setting.personal.delete_account.title')}\n \n \n {t('setting.personal.delete_account.warning')} \n \n \n \n \n \n \n );\n};\n\nexport default PersonalSetting;\n"
},
{
"path": "web/default/src/components/PrivateRoute.js",
"content": "import { Navigate } from 'react-router-dom';\n\nimport { history } from '../helpers';\n\n\nfunction PrivateRoute({ children }) {\n if (!localStorage.getItem('user')) {\n return \n {t('redemption.status.unused')}\n \n );\n case 2:\n return (\n \n {t('redemption.status.disabled')}\n \n );\n case 3:\n return (\n \n {t('redemption.status.used')}\n \n );\n default:\n return (\n \n {t('redemption.status.unknown')}\n \n );\n }\n}\n\nconst RedemptionsTable = () => {\n const { t } = useTranslation();\n const [redemptions, setRedemptions] = useState([]);\n const [loading, setLoading] = useState(true);\n const [activePage, setActivePage] = useState(1);\n const [searchKeyword, setSearchKeyword] = useState('');\n const [searching, setSearching] = useState(false);\n\n const loadRedemptions = async (startIdx) => {\n const res = await API.get(`/api/redemption/?p=${startIdx}`);\n const { success, message, data } = res.data;\n if (success) {\n if (startIdx === 0) {\n setRedemptions(data);\n } else {\n let newRedemptions = redemptions;\n newRedemptions.push(...data);\n setRedemptions(newRedemptions);\n }\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n const onPaginationChange = (e, { activePage }) => {\n (async () => {\n if (activePage === Math.ceil(redemptions.length / ITEMS_PER_PAGE) + 1) {\n // In this case we have to load more data and then append them.\n await loadRedemptions(activePage - 1);\n }\n setActivePage(activePage);\n })();\n };\n\n useEffect(() => {\n loadRedemptions(0)\n .then()\n .catch((reason) => {\n showError(reason);\n });\n }, []);\n\n const manageRedemption = async (id, action, idx) => {\n let data = { id };\n let res;\n switch (action) {\n case 'delete':\n res = await API.delete(`/api/redemption/${id}/`);\n break;\n case 'enable':\n data.status = 1;\n res = await API.put('/api/redemption/?status_only=true', data);\n break;\n case 'disable':\n data.status = 2;\n res = await API.put('/api/redemption/?status_only=true', data);\n break;\n }\n const { success, message } = res.data;\n if (success) {\n showSuccess(t('token.messages.operation_success'));\n let redemption = res.data.data;\n let newRedemptions = [...redemptions];\n let realIdx = (activePage - 1) * ITEMS_PER_PAGE + idx;\n if (action === 'delete') {\n newRedemptions[realIdx].deleted = true;\n } else {\n newRedemptions[realIdx].status = redemption.status;\n }\n setRedemptions(newRedemptions);\n } else {\n showError(message);\n }\n };\n\n const searchRedemptions = async () => {\n if (searchKeyword === '') {\n // if keyword is blank, load files instead.\n await loadRedemptions(0);\n setActivePage(1);\n return;\n }\n setSearching(true);\n const res = await API.get(\n `/api/redemption/search?keyword=${searchKeyword}`\n );\n const { success, message, data } = res.data;\n if (success) {\n setRedemptions(data);\n setActivePage(1);\n } else {\n showError(message);\n }\n setSearching(false);\n };\n\n const handleKeywordChange = async (e, { value }) => {\n setSearchKeyword(value.trim());\n };\n\n const sortRedemption = (key) => {\n if (redemptions.length === 0) return;\n setLoading(true);\n let sortedRedemptions = [...redemptions];\n sortedRedemptions.sort((a, b) => {\n if (!isNaN(a[key])) {\n // If the value is numeric, subtract to sort\n return a[key] - b[key];\n } else {\n // If the value is not numeric, sort as strings\n return ('' + a[key]).localeCompare(b[key]);\n }\n });\n if (sortedRedemptions[0].id === redemptions[0].id) {\n sortedRedemptions.reverse();\n }\n setRedemptions(sortedRedemptions);\n setLoading(false);\n };\n\n const refresh = async () => {\n setLoading(true);\n await loadRedemptions(0);\n setActivePage(1);\n };\n\n return (\n <>\n \n\n \n \n \n {\n sortRedemption('id');\n }}\n >\n {t('redemption.table.id')}\n \n {\n sortRedemption('name');\n }}\n >\n {t('redemption.table.name')}\n \n {\n sortRedemption('status');\n }}\n >\n {t('redemption.table.status')}\n \n {\n sortRedemption('quota');\n }}\n >\n {t('redemption.table.quota')}\n \n {\n sortRedemption('created_time');\n }}\n >\n {t('redemption.table.created_time')}\n \n {\n sortRedemption('redeemed_time');\n }}\n >\n {t('redemption.table.redeemed_time')}\n \n {t('redemption.table.actions')} \n \n \n\n \n {redemptions\n .slice(\n (activePage - 1) * ITEMS_PER_PAGE,\n activePage * ITEMS_PER_PAGE\n )\n .map((redemption, idx) => {\n if (redemption.deleted) return <>;\n return (\n \n {redemption.id} \n \n {redemption.name ? redemption.name : t('redemption.table.no_name')}\n \n {renderStatus(redemption.status, t)} \n {renderQuota(redemption.quota, t)} \n \n {renderTimestamp(redemption.created_time)}\n \n \n {redemption.redeemed_time\n ? renderTimestamp(redemption.redeemed_time)\n : t('redemption.table.not_redeemed')}{' '}\n \n \n \n
{\n if (await copy(redemption.key)) {\n showSuccess(t('token.messages.copy_success'));\n } else {\n showWarning(t('token.messages.copy_failed'));\n setSearchKeyword(redemption.key);\n }\n }}\n >\n {t('redemption.buttons.copy')}\n \n \n {t('redemption.buttons.delete')}\n \n }\n on='click'\n flowing\n hoverable\n >\n {\n manageRedemption(redemption.id, 'delete', idx);\n }}\n >\n {t('redemption.buttons.confirm_delete')}\n \n \n {\n manageRedemption(\n redemption.id,\n redemption.status === 1 ? 'disable' : 'enable',\n idx\n );\n }}\n >\n {redemption.status === 1\n ? t('redemption.buttons.disable')\n : t('redemption.buttons.enable')}\n \n \n {t('redemption.buttons.edit')}\n \n \n \n );\n })}\n \n\n \n \n \n \n {t('redemption.buttons.add')}\n \n \n {t('redemption.buttons.refresh')}\n \n \n \n \n
\n \n );\n};\n\nexport default RedemptionsTable;\n"
},
{
"path": "web/default/src/components/RegisterForm.js",
"content": "import React, { useEffect, useState } from 'react';\nimport {\n Button,\n Form,\n Grid,\n Header,\n Image,\n Message,\n Card,\n Divider,\n} from 'semantic-ui-react';\nimport { Link, useNavigate } from 'react-router-dom';\nimport { useTranslation } from 'react-i18next';\nimport { API, getLogo, showError, showInfo, showSuccess } from '../helpers';\nimport Turnstile from 'react-turnstile';\n\nconst RegisterForm = () => {\n const { t } = useTranslation();\n const [inputs, setInputs] = useState({\n username: '',\n password: '',\n password2: '',\n email: '',\n verification_code: '',\n });\n const { username, password, password2 } = inputs;\n const [showEmailVerification, setShowEmailVerification] = useState(false);\n const [turnstileEnabled, setTurnstileEnabled] = useState(false);\n const [turnstileSiteKey, setTurnstileSiteKey] = useState('');\n const [turnstileToken, setTurnstileToken] = useState('');\n const [loading, setLoading] = useState(false);\n const [disableButton, setDisableButton] = useState(false);\n const [countdown, setCountdown] = useState(30);\n const logo = getLogo();\n let affCode = new URLSearchParams(window.location.search).get('aff');\n if (affCode) {\n localStorage.setItem('aff', affCode);\n }\n\n useEffect(() => {\n let status = localStorage.getItem('status');\n if (status) {\n status = JSON.parse(status);\n setShowEmailVerification(status.email_verification);\n if (status.turnstile_check) {\n setTurnstileEnabled(true);\n setTurnstileSiteKey(status.turnstile_site_key);\n }\n }\n });\n\n useEffect(() => {\n let countdownInterval = null;\n if (disableButton && countdown > 0) {\n countdownInterval = setInterval(() => {\n setCountdown(countdown - 1);\n }, 1000);\n } else if (countdown === 0) {\n setDisableButton(false);\n setCountdown(30);\n }\n return () => clearInterval(countdownInterval);\n }, [disableButton, countdown]);\n\n let navigate = useNavigate();\n\n function handleChange(e) {\n const { name, value } = e.target;\n console.log(name, value);\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n }\n\n async function handleSubmit(e) {\n if (password.length < 8) {\n showInfo(t('messages.error.password_length'));\n return;\n }\n if (password !== password2) {\n showInfo(t('messages.error.password_mismatch'));\n return;\n }\n if (username && password) {\n if (turnstileEnabled && turnstileToken === '') {\n showInfo(t('messages.error.turnstile_wait'));\n return;\n }\n setLoading(true);\n if (!affCode) {\n affCode = localStorage.getItem('aff');\n }\n inputs.aff_code = affCode;\n const res = await API.post(\n `/api/user/register?turnstile=${turnstileToken}`,\n inputs\n );\n const { success, message } = res.data;\n if (success) {\n navigate('/login');\n showSuccess(t('messages.success.register'));\n } else {\n showError(message);\n }\n setLoading(false);\n }\n }\n\n const sendVerificationCode = async () => {\n if (inputs.email === '') return;\n if (turnstileEnabled && turnstileToken === '') {\n showInfo(t('messages.error.turnstile_wait'));\n return;\n }\n setDisableButton(true);\n setLoading(true);\n const res = await API.get(\n `/api/verification?email=${inputs.email}&turnstile=${turnstileToken}`\n );\n const { success, message } = res.data;\n if (success) {\n showSuccess(t('messages.success.verification_code'));\n } else {\n showError(message);\n setDisableButton(false);\n setCountdown(30);\n }\n setLoading(false);\n };\n\n return (\n \n \n \n \n \n \n {t('auth.register.title')} \n \n \n \n\n \n \n {t('auth.register.has_account')}\n \n {t('auth.register.login')}\n \n \n \n \n \n \n \n );\n};\n\nexport default RegisterForm;\n"
},
{
"path": "web/default/src/components/SystemSetting.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { useTranslation } from 'react-i18next';\nimport {\n Button,\n Divider,\n Form,\n Grid,\n Header,\n Modal,\n Message,\n} from 'semantic-ui-react';\nimport { API, removeTrailingSlash, showError } from '../helpers';\n\nconst SystemSetting = () => {\n const { t } = useTranslation();\n let [inputs, setInputs] = useState({\n PasswordLoginEnabled: '',\n PasswordRegisterEnabled: '',\n EmailVerificationEnabled: '',\n GitHubOAuthEnabled: '',\n GitHubClientId: '',\n GitHubClientSecret: '',\n LarkClientId: '',\n LarkClientSecret: '',\n Notice: '',\n SMTPServer: '',\n SMTPPort: '',\n SMTPAccount: '',\n SMTPFrom: '',\n SMTPToken: '',\n ServerAddress: '',\n Footer: '',\n WeChatAuthEnabled: '',\n WeChatServerAddress: '',\n WeChatServerToken: '',\n WeChatAccountQRCodeImageURL: '',\n MessagePusherAddress: '',\n MessagePusherToken: '',\n TurnstileCheckEnabled: '',\n TurnstileSiteKey: '',\n TurnstileSecretKey: '',\n RegisterEnabled: '',\n EmailDomainRestrictionEnabled: '',\n EmailDomainWhitelist: '',\n });\n const [originInputs, setOriginInputs] = useState({});\n let [loading, setLoading] = useState(false);\n const [EmailDomainWhitelist, setEmailDomainWhitelist] = useState([]);\n const [restrictedDomainInput, setRestrictedDomainInput] = useState('');\n const [showPasswordWarningModal, setShowPasswordWarningModal] =\n useState(false);\n\n const getOptions = async () => {\n const res = await API.get('/api/option/');\n const { success, message, data } = res.data;\n if (success) {\n let newInputs = {};\n data.forEach((item) => {\n newInputs[item.key] = item.value;\n });\n setInputs({\n ...newInputs,\n EmailDomainWhitelist: newInputs.EmailDomainWhitelist.split(','),\n });\n setOriginInputs(newInputs);\n\n setEmailDomainWhitelist(\n newInputs.EmailDomainWhitelist.split(',').map((item) => {\n return { key: item, text: item, value: item };\n })\n );\n } else {\n showError(message);\n }\n };\n\n useEffect(() => {\n getOptions().then();\n }, []);\n\n const updateOption = async (key, value) => {\n setLoading(true);\n switch (key) {\n case 'PasswordLoginEnabled':\n case 'PasswordRegisterEnabled':\n case 'EmailVerificationEnabled':\n case 'GitHubOAuthEnabled':\n case 'WeChatAuthEnabled':\n case 'TurnstileCheckEnabled':\n case 'EmailDomainRestrictionEnabled':\n case 'RegisterEnabled':\n value = inputs[key] === 'true' ? 'false' : 'true';\n break;\n default:\n break;\n }\n const res = await API.put('/api/option/', {\n key,\n value,\n });\n const { success, message } = res.data;\n if (success) {\n if (key === 'EmailDomainWhitelist') {\n value = value.split(',');\n }\n setInputs((inputs) => ({\n ...inputs,\n [key]: value,\n }));\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n const handleInputChange = async (e, { name, value }) => {\n if (name === 'PasswordLoginEnabled' && inputs[name] === 'true') {\n // block disabling password login\n setShowPasswordWarningModal(true);\n return;\n }\n if (\n name === 'Notice' ||\n name.startsWith('SMTP') ||\n name === 'ServerAddress' ||\n name === 'GitHubClientId' ||\n name === 'GitHubClientSecret' ||\n name === 'LarkClientId' ||\n name === 'LarkClientSecret' ||\n name === 'WeChatServerAddress' ||\n name === 'WeChatServerToken' ||\n name === 'WeChatAccountQRCodeImageURL' ||\n name === 'TurnstileSiteKey' ||\n name === 'TurnstileSecretKey' ||\n name === 'EmailDomainWhitelist'\n ) {\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n } else {\n await updateOption(name, value);\n }\n };\n\n const submitServerAddress = async () => {\n let ServerAddress = removeTrailingSlash(inputs.ServerAddress);\n await updateOption('ServerAddress', ServerAddress);\n };\n\n const submitSMTP = async () => {\n if (originInputs['SMTPServer'] !== inputs.SMTPServer) {\n await updateOption('SMTPServer', inputs.SMTPServer);\n }\n if (originInputs['SMTPAccount'] !== inputs.SMTPAccount) {\n await updateOption('SMTPAccount', inputs.SMTPAccount);\n }\n if (originInputs['SMTPFrom'] !== inputs.SMTPFrom) {\n await updateOption('SMTPFrom', inputs.SMTPFrom);\n }\n if (\n originInputs['SMTPPort'] !== inputs.SMTPPort &&\n inputs.SMTPPort !== ''\n ) {\n await updateOption('SMTPPort', inputs.SMTPPort);\n }\n if (\n originInputs['SMTPToken'] !== inputs.SMTPToken &&\n inputs.SMTPToken !== ''\n ) {\n await updateOption('SMTPToken', inputs.SMTPToken);\n }\n };\n\n const submitEmailDomainWhitelist = async () => {\n if (\n originInputs['EmailDomainWhitelist'] !==\n inputs.EmailDomainWhitelist.join(',') &&\n inputs.SMTPToken !== ''\n ) {\n await updateOption(\n 'EmailDomainWhitelist',\n inputs.EmailDomainWhitelist.join(',')\n );\n }\n };\n\n const submitWeChat = async () => {\n if (originInputs['WeChatServerAddress'] !== inputs.WeChatServerAddress) {\n await updateOption(\n 'WeChatServerAddress',\n removeTrailingSlash(inputs.WeChatServerAddress)\n );\n }\n if (\n originInputs['WeChatAccountQRCodeImageURL'] !==\n inputs.WeChatAccountQRCodeImageURL\n ) {\n await updateOption(\n 'WeChatAccountQRCodeImageURL',\n inputs.WeChatAccountQRCodeImageURL\n );\n }\n if (\n originInputs['WeChatServerToken'] !== inputs.WeChatServerToken &&\n inputs.WeChatServerToken !== ''\n ) {\n await updateOption('WeChatServerToken', inputs.WeChatServerToken);\n }\n };\n\n const submitMessagePusher = async () => {\n if (originInputs['MessagePusherAddress'] !== inputs.MessagePusherAddress) {\n await updateOption(\n 'MessagePusherAddress',\n removeTrailingSlash(inputs.MessagePusherAddress)\n );\n }\n if (\n originInputs['MessagePusherToken'] !== inputs.MessagePusherToken &&\n inputs.MessagePusherToken !== ''\n ) {\n await updateOption('MessagePusherToken', inputs.MessagePusherToken);\n }\n };\n\n const submitGitHubOAuth = async () => {\n if (originInputs['GitHubClientId'] !== inputs.GitHubClientId) {\n await updateOption('GitHubClientId', inputs.GitHubClientId);\n }\n if (\n originInputs['GitHubClientSecret'] !== inputs.GitHubClientSecret &&\n inputs.GitHubClientSecret !== ''\n ) {\n await updateOption('GitHubClientSecret', inputs.GitHubClientSecret);\n }\n };\n\n const submitLarkOAuth = async () => {\n if (originInputs['LarkClientId'] !== inputs.LarkClientId) {\n await updateOption('LarkClientId', inputs.LarkClientId);\n }\n if (\n originInputs['LarkClientSecret'] !== inputs.LarkClientSecret &&\n inputs.LarkClientSecret !== ''\n ) {\n await updateOption('LarkClientSecret', inputs.LarkClientSecret);\n }\n };\n\n const submitTurnstile = async () => {\n if (originInputs['TurnstileSiteKey'] !== inputs.TurnstileSiteKey) {\n await updateOption('TurnstileSiteKey', inputs.TurnstileSiteKey);\n }\n if (\n originInputs['TurnstileSecretKey'] !== inputs.TurnstileSecretKey &&\n inputs.TurnstileSecretKey !== ''\n ) {\n await updateOption('TurnstileSecretKey', inputs.TurnstileSecretKey);\n }\n };\n\n const submitNewRestrictedDomain = () => {\n const localDomainList = inputs.EmailDomainWhitelist;\n if (\n restrictedDomainInput !== '' &&\n !localDomainList.includes(restrictedDomainInput)\n ) {\n setRestrictedDomainInput('');\n setInputs({\n ...inputs,\n EmailDomainWhitelist: [...localDomainList, restrictedDomainInput],\n });\n setEmailDomainWhitelist([\n ...EmailDomainWhitelist,\n {\n key: restrictedDomainInput,\n text: restrictedDomainInput,\n value: restrictedDomainInput,\n },\n ]);\n }\n };\n\n return (\n \n \n \n \n \n );\n};\n\nexport default SystemSetting;\n"
},
{
"path": "web/default/src/components/TokensTable.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { useTranslation } from 'react-i18next';\nimport {\n Button,\n Dropdown,\n Form,\n Label,\n Pagination,\n Popup,\n Table,\n} from 'semantic-ui-react';\nimport { Link } from 'react-router-dom';\nimport {\n API,\n copy,\n showError,\n showSuccess,\n showWarning,\n timestamp2string,\n} from '../helpers';\n\nimport { ITEMS_PER_PAGE } from '../constants';\nimport { renderQuota } from '../helpers/render';\n\nfunction renderTimestamp(timestamp) {\n return <>{timestamp2string(timestamp)};\n}\n\nfunction renderStatus(status, t) {\n switch (status) {\n case 1:\n return (\n \n {t('token.table.status_enabled')}\n \n );\n case 2:\n return (\n \n {t('token.table.status_disabled')}\n \n );\n case 3:\n return (\n \n {t('token.table.status_expired')}\n \n );\n case 4:\n return (\n \n {t('token.table.status_depleted')}\n \n );\n default:\n return (\n \n {t('token.table.status_unknown')}\n \n );\n }\n}\n\nconst TokensTable = () => {\n const { t } = useTranslation();\n\n const COPY_OPTIONS = [\n { key: 'raw', text: t('token.copy_options.raw'), value: '' },\n { key: 'next', text: t('token.copy_options.next'), value: 'next' },\n { key: 'ama', text: t('token.copy_options.ama'), value: 'ama' },\n { key: 'opencat', text: t('token.copy_options.opencat'), value: 'opencat' },\n { key: 'lobe', text: t('token.copy_options.lobe'), value: 'lobechat' },\n ];\n\n const OPEN_LINK_OPTIONS = [\n { key: 'next', text: t('token.copy_options.next'), value: 'next' },\n { key: 'ama', text: t('token.copy_options.ama'), value: 'ama' },\n { key: 'opencat', text: t('token.copy_options.opencat'), value: 'opencat' },\n { key: 'lobe', text: t('token.copy_options.lobe'), value: 'lobechat' },\n ];\n\n const [tokens, setTokens] = useState([]);\n const [loading, setLoading] = useState(true);\n const [activePage, setActivePage] = useState(1);\n const [searchKeyword, setSearchKeyword] = useState('');\n const [searching, setSearching] = useState(false);\n const [showTopUpModal, setShowTopUpModal] = useState(false);\n const [targetTokenIdx, setTargetTokenIdx] = useState(0);\n const [orderBy, setOrderBy] = useState('');\n\n const loadTokens = async (startIdx) => {\n const res = await API.get(`/api/token/?p=${startIdx}&order=${orderBy}`);\n const { success, message, data } = res.data;\n if (success) {\n if (startIdx === 0) {\n setTokens(data);\n } else {\n let newTokens = [...tokens];\n newTokens.splice(startIdx * ITEMS_PER_PAGE, data.length, ...data);\n setTokens(newTokens);\n }\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n const onPaginationChange = (e, { activePage }) => {\n (async () => {\n if (activePage === Math.ceil(tokens.length / ITEMS_PER_PAGE) + 1) {\n // In this case we have to load more data and then append them.\n await loadTokens(activePage - 1, orderBy);\n }\n setActivePage(activePage);\n })();\n };\n\n const refresh = async () => {\n setLoading(true);\n await loadTokens(activePage - 1);\n };\n\n const onCopy = async (type, key) => {\n let status = localStorage.getItem('status');\n let serverAddress = '';\n if (status) {\n status = JSON.parse(status);\n serverAddress = status.server_address;\n }\n if (serverAddress === '') {\n serverAddress = window.location.origin;\n }\n let encodedServerAddress = encodeURIComponent(serverAddress);\n const nextLink = localStorage.getItem('chat_link');\n let nextUrl;\n\n if (nextLink) {\n nextUrl =\n nextLink + `/#/?settings={\"key\":\"sk-${key}\",\"url\":\"${serverAddress}\"}`;\n } else {\n nextUrl = `https://app.nextchat.dev/#/?settings={\"key\":\"sk-${key}\",\"url\":\"${serverAddress}\"}`;\n }\n\n let url;\n switch (type) {\n case 'ama':\n url = `ama://set-api-key?server=${encodedServerAddress}&key=sk-${key}`;\n break;\n case 'opencat':\n url = `opencat://team/join?domain=${encodedServerAddress}&token=sk-${key}`;\n break;\n case 'next':\n url = nextUrl;\n break;\n case 'lobechat':\n url =\n nextLink +\n `/?settings={\"keyVaults\":{\"openai\":{\"apiKey\":\"sk-${key}\",\"baseURL\":\"${serverAddress}/v1\"}}}`;\n break;\n default:\n url = `sk-${key}`;\n }\n if (await copy(url)) {\n showSuccess(t('token.messages.copy_success'));\n } else {\n showWarning(t('token.messages.copy_failed'));\n setSearchKeyword(url);\n }\n };\n\n const onOpenLink = async (type, key) => {\n let status = localStorage.getItem('status');\n let serverAddress = '';\n if (status) {\n status = JSON.parse(status);\n serverAddress = status.server_address;\n }\n if (serverAddress === '') {\n serverAddress = window.location.origin;\n }\n let encodedServerAddress = encodeURIComponent(serverAddress);\n const chatLink = localStorage.getItem('chat_link');\n let defaultUrl;\n\n if (chatLink) {\n defaultUrl =\n chatLink + `/#/?settings={\"key\":\"sk-${key}\",\"url\":\"${serverAddress}\"}`;\n } else {\n defaultUrl = `https://app.nextchat.dev/#/?settings={\"key\":\"sk-${key}\",\"url\":\"${serverAddress}\"}`;\n }\n let url;\n switch (type) {\n case 'ama':\n url = `ama://set-api-key?server=${encodedServerAddress}&key=sk-${key}`;\n break;\n\n case 'opencat':\n url = `opencat://team/join?domain=${encodedServerAddress}&token=sk-${key}`;\n break;\n\n case 'lobechat':\n url =\n chatLink +\n `/?settings={\"keyVaults\":{\"openai\":{\"apiKey\":\"sk-${key}\",\"baseURL\":\"${serverAddress}/v1\"}}}`;\n break;\n\n default:\n url = defaultUrl;\n }\n\n window.open(url, '_blank');\n };\n\n useEffect(() => {\n loadTokens(0, orderBy)\n .then()\n .catch((reason) => {\n showError(reason);\n });\n }, [orderBy]);\n\n const manageToken = async (id, action, idx) => {\n let data = { id };\n let res;\n switch (action) {\n case 'delete':\n res = await API.delete(`/api/token/${id}/`);\n break;\n case 'enable':\n data.status = 1;\n res = await API.put('/api/token/?status_only=true', data);\n break;\n case 'disable':\n data.status = 2;\n res = await API.put('/api/token/?status_only=true', data);\n break;\n }\n const { success, message } = res.data;\n if (success) {\n showSuccess(t('token.messages.operation_success'));\n let token = res.data.data;\n let newTokens = [...tokens];\n let realIdx = (activePage - 1) * ITEMS_PER_PAGE + idx;\n if (action === 'delete') {\n newTokens[realIdx].deleted = true;\n } else {\n newTokens[realIdx].status = token.status;\n }\n setTokens(newTokens);\n } else {\n showError(message);\n }\n };\n\n const searchTokens = async () => {\n if (searchKeyword === '') {\n // if keyword is blank, load files instead.\n await loadTokens(0);\n setActivePage(1);\n setOrderBy('');\n return;\n }\n setSearching(true);\n const res = await API.get(`/api/token/search?keyword=${searchKeyword}`);\n const { success, message, data } = res.data;\n if (success) {\n setTokens(data);\n setActivePage(1);\n } else {\n showError(message);\n }\n setSearching(false);\n };\n\n const handleKeywordChange = async (e, { value }) => {\n setSearchKeyword(value.trim());\n };\n\n const sortToken = (key) => {\n if (tokens.length === 0) return;\n setLoading(true);\n let sortedTokens = [...tokens];\n sortedTokens.sort((a, b) => {\n if (!isNaN(a[key])) {\n // If the value is numeric, subtract to sort\n return a[key] - b[key];\n } else {\n // If the value is not numeric, sort as strings\n return ('' + a[key]).localeCompare(b[key]);\n }\n });\n if (sortedTokens[0].id === tokens[0].id) {\n sortedTokens.reverse();\n }\n setTokens(sortedTokens);\n setLoading(false);\n };\n\n const handleOrderByChange = (e, { value }) => {\n setOrderBy(value);\n setActivePage(1);\n };\n\n return (\n <>\n \n\n \n \n \n {\n sortToken('name');\n }}\n >\n {t('token.table.name')}\n \n {\n sortToken('status');\n }}\n >\n {t('token.table.status')}\n \n {\n sortToken('used_quota');\n }}\n >\n {t('token.table.used_quota')}\n \n {\n sortToken('remain_quota');\n }}\n >\n {t('token.table.remain_quota')}\n \n {\n sortToken('created_time');\n }}\n >\n {t('token.table.created_time')}\n \n {\n sortToken('expired_time');\n }}\n >\n {t('token.table.expired_time')}\n \n {t('token.table.actions')} \n \n \n\n \n {tokens\n .slice(\n (activePage - 1) * ITEMS_PER_PAGE,\n activePage * ITEMS_PER_PAGE\n )\n .map((token, idx) => {\n if (token.deleted) return <>;\n\n const copyOptionsWithHandlers = COPY_OPTIONS.map((option) => ({\n ...option,\n onClick: async () => {\n await onCopy(option.value, token.key);\n },\n }));\n\n const openLinkOptionsWithHandlers = OPEN_LINK_OPTIONS.map(\n (option) => ({\n ...option,\n onClick: async () => {\n await onOpenLink(option.value, token.key);\n },\n })\n );\n\n return (\n \n \n {token.name ? token.name : t('token.table.no_name')}\n \n {renderStatus(token.status, t)} \n {renderQuota(token.used_quota, t)} \n \n {token.unlimited_quota\n ? t('token.table.unlimited')\n : renderQuota(token.remain_quota, t, 2)}\n \n {renderTimestamp(token.created_time)} \n \n {token.expired_time === -1\n ? t('token.table.never_expire')\n : renderTimestamp(token.expired_time)}\n \n \n \n
\n await onCopy('', token.key)}\n >\n {t('token.buttons.copy')}\n \n }\n />\n {' '}\n
\n onOpenLink('', token.key)}\n >\n {t('token.buttons.chat')}\n \n }\n />\n {' '}\n
\n {t('token.buttons.delete')}\n \n }\n on='click'\n flowing\n hoverable\n >\n {\n manageToken(token.id, 'delete', idx);\n }}\n >\n {t('token.buttons.confirm_delete')} {token.name}\n \n \n {\n manageToken(\n token.id,\n token.status === 1 ? 'disable' : 'enable',\n idx\n );\n }}\n >\n {token.status === 1\n ? t('token.buttons.disable')\n : t('token.buttons.enable')}\n \n \n {t('token.buttons.edit')}\n \n \n \n );\n })}\n \n\n \n \n \n \n {t('token.buttons.add')}\n \n \n {t('token.buttons.refresh')}\n \n \n \n \n
\n \n );\n};\n\nexport default TokensTable;\n"
},
{
"path": "web/default/src/components/UsersTable.js",
"content": "import React, { useEffect, useState } from 'react';\nimport {\n Button,\n Form,\n Label,\n Pagination,\n Popup,\n Table,\n Dropdown,\n} from 'semantic-ui-react';\nimport { Link } from 'react-router-dom';\nimport { API, showError, showSuccess } from '../helpers';\nimport { useTranslation } from 'react-i18next';\n\nimport { ITEMS_PER_PAGE } from '../constants';\nimport {\n renderGroup,\n renderNumber,\n renderQuota,\n renderText,\n} from '../helpers/render';\n\nfunction renderRole(role, t) {\n switch (role) {\n case 1:\n return {t('user.table.role_types.normal')} ;\n case 10:\n return {t('user.table.role_types.admin')} ;\n case 100:\n return (\n {t('user.table.role_types.super_admin')} \n );\n default:\n return {t('user.table.role_types.unknown')} ;\n }\n}\n\nconst UsersTable = () => {\n const { t } = useTranslation();\n const [users, setUsers] = useState([]);\n const [loading, setLoading] = useState(true);\n const [activePage, setActivePage] = useState(1);\n const [searchKeyword, setSearchKeyword] = useState('');\n const [searching, setSearching] = useState(false);\n const [orderBy, setOrderBy] = useState('');\n\n const loadUsers = async (startIdx) => {\n const res = await API.get(`/api/user/?p=${startIdx}&order=${orderBy}`);\n const { success, message, data } = res.data;\n if (success) {\n if (startIdx === 0) {\n setUsers(data);\n } else {\n let newUsers = users;\n newUsers.push(...data);\n setUsers(newUsers);\n }\n } else {\n showError(message);\n }\n setLoading(false);\n };\n\n const onPaginationChange = (e, { activePage }) => {\n (async () => {\n if (activePage === Math.ceil(users.length / ITEMS_PER_PAGE) + 1) {\n // In this case we have to load more data and then append them.\n await loadUsers(activePage - 1, orderBy);\n }\n setActivePage(activePage);\n })();\n };\n\n useEffect(() => {\n loadUsers(0, orderBy)\n .then()\n .catch((reason) => {\n showError(reason);\n });\n }, [orderBy]);\n\n const manageUser = (username, action, idx) => {\n (async () => {\n const res = await API.post('/api/user/manage', {\n username,\n action,\n });\n const { success, message } = res.data;\n if (success) {\n showSuccess(t('user.messages.operation_success'));\n let user = res.data.data;\n let newUsers = [...users];\n let realIdx = (activePage - 1) * ITEMS_PER_PAGE + idx;\n if (action === 'delete') {\n newUsers[realIdx].deleted = true;\n } else {\n newUsers[realIdx].status = user.status;\n newUsers[realIdx].role = user.role;\n }\n setUsers(newUsers);\n } else {\n showError(message);\n }\n })();\n };\n\n const renderStatus = (status) => {\n switch (status) {\n case 1:\n return {t('user.table.status_types.activated')} ;\n case 2:\n return (\n \n {t('user.table.status_types.banned')}\n \n );\n default:\n return (\n \n {t('user.table.status_types.unknown')}\n \n );\n }\n };\n\n const searchUsers = async () => {\n if (searchKeyword === '') {\n // if keyword is blank, load files instead.\n await loadUsers(0);\n setActivePage(1);\n setOrderBy('');\n return;\n }\n setSearching(true);\n const res = await API.get(`/api/user/search?keyword=${searchKeyword}`);\n const { success, message, data } = res.data;\n if (success) {\n setUsers(data);\n setActivePage(1);\n } else {\n showError(message);\n }\n setSearching(false);\n };\n\n const handleKeywordChange = async (e, { value }) => {\n setSearchKeyword(value.trim());\n };\n\n const sortUser = (key) => {\n if (users.length === 0) return;\n setLoading(true);\n let sortedUsers = [...users];\n sortedUsers.sort((a, b) => {\n if (!isNaN(a[key])) {\n // If the value is numeric, subtract to sort\n return a[key] - b[key];\n } else {\n // If the value is not numeric, sort as strings\n return ('' + a[key]).localeCompare(b[key]);\n }\n });\n if (sortedUsers[0].id === users[0].id) {\n sortedUsers.reverse();\n }\n setUsers(sortedUsers);\n setLoading(false);\n };\n\n const handleOrderByChange = (e, { value }) => {\n setOrderBy(value);\n setActivePage(1);\n };\n\n return (\n <>\n \n\n \n \n \n {\n sortUser('id');\n }}\n >\n {t('user.table.id')}\n \n {\n sortUser('username');\n }}\n >\n {t('user.table.username')}\n \n {\n sortUser('group');\n }}\n >\n {t('user.table.group')}\n \n {\n sortUser('quota');\n }}\n >\n {t('user.table.quota')}\n \n {\n sortUser('role');\n }}\n >\n {t('user.table.role_text')}\n \n {\n sortUser('status');\n }}\n >\n {t('user.table.status_text')}\n \n {t('user.table.actions')} \n \n \n\n \n {users\n .slice(\n (activePage - 1) * ITEMS_PER_PAGE,\n activePage * ITEMS_PER_PAGE\n )\n .map((user, idx) => {\n if (user.deleted) return <>;\n return (\n \n {user.id} \n \n {renderText(user.username, 15)}}\n hoverable\n />\n \n {renderGroup(user.group)} \n {/**/}\n {/* {user.email ? {renderText(user.email, 24)}} /> : '无'}*/}\n {/* */}\n \n {renderQuota(user.quota, t)}\n }\n />\n {renderQuota(user.used_quota, t)}\n }\n />\n {renderNumber(user.request_count)}\n }\n />\n \n {renderRole(user.role, t)} \n {renderStatus(user.status)} \n \n \n
{\n manageUser(user.username, 'promote', idx);\n }}\n disabled={user.role === 100}\n >\n {t('user.buttons.promote')}\n \n {\n manageUser(user.username, 'demote', idx);\n }}\n disabled={user.role === 100}\n >\n {t('user.buttons.demote')}\n \n \n {t('user.buttons.delete')}\n \n }\n on='click'\n flowing\n hoverable\n >\n {\n manageUser(user.username, 'delete', idx);\n }}\n >\n {t('user.buttons.delete_user')} {user.username}\n \n \n {\n manageUser(\n user.username,\n user.status === 1 ? 'disable' : 'enable',\n idx\n );\n }}\n disabled={user.role === 100}\n >\n {user.status === 1\n ? t('user.buttons.disable')\n : t('user.buttons.enable')}\n \n \n {t('user.buttons.edit')}\n \n \n \n );\n })}\n \n\n \n \n \n \n {t('user.buttons.add')}\n \n \n \n \n
\n \n );\n};\n\nexport default UsersTable;\n"
},
{
"path": "web/default/src/components/utils.js",
"content": "import { API, showError } from '../helpers';\n\nexport async function getOAuthState() {\n const res = await API.get('/api/oauth/state');\n const { success, message, data } = res.data;\n if (success) {\n return data;\n } else {\n showError(message);\n return '';\n }\n}\n\nexport async function onGitHubOAuthClicked(github_client_id) {\n const state = await getOAuthState();\n if (!state) return;\n window.open(\n `https://github.com/login/oauth/authorize?client_id=${github_client_id}&state=${state}&scope=user:email`\n );\n}\n\nexport async function onLarkOAuthClicked(lark_client_id) {\n const state = await getOAuthState();\n if (!state) return;\n let redirect_uri = `${window.location.origin}/oauth/lark`;\n window.open(\n `https://open.feishu.cn/open-apis/authen/v1/index?redirect_uri=${redirect_uri}&app_id=${lark_client_id}&state=${state}`\n );\n}"
},
{
"path": "web/default/src/constants/channel.constants.js",
"content": "export const CHANNEL_OPTIONS = [\n { key: 1, text: 'OpenAI', value: 1, color: 'green' },\n {\n key: 50,\n text: 'OpenAI 兼容',\n value: 50,\n color: 'olive',\n description: 'OpenAI 兼容渠道，支持设置 Base URL',\n },\n {key: 14, text: 'Anthropic', value: 14, color: 'black'},\n { key: 33, text: 'AWS', value: 33, color: 'black' },\n {key: 3, text: 'Azure', value: 3, color: 'olive'},\n {key: 11, text: 'PaLM2', value: 11, color: 'orange'},\n {key: 24, text: 'Gemini', value: 24, color: 'orange'},\n {\n key: 51,\n text: 'Gemini (OpenAI)',\n value: 51,\n color: 'orange',\n description: 'Gemini OpenAI 兼容格式',\n },\n { key: 28, text: 'Mistral AI', value: 28, color: 'orange' },\n { key: 41, text: 'Novita', value: 41, color: 'purple' },\n {\n key: 40,\n text: '字节火山引擎',\n value: 40,\n color: 'blue',\n description: '原字节跳动豆包',\n },\n {\n key: 15,\n text: '百度文心千帆',\n value: 15,\n color: 'blue',\n tip: '请前往此处 获取 AK（API Key）以及 SK（Secret Key），注意，V2 版本接口请使用 百度文心千帆 V2 渠道类型',\n },\n {\n key: 47,\n text: '百度文心千帆 V2',\n value: 47,\n color: 'blue',\n tip: '请前往此处 获取 API Key，注意本渠道仅支持推理服务 V2 相关模型',\n },\n {\n key: 17,\n text: '阿里通义千问',\n value: 17,\n color: 'orange',\n tip: '如需使用阿里云百炼，请使用阿里云百炼 渠道',\n },\n { key: 49, text: '阿里云百炼', value: 49, color: 'orange' },\n {\n key: 18,\n text: '讯飞星火认知',\n value: 18,\n color: 'blue',\n tip: '本渠道基于讯飞 WebSocket 版本 API，如需 HTTP 版本，请使用讯飞星火认知 V2 渠道',\n },\n {\n key: 48,\n text: '讯飞星火认知 V2',\n value: 48,\n color: 'blue',\n tip: 'HTTP 版本的讯飞接口，前往此处 获取 HTTP 服务接口认证密钥',\n },\n { key: 16, text: '智谱 ChatGLM', value: 16, color: 'violet' },\n { key: 19, text: '360 智脑', value: 19, color: 'blue' },\n { key: 25, text: 'Moonshot AI', value: 25, color: 'black' },\n { key: 23, text: '腾讯混元', value: 23, color: 'teal' },\n { key: 26, text: '百川大模型', value: 26, color: 'orange' },\n { key: 27, text: 'MiniMax', value: 27, color: 'red' },\n { key: 29, text: 'Groq', value: 29, color: 'orange' },\n { key: 30, text: 'Ollama', value: 30, color: 'black' },\n { key: 31, text: '零一万物', value: 31, color: 'green' },\n { key: 32, text: '阶跃星辰', value: 32, color: 'blue' },\n { key: 34, text: 'Coze', value: 34, color: 'blue' },\n { key: 35, text: 'Cohere', value: 35, color: 'blue' },\n { key: 36, text: 'DeepSeek', value: 36, color: 'black' },\n { key: 37, text: 'Cloudflare', value: 37, color: 'orange' },\n { key: 38, text: 'DeepL', value: 38, color: 'black' },\n { key: 39, text: 'together.ai', value: 39, color: 'blue' },\n { key: 42, text: 'VertexAI', value: 42, color: 'blue' },\n { key: 43, text: 'Proxy', value: 43, color: 'blue' },\n { key: 44, text: 'SiliconFlow', value: 44, color: 'blue' },\n { key: 45, text: 'xAI', value: 45, color: 'blue' },\n { key: 46, text: 'Replicate', value: 46, color: 'blue' },\n {\n key: 8,\n text: '自定义渠道',\n value: 8,\n color: 'pink',\n tip: '不推荐使用，请使用 OpenAI 兼容 渠道类型。注意，这里所需要填入的代理地址仅会在实际请求时替换域名部分，如果你想填入 OpenAI SDK 中所要求的 Base URL，请使用 OpenAI 兼容渠道类型',\n description: '不推荐使用，请使用 OpenAI 兼容渠道类型',\n },\n { key: 22, text: '知识库：FastGPT', value: 22, color: 'blue' },\n { key: 21, text: '知识库：AI Proxy', value: 21, color: 'purple' },\n { key: 20, text: 'OpenRouter', value: 20, color: 'black' },\n { key: 2, text: '代理：API2D', value: 2, color: 'blue' },\n { key: 5, text: '代理：OpenAI-SB', value: 5, color: 'brown' },\n { key: 7, text: '代理：OhMyGPT', value: 7, color: 'purple' },\n { key: 10, text: '代理：AI Proxy', value: 10, color: 'purple' },\n { key: 4, text: '代理：CloseAI', value: 4, color: 'teal' },\n { key: 6, text: '代理：OpenAI Max', value: 6, color: 'violet' },\n { key: 9, text: '代理：AI.LS', value: 9, color: 'yellow' },\n { key: 12, text: '代理：API2GPT', value: 12, color: 'blue' },\n { key: 13, text: '代理：AIGC2D', value: 13, color: 'purple' },\n];\n"
},
{
"path": "web/default/src/constants/common.constant.js",
"content": "export const ITEMS_PER_PAGE = 10; // this value must keep same as the one defined in backend!\n"
},
{
"path": "web/default/src/constants/index.js",
"content": "export * from './toast.constants';\nexport * from './user.constants';\nexport * from './common.constant';\nexport * from './channel.constants';"
},
{
"path": "web/default/src/constants/toast.constants.js",
"content": "export const toastConstants = {\n SUCCESS_TIMEOUT: 5000,\n INFO_TIMEOUT: 8000,\n ERROR_TIMEOUT: 10000,\n WARNING_TIMEOUT: 10000,\n NOTICE_TIMEOUT: 20000,\n};\n"
},
{
"path": "web/default/src/constants/user.constants.js",
"content": "export const userConstants = {\n REGISTER_REQUEST: 'USERS_REGISTER_REQUEST',\n REGISTER_SUCCESS: 'USERS_REGISTER_SUCCESS',\n REGISTER_FAILURE: 'USERS_REGISTER_FAILURE',\n\n LOGIN_REQUEST: 'USERS_LOGIN_REQUEST',\n LOGIN_SUCCESS: 'USERS_LOGIN_SUCCESS',\n LOGIN_FAILURE: 'USERS_LOGIN_FAILURE',\n \n LOGOUT: 'USERS_LOGOUT',\n\n GETALL_REQUEST: 'USERS_GETALL_REQUEST',\n GETALL_SUCCESS: 'USERS_GETALL_SUCCESS',\n GETALL_FAILURE: 'USERS_GETALL_FAILURE',\n\n DELETE_REQUEST: 'USERS_DELETE_REQUEST',\n DELETE_SUCCESS: 'USERS_DELETE_SUCCESS',\n DELETE_FAILURE: 'USERS_DELETE_FAILURE' \n};\n"
},
{
"path": "web/default/src/context/Status/index.js",
"content": "// contexts/User/index.jsx\n\nimport React from 'react';\nimport { initialState, reducer } from './reducer';\n\nexport const StatusContext = React.createContext({\n state: initialState,\n dispatch: () => null,\n});\n\nexport const StatusProvider = ({ children }) => {\n const [state, dispatch] = React.useReducer(reducer, initialState);\n\n return (\n \n {children}\n \n );\n};"
},
{
"path": "web/default/src/context/Status/reducer.js",
"content": "export const reducer = (state, action) => {\n switch (action.type) {\n case 'set':\n return {\n ...state,\n status: action.payload,\n };\n case 'unset':\n return {\n ...state,\n status: undefined,\n };\n default:\n return state;\n }\n};\n\nexport const initialState = {\n status: undefined,\n};\n"
},
{
"path": "web/default/src/context/User/index.js",
"content": "// contexts/User/index.jsx\n\nimport React from \"react\"\nimport { reducer, initialState } from \"./reducer\"\n\nexport const UserContext = React.createContext({\n state: initialState,\n dispatch: () => null\n})\n\nexport const UserProvider = ({ children }) => {\n const [state, dispatch] = React.useReducer(reducer, initialState)\n\n return (\n \n { children }\n \n )\n}"
},
{
"path": "web/default/src/context/User/reducer.js",
"content": "export const reducer = (state, action) => {\n switch (action.type) {\n case 'login':\n return {\n ...state,\n user: action.payload\n };\n case 'logout':\n return {\n ...state,\n user: undefined\n };\n\n default:\n return state;\n }\n};\n\nexport const initialState = {\n user: undefined\n};"
},
{
"path": "web/default/src/helpers/api.js",
"content": "import { showError } from './utils';\nimport axios from 'axios';\n\nexport const API = axios.create({\n baseURL: process.env.REACT_APP_SERVER ? process.env.REACT_APP_SERVER : '',\n});\n\nAPI.interceptors.response.use(\n (response) => response,\n (error) => {\n showError(error);\n }\n);\n"
},
{
"path": "web/default/src/helpers/auth-header.js",
"content": "export function authHeader() {\n // return authorization header with jwt token\n let user = JSON.parse(localStorage.getItem('user'));\n\n if (user && user.token) {\n return { 'Authorization': 'Bearer ' + user.token };\n } else {\n return {};\n }\n}"
},
{
"path": "web/default/src/helpers/helper.js",
"content": "import {CHANNEL_OPTIONS} from '../constants';\n\nlet channelMap = undefined;\n\nexport function getChannelOption(channelId) {\n if (channelMap === undefined) {\n channelMap = {};\n CHANNEL_OPTIONS.forEach((option) => {\n channelMap[option.key] = option;\n });\n }\n return channelMap[channelId];\n}\n"
},
{
"path": "web/default/src/helpers/history.js",
"content": "import { createBrowserHistory } from 'history';\n\nexport const history = createBrowserHistory();"
},
{
"path": "web/default/src/helpers/index.js",
"content": "export * from './history';\nexport * from './auth-header';\nexport * from './utils';\nexport * from './api';"
},
{
"path": "web/default/src/helpers/render.js",
"content": "import { Label, Message } from 'semantic-ui-react';\nimport { getChannelOption } from './helper';\nimport React from 'react';\n\nexport function renderText(text, limit) {\n if (text.length > limit) {\n return text.slice(0, limit - 3) + '...';\n }\n return text;\n}\n\nexport function renderGroup(group) {\n if (group === '') {\n return default ;\n }\n let groups = group.split(',');\n groups.sort();\n return (\n \n {groups.map((group) => {\n if (group === 'vip' || group === 'pro') {\n return {group} ;\n } else if (group === 'svip' || group === 'premium') {\n return {group} ;\n }\n return {group} ;\n })}\n \n );\n}\n\nexport function renderNumber(num) {\n if (num >= 1000000000) {\n return (num / 1000000000).toFixed(1) + 'B';\n } else if (num >= 1000000) {\n return (num / 1000000).toFixed(1) + 'M';\n } else if (num >= 10000) {\n return (num / 1000).toFixed(1) + 'k';\n } else {\n return num;\n }\n}\n\nexport function renderQuota(quota, t, precision = 2) {\n const displayInCurrency =\n localStorage.getItem('display_in_currency') === 'true';\n const quotaPerUnit = parseFloat(\n localStorage.getItem('quota_per_unit') || '1'\n );\n\n if (displayInCurrency) {\n const amount = (quota / quotaPerUnit).toFixed(precision);\n return t('common.quota.display_short', { amount });\n }\n\n return renderNumber(quota);\n}\n\nexport function renderQuotaWithPrompt(quota, t) {\n const displayInCurrency =\n localStorage.getItem('display_in_currency') === 'true';\n const quotaPerUnit = parseFloat(\n localStorage.getItem('quota_per_unit') || '1'\n );\n\n if (displayInCurrency) {\n const amount = (quota / quotaPerUnit).toFixed(2);\n return ` (${t('common.quota.display', { amount })})`;\n }\n\n return '';\n}\n\nconst colors = [\n 'red',\n 'orange',\n 'yellow',\n 'olive',\n 'green',\n 'teal',\n 'blue',\n 'violet',\n 'purple',\n 'pink',\n 'brown',\n 'grey',\n 'black',\n];\n\nexport function renderColorLabel(text) {\n let hash = 0;\n for (let i = 0; i < text.length; i++) {\n hash = text.charCodeAt(i) + ((hash << 5) - hash);\n }\n let index = Math.abs(hash % colors.length);\n return (\n \n {text}\n \n );\n}\n\nexport function renderChannelTip(channelId) {\n let channel = getChannelOption(channelId);\n if (channel === undefined || channel.tip === undefined) {\n return <>;\n }\n return (\n \n
\n \n );\n}\n"
},
{
"path": "web/default/src/helpers/utils.js",
"content": "import {toast} from 'react-toastify';\nimport {toastConstants} from '../constants';\nimport React from 'react';\nimport {API} from './api';\n\nconst HTMLToastContent = ({ htmlContent }) => {\n return
;\n};\nexport default HTMLToastContent;\n\nexport function isAdmin() {\n let user = localStorage.getItem('user');\n if (!user) return false;\n user = JSON.parse(user);\n return user.role >= 10;\n}\n\nexport function isRoot() {\n let user = localStorage.getItem('user');\n if (!user) return false;\n user = JSON.parse(user);\n return user.role >= 100;\n}\n\nexport function getSystemName() {\n let system_name = localStorage.getItem('system_name');\n if (!system_name) return 'One API';\n return system_name;\n}\n\nexport function getLogo() {\n let logo = localStorage.getItem('logo');\n if (!logo) return '/logo.png';\n return logo;\n}\n\nexport function getFooterHTML() {\n return localStorage.getItem('footer_html');\n}\n\nexport async function copy(text) {\n let okay = true;\n try {\n await navigator.clipboard.writeText(text);\n } catch (e) {\n okay = false;\n console.error(e);\n }\n return okay;\n}\n\nexport function isMobile() {\n return window.innerWidth <= 600;\n}\n\nlet showErrorOptions = { autoClose: toastConstants.ERROR_TIMEOUT };\nlet showWarningOptions = { autoClose: toastConstants.WARNING_TIMEOUT };\nlet showSuccessOptions = { autoClose: toastConstants.SUCCESS_TIMEOUT };\nlet showInfoOptions = { autoClose: toastConstants.INFO_TIMEOUT };\nlet showNoticeOptions = { autoClose: false };\n\nif (isMobile()) {\n showErrorOptions.position = 'top-center';\n // showErrorOptions.transition = 'flip';\n\n showSuccessOptions.position = 'top-center';\n // showSuccessOptions.transition = 'flip';\n\n showInfoOptions.position = 'top-center';\n // showInfoOptions.transition = 'flip';\n\n showNoticeOptions.position = 'top-center';\n // showNoticeOptions.transition = 'flip';\n}\n\nexport function showError(error) {\n if (!error) return;\n console.error(error);\n if (error.message) {\n if (error.name === 'AxiosError') {\n switch (error.response.status) {\n case 401:\n // toast.error('错误：未登录或登录已过期，请重新登录！', showErrorOptions);\n window.location.href = '/login?expired=true';\n break;\n case 429:\n toast.error('错误：请求次数过多，请稍后再试！', showErrorOptions);\n break;\n case 500:\n toast.error('错误：服务器内部错误，请联系管理员！', showErrorOptions);\n break;\n case 405:\n toast.info('本站仅作演示之用，无服务端！');\n break;\n default:\n toast.error('错误：' + error.message, showErrorOptions);\n }\n return;\n }\n toast.error('错误：' + error.message, showErrorOptions);\n } else {\n toast.error('错误：' + error, showErrorOptions);\n }\n}\n\nexport function showWarning(message) {\n toast.warn(message, showWarningOptions);\n}\n\nexport function showSuccess(message) {\n toast.success(message, showSuccessOptions);\n}\n\nexport function showInfo(message) {\n toast.info(message, showInfoOptions);\n}\n\nexport function showNotice(message, isHTML = false) {\n if (isHTML) {\n toast(\n \n \n \n \n \n \n \n \n \n \n);\n"
},
{
"path": "web/default/src/locales/en/translation.json",
"content": "{\n \"header\": {\n \"home\": \"Home\",\n \"channel\": \"Channel\",\n \"token\": \"Token\",\n \"redemption\": \"Redemption\",\n \"topup\": \"Top Up\",\n \"user\": \"User\",\n \"dashboard\": \"Dashboard\",\n \"log\": \"Log\",\n \"setting\": \"Settings\",\n \"about\": \"About\",\n \"chat\": \"Chat\",\n \"login\": \"Login\",\n \"logout\": \"Logout\",\n \"register\": \"Register\"\n },\n \"topup\": {\n \"title\": \"Top Up Center\",\n \"get_code\": {\n \"title\": \"Get Redemption Code\",\n \"current_quota\": \"Current Available Quota\",\n \"button\": \"Get Code Now\"\n },\n \"redeem_code\": {\n \"title\": \"Redeem Code\",\n \"placeholder\": \"Please enter redemption code\",\n \"paste\": \"Paste\",\n \"paste_error\": \"Cannot access clipboard, please paste manually\",\n \"submit\": \"Redeem Now\",\n \"submitting\": \"Redeeming...\",\n \"empty_code\": \"Please enter the redemption code!\",\n \"success\": \"Top up successful!\",\n \"request_failed\": \"Request failed\",\n \"no_link\": \"Admin has not set up the top-up link!\"\n }\n },\n \"channel\": {\n \"title\": \"Channel Management\",\n \"search\": \"Search channels by ID, name and key...\",\n \"balance_notice\": \"OpenAI channels no longer support getting balance via key, so balance shows as 0. For supported channel types, click balance to refresh.\",\n \"test_notice\": \"Channel testing only supports chat models, preferring gpt-3.5-turbo. If unavailable, uses the first model in your configured list.\",\n \"detail_notice\": \"Click the detail button below to show balance and set additional test models.\",\n \"table\": {\n \"id\": \"ID\",\n \"name\": \"Name\",\n \"group\": \"Group\",\n \"type\": \"Type\",\n \"status\": \"Status\",\n \"response_time\": \"Response Time\",\n \"balance\": \"Balance\",\n \"priority\": \"Priority\",\n \"test_model\": \"Test Model\",\n \"actions\": \"Actions\",\n \"no_name\": \"None\",\n \"status_enabled\": \"Enabled\",\n \"status_disabled\": \"Disabled\",\n \"status_auto_disabled\": \"Disabled\",\n \"status_disabled_tip\": \"This channel is manually disabled\",\n \"status_auto_disabled_tip\": \"This channel is automatically disabled\",\n \"status_unknown\": \"Unknown Status\",\n \"not_tested\": \"Not Tested\",\n \"priority_tip\": \"Channel selection priority, higher is preferred\",\n \"select_test_model\": \"Please select test model\",\n \"click_to_update\": \"Click to update\",\n \"balance_not_supported\": \"-\"\n },\n \"buttons\": {\n \"test\": \"Test\",\n \"delete\": \"Delete\",\n \"confirm_delete\": \"Delete Channel\",\n \"enable\": \"Enable\",\n \"disable\": \"Disable\",\n \"edit\": \"Edit\",\n \"add\": \"Add New Channel\",\n \"test_all\": \"Test All Channels\",\n \"test_disabled\": \"Test Disabled Channels\",\n \"delete_disabled\": \"Delete Disabled Channels\",\n \"confirm_delete_disabled\": \"Confirm Delete\",\n \"refresh\": \"Refresh\",\n \"show_detail\": \"Details\",\n \"hide_detail\": \"Hide Details\"\n },\n \"messages\": {\n \"test_success\": \"Channel {{name}} test successful, model {{model}}, time {{time}}s, output: {{message}}\",\n \"test_all_started\": \"Channel testing started successfully, please refresh page to see results.\",\n \"delete_disabled_success\": \"Deleted all disabled channels, total: {{count}}\",\n \"balance_update_success\": \"Channel {{name}} balance updated successfully!\",\n \"all_balance_updated\": \"All enabled channel balances have been updated!\",\n \"operation_success\": \"Operation completed successfully!\"\n },\n \"edit\": {\n \"title_edit\": \"Update Channel Information\",\n \"title_create\": \"Create New Channel\",\n \"type\": \"Type\",\n \"name\": \"Name\",\n \"name_placeholder\": \"Please enter name\",\n \"group\": \"Group\",\n \"group_placeholder\": \"Please select groups that can use this channel\",\n \"group_addition\": \"Please edit group multipliers in system settings to add new group:\",\n \"models\": \"Models\",\n \"models_placeholder\": \"Please select models supported by this channel\",\n \"model_mapping\": \"Model Mapping\",\n \"model_mapping_placeholder\": \"Optional, used to modify model names in request body. A JSON string where keys are request model names and values are target model names\",\n \"system_prompt\": \"System Prompt\",\n \"system_prompt_placeholder\": \"Optional, used to force set system prompt. Use with custom model & model mapping. First create a unique custom model name above, then map it to a natively supported model\",\n \"proxy_url\": \"Proxy\",\n \"proxy_url_placeholder\": \"This is optional and used for API calls via a proxy. Please enter the proxy URL, formatted as: https://domain.com\",\n \"base_url\": \"Base URL\",\n \"base_url_placeholder\": \"The Base URL required by the OpenAPI SDK\",\n \"key\": \"Key\",\n \"key_placeholder\": \"Please enter key\",\n \"batch\": \"Batch Create\",\n \"batch_placeholder\": \"Please enter keys, one per line\",\n \"buttons\": {\n \"cancel\": \"Cancel\",\n \"submit\": \"Submit\",\n \"fill_models\": \"Fill Related Models\",\n \"fill_all\": \"Fill All Models\",\n \"clear\": \"Clear All Models\",\n \"add_custom\": \"Add\",\n \"custom_placeholder\": \"Enter custom model name\"\n },\n \"messages\": {\n \"name_required\": \"Please enter channel name and key!\",\n \"models_required\": \"Please select at least one model!\",\n \"model_mapping_invalid\": \"Model mapping must be valid JSON format!\",\n \"update_success\": \"Channel updated successfully!\",\n \"create_success\": \"Channel created successfully!\"\n },\n \"spark_version\": \"Model Version\",\n \"spark_version_placeholder\": \"Please enter Spark model version from API URL, e.g.: v2.1\",\n \"knowledge_id\": \"Knowledge Base ID\",\n \"knowledge_id_placeholder\": \"Please enter knowledge base ID, e.g.: 123456\",\n \"plugin_param\": \"Plugin Parameter\",\n \"plugin_param_placeholder\": \"Please enter plugin parameter (X-DashScope-Plugin header value)\",\n \"coze_notice\": \"For Coze, model name is the Bot ID. You can add prefix `bot-`, e.g.: `bot-123456`.\",\n \"douban_notice\": \"For Douban, you need to go to\",\n \"douban_notice_link\": \"Model Inference Page\",\n \"douban_notice_2\": \"to create an inference endpoint, and use the endpoint name as model name, e.g.: `ep-20240608051426-tkxvl`.\",\n \"aws_region_placeholder\": \"region, e.g.: us-west-2\",\n \"aws_ak_placeholder\": \"AWS IAM Access Key\",\n \"aws_sk_placeholder\": \"AWS IAM Secret Key\",\n \"vertex_region_placeholder\": \"Vertex AI Region, e.g.: us-east5\",\n \"vertex_project_id\": \"Vertex AI Project ID\",\n \"vertex_project_id_placeholder\": \"Vertex AI Project ID\",\n \"vertex_credentials\": \"Google Cloud Application Default Credentials JSON\",\n \"vertex_credentials_placeholder\": \"Google Cloud Application Default Credentials JSON\",\n \"user_id\": \"User ID\",\n \"user_id_placeholder\": \"User ID who generated this key\",\n \"key_prompts\": {\n \"default\": \"Please enter the authentication key for this channel\",\n \"zhipu\": \"Enter in format: APIKey|SecretKey\",\n \"spark\": \"Enter in format: APPID|APISecret|APIKey\",\n \"fastgpt\": \"Enter in format: APIKey-AppId, e.g.: fastgpt-0sp2gtvfdgyi4k30jwlgwf1i-64f335d84283f05518e9e041\",\n \"tencent\": \"Enter in format: AppId|SecretId|SecretKey\"\n }\n }\n },\n \"token\": {\n \"title\": \"Token Management\",\n \"search\": \"Search tokens by name ...\",\n \"table\": {\n \"name\": \"Name\",\n \"status\": \"Status\",\n \"used_quota\": \"Used Quota\",\n \"remain_quota\": \"Remaining Quota\",\n \"created_time\": \"Created Time\",\n \"expired_time\": \"Expiry Time\",\n \"actions\": \"Actions\",\n \"no_name\": \"None\",\n \"never_expire\": \"never\",\n \"unlimited\": \"Unlimited\",\n \"status_enabled\": \"Enabled\",\n \"status_disabled\": \"Disabled\",\n \"status_expired\": \"Expired\",\n \"status_depleted\": \"Depleted\",\n \"status_unknown\": \"Unknown Status\"\n },\n \"buttons\": {\n \"copy\": \"Copy\",\n \"chat\": \"Chat\",\n \"delete\": \"Delete\",\n \"confirm_delete\": \"Delete Token\",\n \"enable\": \"Enable\",\n \"disable\": \"Disable\",\n \"edit\": \"Edit\",\n \"add\": \"Add New Token\",\n \"refresh\": \"Refresh\"\n },\n \"edit\": {\n \"title_edit\": \"Update Token Information\",\n \"title_create\": \"Create New Token\",\n \"name\": \"Name\",\n \"name_placeholder\": \"Please enter name\",\n \"models\": \"Model Scope\",\n \"models_placeholder\": \"Please select allowed models, leave empty for no restrictions\",\n \"ip_limit\": \"IP Restriction\",\n \"ip_limit_placeholder\": \"Please enter allowed subnets, e.g.: 192.168.0.0/24, use commas to separate multiple subnets\",\n \"expire_time\": \"Expiry Time\",\n \"expire_time_placeholder\": \"Please enter expiry time in yyyy-MM-dd HH:mm:ss format, -1 for no limit\",\n \"quota_notice\": \"Note: Token quota only limits the maximum usage of the token itself, actual usage is subject to account remaining quota.\",\n \"quota\": \"Quota\",\n \"quota_placeholder\": \"Please enter quota\",\n \"buttons\": {\n \"never_expire\": \"Never Expire\",\n \"expire_1_month\": \"Expire in 1 Month\",\n \"expire_1_day\": \"Expire in 1 Day\",\n \"expire_1_hour\": \"Expire in 1 Hour\",\n \"expire_1_minute\": \"Expire in 1 Minute\",\n \"unlimited_quota\": \"Set Unlimited Quota\",\n \"cancel_unlimited\": \"Cancel Unlimited Quota\",\n \"submit\": \"Submit\",\n \"cancel\": \"Cancel\"\n },\n \"messages\": {\n \"update_success\": \"Token updated successfully!\",\n \"create_success\": \"Token created successfully, please copy it from the list page!\",\n \"expire_time_invalid\": \"Invalid expiry time format!\"\n }\n },\n \"copy_options\": {\n \"raw\": \"Copy Raw Token\",\n \"ama\": \"Copy AMA Link\",\n \"opencat\": \"Copy OpenCat Link\",\n \"next\": \"Copy NextChat Link\",\n \"lobe\": \"Copy LobeChat Link\"\n },\n \"messages\": {\n \"copy_success\": \"Copied to clipboard!\",\n \"copy_failed\": \"Unable to copy to clipboard, please copy manually. Token has been filled in the search box.\",\n \"operation_success\": \"Operation completed successfully!\"\n },\n \"sort\": {\n \"placeholder\": \"Sort By\",\n \"default\": \"Default Order\",\n \"by_remain\": \"Sort by Remaining Quota\",\n \"by_used\": \"Sort by Used Quota\"\n }\n },\n \"common\": {\n \"quota\": {\n \"display\": \"Equivalent: ${{amount}}\",\n \"display_short\": \"${{amount}}\",\n \"unit\": \"$\"\n }\n },\n \"redemption\": {\n \"title\": \"Redemption Management\",\n \"search\": \"Search redemption codes by ID and name ...\",\n \"table\": {\n \"id\": \"ID\",\n \"name\": \"Name\",\n \"status\": \"Status\",\n \"quota\": \"Quota\",\n \"created_time\": \"Created Time\",\n \"redeemed_time\": \"Redeemed Time\",\n \"actions\": \"Actions\",\n \"no_name\": \"None\",\n \"not_redeemed\": \"Not Redeemed\"\n },\n \"buttons\": {\n \"copy\": \"Copy\",\n \"delete\": \"Delete\",\n \"confirm_delete\": \"Confirm Delete\",\n \"enable\": \"Enable\",\n \"disable\": \"Disable\",\n \"edit\": \"Edit\",\n \"add\": \"Add New Code\",\n \"refresh\": \"Refresh\"\n },\n \"status\": {\n \"unused\": \"Unused\",\n \"disabled\": \"Disabled\",\n \"used\": \"Used\",\n \"unknown\": \"Unknown\"\n },\n \"edit\": {\n \"title_edit\": \"Update Redemption Code\",\n \"title_create\": \"Create New Redemption Code\",\n \"name\": \"Name\",\n \"name_placeholder\": \"Please enter name\",\n \"quota\": \"Quota\",\n \"quota_placeholder\": \"Please enter quota per redemption code\",\n \"count\": \"Generate Count\",\n \"count_placeholder\": \"Please enter number of codes to generate\",\n \"buttons\": {\n \"submit\": \"Submit\",\n \"cancel\": \"Cancel\"\n }\n },\n \"messages\": {\n \"update_success\": \"Redemption code updated successfully!\",\n \"create_success\": \"Redemption code created successfully!\"\n }\n },\n \"log\": {\n \"title\": \"Operation Log\",\n \"search\": \"Search logs...\",\n \"usage_details\": \"Usage Details\",\n \"total_quota\": \"Total Quota Used\",\n \"click_to_view\": \"Click to View\",\n \"type\": {\n \"select\": \"Select Log Type\",\n \"all\": \"All\",\n \"topup\": \"Top Up\",\n \"usage\": \"Usage\",\n \"admin\": \"Admin\",\n \"system\": \"System\",\n \"test\": \"Test\"\n },\n \"table\": {\n \"time\": \"Time\",\n \"channel\": \"Channel\",\n \"type\": \"Type\",\n \"model\": \"Model\",\n \"username\": \"Username\",\n \"token_name\": \"Token Name\",\n \"token_name_placeholder\": \"Optional\",\n \"model_name\": \"Model Name\",\n \"model_name_placeholder\": \"Optional\",\n \"start_time\": \"Start Time\",\n \"end_time\": \"End Time\",\n \"channel_id\": \"Channel ID\",\n \"channel_id_placeholder\": \"Optional\",\n \"username_placeholder\": \"Optional\",\n \"prompt_tokens\": \"Prompt Tokens\",\n \"completion_tokens\": \"Completion Tokens\",\n \"quota\": \"Quota\",\n \"detail\": \"Detail\"\n },\n \"buttons\": {\n \"query\": \"Action\",\n \"submit\": \"Query\",\n \"refresh\": \"Refresh\"\n }\n },\n \"user\": {\n \"title\": \"User Management\",\n \"edit\": {\n \"title\": \"Update User Information\",\n \"username\": \"Username\",\n \"username_placeholder\": \"Please enter new username\",\n \"password\": \"Password\",\n \"password_placeholder\": \"Please enter new password, minimum 8 characters\",\n \"display_name\": \"Display Name\",\n \"display_name_placeholder\": \"Please enter new display name\",\n \"group\": \"Group\",\n \"group_placeholder\": \"Please select group\",\n \"group_addition\": \"Please edit group multipliers in system settings to add new group:\",\n \"quota\": \"Remaining Quota\",\n \"quota_placeholder\": \"Please enter new remaining quota\",\n \"github_id\": \"Linked GitHub Account\",\n \"github_id_placeholder\": \"Read-only, user must link through personal settings page, cannot be modified directly\",\n \"wechat_id\": \"Linked WeChat Account\",\n \"wechat_id_placeholder\": \"Read-only, user must link through personal settings page, cannot be modified directly\",\n \"email\": \"Linked Email Account\",\n \"email_placeholder\": \"Read-only, user must link through personal settings page, cannot be modified directly\",\n \"buttons\": {\n \"submit\": \"Submit\",\n \"cancel\": \"Cancel\"\n }\n },\n \"add\": {\n \"title\": \"Create New User Account\"\n },\n \"messages\": {\n \"update_success\": \"User information updated successfully!\",\n \"create_success\": \"User account created successfully!\",\n \"operation_success\": \"Operation completed successfully!\"\n },\n \"search\": \"Search users...\",\n \"table\": {\n \"id\": \"ID\",\n \"username\": \"Username\",\n \"group\": \"Group\",\n \"quota\": \"Quota\",\n \"role_text\": \"Role\",\n \"status_text\": \"Status\",\n \"actions\": \"Actions\",\n \"remaining_quota\": \"Remaining Quota\",\n \"used_quota\": \"Used Quota\",\n \"request_count\": \"Request Count\",\n \"role_types\": {\n \"normal\": \"Normal User\",\n \"admin\": \"Admin\",\n \"super_admin\": \"Super Admin\",\n \"unknown\": \"Unknown Role\"\n },\n \"status_types\": {\n \"activated\": \"Activated\",\n \"banned\": \"Banned\",\n \"unknown\": \"Unknown Status\"\n },\n \"sort\": {\n \"default\": \"Default Order\",\n \"by_quota\": \"Sort by Remaining Quota\",\n \"by_used_quota\": \"Sort by Used Quota\",\n \"by_request_count\": \"Sort by Request Count\"\n },\n \"sort_by\": \"Sort By\"\n },\n \"buttons\": {\n \"add\": \"Add New User\",\n \"delete\": \"Delete\",\n \"delete_user\": \"Delete User\",\n \"enable\": \"Enable\",\n \"disable\": \"Disable\",\n \"edit\": \"Edit\",\n \"promote\": \"Promote\",\n \"demote\": \"Demote\"\n }\n },\n \"dashboard\": {\n \"charts\": {\n \"requests\": {\n \"title\": \"Model Request Trend\",\n \"tooltip\": \"Request Count\"\n },\n \"quota\": {\n \"title\": \"Quota Usage Trend\",\n \"tooltip\": \"Quota Used\"\n },\n \"tokens\": {\n \"title\": \"Token Usage Trend\",\n \"tooltip\": \"Token Count\"\n }\n },\n \"statistics\": {\n \"title\": \"Statistics\",\n \"tooltip\": {\n \"date\": \"Date\",\n \"value\": \"Value\"\n }\n }\n },\n \"setting\": {\n \"title\": \"System Settings\",\n \"tabs\": {\n \"personal\": \"Personal Settings\",\n \"operation\": \"Operation Settings\",\n \"system\": \"System Settings\",\n \"other\": \"Other Settings\"\n },\n \"personal\": {\n \"general\": {\n \"title\": \"General Settings\",\n \"system_token_notice\": \"Note: The token generated here is for system management, not for requesting OpenAI related services.\",\n \"buttons\": {\n \"update_profile\": \"Update Profile\",\n \"generate_token\": \"Generate System Token\",\n \"copy_invite\": \"Copy Invite Link\",\n \"delete_account\": \"Delete Account\"\n }\n },\n \"binding\": {\n \"title\": \"Account Binding\",\n \"buttons\": {\n \"bind_wechat\": \"Bind WeChat Account\",\n \"bind_github\": \"Bind GitHub Account\",\n \"bind_email\": \"Bind Email Address\",\n \"bind_lark\": \"Bind Lark Account\"\n },\n \"wechat\": {\n \"title\": \"WeChat Binding\",\n \"description\": \"Scan QR code to follow the official account, enter 'verification code' to get the code (valid for 3 minutes)\",\n \"verification_code\": \"Verification Code\",\n \"bind\": \"Bind\"\n },\n \"email\": {\n \"title\": \"Bind Email Address\",\n \"email_placeholder\": \"Enter email address\",\n \"code_placeholder\": \"Verification code\",\n \"get_code\": \"Get Code\",\n \"get_code_retry\": \"Resend({{countdown}})\",\n \"bind\": \"Confirm Binding\",\n \"cancel\": \"Cancel\"\n }\n },\n \"delete_account\": {\n \"title\": \"Dangerous Operation\",\n \"warning\": \"You are deleting your account. All data will be cleared and cannot be recovered\",\n \"confirm_placeholder\": \"Enter your username {{username}} to confirm deletion\",\n \"buttons\": {\n \"confirm\": \"Confirm Delete\",\n \"cancel\": \"Cancel\"\n }\n }\n },\n \"system\": {\n \"general\": {\n \"title\": \"General Settings\",\n \"server_address\": \"Server Address\",\n \"server_address_placeholder\": \"e.g.: https://yourdomain.com\",\n \"buttons\": {\n \"update\": \"Update Server Address\"\n }\n },\n \"login\": {\n \"title\": \"Login & Registration Settings\",\n \"password_login\": \"Allow Password Login\",\n \"password_register\": \"Allow Password Registration\",\n \"email_verification\": \"Require Email Verification for Password Registration\",\n \"github_oauth\": \"Allow GitHub OAuth Login & Registration\",\n \"wechat_login\": \"Allow WeChat Login & Registration\",\n \"registration\": \"Allow New User Registration (When disabled, new users cannot register by any means)\",\n \"turnstile\": \"Enable Turnstile User Verification\"\n },\n \"email_restriction\": {\n \"title\": \"Email Domain Whitelist\",\n \"subtitle\": \"Used to prevent malicious users from batch registering using temporary emails\",\n \"enable\": \"Enable Email Domain Whitelist\",\n \"allowed_domains\": \"Allowed Email Domains\",\n \"add_domain\": \"Add New Allowed Email Domain\",\n \"add_domain_placeholder\": \"Enter new allowed email domain\",\n \"buttons\": {\n \"fill\": \"Fill\",\n \"save\": \"Save Email Domain Whitelist Settings\"\n }\n },\n \"smtp\": {\n \"title\": \"SMTP Configuration\",\n \"subtitle\": \"Used to support system email sending\",\n \"server\": \"SMTP Server Address\",\n \"server_placeholder\": \"e.g.: smtp.gmail.com\",\n \"port\": \"SMTP Port\",\n \"port_placeholder\": \"Default: 587\",\n \"account\": \"SMTP Account\",\n \"account_placeholder\": \"Usually your email address\",\n \"from\": \"SMTP Sender Email\",\n \"from_placeholder\": \"Usually same as email address\",\n \"token\": \"SMTP Access Token\",\n \"token_placeholder\": \"Sensitive information will not be sent to frontend\",\n \"buttons\": {\n \"save\": \"Save SMTP Settings\"\n }\n },\n \"github\": {\n \"title\": \"GitHub OAuth App Configuration\",\n \"subtitle\": \"Used to support GitHub login and registration\",\n \"manage_link\": \"Click here\",\n \"manage_text\": \"to manage your GitHub OAuth Apps\",\n \"url_notice\": \"Set Homepage URL to {{server_url}}, and Authorization callback URL to {{callback_url}}\",\n \"client_id\": \"GitHub Client ID\",\n \"client_id_placeholder\": \"Enter your registered GitHub OAuth APP ID\",\n \"client_secret\": \"GitHub Client Secret\",\n \"client_secret_placeholder\": \"Sensitive information will not be sent to frontend\",\n \"buttons\": {\n \"save\": \"Save GitHub OAuth Settings\"\n }\n },\n \"lark\": {\n \"title\": \"Lark OAuth Configuration\",\n \"subtitle\": \"Used to support Lark login and registration\",\n \"manage_link\": \"Click here\",\n \"manage_text\": \"to manage your Lark applications\",\n \"url_notice\": \"Set Homepage URL to {{server_url}}, and Redirect URL to {{callback_url}}\",\n \"client_id\": \"App ID\",\n \"client_id_placeholder\": \"Enter App ID\",\n \"client_secret\": \"App Secret\",\n \"client_secret_placeholder\": \"Sensitive information will not be sent to frontend\",\n \"buttons\": {\n \"save\": \"Save Lark OAuth Settings\"\n }\n },\n \"wechat\": {\n \"title\": \"WeChat Server Configuration\",\n \"subtitle\": \"Used to support WeChat login and registration\",\n \"learn_more\": \"Learn about WeChat Server\",\n \"server_address\": \"WeChat Server Address\",\n \"server_address_placeholder\": \"e.g.: https://yourdomain.com\",\n \"token\": \"WeChat Server Access Token\",\n \"token_placeholder\": \"Sensitive information will not be sent to frontend\",\n \"qrcode\": \"WeChat Official Account QR Code Image URL\",\n \"qrcode_placeholder\": \"Enter an image URL\",\n \"buttons\": {\n \"save\": \"Save WeChat Server Settings\"\n },\n \"scan_tip\": \"Scan QR code to follow WeChat Official Account, enter 'code' to get verification code (valid for 3 minutes)\",\n \"code_placeholder\": \"Verification code\"\n },\n \"turnstile\": {\n \"title\": \"Turnstile Configuration\",\n \"subtitle\": \"Used to support user verification\",\n \"manage_link\": \"Click here\",\n \"manage_text\": \"to manage your Turnstile Sites, Invisible Widget Type recommended\",\n \"site_key\": \"Turnstile Site Key\",\n \"site_key_placeholder\": \"Enter your registered Turnstile Site Key\",\n \"secret_key\": \"Turnstile Secret Key\",\n \"secret_key_placeholder\": \"Sensitive information will not be sent to frontend\",\n \"buttons\": {\n \"save\": \"Save Turnstile Settings\"\n }\n },\n \"password_login\": {\n \"warning\": {\n \"title\": \"Warning\",\n \"content\": \"Disabling password login will prevent all users (including administrators) who haven't bound other login methods from logging in via password. Confirm disable?\",\n \"buttons\": {\n \"confirm\": \"Confirm\",\n \"cancel\": \"Cancel\"\n }\n }\n }\n },\n \"operation\": {\n \"quota\": {\n \"title\": \"Quota Settings\",\n \"new_user\": \"Initial Quota for New Users\",\n \"new_user_placeholder\": \"e.g.: 100\",\n \"pre_consume\": \"Pre-consumed Quota per Request\",\n \"pre_consume_placeholder\": \"Refund or charge difference after request\",\n \"inviter_reward\": \"Reward Quota for Inviter\",\n \"inviter_reward_placeholder\": \"e.g.: 2000\",\n \"invitee_reward\": \"Reward Quota for Using Invite Code\",\n \"invitee_reward_placeholder\": \"e.g.: 1000\",\n \"buttons\": {\n \"save\": \"Save Quota Settings\"\n }\n },\n \"ratio\": {\n \"title\": \"Ratio Settings\",\n \"model\": {\n \"title\": \"Model Ratio\",\n \"placeholder\": \"A JSON text where keys are model names and values are ratios\"\n },\n \"completion\": {\n \"title\": \"Completion Ratio\",\n \"placeholder\": \"A JSON text where keys are model names and values are ratios. These ratios are the proportion of completion to prompt ratio, which can override One API's internal ratios\"\n },\n \"group\": {\n \"title\": \"Group Ratio\",\n \"placeholder\": \"A JSON text where keys are group names and values are ratios\"\n },\n \"buttons\": {\n \"save\": \"Save Ratio Settings\"\n }\n },\n \"log\": {\n \"title\": \"Log Settings\",\n \"enable_consume\": \"Enable Quota Consumption Logging\",\n \"target_time\": \"Target Time\",\n \"buttons\": {\n \"clean\": \"Clean Historical Logs\"\n }\n },\n \"monitor\": {\n \"title\": \"Monitor Settings\",\n \"max_response_time\": \"Maximum Response Time\",\n \"max_response_time_placeholder\": \"In seconds, channels exceeding this time during testing will be automatically disabled\",\n \"quota_reminder\": \"Quota Reminder Threshold\",\n \"quota_reminder_placeholder\": \"Users will receive email reminders when quota falls below this value\",\n \"auto_disable\": \"Automatically Disable Channel on Failure\",\n \"auto_enable\": \"Automatically Enable Channel on Success\",\n \"buttons\": {\n \"save\": \"Save Monitor Settings\"\n }\n },\n \"general\": {\n \"title\": \"General Settings\",\n \"topup_link\": \"Top-up Link\",\n \"topup_link_placeholder\": \"e.g.: Card selling website purchase link\",\n \"chat_link\": \"Chat Page Link\",\n \"chat_link_placeholder\": \"e.g.: ChatGPT Next Web deployment address\",\n \"quota_per_unit\": \"Quota per Dollar\",\n \"quota_per_unit_placeholder\": \"Quota exchangeable per unit of currency\",\n \"retry_times\": \"Retry Times on Failure\",\n \"retry_times_placeholder\": \"Number of retry attempts on failure\",\n \"display_in_currency\": \"Display Quota in Currency Format\",\n \"display_token_stat\": \"Show Token Quota Instead of User Quota in Billing APIs\",\n \"approximate_token\": \"Use Approximate Method to Estimate Token Count\",\n \"buttons\": {\n \"save\": \"Save General Settings\"\n }\n }\n },\n \"other\": {\n \"notice\": {\n \"title\": \"Notice Settings\",\n \"content\": \"Notice Content\",\n \"content_placeholder\": \"Enter new notice content here, supports Markdown & HTML code\",\n \"buttons\": {\n \"save\": \"Save Notice\"\n }\n },\n \"system\": {\n \"title\": \"System Settings\",\n \"name\": \"System Name\",\n \"name_placeholder\": \"Please enter system name\",\n \"logo\": \"Logo Image URL\",\n \"logo_placeholder\": \"Enter Logo image URL here\",\n \"theme\": {\n \"title\": \"Theme Name\",\n \"link\": \"Available Themes\",\n \"placeholder\": \"Please enter theme name\"\n },\n \"buttons\": {\n \"save_name\": \"Set System Name\",\n \"save_logo\": \"Set Logo\",\n \"save_theme\": \"Set Theme (Restart Required)\"\n }\n },\n \"content\": {\n \"title\": \"Content Settings\",\n \"homepage\": {\n \"title\": \"Homepage Content\",\n \"placeholder\": \"Enter homepage content here, supports Markdown & HTML code. Status information will not be shown after setting. If a link is entered, it will be used as the src attribute of an iframe, allowing you to set any webpage as homepage.\"\n },\n \"about\": {\n \"title\": \"About System\",\n \"description\": \"You can set about content in settings page, supports HTML & Markdown\",\n \"repository\": \"Project Repository:\",\n \"loading_failed\": \"Failed to load about content...\"\n },\n \"footer\": {\n \"title\": \"Footer\",\n \"placeholder\": \"Enter new footer here, leave empty to use default footer, supports HTML code\"\n },\n \"buttons\": {\n \"save_homepage\": \"Save Homepage Content\",\n \"save_about\": \"Save About\",\n \"save_footer\": \"Set Footer\"\n }\n },\n \"copyright\": {\n \"notice\": \"Removing One API's copyright notice requires authorization. Project maintenance requires significant effort, if this project is meaningful to you, please actively support it.\"\n }\n }\n },\n \"footer\": {\n \"built_by\": \"built by\",\n \"built_by_name\": \"JustSong\",\n \"license\": \", source code is licensed under the\",\n \"mit\": \"MIT License\"\n },\n \"home\": {\n \"welcome\": {\n \"title\": \"Welcome to One API\",\n \"description\": \"One API is a LLM API management and distribution system that helps you better manage and use LLM APIs from various providers.\",\n \"login_notice\": \"To use the service, please login or register first.\"\n },\n \"system_status\": {\n \"title\": \"System Status\",\n \"info\": {\n \"title\": \"System Information\",\n \"name\": \"Name: \",\n \"version\": \"Version: \",\n \"source\": \"Source: \",\n \"source_link\": \"GitHub Repository\",\n \"start_time\": \"Start Time: \"\n },\n \"config\": {\n \"title\": \"System Configuration\",\n \"email_verify\": \"Email Verification: \",\n \"github_oauth\": \"GitHub OAuth: \",\n \"wechat_login\": \"WeChat Login: \",\n \"turnstile\": \"Turnstile Check: \",\n \"enabled\": \"Enabled\",\n \"disabled\": \"Disabled\"\n }\n },\n \"loading_failed\": \"Failed to load homepage content...\"\n },\n \"auth\": {\n \"login\": {\n \"title\": \"User Login\",\n \"username\": \"Username / Email\",\n \"password\": \"Password\",\n \"button\": \"Login\",\n \"forgot_password\": \"Forgot password?\",\n \"reset_password\": \"Reset\",\n \"no_account\": \"No account?\",\n \"register\": \"Register\",\n \"other_methods\": \"Other login methods\",\n \"wechat\": {\n \"scan_tip\": \"Scan QR code to follow WeChat Official Account, enter 'code' to get verification code (valid for 3 minutes)\",\n \"code_placeholder\": \"Verification code\"\n }\n },\n \"register\": {\n \"title\": \"New User Registration\",\n \"username\": \"Username (max 12 characters)\",\n \"password\": \"Password (8-20 characters)\",\n \"confirm_password\": \"Confirm password\",\n \"email\": \"Email address\",\n \"verification_code\": \"Verification code\",\n \"get_code\": \"Get code\",\n \"get_code_retry\": \"Retry ({{countdown}})\",\n \"button\": \"Register\",\n \"has_account\": \"Have an account?\",\n \"login\": \"Login\"\n },\n \"reset\": {\n \"title\": \"Password Reset\",\n \"email\": \"Email address\",\n \"button\": \"Submit\",\n \"notice\": \"The system will send an email containing a reset link to your mailbox. Please check your email.\",\n \"confirm\": {\n \"title\": \"Password Reset Confirmation\",\n \"new_password\": \"New password\",\n \"button\": \"Submit\",\n \"button_disabled\": \"Password reset completed\",\n \"notice\": \"New password has been generated, please click the password field or button above to copy. Please login and change your password as soon as possible!\"\n }\n }\n },\n \"about\": {\n \"title\": \"About\",\n \"description\": \"One API is an open-source API management and proxy platform.\",\n \"repository\": \"Repository: \",\n \"loading_failed\": \"Loading failed\"\n },\n \"messages\": {\n \"success\": {\n \"login\": \"Login successful!\",\n \"register\": \"Registration successful!\",\n \"verification_code\": \"Verification code sent, please check your email!\",\n \"password_reset\": \"Reset email sent, please check your inbox!\"\n },\n \"error\": {\n \"login_expired\": \"Not logged in or session expired, please login again!\",\n \"password_length\": \"Password must be at least 8 characters!\",\n \"password_mismatch\": \"Passwords do not match\",\n \"turnstile_wait\": \"Please wait a few seconds, Turnstile is checking the environment!\",\n \"root_password\": \"Please change the default password immediately!\"\n },\n \"notice\": {\n \"password_copied\": \"New password copied to clipboard: {{password}}\"\n }\n }\n}\n"
},
{
"path": "web/default/src/locales/zh/translation.json",
"content": "{\n \"header\": {\n \"home\": \"首页\",\n \"channel\": \"渠道\",\n \"token\": \"令牌\",\n \"redemption\": \"兑换\",\n \"topup\": \"充值\",\n \"user\": \"用户\",\n \"dashboard\": \"总览\",\n \"log\": \"日志\",\n \"setting\": \"设置\",\n \"about\": \"关于\",\n \"chat\": \"聊天\",\n \"login\": \"登录\",\n \"logout\": \"注销\",\n \"register\": \"注册\"\n },\n \"topup\": {\n \"title\": \"充值中心\",\n \"get_code\": {\n \"title\": \"获取兑换码\",\n \"current_quota\": \"当前可用额度\",\n \"button\": \"立即获取兑换码\"\n },\n \"redeem_code\": {\n \"title\": \"兑换码充值\",\n \"placeholder\": \"请输入兑换码\",\n \"paste\": \"粘贴\",\n \"paste_error\": \"无法访问剪贴板，请手动粘贴\",\n \"submit\": \"立即兑换\",\n \"submitting\": \"兑换中...\",\n \"empty_code\": \"请输入兑换码！\",\n \"success\": \"充值成功！\",\n \"request_failed\": \"请求失败\",\n \"no_link\": \"超级管理员未设置充值链接！\"\n }\n },\n \"channel\": {\n \"title\": \"管理渠道\",\n \"search\": \"搜索渠道的 ID，名称和密钥 ...\",\n \"balance_notice\": \"OpenAI 渠道已经不再支持通过 key 获取余额，因此余额显示为 0。对于支持的渠道类型，请点击余额进行刷新。\",\n \"test_notice\": \"渠道测试仅支持 chat 模型，优先使用 gpt-3.5-turbo，如果该模型不可用则使用你所配置的模型列表中的第一个模型。\",\n \"detail_notice\": \"点击下方详情按钮可以显示余额以及设置额外的测试模型。\",\n \"table\": {\n \"id\": \"ID\",\n \"name\": \"名称\",\n \"group\": \"分组\",\n \"type\": \"类型\",\n \"status\": \"状态\",\n \"response_time\": \"响应时间\",\n \"balance\": \"余额\",\n \"priority\": \"优先级\",\n \"test_model\": \"测试模型\",\n \"actions\": \"操作\",\n \"no_name\": \"无\",\n \"status_enabled\": \"已启用\",\n \"status_disabled\": \"已禁用\",\n \"status_auto_disabled\": \"已禁用\",\n \"status_disabled_tip\": \"本渠道被手动禁用\",\n \"status_auto_disabled_tip\": \"本渠道被程序自动禁用\",\n \"status_unknown\": \"未知状态\",\n \"not_tested\": \"未测试\",\n \"priority_tip\": \"渠道选择优先级，越高越优先\",\n \"select_test_model\": \"请选择测试模型\",\n \"click_to_update\": \"点击更新\",\n \"balance_not_supported\": \"-\"\n },\n \"buttons\": {\n \"test\": \"测试\",\n \"delete\": \"删除\",\n \"confirm_delete\": \"删除渠道\",\n \"enable\": \"启用\",\n \"disable\": \"禁用\",\n \"edit\": \"编辑\",\n \"add\": \"添加新的渠道\",\n \"test_all\": \"测试所有渠道\",\n \"test_disabled\": \"测试禁用渠道\",\n \"delete_disabled\": \"删除禁用渠道\",\n \"confirm_delete_disabled\": \"确认删除\",\n \"refresh\": \"刷新\",\n \"show_detail\": \"详情\",\n \"hide_detail\": \"隐藏详情\"\n },\n \"messages\": {\n \"test_success\": \"渠道 {{name}} 测试成功，模型 {{model}}，耗时 {{time}} 秒，模型输出：{{message}}\",\n \"test_all_started\": \"已成功开始测试渠道，请刷新页面查看结果。\",\n \"delete_disabled_success\": \"已删除所有禁用渠道，共计 {{count}} 个\",\n \"balance_update_success\": \"渠道 {{name}} 余额更新成功！\",\n \"all_balance_updated\": \"已更新完毕所有已启用渠道余额！\",\n \"operation_success\": \"操作成功完成！\"\n },\n \"edit\": {\n \"title_edit\": \"更新渠道信息\",\n \"title_create\": \"创建新的渠道\",\n \"type\": \"类型\",\n \"name\": \"名称\",\n \"name_placeholder\": \"请输入名称\",\n \"group\": \"分组\",\n \"group_placeholder\": \"请选择可以使用该渠道的分组\",\n \"group_addition\": \"请在系统设置页面编辑分组倍率以添加新的分组：\",\n \"models\": \"模型\",\n \"models_placeholder\": \"请选择该渠道所支持的模型\",\n \"model_mapping\": \"模型重定向\",\n \"model_mapping_placeholder\": \"此项可选，用于修改请求体中的模型名称，为一个 JSON 字符串，键为请求中模型名称，值为要替换的模型名称\",\n \"system_prompt\": \"系统提示词\",\n \"system_prompt_placeholder\": \"此项可选，用于强制设置给定的系统提示词，请配合自定义模型 & 模型重定向使用，首先创建一个唯一的自定义模型名称并在上面填入，之后将该自定义模型重定向映射到该渠道一个原生支持的模型\",\n \"proxy_url\": \"代理\",\n \"proxy_url_placeholder\": \"此项可选，用于通过代理站来进行 API 调用，请输入代理站地址，格式为：https://domain.com。注意，这里所需要填入的代理地址仅会在实际请求时替换域名部分，如果你想填入 OpenAI SDK 中所要求的 Base URL，请使用 OpenAI 兼容渠道类型\",\n \"base_url\": \"Base URL\",\n \"base_url_placeholder\": \"OpenAPI SDK 中所要求的 Base URL\",\n \"key\": \"密钥\",\n \"key_placeholder\": \"请输入密钥\",\n \"batch\": \"批量创建\",\n \"batch_placeholder\": \"请输入密钥，一行一个\",\n \"buttons\": {\n \"cancel\": \"取消\",\n \"submit\": \"提交\",\n \"fill_models\": \"填入相关模型\",\n \"fill_all\": \"填入所有模型\",\n \"clear\": \"清除所有模型\",\n \"add_custom\": \"填入\",\n \"custom_placeholder\": \"输入自定义模型名称\"\n },\n \"messages\": {\n \"name_required\": \"请填写渠道名称和渠道密钥！\",\n \"models_required\": \"请至少选择一个模型！\",\n \"model_mapping_invalid\": \"模型映射必须是合法的 JSON 格式！\",\n \"update_success\": \"渠道更新成功！\",\n \"create_success\": \"渠道创建成功！\"\n },\n \"spark_version\": \"模型版本\",\n \"spark_version_placeholder\": \"请输入星火大模型版本，注意是接口地址中的版本号，例如：v2.1\",\n \"knowledge_id\": \"知识库 ID\",\n \"knowledge_id_placeholder\": \"请输入知识库 ID，例如：123456\",\n \"plugin_param\": \"插件参数\",\n \"plugin_param_placeholder\": \"请输入插件参数，即 X-DashScope-Plugin 请求头的取值\",\n \"coze_notice\": \"对于 Coze 而言，模型名称即 Bot ID，你可以添加一个前缀 `bot-`，例如：`bot-123456`。\",\n \"douban_notice\": \"对于豆包而言，需要手动去\",\n \"douban_notice_link\": \"模型推理页面\",\n \"douban_notice_2\": \"创建推理接入点，以接入点名称作为模型名称，例如：`ep-20240608051426-tkxvl`。你可以结合模型重定向功能将其转换为常规的模型名称，例如：doubao-lite-4k -> ep-20240608051426-tkxvl（前者作为 JSON 的 key，后者作为 value）。注意，doubao-lite-4k 和 ep-20240608051426-tkxvl 都需要通过自定义模型的方式填入到本渠道的模型列表中。\",\n \"aws_region_placeholder\": \"region，例如：us-west-2\",\n \"aws_ak_placeholder\": \"AWS IAM Access Key\",\n \"aws_sk_placeholder\": \"AWS IAM Secret Key\",\n \"vertex_region_placeholder\": \"Vertex AI Region，例如：us-east5\",\n \"vertex_project_id\": \"Vertex AI Project ID\",\n \"vertex_project_id_placeholder\": \"Vertex AI Project ID\",\n \"vertex_credentials\": \"Google Cloud Application Default Credentials JSON\",\n \"vertex_credentials_placeholder\": \"Google Cloud Application Default Credentials JSON\",\n \"user_id\": \"User ID\",\n \"user_id_placeholder\": \"生成该密钥的用户 ID\",\n \"key_prompts\": {\n \"default\": \"请输入渠道对应的鉴权密钥\",\n \"zhipu\": \"按照如下格式输入：APIKey|SecretKey\",\n \"spark\": \"按照如下格式输入：APPID|APISecret|APIKey\",\n \"fastgpt\": \"按照如下格式输入：APIKey-AppId，例如：fastgpt-0sp2gtvfdgyi4k30jwlgwf1i-64f335d84283f05518e9e041\",\n \"tencent\": \"按照如下格式输入：AppId|SecretId|SecretKey\"\n }\n }\n },\n \"token\": {\n \"title\": \"令牌管理\",\n \"search\": \"搜索令牌的名称 ...\",\n \"table\": {\n \"name\": \"名称\",\n \"status\": \"状态\",\n \"used_quota\": \"已用额度\",\n \"remain_quota\": \"剩余额度\",\n \"created_time\": \"创建时间\",\n \"expired_time\": \"过期时间\",\n \"actions\": \"操作\",\n \"no_name\": \"无\",\n \"never_expire\": \"永不过期\",\n \"unlimited\": \"无限制\",\n \"status_enabled\": \"已启用\",\n \"status_disabled\": \"已禁用\",\n \"status_expired\": \"已过期\",\n \"status_depleted\": \"已耗尽\",\n \"status_unknown\": \"未知状态\"\n },\n \"buttons\": {\n \"copy\": \"复制\",\n \"chat\": \"聊天\",\n \"delete\": \"删除\",\n \"confirm_delete\": \"删除令牌\",\n \"enable\": \"启用\",\n \"disable\": \"禁用\",\n \"edit\": \"编辑\",\n \"add\": \"添加新的令牌\",\n \"refresh\": \"刷新\"\n },\n \"edit\": {\n \"title_edit\": \"更新令牌信息\",\n \"title_create\": \"创建新的令牌\",\n \"name\": \"名称\",\n \"name_placeholder\": \"请输入名称\",\n \"models\": \"模型范围\",\n \"models_placeholder\": \"请选择允许使用的模型，留空则不进行限制\",\n \"ip_limit\": \"IP 限制\",\n \"ip_limit_placeholder\": \"请输入允许访问的网段，例如：192.168.0.0/24，请使用英文逗号分隔多个网段\",\n \"expire_time\": \"过期时间\",\n \"expire_time_placeholder\": \"请输入过期时间，格式为 yyyy-MM-dd HH:mm:ss，-1 表示无限制\",\n \"quota_notice\": \"注意，令牌的额度仅用于限制令牌本身的最大额度使用量，实际的使用受到账户的剩余额度限制。\",\n \"quota\": \"额度\",\n \"quota_placeholder\": \"请输入额度\",\n \"buttons\": {\n \"never_expire\": \"永不过期\",\n \"expire_1_month\": \"一个月后过期\",\n \"expire_1_day\": \"一天后过期\",\n \"expire_1_hour\": \"一小时后过期\",\n \"expire_1_minute\": \"一分钟后过期\",\n \"unlimited_quota\": \"设为无限额度\",\n \"cancel_unlimited\": \"取消无限额度\",\n \"submit\": \"提交\",\n \"cancel\": \"取消\"\n },\n \"messages\": {\n \"update_success\": \"令牌更新成功！\",\n \"create_success\": \"令牌创建成功，请在列表页面点击复制获取令牌！\",\n \"expire_time_invalid\": \"过期时间格式错误！\"\n }\n },\n \"copy_options\": {\n \"raw\": \"复制原始令牌\",\n \"ama\": \"复制 AMA 链接\",\n \"opencat\": \"复制 OpenCat 链接\",\n \"next\": \"复制 NextChat 链接\",\n \"lobe\": \"复制 LobeChat 链接\"\n },\n \"messages\": {\n \"copy_success\": \"已复制到剪贴板！\",\n \"copy_failed\": \"无法复制到剪贴板，请手动复制，已将令牌填入搜索框。\",\n \"operation_success\": \"操作成功完成！\"\n },\n \"sort\": {\n \"placeholder\": \"排序方式\",\n \"default\": \"默认排序\",\n \"by_remain\": \"按剩余额度排序\",\n \"by_used\": \"按已用额度排序\"\n }\n },\n \"common\": {\n \"quota\": {\n \"display\": \"等价金额：${{amount}}\",\n \"display_short\": \"${{amount}}\",\n \"unit\": \"$\"\n }\n },\n \"redemption\": {\n \"title\": \"兑换管理\",\n \"search\": \"搜索兑换码的 ID 和名称 ...\",\n \"table\": {\n \"id\": \"ID\",\n \"name\": \"名称\",\n \"status\": \"状态\",\n \"quota\": \"额度\",\n \"created_time\": \"创建时间\",\n \"redeemed_time\": \"兑换时间\",\n \"actions\": \"操作\",\n \"no_name\": \"无\",\n \"not_redeemed\": \"尚未兑换\"\n },\n \"buttons\": {\n \"copy\": \"复制\",\n \"delete\": \"删除\",\n \"confirm_delete\": \"确认删除\",\n \"enable\": \"启用\",\n \"disable\": \"禁用\",\n \"edit\": \"编辑\",\n \"add\": \"添加新的兑换码\",\n \"refresh\": \"刷新\"\n },\n \"status\": {\n \"unused\": \"未使用\",\n \"disabled\": \"已禁用\",\n \"used\": \"已使用\",\n \"unknown\": \"未知状态\"\n },\n \"edit\": {\n \"title_edit\": \"更新兑换码信息\",\n \"title_create\": \"创建新的兑换码\",\n \"name\": \"名称\",\n \"name_placeholder\": \"请输入名称\",\n \"quota\": \"额度\",\n \"quota_placeholder\": \"请输入单个兑换码中包含的额度\",\n \"count\": \"生成数量\",\n \"count_placeholder\": \"请输入生成数量\",\n \"buttons\": {\n \"submit\": \"提交\",\n \"cancel\": \"取消\"\n }\n },\n \"messages\": {\n \"update_success\": \"兑换码更新成功！\",\n \"create_success\": \"兑换码创建成功！\"\n }\n },\n \"log\": {\n \"title\": \"操作日志\",\n \"search\": \"搜索日志...\",\n \"usage_details\": \"使用明细\",\n \"total_quota\": \"总消耗额度\",\n \"click_to_view\": \"点击查看\",\n \"type\": {\n \"select\": \"选择明细分类\",\n \"all\": \"全部\",\n \"topup\": \"充值\",\n \"usage\": \"消费\",\n \"admin\": \"管理\",\n \"system\": \"系统\",\n \"test\": \"测试\"\n },\n \"table\": {\n \"time\": \"时间\",\n \"channel\": \"渠道\",\n \"type\": \"类型\",\n \"model\": \"模型\",\n \"username\": \"用户名\",\n \"token_name\": \"令牌名称\",\n \"token_name_placeholder\": \"可选值\",\n \"model_name\": \"模型名称\",\n \"model_name_placeholder\": \"可选值\",\n \"start_time\": \"起始时间\",\n \"end_time\": \"结束时间\",\n \"channel_id\": \"渠道 ID\",\n \"channel_id_placeholder\": \"可选值\",\n \"username_placeholder\": \"可选值\",\n \"prompt_tokens\": \"提示词消耗\",\n \"completion_tokens\": \"补全消耗\",\n \"quota\": \"额度\",\n \"detail\": \"详情\"\n },\n \"buttons\": {\n \"query\": \"操作\",\n \"submit\": \"查询\",\n \"refresh\": \"刷新\"\n }\n },\n \"user\": {\n \"title\": \"用户管理\",\n \"edit\": {\n \"title\": \"更新用户信息\",\n \"username\": \"用户名\",\n \"username_placeholder\": \"请输入新的用户名\",\n \"password\": \"密码\",\n \"password_placeholder\": \"请输入新的密码，最短 8 位\",\n \"display_name\": \"显示名称\",\n \"display_name_placeholder\": \"请输入新的显示名称\",\n \"group\": \"分组\",\n \"group_placeholder\": \"请选择分组\",\n \"group_addition\": \"请在系统设置页面编辑分组倍率以添加新的分组：\",\n \"quota\": \"剩余额度\",\n \"quota_placeholder\": \"请输入新的剩余额度\",\n \"github_id\": \"已绑定的 GitHub 账户\",\n \"github_id_placeholder\": \"此项只读，需要用户通过个人设置页面的相关绑定按钮进行绑定，不可直接修改\",\n \"wechat_id\": \"已绑定的微信账户\",\n \"wechat_id_placeholder\": \"此项只读，需要用户通过个人设置页面的相关绑定按钮进行绑定，不可直接修改\",\n \"email\": \"已绑定的邮箱账户\",\n \"email_placeholder\": \"此项只读，需要用户通过个人设置页面的相关绑定按钮进行绑定，不可直接修改\",\n \"buttons\": {\n \"submit\": \"提交\",\n \"cancel\": \"取消\"\n }\n },\n \"add\": {\n \"title\": \"创建新用户账户\"\n },\n \"messages\": {\n \"update_success\": \"用户信息更新成功！\",\n \"create_success\": \"用户账户创建成功！\",\n \"operation_success\": \"操作成功完成！\"\n },\n \"search\": \"搜索用户...\",\n \"table\": {\n \"id\": \"ID\",\n \"username\": \"用户名\",\n \"group\": \"分组\",\n \"quota\": \"额度\",\n \"role_text\": \"角色\",\n \"status_text\": \"状态\",\n \"actions\": \"操作\",\n \"remaining_quota\": \"剩余额度\",\n \"used_quota\": \"已用额度\",\n \"request_count\": \"请求次数\",\n \"role_types\": {\n \"normal\": \"普通用户\",\n \"admin\": \"管理员\",\n \"super_admin\": \"超级管理员\",\n \"unknown\": \"未知身份\"\n },\n \"status_types\": {\n \"activated\": \"已激活\",\n \"banned\": \"已封禁\",\n \"unknown\": \"未知状态\"\n },\n \"sort\": {\n \"default\": \"默认排序\",\n \"by_quota\": \"按剩余额度排序\",\n \"by_used_quota\": \"按已用额度排序\",\n \"by_request_count\": \"按请求次数排序\"\n },\n \"sort_by\": \"排序方式\"\n },\n \"buttons\": {\n \"add\": \"添加新的用户\",\n \"delete\": \"删除\",\n \"delete_user\": \"删除用户\",\n \"enable\": \"启用\",\n \"disable\": \"禁用\",\n \"edit\": \"编辑\",\n \"promote\": \"提升\",\n \"demote\": \"降级\"\n }\n },\n \"dashboard\": {\n \"charts\": {\n \"requests\": {\n \"title\": \"模型请求趋势\",\n \"tooltip\": \"请求次数\"\n },\n \"quota\": {\n \"title\": \"额度消费趋势\",\n \"tooltip\": \"消费额度\"\n },\n \"tokens\": {\n \"title\": \"Token 消费趋势\",\n \"tooltip\": \"Token 数量\"\n }\n },\n \"statistics\": {\n \"title\": \"统计\",\n \"tooltip\": {\n \"date\": \"日期\",\n \"value\": \"数值\"\n }\n }\n },\n \"setting\": {\n \"title\": \"系统设置\",\n \"tabs\": {\n \"personal\": \"个人设置\",\n \"operation\": \"运营设置\",\n \"system\": \"系统设置\",\n \"other\": \"其他设置\"\n },\n \"personal\": {\n \"general\": {\n \"title\": \"通用设置\",\n \"system_token_notice\": \"注意，此处生成的令牌用于系统管理，而非用于请求 OpenAI 相关的服务，请知悉。\",\n \"buttons\": {\n \"update_profile\": \"更新个人信息\",\n \"generate_token\": \"生成系统访问令牌\",\n \"copy_invite\": \"复制邀请链接\",\n \"delete_account\": \"删除个人账户\"\n }\n },\n \"binding\": {\n \"title\": \"账号绑定\",\n \"buttons\": {\n \"bind_wechat\": \"绑定微信账号\",\n \"bind_github\": \"绑定 GitHub 账号\",\n \"bind_email\": \"绑定邮箱地址\",\n \"bind_lark\": \"绑定飞书账号\"\n },\n \"wechat\": {\n \"title\": \"微信绑定\",\n \"description\": \"微信扫码关注公众号，输入「验证码」获取验证码（三分钟内有效）\",\n \"verification_code\": \"验证码\",\n \"bind\": \"绑定\"\n },\n \"email\": {\n \"title\": \"绑定邮箱地址\",\n \"email_placeholder\": \"输入邮箱地址\",\n \"code_placeholder\": \"验证码\",\n \"get_code\": \"获取验证码\",\n \"get_code_retry\": \"重新发送({{countdown}})\",\n \"bind\": \"确认绑定\",\n \"cancel\": \"取消\"\n }\n },\n \"delete_account\": {\n \"title\": \"危险操作\",\n \"warning\": \"您正在删除自己的帐户，将清空所有数据且不可恢复\",\n \"confirm_placeholder\": \"输入你的账户名 {{username}} 以确认删除\",\n \"buttons\": {\n \"confirm\": \"确认删除\",\n \"cancel\": \"取消\"\n }\n }\n },\n \"system\": {\n \"general\": {\n \"title\": \"通用设置\",\n \"server_address\": \"服务器地址\",\n \"server_address_placeholder\": \"例如：https://yourdomain.com\",\n \"buttons\": {\n \"update\": \"更新服务器地址\"\n }\n },\n \"login\": {\n \"title\": \"配置登录注册\",\n \"password_login\": \"允许通过密码进行登录\",\n \"password_register\": \"允许通过密码进行注册\",\n \"email_verification\": \"通过密码注册时需要进行邮箱验证\",\n \"github_oauth\": \"允许通过 GitHub 账户登录 & 注册\",\n \"wechat_login\": \"允许通过微信登录 & 注册\",\n \"registration\": \"允许新用户注册（此项为否时，新用户将无法以任何方式进行注册）\",\n \"turnstile\": \"启用 Turnstile 用户校验\"\n },\n \"email_restriction\": {\n \"title\": \"配置邮箱域名白名单\",\n \"subtitle\": \"用以防止恶意用户利用临时邮箱批量注册\",\n \"enable\": \"启用邮箱域名白名单\",\n \"allowed_domains\": \"允许的邮箱域名\",\n \"add_domain\": \"添加新的允许的邮箱域名\",\n \"add_domain_placeholder\": \"输入新的允许的邮箱域名\",\n \"buttons\": {\n \"fill\": \"填入\",\n \"save\": \"保存邮箱域名白名单设置\"\n }\n },\n \"smtp\": {\n \"title\": \"配置 SMTP\",\n \"subtitle\": \"用以支持系统的邮件发送\",\n \"server\": \"SMTP 服务器地址\",\n \"server_placeholder\": \"例如：smtp.qq.com\",\n \"port\": \"SMTP 端口\",\n \"port_placeholder\": \"默认: 587\",\n \"account\": \"SMTP 账户\",\n \"account_placeholder\": \"通常是邮箱地址\",\n \"from\": \"SMTP 发送者邮箱\",\n \"from_placeholder\": \"通常和邮箱地址保持一致\",\n \"token\": \"SMTP 访问凭证\",\n \"token_placeholder\": \"敏感信息不会发送到前端显示\",\n \"buttons\": {\n \"save\": \"保存 SMTP 设置\"\n }\n },\n \"github\": {\n \"title\": \"配置 GitHub OAuth App\",\n \"subtitle\": \"用以支持通过 GitHub 进行登录注册\",\n \"manage_link\": \"点击此处\",\n \"manage_text\": \"管理你的 GitHub OAuth App\",\n \"url_notice\": \"Homepage URL 填 {{server_url}}，Authorization callback URL 填 {{callback_url}}\",\n \"client_id\": \"GitHub Client ID\",\n \"client_id_placeholder\": \"输入你注册的 GitHub OAuth APP 的 ID\",\n \"client_secret\": \"GitHub Client Secret\",\n \"client_secret_placeholder\": \"敏感信息不会发送到前端显示\",\n \"buttons\": {\n \"save\": \"保存 GitHub OAuth 设置\"\n }\n },\n \"lark\": {\n \"title\": \"配置飞书授权登录\",\n \"subtitle\": \"用以支持通过飞书进行登录注册\",\n \"manage_link\": \"点击此处\",\n \"manage_text\": \"管理你的飞书应用\",\n \"url_notice\": \"主页链接填 {{server_url}}，重定向 URL 填 {{callback_url}}\",\n \"client_id\": \"App ID\",\n \"client_id_placeholder\": \"输入 App ID\",\n \"client_secret\": \"App Secret\",\n \"client_secret_placeholder\": \"敏感信息不会发送到前端显示\",\n \"buttons\": {\n \"save\": \"保存飞书 OAuth 设置\"\n }\n },\n \"wechat\": {\n \"title\": \"配置 WeChat Server\",\n \"subtitle\": \"用以支持通过微信进行登录注册\",\n \"learn_more\": \"了解 WeChat Server\",\n \"server_address\": \"WeChat Server 服务器地址\",\n \"server_address_placeholder\": \"例如：https://yourdomain.com\",\n \"token\": \"WeChat Server 访问凭证\",\n \"token_placeholder\": \"敏感信息不会发送到前端显示\",\n \"qrcode\": \"微信公众号二维码图片链接\",\n \"qrcode_placeholder\": \"输入一个图片链接\",\n \"buttons\": {\n \"save\": \"保存 WeChat Server 设置\"\n }\n },\n \"turnstile\": {\n \"title\": \"配置 Turnstile\",\n \"subtitle\": \"用以支持用户校验\",\n \"manage_link\": \"点击此处\",\n \"manage_text\": \"管理你的 Turnstile Sites，推荐选择 Invisible Widget Type\",\n \"site_key\": \"Turnstile Site Key\",\n \"site_key_placeholder\": \"输入你注册的 Turnstile Site Key\",\n \"secret_key\": \"Turnstile Secret Key\",\n \"secret_key_placeholder\": \"敏感信息不会发送到前端显示\",\n \"buttons\": {\n \"save\": \"保存 Turnstile 设置\"\n }\n },\n \"password_login\": {\n \"warning\": {\n \"title\": \"警告\",\n \"content\": \"取消密码登录将导致所有未绑定其他登录方式的用户（包括管理员）无法通过密码登录，确认取消？\",\n \"buttons\": {\n \"confirm\": \"确定\",\n \"cancel\": \"取消\"\n }\n }\n }\n },\n \"operation\": {\n \"quota\": {\n \"title\": \"额度设置\",\n \"new_user\": \"新用户初始额度\",\n \"new_user_placeholder\": \"例如：100\",\n \"pre_consume\": \"请求预扣费额度\",\n \"pre_consume_placeholder\": \"请求结束后多退少补\",\n \"inviter_reward\": \"邀请新用户奖励额度\",\n \"inviter_reward_placeholder\": \"例如：2000\",\n \"invitee_reward\": \"新用户使用邀请码奖励额度\",\n \"invitee_reward_placeholder\": \"例如：1000\",\n \"buttons\": {\n \"save\": \"保存额度设置\"\n }\n },\n \"ratio\": {\n \"title\": \"倍率设置\",\n \"model\": {\n \"title\": \"模型倍率\",\n \"placeholder\": \"为一个 JSON 文本，键为模型名称，值为倍率\"\n },\n \"completion\": {\n \"title\": \"补全倍率\",\n \"placeholder\": \"为一个 JSON 文本，键为模型名称，值为倍率，此处的倍率设置是模型补全倍率相较于提示倍率的比例，使用该设置可强制覆盖 One API 的内部比例\"\n },\n \"group\": {\n \"title\": \"分组倍率\",\n \"placeholder\": \"为一个 JSON 文本，键为分组名称，值为倍率\"\n },\n \"buttons\": {\n \"save\": \"保存倍率设置\"\n }\n },\n \"log\": {\n \"title\": \"日志设置\",\n \"enable_consume\": \"启用额度消费日志记录\",\n \"target_time\": \"目标时间\",\n \"buttons\": {\n \"clean\": \"清理历史日志\"\n }\n },\n \"monitor\": {\n \"title\": \"监控设置\",\n \"max_response_time\": \"最长响应时间\",\n \"max_response_time_placeholder\": \"单位秒，当运行渠道全部测试时，超过此时间将自动禁用渠道\",\n \"quota_reminder\": \"额度提醒阈值\",\n \"quota_reminder_placeholder\": \"低于此额度时将发送邮件提醒用户\",\n \"auto_disable\": \"失败时自动禁用渠道\",\n \"auto_enable\": \"成功时自动启用渠道\",\n \"buttons\": {\n \"save\": \"保存监控设置\"\n }\n },\n \"general\": {\n \"title\": \"通用设置\",\n \"topup_link\": \"充值链接\",\n \"topup_link_placeholder\": \"例如发卡网站的购买链接\",\n \"chat_link\": \"聊天页面链接\",\n \"chat_link_placeholder\": \"例如 ChatGPT Next Web 的部署地址\",\n \"quota_per_unit\": \"单位美元额度\",\n \"quota_per_unit_placeholder\": \"一单位货币能兑换的额度\",\n \"retry_times\": \"失败重试次数\",\n \"retry_times_placeholder\": \"失败重试次数\",\n \"display_in_currency\": \"以货币形式显示额度\",\n \"display_token_stat\": \"Billing 相关 API 显示令牌额度而非用户额度\",\n \"approximate_token\": \"使用近似的方式估算 token 数以减少计算量\",\n \"buttons\": {\n \"save\": \"保存通用设置\"\n }\n }\n },\n \"other\": {\n \"notice\": {\n \"title\": \"公告设置\",\n \"content\": \"公告内容\",\n \"content_placeholder\": \"在此输入新的公告内容，支持 Markdown & HTML 代码\",\n \"buttons\": {\n \"save\": \"保存公告\"\n }\n },\n \"system\": {\n \"title\": \"系统设置\",\n \"name\": \"系统名称\",\n \"name_placeholder\": \"请输入系统名称\",\n \"logo\": \"Logo 图片地址\",\n \"logo_placeholder\": \"在此输入 Logo 图片地址\",\n \"theme\": {\n \"title\": \"主题名称\",\n \"link\": \"当前可用主题\",\n \"placeholder\": \"请输入主题名称\"\n },\n \"buttons\": {\n \"save_name\": \"设置系统名称\",\n \"save_logo\": \"设置 Logo\",\n \"save_theme\": \"设置主题（重启生效）\"\n }\n },\n \"content\": {\n \"title\": \"内容设置\",\n \"homepage\": {\n \"title\": \"首页内容\",\n \"placeholder\": \"在此输入首页内容，支持 Markdown & HTML 代码，设置后首页的状态信息将不再显示。如果输入的是一个链接，则会使用该链接作为 iframe 的 src 属性，这允许你设置任意网页作为首页。\"\n },\n \"about\": {\n \"title\": \"关于\",\n \"placeholder\": \"在此输入新的关于内容，支持 Markdown & HTML 代码。如果输入的是一个链接，则会使用该链接作为 iframe 的 src 属性，这允许你设置任意网页作为关于页面。\"\n },\n \"footer\": {\n \"title\": \"页脚\",\n \"placeholder\": \"在此输入新的页脚，留空则使用默认页脚，支持 HTML 代码\"\n },\n \"buttons\": {\n \"save_homepage\": \"保存首页内容\",\n \"save_about\": \"保存关于\",\n \"save_footer\": \"设置页脚\"\n }\n },\n \"copyright\": {\n \"notice\": \"移除 One API 的版权标识必须首先获得授权，项目维护需要花费大量精力，如果本项目对你有意义，请主动支持本项目。\"\n }\n }\n },\n \"about\": {\n \"title\": \"关于\",\n \"description\": \"One API 是一个开源的接口管理和代理平台。\",\n \"repository\": \"项目地址：\",\n \"loading_failed\": \"加载失败\"\n },\n \"footer\": {\n \"built_by\": \"由\",\n \"built_by_name\": \"JustSong\",\n \"license\": \"构建，源代码遵循\",\n \"mit\": \"MIT 协议\"\n },\n \"home\": {\n \"welcome\": {\n \"title\": \"欢迎使用 One API\",\n \"description\": \"One API 是一个 LLM API 接口管理和分发系统，可以帮助您更好地管理和使用各大厂商的 LLM API。\",\n \"login_notice\": \"如需使用，请先登录或注册。\"\n },\n \"system_status\": {\n \"title\": \"系统状况\",\n \"info\": {\n \"title\": \"系统信息\",\n \"name\": \"名称：\",\n \"version\": \"版本：\",\n \"source\": \"源码：\",\n \"source_link\": \"GitHub 仓库\",\n \"start_time\": \"启动时间：\"\n },\n \"config\": {\n \"title\": \"系统配置\",\n \"email_verify\": \"邮箱验证：\",\n \"github_oauth\": \"GitHub 身份验证：\",\n \"wechat_login\": \"微信身份验证：\",\n \"turnstile\": \"Turnstile 校验：\",\n \"enabled\": \"已启用\",\n \"disabled\": \"未启用\"\n }\n },\n \"loading_failed\": \"加载首页内容失败...\"\n },\n \"auth\": {\n \"login\": {\n \"title\": \"用户登录\",\n \"username\": \"用户名 / 邮箱地址\",\n \"password\": \"密码\",\n \"button\": \"登录\",\n \"forgot_password\": \"忘记密码？\",\n \"reset_password\": \"点击重置\",\n \"no_account\": \"没有账户？\",\n \"register\": \"点击注册\",\n \"other_methods\": \"使用其他方式登录\",\n \"wechat\": {\n \"scan_tip\": \"微信扫码关注公众号，输入「验证码」获取验证码（三分钟内有效）\",\n \"code_placeholder\": \"验证码\"\n }\n },\n \"register\": {\n \"title\": \"新用户注册\",\n \"username\": \"输入用户名，最长 12 位\",\n \"password\": \"输入密码，最短 8 位，最长 20 位\",\n \"confirm_password\": \"再次输入密码\",\n \"email\": \"输入邮箱地址\",\n \"verification_code\": \"输入验证码\",\n \"get_code\": \"获取验证码\",\n \"get_code_retry\": \"重试 ({{countdown}})\",\n \"button\": \"注册\",\n \"has_account\": \"已有账户？\",\n \"login\": \"点击登录\"\n },\n \"reset\": {\n \"title\": \"密码重置\",\n \"email\": \"邮箱地址\",\n \"button\": \"提交\",\n \"notice\": \"系统将向您的邮箱发送一封包含重置链接的邮件，请注意查收。\",\n \"confirm\": {\n \"title\": \"密码重置确认\",\n \"new_password\": \"新密码\",\n \"button\": \"提交\",\n \"button_disabled\": \"密码重置完成\",\n \"notice\": \"新密码已生成，请点击密码框或上方按钮复制。请及时登录并修改密码！\"\n }\n }\n },\n \"messages\": {\n \"success\": {\n \"login\": \"登录成功！\",\n \"register\": \"注册成功！\",\n \"verification_code\": \"验证码发送成功，请检查你的邮箱！\",\n \"password_reset\": \"重置邮件发送成功，请检查邮箱！\"\n },\n \"error\": {\n \"login_expired\": \"未登录或登录已过期，请重新登录！\",\n \"password_length\": \"密码长度不得小于 8 位！\",\n \"password_mismatch\": \"两次输入的密码不一致\",\n \"turnstile_wait\": \"请稍后几秒重试，Turnstile 正在检查用户环境！\",\n \"root_password\": \"请立刻修改默认密码！\"\n },\n \"notice\": {\n \"password_copied\": \"新密码已复制到剪贴板：{{password}}\"\n }\n }\n}\n"
},
{
"path": "web/default/src/pages/About/index.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { useTranslation } from 'react-i18next';\nimport { Card } from 'semantic-ui-react';\nimport { API, showError } from '../../helpers';\nimport { marked } from 'marked';\n\nconst About = () => {\n const { t } = useTranslation();\n const [about, setAbout] = useState('');\n const [aboutLoaded, setAboutLoaded] = useState(false);\n\n const displayAbout = async () => {\n setAbout(localStorage.getItem('about') || '');\n const res = await API.get('/api/about');\n const { success, message, data } = res.data;\n if (success) {\n let aboutContent = data;\n if (!data.startsWith('https://')) {\n aboutContent = marked.parse(data);\n }\n setAbout(aboutContent);\n localStorage.setItem('about', aboutContent);\n } else {\n showError(message);\n setAbout(t('about.loading_failed'));\n }\n setAboutLoaded(true);\n };\n\n useEffect(() => {\n displayAbout().then();\n }, []);\n\n return (\n <>\n {aboutLoaded && about === '' ? (\n \n ) : (\n <>\n {about.startsWith('https://') ? (\n \n
\n \n \n {isEdit\n ? t('channel.edit.title_edit')\n : t('channel.edit.title_create')}\n \n \n \n \n
\n \n \n {t('channel.title')} \n \n \n
\n );\n};\n\nexport default Channel;\n"
},
{
"path": "web/default/src/pages/Chat/index.js",
"content": "import React from 'react';\n\nconst Chat = () => {\n const chatLink = localStorage.getItem('chat_link');\n\n return (\n \n {/* 三个并排的折线图 */}\n
\n \n \n \n \n {t('dashboard.charts.requests.title')}\n {/* {summaryData.todayRequests} */}\n \n \n \n \n [\n value,\n t('dashboard.charts.requests.tooltip'),\n ]}\n labelFormatter={(label) =>\n `${t(\n 'dashboard.statistics.tooltip.date'\n )}: ${formatDate(label)}`\n }\n />\n \n \n
\n \n \n \n\n \n \n \n \n {t('dashboard.charts.quota.title')}\n {/* \n ${summaryData.todayQuota.toFixed(3)}\n */}\n \n \n \n \n [\n value.toFixed(6),\n t('dashboard.charts.quota.tooltip'),\n ]}\n labelFormatter={(label) =>\n `${t(\n 'dashboard.statistics.tooltip.date'\n )}: ${formatDate(label)}`\n }\n />\n \n \n
\n \n \n \n\n \n \n \n \n {t('dashboard.charts.tokens.title')}\n {/* {summaryData.todayTokens} */}\n \n \n \n \n [\n value,\n t('dashboard.charts.tokens.tooltip'),\n ]}\n labelFormatter={(label) =>\n `${t(\n 'dashboard.statistics.tooltip.date'\n )}: ${formatDate(label)}`\n }\n />\n \n \n
\n \n \n \n \n\n {/* 模型使用统计 */}\n
\n \n {t('dashboard.statistics.title')} \n \n \n \n \n `${t('dashboard.statistics.tooltip.date')}: ${formatDate(\n label\n )}`\n }\n />\n \n \n
\n \n \n
\n
\n \n \n {t('home.welcome.title')}\n \n \n {t('home.welcome.description')}
\n {!userState.user && {t('home.welcome.login_notice')}
}\n \n \n \n
\n \n \n {t('home.system_status.title')} \n \n \n \n \n \n \n \n {t('home.system_status.info.title')}\n \n \n \n \n \n {t('home.system_status.info.name')}\n \n {statusState?.status?.system_name} \n \n \n \n {t('home.system_status.info.version')}\n \n \n {statusState?.status?.version || 'unknown'}\n \n \n \n \n {t('home.system_status.info.source')}\n \n \n {t('home.system_status.info.source_link')}\n \n \n \n \n {t('home.system_status.info.start_time')}\n \n {getStartTimeString()} \n \n \n \n \n \n\n
\n \n \n \n \n {t('home.system_status.config.title')}\n \n \n \n \n \n {t('home.system_status.config.email_verify')}\n \n \n {statusState?.status?.email_verification\n ? t('home.system_status.config.enabled')\n : t('home.system_status.config.disabled')}\n \n \n \n \n {t('home.system_status.config.github_oauth')}\n \n \n {statusState?.status?.github_oauth\n ? t('home.system_status.config.enabled')\n : t('home.system_status.config.disabled')}\n \n \n \n \n {t('home.system_status.config.wechat_login')}\n \n \n {statusState?.status?.wechat_login\n ? t('home.system_status.config.enabled')\n : t('home.system_status.config.disabled')}\n \n \n \n \n {t('home.system_status.config.turnstile')}\n \n \n {statusState?.status?.turnstile_check\n ? t('home.system_status.config.enabled')\n : t('home.system_status.config.disabled')}\n \n \n \n \n \n \n \n \n \n
\n )}\n \n )}\n \n );\n};\n\nexport default Home;\n"
},
{
"path": "web/default/src/pages/Log/index.js",
"content": "import React from 'react';\nimport { Card } from 'semantic-ui-react';\nimport { useTranslation } from 'react-i18next';\nimport LogsTable from '../../components/LogsTable';\n\nconst Log = () => {\n const { t } = useTranslation();\n \n return (\n \n \n \n {t('log.title')} \n \n \n
\n );\n};\n\nexport default Log;\n"
},
{
"path": "web/default/src/pages/NotFound/index.js",
"content": "import React from 'react';\nimport { Message } from 'semantic-ui-react';\n\nconst NotFound = () => (\n <>\n \n 页面不存在 \n 请检查你的浏览器地址是否正确
\n \n \n);\n\nexport default NotFound;\n"
},
{
"path": "web/default/src/pages/Redemption/EditRedemption.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { useTranslation } from 'react-i18next';\nimport { Button, Form, Card } from 'semantic-ui-react';\nimport { useParams, useNavigate } from 'react-router-dom';\nimport { API, downloadTextAsFile, showError, showSuccess } from '../../helpers';\nimport { renderQuota, renderQuotaWithPrompt } from '../../helpers/render';\n\nconst EditRedemption = () => {\n const { t } = useTranslation();\n const params = useParams();\n const navigate = useNavigate();\n const redemptionId = params.id;\n const isEdit = redemptionId !== undefined;\n const [loading, setLoading] = useState(isEdit);\n const originInputs = {\n name: '',\n quota: 100000,\n count: 1,\n };\n const [inputs, setInputs] = useState(originInputs);\n const { name, quota, count } = inputs;\n\n const handleCancel = () => {\n navigate('/redemption');\n };\n\n const handleInputChange = (e, { name, value }) => {\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n };\n\n const loadRedemption = async () => {\n let res = await API.get(`/api/redemption/${redemptionId}`);\n const { success, message, data } = res.data;\n if (success) {\n setInputs(data);\n } else {\n showError(message);\n }\n setLoading(false);\n };\n useEffect(() => {\n if (isEdit) {\n loadRedemption().then();\n }\n }, []);\n\n const submit = async () => {\n if (!isEdit && inputs.name === '') return;\n let localInputs = inputs;\n localInputs.count = parseInt(localInputs.count);\n localInputs.quota = parseInt(localInputs.quota);\n let res;\n if (isEdit) {\n res = await API.put(`/api/redemption/`, {\n ...localInputs,\n id: parseInt(redemptionId),\n });\n } else {\n res = await API.post(`/api/redemption/`, {\n ...localInputs,\n });\n }\n const { success, message, data } = res.data;\n if (success) {\n if (isEdit) {\n showSuccess(t('redemption.messages.update_success'));\n } else {\n showSuccess(t('redemption.messages.create_success'));\n setInputs(originInputs);\n }\n } else {\n showError(message);\n }\n if (!isEdit && data) {\n let text = '';\n for (let i = 0; i < data.length; i++) {\n text += data[i] + '\\n';\n }\n downloadTextAsFile(text, `${inputs.name}.txt`);\n }\n };\n\n return (\n \n \n \n \n {isEdit ? t('redemption.edit.title_edit') : t('redemption.edit.title_create')}\n \n \n \n \n
\n );\n};\n\nexport default EditRedemption;\n"
},
{
"path": "web/default/src/pages/Redemption/index.js",
"content": "import React from 'react';\nimport { Card } from 'semantic-ui-react';\nimport { useTranslation } from 'react-i18next';\nimport RedemptionsTable from '../../components/RedemptionsTable';\n\nconst Redemption = () => {\n const { t } = useTranslation();\n \n return (\n \n \n \n {t('redemption.title')} \n \n \n
\n );\n};\n\nexport default Redemption;\n"
},
{
"path": "web/default/src/pages/Setting/index.js",
"content": "import React from 'react';\nimport { useTranslation } from 'react-i18next';\nimport { Card, Tab } from 'semantic-ui-react';\nimport SystemSetting from '../../components/SystemSetting';\nimport { isRoot } from '../../helpers';\nimport OtherSetting from '../../components/OtherSetting';\nimport PersonalSetting from '../../components/PersonalSetting';\nimport OperationSetting from '../../components/OperationSetting';\n\nconst Setting = () => {\n const { t } = useTranslation();\n\n let panes = [\n {\n menuItem: t('setting.tabs.personal'),\n render: () => (\n \n \n ),\n },\n ];\n\n if (isRoot()) {\n panes.push({\n menuItem: t('setting.tabs.operation'),\n render: () => (\n \n \n ),\n });\n panes.push({\n menuItem: t('setting.tabs.system'),\n render: () => (\n \n \n ),\n });\n panes.push({\n menuItem: t('setting.tabs.other'),\n render: () => (\n \n \n ),\n });\n }\n\n return (\n \n \n \n {t('setting.title')} \n \n \n
\n );\n};\n\nexport default Setting;\n"
},
{
"path": "web/default/src/pages/Token/EditToken.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { useTranslation } from 'react-i18next';\nimport {\n Button,\n Form,\n Header,\n Message,\n Segment,\n Card,\n} from 'semantic-ui-react';\nimport { useNavigate, useParams } from 'react-router-dom';\nimport {\n API,\n copy,\n showError,\n showSuccess,\n timestamp2string,\n} from '../../helpers';\nimport { renderQuotaWithPrompt } from '../../helpers/render';\n\nconst EditToken = () => {\n const { t } = useTranslation();\n const params = useParams();\n const tokenId = params.id;\n const isEdit = tokenId !== undefined;\n const [loading, setLoading] = useState(isEdit);\n const [modelOptions, setModelOptions] = useState([]);\n const originInputs = {\n name: '',\n remain_quota: isEdit ? 0 : 500000,\n expired_time: -1,\n unlimited_quota: false,\n models: [],\n subnet: '',\n };\n const [inputs, setInputs] = useState(originInputs);\n const { name, remain_quota, expired_time, unlimited_quota } = inputs;\n const navigate = useNavigate();\n const handleInputChange = (e, { name, value }) => {\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n };\n const handleCancel = () => {\n navigate('/token');\n };\n const setExpiredTime = (month, day, hour, minute) => {\n let now = new Date();\n let timestamp = now.getTime() / 1000;\n let seconds = month * 30 * 24 * 60 * 60;\n seconds += day * 24 * 60 * 60;\n seconds += hour * 60 * 60;\n seconds += minute * 60;\n if (seconds !== 0) {\n timestamp += seconds;\n setInputs({ ...inputs, expired_time: timestamp2string(timestamp) });\n } else {\n setInputs({ ...inputs, expired_time: -1 });\n }\n };\n\n const setUnlimitedQuota = () => {\n setInputs({ ...inputs, unlimited_quota: !unlimited_quota });\n };\n\n const loadToken = async () => {\n try {\n let res = await API.get(`/api/token/${tokenId}`);\n const { success, message, data } = res.data || {};\n if (success && data) {\n if (data.expired_time !== -1) {\n data.expired_time = timestamp2string(data.expired_time);\n }\n if (data.models === '') {\n data.models = [];\n } else {\n data.models = data.models.split(',');\n }\n setInputs(data);\n } else {\n showError(message || 'Failed to load token');\n }\n } catch (error) {\n showError(error.message || 'Network error');\n }\n setLoading(false);\n };\n\n const loadAvailableModels = async () => {\n try {\n let res = await API.get(`/api/user/available_models`);\n const { success, message, data } = res.data || {};\n if (success && data) {\n let options = data.map((model) => {\n return {\n key: model,\n text: model,\n value: model,\n };\n });\n setModelOptions(options);\n } else {\n showError(message || 'Failed to load models');\n }\n } catch (error) {\n showError(error.message || 'Network error');\n }\n };\n\n useEffect(() => {\n if (isEdit) {\n loadToken().catch((error) => {\n showError(error.message || 'Failed to load token');\n setLoading(false);\n });\n }\n loadAvailableModels().catch((error) => {\n showError(error.message || 'Failed to load models');\n });\n }, []);\n\n const submit = async () => {\n if (!isEdit && inputs.name === '') return;\n let localInputs = inputs;\n localInputs.remain_quota = parseInt(localInputs.remain_quota);\n if (localInputs.expired_time !== -1) {\n let time = Date.parse(localInputs.expired_time);\n if (isNaN(time)) {\n showError(t('token.edit.messages.expire_time_invalid'));\n return;\n }\n localInputs.expired_time = Math.ceil(time / 1000);\n }\n localInputs.models = localInputs.models.join(',');\n let res;\n if (isEdit) {\n res = await API.put(`/api/token/`, {\n ...localInputs,\n id: parseInt(tokenId),\n });\n } else {\n res = await API.post(`/api/token/`, localInputs);\n }\n const { success, message } = res.data;\n if (success) {\n if (isEdit) {\n showSuccess(t('token.edit.messages.update_success'));\n } else {\n showSuccess(t('token.edit.messages.create_success'));\n setInputs(originInputs);\n }\n } else {\n showError(message);\n }\n };\n\n return (\n \n
\n \n \n {isEdit ? t('token.edit.title_edit') : t('token.edit.title_create')}\n \n \n \n \n
\n \n \n {t('token.title')} \n \n \n
\n );\n};\n\nexport default Token;\n"
},
{
"path": "web/default/src/pages/TopUp/index.js",
"content": "import React, { useEffect, useState } from 'react';\nimport {\n Button,\n Form,\n Grid,\n Header,\n Card,\n Statistic,\n Divider,\n} from 'semantic-ui-react';\nimport { API, showError, showInfo, showSuccess } from '../../helpers';\nimport { renderQuota } from '../../helpers/render';\nimport { useTranslation } from 'react-i18next';\n\nconst TopUp = () => {\n const { t } = useTranslation();\n const [redemptionCode, setRedemptionCode] = useState('');\n const [topUpLink, setTopUpLink] = useState('');\n const [userQuota, setUserQuota] = useState(0);\n const [isSubmitting, setIsSubmitting] = useState(false);\n const [user, setUser] = useState({});\n\n const topUp = async () => {\n if (redemptionCode === '') {\n showInfo(t('topup.redeem_code.empty_code'));\n return;\n }\n setIsSubmitting(true);\n try {\n const res = await API.post('/api/user/topup', {\n key: redemptionCode,\n });\n const { success, message, data } = res.data;\n if (success) {\n showSuccess(t('topup.redeem_code.success'));\n setUserQuota((quota) => {\n return quota + data;\n });\n setRedemptionCode('');\n } else {\n showError(message);\n }\n } catch (err) {\n showError(t('topup.redeem_code.request_failed'));\n } finally {\n setIsSubmitting(false);\n }\n };\n\n const openTopUpLink = () => {\n if (!topUpLink) {\n showError(t('topup.redeem_code.no_link'));\n return;\n }\n let url = new URL(topUpLink);\n let username = user.username;\n let user_id = user.id;\n url.searchParams.append('username', username);\n url.searchParams.append('user_id', user_id);\n url.searchParams.append('transaction_id', crypto.randomUUID());\n window.open(url.toString(), '_blank');\n };\n\n const getUserQuota = async () => {\n let res = await API.get(`/api/user/self`);\n const { success, message, data } = res.data;\n if (success) {\n setUserQuota(data.quota);\n setUser(data);\n } else {\n showError(message);\n }\n };\n\n useEffect(() => {\n let status = localStorage.getItem('status');\n if (status) {\n status = JSON.parse(status);\n if (status.top_up_link) {\n setTopUpLink(status.top_up_link);\n }\n }\n getUserQuota().then();\n }, []);\n\n return (\n \n
\n \n \n \n \n\n \n \n \n \n \n \n \n \n \n \n \n \n \n {renderQuota(userQuota, t)}\n \n \n {t('topup.get_code.current_quota')}\n \n \n
\n\n \n \n {t('topup.get_code.button')}\n \n \n \n \n \n \n \n \n \n \n {\n setRedemptionCode(e.target.value);\n }}\n onPaste={(e) => {\n e.preventDefault();\n const pastedText = e.clipboardData.getData('text');\n setRedemptionCode(pastedText.trim());\n }}\n action={\n {\n try {\n const text =\n await navigator.clipboard.readText();\n setRedemptionCode(text.trim());\n } catch (err) {\n showError(t('topup.redeem_code.paste_error'));\n }\n }}\n />\n }\n />\n\n \n \n {isSubmitting\n ? t('topup.redeem_code.submitting')\n : t('topup.redeem_code.submit')}\n \n
\n \n \n \n \n \n \n \n \n \n );\n};\n\nexport default TopUp;\n"
},
{
"path": "web/default/src/pages/User/AddUser.js",
"content": "import React, { useState } from 'react';\nimport { useTranslation } from 'react-i18next';\nimport { Button, Form, Card } from 'semantic-ui-react';\nimport { API, showError, showSuccess } from '../../helpers';\n\nconst AddUser = () => {\n const { t } = useTranslation();\n const originInputs = {\n username: '',\n display_name: '',\n password: '',\n };\n const [inputs, setInputs] = useState(originInputs);\n const { username, display_name, password } = inputs;\n\n const handleInputChange = (e, { name, value }) => {\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n };\n\n const submit = async () => {\n if (inputs.username === '' || inputs.password === '') return;\n const res = await API.post(`/api/user/`, inputs);\n const { success, message } = res.data;\n if (success) {\n showSuccess(t('user.messages.create_success'));\n setInputs(originInputs);\n } else {\n showError(message);\n }\n };\n\n return (\n \n \n \n {t('user.add.title')} \n \n \n \n
\n );\n};\n\nexport default AddUser;\n"
},
{
"path": "web/default/src/pages/User/EditUser.js",
"content": "import React, { useEffect, useState } from 'react';\nimport { useTranslation } from 'react-i18next';\nimport { Button, Form, Card } from 'semantic-ui-react';\nimport { useParams, useNavigate } from 'react-router-dom';\nimport { API, showError, showSuccess } from '../../helpers';\nimport { renderQuota, renderQuotaWithPrompt } from '../../helpers/render';\n\nconst EditUser = () => {\n const { t } = useTranslation();\n const params = useParams();\n const userId = params.id;\n const [loading, setLoading] = useState(true);\n const [inputs, setInputs] = useState({\n username: '',\n display_name: '',\n password: '',\n github_id: '',\n wechat_id: '',\n email: '',\n quota: 0,\n group: 'default',\n });\n const [groupOptions, setGroupOptions] = useState([]);\n const {\n username,\n display_name,\n password,\n github_id,\n wechat_id,\n email,\n quota,\n group,\n } = inputs;\n const handleInputChange = (e, { name, value }) => {\n setInputs((inputs) => ({ ...inputs, [name]: value }));\n };\n const fetchGroups = async () => {\n try {\n let res = await API.get(`/api/group/`);\n setGroupOptions(\n res.data.data.map((group) => ({\n key: group,\n text: group,\n value: group,\n }))\n );\n } catch (error) {\n showError(error.message);\n }\n };\n const navigate = useNavigate();\n const handleCancel = () => {\n navigate('/setting');\n };\n const loadUser = async () => {\n let res = undefined;\n if (userId) {\n res = await API.get(`/api/user/${userId}`);\n } else {\n res = await API.get(`/api/user/self`);\n }\n const { success, message, data } = res.data;\n if (success) {\n data.password = '';\n setInputs(data);\n } else {\n showError(message);\n }\n setLoading(false);\n };\n useEffect(() => {\n loadUser().then();\n if (userId) {\n fetchGroups().then();\n }\n }, []);\n\n const submit = async () => {\n let res = undefined;\n if (userId) {\n let data = { ...inputs, id: parseInt(userId) };\n if (typeof data.quota === 'string') {\n data.quota = parseInt(data.quota);\n }\n res = await API.put(`/api/user/`, data);\n } else {\n res = await API.put(`/api/user/self`, inputs);\n }\n const { success, message } = res.data;\n if (success) {\n showSuccess(t('user.messages.update_success'));\n } else {\n showError(message);\n }\n };\n\n return (\n \n \n \n {t('user.edit.title')} \n \n \n \n
\n );\n};\n\nexport default EditUser;\n"
},
{
"path": "web/default/src/pages/User/index.js",
"content": "import React from 'react';\nimport { useTranslation } from 'react-i18next';\nimport { Card } from 'semantic-ui-react';\nimport UsersTable from '../../components/UsersTable';\n\nconst User = () => {\n const { t } = useTranslation();\n\n return (\n \n \n \n {t('user.title')} \n \n \n
\n );\n};\n\nexport default User;\n"
},
{
"path": "web/default/vercel.json",
"content": "{\n \"github\": {\n \"silent\": true\n }\n}\n"
}
]
![\"one-api](\"https://raw.githubusercontent.com/songquanpeng/one-api/main/web/default/public/logo.png\"){kind=logo}
\n![\"license\"](\"https://img.shields.io/github/license/songquanpeng/one-api?color=brightgreen\")
![\"release\"](\"https://img.shields.io/github/v/release/songquanpeng/one-api?color=brightgreen&include_prereleases\")
![\"docker](\"https://img.shields.io/docker/pulls/justsong/one-api?color=brightgreen\")
![\"release\"](\"https://img.shields.io/github/downloads/songquanpeng/one-api/total?color=brightgreen&include_prereleases\")