@promptbook/cli 0.113.0-1 → 0.113.0-3

package/apps/agents-server/src/utils/getAdminChatTasksResponse.ts

@@ -4,6 +4,7 @@ import { getAdminChatTasks } from './getAdminChatTasksResponse/getAdminChatTasks
 import { mapVpsSelfUpdateJobToAdminChatTask } from './getAdminChatTasksResponse/mapVpsSelfUpdateJobToAdminChatTask';
 import { parseAdminChatTaskQuery, type ParsedAdminChatTaskQuery } from './getAdminChatTasksResponse/parseAdminChatTaskQuery';
 import { throttledAdminRecovery } from './getAdminChatTasksResponse/throttledAdminRecovery';
+import { listPagePreviewBrowserAdminTasks } from './pagePreviewBrowserSessions';
 import { readVpsSelfUpdateJobSnapshot } from './vpsSelfUpdate';
 
 /**
@@ -63,7 +64,11 @@ export async function getAdminChatTasksResponse(
 
     const adminChatTasks = await getAdminChatTasks(parsedQuery);
     const vpsSelfUpdateTask = await loadVpsSelfUpdateAdminChatTask();
-    const injectedTasks = collectAdminChatTasksToInject(vpsSelfUpdateTask, parsedQuery);
+    const pagePreviewBrowserTasks = listPagePreviewBrowserAdminTasks();
+    const injectableTasks = [vpsSelfUpdateTask, ...pagePreviewBrowserTasks].filter(
+        (task): task is AdminChatTaskRecord => task !== null,
+    );
+    const injectedTasks = collectAdminChatTasksToInject(injectableTasks, parsedQuery);
     const { items, total } = mergeInjectedAdminChatTasks({
         databaseItems: adminChatTasks.items,
         databaseTotal: adminChatTasks.total,
@@ -71,7 +76,7 @@ export async function getAdminChatTasksResponse(
         page: parsedQuery.page,
         pageSize: parsedQuery.pageSize,
     });
-    const counters = mergeInjectedAdminChatTaskCounters(adminChatTasks.counters, vpsSelfUpdateTask);
+    const counters = mergeInjectedAdminChatTaskCounters(adminChatTasks.counters, injectableTasks);
 
     return {
         status: 200,
@@ -109,27 +114,18 @@ async function loadVpsSelfUpdateAdminChatTask(): Promise<AdminChatTaskRecord | n
 /**
  * Filters out injected tasks that do not belong to the requested admin task-manager view or search.
  *
- * @param vpsSelfUpdateTask - Currently persisted self-update task or `null`.
+ * @param injectableTasks - Synthetic task rows collected from process-local state.
  * @param query - Parsed admin task-manager query.
  * @returns Tasks to inject on top of the database-backed items.
  */
 function collectAdminChatTasksToInject(
-    vpsSelfUpdateTask: AdminChatTaskRecord | null,
+    injectableTasks: ReadonlyArray<AdminChatTaskRecord>,
     query: ParsedAdminChatTaskQuery,
 ): ReadonlyArray<AdminChatTaskRecord> {
-    if (!vpsSelfUpdateTask) {
-        return [];
-    }
-
-    if (!matchesAdminChatTaskView(vpsSelfUpdateTask, query, Date.now())) {
-        return [];
-    }
-
-    if (!matchesAdminChatTaskSearch(vpsSelfUpdateTask, query.search)) {
-        return [];
-    }
-
-    return [vpsSelfUpdateTask];
+    const nowTimestamp = Date.now();
+    return injectableTasks.filter(
+        (task) => matchesAdminChatTaskView(task, query, nowTimestamp) && matchesAdminChatTaskSearch(task, query.search),
+    );
 }
 
 /**
@@ -160,34 +156,64 @@ function mergeInjectedAdminChatTasks(options: {
 }
 
 /**
- * Adds the injected self-update task to the summary counters so the header metrics stay accurate.
+ * Adds injected tasks to the summary counters so the header metrics stay accurate.
  *
  * @param databaseCounters - Counters computed from durable database rows.
- * @param vpsSelfUpdateTask - Currently persisted self-update task or `null`.
- * @returns Merged counters including the injected task.
+ * @param injectedTasks - Synthetic task rows collected from process-local state.
+ * @returns Merged counters including the injected tasks.
  */
 function mergeInjectedAdminChatTaskCounters(
     databaseCounters: AdminChatTaskCounters,
-    vpsSelfUpdateTask: AdminChatTaskRecord | null,
+    injectedTasks: ReadonlyArray<AdminChatTaskRecord>,
 ): AdminChatTaskCounters {
-    if (!vpsSelfUpdateTask) {
+    if (injectedTasks.length === 0) {
         return databaseCounters;
     }
 
     const nowTimestamp = Date.now();
-    const isRunning = vpsSelfUpdateTask.status === 'RUNNING';
-    const isFailedInWindow =
-        vpsSelfUpdateTask.status === 'FAILED' &&
-        isIsoTimestampAtOrAfter(vpsSelfUpdateTask.finishedAt, nowTimestamp - 24 * HOUR_IN_MILLISECONDS);
+    const injectedQueuedTimestamps = injectedTasks
+        .filter((task) => task.status === 'QUEUED')
+        .map((task) => Date.parse(task.queuedAt))
+        .filter((timestamp) => Number.isFinite(timestamp));
+    const oldestInjectedQueuedAgeMs =
+        injectedQueuedTimestamps.length === 0 ? null : nowTimestamp - Math.min(...injectedQueuedTimestamps);
 
     return {
-        runningCount: databaseCounters.runningCount + (isRunning ? 1 : 0),
-        queuedCount: databaseCounters.queuedCount,
-        failedLast24hCount: databaseCounters.failedLast24hCount + (isFailedInWindow ? 1 : 0),
-        oldestQueuedAgeMs: databaseCounters.oldestQueuedAgeMs,
+        runningCount: databaseCounters.runningCount + injectedTasks.filter((task) => task.status === 'RUNNING').length,
+        queuedCount: databaseCounters.queuedCount + injectedTasks.filter((task) => task.status === 'QUEUED').length,
+        failedLast24hCount:
+            databaseCounters.failedLast24hCount +
+            injectedTasks.filter(
+                (task) =>
+                    task.status === 'FAILED' &&
+                    isIsoTimestampAtOrAfter(task.finishedAt, nowTimestamp - 24 * HOUR_IN_MILLISECONDS),
+            ).length,
+        oldestQueuedAgeMs: mergeOldestQueuedAge(databaseCounters.oldestQueuedAgeMs, oldestInjectedQueuedAgeMs),
     };
 }
 
+/**
+ * Merges durable and injected queued-task ages.
+ *
+ * @param databaseOldestQueuedAgeMs - Oldest durable queued-task age.
+ * @param injectedOldestQueuedAgeMs - Oldest injected queued-task age.
+ * @returns Oldest queued age across both sources.
+ */
+function mergeOldestQueuedAge(
+    databaseOldestQueuedAgeMs: number | null,
+    injectedOldestQueuedAgeMs: number | null,
+): number | null {
+    if (databaseOldestQueuedAgeMs === null) {
+        return injectedOldestQueuedAgeMs;
+    }
+
+    if (injectedOldestQueuedAgeMs === null) {
+        return databaseOldestQueuedAgeMs;
+    }
+
+    return Math.max(databaseOldestQueuedAgeMs, injectedOldestQueuedAgeMs);
+}
+
 /**
  * Returns whether the injected task belongs in the requested admin task-manager view.
  *

package/apps/agents-server/src/utils/pagePreviewBrowserSessions.ts

@@ -0,0 +1,234 @@
+import type { Page } from 'playwright';
+import type { AdminChatTaskRecord } from './chatTasksAdmin';
+import type { UserInfo } from './getCurrentUser';
+
+/**
+ * Prefix for browser-preview session identifiers created by the citation preview UI.
+ *
+ * @private internal constant of Agents Server page-preview streaming
+ */
+const PAGE_PREVIEW_BROWSER_SESSION_ID_PREFIX = 'page-preview-';
+
+/**
+ * Pattern accepted for client-created page-preview session identifiers.
+ *
+ * @private internal constant of Agents Server page-preview streaming
+ */
+const PAGE_PREVIEW_BROWSER_SESSION_ID_PATTERN = /^page-preview-[a-z0-9-]{16,80}$/;
+
+/**
+ * Queue name used when active browser previews are surfaced in the admin task manager.
+ *
+ * @private internal constant of Agents Server page-preview streaming
+ */
+const PAGE_PREVIEW_BROWSER_TASK_QUEUE_NAME = 'page-preview-browser';
+
+/**
+ * Agent label used by the task manager for browser-preview rows.
+ *
+ * @private internal constant of Agents Server page-preview streaming
+ */
+const PAGE_PREVIEW_BROWSER_TASK_AGENT_NAME = 'Browser preview';
+
+/**
+ * In-memory record for one live browser preview stream.
+ *
+ * @private internal type of Agents Server page-preview streaming
+ */
+export type PagePreviewBrowserSession = {
+    readonly id: string;
+    readonly url: string;
+    readonly userId: number;
+    readonly username: string | null;
+    readonly createdAt: string;
+    readonly startedAt: string;
+    readonly processId: number | null;
+    page: Page | null;
+    viewport: PagePreviewBrowserViewport | null;
+    updatedAt: string;
+    lastFrameAt: string | null;
+};
+
+/**
+ * Browser viewport used to convert pointer ratios into Playwright coordinates.
+ *
+ * @private internal type of Agents Server page-preview streaming
+ */
+export type PagePreviewBrowserViewport = {
+    readonly width: number;
+    readonly height: number;
+};
+
+/**
+ * Options used when registering one browser preview stream.
+ *
+ * @private internal type of Agents Server page-preview streaming
+ */
+export type RegisterPagePreviewBrowserSessionOptions = {
+    readonly sessionId: string;
+    readonly url: string;
+    readonly user: UserInfo;
+};
+
+/**
+ * Shared in-memory browser-preview registry.
+ *
+ * The route streams are short-lived and tied to the current Node process, so a
+ * durable database row would create stale task-manager entries after restarts.
+ *
+ * @private internal constant of Agents Server page-preview streaming
+ */
+const pagePreviewBrowserSessions = new Map<string, PagePreviewBrowserSession>();
+
+/**
+ * Creates a normalized page-preview browser session id.
+ *
+ * @param requestedSessionId - Optional client-created session id.
+ * @returns Valid session id or `null` when the supplied value is invalid.
+ */
+export function normalizePagePreviewBrowserSessionId(requestedSessionId: string | null): string | null {
+    if (!requestedSessionId) {
+        return null;
+    }
+
+    const normalizedSessionId = requestedSessionId.trim().toLowerCase();
+    return PAGE_PREVIEW_BROWSER_SESSION_ID_PATTERN.test(normalizedSessionId) ? normalizedSessionId : null;
+}
+
+/**
+ * Registers or replaces one active browser preview session.
+ *
+ * @param options - Session metadata.
+ * @returns Registered session snapshot.
+ */
+export function registerPagePreviewBrowserSession(
+    options: RegisterPagePreviewBrowserSessionOptions,
+): PagePreviewBrowserSession {
+    const now = new Date().toISOString();
+    const previousSession = pagePreviewBrowserSessions.get(options.sessionId);
+    if (previousSession?.page) {
+        void previousSession.page.close().catch(() => undefined);
+    }
+
+    const session: PagePreviewBrowserSession = {
+        id: options.sessionId,
+        url: options.url,
+        userId: options.user.id ?? 0,
+        username: options.user.username,
+        createdAt: now,
+        startedAt: now,
+        updatedAt: now,
+        lastFrameAt: null,
+        processId: typeof process.pid === 'number' ? process.pid : null,
+        page: null,
+        viewport: null,
+    };
+
+    pagePreviewBrowserSessions.set(options.sessionId, session);
+    return session;
+}
+
+/**
+ * Attaches the Playwright page used by one live browser preview.
+ *
+ * @param sessionId - Active session id.
+ * @param page - Playwright page bound to the stream.
+ * @param viewport - Viewport used by the stream.
+ */
+export function attachPagePreviewBrowserSessionPage(
+    sessionId: string,
+    page: Page,
+    viewport: PagePreviewBrowserViewport,
+): void {
+    const session = pagePreviewBrowserSessions.get(sessionId);
+    if (!session) {
+        return;
+    }
+
+    session.page = page;
+    session.viewport = viewport;
+    session.updatedAt = new Date().toISOString();
+}
+
+/**
+ * Marks that one stream frame has been delivered.
+ *
+ * @param sessionId - Active session id.
+ */
+export function markPagePreviewBrowserSessionFrame(sessionId: string): void {
+    const session = pagePreviewBrowserSessions.get(sessionId);
+    if (!session) {
+        return;
+    }
+
+    const now = new Date().toISOString();
+    session.lastFrameAt = now;
+    session.updatedAt = now;
+}
+
+/**
+ * Removes one browser preview session from the active registry.
+ *
+ * @param sessionId - Active session id.
+ */
+export function finishPagePreviewBrowserSession(sessionId: string): void {
+    pagePreviewBrowserSessions.delete(sessionId);
+}
+
+/**
+ * Returns the active browser preview session when owned by the current user.
+ *
+ * @param sessionId - Browser preview session id.
+ * @param user - Current authenticated user.
+ * @returns Session or `null` when missing or owned by a different user.
+ */
+export function findUserPagePreviewBrowserSession(
+    sessionId: string,
+    user: UserInfo,
+): PagePreviewBrowserSession | null {
+    const session = pagePreviewBrowserSessions.get(sessionId);
+    if (!session) {
+        return null;
+    }
+
+    if ((user.id ?? 0) !== session.userId || user.username !== session.username) {
+        return null;
+    }
+
+    return session;
+}
+
+/**
+ * Lists active browser preview sessions as admin task-manager rows.
+ *
+ * @returns Task-manager rows for streams currently attached to this process.
+ */
+export function listPagePreviewBrowserAdminTasks(): Array<AdminChatTaskRecord> {
+    return Array.from(pagePreviewBrowserSessions.values()).map((session) => ({
+        id: session.id,
+        kind: 'BROWSER_PREVIEW',
+        status: 'RUNNING',
+        createdAt: session.createdAt,
+        queuedAt: session.createdAt,
+        startedAt: session.startedAt,
+        updatedAt: session.updatedAt,
+        finishedAt: null,
+        cancelRequestedAt: null,
+        pausedAt: null,
+        lastHeartbeatAt: session.lastFrameAt,
+        leaseExpiresAt: null,
+        recurrenceIntervalMs: null,
+        priority: null,
+        attemptCount: 1,
+        retryCount: 0,
+        lastErrorSummary: null,
+        lastErrorDetails: session.url,
+        userId: session.userId,
+        username: session.username,
+        agentPermanentId: PAGE_PREVIEW_BROWSER_SESSION_ID_PREFIX,
+        agentName: PAGE_PREVIEW_BROWSER_TASK_AGENT_NAME,
+        chatId: session.url,
+        workerId: session.processId === null ? null : String(session.processId),
+        queueName: PAGE_PREVIEW_BROWSER_TASK_QUEUE_NAME,
+    }));
+}

package/esm/index.es.js

@@ -59,7 +59,7 @@ const BOOK_LANGUAGE_VERSION = '2.0.0';
  * @generated
  * @see https://github.com/webgptorg/promptbook
  */
-const PROMPTBOOK_ENGINE_VERSION = '0.113.0-1';
+const PROMPTBOOK_ENGINE_VERSION = '0.113.0-3';
 /**
  * TODO: string_promptbook_version should be constrained to the all versions of Promptbook engine
  * Note: [💞] Ignore a discrepancy between file name and entity name
@@ -30182,9 +30182,10 @@ function buildAgentMessagePrompt(messageRelativePath, agentSystemMessage) {
 
                 [Up to 50,000 CZK](?message=We are posting an order up to 50,000 CZK)
                 [Up to 100,000 CZK](?message=We are posting an order up to 100,000 CZK)
-                [Over 100,000 CZK](?message=We are posting an order over 100,000 CZK)
+                [Over 100,000 CZK](?messageDraft=We are posting an order over 100,000 CZK)
                 \`\`\`
 
+                - You can use \`message\` for message that will be sent immediately after clicking the button, or \`messageDraft\` for message that will be prefilled in the input field for editing before sending.
 
                 ## This is how you should behave
 
@@ -54635,6 +54636,12 @@ const DEPRECATED_TOP_LEVEL_COMMAND_MESSAGES = {
     'start-agents-server': 'Use `ptbk agents-server start` instead.',
     'start-pipelines-server': OLD_PIPELINE_SYSTEM_DEPRECATION_MESSAGE,
 };
+/**
+ * Raw top-level CLI arguments that print the Promptbook version.
+ *
+ * @private internal constant of `promptbookCli`
+ */
+const VERSION_OPTION_ARGUMENTS = new Set(['-v', '--version']);
 /**
  * Runs CLI utilities of Promptbook package
  *
@@ -54649,8 +54656,13 @@ async function promptbookCli() {
 
             `));
     }
-    const cliArguments = process.argv.slice(2);
-    const isVerbose = cliArguments.some((arg) => arg === '--verbose' || arg === '-v');
+    const commandLineArguments = process.argv.slice(2);
+    const isVersionRequested = isTopLevelVersionRequested(commandLineArguments);
+    if (isVersionRequested) {
+        console.info(PROMPTBOOK_ENGINE_VERSION);
+        return process.exit(0);
+    }
+    const isVerbose = commandLineArguments.some((argument) => argument === '--verbose' || argument === '-v');
     //     <- TODO: Can be this be done with commander before the commander commands are initialized?
     if (isVerbose) {
         console.info(colors.gray(`Promptbook CLI version ${PROMPTBOOK_ENGINE_VERSION} in ${__filename.split('\\').join('/')}`));
@@ -54679,12 +54691,21 @@ async function promptbookCli() {
     $deprecateTopLevelCommands(program);
     // TODO: [🧠] Should it be here or not> $addGlobalOptionsToCommand(program);
     program.commands.forEach($addGlobalOptionsToCommand);
-    if (cliArguments.length === 0) {
+    if (commandLineArguments.length === 0) {
         program.outputHelp();
         return process.exit(0);
     }
     program.parse(process.argv);
 }
+/**
+ * Checks whether the invocation asks for the root Promptbook CLI version.
+ *
+ * @private internal utility of `promptbookCli`
+ */
+function isTopLevelVersionRequested(commandLineArguments) {
+    const firstCommandLineArgument = commandLineArguments[0];
+    return firstCommandLineArgument !== undefined && VERSION_OPTION_ARGUMENTS.has(firstCommandLineArgument);
+}
 /**
  * Adds one deprecation notice to each configured top-level legacy command.
  */