diff --git a/internal/agent/agent.go b/internal/agent/agent.go index d9a5ee837fe9b71da2bede2828e5720bead020e2..a672708c414545b8e61e4d6ece8940e9a7e1c679 100644 --- a/internal/agent/agent.go +++ b/internal/agent/agent.go @@ -239,6 +239,10 @@ func (a *sessionAgent) Run(ctx context.Context, call SessionAgentCall) (*fantasy currentAssistant = &assistantMsg return callContext, prepared, err }, + OnReasoningStart: func(id string, reasoning fantasy.ReasoningContent) error { + currentAssistant.AppendReasoningContent(reasoning.Text) + return a.messages.Update(genCtx, *currentAssistant) + }, OnReasoningDelta: func(id string, text string) error { currentAssistant.AppendReasoningContent(text) return a.messages.Update(genCtx, *currentAssistant)