diff options
Diffstat (limited to 'internal/llm/agent/coder.go')
| -rw-r--r-- | internal/llm/agent/coder.go | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/internal/llm/agent/coder.go b/internal/llm/agent/coder.go index 8ff3c61aa..5deff05a8 100644 --- a/internal/llm/agent/coder.go +++ b/internal/llm/agent/coder.go @@ -1,6 +1,7 @@ package agent import ( + "context" "errors" "github.com/kujtimiihoxha/termai/internal/app" @@ -28,9 +29,9 @@ func (c *coderAgent) setAgentTool(sessionID string) { } } -func (c *coderAgent) Generate(sessionID string, content string) error { +func (c *coderAgent) Generate(ctx context.Context, sessionID string, content string) error { c.setAgentTool(sessionID) - return c.generate(sessionID, content) + return c.generate(ctx, sessionID, content) } func NewCoderAgent(app *app.App) (Agent, error) { |