diff --git a/firebase_admin/dataconnect.py b/firebase_admin/dataconnect.py index e9ef1aa2..63993b79 100644 --- a/firebase_admin/dataconnect.py +++ b/firebase_admin/dataconnect.py @@ -361,6 +361,9 @@ def _get_headers(self) -> Dict[str, str]: "X-Firebase-Client": f"fire-admin-python/{firebase_admin.__version__}", "x-goog-api-client": _utils.get_metrics_header(), "X-Client-Version": f"python/{firebase_admin.__version__}", + "X-Firebase-Sqlconnect-Affinity": ( + f"{self._project_id}{self._connector_config.service_id}" + ), } @staticmethod diff --git a/tests/test_data_connect.py b/tests/test_data_connect.py index 255ce005..5da25811 100644 --- a/tests/test_data_connect.py +++ b/tests/test_data_connect.py @@ -741,6 +741,7 @@ def test_get_headers(self): assert headers.get("X-Firebase-Client") == f"fire-admin-python/{firebase_admin.__version__}" assert headers.get("x-goog-api-client") == _utils.get_metrics_header() assert headers.get("X-Client-Version") == f"python/{firebase_admin.__version__}" + assert headers.get("X-Firebase-Sqlconnect-Affinity") == "test-projectstarterproject" class TestDataConnectApiClientMakeGqlRequest: