admin/geekai
1
0
Fork 0
mirror of https://gitee.com/blackfox/geekai.git synced 2025-12-08 01:38:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

add 'type' field for ChatModel, support Chat and Image model

Browse Source
This commit is contained in:
RockYang 2024-12-25 18:57:18 +08:00
parent cbf06eea24
commit acee2d9d81
35 changed files with 766 additions and 698 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
CHANGELOG.md
View File

@ -3,11 +3,13 @@
## v4.1.8 ## v4.1.8
- 功能优化：**UI 全新改版，支持主题切换**。 :rocket: :rocket: :rocket: - 功能优化：**UI 全新改版，支持主题切换**。 :rocket: :rocket: :rocket:
- 功能新增：Gitee AI API 接口接入，目前支持 Gitee 的 SD 绘图接口，支持 Gitee 的 AI 对话接口。:rocket: :rocket: :rocket:
- Bug 修复：修复音 Luma API 更新导致任务响应解析失败的错误 - Bug 修复：修复音 Luma API 更新导致任务响应解析失败的错误
- 功能优化：支持 Suno v4.0 模型支持 - 功能优化：支持 Suno v4.0 模型支持
- Bug 修复：修复 Suno 已完成任务删除失败的 错误 - Bug 修复：修复 Suno 已完成任务删除失败的 错误
- 功能新增：支持 OpenAI 实时语音通话功能，目前已经支持按次收费，支持管理员设置每次实时语音通话的算力消耗 - 功能新增：支持 OpenAI 实时语音通话功能，目前已经支持按次收费，支持管理员设置每次实时语音通话的算力消耗
- 功能新增：生成提示词需要消耗算力，支持管理员设置每次生成提示词的算力消耗，防止被白嫖 - 功能新增：生成提示词需要消耗算力，支持管理员设置每次生成提示词的算力消耗，防止被白嫖
- 功能新增：DALL-E-3 绘图支持 Gitee AI Flex 接口
## v4.1.7 ## v4.1.7

14
api/core/app_server.go
View File

