From 829142c808d0908b17051ffba539daae52266174 Mon Sep 17 00:00:00 2001 From: Pranav Date: Mon, 6 Oct 2025 22:51:18 -0700 Subject: [PATCH] fix: Update max_turns config (#12604) Pass max_turns in the run config than during the initialization. --------- Co-authored-by: Shivam Mishra --- config/initializers/ai_agents.rb | 1 - .../app/services/captain/assistant/agent_runner_service.rb | 2 +- .../services/captain/assistant/agent_runner_service_spec.rb | 3 ++- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/config/initializers/ai_agents.rb b/config/initializers/ai_agents.rb index 099d637ae..37bdd589f 100644 --- a/config/initializers/ai_agents.rb +++ b/config/initializers/ai_agents.rb @@ -15,7 +15,6 @@ Rails.application.config.after_initialize do config.openai_api_base = api_base end config.default_model = model - config.max_turns = 30 config.debug = false end end diff --git a/enterprise/app/services/captain/assistant/agent_runner_service.rb b/enterprise/app/services/captain/assistant/agent_runner_service.rb index 11a7dcad1..9c4e56841 100644 --- a/enterprise/app/services/captain/assistant/agent_runner_service.rb +++ b/enterprise/app/services/captain/assistant/agent_runner_service.rb @@ -23,7 +23,7 @@ class Captain::Assistant::AgentRunnerService message_to_process = extract_last_user_message(message_history) runner = Agents::Runner.with_agents(*agents) runner = add_callbacks_to_runner(runner) if @callbacks.any? - result = runner.run(message_to_process, context: context) + result = runner.run(message_to_process, context: context, max_turns: 100) process_agent_result(result) rescue StandardError => e diff --git a/spec/enterprise/services/captain/assistant/agent_runner_service_spec.rb b/spec/enterprise/services/captain/assistant/agent_runner_service_spec.rb index 4ee269c48..2c05860e2 100644 --- a/spec/enterprise/services/captain/assistant/agent_runner_service_spec.rb +++ b/spec/enterprise/services/captain/assistant/agent_runner_service_spec.rb @@ -90,7 +90,8 @@ RSpec.describe Captain::Assistant::AgentRunnerService do expect(mock_runner).to receive(:run).with( 'I need help with my account', - context: expected_context + context: expected_context, + max_turns: 100 ) service.generate_response(message_history: message_history)