xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-11-12 04:03:42 +08:00
Code Issues Actions 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
Download Patch File Download Diff File Expand all files Collapse all files

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) {
var items []model.ChatModel
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")
if t != "" {
session = session.Where("type", t)