fix: fix the agent tool

Kujtim Hoxha and Crush created 6 months ago

💖 Generated with Crush
Co-Authored-By: Crush <[e-mail]>

Change summary

internal/tui/components/chat/messages/renderer.go | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)

Detailed changes

internal/tui/components/chat/messages/renderer.go

@@ -502,7 +502,7 @@ func (tr agentRenderer) Render(v *toolCallCmp) string {
 	prompt = strings.ReplaceAll(prompt, "\n", " ")
 
 	header := tr.makeHeader(v, "Agent", v.textWidth())
-	if res, done := earlyState(header, v); done {
+	if res, done := earlyState(header, v); v.cancelled && done {
 		return res
 	}
 	taskTag := t.S().Base.Padding(0, 1).MarginLeft(1).Background(t.BlueLight).Foreground(t.White).Render("Task")
@@ -527,6 +527,7 @@ func (tr agentRenderer) Render(v *toolCallCmp) string {
 	parts := []string{
 		childTools.Enumerator(RoundedEnumerator).String(),
 	}
+
 	if v.result.ToolCallID == "" {
 		v.spinning = true
 		parts = append(parts, v.anim.View().String())