@promptbook/cli 0.113.0-9 → 0.114.0-0

package/src/book-components/Chat/utils/thinkingMessageVariants.ts

@@ -31,15 +31,25 @@ const THINKING_MESSAGE_DELAY_MAX_MS = 5_000;
  * @private internal helper for Promptbook chat placeholders
  */
 export function normalizeThinkingMessageVariants(thinkingMessages?: ReadonlyArray<string>): ReadonlyArray<string> {
+    const normalized = parseThinkingMessageVariants(thinkingMessages);
+
+    return normalized.length > 0 ? normalized : DEFAULT_THINKING_MESSAGES;
+}
+
+/**
+ * Parses configured thinking-message variants without applying default fallbacks.
+ *
+ * @param thinkingMessages Raw thinking-message variants.
+ * @returns Trimmed non-empty variants.
+ *
+ * @private internal helper for Promptbook chat placeholders
+ */
+export function parseThinkingMessageVariants(thinkingMessages?: ReadonlyArray<string> | null): ReadonlyArray<string> {
     if (!thinkingMessages) {
-        return DEFAULT_THINKING_MESSAGES;
+        return [];
     }
 
-    const normalized = thinkingMessages
-        .map((message) => message?.trim())
-        .filter((message): message is string => Boolean(message));
-
-    return normalized.length > 0 ? normalized : DEFAULT_THINKING_MESSAGES;
+    return thinkingMessages.map((message) => message?.trim()).filter((message): message is string => Boolean(message));
 }
 
 /**

package/src/cli/cli-commands/about.ts

@@ -1,6 +1,6 @@
 import colors from 'colors';
 import type {
-  Command as Program /* <- Note: [🔸] Using Program because Command is misleading name */
+    Command as Program /* <- Note: [🔸] Using Program because Command is misleading name */,
 } from 'commander';
 import { spaceTrim } from 'spacetrim';
 import { CLAIM } from '../../config';

package/src/cli/cli-commands/agent/agentCliOptions.ts

@@ -1,8 +1,11 @@
 import type {
-  Command as Program /* <- Note: [🔸] Using Program because Command is misleading name */
+    Command as Program /* <- Note: [🔸] Using Program because Command is misleading name */,
 } from 'commander';
 import { NotAllowed } from '../../../errors/NotAllowed';
-import type { NormalizedPromptRunnerSelectionCliOptions, PromptRunnerSelectionCliOptions } from '../common/promptRunnerCliOptions';
+import type {
+    NormalizedPromptRunnerSelectionCliOptions,
+    PromptRunnerSelectionCliOptions,
+} from '../common/promptRunnerCliOptions';
 import { normalizePromptRunnerSelectionCliOptions } from '../common/promptRunnerCliOptions';
 
 /**

package/src/cli/cli-commands/agent-folder/agentProjectPaths.ts

@@ -1,13 +1,21 @@
 import { join } from 'path';
 import { spaceTrim } from 'spacetrim';
-import { AGENT_BOOK_FILE_PATH, AGENT_FAILED_MESSAGES_DIRECTORY_PATH, AGENT_FINISHED_MESSAGES_DIRECTORY_PATH, AGENT_MESSAGES_DIRECTORY_PATH, AGENT_QUEUED_MESSAGES_DIRECTORY_PATH } from '../../../book-3.0/agentFolderPaths';
+import {
+    AGENT_BOOK_FILE_PATH,
+    AGENT_FAILED_MESSAGES_DIRECTORY_PATH,
+    AGENT_FINISHED_MESSAGES_DIRECTORY_PATH,
+    AGENT_MESSAGES_DIRECTORY_PATH,
+    AGENT_PROJECTS_DIRECTORY_PATH,
+    AGENT_QUEUED_MESSAGES_DIRECTORY_PATH,
+} from '../../../book-3.0/agentFolderPaths';
 
 export {
-  AGENT_BOOK_FILE_PATH,
-  AGENT_FAILED_MESSAGES_DIRECTORY_PATH,
-  AGENT_FINISHED_MESSAGES_DIRECTORY_PATH,
-  AGENT_MESSAGES_DIRECTORY_PATH,
-  AGENT_QUEUED_MESSAGES_DIRECTORY_PATH,
+    AGENT_BOOK_FILE_PATH,
+    AGENT_FAILED_MESSAGES_DIRECTORY_PATH,
+    AGENT_FINISHED_MESSAGES_DIRECTORY_PATH,
+    AGENT_MESSAGES_DIRECTORY_PATH,
+    AGENT_PROJECTS_DIRECTORY_PATH,
+    AGENT_QUEUED_MESSAGES_DIRECTORY_PATH,
 };
 
 /**

package/src/cli/cli-commands/agent-folder/init.ts

@@ -1,8 +1,12 @@
 import type {
-  Command as Program /* <- Note: [🔸] Using Program because Command is misleading name */
+    Command as Program /* <- Note: [🔸] Using Program because Command is misleading name */,
 } from 'commander';
 import { spaceTrim } from 'spacetrim';
