fixed markdown generating styles

api/handler/markmap_handler.go

@@ -74,6 +74,7 @@ func (h *MarkMapHandler) Client(c *gin.Context) {
 			logger.Info("Receive a message: ", message.Content)
 			err = h.sendMessage(client, utils.InterfaceToString(message.Content), modelId, userId)
 			if err != nil {
+				logger.Error(err)
 				utils.ReplyChunkMessage(client, types.WsMessage{Type: types.WsErr, Content: err.Error()})
 			}
 
@@ -102,7 +103,7 @@ func (h *MarkMapHandler) sendMessage(client *types.WsClient, prompt string, mode
 	}
 
 	messages := make([]interface{}, 0)
-	messages = append(messages, types.Message{Role: "system", Content: "你是一位非常优秀的思维导图助手，你会把用户的所有提问都总结成思维导图，然后以 Markdown 格式输出。只输出 Markdown 内容，不要输出任何解释性的语句。"})
+	messages = append(messages, types.Message{Role: "system", Content: "你是一位非常优秀的思维导图助手，你会把用户的所有提问都总结成思维导图，然后以 Markdown 格式输出。不要输出任何解释性的语句。"})
 	messages = append(messages, types.Message{Role: "user", Content: prompt})
 	var req = types.ApiRequest{
 		Model:    chatModel.Value,

api/service/mj/service.go

@@ -53,7 +53,7 @@ func (s *Service) Run() {
 
 		// translate prompt
 		if utils.HasChinese(task.Prompt) {
-			content, err := utils.OpenAIRequest(s.db, fmt.Sprintf(service.TranslatePromptTemplate, task.Prompt))
+			content, err := utils.OpenAIRequest(s.db, fmt.Sprintf(service.RewritePromptTemplate, task.Prompt))
 			if err == nil {
 				task.Prompt = content
 			} else {
@@ -62,7 +62,7 @@ func (s *Service) Run() {
 		}
 		// translate negative prompt
 		if task.NegPrompt != "" && utils.HasChinese(task.NegPrompt) {
-			content, err := utils.OpenAIRequest(s.db, fmt.Sprintf(service.TranslatePromptTemplate, task.NegPrompt))
+			content, err := utils.OpenAIRequest(s.db, fmt.Sprintf(service.RewritePromptTemplate, task.NegPrompt))
 			if err == nil {
 				task.NegPrompt = content
 			} else {

api/store/model/dalle_job.go

@@ -0,0 +1,16 @@
+package model
+
+import "time"
+
+type DallJob struct {
+	Id        uint `gorm:"primarykey;column:id"`
+	UserId    int
+	TaskId      string
+	Prompt    string
+	ImgURL       string
+	Publish       bool
+	Power      int
+	Progress      int
+	ErrMsg      string
+	CreatedAt time.Time
+}

api/store/vo/dalle_job.go

@@ -0,0 +1,14 @@
+package vo
+
+type DallJob struct {
+	Id        uint `json:"id"`
+	UserId    int `json:"user_id"`
+	TaskId      string `json:"task_id"`
+	Prompt    string `json:"prompt"`
+	ImgURL       string `json:"img_url"`
+	Publish       bool `json:"publish"`
+	Power      int `json:"power"`
+	Progress      int `json:"progress"`
+	ErrMsg      string `json:"err_msg"`
+	CreatedAt int64 `json:"created_at"`
+}