Working Github OIDC code example

Author: emmyzhou-db

examples/docs/src/main/java/com/databricks/example/GithubOIDCAuthExample.java

@@ -1,43 +1,36 @@
-package com.databricks.sdk.examples;
+package com.databricks.example;
 
+import com.databricks.sdk.AccountClient;
 import com.databricks.sdk.core.DatabricksConfig;
-import com.databricks.sdk.WorkspaceClient;
 import com.databricks.sdk.service.iam.User;
+import com.databricks.sdk.service.iam.ListAccountGroupsRequest;
+import com.databricks.sdk.service.iam.Group;
 
 /**
  * Example demonstrating how to use GitHub OIDC authentication with Databricks.
- * 
- * This example assumes you have:
- * 1. A Databricks workspace
- * 2. A service principal configured with GitHub OIDC federation
- * 3. The following environment variables set:
- *    - DATABRICKS_HOST: Your Databricks workspace URL
- *    - DATABRICKS_CLIENT_ID: Your service principal's client ID
- *    - ACTIONS_ID_TOKEN_REQUEST_URL: GitHub Actions token request URL
- *    - ACTIONS_ID_TOKEN_REQUEST_TOKEN: GitHub Actions token request token
  */
 public class GithubOIDCAuthExample {
+
     public static void main(String[] args) {
         // Create Databricks configuration with GitHub OIDC authentication
         DatabricksConfig config = new DatabricksConfig()
-            .setAuthType("github-oidc");
+        .setAuthType("github-oidc")
+        .setHost("https://accounts.cloud.databricks.com/")
+        .setAccountId("968367da-7edd-44f7-9dea-3e0b20b0ec97")
+        .setClientId("dd3b5d58-5a6e-45e3-a3ae-81d8f89fc6fd");
 
-        // Create a workspace client
-        WorkspaceClient workspace = new WorkspaceClient(config);
+        AccountClient account = new AccountClient(config);
 
         try {
-            // Test the authentication by getting the current User
-            User currentUser = workspace.currentUser().me();
-            System.out.println("Successfully authenticated as: " + currentUser.getUserName());
-            
-            // You can add more API calls here to test other functionality
-            // For example:
-            // var clusters = workspace.clusters().list();
-            // System.out.println("Available clusters: " + clusters);
-            
+            System.out.println("\nListing account groups:");
+            Iterable<Group> groups = account.groups().list(new ListAccountGroupsRequest());
+            for (Group group : groups) {
+                System.out.println("- Group: " + group.getDisplayName() + " (ID: " + group.getId() + ")");
+            }
+
         } catch (Exception e) {
             System.err.println("Authentication failed: " + e.getMessage());
             e.printStackTrace();
         }
     }
-} 
+}