@amplitude/wizard 1.0.0-beta.0 → 1.0.0-beta.2

package/dist/src/lib/feature-flags.js

@@ -0,0 +1,119 @@
+"use strict";
+/**
+ * Feature flags powered by Amplitude Experiment (server-side local evaluation).
+ *
+ * Flags are fetched once at startup via `initFeatureFlags()` and evaluated
+ * synchronously thereafter — no per-check network calls.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FLAG_AGENT_ANALYTICS = exports.FLAG_LLM_ANALYTICS = void 0;
+exports.initFeatureFlags = initFeatureFlags;
+exports.getFlag = getFlag;
+exports.isFlagEnabled = isFlagEnabled;
+exports.getAllFlags = getAllFlags;
+exports.refreshFlags = refreshFlags;
+const experiment_node_server_1 = require("@amplitude/experiment-node-server");
+const debug_1 = require("../utils/debug");
+// ── Flag keys ────────────────────────────────────────────────────────
+/** Gate for the LLM analytics additional-feature flow. */
+exports.FLAG_LLM_ANALYTICS = 'wizard-llm-analytics';
+/** Gate for agent-level analytics / telemetry instrumented by the wizard. */
+exports.FLAG_AGENT_ANALYTICS = 'wizard-agent-analytics';
+// ── Deployment key ───────────────────────────────────────────────────
+/**
+ * Server deployment key for local evaluation.
+ * Override with `AMPLITUDE_EXPERIMENT_DEPLOYMENT_KEY` env var.
+ */
+const DEFAULT_DEPLOYMENT_KEY = 'server-YOTsk4MS1RWzLyIf1DmvNDUsdOGqkRdM';
+function resolveDeploymentKey() {
+    const fromEnv = process.env.AMPLITUDE_EXPERIMENT_DEPLOYMENT_KEY;
+    return (fromEnv ?? DEFAULT_DEPLOYMENT_KEY).trim();
+}
+// ── Singleton client ─────────────────────────────────────────────────
+let client = null;
+let cachedFlags = {};
+/**
+ * Initialize the Experiment local-evaluation client and pre-fetch flag configs.
+ * Safe to call multiple times — subsequent calls are no-ops.
+ *
+ * @param userId  Optional user ID for targeted flag evaluation.
+ * @param deviceId  Optional device ID for targeted flag evaluation.
+ */
+async function initFeatureFlags(userId, deviceId) {
+    if (client)
+        return; // already initialized
+    const deploymentKey = resolveDeploymentKey();
+    if (!deploymentKey) {
+        (0, debug_1.debug)('feature-flags: no deployment key — all flags default to off');
+        return;
+    }
+    try {
+        client = experiment_node_server_1.Experiment.initializeLocal(deploymentKey);
+        await client.start();
+        const user = {};
+        if (userId)
+            user.user_id = userId;
+        if (deviceId)
+            user.device_id = deviceId;
+        const variants = client.evaluateV2(user);
+        for (const [key, variant] of Object.entries(variants)) {
+            if (variant.value !== undefined && variant.value !== null) {
+                cachedFlags[key] = String(variant.value);
+            }
+        }
+        (0, debug_1.debug)('feature-flags: initialized with flags', cachedFlags);
+    }
+    catch (err) {
+        (0, debug_1.debug)('feature-flags: initialization failed, defaulting all flags off', err);
+        client = null;
+    }
+}
+/**
+ * Evaluate a single feature flag. Returns the string variant value,
+ * or `undefined` if the flag is not set / client not initialized.
+ */
+function getFlag(flagKey) {
+    return cachedFlags[flagKey];
+}
+/**
+ * Check whether a flag is enabled (variant is `'on'` or `'true'`).
+ * Returns `false` when the flag is absent or the client is not initialized.
+ */
+function isFlagEnabled(flagKey) {
+    const value = cachedFlags[flagKey];
+    return value === 'on' || value === 'true';
+}
+/**
+ * Return a snapshot of all evaluated flags (key -> string value).
+ */
+function getAllFlags() {
+    return { ...cachedFlags };
+}
+/**
+ * Re-evaluate flags for a specific user (e.g. after login).
+ * Updates the cached flags in place.
+ */
+async function refreshFlags(userId, deviceId) {
+    if (!client)
+        return;
+    try {
+        // Re-fetch flag configs in case they changed
+        await client.start();
+        const user = {};
+        if (userId)
+            user.user_id = userId;
+        if (deviceId)
+            user.device_id = deviceId;
+        const variants = client.evaluateV2(user);
+        cachedFlags = {};
+        for (const [key, variant] of Object.entries(variants)) {
+            if (variant.value !== undefined && variant.value !== null) {
+                cachedFlags[key] = String(variant.value);
+            }
+        }
+        (0, debug_1.debug)('feature-flags: refreshed flags', cachedFlags);
+    }
+    catch (err) {
+        (0, debug_1.debug)('feature-flags: refresh failed', err);
+    }
+}

