@kafca/agentdock 0.1.23 → 0.1.25

package/dist-electron/services/local-ai-core/src/acp/local-core-acp-store.js

@@ -6,6 +6,8 @@ const node_crypto_1 = require("node:crypto");
 const node_fs_1 = require("node:fs");
 const node_sqlite_1 = require("node:sqlite");
 const node_path_1 = require("node:path");
+const index_js_1 = require("../../../../packages/contracts/src/index.js");
+const job_id_js_1 = require("../scheduler/job-id.js");
 const desktop_js_1 = require("../../../../shared/desktop.js");
 const workspace_thread_mappers_js_1 = require("../thread/workspace-thread-mappers.js");
 const workspace_thread_id_js_1 = require("../thread/workspace-thread-id.js");
@@ -29,7 +31,8 @@ class LocalCoreAcpStore {
         history_count INTEGER NOT NULL DEFAULT 0,
         excerpt TEXT NOT NULL DEFAULT '',
         acp_session_id TEXT,
-        acp_supports_load INTEGER NOT NULL DEFAULT 0
+        acp_supports_load INTEGER NOT NULL DEFAULT 0,
+        agent_mode TEXT NOT NULL DEFAULT 'default'
       );
       CREATE INDEX IF NOT EXISTS idx_threads_workspace_updated ON threads (workspace_id, updated_at DESC);
       CREATE TABLE IF NOT EXISTS messages (
@@ -37,6 +40,7 @@ class LocalCoreAcpStore {
         thread_id TEXT NOT NULL,
         role TEXT NOT NULL,
         content TEXT NOT NULL,
+        tool_call_json TEXT,
         timestamp TEXT NOT NULL,
         kind TEXT NOT NULL,
         seq INTEGER NOT NULL,
@@ -217,14 +221,16 @@ class LocalCoreAcpStore {
       CREATE INDEX IF NOT EXISTS idx_audit_events_task_created ON audit_events (task_id, created_at DESC);
       CREATE INDEX IF NOT EXISTS idx_audit_events_type_created ON audit_events (type, created_at DESC);
     `);
+        this.ensureColumn('messages', 'tool_call_json', 'TEXT');
         this.ensureColumn('scheduled_jobs', 'execution_mode', "TEXT NOT NULL DEFAULT 'same-thread'");
+        this.ensureColumn('threads', 'agent_mode', "TEXT NOT NULL DEFAULT 'default'");
     }
     close() {
         this.db.close();
     }
     listThreadSummaries(workspaceId) {
         const rows = this.db.prepare(`
-      SELECT id, workspace_id, session_id, bridge_session_key, title, agent_type, created_at, updated_at, history_count, excerpt
+      SELECT id, workspace_id, session_id, bridge_session_key, title, agent_type, created_at, updated_at, history_count, excerpt, acp_session_id, acp_supports_load, agent_mode
       FROM threads
       WHERE workspace_id = ?
       ORDER BY updated_at DESC
@@ -247,15 +253,15 @@ class LocalCoreAcpStore {
         const row = this.db.prepare('SELECT COUNT(*) AS total FROM threads WHERE workspace_id = ?').get(workspaceId);
         return Number(row?.total || 0);
     }
-    createThread(workspaceId, title, agentType = desktop_js_1.LOCALCORE_ACP_AGENT_TYPE) {
+    createThread(workspaceId, title, agentType = desktop_js_1.LOCALCORE_ACP_AGENT_TYPE, agentMode = 'default') {
         const sessionId = (0, node_crypto_1.randomUUID)();
         const threadId = (0, workspace_thread_id_js_1.encodeThreadId)(workspaceId, sessionId);
         const now = new Date().toISOString();
         const bridgeSessionKey = `${desktop_js_1.LOCALCORE_ACP_AGENT_TYPE}:${workspaceId}:${sessionId}`;
         this.db.prepare(`
-      INSERT INTO threads (id, workspace_id, session_id, bridge_session_key, title, agent_type, created_at, updated_at, history_count, excerpt)
-      VALUES (?, ?, ?, ?, ?, ?, ?, ?, 0, '')
-    `).run(threadId, workspaceId, sessionId, bridgeSessionKey, title, agentType, now, now);
+      INSERT INTO threads (id, workspace_id, session_id, bridge_session_key, title, agent_type, created_at, updated_at, history_count, excerpt, agent_mode)
+      VALUES (?, ?, ?, ?, ?, ?, ?, ?, 0, '', ?)
+    `).run(threadId, workspaceId, sessionId, bridgeSessionKey, title, agentType, now, now, agentMode || 'default');
         return {
             id: threadId,
             workspaceId,
@@ -278,7 +284,7 @@ class LocalCoreAcpStore {
             throw new Error(`Thread not found: ${threadId}`);
         }
         const messages = this.db.prepare(`
-      SELECT id, thread_id, role, content, timestamp, kind, seq
+      SELECT id, thread_id, role, content, tool_call_json, timestamp, kind, seq
       FROM messages
       WHERE thread_id = ?
       ORDER BY seq ASC
@@ -298,6 +304,7 @@ class LocalCoreAcpStore {
                 id: message.id,
                 role: message.role,
                 content: message.content,
+                toolCall: parseJson(message.tool_call_json || 'null', null) || undefined,
                 timestamp: message.timestamp,
                 kind: message.kind,
             })),
@@ -312,7 +319,7 @@ class LocalCoreAcpStore {
     deleteThread(threadId) {
         this.db.prepare('DELETE FROM threads WHERE id = ?').run(threadId);
     }
-    appendMessage(threadId, role, content, kind) {
+    appendMessage(threadId, role, content, kind, toolCall) {
         const timestamp = new Date().toISOString();
         const nextSequenceRow = this.db.prepare('SELECT COALESCE(MAX(seq), -1) + 1 AS next_seq FROM messages WHERE thread_id = ?').get(threadId);
         const nextSeq = Number(nextSequenceRow?.next_seq || 0);
@@ -321,9 +328,9 @@ class LocalCoreAcpStore {
         this.db.exec('BEGIN IMMEDIATE');
         try {
             this.db.prepare(`
-        INSERT INTO messages (id, thread_id, role, content, timestamp, kind, seq)
-        VALUES (?, ?, ?, ?, ?, ?, ?)
-      `).run(id, threadId, role, content, timestamp, kind, nextSeq);
+        INSERT INTO messages (id, thread_id, role, content, tool_call_json, timestamp, kind, seq)
+        VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+      `).run(id, threadId, role, content, toolCall ? JSON.stringify(toolCall) : null, timestamp, kind, nextSeq);
             this.db.prepare(`
         UPDATE threads
         SET updated_at = ?, history_count = history_count + 1, excerpt = ?
@@ -337,7 +344,7 @@ class LocalCoreAcpStore {
         }
         return { id, timestamp };
     }
-    upsertMessage(threadId, id, role, content, kind) {
+    upsertMessage(threadId, id, role, content, kind, toolCall) {
         const timestamp = new Date().toISOString();
         const excerpt = (0, workspace_thread_mappers_js_1.normalizeMessageContent)(content);
         const existing = this.db.prepare('SELECT id FROM messages WHERE id = ? AND thread_id = ?').get(id, threadId);
@@ -346,17 +353,17 @@ class LocalCoreAcpStore {
             if (existing) {
                 this.db.prepare(`
           UPDATE messages
-          SET content = ?, timestamp = ?, kind = ?
+          SET content = ?, tool_call_json = ?, timestamp = ?, kind = ?
           WHERE id = ? AND thread_id = ?
-        `).run(content, timestamp, kind, id, threadId);
+        `).run(content, toolCall ? JSON.stringify(toolCall) : null, timestamp, kind, id, threadId);
             }
             else {
                 const nextSequenceRow = this.db.prepare('SELECT COALESCE(MAX(seq), -1) + 1 AS next_seq FROM messages WHERE thread_id = ?').get(threadId);
                 const nextSeq = Number(nextSequenceRow?.next_seq || 0);
                 this.db.prepare(`
-          INSERT INTO messages (id, thread_id, role, content, timestamp, kind, seq)
-          VALUES (?, ?, ?, ?, ?, ?, ?)
-        `).run(id, threadId, role, content, timestamp, kind, nextSeq);
+          INSERT INTO messages (id, thread_id, role, content, tool_call_json, timestamp, kind, seq)
+          VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+        `).run(id, threadId, role, content, toolCall ? JSON.stringify(toolCall) : null, timestamp, kind, nextSeq);
                 this.db.prepare(`
           UPDATE threads
           SET history_count = history_count + 1
@@ -378,31 +385,34 @@ class LocalCoreAcpStore {
     }
     updateRun(runId, threadId, status) {
         const now = new Date().toISOString();
+        const normalizedStatus = (0, index_js_1.normalizeRunStatus)(status);
         const existing = this.db.prepare('SELECT id FROM runs WHERE id = ?').get(runId);
         if (existing) {
-            this.db.prepare('UPDATE runs SET status = ?, updated_at = ? WHERE id = ?').run(status, now, runId);
+            this.db.prepare('UPDATE runs SET status = ?, updated_at = ? WHERE id = ?').run(normalizedStatus, now, runId);
             return;
         }
         this.db.prepare(`
       INSERT INTO runs (id, thread_id, status, started_at, updated_at)
       VALUES (?, ?, ?, ?, ?)
-    `).run(runId, threadId, status, now, now);
+    `).run(runId, threadId, normalizedStatus, now, now);
     }
     getLatestRunForThread(threadId) {
-        return this.db.prepare(`
+        const row = this.db.prepare(`
       SELECT id, thread_id, status, started_at, updated_at
       FROM runs
       WHERE thread_id = ?
       ORDER BY updated_at DESC
       LIMIT 1
     `).get(threadId);
+        return row ? { ...row, status: (0, index_js_1.normalizeRunStatus)(row.status) } : undefined;
     }
     getRun(runId) {
-        return this.db.prepare(`
+        const row = this.db.prepare(`
       SELECT id, thread_id, status, started_at, updated_at
       FROM runs
       WHERE id = ?
     `).get(runId);
+        return row ? { ...row, status: (0, index_js_1.normalizeRunStatus)(row.status) } : undefined;
     }
     listWorkspaceRegistry() {
         const rows = this.db.prepare(`
@@ -573,7 +583,7 @@ class LocalCoreAcpStore {
             params.push(query.taskId);
         }
         if (query.status) {
-            const statuses = Array.isArray(query.status) ? query.status : [query.status];
+            const statuses = (Array.isArray(query.status) ? query.status : [query.status]).map((status) => (0, index_js_1.normalizeApprovalRequestStatus)(status));
             predicates.push(`status IN (${statuses.map(() => '?').join(', ')})`);
             params.push(...statuses);
         }
@@ -669,7 +679,7 @@ class LocalCoreAcpStore {
     createAgentTask(input) {
         const id = `task:${(0, node_crypto_1.randomUUID)()}`;
         const now = new Date().toISOString();
-        const status = input.status || 'created';
+        const status = (0, index_js_1.normalizeAgentTaskStatus)(input.status);
         const timeline = [{
                 id: `timeline:${(0, node_crypto_1.randomUUID)()}`,
                 type: 'status_change',
@@ -706,7 +716,7 @@ class LocalCoreAcpStore {
             params.push(query.runtimeId);
         }
         if (query.status) {
-            const statuses = Array.isArray(query.status) ? query.status : [query.status];
+            const statuses = (Array.isArray(query.status) ? query.status : [query.status]).map((status) => (0, index_js_1.normalizeAgentTaskStatus)(status));
             predicates.push(`status IN (${statuses.map(() => '?').join(', ')})`);
             params.push(...statuses);
         }
@@ -735,17 +745,17 @@ class LocalCoreAcpStore {
             throw new Error(`Task not found: ${taskId}`);
         }
         const now = new Date().toISOString();
-        const nextStatus = input.status || existing.status;
+        const nextStatus = input.status ? (0, index_js_1.normalizeAgentTaskStatus)(input.status) : existing.status;
         const timeline = [...existing.timeline];
         const logs = [...existing.logs];
         const artifacts = [...existing.artifacts];
         const approvalIds = [...existing.approvalIds];
-        if (input.status && input.status !== existing.status) {
+        if (input.status && nextStatus !== existing.status) {
             timeline.push({
                 id: `timeline:${(0, node_crypto_1.randomUUID)()}`,
                 type: 'status_change',
-                title: `Task ${input.status}`,
-                status: input.status,
+                title: `Task ${nextStatus}`,
+                status: nextStatus,
                 timestamp: now,
             });
         }
@@ -780,14 +790,14 @@ class LocalCoreAcpStore {
       WHERE id = ?
     `).run(nextStatus, input.threadId || existing.threadId || null, input.runId || existing.runId || null, input.title || existing.title, now, existing.queuedAt || (nextStatus === 'queued' ? now : null), existing.startedAt || (nextStatus === 'running' ? now : null), existing.completedAt || (['completed', 'failed', 'cancelled'].includes(nextStatus) ? now : null), input.summary ?? existing.summary ?? null, input.error === null ? null : input.error ?? existing.error ?? null, JSON.stringify(timeline), JSON.stringify(logs), JSON.stringify(artifacts), JSON.stringify(approvalIds), JSON.stringify(input.metadata || existing.metadata || {}), taskId);
         const task = this.getAgentTask(taskId);
-        if (input.status && input.status !== existing.status) {
+        if (input.status && nextStatus !== existing.status) {
             this.createAuditEvent({
                 type: 'task.updated',
                 workspaceId: task.workspaceId,
                 taskId: task.taskId,
                 actor: 'local',
-                summary: `Task status changed to ${input.status}.`,
-                metadata: { previousStatus: existing.status, status: input.status, runId: task.runId },
+                summary: `Task status changed to ${nextStatus}.`,
+                metadata: { previousStatus: existing.status, status: nextStatus, runId: task.runId },
             });
         }
         return task;
@@ -820,14 +830,20 @@ class LocalCoreAcpStore {
         return row ? this.toScheduledJob(row) : undefined;
     }
     createScheduledJob(input) {
-        const id = `job:${(0, node_crypto_1.randomUUID)()}`;
+        let id = (0, job_id_js_1.createScheduledJobId)();
+        for (let attempt = 0; attempt < 5 && this.getScheduledJob(id); attempt += 1) {
+            id = (0, job_id_js_1.createScheduledJobId)();
+        }
+        if (this.getScheduledJob(id)) {
+            throw new Error('Unable to allocate a unique scheduled job id.');
+        }
         const now = new Date().toISOString();
         this.db.prepare(`
       INSERT INTO scheduled_jobs (
         id, workspace_id, platform, route_type, route_config, execution_mode, trigger_type, cron_expr, run_at,
         prompt_template, description, enabled, concurrency_policy, created_at, updated_at
       ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, 'skip_if_running', ?, ?)
-    `).run(id, input.workspaceId, input.platform, input.route.type, JSON.stringify(input.route), input.executionMode || 'same-thread', input.triggerType, input.cronExpr || null, input.runAt || null, input.promptTemplate, input.description || '', input.enabled === false ? 0 : 1, now, now);
+    `).run(id, input.workspaceId, (0, index_js_1.normalizeChannelPlatform)(input.platform), input.route.type, JSON.stringify(input.route), (0, index_js_1.normalizeScheduledJobExecutionMode)(input.executionMode), (0, index_js_1.normalizeScheduledJobTriggerType)(input.triggerType), input.cronExpr || null, input.runAt || null, input.promptTemplate, input.description || '', input.enabled === false ? 0 : 1, now, now);
         return this.getScheduledJob(id);
     }
     updateScheduledJob(jobId, input) {
@@ -838,8 +854,8 @@ class LocalCoreAcpStore {
         const next = {
             ...existing,
             ...(input.route ? { route: input.route } : {}),
-            ...(input.executionMode ? { executionMode: input.executionMode } : {}),
-            ...(input.triggerType ? { triggerType: input.triggerType } : {}),
+            ...(input.executionMode ? { executionMode: (0, index_js_1.normalizeScheduledJobExecutionMode)(input.executionMode) } : {}),
+            ...(input.triggerType ? { triggerType: (0, index_js_1.normalizeScheduledJobTriggerType)(input.triggerType) } : {}),
             ...(Object.prototype.hasOwnProperty.call(input, 'cronExpr') ? { cronExpr: input.cronExpr || undefined } : {}),
             ...(Object.prototype.hasOwnProperty.call(input, 'runAt') ? { runAt: input.runAt || undefined } : {}),
             ...(typeof input.promptTemplate === 'string' ? { promptTemplate: input.promptTemplate } : {}),
@@ -856,7 +872,10 @@ class LocalCoreAcpStore {
         return this.getScheduledJob(jobId);
     }
     deleteScheduledJob(jobId) {
-        this.db.prepare('DELETE FROM scheduled_jobs WHERE id = ?').run(jobId);
+        const result = this.db.prepare('DELETE FROM scheduled_jobs WHERE id = ?').run(jobId);
+        if (result.changes === 0) {
+            throw new Error(`Scheduled job not found: ${jobId}`);
+        }
         return { deleted: true };
     }
     listScheduledJobRuns(jobId) {
@@ -871,13 +890,14 @@ class LocalCoreAcpStore {
     createScheduledJobRun(jobId, status, input = {}) {
         const id = `jobrun:${(0, node_crypto_1.randomUUID)()}`;
         const triggeredAt = input.triggeredAt || new Date().toISOString();
+        const normalizedStatus = (0, index_js_1.normalizeScheduledJobRunStatus)(status);
         this.db.prepare(`
       INSERT INTO scheduled_job_runs (id, job_id, status, triggered_at, started_at, finished_at, error, thread_id, run_id, platform_message_id)
       VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
-    `).run(id, jobId, status, triggeredAt, input.startedAt || null, input.finishedAt || null, input.error || null, input.threadId || null, input.runId || null, input.platformMessageId || null);
+    `).run(id, jobId, normalizedStatus, triggeredAt, input.startedAt || null, input.finishedAt || null, input.error || null, input.threadId || null, input.runId || null, input.platformMessageId || null);
         this.updateScheduledJobStatus(jobId, {
             lastRunAt: triggeredAt,
-            lastStatus: status,
+            lastStatus: normalizedStatus,
             lastError: input.error || '',
         });
         return this.getScheduledJobRun(id);
@@ -898,6 +918,7 @@ class LocalCoreAcpStore {
         const next = {
             ...existing,
             ...input,
+            ...(input.status ? { status: (0, index_js_1.normalizeScheduledJobRunStatus)(input.status) } : {}),
         };
         this.db.prepare(`
       UPDATE scheduled_job_runs
@@ -926,11 +947,18 @@ class LocalCoreAcpStore {
     }
     getThreadRow(threadId) {
         return this.db.prepare(`
-      SELECT id, workspace_id, session_id, bridge_session_key, title, agent_type, created_at, updated_at, history_count, excerpt, acp_session_id, acp_supports_load
+      SELECT id, workspace_id, session_id, bridge_session_key, title, agent_type, created_at, updated_at, history_count, excerpt, acp_session_id, acp_supports_load, agent_mode
       FROM threads
       WHERE id = ?
     `).get(threadId);
     }
+    updateThreadAgentMode(threadId, mode) {
+        this.db.prepare(`
+      UPDATE threads
+      SET agent_mode = ?, updated_at = ?
+      WHERE id = ?
+    `).run(mode, new Date().toISOString(), threadId);
+    }
     updateThreadSession(threadId, sessionId, supportsLoad) {
         this.db.prepare(`
       UPDATE threads
@@ -1113,10 +1141,10 @@ class LocalCoreAcpStore {
         return {
             id: row.id,
             workspaceId: row.workspace_id,
-            platform: row.platform,
+            platform: (0, index_js_1.normalizeChannelPlatform)(row.platform),
             route,
-            executionMode: row.execution_mode || 'same-thread',
-            triggerType: row.trigger_type,
+            executionMode: (0, index_js_1.normalizeScheduledJobExecutionMode)(row.execution_mode),
+            triggerType: (0, index_js_1.normalizeScheduledJobTriggerType)(row.trigger_type),
             cronExpr: row.cron_expr || undefined,
             runAt: row.run_at || undefined,
             promptTemplate: row.prompt_template,
@@ -1126,7 +1154,7 @@ class LocalCoreAcpStore {
             createdAt: row.created_at,
             updatedAt: row.updated_at,
             lastRunAt: row.last_run_at || undefined,
-            lastStatus: row.last_status || undefined,
+            lastStatus: row.last_status ? (0, index_js_1.normalizeScheduledJobRunStatus)(row.last_status) : undefined,
             lastError: row.last_error || undefined,
         };
     }
@@ -1134,7 +1162,7 @@ class LocalCoreAcpStore {
         return {
             id: row.id,
             jobId: row.job_id,
-            status: row.status,
+            status: (0, index_js_1.normalizeScheduledJobRunStatus)(row.status),
             triggeredAt: row.triggered_at,
             startedAt: row.started_at || undefined,
             finishedAt: row.finished_at || undefined,
@@ -1187,7 +1215,7 @@ class LocalCoreAcpStore {
             runId: row.run_id || undefined,
             title: row.title,
             prompt: row.prompt || undefined,
-            status: row.status,
+            status: (0, index_js_1.normalizeAgentTaskStatus)(row.status),
             createdAt: row.created_at,
             updatedAt: row.updated_at,
             queuedAt: row.queued_at || undefined,
@@ -1224,7 +1252,7 @@ class LocalCoreAcpStore {
             runId: row.run_id || undefined,
             deviceId: row.device_id,
             kind: row.kind,
-            status: row.status,
+            status: (0, index_js_1.normalizeApprovalRequestStatus)(row.status),
             riskLevel: row.risk_level,
             title: row.title,
             description: row.description,