Improve resolve() typing

Author: sobolevn

rodi/__init__.py

@@ -20,6 +20,7 @@
 from typing import (
     cast,
     get_type_hints,
+    overload,
 )
 
 T = TypeVar("T")
@@ -31,10 +32,16 @@ class ContainerProtocol(Protocol):
     and tell if a type is configured.
     """
 
-    def register(self, obj_type: Type | str, *args, **kwargs):
+    def register(self, obj_type: Type | str, *args, **kwargs) -> None:
         """Registers a type in the container, with optional arguments."""
 
-    def resolve(self, obj_type: Type[T] | str, *args, **kwargs) -> T:
+    @overload
+    def resolve(self, obj_type: Type[T], *args, **kwargs) -> T: ...
+
+    @overload
+    def resolve(self, obj_type: str, *args, **kwargs) -> Any: ...
+
+    def resolve(self, obj_type: Type[T] | str, *args, **kwargs) -> Any:
         """Activates an instance of the given type, with optional arguments."""
 
     def __contains__(self, item) -> bool:
@@ -934,13 +941,31 @@ def register(
             self.add_transient(obj_type, sub_type)
         return self
 
+    @overload
+    def resolve(
+        self,
+        obj_type: Type[T],
+        scope: Any = None,
+        *args,
+        **kwargs,
+    ) -> T: ...
+
+    @overload
+    def resolve(
+        self,
+        obj_type: str,
+        scope: Any = None,
+        *args,
+        **kwargs,
+    ) -> Any: ...
+
     def resolve(
         self,
         obj_type: Type[T] | str,
         scope: Any = None,
         *args,
         **kwargs,
-    ) -> T:
+    ) -> Any:
         """
         Resolves a service by type, obtaining an instance of that type.
         """