Improve message filtering in Neo API assistant stream processing
- Added filtering for 'function' type messages in requestMessages method - Removed debug print statement in streamChat method - Enhanced message handling to ignore function call messages during processing This change refines the message filtering logic in the Neo API assistant, improving the clarity and efficiency of message stream processing.
This commit is contained in:
parent
63807e1944
commit
bc662d95fa
1 changed files with 6 additions and 2 deletions
|
|
@ -308,8 +308,6 @@ func (ast *Assistant) streamChat(
|
||||||
// msg.Write(c.Writer)
|
// msg.Write(c.Writer)
|
||||||
// }
|
// }
|
||||||
|
|
||||||
fmt.Println("Done", contents.JSON())
|
|
||||||
|
|
||||||
res, hookErr := ast.HookDone(c, ctx, messages, contents)
|
res, hookErr := ast.HookDone(c, ctx, messages, contents)
|
||||||
if hookErr == nil && res != nil {
|
if hookErr == nil && res != nil {
|
||||||
|
|
||||||
|
|
@ -468,6 +466,12 @@ func (ast *Assistant) requestMessages(ctx context.Context, messages []chatMessag
|
||||||
length := len(messages)
|
length := len(messages)
|
||||||
|
|
||||||
for index, message := range messages {
|
for index, message := range messages {
|
||||||
|
|
||||||
|
// Ignore the function call message
|
||||||
|
if message.Type == "function" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
role := message.Role
|
role := message.Role
|
||||||
if role == "" {
|
if role == "" {
|
||||||
return nil, fmt.Errorf("role must be string")
|
return nil, fmt.Errorf("role must be string")
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue