mirror of
https://github.com/yangjian102621/geekai.git
synced 2025-09-17 16:56:38 +08:00
50 lines
1.2 KiB
Go
50 lines
1.2 KiB
Go
package fun
|
|
|
|
import (
|
|
"chatplus/core/types"
|
|
"chatplus/service/mj"
|
|
"chatplus/utils"
|
|
"errors"
|
|
)
|
|
|
|
// AI 绘画函数
|
|
|
|
type FuncMidJourney struct {
|
|
name string
|
|
service *mj.Service
|
|
config types.MidJourneyConfig
|
|
}
|
|
|
|
func NewMidJourneyFunc(mjService *mj.Service, config types.MidJourneyConfig) FuncMidJourney {
|
|
return FuncMidJourney{
|
|
name: "MidJourney AI 绘画",
|
|
config: config,
|
|
service: mjService}
|
|
}
|
|
|
|
func (f FuncMidJourney) Invoke(params map[string]interface{}) (string, error) {
|
|
if !f.config.Enabled {
|
|
return "", errors.New("MidJourney AI 绘画功能没有启用")
|
|
}
|
|
|
|
logger.Infof("MJ 绘画参数：%+v", params)
|
|
prompt := utils.InterfaceToString(params["prompt"])
|
|
f.service.PushTask(types.MjTask{
|
|
SessionId: utils.InterfaceToString(params["session_id"]),
|
|
Src: types.TaskSrcChat,
|
|
Type: types.TaskImage,
|
|
Prompt: prompt,
|
|
UserId: utils.IntValue(utils.InterfaceToString(params["user_id"]), 0),
|
|
RoleId: utils.IntValue(utils.InterfaceToString(params["role_id"]), 0),
|
|
Icon: utils.InterfaceToString(params["icon"]),
|
|
ChatId: utils.InterfaceToString(params["chat_id"]),
|
|
})
|
|
return prompt, nil
|
|
}
|
|
|
|
func (f FuncMidJourney) Name() string {
|
|
return f.name
|
|
}
|
|
|
|
var _ Function = &FuncMidJourney{}
|