diff options
| author | Kujtim Hoxha <[email protected]> | 2025-04-18 21:24:35 +0200 |
|---|---|---|
| committer | Kujtim Hoxha <[email protected]> | 2025-04-21 13:42:29 +0200 |
| commit | 72afeb9f54cee8e248093a52ac0779441c79aea3 (patch) | |
| tree | 70f5966bcf2cb8a6186780ea5b22a3515bc8ee22 /internal/llm | |
| parent | 333ea6ec4b2abfc2c1a9c3f6b0918ca5d296347f (diff) | |
| download | opencode-72afeb9f54cee8e248093a52ac0779441c79aea3.tar.gz opencode-72afeb9f54cee8e248093a52ac0779441c79aea3.zip | |
small fixes
Diffstat (limited to 'internal/llm')
| -rw-r--r-- | internal/llm/agent/agent.go | 2 | ||||
| -rw-r--r-- | internal/llm/tools/edit.go | 6 |
2 files changed, 5 insertions, 3 deletions
diff --git a/internal/llm/agent/agent.go b/internal/llm/agent/agent.go index 5e9785991..7542d9adf 100644 --- a/internal/llm/agent/agent.go +++ b/internal/llm/agent/agent.go @@ -221,6 +221,8 @@ func (a *agent) processGeneration(ctx context.Context, sessionID, content string agentMessage, toolResults, err := a.streamAndHandleEvents(ctx, sessionID, msgHistory) if err != nil { if errors.Is(err, context.Canceled) { + agentMessage.AddFinish(message.FinishReasonCanceled) + a.messages.Update(context.Background(), agentMessage) return a.err(ErrRequestCancelled) } return a.err(fmt.Errorf("failed to process events: %w", err)) diff --git a/internal/llm/tools/edit.go b/internal/llm/tools/edit.go index 6a1616010..83cec5dba 100644 --- a/internal/llm/tools/edit.go +++ b/internal/llm/tools/edit.go @@ -141,20 +141,20 @@ func (e *editTool) Run(ctx context.Context, call ToolCall) (ToolResponse, error) if params.OldString == "" { response, err = e.createNewFile(ctx, params.FilePath, params.NewString) if err != nil { - return response, nil + return response, err } } if params.NewString == "" { response, err = e.deleteContent(ctx, params.FilePath, params.OldString) if err != nil { - return response, nil + return response, err } } response, err = e.replaceContent(ctx, params.FilePath, params.OldString, params.NewString) if err != nil { - return response, nil + return response, err } if response.IsError { // Return early if there was an error during content replacement |