weave-typescript 0.43.0 → 0.45.0

package/dist/weavesql/weavedb/agent_sql.js

@@ -0,0 +1,844 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.listAgentRunEventsQuery = exports.insertAgentRunEventQuery = exports.updateAgentRunStatusQuery = exports.listAgentRunsQuery = exports.getAgentRunQuery = exports.createAgentRunQuery = exports.listAgentTemplateVersionsQuery = exports.getAgentTemplateVersionByIDQuery = exports.getAgentTemplateVersionByNumberQuery = exports.setAgentTemplateLatestVersionQuery = exports.insertAgentTemplateVersionQuery = exports.insertAgentDraftPatchQuery = exports.updateAgentDraftQuery = exports.updateAgentTemplateStatusQuery = exports.updateAgentTemplateDraftMetadataQuery = exports.listAgentTemplatesQuery = exports.getAgentTemplateDraftQuery = exports.getAgentTemplateBySlugQuery = exports.getAgentTemplateByIDQuery = exports.createAgentTemplateDraftQuery = exports.createAgentTemplateQuery = void 0;
+exports.createAgentTemplate = createAgentTemplate;
+exports.createAgentTemplateDraft = createAgentTemplateDraft;
+exports.getAgentTemplateByID = getAgentTemplateByID;
+exports.getAgentTemplateBySlug = getAgentTemplateBySlug;
+exports.getAgentTemplateDraft = getAgentTemplateDraft;
+exports.listAgentTemplates = listAgentTemplates;
+exports.updateAgentTemplateDraftMetadata = updateAgentTemplateDraftMetadata;
+exports.updateAgentTemplateStatus = updateAgentTemplateStatus;
+exports.updateAgentDraft = updateAgentDraft;
+exports.insertAgentDraftPatch = insertAgentDraftPatch;
+exports.insertAgentTemplateVersion = insertAgentTemplateVersion;
+exports.setAgentTemplateLatestVersion = setAgentTemplateLatestVersion;
+exports.getAgentTemplateVersionByNumber = getAgentTemplateVersionByNumber;
+exports.getAgentTemplateVersionByID = getAgentTemplateVersionByID;
+exports.listAgentTemplateVersions = listAgentTemplateVersions;
+exports.createAgentRun = createAgentRun;
+exports.getAgentRun = getAgentRun;
+exports.listAgentRuns = listAgentRuns;
+exports.updateAgentRunStatus = updateAgentRunStatus;
+exports.insertAgentRunEvent = insertAgentRunEvent;
+exports.listAgentRunEvents = listAgentRunEvents;
+exports.createAgentTemplateQuery = `-- name: CreateAgentTemplate :one
+INSERT INTO weave.agent_templates (
+    id,
+    organization_id,
+    slug,
+    name,
+    description,
+    status,
+    created_by_user_id
+) VALUES (
+    $1,
+    $2,
+    $3,
+    $4,
+    $5,
+    $6,
+    $7
+)
+RETURNING id, organization_id, slug, name, description, status, created_by_user_id, draft_revision, latest_version_id, latest_version, created_at, updated_at`;
+async function createAgentTemplate(client, args) {
+    const result = await client.query({
+        text: exports.createAgentTemplateQuery,
+        values: [args.id, args.organizationId, args.slug, args.name, args.description, args.status, args.createdByUserId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        slug: row[2],
+        name: row[3],
+        description: row[4],
+        status: row[5],
+        createdByUserId: row[6],
+        draftRevision: row[7],
+        latestVersionId: row[8],
+        latestVersion: row[9],
+        createdAt: row[10],
+        updatedAt: row[11]
+    };
+}
+exports.createAgentTemplateDraftQuery = `-- name: CreateAgentTemplateDraft :one
+INSERT INTO weave.agent_template_drafts (
+    id,
+    template_id,
+    organization_id,
+    base_version_id,
+    revision,
+    name,
+    description,
+    instructions,
+    model_id,
+    provider_configuration_id,
+    tool_descriptors,
+    input_schema,
+    output_schema,
+    metadata,
+    updated_by_user_id
+) VALUES (
+    $1,
+    $2,
+    $3,
+    $4::uuid,
+    $5,
+    $6,
+    $7,
+    $8,
+    $9,
+    $10::uuid,
+    $11,
+    $12,
+    $13,
+    $14,
+    $15
+)
+RETURNING id, template_id, organization_id, base_version_id, revision, name, description, instructions, model_id, provider_configuration_id, tool_descriptors, input_schema, output_schema, metadata, updated_by_user_id, created_at, updated_at`;
+async function createAgentTemplateDraft(client, args) {
+    const result = await client.query({
+        text: exports.createAgentTemplateDraftQuery,
+        values: [args.id, args.templateId, args.organizationId, args.baseVersionId, args.revision, args.name, args.description, args.instructions, args.modelId, args.providerConfigurationId, args.toolDescriptors, args.inputSchema, args.outputSchema, args.metadata, args.updatedByUserId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        templateId: row[1],
+        organizationId: row[2],
+        baseVersionId: row[3],
+        revision: row[4],
+        name: row[5],
+        description: row[6],
+        instructions: row[7],
+        modelId: row[8],
+        providerConfigurationId: row[9],
+        toolDescriptors: row[10],
+        inputSchema: row[11],
+        outputSchema: row[12],
+        metadata: row[13],
+        updatedByUserId: row[14],
+        createdAt: row[15],
+        updatedAt: row[16]
+    };
+}
+exports.getAgentTemplateByIDQuery = `-- name: GetAgentTemplateByID :one
+SELECT id, organization_id, slug, name, description, status, created_by_user_id, draft_revision, latest_version_id, latest_version, created_at, updated_at FROM weave.agent_templates
+WHERE organization_id = $1
+  AND id = $2`;
+async function getAgentTemplateByID(client, args) {
+    const result = await client.query({
+        text: exports.getAgentTemplateByIDQuery,
+        values: [args.organizationId, args.templateId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        slug: row[2],
+        name: row[3],
+        description: row[4],
+        status: row[5],
+        createdByUserId: row[6],
+        draftRevision: row[7],
+        latestVersionId: row[8],
+        latestVersion: row[9],
+        createdAt: row[10],
+        updatedAt: row[11]
+    };
+}
+exports.getAgentTemplateBySlugQuery = `-- name: GetAgentTemplateBySlug :one
+SELECT id, organization_id, slug, name, description, status, created_by_user_id, draft_revision, latest_version_id, latest_version, created_at, updated_at FROM weave.agent_templates
+WHERE organization_id = $1
+  AND slug = $2`;
+async function getAgentTemplateBySlug(client, args) {
+    const result = await client.query({
+        text: exports.getAgentTemplateBySlugQuery,
+        values: [args.organizationId, args.slug],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        slug: row[2],
+        name: row[3],
+        description: row[4],
+        status: row[5],
+        createdByUserId: row[6],
+        draftRevision: row[7],
+        latestVersionId: row[8],
+        latestVersion: row[9],
+        createdAt: row[10],
+        updatedAt: row[11]
+    };
+}
+exports.getAgentTemplateDraftQuery = `-- name: GetAgentTemplateDraft :one
+SELECT id, template_id, organization_id, base_version_id, revision, name, description, instructions, model_id, provider_configuration_id, tool_descriptors, input_schema, output_schema, metadata, updated_by_user_id, created_at, updated_at FROM weave.agent_template_drafts
+WHERE organization_id = $1
+  AND template_id = $2`;
+async function getAgentTemplateDraft(client, args) {
+    const result = await client.query({
+        text: exports.getAgentTemplateDraftQuery,
+        values: [args.organizationId, args.templateId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        templateId: row[1],
+        organizationId: row[2],
+        baseVersionId: row[3],
+        revision: row[4],
+        name: row[5],
+        description: row[6],
+        instructions: row[7],
+        modelId: row[8],
+        providerConfigurationId: row[9],
+        toolDescriptors: row[10],
+        inputSchema: row[11],
+        outputSchema: row[12],
+        metadata: row[13],
+        updatedByUserId: row[14],
+        createdAt: row[15],
+        updatedAt: row[16]
+    };
+}
+exports.listAgentTemplatesQuery = `-- name: ListAgentTemplates :many
+SELECT id, organization_id, slug, name, description, status, created_by_user_id, draft_revision, latest_version_id, latest_version, created_at, updated_at FROM weave.agent_templates
+WHERE organization_id = $1
+  AND (
+      $2::text IS NULL
+      OR status = $2::text
+  )
+ORDER BY updated_at DESC, created_at DESC
+LIMIT $4 OFFSET $3`;
+async function listAgentTemplates(client, args) {
+    const result = await client.query({
+        text: exports.listAgentTemplatesQuery,
+        values: [args.organizationId, args.status, args.pageOffset, args.pageSize],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            id: row[0],
+            organizationId: row[1],
+            slug: row[2],
+            name: row[3],
+            description: row[4],
+            status: row[5],
+            createdByUserId: row[6],
+            draftRevision: row[7],
+            latestVersionId: row[8],
+            latestVersion: row[9],
+            createdAt: row[10],
+            updatedAt: row[11]
+        };
+    });
+}
+exports.updateAgentTemplateDraftMetadataQuery = `-- name: UpdateAgentTemplateDraftMetadata :one
+UPDATE weave.agent_templates
+SET
+    name = $1,
+    description = $2,
+    draft_revision = $3,
+    updated_at = now()
+WHERE organization_id = $4
+  AND id = $5
+RETURNING id, organization_id, slug, name, description, status, created_by_user_id, draft_revision, latest_version_id, latest_version, created_at, updated_at`;
+async function updateAgentTemplateDraftMetadata(client, args) {
+    const result = await client.query({
+        text: exports.updateAgentTemplateDraftMetadataQuery,
+        values: [args.name, args.description, args.draftRevision, args.organizationId, args.templateId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        slug: row[2],
+        name: row[3],
+        description: row[4],
+        status: row[5],
+        createdByUserId: row[6],
+        draftRevision: row[7],
+        latestVersionId: row[8],
+        latestVersion: row[9],
+        createdAt: row[10],
+        updatedAt: row[11]
+    };
+}
+exports.updateAgentTemplateStatusQuery = `-- name: UpdateAgentTemplateStatus :one
+UPDATE weave.agent_templates
+SET
+    status = $1,
+    updated_at = now()
+WHERE organization_id = $2
+  AND id = $3
+RETURNING id, organization_id, slug, name, description, status, created_by_user_id, draft_revision, latest_version_id, latest_version, created_at, updated_at`;
+async function updateAgentTemplateStatus(client, args) {
+    const result = await client.query({
+        text: exports.updateAgentTemplateStatusQuery,
+        values: [args.status, args.organizationId, args.templateId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        slug: row[2],
+        name: row[3],
+        description: row[4],
+        status: row[5],
+        createdByUserId: row[6],
+        draftRevision: row[7],
+        latestVersionId: row[8],
+        latestVersion: row[9],
+        createdAt: row[10],
+        updatedAt: row[11]
+    };
+}
+exports.updateAgentDraftQuery = `-- name: UpdateAgentDraft :one
+UPDATE weave.agent_template_drafts
+SET
+    base_version_id = $1::uuid,
+    revision = revision + 1,
+    name = $2,
+    description = $3,
+    instructions = $4,
+    model_id = $5,
+    provider_configuration_id = $6::uuid,
+    tool_descriptors = $7,
+    input_schema = $8,
+    output_schema = $9,
+    metadata = $10,
+    updated_by_user_id = $11,
+    updated_at = now()
+WHERE organization_id = $12
+  AND template_id = $13
+  AND revision = $14
+RETURNING id, template_id, organization_id, base_version_id, revision, name, description, instructions, model_id, provider_configuration_id, tool_descriptors, input_schema, output_schema, metadata, updated_by_user_id, created_at, updated_at`;
+async function updateAgentDraft(client, args) {
+    const result = await client.query({
+        text: exports.updateAgentDraftQuery,
+        values: [args.baseVersionId, args.name, args.description, args.instructions, args.modelId, args.providerConfigurationId, args.toolDescriptors, args.inputSchema, args.outputSchema, args.metadata, args.updatedByUserId, args.organizationId, args.templateId, args.expectedRevision],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        templateId: row[1],
+        organizationId: row[2],
+        baseVersionId: row[3],
+        revision: row[4],
+        name: row[5],
+        description: row[6],
+        instructions: row[7],
+        modelId: row[8],
+        providerConfigurationId: row[9],
+        toolDescriptors: row[10],
+        inputSchema: row[11],
+        outputSchema: row[12],
+        metadata: row[13],
+        updatedByUserId: row[14],
+        createdAt: row[15],
+        updatedAt: row[16]
+    };
+}
+exports.insertAgentDraftPatchQuery = `-- name: InsertAgentDraftPatch :one
+INSERT INTO weave.agent_template_draft_patches (
+    id,
+    template_id,
+    organization_id,
+    revision,
+    patch,
+    created_by_user_id
+) VALUES (
+    $1,
+    $2,
+    $3,
+    $4,
+    $5,
+    $6
+)
+RETURNING id, template_id, organization_id, revision, patch, created_by_user_id, created_at`;
+async function insertAgentDraftPatch(client, args) {
+    const result = await client.query({
+        text: exports.insertAgentDraftPatchQuery,
+        values: [args.id, args.templateId, args.organizationId, args.revision, args.patch, args.createdByUserId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        templateId: row[1],
+        organizationId: row[2],
+        revision: row[3],
+        patch: row[4],
+        createdByUserId: row[5],
+        createdAt: row[6]
+    };
+}
+exports.insertAgentTemplateVersionQuery = `-- name: InsertAgentTemplateVersion :one
+INSERT INTO weave.agent_template_versions (
+    id,
+    template_id,
+    organization_id,
+    version,
+    source_draft_revision,
+    name,
+    description,
+    instructions,
+    model_id,
+    provider_configuration_id,
+    tool_descriptors,
+    input_schema,
+    output_schema,
+    metadata,
+    published_by_user_id
+) VALUES (
+    $1,
+    $2,
+    $3,
+    $4,
+    $5,
+    $6,
+    $7,
+    $8,
+    $9,
+    $10::uuid,
+    $11,
+    $12,
+    $13,
+    $14,
+    $15
+)
+RETURNING id, template_id, organization_id, version, source_draft_revision, name, description, instructions, model_id, provider_configuration_id, tool_descriptors, input_schema, output_schema, metadata, published_by_user_id, published_at`;
+async function insertAgentTemplateVersion(client, args) {
+    const result = await client.query({
+        text: exports.insertAgentTemplateVersionQuery,
+        values: [args.id, args.templateId, args.organizationId, args.version, args.sourceDraftRevision, args.name, args.description, args.instructions, args.modelId, args.providerConfigurationId, args.toolDescriptors, args.inputSchema, args.outputSchema, args.metadata, args.publishedByUserId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        templateId: row[1],
+        organizationId: row[2],
+        version: row[3],
+        sourceDraftRevision: row[4],
+        name: row[5],
+        description: row[6],
+        instructions: row[7],
+        modelId: row[8],
+        providerConfigurationId: row[9],
+        toolDescriptors: row[10],
+        inputSchema: row[11],
+        outputSchema: row[12],
+        metadata: row[13],
+        publishedByUserId: row[14],
+        publishedAt: row[15]
+    };
+}
+exports.setAgentTemplateLatestVersionQuery = `-- name: SetAgentTemplateLatestVersion :one
+UPDATE weave.agent_templates
+SET
+    latest_version_id = $1,
+    latest_version = $2,
+    updated_at = now()
+WHERE organization_id = $3
+  AND id = $4
+RETURNING id, organization_id, slug, name, description, status, created_by_user_id, draft_revision, latest_version_id, latest_version, created_at, updated_at`;
+async function setAgentTemplateLatestVersion(client, args) {
+    const result = await client.query({
+        text: exports.setAgentTemplateLatestVersionQuery,
+        values: [args.versionId, args.version, args.organizationId, args.templateId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        slug: row[2],
+        name: row[3],
+        description: row[4],
+        status: row[5],
+        createdByUserId: row[6],
+        draftRevision: row[7],
+        latestVersionId: row[8],
+        latestVersion: row[9],
+        createdAt: row[10],
+        updatedAt: row[11]
+    };
+}
+exports.getAgentTemplateVersionByNumberQuery = `-- name: GetAgentTemplateVersionByNumber :one
+SELECT id, template_id, organization_id, version, source_draft_revision, name, description, instructions, model_id, provider_configuration_id, tool_descriptors, input_schema, output_schema, metadata, published_by_user_id, published_at FROM weave.agent_template_versions
+WHERE organization_id = $1
+  AND template_id = $2
+  AND version = $3`;
+async function getAgentTemplateVersionByNumber(client, args) {
+    const result = await client.query({
+        text: exports.getAgentTemplateVersionByNumberQuery,
+        values: [args.organizationId, args.templateId, args.version],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        templateId: row[1],
+        organizationId: row[2],
+        version: row[3],
+        sourceDraftRevision: row[4],
+        name: row[5],
+        description: row[6],
+        instructions: row[7],
+        modelId: row[8],
+        providerConfigurationId: row[9],
+        toolDescriptors: row[10],
+        inputSchema: row[11],
+        outputSchema: row[12],
+        metadata: row[13],
+        publishedByUserId: row[14],
+        publishedAt: row[15]
+    };
+}
+exports.getAgentTemplateVersionByIDQuery = `-- name: GetAgentTemplateVersionByID :one
+SELECT id, template_id, organization_id, version, source_draft_revision, name, description, instructions, model_id, provider_configuration_id, tool_descriptors, input_schema, output_schema, metadata, published_by_user_id, published_at FROM weave.agent_template_versions
+WHERE organization_id = $1
+  AND id = $2`;
+async function getAgentTemplateVersionByID(client, args) {
+    const result = await client.query({
+        text: exports.getAgentTemplateVersionByIDQuery,
+        values: [args.organizationId, args.versionId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        templateId: row[1],
+        organizationId: row[2],
+        version: row[3],
+        sourceDraftRevision: row[4],
+        name: row[5],
+        description: row[6],
+        instructions: row[7],
+        modelId: row[8],
+        providerConfigurationId: row[9],
+        toolDescriptors: row[10],
+        inputSchema: row[11],
+        outputSchema: row[12],
+        metadata: row[13],
+        publishedByUserId: row[14],
+        publishedAt: row[15]
+    };
+}
+exports.listAgentTemplateVersionsQuery = `-- name: ListAgentTemplateVersions :many
+SELECT id, template_id, organization_id, version, source_draft_revision, name, description, instructions, model_id, provider_configuration_id, tool_descriptors, input_schema, output_schema, metadata, published_by_user_id, published_at FROM weave.agent_template_versions
+WHERE organization_id = $1
+  AND template_id = $2
+ORDER BY version DESC
+LIMIT $4 OFFSET $3`;
+async function listAgentTemplateVersions(client, args) {
+    const result = await client.query({
+        text: exports.listAgentTemplateVersionsQuery,
+        values: [args.organizationId, args.templateId, args.pageOffset, args.pageSize],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            id: row[0],
+            templateId: row[1],
+            organizationId: row[2],
+            version: row[3],
+            sourceDraftRevision: row[4],
+            name: row[5],
+            description: row[6],
+            instructions: row[7],
+            modelId: row[8],
+            providerConfigurationId: row[9],
+            toolDescriptors: row[10],
+            inputSchema: row[11],
+            outputSchema: row[12],
+            metadata: row[13],
+            publishedByUserId: row[14],
+            publishedAt: row[15]
+        };
+    });
+}
+exports.createAgentRunQuery = `-- name: CreateAgentRun :one
+INSERT INTO weave.agent_runs (
+    id,
+    organization_id,
+    agent_template_id,
+    agent_version_id,
+    status,
+    created_by_user_id,
+    chat_session_id,
+    workflow_run_id,
+    workflow_step_run_id,
+    input
+) VALUES (
+    $1,
+    $2,
+    $3,
+    $4,
+    $5,
+    $6,
+    $7::uuid,
+    $8::uuid,
+    $9::uuid,
+    $10
+)
+RETURNING id, organization_id, agent_template_id, agent_version_id, status, created_by_user_id, chat_session_id, workflow_run_id, workflow_step_run_id, input, output, error_code, safe_error_message, created_at, started_at, finished_at`;
+async function createAgentRun(client, args) {
+    const result = await client.query({
+        text: exports.createAgentRunQuery,
+        values: [args.id, args.organizationId, args.agentTemplateId, args.agentVersionId, args.status, args.createdByUserId, args.chatSessionId, args.workflowRunId, args.workflowStepRunId, args.input],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        agentTemplateId: row[2],
+        agentVersionId: row[3],
+        status: row[4],
+        createdByUserId: row[5],
+        chatSessionId: row[6],
+        workflowRunId: row[7],
+        workflowStepRunId: row[8],
+        input: row[9],
+        output: row[10],
+        errorCode: row[11],
+        safeErrorMessage: row[12],
+        createdAt: row[13],
+        startedAt: row[14],
+        finishedAt: row[15]
+    };
+}
+exports.getAgentRunQuery = `-- name: GetAgentRun :one
+SELECT id, organization_id, agent_template_id, agent_version_id, status, created_by_user_id, chat_session_id, workflow_run_id, workflow_step_run_id, input, output, error_code, safe_error_message, created_at, started_at, finished_at FROM weave.agent_runs
+WHERE organization_id = $1
+  AND id = $2`;
+async function getAgentRun(client, args) {
+    const result = await client.query({
+        text: exports.getAgentRunQuery,
+        values: [args.organizationId, args.agentRunId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        agentTemplateId: row[2],
+        agentVersionId: row[3],
+        status: row[4],
+        createdByUserId: row[5],
+        chatSessionId: row[6],
+        workflowRunId: row[7],
+        workflowStepRunId: row[8],
+        input: row[9],
+        output: row[10],
+        errorCode: row[11],
+        safeErrorMessage: row[12],
+        createdAt: row[13],
+        startedAt: row[14],
+        finishedAt: row[15]
+    };
+}
+exports.listAgentRunsQuery = `-- name: ListAgentRuns :many
+SELECT id, organization_id, agent_template_id, agent_version_id, status, created_by_user_id, chat_session_id, workflow_run_id, workflow_step_run_id, input, output, error_code, safe_error_message, created_at, started_at, finished_at FROM weave.agent_runs
+WHERE organization_id = $1
+  AND (
+      $2::uuid IS NULL
+      OR agent_template_id = $2::uuid
+  )
+  AND (
+      $3::text IS NULL
+      OR status = $3::text
+  )
+ORDER BY created_at DESC
+LIMIT $5 OFFSET $4`;
+async function listAgentRuns(client, args) {
+    const result = await client.query({
+        text: exports.listAgentRunsQuery,
+        values: [args.organizationId, args.agentTemplateId, args.status, args.pageOffset, args.pageSize],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            id: row[0],
+            organizationId: row[1],
+            agentTemplateId: row[2],
+            agentVersionId: row[3],
+            status: row[4],
+            createdByUserId: row[5],
+            chatSessionId: row[6],
+            workflowRunId: row[7],
+            workflowStepRunId: row[8],
+            input: row[9],
+            output: row[10],
+            errorCode: row[11],
+            safeErrorMessage: row[12],
+            createdAt: row[13],
+            startedAt: row[14],
+            finishedAt: row[15]
+        };
+    });
+}
+exports.updateAgentRunStatusQuery = `-- name: UpdateAgentRunStatus :one
+UPDATE weave.agent_runs
+SET
+    status = $1,
+    output = $2,
+    error_code = $3,
+    safe_error_message = $4,
+    started_at = COALESCE(started_at, $5::timestamptz),
+    finished_at = $6::timestamptz
+WHERE organization_id = $7
+  AND id = $8
+RETURNING id, organization_id, agent_template_id, agent_version_id, status, created_by_user_id, chat_session_id, workflow_run_id, workflow_step_run_id, input, output, error_code, safe_error_message, created_at, started_at, finished_at`;
+async function updateAgentRunStatus(client, args) {
+    const result = await client.query({
+        text: exports.updateAgentRunStatusQuery,
+        values: [args.status, args.output, args.errorCode, args.safeErrorMessage, args.startedAt, args.finishedAt, args.organizationId, args.agentRunId],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        agentTemplateId: row[2],
+        agentVersionId: row[3],
+        status: row[4],
+        createdByUserId: row[5],
+        chatSessionId: row[6],
+        workflowRunId: row[7],
+        workflowStepRunId: row[8],
+        input: row[9],
+        output: row[10],
+        errorCode: row[11],
+        safeErrorMessage: row[12],
+        createdAt: row[13],
+        startedAt: row[14],
+        finishedAt: row[15]
+    };
+}
+exports.insertAgentRunEventQuery = `-- name: InsertAgentRunEvent :one
+INSERT INTO weave.agent_run_events (
+    id,
+    organization_id,
+    agent_run_id,
+    sequence,
+    kind,
+    message,
+    payload
+) VALUES (
+    $1,
+    $2,
+    $3,
+    $4,
+    $5,
+    $6,
+    $7
+)
+RETURNING id, organization_id, agent_run_id, sequence, kind, message, payload, created_at`;
+async function insertAgentRunEvent(client, args) {
+    const result = await client.query({
+        text: exports.insertAgentRunEventQuery,
+        values: [args.id, args.organizationId, args.agentRunId, args.sequence, args.kind, args.message, args.payload],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        agentRunId: row[2],
+        sequence: row[3],
+        kind: row[4],
+        message: row[5],
+        payload: row[6],
+        createdAt: row[7]
+    };
+}
+exports.listAgentRunEventsQuery = `-- name: ListAgentRunEvents :many
+SELECT id, organization_id, agent_run_id, sequence, kind, message, payload, created_at FROM weave.agent_run_events
+WHERE organization_id = $1
+  AND agent_run_id = $2
+  AND sequence > $3
+ORDER BY sequence
+LIMIT $4`;
+async function listAgentRunEvents(client, args) {
+    const result = await client.query({
+        text: exports.listAgentRunEventsQuery,
+        values: [args.organizationId, args.agentRunId, args.afterSequence, args.pageSize],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            id: row[0],
+            organizationId: row[1],
+            agentRunId: row[2],
+            sequence: row[3],
+            kind: row[4],
+            message: row[5],
+            payload: row[6],
+            createdAt: row[7]
+        };
+    });
+}