765a9cc · crush · git.iko.soy

internal/agent/agent.go+3

compare with main

unifiedsplit

@@ -73,6 +73,7 @@ type sessionAgent struct {

73 sessions session.Service

74 messages message.Service

75 disableAutoSummarize bool

76+ isYolo bool

76

77

77 messageQueue *csync.Map[string, []SessionAgentCall]

78 messageQueue *csync.Map[string, []SessionAgentCall]

78 activeRequests *csync.Map[string, context.CancelFunc]

79 activeRequests *csync.Map[string, context.CancelFunc]

@@ -83,6 +84,7 @@ type SessionAgentOptions struct {

83 SmallModel Model

84 SmallModel Model

84 SystemPrompt string

85 SystemPrompt string

85 DisableAutoSummarize bool

86 DisableAutoSummarize bool

87+ IsYolo bool

86 Sessions session.Service

88 Sessions session.Service

87 Messages message.Service

89 Messages message.Service

88 Tools []fantasy.AgentTool

90 Tools []fantasy.AgentTool

@@ -99,6 +101,7 @@ func NewSessionAgent(

99 messages: opts.Messages,

101 messages: opts.Messages,

100 disableAutoSummarize: opts.DisableAutoSummarize,

102 disableAutoSummarize: opts.DisableAutoSummarize,

101 tools: opts.Tools,

103 tools: opts.Tools,

104+ isYolo: opts.IsYolo,

102 messageQueue: csync.NewMap[string, []SessionAgentCall](),

105 messageQueue: csync.NewMap[string, []SessionAgentCall](),

103 activeRequests: csync.NewMap[string, context.CancelFunc](),

106 activeRequests: csync.NewMap[string, context.CancelFunc](),

104 }

107 }

internal/agent/common_test.go+1−1

@@ -145,7 +145,7 @@ func testSessionAgent(env env, large, small fantasy.LanguageModel, systemPrompt

145 DefaultMaxTokens: 10000,

146 },

147 }

148− agent := NewSessionAgent(SessionAgentOptions{largeModel, smallModel, systemPrompt, false, env.sessions, env.messages, tools})

148+ agent := NewSessionAgent(SessionAgentOptions{largeModel, smallModel, systemPrompt, false, true, env.sessions, env.messages, tools})

149 return agent

150}

151

internal/agent/coordinator.go+1−1

@@ -280,7 +280,7 @@ func (c *coordinator) buildAgent(ctx context.Context, prompt *prompt.Prompt, age

280 if err != nil {

281 return nil, err

282 }

283− return NewSessionAgent(SessionAgentOptions{large, small, systemPrompt, c.cfg.Options.DisableAutoSummarize, c.sessions, c.messages, tools}), nil

283+ return NewSessionAgent(SessionAgentOptions{large, small, systemPrompt, c.cfg.Options.DisableAutoSummarize, c.permissions.SkipRequests(), c.sessions, c.messages, tools}), nil

284}

285

286func (c *coordinator) buildTools(ctx context.Context, agent config.Agent) ([]fantasy.AgentTool, error) {

internal/agent/event.go+1

@@ -46,5 +46,6 @@ func (a sessionAgent) eventCommon(sessionID string, model Model) []any {

46 "model", m.Model,

47 "reasoning effort", m.ReasoningEffort,

48 "thinking mode", m.Think,

49+ "yolo mode", a.isYolo,

49 }

50 }

50}

51}