codecane 1.0.207 → 1.0.237

package/dist/client.js

@@ -1,43 +1,25 @@
 "use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
-    Object.defineProperty(o, "default", { enumerable: true, value: v });
-}) : function(o, v) {
-    o["default"] = v;
-});
-var __importStar = (this && this.__importStar) || function (mod) {
-    if (mod && mod.__esModule) return mod;
-    var result = {};
-    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
-    __setModuleDefault(result, mod);
-    return result;
-};
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Client = void 0;
 const child_process_1 = require("child_process");
-const fs = __importStar(require("fs"));
+const fs_1 = require("fs");
 const path_1 = __importDefault(require("path"));
-const picocolors_1 = require("picocolors");
-const ts_pattern_1 = require("ts-pattern");
 const actions_1 = require("./common/actions");
-const constants_1 = require("./common/constants");
+const constants_1 = require("./common/api-keys/constants");
+const constants_2 = require("./common/constants");
+const constants_3 = require("./common/json-config/constants");
 const agent_state_1 = require("./common/types/agent-state");
+const string_1 = require("./common/util/string");
 const websocket_client_1 = require("./common/websockets/websocket-client");
-const chat_storage_1 = require("./chat-storage");
+const picocolors_1 = require("picocolors");
+const ts_pattern_1 = require("ts-pattern");
+const zod_1 = require("zod");
+const background_process_manager_1 = require("./background-process-manager");
 const browser_runner_1 = require("./browser-runner");
+const chat_storage_1 = require("./chat-storage");
 const checkpoint_manager_1 = require("./checkpoints/checkpoint-manager");
 const config_1 = require("./config");
 const credentials_1 = require("./credentials");
@@ -46,37 +28,81 @@ const menu_1 = require("./menu");
 const project_files_1 = require("./project-files");
 const tool_handlers_1 = require("./tool-handlers");
 const spinner_1 = require("./utils/spinner");
-const xml_stream_parser_1 = require("./utils/xml-stream-parser");
 const tool_renderers_1 = require("./utils/tool-renderers");
