kujtimiihoxha пре 3 месеци
родитељ
комит
765a9cc222
4 измењених фајлова са 6 додато и 2 уклоњено
  1. 3 0
      internal/agent/agent.go
  2. 1 1
      internal/agent/common_test.go
  3. 1 1
      internal/agent/coordinator.go
  4. 1 0
      internal/agent/event.go

+ 3 - 0
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](),
 	}

+ 1 - 1
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
 }
 

+ 1 - 1
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) {

+ 1 - 0
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,
 	}
 }