fix: implementation, update schema

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

Author: leakonvalinka

providers/openfeature-provider-flagd/openfeature/schemas

@@ -6,7 +6,6 @@
 
 import grpc
 from google.protobuf.json_format import MessageToDict
-from google.protobuf.struct_pb2 import Struct
 
 from openfeature.evaluation_context import EvaluationContext
 from openfeature.event import ProviderEventDetails
@@ -210,6 +209,17 @@ def _create_request_args(self) -> dict:
 
         return request_args
 
+    def _fetch_metadata(self) -> typing.Optional[sync_pb2.GetMetadataResponse]:
+        if self.config.sync_metadata_disabled:
+            return None
+
+        context_values_request = sync_pb2.GetMetadataRequest()
+        try:
+            return self.stub.GetMetadata(context_values_request, wait_for_ready=True)
+        except grpc.RpcError as e:
+            logger.debug(f"Error getting sync metadata: {e}")
+            return None
+
     def listen(self) -> None:
         call_args = (
             {"timeout": self.streamline_deadline_seconds}
@@ -220,18 +230,9 @@ def listen(self) -> None:
 
         while self.active:
             try:
-                context_values_response: sync_pb2.GetMetadataResponse
-                if self.config.sync_metadata_disabled:
-                    context_values_response = sync_pb2.GetMetadataResponse(
-                        metadata=Struct()
-                    )
-                else:
-                    context_values_request = sync_pb2.GetMetadataRequest()
-                    context_values_response = self.stub.GetMetadata(
-                        context_values_request, wait_for_ready=True
-                    )
-
-                context_values = MessageToDict(context_values_response)
+                context_values_response: typing.Optional[
+                    sync_pb2.GetMetadataResponse
+                ] = self._fetch_metadata()
 
                 request = sync_pb2.SyncFlagsRequest(**request_args)
 
@@ -245,12 +246,20 @@ def listen(self) -> None:
                     )
                     self.flag_store.update(json.loads(flag_str))
 
+                    context_values = None
+                    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(
                             ProviderEventDetails(
                                 message="gRPC sync connection established"
                             ),
-                            context_values["metadata"],
+                            context_values,
                         )
                         self.connected = True