Skip to content

Commit d4f05b6

Browse files
author
quanbisen
committed
fix: deekseek model tool call fail cause by "\nObservation" args input
1 parent 509308f commit d4f05b6

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

agents/conversational.go

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -157,8 +157,10 @@ func (a *ConversationalAgent) parseOutput(output string) ([]schema.AgentAction,
157157
return nil, nil, fmt.Errorf("%w: %s", ErrUnableToParseOutput, output)
158158
}
159159

160+
toolInput := strings.TrimRight(matches[2], "\nObservation:")
161+
toolInput = strings.TrimRight(toolInput, "\nObservation")
160162
return []schema.AgentAction{
161-
{Tool: strings.TrimSpace(matches[1]), ToolInput: strings.TrimSpace(matches[2]), Log: output},
163+
{Tool: strings.TrimSpace(matches[1]), ToolInput: strings.TrimSpace(toolInput), Log: output},
162164
}, nil, nil
163165
}
164166

agents/executor.go

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -135,7 +135,9 @@ func (e *Executor) doAction(
135135
}), nil
136136
}
137137

138-
observation, err := tool.Call(ctx, strings.TrimSuffix(action.ToolInput, "\nObservation:"))
138+
toolInput := strings.TrimSuffix(action.ToolInput, "\nObservation:")
139+
toolInput = strings.TrimSuffix(toolInput, "\nObservation")
140+
observation, err := tool.Call(ctx, toolInput)
139141
if err != nil {
140142
return nil, err
141143
}

0 commit comments

Comments
 (0)