Merge pull request #840 from trheyi/main

Improve message filtering in Neo API assistant stream processing
This commit is contained in:
Max 2025-01-29 16:45:36 +08:00 committed by GitHub
commit 86b88387a0
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -308,8 +308,6 @@ func (ast *Assistant) streamChat(
// msg.Write(c.Writer)
// }
fmt.Println("Done", contents.JSON())
res, hookErr := ast.HookDone(c, ctx, messages, contents)
if hookErr == nil && res != nil {
@ -468,6 +466,12 @@ func (ast *Assistant) requestMessages(ctx context.Context, messages []chatMessag
length := len(messages)
for index, message := range messages {
// Ignore the function call message
if message.Type == "function" {
continue
}
role := message.Role
if role == "" {
return nil, fmt.Errorf("role must be string")