refactor: Update message handling to ignore progress messages

- Modify message handling logic to include "progress" in the list of ignored message types alongside "loading", "error", and "action".
- Comment out the loading message sending logic in the Execute method to streamline the process and improve clarity in message flow.
This commit is contained in:
Max 2025-04-19 15:38:12 +08:00
parent 6203b5ccf2
commit 57e8389afd
2 changed files with 12 additions and 11 deletions

View file

@ -264,18 +264,19 @@ func (next *NextAction) Execute(c *gin.Context, ctx chatctx.Context, contents *c
return nil, fmt.Errorf("with history error: %s", err.Error())
}
// Send the progress message from application side instead
// Create a new Text
// Send loading message and mark as new
if !ctx.Silent {
msg := chatMessage.New().Map(map[string]interface{}{
"new": true,
"role": "assistant",
"type": "loading",
"props": map[string]interface{}{"placeholder": "Calling " + assistant.Name},
})
msg.Assistant(assistant.ID, assistant.Name, assistant.Avatar)
msg.Write(c.Writer)
}
// if !ctx.Silent {
// msg := chatMessage.New().Map(map[string]interface{}{
// "new": true,
// "role": "assistant",
// "type": "loading",
// "props": map[string]interface{}{"placeholder": "Calling " + assistant.Name},
// })
// msg.Assistant(assistant.ID, assistant.Name, assistant.Avatar)
// msg.Write(c.Writer)
// }
newContents := chatMessage.NewContents()
// Update the context id

View file

@ -432,7 +432,7 @@ func (m *Message) AppendTo(contents *Contents) *Message {
}
return m
case "loading", "error", "action": // Ignore loading, action and error messages
case "loading", "error", "action", "progress": // Ignore progress, loading, action and error messages
return m
default: