mirror of
				https://github.com/songquanpeng/one-api.git
				synced 2025-10-31 13:53:41 +08:00 
			
		
		
		
	Compare commits
	
		
			25 Commits
		
	
	
		
			v0.2.3-alp
			...
			v0.2.6-alp
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
|  | 443a22b75d | ||
|  | b44f0519a0 | ||
|  | 4a0e81fe83 | ||
|  | 976c29ea9f | ||
|  | 926951ee03 | ||
|  | 2cdc718fde | ||
|  | 57cb150177 | ||
|  | 6167e20b34 | ||
|  | 8835d8302e | ||
|  | 224bebe67a | ||
|  | cf6883778e | ||
|  | 246b981e23 | ||
|  | 2edd52e851 | ||
|  | e123c66bc7 | ||
|  | 9edc82bde0 | ||
|  | d84c2f5c70 | ||
|  | 46e77389a4 | ||
|  | f5f4e6fbc6 | ||
|  | dc4a6cb711 | ||
|  | 5798fdac50 | ||
|  | 3710688efd | ||
|  | 83e86b9f8a | ||
|  | 74c1ba7cbc | ||
|  | 73aa53f536 | ||
|  | da9ccb528d | 
							
								
								
									
										25
									
								
								README.md
									
									
									
									
									
								
							
							
						
						
									
										25
									
								
								README.md
									
									
									
									
									
								
							| @@ -42,28 +42,29 @@ _✨ All in one 的 OpenAI 接口,整合各种 API 访问方式,开箱即用 | ||||
