feat: ✨ CLI: Add aws credentials onboarding

Author: chezsmithy

extensions/cli/spec/onboarding.md

@@ -7,10 +7,11 @@ When a user first runs `cn` in interactive mode, they will be taken through "onb
 **The onboarding flow runs when the user hasn't completed onboarding before, regardless of whether they have a valid config.yaml file.**
 
 1. If the --config flag is provided, load this config
-2. Otherwise, we will present the user with two options:
-
+2. Otherwise, check if AWS credentials are available through the standard AWS credential chain
+3. Present the user with available options:
    - Log in with Continue: log them in, which will automatically create their assistant and then we can load the first assistant from the first org
    - Enter your Anthropic API key: let them enter the key, and then either create a ~/.continue/config.yaml with the following contents OR update the existing config.yaml to add the model
+   - Use AWS credentials: (only shown if AWS credentials are available) automatically create a config using AWS Bedrock Claude and skip further onboarding
 
    ```yaml
    name: Local Config
@@ -23,6 +24,19 @@ When a user first runs `cn` in interactive mode, they will be taken through "onb
          ANTHROPIC_API_KEY: <THEIR_ANTHROPIC_API_KEY>
    ```
 
+   For the AWS option, it will create/update the config with:
+
+   ```yaml
+   name: Local Config
+   version: 1.0.0
+   schema: v1
+
+   models:
+     - uses: bedrock/anthropic.claude-3-5-sonnet-20241022-v2:0
+       with:
+         region: us-east-1
+   ```
+
 When something in the onboarding flow is done automatically, we should tell the user what happened.
 
 ## Normal flow

extensions/cli/src/onboarding.ts

@@ -16,7 +16,9 @@ import {
   getApiKeyValidationError,
   isValidAnthropicApiKey,
 } from "./util/apiKeyValidation.js";
-import { updateAnthropicModelInYaml } from "./util/yamlConfigUpdater.js";
+import {
+  updateAnthropicModelInYaml,
+} from "./util/yamlConfigUpdater.js";
 
 const CONFIG_PATH = path.join(env.continueHome, "config.yaml");
 
@@ -92,14 +94,18 @@ async function runOnboardingFlow(
     return { ...result, wasOnboarded: false };
   }
 
-  // Step 3: Present user with two options
+  // Step 3: Present user with three options
   console.log(chalk.yellow("How do you want to get started?"));
   console.log(chalk.white("1. ⏩ Log in with Continue"));
   console.log(chalk.white("2. 🔑 Enter your Anthropic API key"));
+  console.log(chalk.white("3. ☁️  Use AWS credentials"));
 
+  const validChoices = ["1", "2", "3", ""];
   const choice = readlineSync.question(chalk.yellow("\nEnter choice (1): "), {
-    limit: ["1", "2", ""],
-    limitMessage: chalk.dim("Please enter 1 or 2"),
+    limit: validChoices,
+    limitMessage: chalk.dim(
+      "Please enter 1, 2, or 3",
+    ),
   });
 
   if (choice === "1" || choice === "") {
@@ -124,10 +130,17 @@ async function runOnboardingFlow(
       chalk.green(`✓ Config file updated successfully at ${CONFIG_PATH}`),
     );
 
+    const result = await initialize(authConfig, CONFIG_PATH);
+    return { ...result, wasOnboarded: true };
+  } else if (choice === "3") {
+    console.log(chalk.blue("✓ Using AWS credentials"));
+
     const result = await initialize(authConfig, CONFIG_PATH);
     return { ...result, wasOnboarded: true };
   } else {
-    throw new Error("Invalid choice. Please select 1 or 2.");
+    throw new Error(
+      `Invalid choice. Please select "1, 2, or 3"`,
+    );
   }
 }