From 8dc8a516d52849371672d68a5f7ca9bac67d099a Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Wed, 28 Sep 2022 17:45:26 +0530 Subject: [PATCH] Agent: Fix type hint in HTTPIslandAPIClient.send_log() --- .../island_api_client/http_island_api_client.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/monkey/infection_monkey/island_api_client/http_island_api_client.py b/monkey/infection_monkey/island_api_client/http_island_api_client.py index 7c7215589..2c0e2319a 100644 --- a/monkey/infection_monkey/island_api_client/http_island_api_client.py +++ b/monkey/infection_monkey/island_api_client/http_island_api_client.py @@ -16,7 +16,7 @@ from common.common_consts.timeouts import ( SHORT_REQUEST_TIMEOUT, ) from common.credentials import Credentials -from common.types import SocketAddress +from common.types import AgentID, SocketAddress from . import ( AbstractIslandAPIClientFactory, @@ -92,7 +92,7 @@ class HTTPIslandAPIClient(IIslandAPIClient): self._api_url = f"https://{island_server}/api" @handle_island_errors - def send_log(self, agent_id: int, log_contents: str): + def send_log(self, agent_id: AgentID, log_contents: str): response = requests.post( # noqa: DUO123 f"{self._api_url}/agent-logs/{agent_id}", json=log_contents,