small fixes

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))

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