diff options
Diffstat (limited to 'lib/gitlab/kas/client.rb')
-rw-r--r-- | lib/gitlab/kas/client.rb | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/lib/gitlab/kas/client.rb b/lib/gitlab/kas/client.rb index fe244bd88a0..464c049ee52 100644 --- a/lib/gitlab/kas/client.rb +++ b/lib/gitlab/kas/client.rb @@ -19,13 +19,13 @@ module Gitlab raise ConfigurationError, 'KAS internal URL is not configured' unless Gitlab::Kas.internal_url.present? end - def get_connected_agents(project:) - request = Gitlab::Agent::AgentTracker::Rpc::GetConnectedAgentsRequest.new(project_id: project.id) + def get_connected_agents_by_agent_ids(agent_ids:) + request = Gitlab::Agent::AgentTracker::Rpc::GetConnectedAgentsByAgentIdsRequest.new(agent_ids: agent_ids) stub_for(:agent_tracker) - .get_connected_agents(request, metadata: metadata) - .agents - .to_a + .get_connected_agents_by_agent_ids(request, metadata: metadata) + .agents + .to_a end def list_agent_config_files(project:) |