fixup: changing to mypy-protobuf

Signed-off-by: Simon Schrottner <[email protected]>

Author: aepfli

.github/workflows/build.yml

@@ -38,16 +38,14 @@ jobs:
           python-version: ${{ matrix.python-version }}
           cache: "pip"
 
-      - uses: bufbuild/buf-action@v1
-        with:
-          github_token: ${{ github.token }}
-          setup_only: true
+      - name: Install Hatch
+        uses: pypa/hatch@install
 
-      - name: Install hatch
-        run: pip install hatch
+      - name: Building first to generate files
+        run: hatch build
 
       - name: Test with pytest
-        run: hatch run cov
+        run: hatch test -c
         working-directory: ${{ matrix.package }}
 
       - if: matrix.python-version == '3.11'

providers/openfeature-provider-flagd/pyproject.toml

@@ -32,7 +32,7 @@ Homepage = "https://github.com/open-feature/python-sdk-contrib"
 
 [tool.hatch]
 
-[tool.hatch.envs.default]
+[tool.hatch.envs.hatch-test]
 dependencies = [
   "coverage[toml]>=6.5",
   "pytest",
@@ -41,28 +41,49 @@ dependencies = [
   "asserts",
   "grpcio-health-checking==1.60.0",
 ]
+pre-install-commands = [
+  "hatch build",
+]
 
-[tool.hatch.envs.default.scripts]
-test = "pytest {args:tests}"
-test-cov = "coverage run -m pytest {args:tests}"
+[tool.hatch.envs.hatch-test.scripts]
+run = "pytest {args:tests}"
+run-cov = "coverage run -m pytest {args:tests}"
+cov-combine = "coverage combine"
 cov-report = [
   "coverage xml",
   "coverage html",
+  "coverage report",
 ]
 cov = [
   "test-cov",
   "cov-report",
 ]
 
 [tool.hatch.build.hooks.protobuf]
-dependencies = ["hatch-protobuf"]
-paths = ["schemas/protobuf/schema/v1/schema.proto", "schemas/protobuf/sync/v1/sync_service.proto"]
-output_path = "src"
+generate_pyi = false
+dependencies = [
+    "hatch-protobuf",
+    "mypy-protobuf~=3.0",
+]
+proto_paths = [
+  "schemas/protobuf/",
+]
+output_path = "src/"
+
+[[tool.hatch.build.hooks.protobuf.generators]]
+name = "mypy"
+outputs = ["{proto_path}/{proto_name}_pb2.pyi"]
+
+[[tool.hatch.build.hooks.protobuf.generators]]
+name = "mypy_grpc"
+outputs = ["{proto_path}/{proto_name}_pb2_grpc.pyi"]
 
 [tool.hatch.build.targets.sdist]
 exclude = [
   ".gitignore",
   "schemas",
+  "src/schema",
+  "src/sync",
 ]
 
 [tool.hatch.build.targets.wheel]

providers/openfeature-provider-flagd/src/.gitignore

@@ -0,0 +1,3 @@
+flagd
+schema
+sync

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

@@ -3,11 +3,11 @@
 import grpc
 from google.protobuf.json_format import MessageToDict
 from google.protobuf.struct_pb2 import Struct
-from schemas.protobuf.flagd.evaluation.v1 import (  # type:ignore[import-not-found]
+
+from flagd.evaluation.v1 import (  # type:ignore[import-not-found]
     evaluation_pb2,
     evaluation_pb2_grpc,
 )
-
 from openfeature.evaluation_context import EvaluationContext
 from openfeature.exception import (
     FlagNotFoundError,
@@ -86,6 +86,13 @@ def _resolve(  # noqa: PLR0915
         context = self._convert_context(evaluation_context)
         call_args = {"timeout": self.config.timeout}
         try:
+            request: typing.Union[
+                evaluation_pb2.ResolveBooleanRequest,
+                evaluation_pb2.ResolveIntRequest,
+                evaluation_pb2.ResolveStringRequest,
+                evaluation_pb2.ResolveObjectRequest,
+                evaluation_pb2.ResolveFloatRequest,
+            ]
             if flag_type == FlagType.BOOLEAN:
                 request = evaluation_pb2.ResolveBooleanRequest(
                     flag_key=flag_key, context=context