feat: midjourney drawing image function is ready

api/core/types/chat.go

@@ -43,7 +43,6 @@ type ChatSession struct {
 }
 
 type MjTask struct {
-	Client      Client
 	ChatId      string
 	MessageId   string
 	MessageHash string
@@ -63,6 +62,7 @@ type ApiError struct {
 
 const PromptMsg = "prompt" // prompt message
 const ReplyMsg = "reply"   // reply message
+const MjMsg = "mj"
 
 var ModelToTokens = map[string]int{
 	"gpt-3.5-turbo":     4096,
@@ -70,3 +70,5 @@ var ModelToTokens = map[string]int{
 	"gpt-4":             8192,
 	"gpt-4-32k":         32768,
 }
+
+const TaskStorePrefix = "/tasks/"

api/core/types/function.go

@@ -87,7 +87,11 @@ var InnerFunctions = []Function{
 				},
 				"ar": {
 					Type:        "string",
-					Description: "图片长宽比，如 16:9",
+					Description: "图片长宽比，如 16:9, --ar 3:2",
+				},
+				"niji": {
+					Type:        "string",
+					Description: "动漫模型版本，如 --niji 5",
 				},
 			},
 			Required: []string{},

api/core/types/web.go

@@ -21,7 +21,7 @@ const (
 	WsStart  = WsMsgType("start")
 	WsMiddle = WsMsgType("middle")
 	WsEnd    = WsMsgType("end")
-	WsImg    = WsMsgType("img")
+	WsMjImg  = WsMsgType("mj")
 )
 
 type BizCode int