From 765a9cc222548c7ac2b6078ee9c6747d7f6f5d15 Mon Sep 17 00:00:00 2001 From: kujtimiihoxha Date: Thu, 23 Oct 2025 23:05:13 +0200 Subject: [PATCH] chore: add yolo back --- internal/agent/agent.go | 3 +++ internal/agent/common_test.go | 2 +- internal/agent/coordinator.go | 2 +- internal/agent/event.go | 1 + 4 files changed, 6 insertions(+), 2 deletions(-) diff --git a/internal/agent/agent.go b/internal/agent/agent.go index 27626fe90292ca7c14db5414571430268b089821..d9a5ee837fe9b71da2bede2828e5720bead020e2 100644 --- a/internal/agent/agent.go +++ b/internal/agent/agent.go @@ -73,6 +73,7 @@ type sessionAgent struct { sessions session.Service messages message.Service disableAutoSummarize bool + isYolo bool messageQueue *csync.Map[string, []SessionAgentCall] activeRequests *csync.Map[string, context.CancelFunc] @@ -83,6 +84,7 @@ type SessionAgentOptions struct { SmallModel Model SystemPrompt string DisableAutoSummarize bool + IsYolo bool Sessions session.Service Messages message.Service Tools []fantasy.AgentTool @@ -99,6 +101,7 @@ func NewSessionAgent( messages: opts.Messages, disableAutoSummarize: opts.DisableAutoSummarize, tools: opts.Tools, + isYolo: opts.IsYolo, messageQueue: csync.NewMap[string, []SessionAgentCall](), activeRequests: csync.NewMap[string, context.CancelFunc](), } diff --git a/internal/agent/common_test.go b/internal/agent/common_test.go index c74ea744c3a7af46d1271cf33ec465270ea886d8..84300cf8cff06bad554af233576896a4bddcf6e6 100644 --- a/internal/agent/common_test.go +++ b/internal/agent/common_test.go @@ -145,7 +145,7 @@ func testSessionAgent(env env, large, small fantasy.LanguageModel, systemPrompt DefaultMaxTokens: 10000, }, } - agent := NewSessionAgent(SessionAgentOptions{largeModel, smallModel, systemPrompt, false, env.sessions, env.messages, tools}) + agent := NewSessionAgent(SessionAgentOptions{largeModel, smallModel, systemPrompt, false, true, env.sessions, env.messages, tools}) return agent } diff --git a/internal/agent/coordinator.go b/internal/agent/coordinator.go index 1c05493cd363fc54a0dbc09d1ac9c95f0406afa5..2458f3bb0fdf1b089aaa9b312ffea92514223d3e 100644 --- a/internal/agent/coordinator.go +++ b/internal/agent/coordinator.go @@ -280,7 +280,7 @@ func (c *coordinator) buildAgent(ctx context.Context, prompt *prompt.Prompt, age if err != nil { return nil, err } - return NewSessionAgent(SessionAgentOptions{large, small, systemPrompt, c.cfg.Options.DisableAutoSummarize, c.sessions, c.messages, tools}), nil + return NewSessionAgent(SessionAgentOptions{large, small, systemPrompt, c.cfg.Options.DisableAutoSummarize, c.permissions.SkipRequests(), c.sessions, c.messages, tools}), nil } func (c *coordinator) buildTools(ctx context.Context, agent config.Agent) ([]fantasy.AgentTool, error) { diff --git a/internal/agent/event.go b/internal/agent/event.go index 87e56e50344240fdd41d0f4740592edfd4e71347..bf36ec84bf4270bd2e63ae0efae0440474288565 100644 --- a/internal/agent/event.go +++ b/internal/agent/event.go @@ -46,5 +46,6 @@ func (a sessionAgent) eventCommon(sessionID string, model Model) []any { "model", m.Model, "reasoning effort", m.ReasoningEffort, "thinking mode", m.Think, + "yolo mode", a.isYolo, } }