From 394ba4ff73284a6a7ac07e6bcd61db75b7da34cf Mon Sep 17 00:00:00 2001 From: Shivam Mishra Date: Tue, 15 Jul 2025 20:21:22 +0530 Subject: [PATCH] fix: arguments --- enterprise/app/models/concerns/agentable.rb | 4 ++-- .../app/services/captain/assistant/agent_runner_service.rb | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/enterprise/app/models/concerns/agentable.rb b/enterprise/app/models/concerns/agentable.rb index 2df4d5e18..4f98bde77 100644 --- a/enterprise/app/models/concerns/agentable.rb +++ b/enterprise/app/models/concerns/agentable.rb @@ -2,11 +2,11 @@ module Agentable extend ActiveSupport::Concern included do - def agent(user) + def agent Agents::Agent.new( name: agent_name, instructions: ->(context) { agent_instructions(context) }, - tools: agent_tools(user), + tools: agent_tools, model: agent_model ) end diff --git a/enterprise/app/services/captain/assistant/agent_runner_service.rb b/enterprise/app/services/captain/assistant/agent_runner_service.rb index dea53fa23..4ec3594e2 100644 --- a/enterprise/app/services/captain/assistant/agent_runner_service.rb +++ b/enterprise/app/services/captain/assistant/agent_runner_service.rb @@ -109,8 +109,8 @@ class Captain::Assistant::AgentRunnerService end def build_and_wire_agents - assistant_agent = @assistant.agent(nil) - scenario_agents = @assistant.scenarios.enabled.map { |scenario| scenario.agent(nil) } + assistant_agent = @assistant.agent + scenario_agents = @assistant.scenarios.enabled.map { |scenario| scenario.agent } assistant_agent.register_handoffs(*scenario_agents) if scenario_agents.any? scenario_agents.each { |scenario_agent| scenario_agent.register_handoffs(assistant_agent) }