-import { AGENT_BOOK_FILE_PATH, AGENT_FINISHED_MESSAGES_DIRECTORY_PATH, AGENT_QUEUED_MESSAGES_DIRECTORY_PATH } from '../../../book-3.0/agentFolderPaths';
+import {
+    AGENT_BOOK_FILE_PATH,
+    AGENT_FINISHED_MESSAGES_DIRECTORY_PATH,
+    AGENT_QUEUED_MESSAGES_DIRECTORY_PATH,
+} from '../../../book-3.0/agentFolderPaths';
 import type { $side_effect } from '../../../utils/organization/$side_effect';
 import { handleActionErrors } from '../common/handleActionErrors';
 import { AGENT_BOOK_LANGUAGE_MANUAL_FILE_PATH, AGENT_KNOWLEDGE_DIRECTORY_PATH } from './agentProjectPaths';

package/src/cli/cli-commands/agent-folder/initializeAgentProjectConfiguration.ts

@@ -1,8 +1,19 @@
-import { AGENT_BOOK_FILE_PATH, AGENT_FINISHED_MESSAGES_DIRECTORY_PATH, AGENT_MESSAGES_DIRECTORY_PATH, AGENT_QUEUED_MESSAGES_DIRECTORY_PATH } from '../../../book-3.0/agentFolderPaths';
+import {
+    AGENT_BOOK_FILE_PATH,
+    AGENT_FINISHED_MESSAGES_DIRECTORY_PATH,
+    AGENT_MESSAGES_DIRECTORY_PATH,
+    AGENT_QUEUED_MESSAGES_DIRECTORY_PATH,
+} from '../../../book-3.0/agentFolderPaths';
 import type { InitializationStatus } from '../coder/boilerplateTemplates';
 import { ensureCoderMarkdownFile } from '../coder/ensureCoderMarkdownFile';
 import { ensureDirectory } from '../coder/ensureDirectory';
