From 1f04c35627124d6997f61c057951cd9942909d49 Mon Sep 17 00:00:00 2001
From: MilTy <shveitsar215@gmail.com>
Date: Tue, 28 Jan 2025 04:32:48 +0100
Subject: [PATCH] overloaded methods added

---
 chatsky/core/ctx_dict.py | 32 ++++++++++++++++++++++----------
 1 file changed, 22 insertions(+), 10 deletions(-)

diff --git a/chatsky/core/ctx_dict.py b/chatsky/core/ctx_dict.py
index c0d75bb2e..6d821eac8 100644
--- a/chatsky/core/ctx_dict.py
+++ b/chatsky/core/ctx_dict.py
@@ -396,6 +396,9 @@ async def __getitem__(self, key: int) -> AbsoluteNodeLabel: ...  # noqa: E704
     @overload
     async def __getitem__(self, key: slice) -> List[AbsoluteNodeLabel]: ...  # noqa: E704
 
+    async def __getitem__(self, key):
+        return await super().__getitem__(key)
+
     def __setitem__(self, key: Union[int, slice], value: Union[AbsoluteNodeLabel, Sequence[AbsoluteNodeLabel]]) -> None:
         return super().__setitem__(key, value)
 
@@ -405,20 +408,23 @@ async def get(self, key: int, default=None) -> AbsoluteNodeLabel: ...  # noqa: E
     @overload
     async def get(self, key: Iterable[int], default=None) -> List[AbsoluteNodeLabel]: ...  # noqa: E704
 
+    async def get(self, key, default=None):
+        return await super().get(key, default)
+
     async def values(self) -> List[AbsoluteNodeLabel]:
-        return super().values()
+        return await super().values()
 
     async def items(self) -> List[Tuple[int, AbsoluteNodeLabel]]:
-        return super().items()
+        return await super().items()
 
     async def pop(self, key: int, default=None) -> AbsoluteNodeLabel:
-        return super().pop(key, default)
+        return await super().pop(key, default)
 
     async def popitem(self) -> Tuple[int, AbsoluteNodeLabel]:
-        return super().popitem()
+        return await super().popitem()
 
     async def setdefault(self, key: int, default=None) -> AbsoluteNodeLabel:
-        return super().setdefault(key, default)
+        return await super().setdefault(key, default)
 
     @model_serializer()
     def _serialize_model(self) -> Dict[int, AbsoluteNodeLabel]:
@@ -442,6 +448,9 @@ async def __getitem__(self, key: int) -> Message: ...  # noqa: E704
     @overload
     async def __getitem__(self, key: slice) -> List[Message]: ...  # noqa: E704
 
+    async def __getitem__(self, key):
+        return await super().__getitem__(key)
+
     def __setitem__(self, key: Union[int, slice], value: Union[Message, Sequence[Message]]) -> None:
         return super().__setitem__(key, value)
 
@@ -451,20 +460,23 @@ async def get(self, key: int, default=None) -> Message: ...  # noqa: E704
     @overload
     async def get(self, key: Iterable[int], default=None) -> List[Message]: ...  # noqa: E704
 
+    async def get(self, key, default=None):
+        return await super().get(key, default)
+
     async def values(self) -> List[Message]:
-        return super().values()
+        return await super().values()
 
     async def items(self) -> List[Tuple[int, Message]]:
-        return super().items()
+        return await super().items()
 
     async def pop(self, key: int, default=None) -> Message:
-        return super().pop(key, default)
+        return await super().pop(key, default)
 
     async def popitem(self) -> Tuple[int, Message]:
-        return super().popitem()
+        return await super().popitem()
 
     async def setdefault(self, key: int, default=None) -> Message:
-        return super().setdefault(key, default)
+        return await super().setdefault(key, default)
 
     @model_serializer()
     def _serialize_model(self) -> Dict[int, Message]: