typing

Author: ElectricalBoy

src/lpdb/async_session/async_session.py

@@ -5,7 +5,7 @@
 
 import aiohttp
 
-from ..session import AbstractLpdbSession, LpdbError
+from ..session import AbstractLpdbSession, LpdbDataType, LpdbError
 from ..defs import TeamTemplate
 
 __all__ = ["AsyncLpdbSession"]
@@ -70,7 +70,7 @@ async def __handle_response(
     @override
     async def make_request(
         self,
-        lpdb_datatype,
+        lpdb_datatype: LpdbDataType,
         wiki: str | list[str],
         limit: int = 20,
         offset: int = 0,

src/lpdb/session.py

@@ -127,7 +127,7 @@ def get_wikis() -> set[str]:
     @abstractmethod
     def make_request(
         self,
-        lpdb_datatype,
+        lpdb_datatype: LpdbDataType,
         wiki: str | list[str],
         limit: int = 20,
         offset: int = 0,
@@ -155,7 +155,7 @@ def make_request(
 
     def make_count_request(
         self,
-        lpdb_datatype,
+        lpdb_datatype: LpdbDataType,
         wiki: str,
         conditions: Optional[str] = None,
     ) -> int:
@@ -288,7 +288,7 @@ def __handle_response(response: requests.Response) -> list[dict[str, Any]]:
     @override
     def make_request(
         self,
-        lpdb_datatype: str,
+        lpdb_datatype: LpdbDataType,
         wiki: str | list[str],
         limit: int = 20,
         offset: int = 0,