adjust protobuf hatch plugin and mypy

Signed-off-by: gruebel <[email protected]>

Author: gruebel

.github/workflows/build.yml

@@ -45,6 +45,11 @@ jobs:
         run: hatch build
         working-directory: ${{ matrix.package }}
 
+      - name: Type checking
+        if: matrix.package == 'providers/openfeature-provider-flagd'
+        working-directory: ${{ matrix.package }}
+        run: hatch run mypy
+
       - name: Test with pytest
         run: hatch test -c
         working-directory: ${{ matrix.package }}

.pre-commit-config.yaml

@@ -29,4 +29,4 @@ repos:
           - mmh3
           - semver
           - panzi-json-logic
-        exclude: proto|tests
+        exclude: providers/openfeature-provider-flagd|tests

mypy.ini

@@ -1,5 +1,5 @@
 [mypy]
-files = hooks,providers
+files = hooks,providers/openfeature-provider-ofrep
 exclude = proto|tests
 untyped_calls_exclude = flagd.proto
 

providers/openfeature-provider-flagd/pyproject.toml

@@ -17,7 +17,7 @@ classifiers = [
 ]
 keywords = []
 dependencies = [
-  "openfeature-sdk>=0.4.0",
+  "openfeature-sdk>=0.6.0",
   "grpcio>=1.60.0",
   "protobuf>=4.25.2",
   "mmh3>=4.1.0",
@@ -40,6 +40,9 @@ dependencies = [
   "testcontainers",
   "asserts",
   "grpcio-health-checking==1.60.0",
+  "mypy[faster-cache]>=1.13.0",
+  "types-protobuf",
+  "types-pyyaml",
 ]
 pre-install-commands = [
   "hatch build",
@@ -93,3 +96,30 @@ omit = [
   "src/openfeature/contrib/provider/flagd/proto/*",
   "tests/**",
 ]
+
+[tool.mypy]
+mypy_path = "src"
+files = "src"
+exclude = [
+  "src/openfeature/contrib/provider/flagd/protobuf"
+]
+
+namespace_packages = true
+explicit_package_bases = true
+local_partial_types = true
+pretty = true
+
+strict = true
+disallow_any_generics = false
+
+[[tool.mypy.overrides]]
+module = [
+    "grpc"
+]
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = [
+    "openfeature.contrib.provider.flagd.protobuf.*"
+]
+follow_imports = "silent"

providers/openfeature-provider-flagd/src/openfeature/contrib/provider/flagd/resolvers/grpc.py

@@ -20,6 +20,10 @@
 
 from ..config import Config
 from ..flag_type import FlagType
+from ..protobuf.flagd.evaluation.v1 import evaluation_pb2, evaluation_pb2_grpc
+
+if typing.TYPE_CHECKING:
+    from google.protobuf.message import Message
 
 T = typing.TypeVar("T")
 
@@ -31,7 +35,7 @@ def __init__(self, config: Config):
             grpc.secure_channel if self.config.tls else grpc.insecure_channel
         )
         self.channel = channel_factory(f"{self.config.host}:{self.config.port}")
-        self.stub = evaluation_pb2_grpc.ServiceStub(self.channel)
+        self.stub = evaluation_pb2_grpc.ServiceStub(self.channel)  # type: ignore[no-untyped-call]
 
     def shutdown(self) -> None:
         self.channel.close()
@@ -94,7 +98,7 @@ def _resolve(  # noqa: PLR0915
                 evaluation_pb2.ResolveFloatRequest,
             ]
             if flag_type == FlagType.BOOLEAN:
-                request = evaluation_pb2.ResolveBooleanRequest(
+                request: Message = evaluation_pb2.ResolveBooleanRequest(
                     flag_key=flag_key, context=context
                 )
                 response = self.stub.ResolveBoolean(request, **call_args)