From 36f201d5d3aaba7e0285d86cf1c0cf6b54769cff Mon Sep 17 00:00:00 2001 From: Kujtim Hoxha Date: Thu, 10 Apr 2025 13:29:44 +0200 Subject: fix logs and add cancellation --- internal/llm/agent/task.go | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'internal/llm/agent/task.go') diff --git a/internal/llm/agent/task.go b/internal/llm/agent/task.go index 9737d41b8..034e93460 100644 --- a/internal/llm/agent/task.go +++ b/internal/llm/agent/task.go @@ -1,6 +1,7 @@ package agent import ( + "context" "errors" "github.com/kujtimiihoxha/termai/internal/app" @@ -13,8 +14,8 @@ type taskAgent struct { *agent } -func (c *taskAgent) Generate(sessionID string, content string) error { - return c.generate(sessionID, content) +func (c *taskAgent) Generate(ctx context.Context, sessionID string, content string) error { + return c.generate(ctx, sessionID, content) } func NewTaskAgent(app *app.App) (Agent, error) { -- cgit v1.2.3