-import { AGENT_BOOK_LANGUAGE_MANUAL_FILE_PATH, AGENT_DOCS_DIRECTORY_PATH, AGENT_KNOWLEDGE_DIRECTORY_PATH, getDefaultAgentBookContent, getDefaultBookLanguageManualContent } from './agentProjectPaths';
+import {
+    AGENT_BOOK_LANGUAGE_MANUAL_FILE_PATH,
+    AGENT_DOCS_DIRECTORY_PATH,
+    AGENT_KNOWLEDGE_DIRECTORY_PATH,
+    getDefaultAgentBookContent,
+    getDefaultBookLanguageManualContent,
+} from './agentProjectPaths';
 
 /**
  * Result summary returned after local agent configuration initialization.

package/src/cli/cli-commands/agent-folder/printAgentInitializationSummary.ts

@@ -1,7 +1,16 @@
 import colors from 'colors';
-import { AGENT_BOOK_FILE_PATH, AGENT_FINISHED_MESSAGES_DIRECTORY_PATH, AGENT_MESSAGES_DIRECTORY_PATH, AGENT_QUEUED_MESSAGES_DIRECTORY_PATH } from '../../../book-3.0/agentFolderPaths';
+import {
+    AGENT_BOOK_FILE_PATH,
+    AGENT_FINISHED_MESSAGES_DIRECTORY_PATH,
+    AGENT_MESSAGES_DIRECTORY_PATH,
+    AGENT_QUEUED_MESSAGES_DIRECTORY_PATH,
+} from '../../../book-3.0/agentFolderPaths';
 import type { InitializationStatus } from '../coder/boilerplateTemplates';
-import { AGENT_BOOK_LANGUAGE_MANUAL_FILE_PATH, AGENT_DOCS_DIRECTORY_PATH, AGENT_KNOWLEDGE_DIRECTORY_PATH } from './agentProjectPaths';
+import {
+    AGENT_BOOK_LANGUAGE_MANUAL_FILE_PATH,
+    AGENT_DOCS_DIRECTORY_PATH,
+    AGENT_KNOWLEDGE_DIRECTORY_PATH,
+} from './agentProjectPaths';
 import type { AgentInitializationSummary } from './initializeAgentProjectConfiguration';
 
 /**

package/src/cli/cli-commands/agents-server/buildAgentsServer/createAgentsServerBuildSourceFingerprint.ts

@@ -43,7 +43,9 @@ async function addAgentsServerBuildInputToFingerprint(
     try {
         inputStats = await stat(options.inputPath);
     } catch {
-        fingerprint.update(`missing:${normalizeAgentsServerBuildInputPath(options.runtimeRootPath, options.inputPath)}\n`);
+        fingerprint.update(
+            `missing:${normalizeAgentsServerBuildInputPath(options.runtimeRootPath, options.inputPath)}\n`,
+        );
         return;
     }
 
@@ -58,7 +60,9 @@ async function addAgentsServerBuildInputToFingerprint(
         return;
     }
 
-    fingerprint.update(`directory:${normalizeAgentsServerBuildInputPath(options.runtimeRootPath, options.inputPath)}\n`);
+    fingerprint.update(
+        `directory:${normalizeAgentsServerBuildInputPath(options.runtimeRootPath, options.inputPath)}\n`,
+    );
     const inputDirents = await readdir(options.inputPath, { withFileTypes: true });
     const sortedInputDirents = [...inputDirents].sort((left, right) => left.name.localeCompare(right.name));
 

package/src/cli/cli-commands/agents-server/buildAgentsServer/runNextBuild.ts

@@ -5,11 +5,10 @@ import { NotAllowed } from '../../../../errors/NotAllowed';
 /**
  * Node.js heap size limit (in MiB) injected into `NODE_OPTIONS` for the Next.js production build.
  *
- * Next.js webpack peaks at ~1.9 GiB on a fresh install; the Node.js default cap is ~1.7 GiB,
- * which causes an OOM crash on first-run VPS installations where swap is the primary resource.
- * Raising the limit lets the build complete on the first attempt.
+ * Next.js webpack can exceed the default ~4 GiB cap after the full verification suite warms the
+ * workspace, and VPS self-updates use this same production build path.
  */