|  | ||||
| ## 功能 | ||||
| 1. 支持多种 API 访问渠道,欢迎 PR 或提 issue 添加更多渠道: | ||||
|    + [x] One API 服务端中继 | ||||
|    + [x] OpenAI 官方通道 | ||||
|    + [x] **Azure OpenAI API** | ||||
|    + [x] [API2D](https://api2d.com/r/197971) | ||||
|    + [ ] Azure OpenAI API | ||||
|    + [x] [CloseAI](https://console.openai-asia.com) | ||||
|    + [x] [OpenAI-SB](https://openai-sb.com) | ||||
|    + [x] [OpenAI Max](https://openaimax.com) | ||||
|    + [x] [OhMyGPT](https://www.ohmygpt.com) | ||||
|    + [x] 自定义渠道:例如使用自行搭建的 OpenAI 代理 | ||||
| 2. 支持通过负载均衡的方式访问多个渠道。 | ||||
| 3. 支持单个访问渠道设置多个 API Key,利用起来你的多个 API Key。 | ||||
| 4. 支持 HTTP SSE,可以通过流式传输实现打字机效果。 | ||||
| 5. 支持设置令牌的过期时间和使用次数。 | ||||
| 6. 支持批量生成和导出兑换码,可使用兑换码为令牌进行充值。 | ||||
| 7. 支持为新用户设置初始配额。 | ||||
| 8. 支持发布公告,在线修改关于页面,设置充值链接,自定义页脚。 | ||||
| 2. 支持通过**负载均衡**的方式访问多个渠道。 | ||||
| 3. 支持 **stream 模式**,可以通过流式传输实现打字机效果。 | ||||
| 4. 支持**令牌管理**,设置令牌的过期时间和使用次数。 | ||||
| 5. 支持**兑换码管理**,支持批量生成和导出兑换码,可使用兑换码为令牌进行充值。 | ||||
| 6. 支持**通道管理**,批量创建通道。 | ||||
| 7. 支持发布公告,设置充值链接,设置新用户初始额度。 | ||||
| 8. 支持丰富的**自定义**设置, | ||||
|    1. 支持自定义系统名称,logo 以及页脚。 | ||||
|    2. 支持自定义首页和关于页面,可以选择使用 HTML & Markdown 代码进行自定义,或者使用一个单独的网页通过 iframe 嵌入。 | ||||
| 9. 支持通过系统访问令牌访问管理 API。 | ||||
| 10. 多种用户登录注册方式: | ||||
| 10. 支持用户管理,支持**多种用户登录注册方式**: | ||||
|     + 邮箱登录注册以及通过邮箱进行密码重置。 | ||||
|     + [GitHub 开放授权](https://github.com/settings/applications/new)。 | ||||
|     + 微信公众号授权(需要额外部署 [WeChat Server](https://github.com/songquanpeng/wechat-server))。 | ||||
| 11. 支持用户管理。 | ||||
| 12. 未来其他大模型开放 API 后,将第一时间支持,并将其封装成同样的 API 访问方式。 | ||||
| 11. 未来其他大模型开放 API 后,将第一时间支持,并将其封装成同样的 API 访问方式。 | ||||
|  | ||||
| ## 部署 | ||||
| ### 基于 Docker 进行部署 | ||||
|   | ||||
| @@ -11,6 +11,7 @@ var Version = "v0.0.0"            // this hard coding will be replaced automatic | ||||
| var SystemName = "One API" | ||||
| var ServerAddress = "http://localhost:3000" | ||||
| var Footer = "" | ||||
| var Logo = "" | ||||
| var TopUpLink = "" | ||||
|  | ||||
| var UsingSQLite = false | ||||
| @@ -36,6 +37,7 @@ var RegisterEnabled = true | ||||
| var SMTPServer = "" | ||||
| var SMTPPort = 587 | ||||
| var SMTPAccount = "" | ||||
| var SMTPFrom = "" | ||||
| var SMTPToken = "" | ||||
|  | ||||
| var GitHubClientId = "" | ||||
|   | ||||
| @@ -1,14 +1,67 @@ | ||||
| package common | ||||
|  | ||||
| import "gopkg.in/gomail.v2" | ||||
| import ( | ||||
| 	"crypto/tls" | ||||
| 	"encoding/base64" | ||||
| 	"fmt" | ||||
| 	"net/smtp" | ||||
| 	"strings" | ||||
| ) | ||||
|  | ||||
| func SendEmail(subject string, receiver string, content string) error { | ||||
| 	m := gomail.NewMessage() | ||||
| 	m.SetHeader("From", SMTPAccount) | ||||
| 	m.SetHeader("To", receiver) | ||||
| 	m.SetHeader("Subject", subject) | ||||
| 	m.SetBody("text/html", content) | ||||
| 	d := gomail.NewDialer(SMTPServer, SMTPPort, SMTPAccount, SMTPToken) | ||||
| 	err := d.DialAndSend(m) | ||||
| 	if SMTPFrom == "" { // for compatibility | ||||
| 		SMTPFrom = SMTPAccount | ||||
| 	} | ||||
| 	encodedSubject := fmt.Sprintf("=?UTF-8?B?%s?=", base64.StdEncoding.EncodeToString([]byte(subject))) | ||||
| 	mail := []byte(fmt.Sprintf("To: %s\r\n"+ | ||||
| 		"From: %s<%s>\r\n"+ | ||||
| 		"Subject: %s\r\n"+ | ||||
| 		"Content-Type: text/html; charset=UTF-8\r\n\r\n%s\r\n", | ||||
| 		receiver, SystemName, SMTPFrom, encodedSubject, content)) | ||||
| 	auth := smtp.PlainAuth("", SMTPAccount, SMTPToken, SMTPServer) | ||||
| 	addr := fmt.Sprintf("%s:%d", SMTPServer, SMTPPort) | ||||
| 	to := strings.Split(receiver, ";") | ||||
| 	var err error | ||||
| 	if SMTPPort == 465 { | ||||
| 		tlsConfig := &tls.Config{ | ||||
| 			InsecureSkipVerify: true, | ||||
| 			ServerName:         SMTPServer, | ||||
| 		} | ||||
| 		conn, err := tls.Dial("tcp", fmt.Sprintf("%s:%d", SMTPServer, SMTPPort), tlsConfig) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		client, err := smtp.NewClient(conn, SMTPServer) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		defer client.Close() | ||||
| 		if err = client.Auth(auth); err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		if err = client.Mail(SMTPFrom); err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		receiverEmails := strings.Split(receiver, ";") | ||||
| 		for _, receiver := range receiverEmails { | ||||
| 			if err = client.Rcpt(receiver); err != nil { | ||||
| 				return err | ||||
| 			} | ||||
| 		} | ||||
| 		w, err := client.Data() | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		_, err = w.Write(mail) | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 		err = w.Close() | ||||
| 		if err != nil { | ||||
| 			return err | ||||
| 		} | ||||
| 	} else { | ||||
| 		err = smtp.SendMail(addr, auth, SMTPAccount, to, mail) | ||||
| 	} | ||||
| 	return err | ||||
| } | ||||
|   | ||||
| @@ -1,11 +1,17 @@ | ||||
| package controller | ||||
|  | ||||
| import ( | ||||
| 	"bytes" | ||||
| 	"encoding/json" | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 	"github.com/gin-gonic/gin" | ||||
| 	"net/http" | ||||
| 	"one-api/common" | ||||
| 	"one-api/model" | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
| 	"time" | ||||
| ) | ||||
|  | ||||
| func GetAllChannels(c *gin.Context) { | ||||
| @@ -84,7 +90,17 @@ func AddChannel(c *gin.Context) { | ||||
| 	} | ||||
| 	channel.CreatedTime = common.GetTimestamp() | ||||
| 	channel.AccessedTime = common.GetTimestamp() | ||||
| 	err = channel.Insert() | ||||
| 	keys := strings.Split(channel.Key, "\n") | ||||
| 	channels := make([]model.Channel, 0) | ||||
| 	for _, key := range keys { | ||||
| 		if key == "" { | ||||
| 			continue | ||||
| 		} | ||||
| 		localChannel := channel | ||||
| 		localChannel.Key = key | ||||
| 		channels = append(channels, localChannel) | ||||
| 	} | ||||
| 	err = model.BatchInsertChannels(channels) | ||||
| 	if err != nil { | ||||
| 		c.JSON(http.StatusOK, gin.H{ | ||||
| 			"success": false, | ||||
| @@ -142,3 +158,97 @@ func UpdateChannel(c *gin.Context) { | ||||
| 	}) | ||||
| 	return | ||||
| } | ||||
|  | ||||
| func testChannel(channel *model.Channel, request *ChatRequest) error { | ||||
| 	if request.Model == "" { | ||||
| 		request.Model = "gpt-3.5-turbo" | ||||
| 		if channel.Type == common.ChannelTypeAzure { | ||||
| 			request.Model = "gpt-35-turbo" | ||||
| 		} | ||||
| 	} | ||||
| 	requestURL := common.ChannelBaseURLs[channel.Type] | ||||
| 	if channel.Type == common.ChannelTypeAzure { | ||||
| 		requestURL = fmt.Sprintf("%s/openai/deployments/%s/chat/completions?api-version=2023-03-15-preview", channel.BaseURL, request.Model) | ||||
| 	} else { | ||||
| 		if channel.Type == common.ChannelTypeCustom { | ||||
| 			requestURL = channel.BaseURL | ||||
| 		} | ||||
| 		requestURL += "/v1/chat/completions" | ||||
| 	} | ||||
|  | ||||
| 	jsonData, err := json.Marshal(request) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	req, err := http.NewRequest("POST", requestURL, bytes.NewBuffer(jsonData)) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	if channel.Type == common.ChannelTypeAzure { | ||||
| 		req.Header.Set("api-key", channel.Key) | ||||
| 	} else { | ||||
| 		req.Header.Set("Authorization", "Bearer "+channel.Key) | ||||
| 	} | ||||
| 	req.Header.Set("Content-Type", "application/json") | ||||
| 	client := &http.Client{} | ||||
| 	resp, err := client.Do(req) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	defer resp.Body.Close() | ||||
| 	var response TextResponse | ||||
| 	err = json.NewDecoder(resp.Body).Decode(&response) | ||||
| 	if err != nil { | ||||
| 		return err | ||||
| 	} | ||||
| 	if response.Error.Type != "" { | ||||
| 		return errors.New(fmt.Sprintf("type %s, code %s, message %s", response.Error.Type, response.Error.Code, response.Error.Message)) | ||||
| 	} | ||||
| 	return nil | ||||
| } | ||||
|  | ||||
| func TestChannel(c *gin.Context) { | ||||
| 	id, err := strconv.Atoi(c.Param("id")) | ||||
| 	if err != nil { | ||||
| 		c.JSON(http.StatusOK, gin.H{ | ||||
| 			"success": false, | ||||
| 			"message": err.Error(), | ||||
| 		}) | ||||
| 		return | ||||
| 	} | ||||
| 	channel, err := model.GetChannelById(id, true) | ||||
| 	if err != nil { | ||||
| 		c.JSON(http.StatusOK, gin.H{ | ||||
| 			"success": false, | ||||
| 			"message": err.Error(), | ||||
| 		}) | ||||
| 		return | ||||
| 	} | ||||
| 	model_ := c.Query("model") | ||||
| 	chatRequest := &ChatRequest{ | ||||
| 		Model: model_, | ||||
| 	} | ||||
| 	testMessage := Message{ | ||||
| 		Role:    "user", | ||||
| 		Content: "echo hi", | ||||
| 	} | ||||
| 	chatRequest.Messages = append(chatRequest.Messages, testMessage) | ||||
| 	tik := time.Now() | ||||
| 	err = testChannel(channel, chatRequest) | ||||
| 	tok := time.Now() | ||||
| 	consumedTime := float64(tok.Sub(tik).Milliseconds()) / 1000.0 | ||||
| 	if err != nil { | ||||
| 		c.JSON(http.StatusOK, gin.H{ | ||||
| 			"success": false, | ||||
| 			"message": err.Error(), | ||||
| 			"time":    consumedTime, | ||||
| 		}) | ||||
| 		return | ||||
| 	} | ||||
| 	c.JSON(http.StatusOK, gin.H{ | ||||
| 		"success": true, | ||||
| 		"message": "", | ||||
| 		"time":    consumedTime, | ||||
| 	}) | ||||
| 	return | ||||
| } | ||||
|   | ||||
| @@ -20,6 +20,7 @@ func GetStatus(c *gin.Context) { | ||||
| 			"github_oauth":       common.GitHubOAuthEnabled, | ||||
| 			"github_client_id":   common.GitHubClientId, | ||||
| 			"system_name":        common.SystemName, | ||||
| 			"logo":               common.Logo, | ||||
| 			"footer_html":        common.Footer, | ||||
| 			"wechat_qrcode":      common.WeChatAccountQRCodeImageURL, | ||||
| 			"wechat_login":       common.WeChatAuthEnabled, | ||||
| @@ -54,6 +55,17 @@ func GetAbout(c *gin.Context) { | ||||
| 	return | ||||
| } | ||||
|  | ||||
| func GetHomePageContent(c *gin.Context) { | ||||
| 	common.OptionMapRWMutex.RLock() | ||||
| 	defer common.OptionMapRWMutex.RUnlock() | ||||
| 	c.JSON(http.StatusOK, gin.H{ | ||||
| 		"success": true, | ||||
| 		"message": "", | ||||
| 		"data":    common.OptionMap["HomePageContent"], | ||||
| 	}) | ||||
| 	return | ||||
| } | ||||
|  | ||||
| func SendEmailVerification(c *gin.Context) { | ||||
| 	email := c.Query("email") | ||||
| 	if err := common.Validate.Var(email, "required,email"); err != nil { | ||||
|   | ||||
| @@ -19,6 +19,11 @@ type Message struct { | ||||
| 	Content string `json:"content"` | ||||
| } | ||||
|  | ||||
| type ChatRequest struct { | ||||
| 	Model    string    `json:"model"` | ||||
| 	Messages []Message `json:"messages"` | ||||
| } | ||||
|  | ||||
| type TextRequest struct { | ||||
| 	Model    string    `json:"model"` | ||||
| 	Messages []Message `json:"messages"` | ||||
| @@ -32,8 +37,16 @@ type Usage struct { | ||||
| 	TotalTokens      int `json:"total_tokens"` | ||||
| } | ||||
|  | ||||
| type OpenAIError struct { | ||||
| 	Message string `json:"message"` | ||||
| 	Type    string `json:"type"` | ||||
| 	Param   string `json:"param"` | ||||
| 	Code    string `json:"code"` | ||||
| } | ||||
|  | ||||
| type TextResponse struct { | ||||
| 	Usage `json:"usage"` | ||||
| 	Error OpenAIError `json:"error"` | ||||
| } | ||||
|  | ||||
| type StreamResponse struct { | ||||
| @@ -94,12 +107,16 @@ func relayHelper(c *gin.Context) error { | ||||
| 	if channelType == common.ChannelTypeAzure { | ||||
| 		// https://learn.microsoft.com/en-us/azure/cognitive-services/openai/chatgpt-quickstart?pivots=rest-api&tabs=command-line#rest-api | ||||
| 		query := c.Request.URL.Query() | ||||
| 		if query.Get("api-version") == "" { | ||||
| 			requestURL = fmt.Sprintf("%s?api-version=2023-03-15-preview", requestURL) | ||||
| 		apiVersion := query.Get("api-version") | ||||
| 		if apiVersion == "" { | ||||
| 			apiVersion = c.GetString("api_version") | ||||
| 		} | ||||
| 		requestURL := strings.Split(requestURL, "?")[0] | ||||
| 		requestURL = fmt.Sprintf("%s?api-version=%s", requestURL, apiVersion) | ||||
| 		baseURL = c.GetString("base_url") | ||||
| 		task := strings.TrimPrefix(requestURL, "/v1/") | ||||
| 		model_ := textRequest.Model | ||||
| 		model_ = strings.Replace(model_, ".", "", -1) | ||||
| 		fullRequestURL = fmt.Sprintf("%s/openai/deployments/%s/%s", baseURL, model_, task) | ||||
| 	} | ||||
| 	req, err := http.NewRequest(c.Request.Method, fullRequestURL, c.Request.Body) | ||||
| @@ -184,7 +201,7 @@ func relayHelper(c *gin.Context) error { | ||||
| 				data := scanner.Text() | ||||
| 				dataChan <- data | ||||
| 				data = data[6:] | ||||
| 				if data != "[DONE]" { | ||||
| 				if !strings.HasPrefix(data, "[DONE]") { | ||||
| 					var streamResponse StreamResponse | ||||
| 					err = json.Unmarshal([]byte(data), &streamResponse) | ||||
| 					if err != nil { | ||||
| @@ -205,6 +222,9 @@ func relayHelper(c *gin.Context) error { | ||||
| 		c.Stream(func(w io.Writer) bool { | ||||
| 			select { | ||||
| 			case data := <-dataChan: | ||||
| 				if strings.HasPrefix(data, "data: [DONE]") { | ||||
| 					data = data[:12] | ||||
| 				} | ||||
| 				c.Render(-1, common.CustomEvent{Data: data}) | ||||
| 				return true | ||||
| 			case <-stopChan: | ||||
|   | ||||
							
								
								
									
										5
									
								
								go.mod
									
									
									
									
									
								
							
							
						
						
									
										5
									
								
								go.mod
									
									
									
									
									
								
							| @@ -12,8 +12,8 @@ require ( | ||||
| 	github.com/go-playground/validator/v10 v10.12.0 | ||||
| 	github.com/go-redis/redis/v8 v8.11.5 | ||||
| 	github.com/google/uuid v1.3.0 | ||||
| 	github.com/pkoukk/tiktoken-go v0.1.1 | ||||
| 	golang.org/x/crypto v0.8.0 | ||||
| 	gopkg.in/gomail.v2 v2.0.0-20160411212932-81ebce5c23df | ||||
| 	gorm.io/driver/mysql v1.4.3 | ||||
| 	gorm.io/driver/sqlite v1.4.3 | ||||
| 	gorm.io/gorm v1.24.0 | ||||
| @@ -45,7 +45,6 @@ require ( | ||||
| 	github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect | ||||
| 	github.com/modern-go/reflect2 v1.0.2 // indirect | ||||
| 	github.com/pelletier/go-toml/v2 v2.0.7 // indirect | ||||
| 	github.com/pkoukk/tiktoken-go v0.1.1 // indirect | ||||
| 	github.com/twitchyliquid64/golang-asm v0.15.1 // indirect | ||||
| 	github.com/ugorji/go/codec v1.2.11 // indirect | ||||
| 	golang.org/x/arch v0.3.0 // indirect | ||||
| @@ -53,7 +52,5 @@ require ( | ||||
| 	golang.org/x/sys v0.7.0 // indirect | ||||
| 	golang.org/x/text v0.9.0 // indirect | ||||
| 	google.golang.org/protobuf v1.30.0 // indirect | ||||
| 	gopkg.in/alexcesaro/quotedprintable.v3 v3.0.0-20150716171945-2caba252f4dc // indirect | ||||
| 	gopkg.in/yaml.v2 v2.4.0 // indirect | ||||
| 	gopkg.in/yaml.v3 v3.0.1 // indirect | ||||
| ) | ||||
|   | ||||
							
								
								
									
										31
									
								
								go.sum
									
									
									
									
									
								
							
							
						
						
									
										31
									
								
								go.sum
									
									
									
									
									
								
							| @@ -28,33 +28,27 @@ github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm | ||||
| github.com/gin-contrib/static v0.0.1 h1:JVxuvHPuUfkoul12N7dtQw7KRn/pSMq7Ue1Va9Swm1U= | ||||
| github.com/gin-contrib/static v0.0.1/go.mod h1:CSxeF+wep05e0kCOsqWdAWbSszmc31zTIbD8TvWl7Hs= | ||||
| github.com/gin-gonic/gin v1.6.3/go.mod h1:75u5sXoLsGZoRN5Sgbi1eraJ4GU3++wFwWzhwvtwp4M= | ||||
| github.com/gin-gonic/gin v1.8.1 h1:4+fr/el88TOO3ewCmQr8cx/CtZ/umlIRIs5M4NTNjf8= | ||||
| github.com/gin-gonic/gin v1.8.1/go.mod h1:ji8BvRH1azfM+SYow9zQ6SZMvR8qOMZHmsCuWR9tTTk= | ||||
| github.com/gin-gonic/gin v1.9.0 h1:OjyFBKICoexlu99ctXNR2gg+c5pKrKMuyjgARg9qeY8= | ||||
| github.com/gin-gonic/gin v1.9.0/go.mod h1:W1Me9+hsUSyj3CePGrd1/QrKJMSJ1Tu/0hFEH89961k= | ||||
| github.com/go-playground/assert/v2 v2.0.1 h1:MsBgLAaY856+nPRTKrp3/OZK38U/wa0CcBYNjji3q3A= | ||||
| github.com/go-playground/assert/v2 v2.0.1/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4= | ||||
| github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s= | ||||
| github.com/go-playground/locales v0.13.0/go.mod h1:taPMhCMXrRLJO55olJkUXHZBHCxTMfnGwq/HNwmWNS8= | ||||
| github.com/go-playground/locales v0.14.0 h1:u50s323jtVGugKlcYeyzC0etD1HifMjqmJqb8WugfUU= | ||||
| github.com/go-playground/locales v0.14.0/go.mod h1:sawfccIbzZTqEDETgFXqTho0QybSa7l++s0DH+LDiLs= | ||||
| github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA= | ||||
| github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY= | ||||
| github.com/go-playground/universal-translator v0.17.0/go.mod h1:UkSxE5sNxxRwHyU+Scu5vgOQjsIJAF8j9muTVoKLVtA= | ||||
| github.com/go-playground/universal-translator v0.18.0 h1:82dyy6p4OuJq4/CByFNOn/jYrnRPArHwAcmLoJZxyho= | ||||
| github.com/go-playground/universal-translator v0.18.0/go.mod h1:UvRDBj+xPUEGrFYl+lu/H90nyDXpg0fqeB/AQUGNTVA= | ||||
| github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY= | ||||
| github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY= | ||||
| github.com/go-playground/validator/v10 v10.2.0/go.mod h1:uOYAAleCW8F/7oMFd6aG0GOhaH6EGOAJShg8Id5JGkI= | ||||
| github.com/go-playground/validator/v10 v10.10.0/go.mod h1:74x4gJWsvQexRdW8Pn3dXSGrTK4nAUsbPlLADvpJkos= | ||||
| github.com/go-playground/validator/v10 v10.11.1 h1:prmOlTVv+YjZjmRmNSF3VmspqJIxJWXmqUsHwfTRRkQ= | ||||
| github.com/go-playground/validator/v10 v10.11.1/go.mod h1:i+3WkQ1FvaUjjxh1kSvIA4dMGDBiPU55YFDl0WbKdWU= | ||||
| github.com/go-playground/validator/v10 v10.12.0 h1:E4gtWgxWxp8YSxExrQFv5BpCahla0PVF2oTTEYaWQGI= | ||||
| github.com/go-playground/validator/v10 v10.12.0/go.mod h1:hCAPuzYvKdP33pxWa+2+6AIKXEKqjIUyqsNCtbsSJrA= | ||||
| github.com/go-redis/redis/v8 v8.11.5 h1:AcZZR7igkdvfVmQTPnu9WE37LRrO/YrBH5zWyjDC0oI= | ||||
| github.com/go-redis/redis/v8 v8.11.5/go.mod h1:gREzHqY1hg6oD9ngVRbLStwAWKhA0FEgq8Jd4h5lpwo= | ||||
| github.com/go-sql-driver/mysql v1.6.0 h1:BCTh4TKNUYmOmMUcQ3IipzF5prigylS7XXjEkfCHuOE= | ||||
| github.com/go-sql-driver/mysql v1.6.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg= | ||||
| github.com/goccy/go-json v0.9.7 h1:IcB+Aqpx/iMHu5Yooh7jEzJk1JZ7Pjtmys2ukPr7EeM= | ||||
| github.com/goccy/go-json v0.9.7/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I= | ||||
| github.com/goccy/go-json v0.10.2 h1:CrxCmQqYDkv1z7lO7Wbh2HN93uovUHgrECaO5ZrCXAU= | ||||
| github.com/goccy/go-json v0.10.2/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I= | ||||
| @@ -94,19 +88,16 @@ github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= | ||||
| github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY= | ||||
| github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE= | ||||
| github.com/leodido/go-urn v1.2.0/go.mod h1:+8+nEpDfqqsY+g338gtMEUOtuK+4dEMhiQEgxpxOKII= | ||||
| github.com/leodido/go-urn v1.2.1 h1:BqpAaACuzVSgi/VLzGZIobT2z4v53pjosyNd9Yv6n/w= | ||||
| github.com/leodido/go-urn v1.2.1/go.mod h1:zt4jvISO2HfUBqxjfIshjdMTYS56ZS/qv49ictyFfxY= | ||||
| github.com/leodido/go-urn v1.2.3 h1:6BE2vPT0lqoz3fmOesHZiaiFh7889ssCo2GMvLCfiuA= | ||||
| github.com/leodido/go-urn v1.2.3/go.mod h1:7ZrI8mTSeBSHl/UaRyKQW1qZeMgak41ANeCNaVckg+4= | ||||
| github.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU= | ||||
| github.com/mattn/go-isatty v0.0.14 h1:yVuAays6BHfxijgZPzw+3Zlu5yQgKGP2/hcQbHb7S9Y= | ||||
| github.com/mattn/go-isatty v0.0.14/go.mod h1:7GGIvUiUoEMVVmxf/4nioHXj79iQHKdU27kJ6hsGG94= | ||||
| github.com/mattn/go-isatty v0.0.18 h1:DOKFKCQ7FNG2L1rbrmstDN4QVRdS89Nkh85u68Uwp98= | ||||
| github.com/mattn/go-isatty v0.0.18/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= | ||||
| github.com/mattn/go-sqlite3 v1.14.15/go.mod h1:2eHXhiwb8IkHr+BDWZGa96P6+rkvnG63S2DGjv9HUNg= | ||||
| github.com/mattn/go-sqlite3 v2.0.3+incompatible h1:gXHsfypPkaMZrKbD5209QV9jbUTJKjyR5WD3HYQSd+U= | ||||
| github.com/mattn/go-sqlite3 v2.0.3+incompatible/go.mod h1:FPy6KqzDD04eiIsT53CuJW3U88zkxoIYsOqkbpncsNc= | ||||
| github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421 h1:ZqeYNhU3OHLH3mGKHDcjJRFFRrJa6eAM5H+CtDdOsPc= | ||||
| github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= | ||||
| github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg= | ||||
| github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= | ||||
| @@ -116,7 +107,6 @@ github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjY | ||||
| github.com/nxadm/tail v1.4.8 h1:nPr65rt6Y5JFSKQO7qToXr7pePgD6Gwiw05lkbyAQTE= | ||||
| github.com/onsi/ginkgo v1.16.5 h1:8xi0RTUf59SOSfEtZMvwTvXYMzG4gV23XVHOZiXNtnE= | ||||
| github.com/onsi/gomega v1.18.1 h1:M1GfJqGRrBrrGGsbxzV5dqM2U2ApXefZCQpkukxYRLE= | ||||
| github.com/pelletier/go-toml/v2 v2.0.1 h1:8e3L2cCQzLFi2CR4g7vGFuFxX7Jl1kKX8gW+iV0GUKU= | ||||
| github.com/pelletier/go-toml/v2 v2.0.1/go.mod h1:r9LEWfGN8R5k0VXJ+0BkIe7MYkRdwZOjgMj2KwnJFUo= | ||||
| github.com/pelletier/go-toml/v2 v2.0.7 h1:muncTPStnKRos5dpVKULv2FVd4bMOhNePj9CjgDb8Us= | ||||
| github.com/pelletier/go-toml/v2 v2.0.7/go.mod h1:eumQOmlWiOPt5WriQQqoM5y18pDHwha2N+QD+EUNTek= | ||||
| @@ -136,7 +126,6 @@ github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81P | ||||
| github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= | ||||
| github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= | ||||
| github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= | ||||
| github.com/stretchr/testify v1.8.0 h1:pSgiaMZlXftHpm5L7V1+rVB+AZJydKsMxsQBIJw4PKk= | ||||
| github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= | ||||
| github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= | ||||
| github.com/stretchr/testify v1.8.2 h1:+h33VjcLVPDHtOdpUCuF+7gSuG3yGIftsP1YvFihtJ8= | ||||
| @@ -146,7 +135,6 @@ github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2 | ||||
| github.com/ugorji/go v1.1.7/go.mod h1:kZn38zHttfInRq0xu/PH0az30d+z6vm202qpg1oXVMw= | ||||
| github.com/ugorji/go v1.2.7/go.mod h1:nF9osbDWLy6bDVv/Rtoh6QgnvNDpmCalQV5urGCCS6M= | ||||
| github.com/ugorji/go/codec v1.1.7/go.mod h1:Ax+UKWsSmolVDwsd+7N3ZtXu+yMGCf907BLYF3GoBXY= | ||||
| github.com/ugorji/go/codec v1.2.7 h1:YPXUKf7fYbp/y8xloBqZOw2qaVggbfwMlI8WM3wZUJ0= | ||||
| github.com/ugorji/go/codec v1.2.7/go.mod h1:WGN1fab3R1fzQlVQTkfxVtIBhWDRqOviHU95kRgeqEY= | ||||
| github.com/ugorji/go/codec v1.2.11 h1:BMaWp1Bb6fHwEtbplGBGJ498wD+LKlNSl25MjdZY4dU= | ||||
| github.com/ugorji/go/codec v1.2.11/go.mod h1:UNopzCgEMSXjBc6AOMqYvWC1ktqTAfzJZUZgYf6w6lg= | ||||
| @@ -154,26 +142,17 @@ golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUu | ||||
| golang.org/x/arch v0.3.0 h1:02VY4/ZcO/gBOH6PUaoiptASxtXU10jazRCP865E97k= | ||||
| golang.org/x/arch v0.3.0/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8= | ||||
| golang.org/x/crypto v0.0.0-20210711020723-a769d52b0f97/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc= | ||||
| golang.org/x/crypto v0.0.0-20211215153901-e495a2d5b3d3/go.mod h1:IxCIyHEi3zRg3s0A5j5BB6A9Jmi73HwBIUl50j+osU4= | ||||
| golang.org/x/crypto v0.1.0 h1:MDRAIl0xIo9Io2xV565hzXHw3zVseKrJKodhohM5CjU= | ||||
| golang.org/x/crypto v0.1.0/go.mod h1:RecgLatLF4+eUMCP1PoPZQb+cVrJcOPbHkTkbkB9sbw= | ||||
| golang.org/x/crypto v0.8.0 h1:pd9TJtTueMTVQXzk8E2XESSMQDj/U7OUu0PqJqPXQjQ= | ||||
| golang.org/x/crypto v0.8.0/go.mod h1:mRqEX+O9/h5TFCrQhkgjo2yKi0yYA+9ecGkdQoHrywE= | ||||
| golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg= | ||||
| golang.org/x/net v0.0.0-20211112202133-69e39bad7dc2/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y= | ||||
| golang.org/x/net v0.7.0 h1:rJrUqqhjsgNp7KqAIc25s9pZnjU7TUcSY7HcVZjdn1g= | ||||
| golang.org/x/net v0.7.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs= | ||||
| golang.org/x/net v0.9.0 h1:aWJ/m6xSmxWBx+V0XRHTlrYrPG56jKsLdTFmsSsCzOM= | ||||
| golang.org/x/net v0.9.0/go.mod h1:d48xBJpPfHeWQsugry2m+kC02ZBRGRgulfHnEXEuWns= | ||||
| golang.org/x/sys v0.0.0-20200116001909-b77594299b42/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= | ||||
| golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= | ||||
| golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= | ||||
| golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||||
| golang.org/x/sys v0.0.0-20210630005230-0f9fa26af87c/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||||
| golang.org/x/sys v0.0.0-20210806184541-e5e7981a1069/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||||
| golang.org/x/sys v0.0.0-20220704084225-05e143d24a9e/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||||
| golang.org/x/sys v0.5.0 h1:MUK/U/4lj1t1oPg0HfuXDN/Z1wv31ZJ/YcPiGccS4DU= | ||||
| golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||||
| golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||||
| golang.org/x/sys v0.7.0 h1:3jlCCIQZPdOYu1h8BkNvLz8Kgwtae2cagcG/VamtZRU= | ||||
| golang.org/x/sys v0.7.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= | ||||
| @@ -181,28 +160,20 @@ golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9sn | ||||
| golang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk= | ||||
| golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= | ||||
| golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= | ||||
| golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ= | ||||
| golang.org/x/text v0.7.0 h1:4BRB4x83lYWy72KwLD/qYDuTu7q9PjSagHvijDw7cLo= | ||||
| golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8= | ||||
| golang.org/x/text v0.9.0 h1:2sjJmO8cDvYveuX97RDLsxlyUxLl+GHoLxBiRdHllBE= | ||||
| golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8= | ||||
| golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= | ||||
| golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4= | ||||
| golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= | ||||
| google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw= | ||||
| google.golang.org/protobuf v1.28.0 h1:w43yiav+6bVFTBQFZX0r7ipe9JQ1QsbMgHwbBziscLw= | ||||
| google.golang.org/protobuf v1.28.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I= | ||||
| google.golang.org/protobuf v1.30.0 h1:kPPoIgf3TsEvrm0PFe15JQ+570QVxYzEvvHqChK+cng= | ||||
| google.golang.org/protobuf v1.30.0/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I= | ||||
| gopkg.in/alexcesaro/quotedprintable.v3 v3.0.0-20150716171945-2caba252f4dc h1:2gGKlE2+asNV9m7xrywl36YYNnBG5ZQ0r/BOOxqPpmk= | ||||
| gopkg.in/alexcesaro/quotedprintable.v3 v3.0.0-20150716171945-2caba252f4dc/go.mod h1:m7x9LTH6d71AHyAX77c9yqWCCa3UKHcVEj9y7hAtKDk= | ||||
| gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= | ||||
| gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= | ||||
| gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk= | ||||
| gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q= | ||||
| gopkg.in/errgo.v2 v2.1.0/go.mod h1:hNsd1EY+bozCKY1Ytp96fpM3vjJbqLJn88ws8XvfDNI= | ||||
| gopkg.in/gomail.v2 v2.0.0-20160411212932-81ebce5c23df h1:n7WqCuqOuCbNr617RXOY0AWRXxgwEyPp2z+p0+hgMuE= | ||||
| gopkg.in/gomail.v2 v2.0.0-20160411212932-81ebce5c23df/go.mod h1:LRQQ+SO6ZHR7tOkpBDuZnXENFzX8qRjMDMyPD6BRkCw= | ||||
| gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 h1:uRGJdciOHaEIrze2W8Q3AKkepLTh2hOroT7a+7czfdQ= | ||||
| gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= | ||||
| gopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= | ||||
|   | ||||
| @@ -111,14 +111,9 @@ func TokenAuth() func(c *gin.Context) { | ||||
| 		c.Set("id", token.UserId) | ||||
| 		c.Set("token_id", token.Id) | ||||
| 		requestURL := c.Request.URL.String() | ||||
| 		consumeQuota := false | ||||
| 		switch requestURL { | ||||
| 		case "/v1/chat/completions": | ||||
| 			consumeQuota = !token.UnlimitedQuota | ||||
| 		case "/v1/completions": | ||||
| 			consumeQuota = !token.UnlimitedQuota | ||||
| 		case "/v1/edits": | ||||
| 			consumeQuota = !token.UnlimitedQuota | ||||
| 		consumeQuota := !token.UnlimitedQuota | ||||
| 		if strings.HasPrefix(requestURL, "/models") { | ||||
| 			consumeQuota = false | ||||
| 		} | ||||
| 		c.Set("consume_quota", consumeQuota) | ||||
| 		if len(parts) > 1 { | ||||
|   | ||||
| @@ -65,6 +65,9 @@ func Distribute() func(c *gin.Context) { | ||||
| 		c.Request.Header.Set("Authorization", fmt.Sprintf("Bearer %s", channel.Key)) | ||||
| 		if channel.Type == common.ChannelTypeCustom || channel.Type == common.ChannelTypeAzure { | ||||
| 			c.Set("base_url", channel.BaseURL) | ||||
| 			if channel.Type == common.ChannelTypeAzure { | ||||
| 				c.Set("api_version", channel.Other) | ||||
| 			} | ||||
| 		} | ||||
| 		c.Next() | ||||
| 	} | ||||
|   | ||||
| @@ -15,6 +15,7 @@ type Channel struct { | ||||
| 	CreatedTime  int64  `json:"created_time" gorm:"bigint"` | ||||
| 	AccessedTime int64  `json:"accessed_time" gorm:"bigint"` | ||||
| 	BaseURL      string `json:"base_url" gorm:"column:base_url"` | ||||
| 	Other        string `json:"other"` | ||||
| } | ||||
|  | ||||
| func GetAllChannels(startIdx int, num int) ([]*Channel, error) { | ||||
| @@ -52,6 +53,12 @@ func GetRandomChannel() (*Channel, error) { | ||||
| 	return &channel, err | ||||
| } | ||||
|  | ||||
| func BatchInsertChannels(channels []Channel) error { | ||||
| 	var err error | ||||
| 	err = DB.Create(&channels).Error | ||||
| 	return err | ||||
| } | ||||
|  | ||||
| func (channel *Channel) Insert() error { | ||||
| 	var err error | ||||
| 	err = DB.Create(channel).Error | ||||
|   | ||||
| @@ -33,12 +33,16 @@ func InitOptionMap() { | ||||
| 	common.OptionMap["TurnstileCheckEnabled"] = strconv.FormatBool(common.TurnstileCheckEnabled) | ||||
| 	common.OptionMap["RegisterEnabled"] = strconv.FormatBool(common.RegisterEnabled) | ||||
| 	common.OptionMap["SMTPServer"] = "" | ||||
| 	common.OptionMap["SMTPFrom"] = "" | ||||
| 	common.OptionMap["SMTPPort"] = strconv.Itoa(common.SMTPPort) | ||||
| 	common.OptionMap["SMTPAccount"] = "" | ||||
| 	common.OptionMap["SMTPToken"] = "" | ||||
| 	common.OptionMap["Notice"] = "" | ||||
| 	common.OptionMap["About"] = "" | ||||
| 	common.OptionMap["HomePageContent"] = "" | ||||
| 	common.OptionMap["Footer"] = common.Footer | ||||
| 	common.OptionMap["SystemName"] = common.SystemName | ||||
| 	common.OptionMap["Logo"] = common.Logo | ||||
| 	common.OptionMap["ServerAddress"] = "" | ||||
| 	common.OptionMap["GitHubClientId"] = "" | ||||
| 	common.OptionMap["GitHubClientSecret"] = "" | ||||
| @@ -120,6 +124,8 @@ func updateOptionMap(key string, value string) (err error) { | ||||
| 		common.SMTPPort = intValue | ||||
| 	case "SMTPAccount": | ||||
| 		common.SMTPAccount = value | ||||
| 	case "SMTPFrom": | ||||
| 		common.SMTPFrom = value | ||||
| 	case "SMTPToken": | ||||
| 		common.SMTPToken = value | ||||
| 	case "ServerAddress": | ||||
| @@ -130,6 +136,10 @@ func updateOptionMap(key string, value string) (err error) { | ||||
| 		common.GitHubClientSecret = value | ||||
| 	case "Footer": | ||||
| 		common.Footer = value | ||||
| 	case "SystemName": | ||||
| 		common.SystemName = value | ||||
| 	case "Logo": | ||||
| 		common.Logo = value | ||||
| 	case "WeChatServerAddress": | ||||
| 		common.WeChatServerAddress = value | ||||
| 	case "WeChatServerToken": | ||||
|   | ||||
| @@ -15,6 +15,7 @@ func SetApiRouter(router *gin.Engine) { | ||||
| 		apiRouter.GET("/status", controller.GetStatus) | ||||
| 		apiRouter.GET("/notice", controller.GetNotice) | ||||
| 		apiRouter.GET("/about", controller.GetAbout) | ||||
| 		apiRouter.GET("/home_page_content", controller.GetHomePageContent) | ||||
| 		apiRouter.GET("/verification", middleware.CriticalRateLimit(), middleware.TurnstileCheck(), controller.SendEmailVerification) | ||||
| 		apiRouter.GET("/reset_password", middleware.CriticalRateLimit(), middleware.TurnstileCheck(), controller.SendPasswordResetEmail) | ||||
| 		apiRouter.POST("/user/reset", middleware.CriticalRateLimit(), controller.ResetPassword) | ||||
| @@ -62,6 +63,7 @@ func SetApiRouter(router *gin.Engine) { | ||||
| 			channelRoute.GET("/", controller.GetAllChannels) | ||||
| 			channelRoute.GET("/search", controller.SearchChannels) | ||||
| 			channelRoute.GET("/:id", controller.GetChannel) | ||||
| 			channelRoute.GET("/test/:id", controller.TestChannel) | ||||
| 			channelRoute.POST("/", controller.AddChannel) | ||||
| 			channelRoute.PUT("/", controller.UpdateChannel) | ||||
| 			channelRoute.DELETE("/:id", controller.DeleteChannel) | ||||
|   | ||||
| @@ -42,6 +42,8 @@ function App() { | ||||
|     if (success) { | ||||
|       localStorage.setItem('status', JSON.stringify(data)); | ||||
|       statusDispatch({ type: 'set', payload: data }); | ||||
|       localStorage.setItem('system_name', data.system_name); | ||||
|       localStorage.setItem('logo', data.logo); | ||||
|       localStorage.setItem('footer_html', data.footer_html); | ||||
|       if ( | ||||
|         data.version !== process.env.REACT_APP_VERSION && | ||||
|   | ||||
| @@ -1,7 +1,7 @@ | ||||
| import React, { useEffect, useState } from 'react'; | ||||
| import { Button, Form, Label, Pagination, Table } from 'semantic-ui-react'; | ||||
| import { Button, Form, Label, Pagination, Popup, Table } from 'semantic-ui-react'; | ||||
| import { Link } from 'react-router-dom'; | ||||
| import { API, copy, showError, showSuccess, timestamp2string } from '../helpers'; | ||||
| import { API, copy, showError, showInfo, showSuccess, timestamp2string } from '../helpers'; | ||||
|  | ||||
| import { CHANNEL_OPTIONS, ITEMS_PER_PAGE } from '../constants'; | ||||
|  | ||||
| @@ -139,6 +139,16 @@ const ChannelsTable = () => { | ||||
|     setSearching(false); | ||||
|   }; | ||||
|  | ||||
|   const testChannel = async (id, name) => { | ||||
|     const res = await API.get(`/api/channel/test/${id}/`); | ||||
|     const { success, message, time } = res.data; | ||||
|     if (success) { | ||||
|       showInfo(`通道 ${name} 测试成功,耗时 ${time} 秒。`); | ||||
|     } else { | ||||
|       showError(message); | ||||
|     } | ||||
|   } | ||||
|  | ||||
|   const handleKeywordChange = async (e, { value }) => { | ||||
|     setSearchKeyword(value.trim()); | ||||
|   }; | ||||
| @@ -246,13 +256,32 @@ const ChannelsTable = () => { | ||||
|                     <div> | ||||
|                       <Button | ||||
|                         size={'small'} | ||||
|                         positive | ||||
|                         onClick={() => { | ||||
|                           testChannel(channel.id, channel.name); | ||||
|                         }} | ||||
|                       > | ||||
|                         测试 | ||||
|                       </Button> | ||||
|                       <Popup | ||||
|                         trigger={ | ||||
|                           <Button size='small' negative> | ||||
|                             删除 | ||||
|                           </Button> | ||||
|                         } | ||||
|                         on='click' | ||||
|                         flowing | ||||
|                         hoverable | ||||
|                       > | ||||
|                         <Button | ||||
|                           negative | ||||
|                           onClick={() => { | ||||
|                             manageChannel(channel.id, 'delete', idx); | ||||
|                           }} | ||||
|                         > | ||||
|                         删除 | ||||
|                           删除渠道 {channel.name} | ||||
|                         </Button> | ||||
|                       </Popup> | ||||
|                       <Button | ||||
|                         size={'small'} | ||||
|                         onClick={() => { | ||||
|   | ||||
| @@ -1,40 +1,37 @@ | ||||
| import React, { useEffect, useState } from 'react'; | ||||
| import React from 'react'; | ||||
|  | ||||
| import { Container, Segment } from 'semantic-ui-react'; | ||||
| import { getFooterHTML, getSystemName } from '../helpers'; | ||||
|  | ||||
| const Footer = () => { | ||||
|   const [Footer, setFooter] = useState(''); | ||||
|   useEffect(() => { | ||||
|     let savedFooter = localStorage.getItem('footer_html'); | ||||
|     if (!savedFooter) savedFooter = ''; | ||||
|     setFooter(savedFooter); | ||||
|   }); | ||||
|   const systemName = getSystemName(); | ||||
|   const footer = getFooterHTML(); | ||||
|  | ||||
|   return ( | ||||
|     <Segment vertical> | ||||
|       <Container textAlign="center"> | ||||
|         {Footer === '' ? ( | ||||
|           <div className="custom-footer"> | ||||
|       <Container textAlign='center'> | ||||
|         {footer ? ( | ||||
|           <div | ||||
|             className='custom-footer' | ||||
|             dangerouslySetInnerHTML={{ __html: footer }} | ||||
|           ></div> | ||||
|         ) : ( | ||||
|           <div className='custom-footer'> | ||||
|             <a | ||||
|               href="https://github.com/songquanpeng/one-api" | ||||
|               target="_blank" | ||||
|               href='https://github.com/songquanpeng/one-api' | ||||
|               target='_blank' | ||||
|             > | ||||
|               One API {process.env.REACT_APP_VERSION}{' '} | ||||
|               {systemName} {process.env.REACT_APP_VERSION}{' '} | ||||
|             </a> | ||||
|             由{' '} | ||||
|             <a href="https://github.com/songquanpeng" target="_blank"> | ||||
|             <a href='https://github.com/songquanpeng' target='_blank'> | ||||
|               JustSong | ||||
|             </a>{' '} | ||||
|             构建,源代码遵循{' '} | ||||
|             <a href="https://opensource.org/licenses/mit-license.php"> | ||||
|             <a href='https://opensource.org/licenses/mit-license.php'> | ||||
|               MIT 协议 | ||||
|             </a> | ||||
|           </div> | ||||
|         ) : ( | ||||
|           <div | ||||
|             className="custom-footer" | ||||
|             dangerouslySetInnerHTML={{ __html: Footer }} | ||||
|           ></div> | ||||
|         )} | ||||
|       </Container> | ||||
|     </Segment> | ||||
|   | ||||
| @@ -3,7 +3,7 @@ import { Link, useNavigate } from 'react-router-dom'; | ||||
| import { UserContext } from '../context/User'; | ||||
|  | ||||
| import { Button, Container, Dropdown, Icon, Menu, Segment } from 'semantic-ui-react'; | ||||
| import { API, isAdmin, isMobile, showSuccess } from '../helpers'; | ||||
| import { API, getLogo, getSystemName, isAdmin, isMobile, showSuccess } from '../helpers'; | ||||
| import '../index.css'; | ||||
|  | ||||
| // Header Buttons | ||||
| @@ -53,6 +53,8 @@ const Header = () => { | ||||
|   let navigate = useNavigate(); | ||||
|  | ||||
|   const [showSidebar, setShowSidebar] = useState(false); | ||||
|   const systemName = getSystemName(); | ||||
|   const logo = getLogo(); | ||||
|  | ||||
|   async function logout() { | ||||
|     setShowSidebar(false); | ||||
| @@ -111,12 +113,12 @@ const Header = () => { | ||||
|           <Container> | ||||
|             <Menu.Item as={Link} to='/'> | ||||
|               <img | ||||
|                 src='/logo.png' | ||||
|                 src={logo} | ||||
|                 alt='logo' | ||||
|                 style={{ marginRight: '0.75em' }} | ||||
|               /> | ||||
|               <div style={{ fontSize: '20px' }}> | ||||
|                 <b>One API</b> | ||||
|                 <b>{systemName}</b> | ||||
|               </div> | ||||
|             </Menu.Item> | ||||
|             <Menu.Menu position='right'> | ||||
| @@ -168,9 +170,9 @@ const Header = () => { | ||||
|       <Menu borderless style={{ borderTop: 'none' }}> | ||||
|         <Container> | ||||
|           <Menu.Item as={Link} to='/' className={'hide-on-mobile'}> | ||||
|             <img src='/logo.png' alt='logo' style={{ marginRight: '0.75em' }} /> | ||||
|             <img src={logo} alt='logo' style={{ marginRight: '0.75em' }} /> | ||||
|             <div style={{ fontSize: '20px' }}> | ||||
|               <b>One API</b> | ||||
|               <b>{systemName}</b> | ||||
|             </div> | ||||
|           </Menu.Item> | ||||
|           {renderButtons(false)} | ||||
|   | ||||
| @@ -12,7 +12,7 @@ import { | ||||
| } from 'semantic-ui-react'; | ||||
| import { Link, useNavigate, useSearchParams } from 'react-router-dom'; | ||||
| import { UserContext } from '../context/User'; | ||||
| import { API, showError, showSuccess } from '../helpers'; | ||||
| import { API, getLogo, showError, showSuccess } from '../helpers'; | ||||
|  | ||||
| const LoginForm = () => { | ||||
|   const [inputs, setInputs] = useState({ | ||||
| @@ -27,6 +27,7 @@ const LoginForm = () => { | ||||
|   let navigate = useNavigate(); | ||||
|  | ||||
|   const [status, setStatus] = useState({}); | ||||
|   const logo = getLogo(); | ||||
|  | ||||
|   useEffect(() => { | ||||
|     if (searchParams.get("expired")) { | ||||
| @@ -95,7 +96,7 @@ const LoginForm = () => { | ||||
|     <Grid textAlign="center" style={{ marginTop: '48px' }}> | ||||
|       <Grid.Column style={{ maxWidth: 450 }}> | ||||
|         <Header as="h2" color="" textAlign="center"> | ||||
|           <Image src="/logo.png" /> 用户登录 | ||||
|           <Image src={logo} /> 用户登录 | ||||
|         </Header> | ||||
|         <Form size="large"> | ||||
|           <Segment> | ||||
|   | ||||
| @@ -8,6 +8,9 @@ const OtherSetting = () => { | ||||
|     Footer: '', | ||||
|     Notice: '', | ||||
|     About: '', | ||||
|     SystemName: '', | ||||
|     Logo: '', | ||||
|     HomePageContent: '', | ||||
|   }); | ||||
|   let originInputs = {}; | ||||
|   let [loading, setLoading] = useState(false); | ||||
| @@ -65,10 +68,22 @@ const OtherSetting = () => { | ||||
|     await updateOption('Footer', inputs.Footer); | ||||
|   }; | ||||
|  | ||||
|   const submitSystemName = async () => { | ||||
|     await updateOption('SystemName', inputs.SystemName); | ||||
|   }; | ||||
|  | ||||
|   const submitLogo = async () => { | ||||
|     await updateOption('Logo', inputs.Logo); | ||||
|   }; | ||||
|  | ||||
|   const submitAbout = async () => { | ||||
|     await updateOption('About', inputs.About); | ||||
|   }; | ||||
|  | ||||
|   const submitOption = async (key) => { | ||||
|     await updateOption(key, inputs[key]); | ||||
|   }; | ||||
|  | ||||
|   const openGitHubRelease = () => { | ||||
|     window.location = | ||||
|       'https://github.com/songquanpeng/one-api/releases/latest'; | ||||
| @@ -109,10 +124,42 @@ const OtherSetting = () => { | ||||
|           <Form.Button onClick={submitNotice}>保存公告</Form.Button> | ||||
|           <Divider /> | ||||
|           <Header as='h3'>个性化设置</Header> | ||||
|           <Form.Group widths='equal'> | ||||
|             <Form.Input | ||||
|               label='系统名称' | ||||
|               placeholder='在此输入系统名称' | ||||
|               value={inputs.SystemName} | ||||
|               name='SystemName' | ||||
|               onChange={handleInputChange} | ||||
|             /> | ||||
|           </Form.Group> | ||||
|           <Form.Button onClick={submitSystemName}>设置系统名称</Form.Button> | ||||
|           <Form.Group widths='equal'> | ||||
|             <Form.Input | ||||
|               label='Logo 图片地址' | ||||
|               placeholder='在此输入 Logo 图片地址' | ||||
|               value={inputs.Logo} | ||||
|               name='Logo' | ||||
|               type='url' | ||||
|               onChange={handleInputChange} | ||||
|             /> | ||||
|           </Form.Group> | ||||
|           <Form.Button onClick={submitLogo}>设置 Logo</Form.Button> | ||||
|           <Form.Group widths='equal'> | ||||
|             <Form.TextArea | ||||
|               label='首页内容' | ||||
|               placeholder='在此输入首页内容,支持 Markdown & HTML 代码,设置后首页的状态信息将不再显示。如果输入的是一个链接,则会使用该链接作为 iframe 的 src 属性,这允许你设置任意网页作为首页。' | ||||
|               value={inputs.HomePageContent} | ||||
|               name='HomePageContent' | ||||
|               onChange={handleInputChange} | ||||
|               style={{ minHeight: 150, fontFamily: 'JetBrains Mono, Consolas' }} | ||||
|             /> | ||||
|           </Form.Group> | ||||
|           <Form.Button onClick={()=>submitOption('HomePageContent')}>保存首页内容</Form.Button> | ||||
|           <Form.Group widths='equal'> | ||||
|             <Form.TextArea | ||||
|               label='关于' | ||||
|               placeholder='在此输入新的关于内容,支持 Markdown & HTML 代码' | ||||
|               placeholder='在此输入新的关于内容,支持 Markdown & HTML 代码。如果输入的是一个链接,则会使用该链接作为 iframe 的 src 属性,这允许你设置任意网页作为关于页面。' | ||||
|               value={inputs.About} | ||||
|               name='About' | ||||
|               onChange={handleInputChange} | ||||
|   | ||||
| @@ -9,7 +9,7 @@ import { | ||||
|   Segment, | ||||
| } from 'semantic-ui-react'; | ||||
| import { Link, useNavigate } from 'react-router-dom'; | ||||
| import { API, showError, showInfo, showSuccess } from '../helpers'; | ||||
| import { API, getLogo, showError, showInfo, showSuccess } from '../helpers'; | ||||
| import Turnstile from 'react-turnstile'; | ||||
|  | ||||
| const RegisterForm = () => { | ||||
| @@ -26,6 +26,7 @@ const RegisterForm = () => { | ||||
|   const [turnstileSiteKey, setTurnstileSiteKey] = useState(''); | ||||
|   const [turnstileToken, setTurnstileToken] = useState(''); | ||||
|   const [loading, setLoading] = useState(false); | ||||
|   const logo = getLogo(); | ||||
|  | ||||
|   useEffect(() => { | ||||
|     let status = localStorage.getItem('status'); | ||||
| @@ -100,7 +101,7 @@ const RegisterForm = () => { | ||||
|     <Grid textAlign='center' style={{ marginTop: '48px' }}> | ||||
|       <Grid.Column style={{ maxWidth: 450 }}> | ||||
|         <Header as='h2' color='' textAlign='center'> | ||||
|           <Image src='/logo.png' /> 新用户注册 | ||||
|           <Image src={logo} /> 新用户注册 | ||||
|         </Header> | ||||
|         <Form size='large'> | ||||
|           <Segment> | ||||
|   | ||||
| @@ -14,6 +14,7 @@ const SystemSetting = () => { | ||||
|     SMTPServer: '', | ||||
|     SMTPPort: '', | ||||
|     SMTPAccount: '', | ||||
|     SMTPFrom: '', | ||||
|     SMTPToken: '', | ||||
|     ServerAddress: '', | ||||
|     Footer: '', | ||||
| @@ -129,6 +130,9 @@ const SystemSetting = () => { | ||||
|     if (originInputs['SMTPAccount'] !== inputs.SMTPAccount) { | ||||
|       await updateOption('SMTPAccount', inputs.SMTPAccount); | ||||
|     } | ||||
|     if (originInputs['SMTPFrom'] !== inputs.SMTPFrom) { | ||||
|       await updateOption('SMTPFrom', inputs.SMTPFrom); | ||||
|     } | ||||
|     if ( | ||||
|       originInputs['SMTPPort'] !== inputs.SMTPPort && | ||||
|       inputs.SMTPPort !== '' | ||||
| @@ -298,7 +302,7 @@ const SystemSetting = () => { | ||||
|             配置 SMTP | ||||
|             <Header.Subheader>用以支持系统的邮件发送</Header.Subheader> | ||||
|           </Header> | ||||
|           <Form.Group widths={4}> | ||||
|           <Form.Group widths={3}> | ||||
|             <Form.Input | ||||
|               label='SMTP 服务器地址' | ||||
|               name='SMTPServer' | ||||
| @@ -323,6 +327,16 @@ const SystemSetting = () => { | ||||
|               value={inputs.SMTPAccount} | ||||
|               placeholder='通常是邮箱地址' | ||||
|             /> | ||||
|           </Form.Group> | ||||
|           <Form.Group widths={3}> | ||||
|             <Form.Input | ||||
|               label='SMTP 发送者邮箱' | ||||
|               name='SMTPFrom' | ||||
|               onChange={handleInputChange} | ||||
|               autoComplete='new-password' | ||||
|               value={inputs.SMTPFrom} | ||||
|               placeholder='通常和邮箱地址保持一致' | ||||
|             /> | ||||
|             <Form.Input | ||||
|               label='SMTP 访问凭证' | ||||
|               name='SMTPToken' | ||||
|   | ||||
| @@ -1,5 +1,5 @@ | ||||
| import React, { useEffect, useState } from 'react'; | ||||
| import { Button, Form, Label, Modal, Pagination, Table } from 'semantic-ui-react'; | ||||
| import { Button, Form, Label, Modal, Pagination, Popup, Table } from 'semantic-ui-react'; | ||||
| import { Link } from 'react-router-dom'; | ||||
| import { API, copy, showError, showSuccess, showWarning, timestamp2string } from '../helpers'; | ||||
|  | ||||
| @@ -283,15 +283,25 @@ const TokensTable = () => { | ||||
|                         }}> | ||||
|                         充值 | ||||
|                       </Button> | ||||
|                       <Popup | ||||
|                         trigger={ | ||||
|                           <Button size='small' negative> | ||||
|                             删除 | ||||
|                           </Button> | ||||
|                         } | ||||
|                         on='click' | ||||
|                         flowing | ||||
|                         hoverable | ||||
|                       > | ||||
|                         <Button | ||||
|                         size={'small'} | ||||
|                           negative | ||||
|                           onClick={() => { | ||||
|                             manageToken(token.id, 'delete', idx); | ||||
|                           }} | ||||
|                         > | ||||
|                         删除 | ||||
|                           删除令牌 {token.name} | ||||
|                         </Button> | ||||
|                       </Popup> | ||||
|                       <Button | ||||
|                         size={'small'} | ||||
|                         onClick={() => { | ||||
|   | ||||
| @@ -15,6 +15,22 @@ export function isRoot() { | ||||
|   return user.role >= 100; | ||||
| } | ||||
|  | ||||
| export function getSystemName() { | ||||
|   let system_name = localStorage.getItem('system_name'); | ||||
|   if (!system_name) return 'One API'; | ||||
|   return system_name; | ||||
| } | ||||
|  | ||||
| export function getLogo() { | ||||
|   let logo = localStorage.getItem('logo'); | ||||
|   if (!logo) return '/logo.png'; | ||||
|   return logo | ||||
| } | ||||
|  | ||||
| export function getFooterHTML() { | ||||
|   return localStorage.getItem('footer_html'); | ||||
| } | ||||
|  | ||||
| export async function copy(text) { | ||||
|   let okay = true; | ||||
|   try { | ||||
|   | ||||
| @@ -5,6 +5,11 @@ body { | ||||
|     font-family: Lato, 'Helvetica Neue', Arial, Helvetica, "Microsoft YaHei", sans-serif; | ||||
|     -webkit-font-smoothing: antialiased; | ||||
|     -moz-osx-font-smoothing: grayscale; | ||||
|     scrollbar-width: none; | ||||
| } | ||||
|  | ||||
| body::-webkit-scrollbar { | ||||
|     display: none; | ||||
| } | ||||
|  | ||||
| code { | ||||
|   | ||||
| @@ -5,18 +5,24 @@ import { marked } from 'marked'; | ||||
|  | ||||
| const About = () => { | ||||
|   const [about, setAbout] = useState(''); | ||||
|   const [aboutLoaded, setAboutLoaded] = useState(false); | ||||
|  | ||||
|   const displayAbout = async () => { | ||||
|     setAbout(localStorage.getItem('about') || ''); | ||||
|     const res = await API.get('/api/about'); | ||||
|     const { success, message, data } = res.data; | ||||
|     if (success) { | ||||
|       let HTMLAbout = marked.parse(data); | ||||
|       localStorage.setItem('about', HTMLAbout); | ||||
|       setAbout(HTMLAbout); | ||||
|       let aboutContent = data; | ||||
|       if (!data.startsWith('https://')) { | ||||
|         aboutContent = marked.parse(data); | ||||
|       } | ||||
|       setAbout(aboutContent); | ||||
|       localStorage.setItem('about', aboutContent); | ||||
|     } else { | ||||
|       showError(message); | ||||
|       setAbout('加载关于内容失败...'); | ||||
|     } | ||||
|     setAboutLoaded(true); | ||||
|   }; | ||||
|  | ||||
|   useEffect(() => { | ||||
| @@ -25,20 +31,27 @@ const About = () => { | ||||
|  | ||||
|   return ( | ||||
|     <> | ||||
|       <Segment> | ||||
|       { | ||||
|           about === '' ? <> | ||||
|         aboutLoaded && about === '' ? <> | ||||
|           <Segment> | ||||
|             <Header as='h3'>关于</Header> | ||||
|             <p>可在设置页面设置关于内容,支持 HTML & Markdown</p> | ||||
|             项目仓库地址: | ||||
|             <a href="https://github.com/songquanpeng/one-api"> | ||||
|             <a href='https://github.com/songquanpeng/one-api'> | ||||
|               https://github.com/songquanpeng/one-api | ||||
|             </a> | ||||
|           </Segment> | ||||
|         </> : <> | ||||
|             <div dangerouslySetInnerHTML={{ __html: about}}></div> | ||||
|           { | ||||
|             about.startsWith('https://') ? <iframe | ||||
|               src={about} | ||||
|               style={{ width: '100%', height: '100vh', border: 'none' }} | ||||
|             /> : <Segment> | ||||
|               <div style={{ fontSize: 'larger' }} dangerouslySetInnerHTML={{ __html: about }}></div> | ||||
|             </Segment> | ||||
|           } | ||||
|         </> | ||||
|       } | ||||
|       </Segment> | ||||
|     </> | ||||
|   ); | ||||
| }; | ||||
|   | ||||
| @@ -13,10 +13,13 @@ const EditChannel = () => { | ||||
|     name: '', | ||||
|     type: 1, | ||||
|     key: '', | ||||
|     base_url: '' | ||||
|     base_url: '', | ||||
|     other: '' | ||||
|   }; | ||||
|   const [batch, setBatch] = useState(false); | ||||
|   const [inputs, setInputs] = useState(originInputs); | ||||
|   const handleInputChange = (e, { name, value }) => { | ||||
|     console.log(name, value); | ||||
|     setInputs((inputs) => ({ ...inputs, [name]: value })); | ||||
|   }; | ||||
|  | ||||
| @@ -80,7 +83,7 @@ const EditChannel = () => { | ||||
|             inputs.type === 3 && ( | ||||
|               <> | ||||
|                 <Message> | ||||
|                   注意,创建资源时,部署名称必须和 OpenAI 官方的模型名称保持一致,因为 One API 会把请求体中的 model 参数替换为你的部署名称。 | ||||
|                   注意,<strong>模型部署名称必须和模型名称保持一致</strong>,因为 One API 会把请求体中的 model 参数替换为你的部署名称(模型名称中的点会被剔除)。 | ||||
|                 </Message> | ||||
|                 <Form.Field> | ||||
|                   <Form.Input | ||||
| @@ -92,6 +95,16 @@ const EditChannel = () => { | ||||
|                     autoComplete='new-password' | ||||
|                   /> | ||||
|                 </Form.Field> | ||||
|                 <Form.Field> | ||||
|                   <Form.Input | ||||
|                     label='默认 API 版本' | ||||
|                     name='other' | ||||
|                     placeholder={'请输入默认 API 版本,例如:2023-03-15-preview,该配置可以被实际的请求查询参数所覆盖'} | ||||
|                     onChange={handleInputChange} | ||||
|                     value={inputs.other} | ||||
|                     autoComplete='new-password' | ||||
|                   /> | ||||
|                 </Form.Field> | ||||
|               </> | ||||
|             ) | ||||
|           } | ||||
| @@ -119,17 +132,38 @@ const EditChannel = () => { | ||||
|               autoComplete='new-password' | ||||
|             /> | ||||
|           </Form.Field> | ||||
|           <Form.Field> | ||||
|           { | ||||
|             batch ? <Form.Field> | ||||
|               <Form.TextArea | ||||
|                 label='密钥' | ||||
|                 name='key' | ||||
|                 placeholder={'请输入密钥,一行一个'} | ||||
|                 onChange={handleInputChange} | ||||
|                 value={inputs.key} | ||||
|                 style={{ minHeight: 150, fontFamily: 'JetBrains Mono, Consolas' }} | ||||
|                 autoComplete='new-password' | ||||
|               /> | ||||
|             </Form.Field> : <Form.Field> | ||||
|               <Form.Input | ||||
|                 label='密钥' | ||||
|                 name='key' | ||||
|                 placeholder={'请输入密钥'} | ||||
|                 onChange={handleInputChange} | ||||
|                 value={inputs.key} | ||||
|               // type='password' | ||||
|                 autoComplete='new-password' | ||||
|                 /> | ||||
|             </Form.Field> | ||||
|           } | ||||
|           { | ||||
|             !isEdit && ( | ||||
|               <Form.Checkbox | ||||
|                 checked={batch} | ||||
|                 label='批量创建' | ||||
|                 name='batch' | ||||
|                 onChange={() => setBatch(!batch)} | ||||
|               /> | ||||
|             ) | ||||
|           } | ||||
|           <Button onClick={submit}>提交</Button> | ||||
|         </Form> | ||||
|       </Segment> | ||||
|   | ||||
| @@ -1,10 +1,13 @@ | ||||
| import React, { useContext, useEffect } from 'react'; | ||||
| import React, { useContext, useEffect, useState } from 'react'; | ||||
| import { Card, Grid, Header, Segment } from 'semantic-ui-react'; | ||||
| import { API, showError, showNotice, timestamp2string } from '../../helpers'; | ||||
| import { StatusContext } from '../../context/Status'; | ||||
| import { marked } from 'marked'; | ||||
|  | ||||
| const Home = () => { | ||||
|   const [statusState, statusDispatch] = useContext(StatusContext); | ||||
|   const [homePageContentLoaded, setHomePageContentLoaded] = useState(false); | ||||
|   const [homePageContent, setHomePageContent] = useState(''); | ||||
|  | ||||
|   const displayNotice = async () => { | ||||
|     const res = await API.get('/api/notice'); | ||||
| @@ -20,6 +23,24 @@ const Home = () => { | ||||
|     } | ||||
|   }; | ||||
|  | ||||
|   const displayHomePageContent = async () => { | ||||
|     setHomePageContent(localStorage.getItem('home_page_content') || ''); | ||||
|     const res = await API.get('/api/home_page_content'); | ||||
|     const { success, message, data } = res.data; | ||||
|     if (success) { | ||||
|       let content = data; | ||||
|       if (!data.startsWith('https://')) { | ||||
|         content = marked.parse(data); | ||||
|       } | ||||
|       setHomePageContent(content); | ||||
|       localStorage.setItem('home_page_content', content); | ||||
|     } else { | ||||
|       showError(message); | ||||
|       setHomePageContent('加载首页内容失败...'); | ||||
|     } | ||||
|     setHomePageContentLoaded(true); | ||||
|   }; | ||||
|  | ||||
|   const getStartTimeString = () => { | ||||
|     const timestamp = statusState?.status?.start_time; | ||||
|     return timestamp2string(timestamp); | ||||
| @@ -27,9 +48,12 @@ const Home = () => { | ||||
|  | ||||
|   useEffect(() => { | ||||
|     displayNotice().then(); | ||||
|     displayHomePageContent().then(); | ||||
|   }, []); | ||||
|   return ( | ||||
|     <> | ||||
|       { | ||||
|         homePageContentLoaded && homePageContent === '' ? <> | ||||
|           <Segment> | ||||
|             <Header as='h3'>系统状况</Header> | ||||
|             <Grid columns={2} stackable> | ||||
| @@ -91,6 +115,16 @@ const Home = () => { | ||||
|               </Grid.Column> | ||||
|             </Grid> | ||||
|           </Segment> | ||||
|         </> : <> | ||||
|           { | ||||
|             homePageContent.startsWith('https://') ? <iframe | ||||
|               src={homePageContent} | ||||
|               style={{ width: '100%', height: '100vh', border: 'none' }} | ||||
|             /> : <div style={{ fontSize: 'larger' }} dangerouslySetInnerHTML={{ __html: homePageContent }}></div> | ||||
|           } | ||||
|         </> | ||||
|       } | ||||
|  | ||||
|     </> | ||||
|   ); | ||||
| }; | ||||
|   | ||||
		Reference in New Issue
	
	Block a user