fix

Author: madhav-db

.github/workflows/token-federation-test.yml

@@ -191,7 +191,7 @@ jobs:
           OIDC_TOKEN: ${{ steps.get-id-token.outputs.token }}
         run: |
           echo "Attempting direct token exchange with curl..."
-          echo "Host: $DATABRICKS_HOST"
+          echo "Host: $DATABRICKS_HOST_FOR_TF"
           echo "Client ID: $IDENTITY_FEDERATION_CLIENT_ID"
           
           # Debug token claims before making the request
@@ -227,7 +227,7 @@ jobs:
           curl_data=$(eval echo "$curl_data")
           
           # Print request details (except the token)
-          echo "Request URL: https://$DATABRICKS_HOST/oidc/v1/token"
+          echo "Request URL: https://$DATABRICKS_HOST_FOR_TF/oidc/v1/token"
           echo "Request data: $(echo "$curl_data" | sed 's/subject_token=.*&/subject_token=REDACTED&/')"
           
           # Make the request with detailed info
@@ -349,7 +349,7 @@ jobs:
               # Get Databricks connection parameters
               host = os.environ.get("DATABRICKS_HOST_FOR_TF")
               http_path = os.environ.get("DATABRICKS_HTTP_PATH_FOR_TF")
-              identity_federation_client_id = os.environ.get("IDENTITY_FEDERATION_CLIENT_ID_FOR_TF")
+              identity_federation_client_id = os.environ.get("IDENTITY_FEDERATION_CLIENT_ID")
               
               if not host or not http_path:
                   print("Missing Databricks connection parameters")