-const AGENTS_SERVER_BUILD_MAX_OLD_SPACE_MIB = 4096;
+const AGENTS_SERVER_BUILD_MAX_OLD_SPACE_MIB = 8192;
 
 /**
  * Maximum attempts for a Next.js build that was killed by the operating system.

package/src/cli/cli-commands/agents-server/run.ts

@@ -16,9 +16,10 @@ import {
     normalizePromptRunnerSelectionCliOptions,
     PROMPT_RUNNER_DESCRIPTION,
 } from '../common/promptRunnerCliOptions';
-import { ensureAgentsServerBuild } from './buildAgentsServer';
-import type { AgentsServerNextRuntimeMode } from './startAgentsServer';
-import { loadAgentsServerProjectEnvironment, startAgentsServer } from './startAgentsServer';
+import { ensureAgentsServerBuild } from './buildAgentsServer/ensureAgentsServerBuild';
+import { startAgentsServer } from './startAgentsServer';
+import { loadAgentsServerProjectEnvironment } from './startAgentsServer/loadAgentsServerProjectEnvironment';
+import type { AgentsServerNextRuntimeMode } from './startAgentsServer/StartAgentsServerOptions';
 
 /**
  * Default port used by `ptbk agents-server start` and `ptbk agents-server dev`.

package/src/cli/cli-commands/agents-server/startAgentsServer/AgentsServerChildEnvironment.ts

@@ -1,5 +1,9 @@
 import { randomBytes } from 'crypto';
 import { join } from 'path';
+import {
+    PTBK_AGENTS_SERVER_URL_ENV,
+    PTBK_AGENTS_SERVER_USER_CHAT_WORKER_TOKEN_ENV,
+} from '../../../../../apps/agents-server/src/utils/agentProjects/agentProjectRuntimeConstants';
 import type { number_port } from '../../../../types/number_positive';
 
 /**
@@ -36,6 +40,7 @@ const LOCAL_USER_CHAT_WORKER_TOKEN_BYTE_LENGTH = 32;
  * @private internal type of `startAgentsServer`
  */
 export type AgentsServerChildEnvironment = NodeJS.ProcessEnv & {
+    readonly PTBK_AGENTS_SERVER_URL: string;
     readonly PTBK_AGENTS_SERVER_USER_CHAT_WORKER_TOKEN: string;
 };
 
@@ -49,19 +54,22 @@ export function createAgentsServerChildEnvironment(
     agentRootPath: string,
 ): AgentsServerChildEnvironment {
     const launchWorkingDirectory = process.cwd();
+    const localAgentsServerUrl = `http://localhost:${port}`;
+    const publicSiteUrl = process.env.NEXT_PUBLIC_SITE_URL || localAgentsServerUrl;
 
     return {
         ...process.env,
         PORT: String(port),
-        NEXT_PUBLIC_SITE_URL: process.env.NEXT_PUBLIC_SITE_URL || `http://localhost:${port}`,
+        NEXT_PUBLIC_SITE_URL: publicSiteUrl,
+        [PTBK_AGENTS_SERVER_URL_ENV]: localAgentsServerUrl,
         [PTBK_AGENTS_SERVER_AGENT_ROOT_ENV]: agentRootPath,
         [PTBK_AGENTS_SERVER_SQLITE_PATH_ENV]:
             process.env[PTBK_AGENTS_SERVER_SQLITE_PATH_ENV] ||
             join(launchWorkingDirectory, '.promptbook', 'agents-server.sqlite'),
         [PTBK_AGENTS_SERVER_DATABASE_ENV]: process.env[PTBK_AGENTS_SERVER_DATABASE_ENV] || 'supabase',
         // Next loads app-local `.env` values after the CLI has prepared this bridge environment.
-        PTBK_AGENTS_SERVER_USER_CHAT_WORKER_TOKEN:
-            process.env.PTBK_AGENTS_SERVER_USER_CHAT_WORKER_TOKEN ||
+        [PTBK_AGENTS_SERVER_USER_CHAT_WORKER_TOKEN_ENV]:
+            process.env[PTBK_AGENTS_SERVER_USER_CHAT_WORKER_TOKEN_ENV] ||
             randomBytes(LOCAL_USER_CHAT_WORKER_TOKEN_BYTE_LENGTH).toString('hex'),
     };
 }

package/src/cli/cli-commands/agents-server/startAgentsServer/AgentsServerRuntimePaths.ts

@@ -1,6 +1,6 @@
 import { join } from 'path';
 import { resolvePromptbookTemporaryPath } from '../../../../utils/filesystem/promptbookTemporaryPath';
