Agent: Fix type hint in HTTPIslandAPIClient.send_log()

This commit is contained in:
Shreya Malviya 2022-09-28 17:45:26 +05:30 committed by Mike Salvatore
parent 29c08ff40c
commit 8dc8a516d5
1 changed files with 2 additions and 2 deletions

View File

@ -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,