xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2026-02-03 04:56:00 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

修改模型选择功能，提升用户体验

Browse Source
This commit is contained in:
清柯
2025-04-25 03:07:51 +08:00
parent 51c270fb29
commit c7b09f29ca
5 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/types/chat.go
View File

@@ -70,6 +70,8 @@ type ChatModel struct {
Power int `json:"power"`
MaxTokens int `json:"max_tokens"` // 最大响应长度
MaxContext int `json:"max_context"` // 最大上下文长度
Description string `json:"description"` //模型描述
Category string `json:"category"` //模型类别
Temperature float32 `json:"temperature"` // 模型温度
KeyId int `json:"key_id"` // 绑定 API KEY
}

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

@@ -40,6 +40,8 @@ func (h *ChatModelHandler) Save(c *gin.Context) {
Power int `json:"power"`
MaxTokens int `json:"max_tokens"` // 最大响应长度
MaxContext int `json:"max_context"` // 最大上下文长度
Description string `json:"description"` //模型描述
Category string `json:"category"` //模型类别
Temperature float32 `json:"temperature"` // 模型温度
KeyId int `json:"key_id,omitempty"`
CreatedAt int64 `json:"created_at"`
@@ -64,6 +66,8 @@ func (h *ChatModelHandler) Save(c *gin.Context) {
item.Power = data.Power
item.MaxTokens = data.MaxTokens
item.MaxContext = data.MaxContext
item.Description = data.Description
item.Category = data.Category
item.Temperature = data.Temperature
item.KeyId = data.KeyId
item.Type = data.Type

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

@@ -10,6 +10,8 @@ type ChatModel struct {
Open bool // 是否开放模型给所有人使用
MaxTokens int // 最大响应长度
MaxContext int // 最大上下文长度
Description string //模型描述
Category string //模型类别
Temperature float32 // 模型温度
KeyId int // 绑定 API KEY ID
Type string // 模型类型

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

@@ -10,6 +10,8 @@ type ChatModel struct {
Open bool `json:"open"`
MaxTokens int `json:"max_tokens"` // 最大响应长度
MaxContext int `json:"max_context"` // 最大上下文长度
Description string `json:"description"` // 模型描述
Category string `json:"category"` //模型类别
Temperature float32 `json:"temperature"` // 模型温度
KeyId int `json:"key_id,omitempty"`
KeyName string `json:"key_name"`

2
database/update-v4.2.3.sql Normal file
View File

@@ -0,0 +1,2 @@
ALTER TABLE `chatgpt_chat_models` ADD `category` VARCHAR(1024) NOT NULL DEFAULT '' COMMENT '模型类别' AFTER `id`;
ALTER TABLE `chatgpt_chat_models` ADD `description` VARCHAR(1024) NOT NULL DEFAULT '' COMMENT '模型类型描述' AFTER `id`;