@taskforcehq/taskforce 0.3.326 → 0.3.328

package/dist/storage/taskforceAgentConversationStore.js

@@ -0,0 +1,264 @@
+import { randomUUID } from 'crypto';
+import { normalizePersistedModelGatewayMessages } from '../services/modelGateway.js';
+function normalizeId(value) {
+    return String(value || '').trim();
+}
+function normalizeWorkspaceId(value) {
+    return normalizeId(value) || 'default';
+}
+function normalizeMessages(messages) {
+    return normalizePersistedModelGatewayMessages(messages);
+}
+function titleFromMessages(messages) {
+    const firstUserMessage = messages.find((message) => message.role === 'user')?.content || 'Agent chat';
+    return firstUserMessage.length > 80 ? `${firstUserMessage.slice(0, 77)}...` : firstUserMessage;
+}
+function normalizeTitle(value) {
+    const title = String(value || '').trim();
+    if (!title)
+        return '';
+    return title.length > 120 ? title.slice(0, 120) : title;
+}
+function parseJson(value, fallback) {
+    try {
+        const parsed = JSON.parse(String(value || ''));
+        return parsed === null || parsed === undefined ? fallback : parsed;
+    }
+    catch {
+        return fallback;
+    }
+}
+function mapRow(row) {
+    return {
+        id: String(row.id || ''),
+        tenantId: String(row.tenant_id || 'default'),
+        workspaceId: String(row.workspace_id || 'default'),
+        agentId: String(row.agent_id || ''),
+        taskId: row.task_id ? String(row.task_id) : null,
+        title: String(row.title || 'Agent chat'),
+        messages: normalizeMessages(parseJson(row.messages_json, [])),
+        modelProvider: row.model_provider ? String(row.model_provider) : null,
+        modelId: row.model_id ? String(row.model_id) : null,
+        usage: parseJson(row.usage_json, null),
+        latencyMs: row.latency_ms === null || row.latency_ms === undefined ? null : Number(row.latency_ms),
+        createdByUserId: row.created_by_user_id ? String(row.created_by_user_id) : null,
+        updatedByUserId: row.updated_by_user_id ? String(row.updated_by_user_id) : null,
+        createdAt: String(row.created_at || ''),
+        updatedAt: String(row.updated_at || ''),
+        deletedAt: row.deleted_at ? String(row.deleted_at) : null,
+    };
+}
+export class TaskforceAgentConversationStore {
+    db;
+    tenantId;
+    constructor(db, tenantId) {
+        this.db = db;
+        this.tenantId = tenantId;
+    }
+    getLatest(workspaceIdRaw, agentIdRaw) {
+        const workspaceId = normalizeWorkspaceId(workspaceIdRaw);
+        const agentId = normalizeId(agentIdRaw);
+        if (!agentId)
+            return null;
+        const row = this.db.prepare(`
+            SELECT *
+            FROM agent_conversations
+            WHERE tenant_id = ? AND workspace_id = ? AND agent_id = ? AND deleted_at IS NULL
+            ORDER BY updated_at DESC, created_at DESC
+            LIMIT 1
+        `).get(this.tenantId, workspaceId, agentId);
+        return row ? mapRow(row) : null;
+    }
+    listForAgent(workspaceIdRaw, agentIdRaw) {
+        const workspaceId = normalizeWorkspaceId(workspaceIdRaw);
+        const agentId = normalizeId(agentIdRaw);
+        if (!agentId)
+            return [];
+        const rows = this.db.prepare(`
+            SELECT *
+            FROM agent_conversations
+            WHERE tenant_id = ? AND workspace_id = ? AND agent_id = ? AND deleted_at IS NULL
+            ORDER BY updated_at DESC, created_at DESC
+        `).all(this.tenantId, workspaceId, agentId);
+        return rows ? rows.map(mapRow) : [];
+    }
+    getById(workspaceIdRaw, agentIdRaw, idRaw) {
+        const workspaceId = normalizeWorkspaceId(workspaceIdRaw);
+        const agentId = normalizeId(agentIdRaw);
+        const id = normalizeId(idRaw);
+        if (!agentId || !id)
+            return null;
+        const row = this.db.prepare(`
+            SELECT *
+            FROM agent_conversations
+            WHERE tenant_id = ? AND workspace_id = ? AND agent_id = ? AND id = ? AND deleted_at IS NULL
+            LIMIT 1
+        `).get(this.tenantId, workspaceId, agentId, id);
+        return row ? mapRow(row) : null;
+    }
+    listForSync(workspaceIdRaw, options) {
+        const workspaceId = normalizeWorkspaceId(workspaceIdRaw);
+        const rows = this.db.prepare(`
+            SELECT *
+            FROM agent_conversations
+            WHERE tenant_id = ? AND workspace_id = ?
+                ${options?.includeDeleted ? '' : 'AND deleted_at IS NULL'}
+            ORDER BY updated_at ASC, id ASC
+        `).all(this.tenantId, workspaceId);
+        return rows ? rows.map(mapRow) : [];
+    }
+    createEmpty(workspaceIdRaw, agentIdRaw, actorUserId, taskId) {
+        return this.save({
+            workspaceId: workspaceIdRaw,
+            agentId: agentIdRaw,
+            taskId,
+            messages: [],
+            actorUserId,
+        });
+    }
+    save(input) {
+        const workspaceId = normalizeWorkspaceId(input.workspaceId);
+        const agentId = normalizeId(input.agentId);
+        if (!agentId)
+            throw new Error('Agent id is required.');
+        const messages = normalizeMessages(input.messages);
+        const existingId = normalizeId(input.conversationId || '');
+        const existing = existingId ? this.getById(workspaceId, agentId, existingId) : null;
+        const nowIso = new Date().toISOString();
+        const providedTitle = normalizeTitle(input.title);
+        const title = providedTitle || (existing && existing.title !== 'Agent chat'
+            ? existing.title
+            : titleFromMessages(messages));
+        if (!existing) {
+            const id = `tfc-${randomUUID()}`;
+            this.db.prepare(`
+                INSERT INTO agent_conversations (
+                    id,
+                    tenant_id,
+                    workspace_id,
+                    agent_id,
+                    task_id,
+                    title,
+                    messages_json,
+                    model_provider,
+                    model_id,
+                    usage_json,
+                    latency_ms,
+                    created_by_user_id,
+                    updated_by_user_id,
+                    created_at,
+                    updated_at
+                ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+            `).run(id, this.tenantId, workspaceId, agentId, normalizeId(input.taskId || '') || null, title, JSON.stringify(messages), input.modelProvider || null, input.modelId || null, input.usage ? JSON.stringify(input.usage) : null, input.latencyMs ?? null, input.actorUserId || null, input.actorUserId || null, nowIso, nowIso);
+            return this.getById(workspaceId, agentId, id);
+        }
+        const nextTaskId = Object.prototype.hasOwnProperty.call(input, 'taskId')
+            ? normalizeId(input.taskId || '') || null
+            : existing.taskId || null;
+        this.db.prepare(`
+            UPDATE agent_conversations
+            SET title = ?,
+                task_id = ?,
+                messages_json = ?,
+                model_provider = ?,
+                model_id = ?,
+                usage_json = ?,
+                latency_ms = ?,
+                updated_by_user_id = ?,
+                updated_at = ?
+            WHERE tenant_id = ? AND workspace_id = ? AND agent_id = ? AND id = ? AND deleted_at IS NULL
+        `).run(title, nextTaskId, JSON.stringify(messages), input.modelProvider || null, input.modelId || null, input.usage ? JSON.stringify(input.usage) : null, input.latencyMs ?? null, input.actorUserId || null, nowIso, this.tenantId, workspaceId, agentId, existing.id);
+        return this.getById(workspaceId, agentId, existing.id);
+    }
+    rename(workspaceIdRaw, agentIdRaw, idRaw, titleRaw, actorUserId) {
+        const workspaceId = normalizeWorkspaceId(workspaceIdRaw);
+        const agentId = normalizeId(agentIdRaw);
+        const id = normalizeId(idRaw);
+        const title = normalizeTitle(titleRaw);
+        if (!agentId || !id || !title)
+            return null;
+        const existing = this.getById(workspaceId, agentId, id);
+        if (!existing)
+            return null;
+        const nowIso = new Date().toISOString();
+        this.db.prepare(`
+            UPDATE agent_conversations
+            SET title = ?,
+                updated_by_user_id = ?,
+                updated_at = ?
+            WHERE tenant_id = ? AND workspace_id = ? AND agent_id = ? AND id = ? AND deleted_at IS NULL
+        `).run(title, actorUserId || null, nowIso, this.tenantId, workspaceId, agentId, id);
+        return this.getById(workspaceId, agentId, id);
+    }
+    getByIdForSync(workspaceIdRaw, idRaw) {
+        const workspaceId = normalizeWorkspaceId(workspaceIdRaw);
+        const id = normalizeId(idRaw);
+        if (!id)
+            return null;
+        const row = this.db.prepare(`
+            SELECT *
+            FROM agent_conversations
+            WHERE tenant_id = ? AND workspace_id = ? AND id = ?
+            LIMIT 1
+        `).get(this.tenantId, workspaceId, id);
+        return row ? mapRow(row) : null;
+    }
+    upsertForSync(input) {
+        const id = normalizeId(input.id);
+        const workspaceId = normalizeWorkspaceId(input.workspaceId);
+        const agentId = normalizeId(input.agentId);
+        if (!id || !agentId)
+            return null;
+        const messages = normalizeMessages(input.messages);
+        const nowIso = new Date().toISOString();
+        const createdAt = String(input.createdAt || input.updatedAt || nowIso).trim();
+        const updatedAt = String(input.updatedAt || input.createdAt || nowIso).trim();
+        const deletedAt = String(input.deletedAt || '').trim() || null;
+        const title = String(input.title || titleFromMessages(messages)).trim() || 'Agent chat';
+        const existing = this.getByIdForSync(workspaceId, id);
+        if (!existing) {
+            this.db.prepare(`
+                INSERT INTO agent_conversations (
+                    id,
+                    tenant_id,
+                    workspace_id,
+                    agent_id,
+                    task_id,
+                    title,
+                    messages_json,
+                    model_provider,
+                    model_id,
+                    usage_json,
+                    latency_ms,
+                    created_by_user_id,
+                    updated_by_user_id,
+                    created_at,
+                    updated_at,
+                    deleted_at
+                ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+            `).run(id, this.tenantId, workspaceId, agentId, normalizeId(input.taskId || '') || null, title, JSON.stringify(messages), input.modelProvider || null, input.modelId || null, input.usage ? JSON.stringify(input.usage) : null, input.latencyMs ?? null, input.createdByUserId || null, input.updatedByUserId || null, createdAt, updatedAt, deletedAt);
+            return this.getByIdForSync(workspaceId, id);
+        }
+        const nextTaskId = Object.prototype.hasOwnProperty.call(input, 'taskId')
+            ? normalizeId(input.taskId || '') || null
+            : existing.taskId || null;
+        this.db.prepare(`
+            UPDATE agent_conversations
+            SET agent_id = ?,
+                task_id = ?,
+                title = ?,
+                messages_json = ?,
+                model_provider = ?,
+                model_id = ?,
+                usage_json = ?,
+                latency_ms = ?,
+                created_by_user_id = ?,
+                updated_by_user_id = ?,
+                created_at = ?,
+                updated_at = ?,
+                deleted_at = ?
+            WHERE tenant_id = ? AND workspace_id = ? AND id = ?
+        `).run(agentId, nextTaskId, title, JSON.stringify(messages), input.modelProvider || null, input.modelId || null, input.usage ? JSON.stringify(input.usage) : null, input.latencyMs ?? null, input.createdByUserId ?? existing.createdByUserId ?? null, input.updatedByUserId ?? null, createdAt || existing.createdAt, updatedAt, deletedAt, this.tenantId, workspaceId, id);
+        return this.getByIdForSync(workspaceId, id);
+    }
+}

