From bd2ec29b65e430f83f430db5fdc424c7d631989d Mon Sep 17 00:00:00 2001 From: Kujtim Hoxha Date: Sat, 12 Apr 2025 18:45:36 +0200 Subject: add initial git support --- internal/llm/agent/agent.go | 3 +++ 1 file changed, 3 insertions(+) (limited to 'internal/llm/agent') diff --git a/internal/llm/agent/agent.go b/internal/llm/agent/agent.go index b01ffec3c..89de627f7 100644 --- a/internal/llm/agent/agent.go +++ b/internal/llm/agent/agent.go @@ -246,6 +246,7 @@ func (c *agent) handleToolExecution( } func (c *agent) generate(ctx context.Context, sessionID string, content string) error { + ctx = context.WithValue(ctx, tools.SessionIDContextKey, sessionID) messages, err := c.Messages.List(sessionID) if err != nil { return err @@ -310,6 +311,8 @@ func (c *agent) generate(ctx context.Context, sessionID string, content string) if err != nil { return err } + + ctx = context.WithValue(ctx, tools.MessageIDContextKey, assistantMsg.ID) for event := range eventChan { err = c.processEvent(sessionID, &assistantMsg, event) if err != nil { -- cgit v1.2.3