diff --git a/pkg/seahorse/tool_expand.go b/pkg/seahorse/tool_expand.go index 749c9cd6..7731cb69 100644 --- a/pkg/seahorse/tool_expand.go +++ b/pkg/seahorse/tool_expand.go @@ -124,6 +124,9 @@ func (t *ExpandTool) Execute(ctx context.Context, args map[string]any) *tools.To "tokenCount": result.TokenCount, "messages": messages, } - data, _ := json.Marshal(output) + data, err := json.Marshal(output) + if err != nil { + return tools.ErrorResult(fmt.Sprintf("failed to marshal expand result: %v", err)) + } return tools.NewToolResult(string(data)) } diff --git a/pkg/seahorse/tool_grep.go b/pkg/seahorse/tool_grep.go index 9671d2a7..5ea32624 100644 --- a/pkg/seahorse/tool_grep.go +++ b/pkg/seahorse/tool_grep.go @@ -167,6 +167,9 @@ func (t *GrepTool) Execute(ctx context.Context, args map[string]any) *tools.Tool output["hint"] = result.Hint } - data, _ := json.Marshal(output) + data, err := json.Marshal(output) + if err != nil { + return tools.ErrorResult(fmt.Sprintf("failed to marshal grep result: %v", err)) + } return tools.NewToolResult(string(data)) }