package/dist/storage/taskforceAgentSkillStore.d.ts

@@ -0,0 +1,93 @@
+import type { DatabaseAdapter } from './databaseAdapter.js';
+import { type TaskforceAgentSkillDefinitionV1 } from '../shared/taskforceAgentSkill.js';
+import type { BehaviorReference } from '../shared/taskforceAgentDefinition.js';
+import type { ResolvedAgentConfigurationReference } from '../shared/taskforceAgentRuntimeResolution.js';
+export type TaskforceAgentSkillLifecycleStatus = 'active' | 'retired';
+export interface TaskforceAgentSkillRevisionRecord {
+    revision: number;
+    contentHash: string;
+    definition: TaskforceAgentSkillDefinitionV1;
+    createdByActorId: string | null;
+    createdAt: string;
+}
+export interface TaskforceAgentSkillRecord {
+    id: string;
+    tenantId: string;
+    workspaceId: string;
+    lifecycleStatus: TaskforceAgentSkillLifecycleStatus;
+    definition: TaskforceAgentSkillDefinitionV1;
+    revision: number;
+    contentHash: string;
+    createdByActorId: string | null;
+    updatedByActorId: string | null;
+    createdAt: string;
+    updatedAt: string;
+    retiredAt: string | null;
+    retiredByActorId: string | null;
+    revisions?: TaskforceAgentSkillRevisionRecord[];
+}
+export interface UpsertTaskforceAgentSkillForSyncInput {
+    id: string;
+    workspaceId: string;
+    lifecycleStatus: TaskforceAgentSkillLifecycleStatus;
+    definition: unknown;
+    revision: number;
+    contentHash: string;
+    revisions: Array<{
+        revision: number;
+        contentHash: string;
+        definition: unknown;
+        createdByActorId?: string | null;
+        createdAt: string;
+    }>;
+    createdByActorId?: string | null;
+    updatedByActorId?: string | null;
+    createdAt: string;
+    updatedAt: string;
+    retiredAt?: string | null;
+    retiredByActorId?: string | null;
+}
+export declare class TaskforceAgentSkillConflictError extends Error {
+    readonly currentRevision: number;
+    readonly currentContentHash: string;
+    readonly statusCode = 409;
+    readonly code = "TASKFORCE_AGENT_SKILL_CONFLICT";
+    constructor(currentRevision: number, currentContentHash: string);
+}
+export declare class TaskforceAgentSkillSyncError extends Error {
+    readonly code: 'TASKFORCE_AGENT_SKILL_SYNC_INVALID_ENVELOPE' | 'TASKFORCE_AGENT_SKILL_SYNC_HASH_MISMATCH' | 'TASKFORCE_AGENT_SKILL_SYNC_MISSING_REVISION';
+    readonly statusCode = 409;
+    constructor(message: string, code: 'TASKFORCE_AGENT_SKILL_SYNC_INVALID_ENVELOPE' | 'TASKFORCE_AGENT_SKILL_SYNC_HASH_MISMATCH' | 'TASKFORCE_AGENT_SKILL_SYNC_MISSING_REVISION');
+}
+export declare class TaskforceAgentSkillStore {
+    private readonly db;
+    private readonly tenantId;
+    constructor(db: DatabaseAdapter, tenantId: string);
+    list(workspaceIdRaw: string, options?: {
+        includeRetired?: boolean;
+        includeRevisions?: boolean;
+    }): TaskforceAgentSkillRecord[];
+    listForSync(workspaceIdRaw: string): TaskforceAgentSkillRecord[];
+    get(workspaceIdRaw: string, idRaw: string, options?: {
+        includeRevisions?: boolean;
+    }): TaskforceAgentSkillRecord | null;
+    listRevisions(workspaceIdRaw: string, idRaw: string): TaskforceAgentSkillRevisionRecord[];
+    create(input: {
+        workspaceId: string;
+        definition: unknown;
+        actorId: string;
+    }): TaskforceAgentSkillRecord;
+    update(workspaceIdRaw: string, idRaw: string, input: {
+        definition: unknown;
+        expectedContentHash: string;
+        actorId: string;
+    }): TaskforceAgentSkillRecord | null;
+    retire(workspaceIdRaw: string, idRaw: string, actorId: string): TaskforceAgentSkillRecord | null;
+    restore(workspaceIdRaw: string, idRaw: string, actorId: string): TaskforceAgentSkillRecord | null;
+    resolve(workspaceIdRaw: string, reference: BehaviorReference): ResolvedAgentConfigurationReference | null;
+    upsertForSync(input: UpsertTaskforceAgentSkillForSyncInput): TaskforceAgentSkillRecord;
+    private setLifecycle;
+    private insertRevision;
+    private insertEvent;
+}
+export declare function isValidTaskforceAgentSkillSyncPayload(value: unknown): value is UpsertTaskforceAgentSkillForSyncInput;