@promptbook/cli 0.112.0-111 → 0.112.0-113

package/apps/agents-server/src/utils/serverManagement/createManagedServer/seedServerDefaultAgents.ts

@@ -1,5 +1,6 @@
 import { createAgentPersistenceRecords } from '../../../../../../src/collection/agent-collection/constructors/agent-collection-in-supabase/createAgentPersistenceRecords';
 import type { Client } from 'pg';
+import { DEFAULT_AGENT_VISIBILITY } from '../../agentVisibility';
 import { loadDefaultAgentBooks } from '../../defaultAgents/loadDefaultAgentBooks';
 import { createInsertStatement, quoteIdentifier, type SqlRecorder } from './createSqlRecorder';
 import type { NormalizedCreateServerInput } from './normalizeCreateServerInput';
@@ -28,7 +29,7 @@ export async function seedServerDefaultAgents(
         const createdAt = new Date().toISOString();
         const { agentInsertRecord, agentHistoryInsertRecord } = createAgentPersistenceRecords(
             defaultAgentBook,
-            { sortOrder: index },
+            { sortOrder: index, visibility: DEFAULT_AGENT_VISIBILITY },
             createdAt,
         );
 
@@ -45,9 +46,10 @@ export async function seedServerDefaultAgents(
                     "promptbookEngineVersion",
                     "usage",
                     "folderId",
-                    "sortOrder"
+                    "sortOrder",
+                    "visibility"
                 )
-                VALUES ($1, $2, $3, $4, $5, $6, $7::jsonb, $8, $9::jsonb, $10, $11)
+                VALUES ($1, $2, $3, $4, $5, $6, $7::jsonb, $8, $9::jsonb, $10, $11, $12)
             `,
             [
                 agentInsertRecord.agentName,
@@ -61,6 +63,7 @@ export async function seedServerDefaultAgents(
                 JSON.stringify(agentInsertRecord.usage),
                 agentInsertRecord.folderId ?? null,
                 agentInsertRecord.sortOrder ?? index,
+                agentInsertRecord.visibility ?? DEFAULT_AGENT_VISIBILITY,
             ],
         );
         sqlRecorder.addStatement(
@@ -70,6 +73,7 @@ export async function seedServerDefaultAgents(
                 usage: JSON.stringify(agentInsertRecord.usage),
                 folderId: agentInsertRecord.folderId ?? null,
                 sortOrder: agentInsertRecord.sortOrder ?? index,
+                visibility: agentInsertRecord.visibility ?? DEFAULT_AGENT_VISIBILITY,
             }),
         );
 

package/apps/agents-server/src/utils/serverManagement/createManagedServer.ts

@@ -108,6 +108,11 @@ export type CreateServerInput = {
      */
     readonly additionalUsers?: ReadonlyArray<ServerSeedUserInput>;
 
+    /**
+     * Whether bundled default agents from `agents/default` should be created during bootstrap.
+     */
+    readonly isDefaultAgentsInstalled?: boolean;
+
     /**
      * Initial metadata values for the new server.
      */

package/apps/agents-server/src/utils/userChat/listUserChats.ts

@@ -1,5 +1,7 @@
 import { $getTableName } from '@/src/database/$getTableName';
 import { $provideClientSql } from '@/src/database/$provideClientSql';
+import { isAgentsServerSqliteMode } from '@/src/database/agentsServerDatabaseMode';
+import { $provideAgentsServerSqliteDatabase } from '@/src/database/sqlite/$provideAgentsServerSqliteDatabase';
 import type { ListUserChatsOptions, UserChatRecord } from './UserChatRecord';
 import type { UserChatSource } from './UserChatSource';
 import type { UserChatRow } from './UserChatRow';
@@ -30,6 +32,13 @@ const POSTGRES_UNDEFINED_COLUMN_ERROR_CODE = '42703';
 const CLIENT_SQL_MISSING_CONNECTION_MESSAGE_FRAGMENT =
     'Environment variable `POSTGRES_URL` or `DATABASE_URL` must be defined.';
 
+/**
+ * SQLite expression that safely exposes chat messages as a JSON array.
+ *
+ * @private function of `userChat`
+ */
+const SQLITE_CHAT_MESSAGES_JSON_EXPRESSION = `CASE WHEN json_valid(chat."messages") THEN chat."messages" ELSE '[]' END`;
+
 /**
  * Lists all user chats for one agent ordered by last activity.
  */
@@ -69,6 +78,10 @@ export async function listUserChats(options: ListUserChatsOptions): Promise<Arra
  * @private function of `userChat`
  */
 export async function listUserChatSummarySeeds(options: ListUserChatsOptions): Promise<Array<UserChatSummarySeed>> {
+    if (isAgentsServerSqliteMode()) {
+        return listUserChatSummarySeedsViaSqlite(options);
+    }
+
     if (!isDirectSqlConnectionConfigured()) {
         return listUserChatSummarySeedsViaSupabase(options);
     }
@@ -151,6 +164,92 @@ export async function listUserChatSummarySeeds(options: ListUserChatsOptions): P
     }
 }
 
+/**
+ * Lists lightweight chat-summary seeds through direct SQLite JSON queries.
+ *
+ * @private function of `userChat`
+ */
+async function listUserChatSummarySeedsViaSqlite(options: ListUserChatsOptions): Promise<Array<UserChatSummarySeed>> {
+    const userChatTableName = quoteIdentifier(await $getTableName('UserChat'));
+    const shouldLoadExternalChats = options.viewerIsAdmin && options.includeExternalChats;
+    const whereClause = shouldLoadExternalChats
+        ? `
+            chat."agentPermanentId" = ?
+            AND (chat."source" <> ? OR chat."userId" = ?)
+        `
+        : `
+            chat."userId" = ?
+            AND chat."agentPermanentId" = ?
+            AND chat."source" = ?
+        `;
+    const queryValues = shouldLoadExternalChats
+        ? [options.agentPermanentId, USER_CHAT_SOURCES.WEB_UI, options.userId]
+        : [options.userId, options.agentPermanentId, USER_CHAT_SOURCES.WEB_UI];
+
+    try {
+        const database = $provideAgentsServerSqliteDatabase();
+        const summaryRows = database
+            .prepare(
+                `
+                    SELECT
+                        chat."id",
+                        chat."createdAt",
+                        chat."updatedAt",
+                        chat."lastMessageAt",
+                        chat."title",
+                        chat."source",
+                        COALESCE(json_array_length(${SQLITE_CHAT_MESSAGES_JSON_EXPRESSION}), 0) AS "messagesCount",
+                        COALESCE(
+                            (
+                                SELECT CAST(json_extract(message.value, '$.content') AS TEXT)
+                                FROM json_each(${SQLITE_CHAT_MESSAGES_JSON_EXPRESSION}) AS message
+                                WHERE UPPER(CAST(COALESCE(json_extract(message.value, '$.sender'), '') AS TEXT)) = 'USER'
+                                ORDER BY CAST(message.key AS INTEGER) ASC
+                                LIMIT 1
+                            ),
+                            ''
+                        ) AS "firstUserMessageContent",
+                        COALESCE(
+                            (
+                                SELECT CAST(json_extract(message.value, '$.content') AS TEXT)
+                                FROM json_each(${SQLITE_CHAT_MESSAGES_JSON_EXPRESSION}) AS message
+                                WHERE LENGTH(TRIM(CAST(COALESCE(json_extract(message.value, '$.content'), '') AS TEXT))) > 0
+                                ORDER BY CAST(message.key AS INTEGER) DESC
+                                LIMIT 1
+                            ),
+                            ''
+                        ) AS "lastPreviewMessageContent",
+                        COALESCE(
+                            (
+                                SELECT COUNT(*)
+                                FROM json_each(${SQLITE_CHAT_MESSAGES_JSON_EXPRESSION}) AS message
+                                WHERE
+                                    UPPER(CAST(COALESCE(json_extract(message.value, '$.sender'), '') AS TEXT)) IN ('AGENT', 'MODEL')
+                                    AND (
+                                        json_extract(message.value, '$.isComplete') = 0
+                                        OR LOWER(CAST(COALESCE(json_extract(message.value, '$.isComplete'), '') AS TEXT)) = 'false'
+                                        OR LOWER(CAST(COALESCE(json_extract(message.value, '$.lifecycleState'), '') AS TEXT)) IN ('queued', 'running')
+                                    )
+                            ),
+                            0
+                        ) AS "pendingAssistantMessageCount"
+                    FROM ${userChatTableName} AS chat
+                    WHERE ${whereClause}
+                    ORDER BY chat."lastMessageAt" IS NULL ASC, chat."lastMessageAt" DESC, chat."updatedAt" DESC
+                `,
+            )
+            .all(...queryValues) as Array<UserChatSummarySeedSqlRow>;
+
+        return summaryRows.map(mapUserChatSummarySeedSqlRow);
+    } catch (error) {
+        if (!isUserChatSummarySeedSqliteFallbackError(error)) {
+            throw error;
+        }
+
+        return listUserChatSummarySeedsViaSupabase(options);
+    }
+}
+
 /**
  * Raw SQL row shape loaded for lightweight chat summary seeds.
  *
@@ -260,6 +359,16 @@ function isUserChatSummarySeedSqlFallbackError(error: unknown): boolean {
     return /relation .* does not exist|column .* does not exist/i.test(errorMessage);
 }
 
+/**
+ * Returns true when SQLite summary optimization should gracefully fallback to Supabase-shaped reads.
+ *
+ * @private function of `userChat`
+ */
+function isUserChatSummarySeedSqliteFallbackError(error: unknown): boolean {
+    const errorMessage = error instanceof Error ? error.message : String(error);
+    return /no such table|no such column|no such function: json_|malformed JSON/i.test(errorMessage);
+}
+
 /**
  * Loads chat-summary seeds using standard Supabase reads when SQL optimization is unavailable.
  *

package/apps/agents-server/src/utils/userChat/triggerUserChatJobWorker.ts

@@ -1,28 +1,63 @@
 import { resolveUserChatWorkerInternalToken } from './resolveUserChatWorkerInternalToken';
+import { retryWithBackoff } from '../retryWithBackoff';
+
+/**
+ * Number of retries for transient transport failures while waking the durable chat worker.
+ */
+const USER_CHAT_WORKER_TRIGGER_FETCH_RETRIES = 2;
+
+/**
+ * Initial wait before retrying a failed worker wake-up request.
+ */
+const USER_CHAT_WORKER_TRIGGER_FETCH_INITIAL_DELAY_MS = 250;
+
+/**
+ * Maximum wait before retrying a failed worker wake-up request.
+ */
+const USER_CHAT_WORKER_TRIGGER_FETCH_MAX_DELAY_MS = 1_000;
+
+/**
+ * Multiplier used between worker wake-up retry waits.
+ */
+const USER_CHAT_WORKER_TRIGGER_FETCH_BACKOFF_FACTOR = 2;
+
+/**
+ * Randomized extra delay ratio for worker wake-up retry waits.
+ */
+const USER_CHAT_WORKER_TRIGGER_FETCH_JITTER_RATIO = 0.2;
 
 /**
  * Schedules one best-effort internal worker invocation for durable chat jobs.
  */
-export async function triggerUserChatJobWorker(options: {
-    origin: string;
-    preferredJobId?: string;
-}): Promise<void> {
+export async function triggerUserChatJobWorker(options: { origin: string; preferredJobId?: string }): Promise<void> {
     const workerUrl = new URL('/api/internal/user-chat-jobs/run', ensureTrailingSlashlessOrigin(options.origin));
-    const response = await fetch(workerUrl, {
-        method: 'POST',
-        cache: 'no-store',
-        headers: {
-            'Content-Type': 'application/json',
-            'x-user-chat-worker-token': resolveUserChatWorkerInternalToken(),
-        },
-        body: JSON.stringify(
-            options.preferredJobId
-                ? {
-                      preferredJobId: options.preferredJobId,
-                  }
-                : {},
-        ),
-    });
+    const response = (
+        await retryWithBackoff(
+            async () =>
+                await fetch(workerUrl, {
+                    method: 'POST',
+                    cache: 'no-store',
+                    headers: {
+                        'Content-Type': 'application/json',
+                        'x-user-chat-worker-token': resolveUserChatWorkerInternalToken(),
+                    },
+                    body: JSON.stringify(
+                        options.preferredJobId
+                            ? {
+                                  preferredJobId: options.preferredJobId,
+                              }
+                            : {},
+                    ),
+                }),
+            {
+                retries: USER_CHAT_WORKER_TRIGGER_FETCH_RETRIES,
+                initialDelayMs: USER_CHAT_WORKER_TRIGGER_FETCH_INITIAL_DELAY_MS,
+                maxDelayMs: USER_CHAT_WORKER_TRIGGER_FETCH_MAX_DELAY_MS,
+                backoffFactor: USER_CHAT_WORKER_TRIGGER_FETCH_BACKOFF_FACTOR,
+                jitterRatio: USER_CHAT_WORKER_TRIGGER_FETCH_JITTER_RATIO,
+            },
+        )
+    ).value;
 
     if (!response.ok && response.status !== 204) {
         throw new Error(`Failed to trigger user chat worker: ${response.status} ${response.statusText}`);

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

@@ -23,7 +23,7 @@ export const VPS_ENVIRONMENT_VARIABLE_KEYS = [
     'NEXT_PUBLIC_SITE_URL',
     'ADMIN_PASSWORD',
     'OPENAI_API_KEY',
-    'PTBK_AGENT',
+    'PTBK_HARNESS',
     'PTBK_MODEL',
     'PTBK_THINKING_LEVEL',
     'PORT',