mirror of
https://github.com/yangjian102621/geekai.git
synced 2025-09-17 16:56:38 +08:00
358 lines
9.4 KiB
Go
358 lines
9.4 KiB
Go
package server
|
|
|
|
import (
|
|
"encoding/json"
|
|
"github.com/gin-gonic/gin"
|
|
"net/http"
|
|
"openai/types"
|
|
"openai/utils"
|
|
"strconv"
|
|
)
|
|
|
|
// ConfigSetHandle set configs
|
|
func (s *Server) ConfigSetHandle(c *gin.Context) {
|
|
var data map[string]string
|
|
err := json.NewDecoder(c.Request.Body).Decode(&data)
|
|
if err != nil {
|
|
logger.Errorf("Error decode json data: %s", err.Error())
|
|
c.JSON(http.StatusBadRequest, nil)
|
|
return
|
|
}
|
|
|
|
// Model
|
|
if model, ok := data["model"]; ok {
|
|
s.Config.Chat.Model = model
|
|
}
|
|
|
|
// Temperature
|
|
if temperature, ok := data["temperature"]; ok {
|
|
v, err := strconv.ParseFloat(temperature, 32)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, types.BizVo{
|
|
Code: types.InvalidParams,
|
|
Message: "temperature must be a float parameter",
|
|
})
|
|
return
|
|
}
|
|
s.Config.Chat.Temperature = float32(v)
|
|
}
|
|
|
|
// max_tokens
|
|
if maxTokens, ok := data["max_tokens"]; ok {
|
|
v, err := strconv.Atoi(maxTokens)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, types.BizVo{
|
|
Code: types.InvalidParams,
|
|
Message: "max_tokens must be a int parameter",
|
|
})
|
|
return
|
|
}
|
|
s.Config.Chat.MaxTokens = v
|
|
}
|
|
|
|
// enable Context
|
|
if enableContext, ok := data["enable_context"]; ok {
|
|
v, err := strconv.ParseBool(enableContext)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, types.BizVo{
|
|
Code: types.InvalidParams,
|
|
Message: "enable_context must be a bool parameter",
|
|
})
|
|
return
|
|
}
|
|
s.Config.Chat.EnableContext = v
|
|
}
|
|
|
|
// enable auth
|
|
if enableAuth, ok := data["enable_auth"]; ok {
|
|
v, err := strconv.ParseBool(enableAuth)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, types.BizVo{
|
|
Code: types.InvalidParams,
|
|
Message: "enable_auth must be a bool parameter",
|
|
})
|
|
return
|
|
}
|
|
s.Config.EnableAuth = v
|
|
}
|
|
|
|
// 保存配置文件
|
|
err = utils.SaveConfig(s.Config, s.ConfigPath)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save config file"})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg})
|
|
}
|
|
|
|
// AddToken 添加 Token
|
|
func (s *Server) AddToken(c *gin.Context) {
|
|
var data types.Token
|
|
err := json.NewDecoder(c.Request.Body).Decode(&data)
|
|
if err != nil {
|
|
logger.Errorf("Error decode json data: %s", err.Error())
|
|
c.JSON(http.StatusBadRequest, nil)
|
|
return
|
|
}
|
|
|
|
// 参数处理
|
|
if data.Name == "" || data.MaxCalls < 0 {
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Invalid args"})
|
|
return
|
|
}
|
|
|
|
// 检查当前要添加的 token 是否已经存在
|
|
_, err = GetToken(data.Name)
|
|
if err == nil {
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Token " + data.Name + " already exists"})
|
|
return
|
|
}
|
|
|
|
err = PutToken(types.Token{Name: data.Name, MaxCalls: data.MaxCalls, RemainingCalls: data.MaxCalls})
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save configs"})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: GetTokens()})
|
|
}
|
|
|
|
func (s *Server) SetToken(c *gin.Context) {
|
|
var data types.Token
|
|
err := json.NewDecoder(c.Request.Body).Decode(&data)
|
|
if err != nil {
|
|
logger.Errorf("Error decode json data: %s", err.Error())
|
|
c.JSON(http.StatusBadRequest, nil)
|
|
return
|
|
}
|
|
|
|
logger.Info(data)
|
|
|
|
// 参数处理
|
|
if data.Name == "" || data.MaxCalls < 0 {
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Invalid args"})
|
|
return
|
|
}
|
|
|
|
token, err := GetToken(data.Name)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Token not found"})
|
|
return
|
|
}
|
|
|
|
token.RemainingCalls += data.MaxCalls - token.MaxCalls
|
|
token.MaxCalls = data.MaxCalls
|
|
|
|
err = PutToken(*token)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save configs"})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: token})
|
|
}
|
|
|
|
// RemoveToken 删除 Token
|
|
func (s *Server) RemoveToken(c *gin.Context) {
|
|
var data types.Token
|
|
err := json.NewDecoder(c.Request.Body).Decode(&data)
|
|
if err != nil {
|
|
logger.Errorf("Error decode json data: %s", err.Error())
|
|
c.JSON(http.StatusBadRequest, nil)
|
|
return
|
|
}
|
|
|
|
err = RemoveToken(data.Name)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save configs"})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: GetTokens()})
|
|
}
|
|
|
|
// AddApiKey 添加一个 API key
|
|
func (s *Server) AddApiKey(c *gin.Context) {
|
|
var data map[string]string
|
|
err := json.NewDecoder(c.Request.Body).Decode(&data)
|
|
if err != nil {
|
|
logger.Errorf("Error decode json data: %s", err.Error())
|
|
c.JSON(http.StatusBadRequest, nil)
|
|
return
|
|
}
|
|
if key, ok := data["api_key"]; ok && len(key) > 20 {
|
|
s.Config.Chat.ApiKeys = append(s.Config.Chat.ApiKeys, key)
|
|
}
|
|
|
|
// 保存配置文件
|
|
err = utils.SaveConfig(s.Config, s.ConfigPath)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save config file"})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config.Chat.ApiKeys})
|
|
}
|
|
|
|
// RemoveApiKey 移除一个 API key
|
|
func (s *Server) RemoveApiKey(c *gin.Context) {
|
|
var data map[string]string
|
|
err := json.NewDecoder(c.Request.Body).Decode(&data)
|
|
if err != nil {
|
|
logger.Errorf("Error decode json data: %s", err.Error())
|
|
c.JSON(http.StatusBadRequest, nil)
|
|
return
|
|
}
|
|
|
|
if key, ok := data["api_key"]; ok {
|
|
for i, v := range s.Config.Chat.ApiKeys {
|
|
if v == key {
|
|
s.Config.Chat.ApiKeys = append(s.Config.Chat.ApiKeys[:i], s.Config.Chat.ApiKeys[i+1:]...)
|
|
}
|
|
}
|
|
}
|
|
|
|
// 保存配置文件
|
|
err = utils.SaveConfig(s.Config, s.ConfigPath)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save config file"})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config.Chat.ApiKeys})
|
|
}
|
|
|
|
// ListApiKeys 获取 API key 列表
|
|
func (s *Server) ListApiKeys(c *gin.Context) {
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config.Chat.ApiKeys})
|
|
}
|
|
|
|
func (s *Server) GetChatRoleList(c *gin.Context) {
|
|
var rolesOrder = []string{"gpt", "programmer", "teacher", "artist", "philosopher", "lu-xun", "english_trainer", "seller"}
|
|
var res = make([]interface{}, 0)
|
|
var roles = GetChatRoles()
|
|
for _, k := range rolesOrder {
|
|
if v, ok := roles[k]; ok && v.Enable {
|
|
res = append(res, struct {
|
|
Key string `json:"key"`
|
|
Name string `json:"name"`
|
|
Icon string `json:"icon"`
|
|
}{
|
|
Key: v.Key,
|
|
Name: v.Name,
|
|
Icon: v.Icon,
|
|
})
|
|
}
|
|
}
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: res})
|
|
}
|
|
|
|
// UpdateChatRole 更新某个聊天角色信息，这里只允许更改名称以及启用和禁用角色操作
|
|
func (s *Server) UpdateChatRole(c *gin.Context) {
|
|
var data map[string]string
|
|
err := json.NewDecoder(c.Request.Body).Decode(&data)
|
|
if err != nil {
|
|
logger.Errorf("Error decode json data: %s", err.Error())
|
|
c.JSON(http.StatusBadRequest, nil)
|
|
return
|
|
}
|
|
key := data["key"]
|
|
if key == "" {
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Must specified the role key"})
|
|
return
|
|
}
|
|
|
|
roles := GetChatRoles()
|
|
role := roles[key]
|
|
if role.Key == "" {
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Role key not exists"})
|
|
return
|
|
}
|
|
|
|
if enable, ok := data["enable"]; ok {
|
|
v, err := strconv.ParseBool(enable)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, types.BizVo{
|
|
Code: types.InvalidParams,
|
|
Message: "enable must be a bool parameter",
|
|
})
|
|
return
|
|
}
|
|
role.Enable = v
|
|
}
|
|
|
|
if name, ok := data["name"]; ok {
|
|
role.Name = name
|
|
}
|
|
if helloMsg, ok := data["hello_msg"]; ok {
|
|
role.HelloMsg = helloMsg
|
|
}
|
|
if icon, ok := data["icon"]; ok {
|
|
role.Icon = icon
|
|
}
|
|
|
|
// 保存到 leveldb
|
|
err = PutChatRole(role)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save config"})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: role})
|
|
}
|
|
|
|
// AddProxy 添加一个代理
|
|
func (s *Server) AddProxy(c *gin.Context) {
|
|
var data map[string]string
|
|
err := json.NewDecoder(c.Request.Body).Decode(&data)
|
|
if err != nil {
|
|
logger.Errorf("Error decode json data: %s", err.Error())
|
|
c.JSON(http.StatusBadRequest, nil)
|
|
return
|
|
}
|
|
|
|
if proxy, ok := data["proxy"]; ok {
|
|
if !utils.ContainsStr(s.Config.ProxyURL, proxy) {
|
|
s.Config.ProxyURL = append(s.Config.ProxyURL, proxy)
|
|
}
|
|
}
|
|
|
|
// 保存配置文件
|
|
err = utils.SaveConfig(s.Config, s.ConfigPath)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save config file"})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config.ProxyURL})
|
|
}
|
|
|
|
func (s *Server) RemoveProxy(c *gin.Context) {
|
|
var data map[string]string
|
|
err := json.NewDecoder(c.Request.Body).Decode(&data)
|
|
if err != nil {
|
|
logger.Errorf("Error decode json data: %s", err.Error())
|
|
c.JSON(http.StatusBadRequest, nil)
|
|
return
|
|
}
|
|
|
|
if proxy, ok := data["proxy"]; ok {
|
|
for i, v := range s.Config.ProxyURL {
|
|
if v == proxy {
|
|
s.Config.ProxyURL = append(s.Config.ProxyURL[:i], s.Config.ProxyURL[i+1:]...)
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
// 保存配置文件
|
|
err = utils.SaveConfig(s.Config, s.ConfigPath)
|
|
if err != nil {
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Failed, Message: "Failed to save config file"})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, types.BizVo{Code: types.Success, Message: types.OkMsg, Data: s.Config.ProxyURL})
|
|
}
|