literal keys instead of strings

deeppavlov · Nov 28, 2024 · aad2c49 · aad2c49
1 parent 61619e3
commit aad2c49
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chatsky/context_storages/database.py b/chatsky/context_storages/database.py
@@ -21,7 +21,7 @@
 from .protocol import PROTOCOLS
 
 _SUBSCRIPT_TYPE = Union[Literal["__all__"], int]
-_SUBSCRIPT_DICT = Dict[str, Union[_SUBSCRIPT_TYPE]]
+_SUBSCRIPT_DICT = Dict[Literal["labels", "requests", "responses"], Union[_SUBSCRIPT_TYPE]]
 
 logger = getLogger(__name__)