package/dist/src/lib/wizard-session.d.ts

@@ -132,6 +132,14 @@ export interface WizardSession {
         workspaces: Array<{
             id: string;
             name: string;
+            environments?: Array<{
+                name: string;
+                rank: number;
+                app: {
+                    id: string;
+                    apiKey?: string | null;
+                } | null;
+            }> | null;
         }>;
     }> | null;
     /** OAuth id_token held during SUSI account-setup steps. */
@@ -146,6 +154,8 @@ export interface WizardSession {
     /** Workspace selected during SUSI (written to ampli.json). */
     selectedWorkspaceId: string | null;
     selectedWorkspaceName: string | null;
+    /** Project/environment selected during SUSI (displayed in TitleBar). */
+    selectedProjectName: string | null;
     /**
      * Notice shown on the API key entry step of AuthScreen.
      * Set when auto-fetch fails (e.g. user is not an org admin) so the user
@@ -185,6 +195,12 @@ export interface WizardSession {
      * runPhase is set to Completed.
      */
     amplitudePreDetected: boolean;
+    /**
+     * While true, McpScreen shows a prompt: skip the agent (default) or run
+     * the setup wizard anyway. Cleared when the user chooses or when resetting
+     * for a forced wizard run.
+     */
+    amplitudePreDetectedChoicePending: boolean;
     /**
      * True once the activation check confirms events are flowing into the project.
      * Set immediately if activationLevel is already 'full', otherwise set after

package/dist/src/lib/wizard-session.js

@@ -109,6 +109,7 @@ function buildSession(args) {
         selectedOrgName: null,
         selectedWorkspaceId: null,
         selectedWorkspaceName: null,
+        selectedProjectName: null,
         loginUrl: null,
         credentials: null,
         apiKeyNotice: null,
@@ -119,6 +120,7 @@ function buildSession(args) {
         additionalFeatureQueue: [],
         frameworkConfig: null,
         amplitudePreDetected: false,
+        amplitudePreDetectedChoicePending: false,
         dataIngestionConfirmed: false,
         checklistChartComplete: false,
         checklistDashboardComplete: false,

package/dist/src/run.js

@@ -59,7 +59,7 @@ async function runWizard(argv, session) {
     const integration = session.integration ?? (await detectAndResolveIntegration(session));
     session.integration = integration;
     analytics_1.analytics.setTag('integration', integration);
-    analytics_1.analytics.wizardCapture('session started', {
+    analytics_1.analytics.wizardCapture('Session Started', {
         integration,
         ci: session.ci ?? false,
     });

package/dist/src/steps/add-mcp-server-to-clients/index.js

@@ -58,7 +58,7 @@ const addMCPServerToClientsStep = async ({ integration, local = false, ci = fals
     });
     ui.log.success(`Added the MCP server to:
   ${supportedClients.map((c) => `- ${c.name}`).join('\n  ')} `);
-    analytics_1.analytics.wizardCapture('mcp servers added', {
+    analytics_1.analytics.wizardCapture('MCP Servers Added', {
         clients: supportedClients.map((c) => c.name),
         integration,
     });
@@ -68,7 +68,7 @@ exports.addMCPServerToClientsStep = addMCPServerToClientsStep;
 const removeMCPServerFromClientsStep = async ({ integration, local = false, }) => {
     const installedClients = await (0, exports.getInstalledClients)(local);
     if (installedClients.length === 0) {
-        analytics_1.analytics.wizardCapture('mcp no servers to remove', {
+        analytics_1.analytics.wizardCapture('MCP No Servers To Remove', {
             integration,
         });
         return [];
@@ -78,7 +78,7 @@ const removeMCPServerFromClientsStep = async ({ integration, local = false, }) =
         await (0, exports.removeMCPServer)(installedClients, local);
         return installedClients.map((c) => c.name);
     });
-    analytics_1.analytics.wizardCapture('mcp servers removed', {
+    analytics_1.analytics.wizardCapture('MCP Servers Removed', {
         clients: results,
         integration,
     });

package/dist/src/steps/add-or-update-environment-variables.js

@@ -90,10 +90,7 @@ async function addOrUpdateEnvironmentVariablesStep({ installDir, variables, inte
             }
             catch (error) {
                 (0, ui_1.getUI)().log.warn(`Failed to update environment variables in ${chalk_1.default.bold.cyan(relativeEnvFilePath)}. Please update them manually.`);
-                analytics_1.analytics.wizardCapture('env vars error', {
-                    integration,
-                    error: error instanceof Error ? error.message : 'Unknown error',
-                });
+                (0, analytics_1.captureWizardError)('Environment Variables', error instanceof Error ? error.message : 'Unknown error', 'add-or-update-env:update-existing', { integration });
                 return {
                     relativeEnvFilePath,
                     addedEnvVariables,
@@ -112,10 +109,7 @@ async function addOrUpdateEnvironmentVariablesStep({ installDir, variables, inte
             }
             catch (error) {
                 (0, ui_1.getUI)().log.warn(`Failed to create ${chalk_1.default.bold.cyan(relativeEnvFilePath)} with environment variables. Please add them manually.`);
-                analytics_1.analytics.wizardCapture('env vars error', {
-                    integration,
-                    error: error instanceof Error ? error.message : 'Unknown error',
-                });
+                (0, analytics_1.captureWizardError)('Environment Variables', error instanceof Error ? error.message : 'Unknown error', 'add-or-update-env:create-new', { integration });
                 return {
                     relativeEnvFilePath,
                     addedEnvVariables,
@@ -141,10 +135,7 @@ async function addOrUpdateEnvironmentVariablesStep({ installDir, variables, inte
                 }
                 catch (error) {
                     (0, ui_1.getUI)().log.warn(`Failed to update ${chalk_1.default.bold.cyan('.gitignore')} to include ${chalk_1.default.bold.cyan(envFileName)}.`);
-                    analytics_1.analytics.wizardCapture('env vars error', {
-                        integration,
-                        error: error instanceof Error ? error.message : 'Unknown error',
-                    });
+                    (0, analytics_1.captureWizardError)('Environment Variables', error instanceof Error ? error.message : 'Unknown error', 'add-or-update-env:gitignore-update', { integration });
                     return {
                         relativeEnvFilePath,
                         addedEnvVariables,
@@ -165,10 +156,7 @@ async function addOrUpdateEnvironmentVariablesStep({ installDir, variables, inte
             }
             catch (error) {
                 (0, ui_1.getUI)().log.warn(`Failed to create ${chalk_1.default.bold.cyan('.gitignore')} with environment files.`);
-                analytics_1.analytics.wizardCapture('env vars error', {
-                    integration,
-                    error: error instanceof Error ? error.message : 'Unknown error',
-                });
+                (0, analytics_1.captureWizardError)('Environment Variables', error instanceof Error ? error.message : 'Unknown error', 'add-or-update-env:gitignore-create', { integration });
                 return {
                     relativeEnvFilePath,
                     addedEnvVariables,
@@ -176,7 +164,7 @@ async function addOrUpdateEnvironmentVariablesStep({ installDir, variables, inte
                 };
             }
         }
-        analytics_1.analytics.wizardCapture('env vars added', {
+        analytics_1.analytics.wizardCapture('Environment Variables Added', {
             integration,
         });
         return {

package/dist/src/steps/run-prettier.js

@@ -83,7 +83,7 @@ async function runPrettierStep({ installDir, integration, }) {
             return;
         }
         prettierSpinner.stop('Prettier has formatted your files.');
-        analytics_1.analytics.wizardCapture('ran prettier', {
+        analytics_1.analytics.wizardCapture('Prettier Ran', {
             integration,
         });
     });

package/dist/src/steps/upload-environment-variables/index.js

@@ -17,7 +17,7 @@ const uploadEnvironmentVariablesStep = async (envVars, { integration, session, }
         }
     }
     if (!provider) {
-        analytics_1.analytics.wizardCapture('env upload skipped', {
+        analytics_1.analytics.wizardCapture('Env Upload Skipped', {
             reason: 'no environment provider found',
             integration,
         });
@@ -28,7 +28,7 @@ const uploadEnvironmentVariablesStep = async (envVars, { integration, session, }
     const results = await (0, telemetry_1.traceStep)('uploading environment variables', async () => {
         return await provider.uploadEnvVars(envVars);
     });
-    analytics_1.analytics.wizardCapture('env uploaded', {
+    analytics_1.analytics.wizardCapture('Env Uploaded', {
         provider: provider.name,
         integration,
     });

package/dist/src/ui/tui/App.js

@@ -30,5 +30,5 @@ export const App = ({ store }) => {
     const contentAreaWidth = Math.max(10, innerWidth - 2);
     const direction = store.lastNavDirection === 'pop' ? 'right' : 'left';
     const activeScreen = screens[store.currentScreen] ?? null;
-    return (_jsx(CommandModeContext.Provider, { value: store.commandMode, children: _jsx(Box, { flexDirection: "column", height: rows, width: columns, alignItems: "center", justifyContent: "flex-start", children: _jsxs(ConsoleView, { store: store, width: width, height: rows, children: [_jsx(TitleBar, { version: store.version, width: innerWidth }), _jsx(Box, { height: 1 }), _jsx(Box, { flexDirection: "column", flexGrow: 1, paddingX: 1, overflow: "hidden", children: _jsx(DissolveTransition, { transitionKey: store.currentScreen, width: contentAreaWidth, height: contentHeight, direction: direction, children: _jsx(ScreenErrorBoundary, { store: store, retryToken: store.screenErrorRetry, children: activeScreen }) }) })] }) }) }));
+    return (_jsx(CommandModeContext.Provider, { value: store.commandMode, children: _jsx(Box, { flexDirection: "column", height: rows, width: columns, alignItems: "center", justifyContent: "flex-start", children: _jsxs(ConsoleView, { store: store, width: width, height: rows, children: [_jsx(TitleBar, { version: store.version, width: innerWidth, orgName: store.session.selectedOrgName, projectName: store.session.selectedProjectName }), _jsx(Box, { height: 1 }), _jsx(Box, { flexDirection: "column", flexGrow: 1, paddingX: 1, overflow: "hidden", children: _jsx(DissolveTransition, { transitionKey: store.currentScreen, width: contentAreaWidth, height: contentHeight, direction: direction, children: _jsx(ScreenErrorBoundary, { store: store, retryToken: store.screenErrorRetry, children: activeScreen }) }) })] }) }) }));
 };

package/dist/src/ui/tui/components/ConsoleView.js

@@ -21,8 +21,9 @@ import { PickerMenu } from '../primitives/PickerMenu.js';
 import { Colors, Icons } from '../styles.js';
 import { Overlay } from '../router.js';
 import { queryConsole, resolveConsoleCredentials, buildSessionContext, } from '../../../lib/console-query.js';
-import { COMMANDS, getWhoamiText, getHelpText, TEST_PROMPT, } from '../console-commands.js';
+import { COMMANDS, getWhoamiText, parseFeedbackSlashInput, TEST_PROMPT, } from '../console-commands.js';
 import { analytics } from '../../../utils/analytics.js';
+import { trackWizardFeedback } from '../../../utils/track-wizard-feedback.js';
 function executeCommand(raw, store) {
     const [cmd] = raw.trim().split(/\s+/);
     switch (cmd) {
@@ -41,6 +42,19 @@ function executeCommand(raw, store) {
         case '/slack':
             store.showSlackOverlay();
             break;
+        case '/feedback': {
+            const message = parseFeedbackSlashInput(raw);
+            if (!message) {
+                store.setCommandFeedback('Usage: /feedback <your message>');
+                break;
+            }
+            void trackWizardFeedback(message)
+                .then(() => store.setCommandFeedback('Thanks — your feedback was sent.'))
+                .catch((err) => {
+                store.setCommandFeedback(`Could not send feedback: ${err instanceof Error ? err.message : String(err)}`);
+            });
+            break;
+        }
         case '/test':
             return TEST_PROMPT;
         case '/mcp':
@@ -52,12 +66,9 @@ function executeCommand(raw, store) {
         case '/exit':
             store.setOutroData({ kind: OutroKind.Cancel, message: 'Exited.' });
             break;
-        case '/help':
-            store.setCommandFeedback(getHelpText());
-            break;
         default:
             if (cmd)
-                store.setCommandFeedback(`Unknown command: ${cmd}. Type /help.`);
+                store.setCommandFeedback(`Unknown command: ${cmd}. Type / to see available commands.`);
     }
 }
 export const ConsoleView = ({ store, width, height, children, }) => {
@@ -116,7 +127,7 @@ export const ConsoleView = ({ store, width, height, children, }) => {
     }, { isActive: !inputActive });
     const handleSubmit = (value) => {
         const isSlashCommand = value.startsWith('/');
-        analytics.wizardCapture('agent message sent', {
+        analytics.wizardCapture('Agent Message Sent', {
             message_length: value.length,
             is_slash_command: isSlashCommand,
         });

package/dist/src/ui/tui/components/TitleBar.d.ts

@@ -1,6 +1,8 @@
 interface TitleBarProps {
     version: string;
     width: number;
+    orgName?: string | null;
+    projectName?: string | null;
 }
-export declare const TitleBar: ({ version, width }: TitleBarProps) => import("react/jsx-runtime").JSX.Element;
+export declare const TitleBar: ({ version, width, orgName, projectName, }: TitleBarProps) => import("react/jsx-runtime").JSX.Element;
 export {};

package/dist/src/ui/tui/components/TitleBar.js

@@ -1,16 +1,27 @@
-import { jsxs as _jsxs, jsx as _jsx } from "react/jsx-runtime";
+import { jsx as _jsx, jsxs as _jsxs } from "react/jsx-runtime";
 import { Box, Text } from 'ink';
 import { Colors } from '../styles.js';
 const FEEDBACK = 'Feedback: wizard@amplitude.com ';
 const FEEDBACK_SHORT = ' wizard@amplitude.com ';
-export const TitleBar = ({ version, width }) => {
+export const TitleBar = ({ version, width, orgName, projectName, }) => {
     const fullTitle = ` Amplitude Wizard v${version}`;
-    const needShort = width < fullTitle.length + FEEDBACK.length;
+    // Build context string from org + project names
+    const contextParts = [];
+    if (orgName)
+        contextParts.push(orgName);
+    if (projectName)
+        contextParts.push(projectName);
+    const contextStr = contextParts.length > 0 ? ` ${contextParts.join(' / ')} ` : '';
+    const needShort = width < fullTitle.length + contextStr.length + FEEDBACK.length;
     const feedback = needShort ? FEEDBACK_SHORT : FEEDBACK;
-    const title = needShort && fullTitle.length + feedback.length > width
+    const title = needShort && fullTitle.length + contextStr.length + feedback.length > width
         ? ` Wizard v${version}`
         : fullTitle;
-    const gap = Math.max(0, width - title.length - feedback.length);
+    // If context doesn't fit even with the short title, drop it
+    const showContext = contextStr.length > 0 &&
+        title.length + contextStr.length + feedback.length <= width;
+    const middleText = showContext ? contextStr : '';
+    const gap = Math.max(0, width - title.length - middleText.length - feedback.length);
     const padding = ' '.repeat(gap);
-    return (_jsx(Box, { width: width, overflow: "hidden", children: _jsxs(Text, { backgroundColor: Colors.accent, color: "white", bold: true, children: [title, padding, feedback] }) }));
+    return (_jsxs(Box, { width: width, overflow: "hidden", children: [_jsx(Text, { backgroundColor: Colors.accent, color: "white", bold: true, children: title }), middleText ? (_jsxs(Text, { backgroundColor: Colors.accent, color: "white", dimColor: true, children: [padding, middleText] })) : (_jsx(Text, { backgroundColor: Colors.accent, color: "white", children: padding })), _jsx(Text, { backgroundColor: Colors.accent, color: "white", bold: true, children: feedback })] }));
 };

package/dist/src/ui/tui/console-commands.d.ts

@@ -12,5 +12,8 @@ export declare const COMMANDS: {
 export declare const TEST_PROMPT: string;
 /** Returns the feedback text for the /whoami command. */
 export declare function getWhoamiText(session: Pick<WizardSession, 'selectedOrgName' | 'selectedWorkspaceName' | 'region'>): string;
-/** Returns the feedback text for the /help command. */
-export declare function getHelpText(): string;
+/**
+ * Parses `/feedback <message>` from a slash command line.
+ * Returns `undefined` if the line is not a feedback command or the message is empty.
+ */
+export declare function parseFeedbackSlashInput(raw: string): string | undefined;

package/dist/src/ui/tui/console-commands.js

@@ -11,10 +11,13 @@ export const COMMANDS = [
     { cmd: '/whoami', desc: 'Show current user, org, and project' },
     { cmd: '/mcp', desc: 'Install or remove the Amplitude MCP server' },
     { cmd: '/slack', desc: 'Set up Amplitude Slack integration' },
+    {
+        cmd: '/feedback',
+        desc: 'Send product feedback (event: wizard: feedback submitted)',
+    },
     { cmd: '/test', desc: 'Run a prompt-skill demo (confirm + choose)' },
     { cmd: '/snake', desc: 'Play Snake' },
     { cmd: '/exit', desc: 'Exit the wizard' },
-    { cmd: '/help', desc: 'List available slash commands' },
 ];
 export const TEST_PROMPT = 'Demo the wizard prompt tools. ' +
     'First, use the wizard-tools:confirm tool to ask if I want to continue. ' +
@@ -24,8 +27,14 @@ export const TEST_PROMPT = 'Demo the wizard prompt tools. ' +
 export function getWhoamiText(session) {
     return `org: ${session.selectedOrgName ?? '(none)'}  workspace: ${session.selectedWorkspaceName ?? '(none)'}  region: ${session.region ?? '(none)'}`;
 }
-/** Returns the feedback text for the /help command. */
-export function getHelpText() {
-    const maxCmd = Math.max(...COMMANDS.map((c) => c.cmd.length));
-    return COMMANDS.map((c) => `${c.cmd.padEnd(maxCmd)}  ${c.desc}`).join('\n');
+/**
+ * Parses `/feedback <message>` from a slash command line.
+ * Returns `undefined` if the line is not a feedback command or the message is empty.
+ */
+export function parseFeedbackSlashInput(raw) {
+    const m = /^\s*\/feedback(?:\s+(.*))?\s*$/i.exec(raw);
+    if (!m)
+        return undefined;
+    const body = m[1]?.trim();
+    return body || undefined;
 }

package/dist/src/ui/tui/screens/AuthScreen.d.ts

@@ -5,7 +5,8 @@
  *   1. OAuth waiting — spinner + login URL while browser auth happens
  *   2. Org selection — picker if the user belongs to multiple orgs
  *   3. Workspace selection — picker if the org has multiple workspaces
- *   4. API key entry — text input for the Amplitude analytics write key
+ *   4. Project selection — picker if the workspace has multiple environments
+ *   5. API key entry — text input (only if no project key could be resolved)
  *
  * The screen drives itself from session.pendingOrgs + session.credentials.
  * When credentials are set the router resolves past this screen.