fix: type issues

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

Author: leakonvalinka

providers/openfeature-provider-flagd/src/openfeature/contrib/provider/flagd/resolvers/process/connector/grpc_watcher.py

@@ -210,21 +210,21 @@ def _create_request_args(self) -> dict:
 
         return request_args
 
-    def _fetch_metadata(self) -> dict:
+    def _fetch_metadata(self) -> typing.Optional[sync_pb2.GetMetadataResponse]:
         if self.config.sync_metadata_disabled:
-            return {}
+            return None
 
         context_values_request = sync_pb2.GetMetadataRequest()
         context_values_response: sync_pb2.GetMetadataResponse
         try:
             context_values_response = self.stub.GetMetadata(
                 context_values_request, wait_for_ready=True
             )
-            return MessageToDict(context_values_response)["metadata"]
+            return context_values_response
         except grpc.RpcError as e:
             if e.code() == StatusCode.UNIMPLEMENTED:
-                logger.debug("Metadata endpoint disabled")
-                return {}
+                logger.debug(f"Error getting sync metadata: {e}")
+                return None
             else:
                 raise e
 
@@ -238,7 +238,7 @@ def listen(self) -> None:
 
         while self.active:
             try:
-                context_values = self._fetch_metadata()
+                context_values_response = self._fetch_metadata()
 
                 request = sync_pb2.SyncFlagsRequest(**request_args)
 
@@ -252,8 +252,13 @@ def listen(self) -> None:
                     )
                     self.flag_store.update(json.loads(flag_str))
 
+                    context_values = {}
                     if flag_rsp.sync_context:
                         context_values = MessageToDict(flag_rsp.sync_context)
+                    elif context_values_response:
+                        context_values = MessageToDict(context_values_response)[
+                            "metadata"
+                        ]
 
                     if not self.connected:
                         self.emit_provider_ready(