fix annotations for SmartAppToMessage

salute-developers · Oct 4, 2023 · 64282d6 · 64282d6
1 parent a5cd58a
commit 64282d6
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/smart_kit/message/smartapp_to_message.py b/smart_kit/message/smartapp_to_message.py
@@ -1,14 +1,18 @@
+from __future__ import annotations
+
 from functools import cached_property
-from typing import Iterable, Optional, List
+from typing import TYPE_CHECKING, Iterable, Optional, List
 import json
 from copy import copy
 
-from core.basic_models.actions.command import Command
 from core.utils.masking_message import masking
-from core.message.msg_validator import MessageValidator
-from smart_kit.request.kafka_request import SmartKitKafkaRequest
 from smart_kit.utils import SmartAppToMessage_pb2
 
+if TYPE_CHECKING:
+    from core.basic_models.actions.command import Command
+    from core.message.msg_validator import MessageValidator
+    from smart_kit.request.kafka_request import SmartKitKafkaRequest
+
 
 class SmartAppToMessage:
     ROOT_NODES_KEY = "root_nodes"