-import { resolveAgentsServerAppPath } from '../buildAgentsServer';
+import { resolveAgentsServerAppPath } from '../buildAgentsServer/resolveAgentsServerAppPath';
 
 /**
  * Name of the local Agents Server log folder relative to the launch working directory.

package/src/cli/cli-commands/agents-server/startAgentsServer/PreparedAgentsServerLaunch.ts

@@ -1,4 +1,4 @@
-import type { PreparedAgentsServerRuntime } from '../buildAgentsServer';
+import type { PreparedAgentsServerRuntime } from '../buildAgentsServer/PreparedAgentsServerRuntime';
 import type { AgentsServerChildEnvironment } from './AgentsServerChildEnvironment';
 import type { AgentsServerRuntimePaths } from './AgentsServerRuntimePaths';
 

package/src/cli/cli-commands/agents-server/startAgentsServer/createInternalRouteErrorMessage.ts

@@ -3,7 +3,11 @@
  *
  * @private internal utility of `startAgentsServer`
  */
-export function createInternalRouteErrorMessage(routeLabel: string, response: Response, details: string | null): string {
+export function createInternalRouteErrorMessage(
+    routeLabel: string,
+    response: Response,
+    details: string | null,
+): string {
     const statusText = response.statusText ? ` ${response.statusText}` : '';
     const statusMessage = `${response.status}${statusText}`;
 

package/src/cli/cli-commands/agents-server/startAgentsServer/forwardChildOutput.ts

@@ -1,5 +1,6 @@
 import type { WriteStream } from 'fs';
-import { addUiOutput, type AgentsServerSupervisorState } from './AgentsServerSupervisorState';
+import type { AgentsServerSupervisorState } from './AgentsServerSupervisorState';
+import { addUiOutput } from './AgentsServerSupervisorState';
 
 /**
  * Persists one child output chunk and forwards readable lines into UI or plain output.

package/src/cli/cli-commands/agents-server/startAgentsServer/prepareAgentsServerLaunch.ts

@@ -1,10 +1,8 @@
 import type { WriteStream } from 'fs';
-import {
-    createAgentsServerRuntimeEnvironment,
-    ensureAgentsServerBuild,
-    prepareAgentsServerRuntime,
-    type PreparedAgentsServerRuntime,
-} from '../buildAgentsServer';
+import { createAgentsServerRuntimeEnvironment } from '../buildAgentsServer/createAgentsServerRuntimeEnvironment';
+import { ensureAgentsServerBuild } from '../buildAgentsServer/ensureAgentsServerBuild';
+import { prepareAgentsServerRuntime } from '../buildAgentsServer/prepareAgentsServerRuntime';
+import type { PreparedAgentsServerRuntime } from '../buildAgentsServer/PreparedAgentsServerRuntime';
 import type { AgentsServerChildEnvironment } from './AgentsServerChildEnvironment';
 import type { AgentsServerLogStreams } from './AgentsServerLogStreams';
 import { logRunnerEvent } from './AgentsServerLogStreams';

package/src/cli/cli-commands/agents-server/startAgentsServer/startNextServer.ts

@@ -3,7 +3,8 @@ import type { AgentsServerChildEnvironment } from './AgentsServerChildEnvironmen
 import type { AgentsServerLogStreams } from './AgentsServerLogStreams';
 import { logRunnerEvent } from './AgentsServerLogStreams';
 import type { AgentsServerRuntimePaths } from './AgentsServerRuntimePaths';
-import { addUiError, type AgentsServerSupervisorState } from './AgentsServerSupervisorState';
+import type { AgentsServerSupervisorState } from './AgentsServerSupervisorState';
+import { addUiError } from './AgentsServerSupervisorState';
 import { forwardChildOutput } from './forwardChildOutput';
 import { resolveAgentsServerChildHostname } from './resolveAgentsServerChildHostname';
 import type { AgentsServerNextRuntimeMode, StartAgentsServerOptions } from './StartAgentsServerOptions';

package/src/cli/cli-commands/agents-server/startAgentsServer/startUserChatJobWorkerPump.ts

@@ -3,7 +3,8 @@ import type { number_port } from '../../../../types/number_positive';
 import type { AgentsServerChildEnvironment } from './AgentsServerChildEnvironment';
 import type { AgentsServerLogStreams } from './AgentsServerLogStreams';
 import { logRunnerEvent } from './AgentsServerLogStreams';
-import { addUiError, type AgentsServerSupervisorState } from './AgentsServerSupervisorState';
+import type { AgentsServerSupervisorState } from './AgentsServerSupervisorState';
+import { addUiError } from './AgentsServerSupervisorState';
 import { createInternalRouteErrorMessage } from './createInternalRouteErrorMessage';
 import { readInternalRouteErrorDetails } from './readInternalRouteErrorDetails';
 

package/src/cli/cli-commands/agents-server/startAgentsServer.ts

@@ -1,5 +1,9 @@
 import type { ChildProcess } from 'child_process';
 import { mkdir } from 'fs/promises';
+import {
+    PTBK_AGENTS_SERVER_URL_ENV,
+    PTBK_AGENTS_SERVER_USER_CHAT_WORKER_TOKEN_ENV,
+} from '../../../../apps/agents-server/src/utils/agentProjects/agentProjectRuntimeConstants';
 import { runMultipleAgentMessages } from '../../../../scripts/run-agent-messages/main/runMultipleAgentMessages';
 import { withCurrentWorkingDirectory } from '../../../../scripts/run-agent-messages/main/withCurrentWorkingDirectory';
 import { createAgentsServerChildEnvironment } from './startAgentsServer/AgentsServerChildEnvironment';
@@ -9,11 +13,8 @@ import {
     logRunnerEvent,
 } from './startAgentsServer/AgentsServerLogStreams';
 import { resolveAgentsServerRuntimePaths } from './startAgentsServer/AgentsServerRuntimePaths';
-import {
-    addUiOutput,
-    assertNextServerStillRunning,
-    type AgentsServerSupervisorState,
-} from './startAgentsServer/AgentsServerSupervisorState';
+import type { AgentsServerSupervisorState } from './startAgentsServer/AgentsServerSupervisorState';
+import { addUiOutput, assertNextServerStillRunning } from './startAgentsServer/AgentsServerSupervisorState';
 import { createLocalAgentRunOptions } from './startAgentsServer/createLocalAgentRunOptions';
 import { loadAgentsServerProjectEnvironment } from './startAgentsServer/loadAgentsServerProjectEnvironment';
 import { prepareAgentsServerLaunch } from './startAgentsServer/prepareAgentsServerLaunch';
@@ -24,7 +25,10 @@ import { stopChildProcess } from './startAgentsServer/stopChildProcess';
 import { waitForLocalAgentRunnerLimits } from './startAgentsServer/waitForLocalAgentRunnerLimits';
 
 export { loadAgentsServerProjectEnvironment } from './startAgentsServer/loadAgentsServerProjectEnvironment';
-export type { AgentsServerNextRuntimeMode, StartAgentsServerOptions } from './startAgentsServer/StartAgentsServerOptions';
+export type {
+    AgentsServerNextRuntimeMode,
+    StartAgentsServerOptions,
+} from './startAgentsServer/StartAgentsServerOptions';
 
 /**
  * Starts the Agents Server web app and local coding-agent queue workers in the foreground.
@@ -84,18 +88,26 @@ export async function startAgentsServer(options: StartAgentsServerOptions): Prom
             state,
         });
 
+        const restoreLocalAgentRunnerEnvironment = applyLocalAgentRunnerInternalEnvironment(
+            preparedLaunch.runtimeChildEnvironment,
+        );
+
         await withCurrentWorkingDirectory(runtimePaths.agentRootPath, async () => {
-            await runMultipleAgentMessages(createLocalAgentRunOptions(options, localAgentRunnerLimits), {
-                shouldContinue: () => state.isContinuing,
-                watchErrorLogDirectoryPath: runtimePaths.logDirectoryPath,
-                onUiInitialized: (uiHandle) => {
-                    state.uiHandle = uiHandle;
-                    addUiOutput(
-                        state,
-                        `Agents Server running at http://localhost:${options.port}. Next logs: ${runtimePaths.nextLogPath}`,
-                    );
-                },
-            });
+            try {
+                await runMultipleAgentMessages(createLocalAgentRunOptions(options, localAgentRunnerLimits), {
+                    shouldContinue: () => state.isContinuing,
+                    watchErrorLogDirectoryPath: runtimePaths.logDirectoryPath,
+                    onUiInitialized: (uiHandle) => {
+                        state.uiHandle = uiHandle;
+                        addUiOutput(
+                            state,
+                            `Agents Server running at http://localhost:${options.port}. Next logs: ${runtimePaths.nextLogPath}`,
+                        );
+                    },
+                });
+            } finally {
+                restoreLocalAgentRunnerEnvironment();
+            }
         });
 
         assertNextServerStillRunning(state);
@@ -108,4 +120,33 @@ export async function startAgentsServer(options: StartAgentsServerOptions): Prom
     }
 }
 
+/**
+ * Applies internal Agents Server environment variables to local coding-runner subprocesses.
+ */
+function applyLocalAgentRunnerInternalEnvironment(environment: NodeJS.ProcessEnv): () => void {
+    const previousAgentsServerUrl = process.env[PTBK_AGENTS_SERVER_URL_ENV];
+    const previousWorkerToken = process.env[PTBK_AGENTS_SERVER_USER_CHAT_WORKER_TOKEN_ENV];
+
+    process.env[PTBK_AGENTS_SERVER_URL_ENV] = environment[PTBK_AGENTS_SERVER_URL_ENV] || '';
+    process.env[PTBK_AGENTS_SERVER_USER_CHAT_WORKER_TOKEN_ENV] =
+        environment[PTBK_AGENTS_SERVER_USER_CHAT_WORKER_TOKEN_ENV] || '';
+
+    return () => {
+        restoreEnvironmentVariable(PTBK_AGENTS_SERVER_URL_ENV, previousAgentsServerUrl);
+        restoreEnvironmentVariable(PTBK_AGENTS_SERVER_USER_CHAT_WORKER_TOKEN_ENV, previousWorkerToken);
+    };
+}
+
+/**
+ * Restores or removes one environment variable.
+ */
+function restoreEnvironmentVariable(environmentVariableName: string, previousValue: string | undefined): void {
+    if (previousValue === undefined) {
+        delete process.env[environmentVariableName];
+        return;
+    }
+
+    process.env[environmentVariableName] = previousValue;
+}
+
 // Note: [🟡] Code for CLI runtime [startAgentsServer](src/cli/cli-commands/agents-server/startAgentsServer.ts) should never be published outside of `@promptbook/cli`

package/src/cli/cli-commands/agents-server.ts

@@ -1,11 +1,15 @@
 import colors from 'colors';
 import type {
-  Command as Program /* <- Note: [🔸] Using Program because Command is misleading name */
+    Command as Program /* <- Note: [🔸] Using Program because Command is misleading name */,
 } from 'commander';
 import { spaceTrim } from 'spacetrim';
 import type { $side_effect } from '../../utils/organization/$side_effect';
 import { $initializeAgentsServerInitCommand } from './agents-server/init';
-import { $initializeAgentsServerBuildCommand, $initializeAgentsServerDevCommand, $initializeAgentsServerStartCommand } from './agents-server/run';
+import {
+    $initializeAgentsServerBuildCommand,
+    $initializeAgentsServerDevCommand,
+    $initializeAgentsServerStartCommand,
+} from './agents-server/run';
 
 /**
  * Initializes `agents-server` command with subcommands for Promptbook CLI utilities.