small fixes
This commit is contained in:
parent
333ea6ec4b
commit
72afeb9f54
10 changed files with 106 additions and 75 deletions
|
|
@ -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))
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue