From f70035a5a7580491c02a54f636ffd0dfc049cf24 Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Wed, 1 Apr 2026 11:30:52 +0000
Subject: [PATCH 1/2] Initial plan


From a6fc301f0e3196c10dc4e378f52a4691913ca7c9 Mon Sep 17 00:00:00 2001
From: "copilot-swe-agent[bot]" <198982749+Copilot@users.noreply.github.com>
Date: Wed, 1 Apr 2026 11:35:55 +0000
Subject: [PATCH 2/2] Fix: add env var override and key validation in
 setup_eventhub_connection

Agent-Logs-Url: https://github.com/microsoft/real-time-intelligence-operations-solution-accelerator/sessions/8e4a1c2b-b566-4dd9-914d-8fa732f98757

Co-authored-by: Yatish-Microsoft <234036280+Yatish-Microsoft@users.noreply.github.com>
---
 infra/scripts/fabric/fabric_eventhub.py | 46 +++++++++++++++++++------
 1 file changed, 35 insertions(+), 11 deletions(-)

diff --git a/infra/scripts/fabric/fabric_eventhub.py b/infra/scripts/fabric/fabric_eventhub.py
index d812531..ee75879 100644
--- a/infra/scripts/fabric/fabric_eventhub.py
+++ b/infra/scripts/fabric/fabric_eventhub.py
@@ -27,6 +27,7 @@
 """
 
 import argparse
+import os
 from azure.identity import AzureCliCredential
 from azure.mgmt.eventhub import EventHubManagementClient
 from fabric_api import FabricApiClient, FabricApiError
@@ -140,7 +141,7 @@ def get_event_hub_namespace_primary_key(namespace_name: str, subscription_id: st
         return result
         
     except Exception as e:
-        print(f"❌ Error: {e}")
+        print(f"❌ Failed to retrieve namespace access keys for '{namespace_name}': {e}")
         raise
 
 
@@ -178,16 +179,39 @@ def setup_eventhub_connection(
         Exception: If connection creation/update fails
     """
     try:
-        # Retrieve access key automatically
-        print(f"🔑 Retrieving access key for Event Hub: {event_hub_name}")
-        key_info = get_event_hub_namespace_primary_key(
-            namespace_name=namespace_name,
-            subscription_id=subscription_id,
-            resource_group_name=resource_group_name,
-            authorization_rule_name=authorization_rule_name
-        )
-        access_key = key_info['primary_key']
-        print(f"✅ Successfully retrieved access key")
+        # Allow an explicit key override via environment variable (useful in CI when listKeys RBAC is unavailable)
+        access_key = os.getenv("AZURE_EVENT_HUB_SHARED_ACCESS_KEY")
+        if access_key and access_key.strip():
+            print("🔑 Using AZURE_EVENT_HUB_SHARED_ACCESS_KEY from environment (override).")
+            access_key = access_key.strip()
+        else:
+            # Retrieve access key from Azure management API
+            print(f"🔑 Retrieving access key for Event Hub namespace: {namespace_name}")
+            try:
+                key_info = get_event_hub_namespace_primary_key(
+                    namespace_name=namespace_name,
+                    subscription_id=subscription_id,
+                    resource_group_name=resource_group_name,
+                    authorization_rule_name=authorization_rule_name
+                )
+                access_key = (key_info or {}).get("primary_key")
+            except Exception as key_error:
+                raise RuntimeError(
+                    f"Failed to retrieve Event Hub shared access key for namespace '{namespace_name}': {key_error}. "
+                    "Either set the AZURE_EVENT_HUB_SHARED_ACCESS_KEY environment variable, or grant the deployment "
+                    "identity permission to list Event Hub namespace keys "
+                    "(Microsoft.EventHub/namespaces/authorizationRules/listKeys/action)."
+                ) from key_error
+
+        if not access_key or not access_key.strip():
+            raise RuntimeError(
+                "Event Hub shared access key is missing or empty. "
+                "Either set the AZURE_EVENT_HUB_SHARED_ACCESS_KEY environment variable, or grant the deployment "
+                "identity permission to list Event Hub namespace keys "
+                "(Microsoft.EventHub/namespaces/authorizationRules/listKeys/action)."
+            )
+
+        print("✅ Successfully obtained Event Hub shared access key.")
         
         # Use the passed fabric_client instead of creating a new one
         client = fabric_client