Agent: Fix type hint in IIslandAPIClient.send_log()

This commit is contained in:
Shreya Malviya 2022-09-28 17:47:04 +05:30 committed by Mike Salvatore
parent 8dc8a516d5
commit 21f01292f7
1 changed files with 2 additions and 2 deletions

View File

@ -5,7 +5,7 @@ from common import AgentRegistrationData, AgentSignals, OperatingSystem
from common.agent_configuration import AgentConfiguration
from common.agent_events import AbstractAgentEvent
from common.credentials import Credentials
from common.types import SocketAddress
from common.types import AgentID, SocketAddress
class IIslandAPIClient(ABC):
@ -30,7 +30,7 @@ class IIslandAPIClient(ABC):
"""
@abstractmethod
def send_log(self, agent_id: int, log_contents: str):
def send_log(self, agent_id: AgentID, log_contents: str):
"""
Send the contents of the agent's log to the island