@ -15,12 +15,6 @@ import (
"geekai/store/model" "geekai/store/model"
"geekai/utils" "geekai/utils"
"geekai/utils/resp" "geekai/utils/resp"
"github.com/gin-gonic/gin"
"github.com/go-redis/redis/v8"
"github.com/golang-jwt/jwt/v5"
"github.com/nfnt/resize"
"golang.org/x/image/webp"
"gorm.io/gorm"
"image" "image"
"image/jpeg" "image/jpeg"
"io" "io"
@ -29,6 +23,13 @@ import (
"runtime/debug" "runtime/debug"
"strings" "strings"
"time" "time"
"github.com/gin-gonic/gin"
"github.com/go-redis/redis/v8"
"github.com/golang-jwt/jwt/v5"
"github.com/nfnt/resize"
"golang.org/x/image/webp"
"gorm.io/gorm"
) )
type AppServer struct { type AppServer struct {
@ -228,6 +229,7 @@ func needLogin(c *gin.Context) bool {
c.Request.URL.Path == "/api/suno/detail" || c.Request.URL.Path == "/api/suno/detail" ||
c.Request.URL.Path == "/api/suno/play" || c.Request.URL.Path == "/api/suno/play" ||
c.Request.URL.Path == "/api/download" || c.Request.URL.Path == "/api/download" ||
c.Request.URL.Path == "/api/dall/models" ||
strings.HasPrefix(c.Request.URL.Path, "/api/test") || strings.HasPrefix(c.Request.URL.Path, "/api/test") ||
strings.HasPrefix(c.Request.URL.Path, "/api/payment/notify/") || strings.HasPrefix(c.Request.URL.Path, "/api/payment/notify/") ||
strings.HasPrefix(c.Request.URL.Path, "/api/user/clogin") || strings.HasPrefix(c.Request.URL.Path, "/api/user/clogin") ||

4
api/core/types/config.go
View File

@ -133,7 +133,7 @@ type SystemConfig struct {
AdminTitle string `json:"admin_title,omitempty"` // 管理后台标题 AdminTitle string `json:"admin_title,omitempty"` // 管理后台标题
Logo string `json:"logo,omitempty"` // 方形 Logo Logo string `json:"logo,omitempty"` // 方形 Logo
InitPower int `json:"init_power,omitempty"` // 新用户注册赠送算力值 InitPower int `json:"init_power,omitempty"` // 新用户注册赠送算力值
DailyPower int `json:"daily_power,omitempty"` // 每日赠送算力 DailyPower int `json:"daily_power,omitempty"` // 每日签到赠送算力
InvitePower int `json:"invite_power,omitempty"` // 邀请新用户赠送算力值 InvitePower int `json:"invite_power,omitempty"` // 邀请新用户赠送算力值
VipMonthPower int `json:"vip_month_power,omitempty"` // VIP 会员每月赠送的算力值 VipMonthPower int `json:"vip_month_power,omitempty"` // VIP 会员每月赠送的算力值
@ -146,7 +146,7 @@ type SystemConfig struct {
MjPower int `json:"mj_power,omitempty"` // MJ 绘画消耗算力 MjPower int `json:"mj_power,omitempty"` // MJ 绘画消耗算力
MjActionPower int `json:"mj_action_power,omitempty"` // MJ 操作（放大，变换）消耗算力 MjActionPower int `json:"mj_action_power,omitempty"` // MJ 操作（放大，变换）消耗算力
SdPower int `json:"sd_power,omitempty"` // SD 绘画消耗算力 SdPower int `json:"sd_power,omitempty"` // SD 绘画消耗算力
DallPower int `json:"dall_power,omitempty"` // DALL-E-3 绘图消耗算力 DallPower int `json:"dall_power,omitempty"` // DALL-E-3 绘图消耗算力
SunoPower int `json:"suno_power,omitempty"` // Suno 生成歌曲消耗算力 SunoPower int `json:"suno_power,omitempty"` // Suno 生成歌曲消耗算力
LumaPower int `json:"luma_power,omitempty"` // Luma 生成视频消耗算力 LumaPower int `json:"luma_power,omitempty"` // Luma 生成视频消耗算力
AdvanceVoicePower int `json:"advance_voice_power,omitempty"` // 高级语音对话消耗算力 AdvanceVoicePower int `json:"advance_voice_power,omitempty"` // 高级语音对话消耗算力

5
api/core/types/task.go
View File

@ -74,6 +74,8 @@ type SdTaskParams struct {
// DallTask DALL-E task // DallTask DALL-E task
type DallTask struct { type DallTask struct {
ClientId string `json:"client_id"` ClientId string `json:"client_id"`
ModelId uint `json:"model_id"`
ModelName string `json:"model_name"`
Id uint `json:"id"` Id uint `json:"id"`
UserId uint `json:"user_id"` UserId uint `json:"user_id"`
Prompt string `json:"prompt"` Prompt string `json:"prompt"`
@ -81,8 +83,7 @@ type DallTask struct {
Quality string `json:"quality"` Quality string `json:"quality"`
Size string `json:"size"` Size string `json:"size"`
Style string `json:"style"` Style string `json:"style"`
Power int `json:"power"`
Power int `json:"power"`
TranslateModelId int `json:"translate_model_id"` // 提示词翻译模型ID TranslateModelId int `json:"translate_model_id"` // 提示词翻译模型ID
} }

3
api/handler/admin/chat_model_handler.go
View File

@ -43,6 +43,7 @@ func (h *ChatModelHandler) Save(c *gin.Context) {
Temperature float32 `json:"temperature"` // 模型温度 Temperature float32 `json:"temperature"` // 模型温度
KeyId int `json:"key_id,omitempty"` KeyId int `json:"key_id,omitempty"`
CreatedAt int64 `json:"created_at"` CreatedAt int64 `json:"created_at"`
Type string `json:"type"`
} }
if err := c.ShouldBindJSON(&data); err != nil { if err := c.ShouldBindJSON(&data); err != nil {
resp.ERROR(c, types.InvalidArgs) resp.ERROR(c, types.InvalidArgs)
@ -65,7 +66,7 @@ func (h *ChatModelHandler) Save(c *gin.Context) {
item.MaxContext = data.MaxContext item.MaxContext = data.MaxContext
item.Temperature = data.Temperature item.Temperature = data.Temperature
item.KeyId = data.KeyId item.KeyId = data.KeyId
item.Type = data.Type
var res *gorm.DB var res *gorm.DB
if data.Id > 0 { if data.Id > 0 {
res = h.DB.Save(&item) res = h.DB.Save(&item)

2
api/handler/chat_model_handler.go
View File

@ -30,7 +30,7 @@ func NewChatModelHandler(app *core.AppServer, db *gorm.DB) *ChatModelHandler {
func (h *ChatModelHandler) List(c *gin.Context) { func (h *ChatModelHandler) List(c *gin.Context) {
var items []model.ChatModel var items []model.ChatModel
var chatModels = make([]vo.ChatModel, 0) var chatModels = make([]vo.ChatModel, 0)
session := h.DB.Session(&gorm.Session{}).Where("enabled", true) session := h.DB.Session(&gorm.Session{}).Where("type", "chat").Where("enabled", true)
t := c.Query("type") t := c.Query("type")
if t != "" { if t != "" {
session = session.Where("type", t) session = session.Where("type", t)

21
api/handler/openai_handler.go → api/handler/chat_openai_handler.go
View File

@ -17,10 +17,11 @@ import (
"geekai/store/model" "geekai/store/model"
"geekai/store/vo" "geekai/store/vo"
"geekai/utils" "geekai/utils"
req2 "github.com/imroc/req/v3"
"io" "io"
"strings" "strings"
"time" "time"
req2 "github.com/imroc/req/v3"
) )
type Usage struct { type Usage struct {
@ -172,16 +173,22 @@ func (h *ChatHandler) sendOpenAiMessage(
var apiRes types.BizVo var apiRes types.BizVo
r, err := req2.C().R().SetHeader("Body-Type", "application/json"). r, err := req2.C().R().SetHeader("Body-Type", "application/json").
SetHeader("Authorization", function.Token). SetHeader("Authorization", function.Token).
SetBody(params). SetBody(params).Post(function.Action)
SetSuccessResult(&apiRes).Post(function.Action)
errMsg := "" errMsg := ""
if err != nil { if err != nil {
errMsg = err.Error() errMsg = err.Error()
} else if r.IsErrorState() { } else {
errMsg = r.Status all, _ := io.ReadAll(r.Body)
err = json.Unmarshal(all, &apiRes)
if err != nil {
errMsg = err.Error()
} else if apiRes.Code != types.Success {
errMsg = apiRes.Message
}
} }
if errMsg != "" || apiRes.Code != types.Success {
errMsg = "调用函数工具出错：" + apiRes.Message + errMsg if errMsg != "" {
errMsg = "调用函数工具出错：" + errMsg
contents = append(contents, errMsg) contents = append(contents, errMsg)
} else { } else {
errMsg = utils.InterfaceToString(apiRes.Data) errMsg = utils.InterfaceToString(apiRes.Data)

79
api/handler/dalle_handler.go
View File

@ -8,6 +8,7 @@ package handler
// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ // * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
import ( import (
"fmt"
"geekai/core" "geekai/core"
"geekai/core/types" "geekai/core/types"
"geekai/service" "geekai/service"
@ -17,14 +18,13 @@ import (
"geekai/store/vo" "geekai/store/vo"
"geekai/utils" "geekai/utils"
"geekai/utils/resp" "geekai/utils/resp"
"github.com/gin-gonic/gin" "github.com/gin-gonic/gin"
"github.com/go-redis/redis/v8"
"gorm.io/gorm" "gorm.io/gorm"
) )
type DallJobHandler struct { type DallJobHandler struct {
BaseHandler BaseHandler
redis *redis.Client
dallService *dalle.Service dallService *dalle.Service
uploader *oss.UploaderManager uploader *oss.UploaderManager
userService *service.UserService userService *service.UserService
@ -42,49 +42,49 @@ func NewDallJobHandler(app *core.AppServer, db *gorm.DB, service *dalle.Service,
} }
} }
func (h *DallJobHandler) preCheck(c *gin.Context) bool {
user, err := h.GetLoginUser(c)
if err != nil {
resp.NotAuth(c)
return false
}
if user.Power < h.App.SysConfig.DallPower {
resp.ERROR(c, "当前用户剩余算力不足以完成本次绘画！")
return false
}
return true
}
// Image 创建一个绘画任务 // Image 创建一个绘画任务
func (h *DallJobHandler) Image(c *gin.Context) { func (h *DallJobHandler) Image(c *gin.Context) {
if !h.preCheck(c) {
return
}
var data types.DallTask var data types.DallTask
if err := c.ShouldBindJSON(&data); err != nil || data.Prompt == "" { if err := c.ShouldBindJSON(&data); err != nil || data.Prompt == "" {
resp.ERROR(c, types.InvalidArgs) resp.ERROR(c, types.InvalidArgs)
return return
} }
var chatModel model.ChatModel
if res := h.DB.Where("id = ?", data.ModelId).First(&chatModel); res.Error != nil {
resp.ERROR(c, "模型不存在")
return
}
// 检查用户剩余算力
user, err := h.GetLoginUser(c)
if err != nil {
resp.NotAuth(c)
return
}
if user.Power < chatModel.Power {
resp.ERROR(c, "当前用户剩余算力不足以完成本次绘画！")
return
}
idValue, _ := c.Get(types.LoginUserID) idValue, _ := c.Get(types.LoginUserID)
userId := utils.IntValue(utils.InterfaceToString(idValue), 0) userId := utils.IntValue(utils.InterfaceToString(idValue), 0)
task := types.DallTask{ task := types.DallTask{
ClientId: data.ClientId, ClientId: data.ClientId,
UserId: uint(userId), UserId: uint(userId),
ModelId: chatModel.Id,
ModelName: chatModel.Value,
Prompt: data.Prompt, Prompt: data.Prompt,
Quality: data.Quality, Quality: data.Quality,
Size: data.Size, Size: data.Size,
Style: data.Style, Style: data.Style,
Power: h.App.SysConfig.DallPower,
TranslateModelId: h.App.SysConfig.TranslateModelId, TranslateModelId: h.App.SysConfig.TranslateModelId,
Power: chatModel.Power,
} }
job := model.DallJob{ job := model.DallJob{
UserId: uint(userId), UserId: uint(userId),
Prompt: data.Prompt, Prompt: data.Prompt,
Power: task.Power, Power: chatModel.Power,
TaskInfo: utils.JsonEncode(task), TaskInfo: utils.JsonEncode(task),
} }
res := h.DB.Create(&job) res := h.DB.Create(&job)
@ -95,6 +95,17 @@ func (h *DallJobHandler) Image(c *gin.Context) {
task.Id = job.Id task.Id = job.Id
h.dallService.PushTask(task) h.dallService.PushTask(task)
// 扣减算力
err = h.userService.DecreasePower(int(user.Id), chatModel.Power, model.PowerLog{
Type: types.PowerConsume,
Model: chatModel.Value,
Remark: fmt.Sprintf("绘画提示词：%s", utils.CutWords(task.Prompt, 10)),
})
if err != nil {
resp.ERROR(c, "error with decrease power: "+err.Error())
return
}
resp.SUCCESS(c) resp.SUCCESS(c)
} }
@ -210,3 +221,25 @@ func (h *DallJobHandler) Publish(c *gin.Context) {
resp.SUCCESS(c) resp.SUCCESS(c)
} }
func (h *DallJobHandler) GetModels(c *gin.Context) {
var models []model.ChatModel
err := h.DB.Where("type", "img").Where("enabled", true).Find(&models).Error
if err != nil {
resp.ERROR(c, err.Error())
return
}
var modelVos []vo.ChatModel
for _, v := range models {
var modelVo vo.ChatModel
err := utils.CopyObject(v, &modelVo)
if err != nil {
continue
}
modelVo.Id = v.Id
modelVos = append(modelVos, modelVo)
}
resp.SUCCESS(c, modelVos)
}

67
api/handler/function_handler.go
View File

@ -12,6 +12,7 @@ import (
"fmt" "fmt"
"geekai/core" "geekai/core"
"geekai/core/types" "geekai/core/types"
"geekai/service"
"geekai/service/dalle" "geekai/service/dalle"
"geekai/service/oss" "geekai/service/oss"
"geekai/store/model" "geekai/store/model"
@ -32,6 +33,7 @@ type FunctionHandler struct {
config types.ApiConfig config types.ApiConfig
uploadManager *oss.UploaderManager uploadManager *oss.UploaderManager
dallService *dalle.Service dallService *dalle.Service
userService *service.UserService
} }
func NewFunctionHandler( func NewFunctionHandler(
@ -39,7 +41,8 @@ func NewFunctionHandler(
db *gorm.DB, db *gorm.DB,
config *types.AppConfig, config *types.AppConfig,
manager *oss.UploaderManager, manager *oss.UploaderManager,
dallService *dalle.Service) *FunctionHandler { dallService *dalle.Service,
userService *service.UserService) *FunctionHandler {
return &FunctionHandler{ return &FunctionHandler{
BaseHandler: BaseHandler{ BaseHandler: BaseHandler{
App: server, App: server,
@ -48,6 +51,7 @@ func NewFunctionHandler(
config: config.ApiConfig, config: config.ApiConfig,
uploadManager: manager, uploadManager: manager,
dallService: dallService, dallService: dallService,
userService: userService,
} }
} }
@ -152,8 +156,12 @@ func (h *FunctionHandler) ZaoBao(c *gin.Context) {
SetHeader("AppId", h.config.AppId). SetHeader("AppId", h.config.AppId).
SetHeader("Authorization", fmt.Sprintf("Bearer %s", h.config.Token)). SetHeader("Authorization", fmt.Sprintf("Bearer %s", h.config.Token)).
SetSuccessResult(&res).Get(url) SetSuccessResult(&res).Get(url)
if err != nil || r.IsErrorState() { if err != nil {
resp.ERROR(c, fmt.Sprintf("%v%v", err, r.Err)) resp.ERROR(c, fmt.Sprintf("%v", err))
return
}
if r.IsErrorState() {
resp.ERROR(c, fmt.Sprintf("%v", r.Err))
return return
} }
@ -167,7 +175,7 @@ func (h *FunctionHandler) ZaoBao(c *gin.Context) {
for _, v := range res.Data.Items { for _, v := range res.Data.Items {
builder = append(builder, v.Title) builder = append(builder, v.Title)
} }
builder = append(builder, fmt.Sprintf("%s", res.Data.Title)) builder = append(builder, res.Data.Title)
resp.SUCCESS(c, strings.Join(builder, "\n\n")) resp.SUCCESS(c, strings.Join(builder, "\n\n"))
} }
@ -199,33 +207,48 @@ func (h *FunctionHandler) Dall3(c *gin.Context) {
// create dall task // create dall task
prompt := utils.InterfaceToString(params["prompt"]) prompt := utils.InterfaceToString(params["prompt"])
job := model.DallJob{ task := types.DallTask{
UserId: user.Id, UserId: user.Id,
Prompt: prompt, Prompt: prompt,
Power: h.App.SysConfig.DallPower, ModelId: 0,
ModelName: "dall-e-3",
TranslateModelId: h.App.SysConfig.TranslateModelId,
N: 1,
Quality: "standard",
Size: "1024x1024",
Style: "vivid",
Power: h.App.SysConfig.DallPower,
} }
res = h.DB.Create(&job) job := model.DallJob{
UserId: user.Id,
if res.Error != nil { Prompt: prompt,
resp.ERROR(c, "创建 DALL-E 绘图任务失败："+res.Error.Error()) Power: h.App.SysConfig.DallPower,
TaskInfo: utils.JsonEncode(task),
}
err := h.DB.Create(&job).Error
if err != nil {
resp.ERROR(c, "创建 DALL-E 绘图任务失败："+err.Error())
return return
} }
content, err := h.dallService.Image(types.DallTask{ task.Id = job.Id
Id: job.Id, content, err := h.dallService.Image(task, true)
UserId: user.Id,
Prompt: job.Prompt,
N: 1,
Quality: "standard",
Size: "1024x1024",
Style: "vivid",
Power: job.Power,
}, true)
if err != nil { if err != nil {
resp.ERROR(c, "任务执行失败："+err.Error()) resp.ERROR(c, "任务执行失败："+err.Error())
return return
} }
// 扣减算力
err = h.userService.DecreasePower(int(user.Id), job.Power, model.PowerLog{
Type: types.PowerConsume,
Model: task.ModelName,
Remark: fmt.Sprintf("绘画提示词：%s", utils.CutWords(job.Prompt, 10)),
})
if err != nil {
resp.ERROR(c, "扣减算力失败："+err.Error())
return
}
resp.SUCCESS(c, content) resp.SUCCESS(c, content)
} }

1
api/main.go
View File

@ -474,6 +474,7 @@ func main() {
group.GET("imgWall", h.ImgWall) group.GET("imgWall", h.ImgWall)
group.GET("remove", h.Remove) group.GET("remove", h.Remove)
group.GET("publish", h.Publish) group.GET("publish", h.Publish)
group.GET("models", h.GetModels)
}), }),
fx.Provide(handler.NewSunoHandler), fx.Provide(handler.NewSunoHandler),
fx.Invoke(func(s *core.AppServer, h *handler.SunoHandler) { fx.Invoke(func(s *core.AppServer, h *handler.SunoHandler) {

85
api/service/dalle/service.go
View File

@ -8,7 +8,6 @@ package dalle
// * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ // * +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
import ( import (
"errors"
"fmt" "fmt"
"geekai/core/types" "geekai/core/types"
logger2 "geekai/logger" logger2 "geekai/logger"
@ -17,9 +16,11 @@ import (
"geekai/store" "geekai/store"
"geekai/store/model" "geekai/store/model"
"geekai/utils" "geekai/utils"
"github.com/go-redis/redis/v8" "io"
"time" "time"
"github.com/go-redis/redis/v8"
"github.com/imroc/req/v3" "github.com/imroc/req/v3"
"gorm.io/gorm" "gorm.io/gorm"
) )
@ -100,17 +101,18 @@ func (s *Service) Run() {
type imgReq struct { type imgReq struct {
Model string `json:"model"` Model string `json:"model"`
Prompt string `json:"prompt"` Prompt string `json:"prompt"`
N int `json:"n"` N int `json:"n,omitempty"`
Size string `json:"size"` Size string `json:"size,omitempty"`
Quality string `json:"quality"` Quality string `json:"quality,omitempty"`
Style string `json:"style"` Style string `json:"style,omitempty"`
} }
type imgRes struct { type imgRes struct {
Created int64 `json:"created"` Created int64 `json:"created"`
Data []struct { Data []struct {
RevisedPrompt string `json:"revised_prompt"` RevisedPrompt string `json:"revised_prompt,omitempty"`
Url string `json:"url"` Url string `json:"url,omitempty"`
B64Json string `json:"b64_json,omitempty"`
} `json:"data"` } `json:"data"`
} }
@ -135,29 +137,20 @@ func (s *Service) Image(task types.DallTask, sync bool) (string, error) {
} }
} }
var user model.User var chatModel model.ChatModel
s.db.Where("id", task.UserId).First(&user) s.db.Where("id = ?", task.ModelId).First(&chatModel)
if user.Power < task.Power {
return "", errors.New("insufficient of power")
}
// 扣减算力
err := s.userService.DecreasePower(int(user.Id), task.Power, model.PowerLog{
Type: types.PowerConsume,
Model: "dall-e-3",
Remark: fmt.Sprintf("绘画提示词：%s", utils.CutWords(task.Prompt, 10)),
})
if err != nil {
return "", fmt.Errorf("error with decrease power: %v", err)
}
// get image generation API KEY // get image generation API KEY
var apiKey model.ApiKey var apiKey model.ApiKey
err = s.db.Where("type", "dalle"). session := s.db.Where("enabled", true)
Where("enabled", true). if chatModel.KeyId > 0 {
Order("last_used_at ASC").First(&apiKey).Error session = session.Where("id = ?", chatModel.KeyId)
} else {
session = session.Where("type = ?", "dalle")
}
err := session.Order("last_used_at ASC").First(&apiKey).Error
if err != nil { if err != nil {
return "", fmt.Errorf("no available DALL-E api key: %v", err) return "", fmt.Errorf("no available Image Generation api key: %v", err)
} }
var res imgRes var res imgRes
@ -167,7 +160,7 @@ func (s *Service) Image(task types.DallTask, sync bool) (string, error) {
} }
apiURL := fmt.Sprintf("%s/v1/images/generations", apiKey.ApiURL) apiURL := fmt.Sprintf("%s/v1/images/generations", apiKey.ApiURL)
reqBody := imgReq{ reqBody := imgReq{
Model: "dall-e-3", Model: chatModel.Value,
Prompt: prompt, Prompt: prompt,
N: 1, N: 1,
Size: task.Size, Size: task.Size,
@ -182,20 +175,39 @@ func (s *Service) Image(task types.DallTask, sync bool) (string, error) {
SetSuccessResult(&res). SetSuccessResult(&res).
Post(apiURL) Post(apiURL)
if err != nil { if err != nil {
logger.Errorf("error with send request: %v", err)
return "", fmt.Errorf("error with send request: %v", err) return "", fmt.Errorf("error with send request: %v", err)
} }
if r.IsErrorState() { if r.IsErrorState() {
logger.Errorf("error with send request, status: %s, %+v", r.Status, errRes.Error)
return "", fmt.Errorf("error with send request, status: %s, %+v", r.Status, errRes.Error) return "", fmt.Errorf("error with send request, status: %s, %+v", r.Status, errRes.Error)
} }
all, _ := io.ReadAll(r.Body)
logger.Debugf("response: %+v", string(all))
// update the api key last use time // update the api key last use time
s.db.Model(&apiKey).UpdateColumn("last_used_at", time.Now().Unix()) s.db.Model(&apiKey).UpdateColumn("last_used_at", time.Now().Unix())
// update task progress var imgURL string
err = s.db.Model(&model.DallJob{Id: task.Id}).UpdateColumns(map[string]interface{}{ var data = map[string]interface{}{
"progress": 100, "progress": 100,
"org_url": res.Data[0].Url,
"prompt": prompt, "prompt": prompt,
}).Error }
// 如果返回的是base64，则需要上传到oss
if res.Data[0].B64Json != "" {
imgURL, err = s.uploadManager.GetUploadHandler().PutBase64(res.Data[0].B64Json)
if err != nil {
return "", fmt.Errorf("error with upload image: %v", err)
}
logger.Infof("upload image to oss: %s", imgURL)
data["img_url"] = imgURL
} else {
imgURL = res.Data[0].Url
}
data["org_url"] = imgURL
// update task progress
err = s.db.Model(&model.DallJob{Id: task.Id}).UpdateColumns(data).Error
if err != nil { if err != nil {
return "", fmt.Errorf("err with update database: %v", err) return "", fmt.Errorf("err with update database: %v", err)
} }
@ -252,9 +264,14 @@ func (s *Service) CheckTaskStatus() {
// 找出失败的任务，并恢复其扣减算力 // 找出失败的任务，并恢复其扣减算力
s.db.Where("progress", service.FailTaskProgress).Where("power > ?", 0).Find(&jobs) s.db.Where("progress", service.FailTaskProgress).Where("power > ?", 0).Find(&jobs)
for _, job := range jobs { for _, job := range jobs {
err := s.userService.IncreasePower(int(job.UserId), job.Power, model.PowerLog{ var task types.DallTask
err := utils.JsonDecode(job.TaskInfo, &task)
if err != nil {
continue
}
err = s.userService.IncreasePower(int(job.UserId), job.Power, model.PowerLog{
Type: types.PowerRefund, Type: types.PowerRefund,
Model: "dall-e-3", Model: task.ModelName,
Remark: fmt.Sprintf("任务失败，退回算力。任务ID：%d，Err: %s", job.Id, job.ErrMsg), Remark: fmt.Sprintf("任务失败，退回算力。任务ID：%d，Err: %s", job.Id, job.ErrMsg),
}) })
if err != nil { if err != nil {

1
api/store/model/chat_model.go
View File

@ -12,4 +12,5 @@ type ChatModel struct {
MaxContext int // 最大上下文长度 MaxContext int // 最大上下文长度
Temperature float32 // 模型温度 Temperature float32 // 模型温度
KeyId int // 绑定 API KEY ID KeyId int // 绑定 API KEY ID
Type string // 模型类型
} }

1
api/store/vo/chat_model.go
View File

@ -13,4 +13,5 @@ type ChatModel struct {
Temperature float32 `json:"temperature"` // 模型温度 Temperature float32 `json:"temperature"` // 模型温度
KeyId int `json:"key_id,omitempty"` KeyId int `json:"key_id,omitempty"`
KeyName string `json:"key_name"` KeyName string `json:"key_name"`
Type string `json:"type"`
} }

1
database/update-v4.1.8.sql
View File

@ -6,3 +6,4 @@ ALTER TABLE `chatgpt_mj_jobs` CHANGE `err_msg` `err_msg` VARCHAR(1024) CHARACTER
ALTER TABLE `chatgpt_dall_jobs` CHANGE `err_msg` `err_msg` VARCHAR(1024) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NOT NULL COMMENT '错误信息'; ALTER TABLE `chatgpt_dall_jobs` CHANGE `err_msg` `err_msg` VARCHAR(1024) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NOT NULL COMMENT '错误信息';
ALTER TABLE `chatgpt_video_jobs` CHANGE `err_msg` `err_msg` VARCHAR(1024) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NULL DEFAULT NULL COMMENT '错误信息'; ALTER TABLE `chatgpt_video_jobs` CHANGE `err_msg` `err_msg` VARCHAR(1024) CHARACTER SET utf8mb4 COLLATE utf8mb4_0900_ai_ci NULL DEFAULT NULL COMMENT '错误信息';
ALTER TABLE `chatgpt_chat_models` ADD `type` VARCHAR(10) NOT NULL DEFAULT 'chat' COMMENT '模型类型（chat,img）' AFTER `id`;

6
web/package-lock.json generated
View File

@ -22,6 +22,7 @@
"json-bigint": "^1.0.0", "json-bigint": "^1.0.0",
"lodash": "^4.17.21", "lodash": "^4.17.21",
"markdown-it": "^13.0.1", "markdown-it": "^13.0.1",
"markdown-it-emoji": "^2.0.0",
"markdown-it-mathjax3": "^4.3.2", "markdown-it-mathjax3": "^4.3.2",
"markmap-common": "^0.16.0", "markmap-common": "^0.16.0",
"markmap-lib": "^0.16.1", "markmap-lib": "^0.16.1",
@ -8648,6 +8649,11 @@
"markdown-it": "bin/markdown-it.js" "markdown-it": "bin/markdown-it.js"
} }
}, },
"node_modules/markdown-it-emoji": {
"version": "2.0.2",
"resolved": "https://registry.npmmirror.com/markdown-it-emoji/-/markdown-it-emoji-2.0.2.tgz",
"integrity": "sha512-zLftSaNrKuYl0kR5zm4gxXjHaOI3FAOEaloKmRA5hijmJZvSjmxcokOLlzycb/HXlUFWzXqpIEoyEMCE4i9MvQ=="
},
"node_modules/markdown-it-mathjax3": { "node_modules/markdown-it-mathjax3": {
"version": "4.3.2", "version": "4.3.2",
"resolved": "https://registry.npmjs.org/markdown-it-mathjax3/-/markdown-it-mathjax3-4.3.2.tgz", "resolved": "https://registry.npmjs.org/markdown-it-mathjax3/-/markdown-it-mathjax3-4.3.2.tgz",

1
web/package.json
View File

@ -22,6 +22,7 @@
"json-bigint": "^1.0.0", "json-bigint": "^1.0.0",
"lodash": "^4.17.21", "lodash": "^4.17.21",
"markdown-it": "^13.0.1", "markdown-it": "^13.0.1",
"markdown-it-emoji": "^2.0.0",
"markdown-it-mathjax3": "^4.3.2", "markdown-it-mathjax3": "^4.3.2",
"markmap-common": "^0.16.0", "markmap-common": "^0.16.0",
"markmap-lib": "^0.16.1", "markmap-lib": "^0.16.1",

BIN
web/public/images/avatar/gpt.png
View File

Binary file not shown.
Side by Side Swipe Overlay

Before

Width:  |  Height:  |  Size: 7.4 KiB

After

Width:  |  Height:  |  Size: 28 KiB

8
web/src/App.vue
View File

@ -108,6 +108,14 @@ html, body {
text-rendering: optimizeLegibility; text-rendering: optimizeLegibility;
--primary-color: #21aa93 --primary-color: #21aa93
h1 { font-size: 2em; } /* 通常是 2em */
h2 { font-size: 1.5em; } /* 通常是 1.5em */
h3 { font-size: 1.17em; } /* 通常是 1.17em */
h4 { font-size: 1em; } /* 通常是 1em */
h5 { font-size: 0.83em; } /* 通常是 0.83em */
h6 { font-size: 0.67em; } /* 通常是 0.67em */
} }
.el-overlay-dialog { .el-overlay-dialog {

2
web/src/assets/css/chat-plus.styl
View File

@ -5,7 +5,7 @@
.chat-page { .chat-page {
height: 100%; height: 100%;
::v-deep (.el-message-box__message){ :deep (.el-message-box__message){
font-size: 18px !important font-size: 18px !important
} }
.newChat{ .newChat{

2
web/src/assets/css/login.styl
View File

@ -54,7 +54,7 @@
width: 306px; width: 306px;
margin-right: 9px; margin-right: 9px;
} }
::v-deep(.el-tabs__item.is-active, .el-tabs__item:hover){ :deep(.el-tabs__item.is-active, .el-tabs__item:hover){
color: var(--common-text-color) !important; color: var(--common-text-color) !important;
} }
.el-tabs__item{ .el-tabs__item{

7
web/src/assets/css/running-job-list.styl
View File

@ -4,8 +4,13 @@
width 100% width 100%
display flex display flex
flex-flow row flex-flow row
.image-slot {
color var(--theme-text-color-primary)
}
} }
.job-item { .job-item {
margin-right 10px margin-right 10px
width 200px width 200px
@ -31,7 +36,7 @@
span { span {
font-size 20px font-size 20px
color #ffffff color var(--theme-text-color-primary)
} }
} }
} }

5
web/src/assets/css/theme-dark.styl
View File

@ -22,7 +22,7 @@
--border-active:rgba(255, 255, 255, 0.1); --border-active:rgba(255, 255, 255, 0.1);
--card-bg:#252d58; --card-bg:#252d58;
--chat-bg:#1f243f --chat-bg:#1f243f
--chat-wel-bg:#2d2f388a; --chat-wel-bg:#2d2f38;
--card-bg-table: rgba(17, 28, 68, 1); --card-bg-table: rgba(17, 28, 68, 1);
--theme-bg:rgb(13, 20, 53); --theme-bg:rgb(13, 20, 53);
--theme-bg-color: rgb(13, 20, 53); --theme-bg-color: rgb(13, 20, 53);
@ -76,4 +76,7 @@
// 加载动画 // 加载动画
--el-mask-color: rgba(255, 255, 255, 0.5); --el-mask-color: rgba(255, 255, 255, 0.5);
--van-toast-background: rgba(255, 255, 255, 0.3); --van-toast-background: rgba(255, 255, 255, 0.3);
--code-bg-color: #424242;
--code-text-color: #fff;
} }

3
web/src/assets/css/theme-light.styl
View File

@ -45,6 +45,9 @@
// 加载动画 // 加载动画
--el-mask-color: rgba(100, 100, 100, 0.2); --el-mask-color: rgba(100, 100, 100, 0.2);
// code 标签背景
--code-bg-color: #ececec;
--code-text-color: var(--el-color-primary);
} }

2
web/src/components/AccountBg.vue
View File

@ -28,7 +28,7 @@ import FooterBar from "@/components/FooterBar.vue";
position: relative; position: relative;
overflow: hidden; overflow: hidden;
z-index: 1; z-index: 1;
::v-deep(.foot-container){ :deep(.foot-container){
position: absolute; position: absolute;
bottom: 20px; bottom: 20px;
width: 100%; width: 100%;

40
web/src/components/ChatPrompt.vue
View File

@ -17,12 +17,7 @@
</div> </div>
<div class="body"> <div class="body">
<div class="title"> <div class="title">
<el-link <el-link :href="file.url" target="_blank" style="--el-font-weight-primary: bold">{{ file.name }}</el-link>
:href="file.url"
target="_blank"
style="--el-font-weight-primary: bold"
>{{ file.name }}</el-link
>
</div> </div>
<div class="info"> <div class="info">
<span>{{ GetFileType(file.ext) }}</span> <span>{{ GetFileType(file.ext) }}</span>
@ -35,8 +30,7 @@
<div class="content" v-html="content"></div> <div class="content" v-html="content"></div>
<div class="bar" v-if="data.created_at > 0"> <div class="bar" v-if="data.created_at > 0">
<span class="bar-item" <span class="bar-item"
><el-icon><Clock /></el-icon> ><el-icon><Clock /></el-icon> {{ dateFormat(data.created_at) }}</span
{{ dateFormat(data.created_at) }}</span
> >
<span class="bar-item">tokens: {{ finalTokens }}</span> <span class="bar-item">tokens: {{ finalTokens }}</span>
</div> </div>
@ -62,12 +56,7 @@
</div> </div>
<div class="body"> <div class="body">
<div class="title"> <div class="title">
<el-link <el-link :href="file.url" target="_blank" style="--el-font-weight-primary: bold">{{ file.name }}</el-link>
:href="file.url"
target="_blank"
style="--el-font-weight-primary: bold"
>{{ file.name }}</el-link
>
</div> </div>
<div class="info"> <div class="info">
<span>{{ GetFileType(file.ext) }}</span> <span>{{ GetFileType(file.ext) }}</span>
@ -82,8 +71,7 @@
</div> </div>
<div class="bar" v-if="data.created_at > 0"> <div class="bar" v-if="data.created_at > 0">
<span class="bar-item" <span class="bar-item"
><el-icon><Clock /></el-icon> ><el-icon><Clock /></el-icon> {{ dateFormat(data.created_at) }}</span
{{ dateFormat(data.created_at) }}</span
> >
<!-- <span class="bar-item">tokens: {{ finalTokens }}</span>--> <!-- <span class="bar-item">tokens: {{ finalTokens }}</span>-->
</div> </div>
@ -99,16 +87,17 @@ import { httpPost } from "@/utils/http";
import hl from "highlight.js"; import hl from "highlight.js";
import { dateFormat, isImage, processPrompt } from "@/utils/libs"; import { dateFormat, isImage, processPrompt } from "@/utils/libs";
import { FormatFileSize, GetFileIcon, GetFileType } from "@/store/system"; import { FormatFileSize, GetFileIcon, GetFileType } from "@/store/system";
import emoji from "markdown-it-emoji";
import mathjaxPlugin from "markdown-it-mathjax3";
import MarkdownIt from "markdown-it";
const mathjaxPlugin = require("markdown-it-mathjax3"); const md = new MarkdownIt({
const md = require("markdown-it")({
breaks: true, breaks: true,
html: true, html: true,
linkify: true, linkify: true,
typographer: true, typographer: true,
highlight: function (str, lang) { highlight: function (str, lang) {
const codeIndex = const codeIndex = parseInt(Date.now()) + Math.floor(Math.random() * 10000000);
parseInt(Date.now()) + Math.floor(Math.random() * 10000000);
// 显示复制代码按钮 // 显示复制代码按钮
const copyBtn = `<span class="copy-code-btn" data-clipboard-action="copy" data-clipboard-target="#copy-target-${codeIndex}">复制</span> const copyBtn = `<span class="copy-code-btn" data-clipboard-action="copy" data-clipboard-target="#copy-target-${codeIndex}">复制</span>
<textarea style="position: absolute;top: -9999px;left: -9999px;z-index: -9999;" id="copy-target-${codeIndex}">${str.replace( <textarea style="position: absolute;top: -9999px;left: -9999px;z-index: -9999;" id="copy-target-${codeIndex}">${str.replace(
@ -127,9 +116,10 @@ const md = require("markdown-it")({
const preCode = md.utils.escapeHtml(str); const preCode = md.utils.escapeHtml(str);
// 将代码包裹在 pre 中 // 将代码包裹在 pre 中
return `<pre class="code-container"><code class="language-${lang} hljs">${preCode}</code>${copyBtn}</pre>`; return `<pre class="code-container"><code class="language-${lang} hljs">${preCode}</code>${copyBtn}</pre>`;
} },
}); });
md.use(mathjaxPlugin); md.use(mathjaxPlugin);
md.use(emoji);
const props = defineProps({ const props = defineProps({
data: { data: {
type: Object, type: Object,
@ -138,13 +128,13 @@ const props = defineProps({
created_at: "", created_at: "",
tokens: 0, tokens: 0,
model: "", model: "",
icon: "" icon: "",
} },
}, },
listStyle: { listStyle: {
type: String, type: String,
default: "list" default: "list",
} },
}); });
const finalTokens = ref(props.data.tokens); const finalTokens = ref(props.data.tokens);
const content = ref(processPrompt(props.data.content)); const content = ref(processPrompt(props.data.content));

110
web/src/components/ChatReply.vue
View File

@ -6,23 +6,14 @@
</div> </div>
<div class="chat-item"> <div class="chat-item">
<div <div class="content" v-html="md.render(processContent(data.content))"></div>
class="content"
v-html="md.render(processContent(data.content))"
></div>
<div class="bar" v-if="data.created_at"> <div class="bar" v-if="data.created_at">
<span class="bar-item" <span class="bar-item"
><el-icon><Clock /></el-icon> ><el-icon><Clock /></el-icon> {{ dateFormat(data.created_at) }}</span
{{ dateFormat(data.created_at) }}</span
> >
<span class="bar-item">tokens: {{ data.tokens }}</span> <span class="bar-item">tokens: {{ data.tokens }}</span>
<span class="bar-item"> <span class="bar-item">
<el-tooltip <el-tooltip class="box-item" effect="dark" content="复制回答" placement="bottom">
class="box-item"
effect="dark"
content="复制回答"
placement="bottom"
>
<el-icon class="copy-reply" :data-clipboard-text="data.content"> <el-icon class="copy-reply" :data-clipboard-text="data.content">
<DocumentCopy /> <DocumentCopy />
</el-icon> </el-icon>
@ -30,23 +21,13 @@
</span> </span>
<span v-if="!readOnly"> <span v-if="!readOnly">
<span class="bar-item" @click="reGenerate(data.prompt)"> <span class="bar-item" @click="reGenerate(data.prompt)">
<el-tooltip <el-tooltip class="box-item" effect="dark" content="重新生成" placement="bottom">
class="box-item"
effect="dark"
content="重新生成"
placement="bottom"
>
<el-icon><Refresh /></el-icon> <el-icon><Refresh /></el-icon>
</el-tooltip> </el-tooltip>
</span> </span>
<span class="bar-item" @click="synthesis(data.content)"> <span class="bar-item" @click="synthesis(data.content)">
<el-tooltip <el-tooltip class="box-item" effect="dark" content="生成语音朗读" placement="bottom">
class="box-item"
effect="dark"
content="生成语音朗读"
placement="bottom"
>
<i class="iconfont icon-speaker"></i> <i class="iconfont icon-speaker"></i>
</el-tooltip> </el-tooltip>
</span> </span>
@ -75,24 +56,15 @@
</div> </div>
<div class="chat-item"> <div class="chat-item">
<div class="content-wrapper"> <div class="content-wrapper">
<div <div class="content" v-html="md.render(processContent(data.content))"></div>
class="content"
v-html="md.render(processContent(data.content))"
></div>
</div> </div>
<div class="bar" v-if="data.created_at"> <div class="bar" v-if="data.created_at">
<span class="bar-item" <span class="bar-item"
><el-icon><Clock /></el-icon> ><el-icon><Clock /></el-icon> {{ dateFormat(data.created_at) }}</span
{{ dateFormat(data.created_at) }}</span
> >
<!-- <span class="bar-item">tokens: {{ data.tokens }}</span>--> <!-- <span class="bar-item">tokens: {{ data.tokens }}</span>-->
<span class="bar-item bg"> <span class="bar-item bg">
<el-tooltip <el-tooltip class="box-item" effect="dark" content="复制回答" placement="bottom">
class="box-item"
effect="dark"
content="复制回答"
placement="bottom"
>
<el-icon class="copy-reply" :data-clipboard-text="data.content"> <el-icon class="copy-reply" :data-clipboard-text="data.content">
<DocumentCopy /> <DocumentCopy />
</el-icon> </el-icon>
@ -100,23 +72,13 @@
</span> </span>
<span v-if="!readOnly"> <span v-if="!readOnly">
<span class="bar-item bg" @click="reGenerate(data.prompt)"> <span class="bar-item bg" @click="reGenerate(data.prompt)">
<el-tooltip <el-tooltip class="box-item" effect="dark" content="重新生成" placement="bottom">
class="box-item"
effect="dark"
content="重新生成"
placement="bottom"
>
<el-icon><Refresh /></el-icon> <el-icon><Refresh /></el-icon>
</el-tooltip> </el-tooltip>
</span> </span>
<span class="bar-item bg" @click="synthesis(data.content)"> <span class="bar-item bg" @click="synthesis(data.content)">
<el-tooltip <el-tooltip class="box-item" effect="dark" content="生成语音朗读" placement="bottom">
class="box-item"
effect="dark"
content="生成语音朗读"
placement="bottom"
>
<i class="iconfont icon-speaker"></i> <i class="iconfont icon-speaker"></i>
</el-tooltip> </el-tooltip>
</span> </span>
@ -132,6 +94,10 @@ import { Clock, DocumentCopy, Refresh } from "@element-plus/icons-vue";
import { ElMessage } from "element-plus"; import { ElMessage } from "element-plus";
import { dateFormat, processContent } from "@/utils/libs"; import { dateFormat, processContent } from "@/utils/libs";
import hl from "highlight.js"; import hl from "highlight.js";
import emoji from "markdown-it-emoji";
import mathjaxPlugin from "markdown-it-mathjax3";
import MarkdownIt from "markdown-it";
// eslint-disable-next-line no-undef,no-unused-vars // eslint-disable-next-line no-undef,no-unused-vars
const props = defineProps({ const props = defineProps({
data: { data: {
@ -140,28 +106,26 @@ const props = defineProps({
icon: "", icon: "",
content: "", content: "",
created_at: "", created_at: "",
tokens: 0 tokens: 0,
} },
}, },
readOnly: { readOnly: {
type: Boolean, type: Boolean,
default: false default: false,
}, },
listStyle: { listStyle: {
type: String, type: String,
default: "list" default: "list",
} },
}); });
const mathjaxPlugin = require("markdown-it-mathjax3"); const md = new MarkdownIt({
const md = require("markdown-it")({
breaks: true, breaks: true,
html: true, html: true,
linkify: true, linkify: true,
typographer: true, typographer: true,
highlight: function (str, lang) { highlight: function (str, lang) {
const codeIndex = const codeIndex = parseInt(Date.now()) + Math.floor(Math.random() * 10000000);
parseInt(Date.now()) + Math.floor(Math.random() * 10000000);
// 显示复制代码按钮 // 显示复制代码按钮
const copyBtn = `<span class="copy-code-btn" data-clipboard-action="copy" data-clipboard-target="#copy-target-${codeIndex}">复制</span> const copyBtn = `<span class="copy-code-btn" data-clipboard-action="copy" data-clipboard-target="#copy-target-${codeIndex}">复制</span>
<textarea style="position: absolute;top: -9999px;left: -9999px;z-index: -9999;" id="copy-target-${codeIndex}">${str.replace( <textarea style="position: absolute;top: -9999px;left: -9999px;z-index: -9999;" id="copy-target-${codeIndex}">${str.replace(
@ -171,7 +135,7 @@ const md = require("markdown-it")({
if (lang && hl.getLanguage(lang)) { if (lang && hl.getLanguage(lang)) {
const langHtml = `<span class="lang-name">${lang}</span>`; const langHtml = `<span class="lang-name">${lang}</span>`;
// 处理代码高亮 // 处理代码高亮
const preCode = hl.highlight(lang, str, true).value; const preCode = hl.highlight(str, { language: lang }).value;
// 将代码包裹在 pre 中 // 将代码包裹在 pre 中
return `<pre class="code-container"><code class="language-${lang} hljs">${preCode}</code>${copyBtn} ${langHtml}</pre>`; return `<pre class="code-container"><code class="language-${lang} hljs">${preCode}</code>${copyBtn} ${langHtml}</pre>`;
} }
@ -180,10 +144,10 @@ const md = require("markdown-it")({
const preCode = md.utils.escapeHtml(str); const preCode = md.utils.escapeHtml(str);
// 将代码包裹在 pre 中 // 将代码包裹在 pre 中
return `<pre class="code-container"><code class="language-${lang} hljs">${preCode}</code>${copyBtn}</pre>`; return `<pre class="code-container"><code class="language-${lang} hljs">${preCode}</code>${copyBtn}</pre>`;
} },
}); });
md.use(mathjaxPlugin); md.use(mathjaxPlugin);
md.use(emoji);
const emits = defineEmits(["regen"]); const emits = defineEmits(["regen"]);
if (!props.data.icon) { if (!props.data.icon) {
@ -205,6 +169,9 @@ const reGenerate = (prompt) => {
<style lang="stylus"> <style lang="stylus">
@import '@/assets/css/markdown/vue.css'; @import '@/assets/css/markdown/vue.css';
.chat-page,.chat-export { .chat-page,.chat-export {
--font-family: Menlo,"微软雅黑","Roboto Mono","Courier New",Courier,monospace,"Inter",sans-serif;
font-family: var(--font-family);
.chat-line-reply-list { .chat-line-reply-list {
justify-content: center; justify-content: center;
background-color: var(--chat-list-bg); background-color: var(--chat-list-bg);
@ -255,12 +222,13 @@ const reGenerate = (prompt) => {
line-height 1.5 line-height 1.5
code { code {
// color:var(--theme-text-color-primary); color:var(--theme-text-color-primary);
color:#fff font-weight 600
// color:#fff
// background-color var(--el-color-primary-light-3) // background-color var(--el-color-primary-light-3)
background-color: var(--el-color-primary); // background-color: var(--el-color-primary);
padding 3px 5px; // padding 3px 5px;
border-radius 5px; // border-radius 5px;
} }
} }
@ -432,12 +400,12 @@ const reGenerate = (prompt) => {
line-height 1.5 line-height 1.5
code { code {
color:#fff; color:var(--code-text-color);
background-color var( --el-color-primary) font-weight bold
padding 0 3px; font-family: var(--font-family);
border-radius 5px; background-color: var(--code-bg-color);
font-size: 16px; border-radius: 4px;
padding: 5px 7px; padding: .2rem .4rem;
} }
} }

13
web/src/components/TaskList.vue
View File

@ -1,15 +1,10 @@
<template> <template>
<div class="running-job-list"> <div class="running-job-list pt-4 pb-4">
<div class="running-job-box" v-if="list.length > 0"> <div class="running-job-box" v-if="list.length > 0">
<div class="job-item" v-for="item in list" :key="item.id"> <div class="job-item" v-for="item in list" :key="item.id">
<div v-if="item.progress > 0" class="job-item-inner"> <div v-if="item.progress > 0" class="job-item-inner">
<div class="progress" v-if="item.progress > 0"> <div class="progress" v-if="item.progress > 0">
<el-progress <el-progress type="circle" :percentage="item.progress" :width="100" color="#47fff1" />
type="circle"
:percentage="item.progress"
:width="100"
color="#47fff1"
/>
</div> </div>
</div> </div>
<el-image fit="cover" v-else> <el-image fit="cover" v-else>
@ -33,8 +28,8 @@ import nodata from "@/assets/img/no-data.png";
const props = defineProps({ const props = defineProps({
list: { list: {
type: Array, type: Array,
default: [] default: [],
} },
}); });
</script> </script>

182
web/src/components/admin/AdminSidebar.vue
View File

@ -1,40 +1,36 @@
<template> <template>
<div :class="'sidebar '+theme"> <div :class="'sidebar ' + theme">
<a class="logo w-full" href="/" target="_blank"> <a class="logo w-full" href="/" target="_blank">
<el-image :src="logo"/> <el-image :src="logo" />
<span class="text" v-show="!sidebar.collapse">{{ title }}</span> <span class="text" v-show="!sidebar.collapse">{{ title }}</span>
</a> </a>
<el-menu <el-menu
class="sidebar-el-menu" class="sidebar-el-menu"
:default-active="onRoutes" :default-active="onRoutes"
:collapse="sidebar.collapse" :collapse="sidebar.collapse"
background-color="#324157" background-color="#324157"
text-color="#bfcbd9" text-color="#bfcbd9"
active-text-color="#20a0ff" active-text-color="#20a0ff"
unique-opened unique-opened
router router
> >
<template v-for="item in items"> <template v-for="item in items">
<template v-if="item.subs"> <template v-if="item.subs">
<el-sub-menu :index="item.index" :key="item.index"> <el-sub-menu :index="item.index" :key="item.index">
<template #title> <template #title>
<i :class="'iconfont icon-'+item.icon"></i> <i :class="'iconfont icon-' + item.icon"></i>
<span>{{ item.title }}</span> <span>{{ item.title }}</span>
</template> </template>
<template v-for="subItem in item.subs"> <template v-for="subItem in item.subs">
<el-sub-menu <el-sub-menu v-if="subItem.subs" :index="subItem.index" :key="subItem.index">
v-if="subItem.subs"
:index="subItem.index"
:key="subItem.index"
>
<template #title>{{ subItem.title }}</template> <template #title>{{ subItem.title }}</template>
<el-menu-item v-for="(threeItem, i) in subItem.subs" :key="i" :index="threeItem.index"> <el-menu-item v-for="(threeItem, i) in subItem.subs" :key="i" :index="threeItem.index">
{{ threeItem.title }} {{ threeItem.title }}
</el-menu-item> </el-menu-item>
</el-sub-menu> </el-sub-menu>
<el-menu-item v-else :index="subItem.index" :key="subItem.index"> <el-menu-item v-else :index="subItem.index" :key="subItem.index">
<i v-if="subItem.icon" :class="'iconfont icon-'+subItem.icon"></i> <i v-if="subItem.icon" :class="'iconfont icon-' + subItem.icon"></i>
{{ subItem.title }} {{ subItem.title }}
</el-menu-item> </el-menu-item>
</template> </template>
@ -42,7 +38,7 @@
</template> </template>
<template v-else> <template v-else>
<el-menu-item :index="item.index" :key="item.index"> <el-menu-item :index="item.index" :key="item.index">
<i :class="'iconfont icon-'+item.icon"></i> <i :class="'iconfont icon-' + item.icon"></i>
<template #title>{{ item.title }}</template> <template #title>{{ item.title }}</template>
</el-menu-item> </el-menu-item>
</template> </template>
@ -52,120 +48,125 @@
</template> </template>
<script setup> <script setup>
import {computed, ref, watch} from 'vue'; import { computed, ref, watch } from "vue";
import {setMenuItems, useSidebarStore} from '@/store/sidebar'; import { setMenuItems, useSidebarStore } from "@/store/sidebar";
import {httpGet} from "@/utils/http"; import { httpGet } from "@/utils/http";
import {ElMessage} from "element-plus"; import { ElMessage } from "element-plus";
import {useRoute} from "vue-router"; import { useRoute } from "vue-router";
import {useSharedStore} from "@/store/sharedata"; import { useSharedStore } from "@/store/sharedata";
const title = ref('') const title = ref("");
const logo = ref('') const logo = ref("");
// 加载系统配置 // 加载系统配置
httpGet('/api/admin/config/get?key=system').then(res => { httpGet("/api/admin/config/get?key=system")
title.value = res.data.admin_title .then((res) => {
logo.value = res.data.logo title.value = res.data.admin_title;
}).catch(e => { logo.value = res.data.logo;
ElMessage.error("加载系统配置失败: " + e.message) })
}) .catch((e) => {
const store = useSharedStore() ElMessage.error("加载系统配置失败: " + e.message);
const theme = ref(store.theme) });
watch(() => store.theme, (val) => { const store = useSharedStore();
theme.value = val const theme = ref(store.theme);
}) watch(
() => store.theme,
(val) => {
theme.value = val;
}
);
const items = [ const items = [
{ {
icon: 'home', icon: "home",
index: '/admin/dashboard', index: "/admin/dashboard",
title: '仪表盘', title: "仪表盘",
}, },
{ {
icon: 'user-fill', icon: "user-fill",
index: '/admin/user', index: "/admin/user",
title: '用户管理', title: "用户管理",
}, },
{ {
icon: 'menu', icon: "menu",
index: '1', index: "1",
title: '应用管理', title: "应用管理",
subs: [ subs: [
{ {
index: '/admin/app', index: "/admin/app",
title: '应用列表', title: "应用列表",
}, },
{ {
index: '/admin/app/type', index: "/admin/app/type",
title: '应用分类', title: "应用分类",
}, },
], ],
}, },
{ {
icon: 'api-key', icon: "api-key",
index: '/admin/apikey', index: "/admin/apikey",
title: 'API-KEY', title: "API-KEY",
}, },
{ {
icon: 'model', icon: "model",
index: '/admin/chat/model', index: "/admin/chat/model",
title: '语言模型', title: "模型管理",
}, },
{ {
icon: 'recharge', icon: "recharge",
index: '/admin/product', index: "/admin/product",
title: '充值产品', title: "充值产品",
}, },
{ {
icon: 'order', icon: "order",
index: '/admin/order', index: "/admin/order",
title: '充值订单', title: "充值订单",
}, },
{ {
icon: 'reward', icon: "reward",
index: '/admin/redeem', index: "/admin/redeem",
title: '兑换码', title: "兑换码",
}, },
{ {
icon: 'control', icon: "control",
index: '/admin/functions', index: "/admin/functions",
title: '函数管理', title: "函数管理",
}, },
{ {
icon: 'prompt', icon: "prompt",
index: '/admin/chats', index: "/admin/chats",
title: '对话管理', title: "对话管理",
}, },
{ {
icon: 'image', icon: "image",
index: '/admin/images', index: "/admin/images",
title: '绘图管理', title: "绘图管理",
}, },
{ {
icon: 'mp3', icon: "mp3",
index: '/admin/medias', index: "/admin/medias",
title: '音视频管理', title: "音视频管理",
}, },
{ {
icon: 'role', icon: "role",
index: '/admin/manger', index: "/admin/manger",
title: '管理员', title: "管理员",
}, },
{ {
icon: 'config', icon: "config",
index: '/admin/system', index: "/admin/system",
title: '系统设置', title: "系统设置",
}, },
{ {
icon: 'log', icon: "log",
index: '/admin/powerLog', index: "/admin/powerLog",
title: '用户算力日志', title: "用户算力日志",
}, },
{ {
icon: 'log', icon: "log",
index: '/admin/loginLog', index: "/admin/loginLog",
title: '用户登录日志', title: "用户登录日志",
}, },
// { // {
// icon: 'menu', // icon: 'menu',
@ -198,7 +199,7 @@ const onRoutes = computed(() => {
}); });
const sidebar = useSidebarStore(); const sidebar = useSidebarStore();
setMenuItems(items) setMenuItems(items);
</script> </script>
<style scoped lang="stylus"> <style scoped lang="stylus">
@ -291,5 +292,4 @@ setMenuItems(items)
border-color var(--el-border-color) border-color var(--el-border-color)
} }
} }
</style> </style>

121
web/src/views/ChatExport.vue
View File

@ -1,87 +1,90 @@
<template> <template>
<div class="chat-export" v-loading="loading"> <div class="chat-export" v-loading="loading">
<div class="chat-box" id="chat-box"> <div class="chat-box" id="chat-box">
<div class="title"> <div class="title pt-4">
<h2>{{ chatTitle }}</h2> <h2>{{ chatTitle }}</h2>
</div> </div>
<div v-for="item in chatData" :key="item.id"> <div v-for="item in chatData" :key="item.id">
<chat-prompt v-if="item.type==='prompt'" :data="item" list-style="list"/> <chat-prompt v-if="item.type === 'prompt'" :data="item" list-style="list" />
<chat-reply v-else-if="item.type==='reply'" :data="item" :read-only="true" list-style="list"/> <chat-reply v-else-if="item.type === 'reply'" :data="item" :read-only="true" list-style="list" />
</div> </div>
</div><!-- end chat box --> </div>
<!-- end chat box -->
</div> </div>
</template> </template>
<script setup> <script setup>
import ChatReply from "@/components/ChatReply.vue"; import ChatReply from "@/components/ChatReply.vue";
import ChatPrompt from "@/components/ChatPrompt.vue"; import ChatPrompt from "@/components/ChatPrompt.vue";
import {nextTick, onMounted, ref} from "vue"; import { nextTick, onMounted, ref } from "vue";
import {useRouter} from "vue-router"; import { useRouter } from "vue-router";
import {httpGet} from "@/utils/http"; import { httpGet } from "@/utils/http";
import 'highlight.js/styles/a11y-dark.css' import "highlight.js/styles/a11y-dark.css";
import hl from "highlight.js"; import hl from "highlight.js";
import {ElMessage} from "element-plus"; import { ElMessage } from "element-plus";
import Clipboard from "clipboard"; import Clipboard from "clipboard";
const chatData = ref([]) const chatData = ref([]);
const router = useRouter() const router = useRouter();
const chatId = router.currentRoute.value.query['chat_id'] const chatId = router.currentRoute.value.query["chat_id"];
const loading = ref(true) const loading = ref(true);
const chatTitle = ref('') const chatTitle = ref("");
httpGet('/api/chat/history?chat_id=' + chatId).then(res => { httpGet("/api/chat/history?chat_id=" + chatId)
const data = res.data .then((res) => {
if (!data) { const data = res.data;
loading.value = false if (!data) {
return loading.value = false;
} return;
const md = require('markdown-it')({breaks: true});
for (let i = 0; i < data.length; i++) {
if (data[i].type === "prompt") {
chatData.value.push(data[i]);
continue;
} else if (data[i].type === "mj") {
data[i].content = JSON.parse(data[i].content)
data[i].content.content = md.render(data[i].content?.content)
chatData.value.push(data[i]);
continue;
} }
data[i].orgContent = data[i].content; for (let i = 0; i < data.length; i++) {
data[i].content = md.render(data[i].content); if (data[i].type === "prompt") {
chatData.value.push(data[i]); chatData.value.push(data[i]);
} continue;
} else if (data[i].type === "mj") {
data[i].content = JSON.parse(data[i].content);
data[i].content.content = data[i].content?.content;
chatData.value.push(data[i]);
continue;
}
nextTick(() => { data[i].orgContent = data[i].content;
hl.configure({ignoreUnescapedHTML: true}) data[i].content = data[i].content;
const blocks = document.querySelector("#chat-box").querySelectorAll('pre code'); chatData.value.push(data[i]);
blocks.forEach((block) => { }
hl.highlightElement(block)
}) nextTick(() => {
hl.configure({ ignoreUnescapedHTML: true });
const blocks = document.querySelector("#chat-box").querySelectorAll("pre code");
blocks.forEach((block) => {
hl.highlightElement(block);
});
});
loading.value = false;
}) })
loading.value = false .catch((e) => {
}).catch(e => { ElMessage.error("加载聊天记录失败：" + e.message);
ElMessage.error('加载聊天记录失败：' + e.message); });
})
httpGet('/api/chat/detail?chat_id=' + chatId).then(res => { httpGet("/api/chat/detail?chat_id=" + chatId)
chatTitle.value = res.data.title .then((res) => {
}).catch(e => { chatTitle.value = res.data.title;
ElMessage.error("加载会失败： " + e.message) })
}) .catch((e) => {
ElMessage.error("加载会失败： " + e.message);
});
onMounted(() => { onMounted(() => {
const clipboard = new Clipboard('.copy-reply'); const clipboard = new Clipboard(".copy-reply");
clipboard.on('success', () => { clipboard.on("success", () => {
ElMessage.success('复制成功！'); ElMessage.success("复制成功！");
}) });
clipboard.on('error', () => { clipboard.on("error", () => {
ElMessage.error('复制失败！'); ElMessage.error("复制失败！");
}) });
}) });
</script> </script>
<style lang="stylus"> <style lang="stylus">
.chat-export { .chat-export {

27
web/src/views/ChatPlus.vue
View File

@ -224,7 +224,7 @@
</el-container> </el-container>
<el-dialog v-model="showNotice" :show-close="true" class="notice-dialog" title="网站公告"> <el-dialog v-model="showNotice" :show-close="true" class="notice-dialog" title="网站公告">
<div class="notice"> <div class="notice p-4">
<div v-html="notice"></div> <div v-html="notice"></div>
</div> </div>
@ -262,24 +262,24 @@
</div> </div>
</template> </template>
<script setup> <script setup>
import {nextTick, onMounted, onUnmounted, ref, watch} from "vue"; import { nextTick, onMounted, onUnmounted, ref, watch } from "vue";
import ChatPrompt from "@/components/ChatPrompt.vue"; import ChatPrompt from "@/components/ChatPrompt.vue";
import ChatReply from "@/components/ChatReply.vue"; import ChatReply from "@/components/ChatReply.vue";
import {Delete, Edit, InfoFilled, More, Plus, Promotion, Search, Share, VideoPause} from "@element-plus/icons-vue"; import { Delete, Edit, InfoFilled, More, Plus, Promotion, Search, Share, VideoPause } from "@element-plus/icons-vue";
import "highlight.js/styles/a11y-dark.css"; import "highlight.js/styles/a11y-dark.css";
import {isMobile, randString, removeArrayItem, UUID} from "@/utils/libs"; import { isMobile, randString, removeArrayItem, UUID } from "@/utils/libs";
import {ElMessage, ElMessageBox} from "element-plus"; import { ElMessage, ElMessageBox } from "element-plus";
import {httpGet, httpPost} from "@/utils/http"; import { httpGet, httpPost } from "@/utils/http";
import {useRouter} from "vue-router"; import { useRouter } from "vue-router";
import Clipboard from "clipboard"; import Clipboard from "clipboard";
import {checkSession, getClientId, getSystemInfo} from "@/store/cache"; import { checkSession, getClientId, getSystemInfo } from "@/store/cache";
import Welcome from "@/components/Welcome.vue"; import Welcome from "@/components/Welcome.vue";
import {useSharedStore} from "@/store/sharedata"; import { useSharedStore } from "@/store/sharedata";
import FileSelect from "@/components/FileSelect.vue"; import FileSelect from "@/components/FileSelect.vue";
import FileList from "@/components/FileList.vue"; import FileList from "@/components/FileList.vue";
import ChatSetting from "@/components/ChatSetting.vue"; import ChatSetting from "@/components/ChatSetting.vue";
import BackTop from "@/components/BackTop.vue"; import BackTop from "@/components/BackTop.vue";
import {closeLoading, showLoading, showMessageError} from "@/utils/dialog"; import { closeLoading, showLoading, showMessageError } from "@/utils/dialog";
const title = ref("GeekAI-智能助手"); const title = ref("GeekAI-智能助手");
const models = ref([]); const models = ref([]);
@ -362,12 +362,15 @@ getSystemInfo()
ElMessage.error("获取系统配置失败：" + e.message); ElMessage.error("获取系统配置失败：" + e.message);
}); });
const md = require("markdown-it")({ import MarkdownIt from "markdown-it";
import emoji from "markdown-it-emoji";
const md = new MarkdownIt({
breaks: true, breaks: true,
html: true, html: true,
linkify: true, linkify: true,
typographer: true, typographer: true,
}); }).use(emoji);
// 获取系统公告 // 获取系统公告
httpGet("/api/config/get?key=notice") httpGet("/api/config/get?key=notice")
.then((res) => { .then((res) => {

50
web/src/views/Dalle.vue
View File

@ -7,7 +7,19 @@
<div class="sd-params"> <div class="sd-params">
<el-form :model="params" label-width="80px" label-position="left"> <el-form :model="params" label-width="80px" label-position="left">
<div class="param-line" style="padding-top: 10px"> <div class="param-line pt-1">
<el-form-item label="生图模型">
<template #default>
<div class="form-item-inner">
<el-select v-model="selectedModel" style="width: 150px" placeholder="请选择模型" @change="changeModel">
<el-option v-for="v in models" :label="v.name" :value="v" :key="v.value" />
</el-select>
</div>
</template>
</el-form-item>
</div>
<div class="param-line">
<el-form-item label="图片质量"> <el-form-item label="图片质量">
<template #default> <template #default>
<div class="form-item-inner"> <div class="form-item-inner">
@ -222,7 +234,7 @@ import { checkSession, getClientId, getSystemInfo } from "@/store/cache";
import { useSharedStore } from "@/store/sharedata"; import { useSharedStore } from "@/store/sharedata";
import TaskList from "@/components/TaskList.vue"; import TaskList from "@/components/TaskList.vue";
import BackTop from "@/components/BackTop.vue"; import BackTop from "@/components/BackTop.vue";
import { showMessageError } from "@/utils/dialog"; import { showMessageError, showMessageOK } from "@/utils/dialog";
const listBoxHeight = ref(0); const listBoxHeight = ref(0);
// const paramBoxHeight = ref(0) // const paramBoxHeight = ref(0)
@ -232,6 +244,7 @@ const colWidth = ref(220);
const isOver = ref(false); const isOver = ref(false);
const previewURL = ref(""); const previewURL = ref("");
const store = useSharedStore(); const store = useSharedStore();
const models = ref([]);
const resizeElement = function () { const resizeElement = function () {
listBoxHeight.value = window.innerHeight - 58; listBoxHeight.value = window.innerHeight - 58;
@ -245,7 +258,9 @@ const qualities = [
{ name: "标准", value: "standard" }, { name: "标准", value: "standard" },
{ name: "高清", value: "hd" }, { name: "高清", value: "hd" },
]; ];
const sizes = ["1024x1024", "1792x1024", "1024x1792"]; const dalleSizes = ["1024x1024", "1792x1024", "1024x1792"];
const fluxSizes = ["1024x1024", "1024x768", "768x1024", "1280x960", "960x1280", "1366x768", "768x1366"];
const sizes = ref(dalleSizes);
const styles = [ const styles = [
{ name: "生动", value: "vivid" }, { name: "生动", value: "vivid" },
{ name: "自然", value: "natural" }, { name: "自然", value: "natural" },
@ -264,15 +279,17 @@ const power = ref(0);
const dallPower = ref(0); // 画一张 SD 图片消耗算力 const dallPower = ref(0); // 画一张 SD 图片消耗算力
const clipboard = ref(null); const clipboard = ref(null);
const userId = ref(0); const userId = ref(0);
const selectedModel = ref(null);
onMounted(() => { onMounted(() => {
initData(); initData();
clipboard.value = new Clipboard(".copy-prompt"); clipboard.value = new Clipboard(".copy-prompt");
clipboard.value.on("success", () => { clipboard.value.on("success", () => {
ElMessage.success("复制成功！"); showMessageOK("复制成功！");
}); });
clipboard.value.on("error", () => { clipboard.value.on("error", () => {
ElMessage.error("复制失败！"); showMessageError("复制失败！");
}); });
getSystemInfo() getSystemInfo()
@ -280,7 +297,7 @@ onMounted(() => {
dallPower.value = res.data["dall_power"]; dallPower.value = res.data["dall_power"];
}) })
.catch((e) => { .catch((e) => {
ElMessage.error("获取系统配置失败：" + e.message); showMessageError("获取系统配置失败：" + e.message);
}); });
store.addMessageHandler("dall", (data) => { store.addMessageHandler("dall", (data) => {
@ -296,6 +313,15 @@ onMounted(() => {
} }
nextTick(() => fetchRunningJobs()); nextTick(() => fetchRunningJobs());
}); });
// 获取模型列表
httpGet("/api/dall/models")
.then((res) => {
models.value = res.data;
})
.catch((e) => {
showMessageError("获取模型列表失败：" + e.message);
});
}); });
onUnmounted(() => { onUnmounted(() => {
@ -367,6 +393,9 @@ const fetchFinishJobs = () => {
// 创建绘图任务 // 创建绘图任务
const promptRef = ref(null); const promptRef = ref(null);
const generate = () => { const generate = () => {
if (!params.value.model_id) {
return ElMessage.error("请选择生图模型！");
}
if (params.value.prompt === "") { if (params.value.prompt === "") {
promptRef.value.focus(); promptRef.value.focus();
return ElMessage.error("请输入绘画提示词！"); return ElMessage.error("请输入绘画提示词！");
@ -446,6 +475,15 @@ const generatePrompt = () => {
isGenerating.value = false; isGenerating.value = false;
}); });
}; };
const changeModel = (model) => {
if (model.value.startsWith("dall")) {
sizes.value = dalleSizes;
} else {
sizes.value = fluxSizes;
}
params.value.model_id = selectedModel.value.id;
};
</script> </script>
<style lang="stylus"> <style lang="stylus">

116
web/src/views/Register.vue
View File

@ -10,83 +10,36 @@
<el-tab-pane label="手机注册" name="mobile" v-if="enableMobile"> <el-tab-pane label="手机注册" name="mobile" v-if="enableMobile">
<el-form-item> <el-form-item>
<div class="form-title">手机号码</div> <div class="form-title">手机号码</div>
<el-input <el-input placeholder="请输入手机号码" size="large" v-model="data.mobile" maxlength="11" autocomplete="off"> </el-input>
placeholder="请输入手机号码"
size="large"
v-model="data.mobile"
maxlength="11"
autocomplete="off"
>
</el-input>
</el-form-item> </el-form-item>
<el-form-item> <el-form-item>
<div class="form-title">验证码</div> <div class="form-title">验证码</div>
<div class="flex w100"> <div class="flex w100">
<el-input <el-input placeholder="请输入验证码" size="large" maxlength="30" class="code-input" v-model="data.code" autocomplete="off"> </el-input>
placeholder="请输入验证码"
size="large"
maxlength="30"
class="code-input"
v-model="data.code"
autocomplete="off"
>
</el-input>
<send-msg <send-msg size="large" :receiver="data.mobile" type="mobile" />
size="large"
:receiver="data.mobile"
type="mobile"
/>
</div> </div>
</el-form-item> </el-form-item>
</el-tab-pane> </el-tab-pane>
<el-tab-pane label="邮箱注册" name="email" v-if="enableEmail"> <el-tab-pane label="邮箱注册" name="email" v-if="enableEmail">
<el-form-item class="block"> <el-form-item class="block">
<div class="form-title">邮箱</div> <div class="form-title">邮箱</div>
<el-input <el-input placeholder="请输入邮箱地址" size="large" v-model="data.email" autocomplete="off"> </el-input>
placeholder="请输入邮箱地址"
size="large"
v-model="data.email"
autocomplete="off"
>
</el-input>
</el-form-item> </el-form-item>
<el-form-item class="block"> <el-form-item class="block">
<div class="form-title">验证码</div> <div class="form-title">验证码</div>
<div class="flex w100"> <div class="flex w100">
<el-input <el-input placeholder="请输入验证码" size="large" maxlength="30" class="code-input" v-model="data.code" autocomplete="off"> </el-input>
placeholder="请输入验证码"
size="large"
maxlength="30"
class="code-input"
v-model="data.code"
autocomplete="off"
>
</el-input>
<send-msg <send-msg size="large" :receiver="data.email" type="email" />
size="large"
:receiver="data.email"
type="email"
/>
</div> </div>
</el-form-item> </el-form-item>
</el-tab-pane> </el-tab-pane>
<el-tab-pane <el-tab-pane label="用户名注册" name="username" v-if="enableUser">
label="用户名注册"
name="username"
v-if="enableUser"
>
<el-form-item class="block"> <el-form-item class="block">
<div class="form-title">用户名</div> <div class="form-title">用户名</div>
<el-input <el-input placeholder="请输入用户名" size="large" v-model="data.username" autocomplete="off"> </el-input>
placeholder="请输入用户名"
size="large"
v-model="data.username"
autocomplete="off"
>
</el-input>
</el-form-item> </el-form-item>
</el-tab-pane> </el-tab-pane>
</el-tabs> </el-tabs>
@ -94,52 +47,24 @@
<el-form-item class="block"> <el-form-item class="block">
<div class="form-title">密码</div> <div class="form-title">密码</div>
<el-input <el-input placeholder="请输入密码(8-16位)" maxlength="16" size="large" v-model="data.password" show-password autocomplete="off"> </el-input>
placeholder="请输入密码(8-16位)"
maxlength="16"
size="large"
v-model="data.password"
show-password
autocomplete="off"
>
</el-input>
</el-form-item> </el-form-item>
<el-form-item class="block"> <el-form-item class="block">
<div class="form-title">重复密码</div> <div class="form-title">重复密码</div>
<el-input <el-input placeholder="请再次输入密码(8-16位)" size="large" maxlength="16" v-model="data.repass" show-password autocomplete="off"> </el-input>
placeholder="请再次输入密码(8-16位)"
size="large"
maxlength="16"
v-model="data.repass"
show-password
autocomplete="off"
>
</el-input>
</el-form-item> </el-form-item>
<el-form-item class="block"> <el-form-item class="block">
<div class="form-title">邀请码</div> <div class="form-title">邀请码</div>
<el-input <el-input placeholder="请输入邀请码(可选)" size="large" v-model="data.invite_code" autocomplete="off"> </el-input>
placeholder="请输入邀请码(可选)"
size="large"
v-model="data.invite_code"
autocomplete="off"
>
</el-input>
</el-form-item> </el-form-item>
<el-row class="btn-row" :gutter="20"> <el-row class="btn-row" :gutter="20">
<el-col :span="24"> <el-col :span="24">
<el-button <el-button class="login-btn" type="primary" size="large" @click="submitRegister">注册</el-button>
class="login-btn"
type="primary"
size="large"
@click="submitRegister"
>注册</el-button
>
</el-col> </el-col>
</el-row> </el-row>
</el-form> </el-form>
@ -156,11 +81,7 @@
</template> </template>
</el-result> </el-result>
</div> </div>
<captcha <captcha v-if="enableVerify" @success="doSubmitRegister" ref="captchaRef" />
v-if="enableVerify"
@success="doSubmitRegister"
ref="captchaRef"
/>
<account-bg /> <account-bg />
</div> </div>
</div> </div>
@ -193,7 +114,7 @@ const data = ref({
password: "", password: "",
code: "", code: "",
repass: "", repass: "",
invite_code: router.currentRoute.value.query["invite_code"] invite_code: router.currentRoute.value.query["invite_code"],
}); });
const enableMobile = ref(false); const enableMobile = ref(false);
@ -272,10 +193,7 @@ const submitRegister = () => {
return showMessageError("两次输入密码不一致"); return showMessageError("两次输入密码不一致");
} }
if ( if ((activeName.value === "mobile" || activeName.value === "email") && data.value.code === "") {
(activeName.value === "mobile" || activeName.value === "email") &&
data.value.code === ""
) {
return showMessageError("请输入验证码"); return showMessageError("请输入验证码");
} }
@ -310,11 +228,11 @@ const doSubmitRegister = (verifyData) => {
<style lang="stylus" scoped> <style lang="stylus" scoped>
@import "@/assets/css/login.styl" @import "@/assets/css/login.styl"
::v-deep(.back){ :deep(.back){
margin-bottom: 10px; margin-bottom: 10px;
} }
::v-deep(.orline){ :deep(.orline){
margin-bottom: 10px; margin-bottom: 10px;
} }
.wechat-card { .wechat-card {

54
web/src/views/admin/ApiKey.vue
View File

@ -7,7 +7,7 @@
<el-button :icon="Search" @click="fetchData">搜索</el-button> <el-button :icon="Search" @click="fetchData">搜索</el-button>
<el-button type="primary" :icon="Plus" @click="add">新增</el-button> <el-button type="primary" :icon="Plus" @click="add">新增</el-button>
<a href="https://api.chat-plus.net" target="_blank" style="margin-left: 10px"> <a href="https://api.geekai.pro" target="_blank" style="margin-left: 10px">
<el-button type="success" :icon="ShoppingCart" @click="add" plain>购买API-KEY</el-button> <el-button type="success" :icon="ShoppingCart" @click="add" plain>购买API-KEY</el-button>
</a> </a>
</div> </div>
@ -66,10 +66,21 @@
<el-dialog v-model="showDialog" :close-on-click-modal="false" :title="title"> <el-dialog v-model="showDialog" :close-on-click-modal="false" :title="title">
<el-form :model="item" label-width="120px" ref="formRef" :rules="rules"> <el-form :model="item" label-width="120px" ref="formRef" :rules="rules">
<el-form-item label="名称：" prop="name"> <el-form-item label="名称：" prop="name">
<el-input v-model="item.name" autocomplete="off" /> <template #default>
<el-row :gutter="10" class="w-full">
<el-col :span="12">
<el-input v-model="item.name" autocomplete="off" />
</el-col>
<el-col :span="12">
<el-select v-model="preset" placeholder="请选择预设" @change="changePreset">
<el-option v-for="item in presets" :value="item" :label="item.label" :key="item.value">{{ item.label }} </el-option>
</el-select>
</el-col>
</el-row>
</template>
</el-form-item> </el-form-item>
<el-form-item label="类型：" prop="type"> <el-form-item label="类型：" prop="type">
<el-select v-model="item.type" placeholder="请选择类型" @change="changeType"> <el-select v-model="item.type" placeholder="请选择类型">
<el-option v-for="item in types" :value="item.value" :label="item.label" :key="item.value">{{ item.label }} </el-option> <el-option v-for="item in types" :value="item.value" :label="item.label" :key="item.value">{{ item.label }} </el-option>
</el-select> </el-select>
</el-form-item> </el-form-item>
@ -111,10 +122,7 @@ import ClipboardJS from "clipboard";
// 变量定义 // 变量定义
const items = ref([]); const items = ref([]);
const query = ref({ type: "" }); const query = ref({ type: "" });
const item = ref({ const item = ref({});
enabled: true,
api_url: "",
});
const showDialog = ref(false); const showDialog = ref(false);
const rules = reactive({ const rules = reactive({
name: [{ required: true, message: "请输入名称", trigger: "change" }], name: [{ required: true, message: "请输入名称", trigger: "change" }],
@ -136,6 +144,14 @@ const types = ref([
]); ]);
const isEdit = ref(false); const isEdit = ref(false);
const clipboard = ref(null); const clipboard = ref(null);
const presets = ref([
{ label: "GiteeAI", value: "https://ai.gitee.com" },
{ label: "中转01", value: "https://api.geekai.pro" },
{ label: "中转03", value: "https://api.geekai.me" },
{ label: "OpenAI", value: "https://api.openai.com" },
]);
const preset = ref(null);
onMounted(() => { onMounted(() => {
clipboard.value = new ClipboardJS(".copy-key"); clipboard.value = new ClipboardJS(".copy-key");
clipboard.value.on("success", () => { clipboard.value.on("success", () => {
@ -153,18 +169,6 @@ onUnmounted(() => {
clipboard.value.destroy(); clipboard.value.destroy();
}); });
const changeType = (event) => {
if (isEdit.value) {
return;
}
if (event === "realtime") {
item.value.api_url = "wss://api.geekai.pro";
} else {
item.value.api_url = "https://api.geekai.pro";
}
};
const getTypeName = (type) => { const getTypeName = (type) => {
for (let v of types.value) { for (let v of types.value) {
if (v.value === type) { if (v.value === type) {
@ -175,7 +179,6 @@ const getTypeName = (type) => {
}; };
// 获取数据 // 获取数据
const fetchData = () => { const fetchData = () => {
httpGet("/api/admin/apikey/list", query.value) httpGet("/api/admin/apikey/list", query.value)
.then((res) => { .then((res) => {
@ -198,6 +201,10 @@ const add = function () {
showDialog.value = true; showDialog.value = true;
title.value = "新增 API KEY"; title.value = "新增 API KEY";
isEdit.value = false; isEdit.value = false;
item.value = {
enabled: true,
api_url: "",
};
}; };
const edit = function (row) { const edit = function (row) {
@ -251,6 +258,13 @@ const set = (filed, row) => {
ElMessage.error("操作失败：" + e.message); ElMessage.error("操作失败：" + e.message);
}); });
}; };
const changePreset = (row) => {
item.value.api_url = row.value;
if (!item.value.name) {
item.value.name = row.label;
}
};
</script> </script>
<style lang="stylus" scoped> <style lang="stylus" scoped>

384
web/src/views/admin/ChatModel.vue
View File

@ -1,6 +1,5 @@
<template> <template>
<div class="container model-list" v-loading="loading"> <div class="container model-list" v-loading="loading">
<div class="handle-box"> <div class="handle-box">
<el-input v-model="query.name" placeholder="模型名称" class="handle-input" /> <el-input v-model="query.name" placeholder="模型名称" class="handle-input" />
@ -9,7 +8,7 @@
</div> </div>
<el-row> <el-row>
<el-table :data="items" :row-key="row => row.id" table-layout="auto"> <el-table :data="items" :row-key="(row) => row.id" table-layout="auto">
<el-table-column type="selection" width="38"></el-table-column> <el-table-column type="selection" width="38"></el-table-column>
<el-table-column prop="name" label="模型名称"> <el-table-column prop="name" label="模型名称">
<template #default="scope"> <template #default="scope">
@ -19,30 +18,36 @@
</span> </span>
</template> </template>
</el-table-column> </el-table-column>
<el-table-column prop="type" label="模型类型">
<template #default="scope">
<el-tag type="primary" v-if="scope.row.type === 'img'">绘图</el-tag>
<el-tag type="success" v-else>聊天</el-tag>
</template>
</el-table-column>
<el-table-column prop="value" label="模型值"> <el-table-column prop="value" label="模型值">
<template #default="scope"> <template #default="scope">
<span>{{ scope.row.value }}</span> <span>{{ scope.row.value }}</span>
<el-icon class="copy-model" :data-clipboard-text="scope.row.value"> <el-icon class="copy-model" :data-clipboard-text="scope.row.value">
<DocumentCopy/> <DocumentCopy />
</el-icon> </el-icon>
</template> </template>
</el-table-column> </el-table-column>
<el-table-column prop="power" label="费率"/> <el-table-column prop="power" label="费率" />
<el-table-column prop="max_tokens" label="最大响应长度"/> <el-table-column prop="max_tokens" label="最大响应长度" />
<el-table-column prop="max_context" label="最大上下文长度"/> <el-table-column prop="max_context" label="最大上下文长度" />
<el-table-column prop="temperature" label="创意度"/> <el-table-column prop="temperature" label="创意度" />
<el-table-column prop="enabled" label="启用状态"> <el-table-column prop="enabled" label="启用状态">
<template #default="scope"> <template #default="scope">
<el-switch v-model="scope.row['enabled']" @change="modelSet('enabled',scope.row)"/> <el-switch v-model="scope.row['enabled']" @change="modelSet('enabled', scope.row)" />
</template> </template>
</el-table-column> </el-table-column>
<el-table-column prop="enabled" label="开放状态"> <el-table-column prop="enabled" label="开放状态">
<template #default="scope"> <template #default="scope">
<el-switch v-model="scope.row['open']" @change="modelSet('open',scope.row)"/> <el-switch v-model="scope.row['open']" @change="modelSet('open', scope.row)" />
</template> </template>
</el-table-column> </el-table-column>
<el-table-column prop="key_name" label="绑定API-KEY"/> <el-table-column prop="key_name" label="绑定API-KEY" />
<el-table-column label="操作" width="180"> <el-table-column label="操作" width="180">
<template #default="scope"> <template #default="scope">
<el-button size="small" type="primary" @click="edit(scope.row)">编辑</el-button> <el-button size="small" type="primary" @click="edit(scope.row)">编辑</el-button>
@ -56,81 +61,70 @@
</el-table> </el-table>
</el-row> </el-row>
<el-dialog <el-dialog v-model="showDialog" :title="title" :close-on-click-modal="false" style="width: 90%; max-width: 600px">
v-model="showDialog"
:title="title"
:close-on-click-modal="false"
style="width: 90%; max-width: 600px;"
>
<el-form :model="item" label-width="120px" ref="formRef" :rules="rules"> <el-form :model="item" label-width="120px" ref="formRef" :rules="rules">
<el-form-item label="模型类型：" prop="type">
<el-select v-model="item.type" placeholder="请选择模型类型">
<el-option v-for="v in type" :value="v.value" :label="v.label" :key="v.value">
{{ v.label }}
</el-option>
</el-select>
</el-form-item>
<el-form-item label="模型名称：" prop="name"> <el-form-item label="模型名称：" prop="name">
<el-input v-model="item.name" autocomplete="off"/> <el-input v-model="item.name" autocomplete="off" />
</el-form-item> </el-form-item>
<el-form-item label="模型值：" prop="value"> <el-form-item label="模型值：" prop="value">
<el-input v-model="item.value" autocomplete="off"/> <el-input v-model="item.value" autocomplete="off" />
</el-form-item> </el-form-item>
<el-form-item label="消耗算力：" prop="weight"> <el-form-item label="消耗算力：" prop="power">
<template #default> <template #label>
<div class="tip-input"> <div class="flex items-center">
<el-input-number :min="0" v-model="item.power" autocomplete="off"/> <span class="mr-1">消耗算力</span>
<div class="info"> <el-tooltip effect="dark" content="每日签到赠送算力" raw-content placement="right">
<el-tooltip <el-icon>
class="box-item" <InfoFilled />
effect="dark" </el-icon>
content="每次对话扣减多少单位算力" </el-tooltip>
placement="right" </div>
> </template>
<el-input v-model.number="item.power" autocomplete="off" placeholder="消耗算力" />
</el-form-item>
<div v-if="item.type === 'chat'">
<el-form-item label="最长响应：" prop="max_tokens">
<el-input v-model.number="item.max_tokens" autocomplete="off" placeholder="模型最大响应长度" />
</el-form-item>
<el-form-item>
<template #label>
<div class="flex items-center">
<span class="mr-1">最大上下文</span>
<el-tooltip effect="dark" content="去各大模型的官方 API 文档查询模型支持的最大上下文长度" raw-content placement="right">
<el-icon> <el-icon>
<InfoFilled/> <InfoFilled />
</el-icon> </el-icon>
</el-tooltip> </el-tooltip>
</div> </div>
</div> </template>
</template> <el-input v-model.number="item.max_context" autocomplete="off" placeholder="模型最大上下文长度" />
</el-form-item> </el-form-item>
<el-form-item label="最长响应：" prop="max_tokens"> <el-form-item label="创意度：" prop="temperature">
<el-input v-model.number="item.max_tokens" autocomplete="off" placeholder="模型最大响应长度"/> <template #label>
</el-form-item> <div class="flex items-center">
<span class="mr-1">创意度</span>
<el-form-item label="最大上下文：" prop="max_context"> <el-tooltip effect="dark" content="OpenAI 0-2，其他模型 0-1" raw-content placement="right">
<div class="tip-input"> <el-icon>
<el-input v-model.number="item.max_context" autocomplete="off" placeholder="模型最大上下文长度"/> <InfoFilled />
<div class="info"> </el-icon>
<el-tooltip </el-tooltip>
class="box-item" </div>
effect="dark" </template>
raw-content <el-input v-model="item.temperature" autocomplete="off" placeholder="模型创意度" />
content="去各大模型的官方 API 文档查询模型支持的最大上下文长度" </el-form-item>
placement="right" </div>
>
<el-icon>
<InfoFilled/>
</el-icon>
</el-tooltip>
</div>
</div>
</el-form-item>
<el-form-item label="创意度：" prop="temperature">
<div class="tip-input">
<el-input v-model="item.temperature" autocomplete="off" placeholder="模型创意度"/>
<div class="info">
<el-tooltip
class="box-item"
effect="dark"
content="OpenAI 0-2，其他模型 0-1"
placement="right"
>
<el-icon>
<InfoFilled/>
</el-icon>
</el-tooltip>
</div>
</div>
</el-form-item>
<el-form-item label="绑定API-KEY：" prop="apikey"> <el-form-item label="绑定API-KEY：" prop="apikey">
<el-select v-model="item.key_id" placeholder="请选择 API KEY" filterable clearable> <el-select v-model="item.key_id" placeholder="请选择 API KEY" filterable clearable>
@ -142,181 +136,193 @@
</el-form-item> </el-form-item>
<el-form-item label="启用状态：" prop="enable"> <el-form-item label="启用状态：" prop="enable">
<el-switch v-model="item.enabled"/> <el-switch v-model="item.enabled" />
</el-form-item> </el-form-item>
<el-form-item label="开放状态：" prop="open"> <el-form-item>
<div class="tip-input"> <template #label>
<el-switch v-model="item.open"/> <div class="flex items-center">
<div class="info"> <span class="mr-1">开放状态</span>
<el-tooltip <el-tooltip effect="dark" content="开放后，该模型将对所有用户可见。<br/> 如果模型没有启用，则当前设置无效。" raw-content placement="right">
class="box-item"
effect="dark"
raw-content
content="开放后，该模型将对所有用户可见。<br/> 如果模型没有启用，则当前设置无效。"
placement="right"
>
<el-icon> <el-icon>
<InfoFilled/> <InfoFilled />
</el-icon> </el-icon>
</el-tooltip> </el-tooltip>
</div> </div>
</div> </template>
<el-switch v-model="item.open" />
</el-form-item> </el-form-item>
</el-form> </el-form>
<template #footer> <template #footer>
<span class="dialog-footer"> <span class="dialog-footer">
<el-button @click="showDialog = false">取消</el-button> <el-button @click="showDialog = false">取消</el-button>
<el-button type="primary" @click="save">提交</el-button> <el-button type="primary" @click="save">提交</el-button>
</span> </span>
</template> </template>
</el-dialog> </el-dialog>
</div> </div>
</template> </template>
<script setup> <script setup>
import {onMounted, onUnmounted, reactive, ref} from "vue"; import { onMounted, onUnmounted, reactive, ref } from "vue";
import {httpGet, httpPost} from "@/utils/http"; import { httpGet, httpPost } from "@/utils/http";
import {ElMessage} from "element-plus"; import { ElMessage } from "element-plus";
import {dateFormat, removeArrayItem, substr} from "@/utils/libs"; import { dateFormat, removeArrayItem, substr } from "@/utils/libs";
import {DocumentCopy, InfoFilled, Plus,Search} from "@element-plus/icons-vue"; import { DocumentCopy, InfoFilled, Plus, Search } from "@element-plus/icons-vue";
import {Sortable} from "sortablejs"; import { Sortable } from "sortablejs";
import ClipboardJS from "clipboard"; import ClipboardJS from "clipboard";
import Default from "md-editor-v3"; import Default from "md-editor-v3";
// 变量定义 // 变量定义
const items = ref([]) const items = ref([]);
const query = ref({name:''}) const query = ref({ name: "" });
const item = ref({}) const item = ref({});
const showDialog = ref(false) const showDialog = ref(false);
const title = ref("") const title = ref("");
const rules = reactive({ const rules = reactive({
name: [{required: true, message: '请输入模型名称', trigger: 'change',}], type: [{ required: true, message: "请选择模型类型", trigger: "change" }],
value: [{required: true, message: '请输入模型值', trigger: 'change',}] name: [{ required: true, message: "请输入模型名称", trigger: "change" }],
}) value: [{ required: true, message: "请输入模型值", trigger: "change" }],
const loading = ref(true) power: [{ required: true, message: "请输入消耗算力", trigger: "change" }],
const formRef = ref(null) });
const loading = ref(true);
const formRef = ref(null);
const type = ref([
{ label: "聊天", value: "chat" },
{ label: "绘图", value: "img" },
]);
// 获取 API KEY // 获取 API KEY
const apiKeys = ref([]) const apiKeys = ref([]);
httpGet('/api/admin/apikey/list?type=chat').then(res => { httpGet("/api/admin/apikey/list?type=chat")
apiKeys.value = res.data .then((res) => {
}).catch(e => { apiKeys.value = res.data;
ElMessage.error("获取 API KEY 失败：" + e.message) })
}) .catch((e) => {
ElMessage.error("获取 API KEY 失败：" + e.message);
});
// 获取数据 // 获取数据
const fetchData = () => { const fetchData = () => {
httpGet('/api/admin/model/list', query.value).then((res) => { httpGet("/api/admin/model/list", query.value)
if (res.data) { .then((res) => {
// 初始化数据 if (res.data) {
const arr = res.data; // 初始化数据
for (let i = 0; i < arr.length; i++) { const arr = res.data;
arr[i].last_used_at = dateFormat(arr[i].last_used_at) for (let i = 0; i < arr.length; i++) {
arr[i].last_used_at = dateFormat(arr[i].last_used_at);
}
items.value = arr;
} }
items.value = arr loading.value = false;
} })
loading.value = false .catch(() => {
}).catch(() => { ElMessage.error("获取数据失败");
ElMessage.error("获取数据失败"); });
}) };
}
const clipboard = ref(null) const clipboard = ref(null);
onMounted(() => { onMounted(() => {
fetchData() fetchData();
const drawBodyWrapper = document.querySelector('.el-table__body tbody') const drawBodyWrapper = document.querySelector(".el-table__body tbody");
// 初始化拖动排序插件 // 初始化拖动排序插件
Sortable.create(drawBodyWrapper, { Sortable.create(drawBodyWrapper, {
sort: true, sort: true,
animation: 500, animation: 500,
onEnd({newIndex, oldIndex, from}) { onEnd({ newIndex, oldIndex, from }) {
if (oldIndex === newIndex) { if (oldIndex === newIndex) {
return return;
} }
const sortedData = Array.from(from.children).map(row => row.querySelector('.sort').getAttribute('data-id')); const sortedData = Array.from(from.children).map((row) => row.querySelector(".sort").getAttribute("data-id"));
const ids = [] const ids = [];
const sorts = [] const sorts = [];
sortedData.forEach((id, index) => { sortedData.forEach((id, index) => {
ids.push(parseInt(id)) ids.push(parseInt(id));
sorts.push(index + 1) sorts.push(index + 1);
items.value[index].sort_num = index + 1 items.value[index].sort_num = index + 1;
}) });
httpPost("/api/admin/model/sort", {ids: ids, sorts: sorts}).then(() => { httpPost("/api/admin/model/sort", { ids: ids, sorts: sorts })
}).catch(e => { .then(() => {})
ElMessage.error("排序失败：" + e.message) .catch((e) => {
}) ElMessage.error("排序失败：" + e.message);
} });
}) },
});
clipboard.value = new ClipboardJS('.copy-model'); clipboard.value = new ClipboardJS(".copy-model");
clipboard.value.on('success', () => { clipboard.value.on("success", () => {
ElMessage.success('复制成功！'); ElMessage.success("复制成功！");
}) });
clipboard.value.on('error', () => { clipboard.value.on("error", () => {
ElMessage.error('复制失败！'); ElMessage.error("复制失败！");
}) });
}) });
onUnmounted(() => { onUnmounted(() => {
clipboard.value.destroy() clipboard.value.destroy();
}) });
const add = function () { const add = function () {
title.value = "新增模型" title.value = "新增模型";
showDialog.value = true showDialog.value = true;
item.value = {enabled: true, power: 1, open: true,max_tokens: 1024,max_context: 8192, temperature: 0.9,} item.value = { enabled: true, power: 1, open: true, max_tokens: 1024, max_context: 8192, temperature: 0.9 };
} };
const edit = function (row) { const edit = function (row) {
title.value = "修改模型" title.value = "修改模型";
showDialog.value = true showDialog.value = true;
item.value = row item.value = row;
} };
const save = function () { const save = function () {
formRef.value.validate((valid) => { formRef.value.validate((valid) => {
item.value.temperature = parseFloat(item.value.temperature) item.value.temperature = parseFloat(item.value.temperature);
if (!item.value.sort_num) { if (!item.value.sort_num) {
item.value.sort_num = items.value.length item.value.sort_num = items.value.length;
} }
if (valid) { if (valid) {
showDialog.value = false showDialog.value = false;
item.value.key_id = parseInt(item.value.key_id) item.value.key_id = parseInt(item.value.key_id);
httpPost('/api/admin/model/save', item.value).then(() => { httpPost("/api/admin/model/save", item.value)
ElMessage.success('操作成功！') .then(() => {
fetchData() ElMessage.success("操作成功！");
}).catch((e) => { fetchData();
ElMessage.error('操作失败，' + e.message) })
}) .catch((e) => {
ElMessage.error("操作失败，" + e.message);
});
} else { } else {
return false return false;
} }
}) });
} };
const modelSet = (filed, row) => { const modelSet = (filed, row) => {
httpPost('/api/admin/model/set', {id: row.id, filed: filed, value: row[filed]}).then(() => { httpPost("/api/admin/model/set", { id: row.id, filed: filed, value: row[filed] })
ElMessage.success("操作成功！") .then(() => {
}).catch(e => { ElMessage.success("操作成功！");
ElMessage.error("操作失败：" + e.message) })
}) .catch((e) => {
} ElMessage.error("操作失败：" + e.message);
});
};
const remove = function (row) { const remove = function (row) {
httpGet('/api/admin/model/remove?id=' + row.id).then(() => { httpGet("/api/admin/model/remove?id=" + row.id)
ElMessage.success("删除成功！") .then(() => {
items.value = removeArrayItem(items.value, row, (v1, v2) => { ElMessage.success("删除成功！");
return v1.id === v2.id items.value = removeArrayItem(items.value, row, (v1, v2) => {
return v1.id === v2.id;
});
}) })
}).catch((e) => { .catch((e) => {
ElMessage.error("删除失败：" + e.message) ElMessage.error("删除失败：" + e.message);
}) });
} };
</script> </script>
<style lang="stylus" scoped> <style lang="stylus" scoped>

39
web/src/views/admin/SysConfig.vue
View File

@ -203,12 +203,18 @@
<el-form-item label="VIP每月赠送算力" prop="vip_month_power"> <el-form-item label="VIP每月赠送算力" prop="vip_month_power">
<el-input v-model.number="system['vip_month_power']" placeholder="VIP用户每月赠送算力" /> <el-input v-model.number="system['vip_month_power']" placeholder="VIP用户每月赠送算力" />
</el-form-item> </el-form-item>
<el-form-item label="每日赠送算力" prop="daily_power"> <el-form-item>
<template #label>
<div class="label-title">
签到赠送算力
<el-tooltip effect="dark" content="每日签到赠送算力" raw-content placement="right">
<el-icon>
<InfoFilled/>
</el-icon>
</el-tooltip>
</div>
</template>
<el-input v-model.number="system['daily_power']" placeholder="默认值0" /> <el-input v-model.number="system['daily_power']" placeholder="默认值0" />
<el-text type="info">
如果设置0表示不赠送，用户享受完免费算力额度之后就不能再发起对话了。如果设置为N，则系统每天将算力值小于N的用户自动补充到N。注意，此功能要配合XXL-JOB启用。
</el-text>
</el-form-item> </el-form-item>
<el-form-item> <el-form-item>
<template #label> <template #label>
@ -240,6 +246,17 @@
<el-input v-model.number="system['sd_power']" placeholder="使用Stable-Diffusion画一张图消耗算力" /> <el-input v-model.number="system['sd_power']" placeholder="使用Stable-Diffusion画一张图消耗算力" />
</el-form-item> </el-form-item>
<el-form-item label="DALL-E-3算力" prop="dall_power"> <el-form-item label="DALL-E-3算力" prop="dall_power">
<template #label>
<div class="label-title">
MJ操作算力
<el-tooltip effect="dark" content="主要用户函数调用 DALL-E-3 进行绘画" raw-content
placement="right">
<el-icon>
<InfoFilled/>
</el-icon>
</el-tooltip>
</div>
</template>
<el-input v-model.number="system['dall_power']" placeholder="使用DALL-E-3画一张图消耗算力" /> <el-input v-model.number="system['dall_power']" placeholder="使用DALL-E-3画一张图消耗算力" />
</el-form-item> </el-form-item>
<el-form-item label="Suno 算力" prop="suno_power"> <el-form-item label="Suno 算力" prop="suno_power">
@ -379,17 +396,17 @@
</template> </template>
<script setup> <script setup>
import { onMounted, reactive, ref } from "vue"; import {onMounted, reactive, ref} from "vue";
import { httpGet, httpPost } from "@/utils/http"; import {httpGet, httpPost} from "@/utils/http";
import Compressor from "compressorjs"; import Compressor from "compressorjs";
import { ElMessage, ElMessageBox } from "element-plus"; import {ElMessage, ElMessageBox} from "element-plus";
import { CloseBold, InfoFilled, Select, UploadFilled } from "@element-plus/icons-vue"; import {CloseBold, InfoFilled, Select, UploadFilled} from "@element-plus/icons-vue";
import MdEditor from "md-editor-v3"; import MdEditor from "md-editor-v3";
import "md-editor-v3/lib/style.css"; import "md-editor-v3/lib/style.css";
import Menu from "@/views/admin/Menu.vue"; import Menu from "@/views/admin/Menu.vue";
import { copyObj, dateFormat } from "@/utils/libs"; import {copyObj, dateFormat} from "@/utils/libs";
import ItemsInput from "@/components/ui/ItemsInput.vue"; import ItemsInput from "@/components/ui/ItemsInput.vue";
import { useSharedStore } from "@/store/sharedata"; import {useSharedStore} from "@/store/sharedata";
const activeName = ref("basic"); const activeName = ref("basic");
const system = ref({ models: [] }); const system = ref({ models: [] });