From 8e45a71a155ee3ed8e37e5029be36b68173c7c57 Mon Sep 17 00:00:00 2001
From: Mike Salvatore <mike.s.salvatore@gmail.com>
Date: Thu, 22 Sep 2022 12:59:51 -0400
Subject: [PATCH] Island: Change agent parameter to agent_id in get_signals()

---
 monkey/monkey_island/cc/services/agent_signals_service.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/monkey/monkey_island/cc/services/agent_signals_service.py b/monkey/monkey_island/cc/services/agent_signals_service.py
index 47c677271..4a3c4ef74 100644
--- a/monkey/monkey_island/cc/services/agent_signals_service.py
+++ b/monkey/monkey_island/cc/services/agent_signals_service.py
@@ -1,18 +1,19 @@
 from datetime import datetime
 
-from monkey_island.cc.models import Agent, AgentSignals
+from common.types import AgentID
 from monkey_island.cc.repository import ISimulationRepository
+from monkey_island.cc.models import AgentSignals
 
 
 class AgentSignalsService:
     def __init__(self, simulation_repository: ISimulationRepository):
         self._simulation_repository = simulation_repository
 
-    def get_signals(self, agent: Agent) -> AgentSignals:
+    def get_signals(self, agent_id: AgentID) -> AgentSignals:
         """
         Gets the signals sent to a particular agent
 
-        :param agent: The agent whose signals need to be retrieved
+        :param agent_id: The ID of the agent whose signals need to be retrieved
         :return: Signals sent to the relevant agent
         """
         return AgentSignals(timestamp=datetime.now())