+const xml_stream_parser_1 = require("./utils/xml-stream-parser");
+const web_scraper_1 = require("./web-scraper");
+const web_scraper_2 = require("./web-scraper");
+const array_1 = require("./common/util/array");
+const LOW_BALANCE_THRESHOLD = 100;
+const WARNING_CONFIG = {
+    [constants_2.UserState.LOGGED_OUT]: {
+        message: () => `Type "login" to unlock full access and get free credits!`,
+        threshold: 100,
+    },
+    [constants_2.UserState.DEPLETED]: {
+        message: () => [
+            (0, picocolors_1.red)(`\n❌ You have used all your credits.`),
+            `Visit ${(0, picocolors_1.bold)((0, picocolors_1.blue)(config_1.websiteUrl + '/usage'))} to add more credits and continue coding.`,
+        ].join('\n'),
+        threshold: 100,
+    },
+    [constants_2.UserState.CRITICAL]: {
+        message: (credits) => [
+            (0, picocolors_1.yellow)(`\n🪫 Only ${(0, picocolors_1.bold)((0, string_1.pluralize)(credits, 'credit'))} remaining!`),
+            (0, picocolors_1.yellow)(`Visit ${(0, picocolors_1.bold)(config_1.websiteUrl + '/usage')} to add more credits.`),
+        ].join('\n'),
+        threshold: 85,
+    },
+    [constants_2.UserState.ATTENTION_NEEDED]: {
+        message: (credits) => [
+            (0, picocolors_1.yellow)(`\n⚠️ ${(0, picocolors_1.bold)((0, string_1.pluralize)(credits, 'credit'))} remaining. Consider topping up soon.`),
+        ].join('\n'),
+        threshold: 75,
+    },
+    [constants_2.UserState.GOOD_STANDING]: {
+        message: () => '',
+        threshold: 0,
+    },
+};
 class Client {
     webSocket;
     returnControlToUser;
+    reconnectWhenNextIdle;
     fingerprintId;
     costMode;
+    hadFileChanges = false;
+    git;
+    rl;
+    responseComplete = false;
+    pendingTopUpMessageAmount = null;
+    oneTimeFlags = Object.fromEntries(constants_2.ONE_TIME_LABELS.map((tag) => [tag, false]));
+    usageData = {
+        usage: 0,
+        remainingBalance: 0,
+        balanceBreakdown: undefined,
+        next_quota_reset: null,
+        nextMonthlyGrant: 0,
+    };
     fileContext;
     lastChanges = [];
     agentState;
     originalFileVersions = {};
+    creditsByPromptId = {};
     user;
     lastWarnedPct = 0;
-    usage = 0;
-    limit = 0;
-    subscription_active = false;
-    lastRequestCredits = 0;
-    sessionCreditsUsed = 0;
-    nextQuotaReset = null;
-    hadFileChanges = false;
-    git;
-    rl;
+    nextMonthlyGrant = 0;
+    storedApiKeyTypes = [];
     lastToolResults = [];
-    pendingRequestId = null;
-    constructor(websocketUrl, onWebSocketError, onWebSocketReconnect, returnControlToUser, costMode, git, rl) {
+    model;
+    constructor(websocketUrl, onWebSocketError, onWebSocketReconnect, returnControlToUser, reconnectWhenNextIdle, costMode, git, rl, model) {
         this.costMode = costMode;
+        this.model = model;
         this.git = git;
         this.webSocket = new websocket_client_1.APIRealtimeClient(websocketUrl, onWebSocketError, onWebSocketReconnect);
         this.user = this.getUser();
-        this.getFingerprintId();
+        this.initFingerprintId();
         this.returnControlToUser = returnControlToUser;
+        this.reconnectWhenNextIdle = reconnectWhenNextIdle;
         this.rl = rl;
     }
     async exit() {
@@ -89,25 +115,104 @@ class Client {
         this.agentState = (0, agent_state_1.getInitialAgentState)(projectFileContext);
         this.fileContext = projectFileContext;
     }
-    async getFingerprintId() {
-        if (this.fingerprintId) {
-            return this.fingerprintId;
+    initFingerprintId() {
+        if (!this.fingerprintId) {
+            this.fingerprintId = this.user?.fingerprintId ?? (0, fingerprint_1.calculateFingerprint)();
         }
-        this.fingerprintId =
-            this.user?.fingerprintId ?? (await (0, fingerprint_1.calculateFingerprint)());
         return this.fingerprintId;
     }
     getUser() {
-        if (!fs.existsSync(credentials_1.CREDENTIALS_PATH)) {
+        if (!(0, fs_1.existsSync)(credentials_1.CREDENTIALS_PATH)) {
             return;
         }
-        const credentialsFile = fs.readFileSync(credentials_1.CREDENTIALS_PATH, 'utf8');
+        const credentialsFile = (0, fs_1.readFileSync)(credentials_1.CREDENTIALS_PATH, 'utf8');
         const user = (0, credentials_1.userFromJson)(credentialsFile);
         return user;
     }
     async connect() {
         await this.webSocket.connect();
         this.setupSubscriptions();
+        await this.fetchStoredApiKeyTypes();
+    }
+    async fetchStoredApiKeyTypes() {
+        if (!this.user || !this.user.authToken) {
+            return;
+        }
+        const TIMEOUT_MS = 5_000;
+        //   try {
+        //     const timeoutPromise = new Promise<Response>((_, reject) => {
+        //       setTimeout(() => reject(new Error('Request timed out')), TIMEOUT_MS)
+        //     })
+        //     const fetchPromise = fetch(
+        //       `${process.env.NEXT_PUBLIC_APP_URL}/api/api-keys`,
+        //       {
+        //         method: 'GET',
+        //         headers: {
+        //           'Content-Type': 'application/json',
+        //           Cookie: `next-auth.session-token=${this.user.authToken}`,
+        //           Authorization: `Bearer ${this.user.authToken}`,
+        //         },
+        //       }
+        //     )
+        //     const response = await Promise.race([fetchPromise, timeoutPromise])
+        //     if (response.ok) {
+        //       const { keyTypes } = await response.json()
+        //       this.storedApiKeyTypes = keyTypes as ApiKeyType[]
+        //     } else {
+        //       this.storedApiKeyTypes = []
+        //     }
+        //   } catch (error) {
+        //     if (process.env.NODE_ENV !== 'production') {
+        //       console.error(
+        //         'Error fetching stored API key types (is there something else on port 3000?):',
+        //         error
+        //       )
+        //     }
+        //     this.storedApiKeyTypes = []
+        //   }
+        this.storedApiKeyTypes = [];
+    }
+    async handleAddApiKey(keyType, apiKey) {
+        if (!this.user || !this.user.authToken) {
+            console.log((0, picocolors_1.yellow)("Please log in first using 'login'."));
+            this.returnControlToUser();
+            return;
+        }
+        const readableKeyType = constants_1.READABLE_NAME[keyType];
+        spinner_1.Spinner.get().start();
+        try {
+            const response = await fetch(`${process.env.NEXT_PUBLIC_APP_URL}/api/api-keys`, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                    Cookie: `next-auth.session-token=${this.user.authToken}`,
+                },
+                body: JSON.stringify({
+                    keyType,
+                    apiKey,
+                    authToken: this.user.authToken,
+                }),
+            });
+            spinner_1.Spinner.get().stop();
+            const respJson = await response.json();
+            if (response.ok) {
+                console.log((0, picocolors_1.green)(`Successfully added ${readableKeyType} API key.`));
+                if (!this.storedApiKeyTypes.includes(keyType)) {
+                    this.storedApiKeyTypes.push(keyType);
+                }
+            }
+            else {
+                throw new Error(respJson.message);
+            }
+        }
+        catch (e) {
+            spinner_1.Spinner.get().stop();
+            const error = e;
+            console.error((0, picocolors_1.red)('Error adding API key: ' + error.message));
+        }
+        finally {
+            this.returnControlToUser();
+        }
     }
     async handleReferralCode(referralCode) {
         if (this.user) {
@@ -127,7 +232,7 @@ class Client {
                 if (redeemReferralResp.ok) {
                     console.log([
                         (0, picocolors_1.green)(`Noice, you've earned an extra ${respJson.credits_redeemed} credits!`),
-                        `(pssst: you can also refer new users and earn ${constants_1.CREDITS_REFERRAL_BONUS} credits for each referral at: ${process.env.NEXT_PUBLIC_APP_URL}/referrals)`,
+                        `(pssst: you can also refer new users and earn ${constants_2.CREDITS_REFERRAL_BONUS} credits for each referral at: ${process.env.NEXT_PUBLIC_APP_URL}/referrals)`,
                     ].join('\n'));
                     this.getUsage();
                 }
@@ -148,7 +253,7 @@ class Client {
     async logout() {
         if (this.user) {
             try {
-                const response = await fetch(`${config_1.backendUrl}/api/auth/cli/logout`, {
+                const response = await fetch(`${config_1.websiteUrl}/api/auth/cli/logout`, {
                     method: 'POST',
                     headers: { 'Content-Type': 'application/json' },
                     body: JSON.stringify({
@@ -163,9 +268,18 @@ class Client {
                     console.error((0, picocolors_1.red)('Failed to log out: ' + error));
                 }
                 try {
-                    fs.unlinkSync(credentials_1.CREDENTIALS_PATH);
+                    (0, fs_1.unlinkSync)(credentials_1.CREDENTIALS_PATH);
                     console.log(`You (${this.user.name}) have been logged out.`);
                     this.user = undefined;
+                    this.pendingTopUpMessageAmount = null;
+                    this.usageData = {
+                        usage: 0,
+                        remainingBalance: 0,
+                        balanceBreakdown: undefined,
+                        next_quota_reset: null,
+                        nextMonthlyGrant: 0,
+                    };
+                    this.oneTimeFlags = Object.fromEntries(constants_2.ONE_TIME_LABELS.map((tag) => [tag, false]));
                 }
                 catch (error) {
                     console.error('Error removing credentials file:', error);
@@ -183,11 +297,11 @@ class Client {
             return;
         }
         try {
-            const response = await fetch(`${config_1.backendUrl}/api/auth/cli/code`, {
+            const response = await fetch(`${config_1.websiteUrl}/api/auth/cli/code`, {
                 method: 'POST',
                 headers: { 'Content-Type': 'application/json' },
                 body: JSON.stringify({
-                    fingerprintId: await this.getFingerprintId(),
+                    fingerprintId: await this.fingerprintId,
                     referralCode,
                 }),
             });
@@ -197,7 +311,7 @@ class Client {
                 this.returnControlToUser();
                 return;
             }
-            const { loginUrl, fingerprintHash } = await response.json();
+            const { loginUrl, fingerprintHash, expiresAt } = await response.json();
             const responseToUser = [
                 '\n',
                 `Press ${(0, picocolors_1.blue)('ENTER')} to open your browser and finish logging in...`,
@@ -207,7 +321,7 @@ class Client {
             this.rl.once('line', () => {
                 if (shouldRequestLogin) {
                     (0, child_process_1.spawn)(`open ${loginUrl}`, { shell: true });
-                    console.log('Done. If nothing happened, copy and paste this link into your browser:');
+                    console.log("Opened a browser window to log you in! If it doesn't open automatically, you can click this link:");
                     console.log();
                     console.log((0, picocolors_1.blue)((0, picocolors_1.bold)((0, picocolors_1.underline)(loginUrl))));
                 }
@@ -226,7 +340,7 @@ class Client {
                     return;
                 }
                 try {
-                    const statusResponse = await fetch(`${config_1.backendUrl}/api/auth/cli/status?fingerprintId=${await this.getFingerprintId()}&fingerprintHash=${fingerprintHash}`);
+                    const statusResponse = await fetch(`${config_1.websiteUrl}/api/auth/cli/status?fingerprintId=${await this.fingerprintId}&fingerprintHash=${fingerprintHash}&expiresAt=${expiresAt}`);
                     if (!statusResponse.ok) {
                         if (statusResponse.status !== 401) {
                             // Ignore 401s during polling
@@ -239,16 +353,17 @@ class Client {
                         shouldRequestLogin = false;
                         this.user = user;
                         const credentialsPathDir = path_1.default.dirname(credentials_1.CREDENTIALS_PATH);
-                        fs.mkdirSync(credentialsPathDir, { recursive: true });
-                        fs.writeFileSync(credentials_1.CREDENTIALS_PATH, JSON.stringify({ default: user }));
+                        (0, fs_1.mkdirSync)(credentialsPathDir, { recursive: true });
+                        (0, fs_1.writeFileSync)(credentials_1.CREDENTIALS_PATH, JSON.stringify({ default: user }));
                         const referralLink = `${process.env.NEXT_PUBLIC_APP_URL}/referrals`;
                         const responseToUser = [
                             'Authentication successful! 🎉',
                             (0, picocolors_1.bold)(`Hey there, ${user.name}.`),
-                            `Refer new users and earn ${constants_1.CREDITS_REFERRAL_BONUS} credits per month: ${(0, picocolors_1.blueBright)(referralLink)}`,
+                            `Refer new users and earn ${constants_2.CREDITS_REFERRAL_BONUS} credits per month: ${(0, picocolors_1.blueBright)(referralLink)}`,
                         ];
                         console.log('\n' + responseToUser.join('\n'));
                         this.lastWarnedPct = 0;
+                        this.oneTimeFlags = Object.fromEntries(constants_2.ONE_TIME_LABELS.map((tag) => [tag, false]));
                         (0, menu_1.displayGreeting)(this.costMode, null);
                         clearInterval(pollInterval);
                         this.returnControlToUser();
@@ -264,19 +379,25 @@ class Client {
             this.returnControlToUser();
         }
     }
-    setUsage({ usage, limit, subscription_active, next_quota_reset, session_credits_used, }) {
-        this.usage = usage;
-        this.limit = limit;
-        this.subscription_active = subscription_active;
-        this.nextQuotaReset = next_quota_reset;
-        if (!!session_credits_used && !this.pendingRequestId) {
-            this.lastRequestCredits = Math.max(session_credits_used - this.sessionCreditsUsed, 0);
-            this.sessionCreditsUsed = session_credits_used;
-        }
+    setUsage(usageData) {
+        this.usageData = usageData;
+    }
+    reconnect() {
+        this.webSocket.forceReconnect();
     }
     setupSubscriptions() {
         this.webSocket.subscribe('action-error', (action) => {
-            console.error(['', (0, picocolors_1.red)(`Error: ${action.message}`)].join('\n'));
+            if (action.error === 'Insufficient credits') {
+                console.error(['', (0, picocolors_1.red)(`Error: ${action.message}`)].join('\n'));
+                console.error(`Visit ${(0, picocolors_1.blue)((0, picocolors_1.bold)(process.env.NEXT_PUBLIC_APP_URL + '/usage'))} to add credits.`);
+            }
+            else if (action.error === 'Auto top-up disabled') {
+                console.error(['', (0, picocolors_1.red)(`Error: ${action.message}`)].join('\n'));
+                console.error((0, picocolors_1.yellow)(`Visit ${(0, picocolors_1.blue)((0, picocolors_1.bold)(process.env.NEXT_PUBLIC_APP_URL + '/usage'))} to update your payment settings.`));
+            }
+            else {
+                console.error(['', (0, picocolors_1.red)(`Error: ${action.message}`)].join('\n'));
+            }
             this.returnControlToUser();
             return;
         });
@@ -295,45 +416,61 @@ class Client {
                 console.warn((0, picocolors_1.yellow)(`\nThere's a new version of Codebuff! Please update to ensure proper functionality.\nUpdate now by running: npm install -g codebuff`));
             }
         });
-        this.webSocket.subscribe('usage-response', (action) => {
-            const parsedAction = actions_1.UsageReponseSchema.safeParse(action);
+        this.webSocket.subscribe('message-cost-response', (action) => {
+            const parsedAction = actions_1.MessageCostResponseSchema.safeParse(action);
             if (!parsedAction.success)
                 return;
-            const a = parsedAction.data;
-            console.log();
-            console.log((0, picocolors_1.green)((0, picocolors_1.underline)(`Codebuff usage:`)), `${a.usage} / ${a.limit} credits`);
-            this.setUsage(a);
-            this.returnControlToUser();
+            const response = parsedAction.data;
+            // Store credits used for this prompt
+            if (!this.creditsByPromptId[response.promptId]) {
+                this.creditsByPromptId[response.promptId] = [];
+            }
+            this.creditsByPromptId[response.promptId].push(response.credits);
         });
-    }
-    showUsageWarning() {
-        const errorCopy = [
-            this.user
-                ? `Visit ${(0, picocolors_1.blue)((0, picocolors_1.bold)(process.env.NEXT_PUBLIC_APP_URL + '/pricing'))} to upgrade – or refer a new user and earn ${constants_1.CREDITS_REFERRAL_BONUS} credits per month: ${(0, picocolors_1.blue)((0, picocolors_1.bold)(process.env.NEXT_PUBLIC_APP_URL + '/referrals'))}`
-                : (0, picocolors_1.green)('Type "login" below to sign up and get more credits!'),
-        ].join('\n');
-        const pct = (0, ts_pattern_1.match)(Math.floor((this.usage / this.limit) * 100))
-            .with(ts_pattern_1.P.number.gte(100), () => 100)
-            .with(ts_pattern_1.P.number.gte(75), () => 75)
-            .otherwise(() => 0);
-        if (pct >= 100) {
-            this.lastWarnedPct = 100;
-            if (!this.subscription_active) {
-                console.error([(0, picocolors_1.red)('You have reached your monthly usage limit.'), errorCopy].join('\n'));
+        this.webSocket.subscribe('usage-response', (action) => {
+            const parsedAction = actions_1.UsageReponseSchema.safeParse(action);
+            if (!parsedAction.success) {
+                console.error((0, picocolors_1.red)('Received invalid usage data from server:'), parsedAction.error.errors);
                 return;
             }
-            if (this.subscription_active && this.lastWarnedPct < 100) {
-                console.warn((0, picocolors_1.yellow)(`You have exceeded your monthly quota, but feel free to keep using Codebuff! We'll continue to charge you for the overage until your next billing cycle. See ${process.env.NEXT_PUBLIC_APP_URL}/usage for more details.`));
-                return;
+            this.setUsage(parsedAction.data);
+            // Store auto-topup info if it occurred
+            if (parsedAction.data.autoTopupAdded) {
+                this.pendingTopUpMessageAmount = parsedAction.data.autoTopupAdded;
             }
+            // Only show warning if the response is complete
+            if (this.responseComplete) {
+                this.showUsageWarning();
+            }
+        });
+        // Used to handle server restarts gracefully
+        this.webSocket.subscribe('request-reconnect', () => {
+            this.reconnectWhenNextIdle();
+        });
+    }
+    showUsageWarning() {
+        // Determine user state based on login status and credit balance
+        const state = (0, ts_pattern_1.match)({
+            isLoggedIn: !!this.user,
+            credits: this.usageData.remainingBalance,
+        })
+            .with({ isLoggedIn: false }, () => constants_2.UserState.LOGGED_OUT)
+            .with({ credits: ts_pattern_1.P.number.gte(100) }, () => constants_2.UserState.GOOD_STANDING)
+            .with({ credits: ts_pattern_1.P.number.gte(20) }, () => constants_2.UserState.ATTENTION_NEEDED)
+            .with({ credits: ts_pattern_1.P.number.gte(1) }, () => constants_2.UserState.CRITICAL)
+            .otherwise(() => constants_2.UserState.DEPLETED);
+        const config = WARNING_CONFIG[state];
+        // Reset warning percentage if in good standing
+        if (state === constants_2.UserState.GOOD_STANDING) {
+            this.lastWarnedPct = 0;
+            return;
         }
-        if (pct > 0 && pct > this.lastWarnedPct) {
-            console.warn([
-                '',
-                (0, picocolors_1.yellow)(`You have used over ${pct}% of your monthly usage limit (${this.usage}/${this.limit} credits).`),
-                errorCopy,
-            ].join('\n'));
-            this.lastWarnedPct = pct;
+        // Show warning if we haven't warned at this threshold yet
+        if (this.lastWarnedPct < config.threshold) {
+            const message = config.message(this.usageData.remainingBalance);
+            console.warn(message);
+            this.lastWarnedPct = config.threshold;
+            this.returnControlToUser();
         }
     }
     async generateCommitMessage(stagedChanges) {
@@ -344,7 +481,7 @@ class Client {
             });
             this.webSocket.sendAction({
                 type: 'generate-commit-message',
-                fingerprintId: await this.getFingerprintId(),
+                fingerprintId: await this.fingerprintId,
                 authToken: this.user?.authToken,
                 stagedChanges,
             });
@@ -355,7 +492,6 @@ class Client {
             throw new Error('Agent state not initialized');
         }
         const userInputId = `mc-input-` + Math.random().toString(36).substring(2, 15);
-        this.pendingRequestId = userInputId;
         const { responsePromise, stopResponse } = this.subscribeToResponse((chunk) => {
             spinner_1.Spinner.get().stop();
             process.stdout.write(chunk);
@@ -363,16 +499,26 @@ class Client {
             spinner_1.Spinner.get().stop();
             process.stdout.write((0, picocolors_1.green)((0, picocolors_1.underline)('\nCodebuff') + ':') + ' ');
         }, prompt);
+        const urls = (0, web_scraper_2.parseUrlsFromContent)(prompt);
+        const scrapedBlocks = await (0, web_scraper_1.getScrapedContentBlocks)(urls);
+        const scrapedContent = scrapedBlocks.length > 0 ? scrapedBlocks.join('\n\n') + '\n\n' : '';
+        // Append process updates to existing tool results
+        const toolResults = (0, array_1.buildArray)(...(this.lastToolResults || []), ...(0, background_process_manager_1.getBackgroundProcessUpdates)(), scrapedContent && {
+            id: 'scraped-content',
+            name: 'web-scraper',
+            result: scrapedContent,
+        });
         spinner_1.Spinner.get().start();
         this.webSocket.sendAction({
             type: 'prompt',
             promptId: userInputId,
             prompt,
             agentState: this.agentState,
-            toolResults: this.lastToolResults,
-            fingerprintId: await this.getFingerprintId(),
+            toolResults,
+            fingerprintId: await this.fingerprintId,
             authToken: this.user?.authToken,
             costMode: this.costMode,
+            model: this.model,
         });
         return {
             responsePromise,
@@ -393,11 +539,11 @@ class Client {
         });
         const stopResponse = () => {
             responseStopped = true;
-            // Only unsubscribe from chunks, keep listening for final credits
             unsubscribeChunks();
+            unsubscribeComplete();
             const assistantMessage = {
-                role: 'assistant',
-                content: responseBuffer + '[RESPONSE_CANCELED_BY_USER]',
+                role: 'user',
+                content: `Received output:\n${responseBuffer}[RESPONSE_CANCELED_BY_USER]`,
             };
             // Update the agent state with just the assistant's response
             const { messageHistory } = this.agentState;
@@ -424,6 +570,21 @@ class Client {
             if (a.userInputId !== userInputId)
                 return;
             const { chunk } = a;
+            const trimmed = chunk.trim();
+            for (const tag of constants_2.ONE_TIME_TAGS) {
+                if (trimmed.startsWith(`<${tag}>`) && trimmed.endsWith(`</${tag}>`)) {
+                    if (this.oneTimeFlags[tag]) {
+                        return;
+                    }
+                    spinner_1.Spinner.get().stop();
+                    const warningMessage = trimmed
+                        .replace(`<${tag}>`, '')
+                        .replace(`</${tag}>`, '');
+                    console.warn((0, picocolors_1.yellow)(`\n${warningMessage}`));
+                    this.oneTimeFlags[tag] = true;
+                    return;
+                }
+            }
             if (chunk && chunk.trim()) {
                 if (!streamStarted && chunk.trim()) {
                     streamStarted = true;
@@ -444,45 +605,51 @@ class Client {
             if (action.promptId !== userInputId)
                 return;
             const a = parsedAction.data;
-            // Only process usage data if this is our pending request
-            const usageData = actions_1.UsageReponseSchema.omit({ type: true }).safeParse(a);
-            if (usageData.success) {
-                this.pendingRequestId = null;
-                this.setUsage(usageData.data);
-            }
-            if (responseStopped) {
-                unsubscribeComplete();
-                xmlStreamParser.end();
-                return;
-            }
-            this.agentState = a.agentState;
-            spinner_1.Spinner.get().stop();
             let isComplete = false;
+            spinner_1.Spinner.get().stop();
+            this.agentState = a.agentState;
             const toolResults = [...a.toolResults];
             for (const toolCall of a.toolCalls) {
-                if (toolCall.name === 'end_turn') {
-                    isComplete = true;
-                    continue;
-                }
-                if (toolCall.name === 'write_file') {
-                    // Save lastChanges for `diff` command
-                    this.lastChanges.push(actions_1.FileChangeSchema.parse(toolCall.parameters));
-                    this.hadFileChanges = true;
+                try {
+                    if (toolCall.name === 'end_turn') {
+                        this.responseComplete = true;
+                        isComplete = true;
+                        continue;
+                    }
+                    if (toolCall.name === 'write_file') {
+                        // Save lastChanges for `diff` command
+                        this.lastChanges.push(actions_1.FileChangeSchema.parse(toolCall.parameters));
+                        this.hadFileChanges = true;
+                    }
+                    if (toolCall.name === 'run_terminal_command' &&
+                        toolCall.parameters.mode === 'user') {
+                        // Special case: when terminal command is run it as a user command, then no need to reprompt assistant.
+                        this.responseComplete = true;
+                        isComplete = true;
+                    }
+                    if (toolCall.name === 'run_terminal_command' &&
+                        toolCall.parameters.mode === 'assistant' &&
+                        toolCall.parameters.process_type === 'BACKGROUND') {
+                        this.oneTimeFlags[constants_2.SHOULD_ASK_CONFIG] = true;
+                    }
+                    const toolResult = await (0, tool_handlers_1.handleToolCall)(toolCall, (0, project_files_1.getProjectRoot)());
+                    toolResults.push(toolResult);
                 }
-                if (toolCall.name === 'run_terminal_command' &&
-                    toolCall.parameters.mode === 'user') {
-                    // Special case: when terminal command is run it as a user command, then no need to reprompt assistant.
-                    isComplete = true;
+                catch (error) {
+                    console.error((0, picocolors_1.red)(`Error parsing tool call ${toolCall.name}:\n${error}`));
                 }
-                const toolResult = await (0, tool_handlers_1.handleToolCall)(toolCall, (0, project_files_1.getProjectRoot)());
-                toolResults.push(toolResult);
+            }
+            if (toolResults.length > 0 ||
+                a.toolCalls.some((call) => call.name === 'end_turn')) {
+                console.log();
             }
             // If we had any file changes, update the project context
             if (this.hadFileChanges) {
                 this.fileContext = await (0, project_files_1.getProjectFileContext)((0, project_files_1.getProjectRoot)(), {});
             }
             if (!isComplete) {
-                spinner_1.Spinner.get().start();
+                // Append process updates to existing tool results
+                toolResults.push(...(0, background_process_manager_1.getBackgroundProcessUpdates)());
                 // Continue the prompt with the tool results.
                 this.webSocket.sendAction({
                     type: 'prompt',
@@ -490,41 +657,117 @@ class Client {
                     prompt: undefined,
                     agentState: this.agentState,
                     toolResults,
-                    fingerprintId: await this.getFingerprintId(),
+                    fingerprintId: await this.fingerprintId,
                     authToken: this.user?.authToken,
                     costMode: this.costMode,
+                    model: this.model,
                 });
                 return;
             }
             this.lastToolResults = toolResults;
             xmlStreamParser.end();
+            askConfig: if (this.oneTimeFlags[constants_2.SHOULD_ASK_CONFIG] &&
+                !this.oneTimeFlags[constants_2.ASKED_CONFIG]) {
+                this.oneTimeFlags[constants_2.ASKED_CONFIG] = true;
+                if ((0, fs_1.existsSync)(path_1.default.join((0, project_files_1.getProjectRoot)(), constants_3.codebuffConfigFile))) {
+                    break askConfig;
+                }
+                console.log((0, picocolors_1.yellow)(`✨ Recommended: run the 'init' command in order to create a configuration file!
+
+If you would like background processes (like this one) to run automatically whenever codebuff starts, creating a ${constants_3.codebuffConfigFile} config file can improve your workflow.
+Go to https://www.codebuff.com/config for more information.
+`));
+            }
             if (this.agentState) {
                 (0, chat_storage_1.setMessages)(this.agentState.messageHistory);
             }
-            this.showUsageWarning();
+            // Show total credits used for this prompt if significant
+            const credits = this.creditsByPromptId[userInputId]?.reduce((a, b) => a + b, 0) ?? 0;
+            if (credits >= constants_2.REQUEST_CREDIT_SHOW_THRESHOLD) {
+                console.log(`${(0, string_1.pluralize)(credits, 'credit')} used for this request.`);
+            }
+            // Show auto top-up success message if it occurred during this response
+            if (this.pendingTopUpMessageAmount) {
+                console.log((0, picocolors_1.green)(`Auto top-up successful! ${this.pendingTopUpMessageAmount.toLocaleString()} credits added.`));
+                this.pendingTopUpMessageAmount = null;
+            }
             if (this.hadFileChanges) {
-                const latestCheckpoint = checkpoint_manager_1.checkpointManager.getLatestCheckpoint();
-                const displayedCheckpointString = latestCheckpoint === null
-                    ? ''
-                    : ` or "checkpoint ${latestCheckpoint.id}" to revert`;
-                console.log(`\nComplete! Type "diff" to review changes${displayedCheckpointString}.`);
+                let checkpointAddendum = '';
+                try {
+                    checkpointAddendum = ` or "checkpoint ${checkpoint_manager_1.checkpointManager.getLatestCheckpoint().id}" to revert`;
+                }
+                catch (error) {
+                    // No latest checkpoint, don't show addendum
+                }
+                console.log(`\nComplete! Type "diff" to review changes${checkpointAddendum}.`);
                 this.hadFileChanges = false;
+                this.returnControlToUser();
             }
             unsubscribeChunks();
             unsubscribeComplete();
             resolveResponse({ ...a, wasStoppedByUser: false });
         });
+        // Reset flags at the start of each response
+        this.responseComplete = false;
         return {
             responsePromise,
             stopResponse,
         };
     }
     async getUsage() {
-        this.webSocket.sendAction({
-            type: 'usage',
-            fingerprintId: await this.getFingerprintId(),
-            authToken: this.user?.authToken,
-        });
+        try {
+            const response = await fetch(`${config_1.backendUrl}/api/usage`, {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                },
+                body: JSON.stringify({
+                    fingerprintId: await this.fingerprintId,
+                    authToken: this.user?.authToken,
+                }),
+            });
+            const data = await response.json();
+            // Use zod schema to validate response
+            const parsedResponse = actions_1.UsageReponseSchema.parse(data);
+            if (data.type === 'action-error') {
+                console.error((0, picocolors_1.red)(data.message));
+                return;
+            }
+            this.setUsage(parsedResponse);
+            const usageLink = `${config_1.websiteUrl}/usage`;
+            const remainingColor = this.usageData.remainingBalance <= 0
+                ? picocolors_1.red
+                : this.usageData.remainingBalance <= LOW_BALANCE_THRESHOLD
+                    ? picocolors_1.red
+                    : picocolors_1.green;
+            const totalCreditsUsedThisSession = Object.values(this.creditsByPromptId)
+                .flat()
+                .reduce((sum, credits) => sum + credits, 0);
+            console.log(`Session usage: ${totalCreditsUsedThisSession.toLocaleString()}. Credits Remaining: ${remainingColor(this.usageData.remainingBalance.toLocaleString())}`);
+            if (this.usageData.next_quota_reset) {
+                const resetDate = new Date(this.usageData.next_quota_reset);
+                const today = new Date();
+                const isToday = resetDate.toDateString() === today.toDateString();
+                const dateDisplay = isToday
+                    ? resetDate.toLocaleString() // Show full date and time for today
+                    : resetDate.toLocaleDateString(); // Just show date otherwise
+                console.log(`Free credits will renew on ${dateDisplay}. Details: ${(0, picocolors_1.underline)((0, picocolors_1.blue)(usageLink))}`);
+            }
+            this.showUsageWarning();
+        }
+        catch (error) {
+            console.error((0, picocolors_1.red)(`Error checking usage: Please reach out to ${process.env.NEXT_PUBLIC_SUPPORT_EMAIL} for help.`));
+            // Check if it's a ZodError for more specific feedback
+            if (error instanceof zod_1.z.ZodError) {
+                console.error((0, picocolors_1.red)('Data validation failed:'), error.errors);
+            }
+            else {
+                console.error(error);
+            }
+        }
+        finally {
+            this.returnControlToUser();
+        }
     }
     async warmContextCache() {
         const fileContext = await (0, project_files_1.getProjectFileContext)((0, project_files_1.getProjectRoot)(), {});
@@ -532,18 +775,16 @@ class Client {
             const parsedAction = actions_1.InitResponseSchema.safeParse(a);
             if (!parsedAction.success)
                 return;
+            // Set initial usage data from the init response
             this.setUsage(parsedAction.data);
         });
-        this.webSocket
-            .sendAction({
+        this.webSocket.sendAction({
             type: 'init',
-            fingerprintId: await this.getFingerprintId(),
+            fingerprintId: await this.fingerprintId,
             authToken: this.user?.authToken,
             fileContext,
-        })
-            .catch((e) => {
-            // console.error('Error warming context cache', e)
         });
+        await this.fetchStoredApiKeyTypes();
     }
 }
 exports.Client = Client;