weave-typescript 0.15.0 → 0.16.0

package/dist/weavesql/atcdb/snapshot_sql.js

@@ -0,0 +1,124 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.listRunSnapshotsQuery = exports.deleteSnapshotQuery = exports.loadSnapshotQuery = exports.upsertSnapshotQuery = void 0;
+exports.upsertSnapshot = upsertSnapshot;
+exports.loadSnapshot = loadSnapshot;
+exports.deleteSnapshot = deleteSnapshot;
+exports.listRunSnapshots = listRunSnapshots;
+exports.upsertSnapshotQuery = `-- name: UpsertSnapshot :one
+INSERT INTO atc.event_snapshots (
+    tenant_id,
+    run_id,
+    sequence,
+    snapshot_type,
+    data
+) VALUES (
+    $1,
+    $2,
+    $3,
+    $4,
+    $5
+)
+ON CONFLICT (tenant_id, run_id, snapshot_type)
+DO UPDATE
+SET
+    sequence = CASE
+        WHEN EXCLUDED.sequence > atc.event_snapshots.sequence THEN EXCLUDED.sequence
+        ELSE atc.event_snapshots.sequence
+    END,
+    data = CASE
+        WHEN EXCLUDED.sequence > atc.event_snapshots.sequence THEN EXCLUDED.data
+        ELSE atc.event_snapshots.data
+    END
+RETURNING tenant_id, run_id, sequence, snapshot_type, data, created_at`;
+async function upsertSnapshot(client, args) {
+    const result = await client.query({
+        text: exports.upsertSnapshotQuery,
+        values: [args.tenantId, args.runId, args.sequence, args.snapshotType, args.data],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        tenantId: row[0],
+        runId: row[1],
+        sequence: row[2],
+        snapshotType: row[3],
+        data: row[4],
+        createdAt: row[5]
+    };
+}
+exports.loadSnapshotQuery = `-- name: LoadSnapshot :one
+SELECT
+    tenant_id,
+    run_id,
+    sequence,
+    snapshot_type,
+    data,
+    created_at
+FROM atc.event_snapshots
+WHERE tenant_id = $1
+  AND run_id = $2
+  AND snapshot_type = $3`;
+async function loadSnapshot(client, args) {
+    const result = await client.query({
+        text: exports.loadSnapshotQuery,
+        values: [args.tenantId, args.runId, args.snapshotType],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        tenantId: row[0],
+        runId: row[1],
+        sequence: row[2],
+        snapshotType: row[3],
+        data: row[4],
+        createdAt: row[5]
+    };
+}
+exports.deleteSnapshotQuery = `-- name: DeleteSnapshot :exec
+DELETE FROM atc.event_snapshots
+WHERE tenant_id = $1
+  AND run_id = $2
+  AND snapshot_type = $3`;
+async function deleteSnapshot(client, args) {
+    await client.query({
+        text: exports.deleteSnapshotQuery,
+        values: [args.tenantId, args.runId, args.snapshotType],
+        rowMode: "array"
+    });
+}
+exports.listRunSnapshotsQuery = `-- name: ListRunSnapshots :many
+SELECT
+    tenant_id,
+    run_id,
+    sequence,
+    snapshot_type,
+    data,
+    created_at
+FROM atc.event_snapshots
+WHERE tenant_id = $1
+  AND run_id = $2
+ORDER BY snapshot_type ASC`;
+async function listRunSnapshots(client, args) {
+    const result = await client.query({
+        text: exports.listRunSnapshotsQuery,
+        values: [args.tenantId, args.runId],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            tenantId: row[0],
+            runId: row[1],
+            sequence: row[2],
+            snapshotType: row[3],
+            data: row[4],
+            createdAt: row[5]
+        };
+    });
+}

package/dist/weavesql/atcdb/team_sql.d.ts

@@ -0,0 +1,111 @@
+import { QueryArrayConfig, QueryArrayResult } from "pg";
+interface Client {
+    query: (config: QueryArrayConfig) => Promise<QueryArrayResult>;
+}
+export declare const insertTeamQuery = "-- name: InsertTeam :one\nINSERT INTO atc.teams (\n    id,\n    tenant_id,\n    parent_id,\n    name,\n    team_type,\n    metadata\n) VALUES (\n    $1,\n    $2,\n    $3,\n    $4,\n    $5,\n    $6\n)\nRETURNING id, tenant_id, parent_id, name, team_type, metadata, created_at, updated_at";
+export interface InsertTeamArgs {
+    id: string;
+    tenantId: string;
+    parentId: string | null;
+    name: string;
+    teamType: string;
+    metadata: any;
+}
+export interface InsertTeamRow {
+    id: string;
+    tenantId: string;
+    parentId: string | null;
+    name: string;
+    teamType: string;
+    metadata: any;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function insertTeam(client: Client, args: InsertTeamArgs): Promise<InsertTeamRow | null>;
+export declare const getTeamQuery = "-- name: GetTeam :one\nSELECT id, tenant_id, parent_id, name, team_type, metadata, created_at, updated_at\nFROM atc.teams\nWHERE tenant_id = $1\n  AND id = $2";
+export interface GetTeamArgs {
+    tenantId: string;
+    id: string;
+}
+export interface GetTeamRow {
+    id: string;
+    tenantId: string;
+    parentId: string | null;
+    name: string;
+    teamType: string;
+    metadata: any;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function getTeam(client: Client, args: GetTeamArgs): Promise<GetTeamRow | null>;
+export declare const listChildTeamsQuery = "-- name: ListChildTeams :many\nSELECT id, tenant_id, parent_id, name, team_type, metadata, created_at, updated_at\nFROM atc.teams\nWHERE tenant_id = $1\n  AND parent_id = $2\nORDER BY name ASC";
+export interface ListChildTeamsArgs {
+    tenantId: string;
+    parentId: string | null;
+}
+export interface ListChildTeamsRow {
+    id: string;
+    tenantId: string;
+    parentId: string | null;
+    name: string;
+    teamType: string;
+    metadata: any;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function listChildTeams(client: Client, args: ListChildTeamsArgs): Promise<ListChildTeamsRow[]>;
+export declare const listTeamAncestorsQuery = "-- name: ListTeamAncestors :many\nWITH RECURSIVE ancestors AS (\n    SELECT\n        id,\n        tenant_id,\n        parent_id,\n        name,\n        team_type,\n        metadata,\n        created_at,\n        updated_at,\n        0 AS depth\n    FROM atc.teams AS team\n    WHERE team.tenant_id = $1\n      AND team.id = $2\n\n    UNION ALL\n\n    SELECT\n        team.id,\n        team.tenant_id,\n        team.parent_id,\n        team.name,\n        team.team_type,\n        team.metadata,\n        team.created_at,\n        team.updated_at,\n        ancestors.depth + 1\n    FROM atc.teams AS team\n    JOIN ancestors\n      ON team.tenant_id = ancestors.tenant_id\n     AND team.id = ancestors.parent_id\n)\nSELECT\n    id,\n    tenant_id,\n    parent_id,\n    name,\n    team_type,\n    metadata,\n    created_at,\n    updated_at\nFROM ancestors\nORDER BY depth ASC";
+export interface ListTeamAncestorsArgs {
+    tenantId: string;
+    teamId: string;
+}
+export interface ListTeamAncestorsRow {
+    id: string;
+    tenantId: string;
+    parentId: string | null;
+    name: string;
+    teamType: string;
+    metadata: any;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function listTeamAncestors(client: Client, args: ListTeamAncestorsArgs): Promise<ListTeamAncestorsRow[]>;
+export declare const upsertTeamBudgetQuery = "-- name: UpsertTeamBudget :one\nINSERT INTO atc.team_budgets (\n    tenant_id,\n    team_id,\n    period_start,\n    period_end,\n    max_cost_cents,\n    max_compute,\n    max_tool_calls\n) VALUES (\n    $1,\n    $2,\n    $3,\n    $4,\n    $5,\n    $6,\n    $7\n)\nON CONFLICT (tenant_id, team_id, period_start)\nDO UPDATE SET\n    period_end = EXCLUDED.period_end,\n    max_cost_cents = EXCLUDED.max_cost_cents,\n    max_compute = EXCLUDED.max_compute,\n    max_tool_calls = EXCLUDED.max_tool_calls\nRETURNING tenant_id, team_id, period_start, period_end, max_cost_cents, max_compute, max_tool_calls, created_at";
+export interface UpsertTeamBudgetArgs {
+    tenantId: string;
+    teamId: string;
+    periodStart: Date;
+    periodEnd: Date;
+    maxCostCents: string;
+    maxCompute: string;
+    maxToolCalls: string;
+}
+export interface UpsertTeamBudgetRow {
+    tenantId: string;
+    teamId: string;
+    periodStart: Date;
+    periodEnd: Date;
+    maxCostCents: string;
+    maxCompute: string;
+    maxToolCalls: string;
+    createdAt: Date;
+}
+export declare function upsertTeamBudget(client: Client, args: UpsertTeamBudgetArgs): Promise<UpsertTeamBudgetRow | null>;
+export declare const getTeamBudgetQuery = "-- name: GetTeamBudget :one\nSELECT tenant_id, team_id, period_start, period_end, max_cost_cents, max_compute, max_tool_calls, created_at\nFROM atc.team_budgets\nWHERE tenant_id = $1\n  AND team_id = $2\n  AND period_start <= $3\n  AND period_end > $3";
+export interface GetTeamBudgetArgs {
+    tenantId: string;
+    teamId: string;
+    atTime: Date;
+}
+export interface GetTeamBudgetRow {
+    tenantId: string;
+    teamId: string;
+    periodStart: Date;
+    periodEnd: Date;
+    maxCostCents: string;
+    maxCompute: string;
+    maxToolCalls: string;
+    createdAt: Date;
+}
+export declare function getTeamBudget(client: Client, args: GetTeamBudgetArgs): Promise<GetTeamBudgetRow | null>;
+export {};

package/dist/weavesql/atcdb/team_sql.js

@@ -0,0 +1,235 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getTeamBudgetQuery = exports.upsertTeamBudgetQuery = exports.listTeamAncestorsQuery = exports.listChildTeamsQuery = exports.getTeamQuery = exports.insertTeamQuery = void 0;
+exports.insertTeam = insertTeam;
+exports.getTeam = getTeam;
+exports.listChildTeams = listChildTeams;
+exports.listTeamAncestors = listTeamAncestors;
+exports.upsertTeamBudget = upsertTeamBudget;
+exports.getTeamBudget = getTeamBudget;
+exports.insertTeamQuery = `-- name: InsertTeam :one
+INSERT INTO atc.teams (
+    id,
+    tenant_id,
+    parent_id,
+    name,
+    team_type,
+    metadata
+) VALUES (
+    $1,
+    $2,
+    $3,
+    $4,
+    $5,
+    $6
+)
+RETURNING id, tenant_id, parent_id, name, team_type, metadata, created_at, updated_at`;
+async function insertTeam(client, args) {
+    const result = await client.query({
+        text: exports.insertTeamQuery,
+        values: [args.id, args.tenantId, args.parentId, args.name, args.teamType, args.metadata],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        tenantId: row[1],
+        parentId: row[2],
+        name: row[3],
+        teamType: row[4],
+        metadata: row[5],
+        createdAt: row[6],
+        updatedAt: row[7]
+    };
+}
+exports.getTeamQuery = `-- name: GetTeam :one
+SELECT id, tenant_id, parent_id, name, team_type, metadata, created_at, updated_at
+FROM atc.teams
+WHERE tenant_id = $1
+  AND id = $2`;
+async function getTeam(client, args) {
+    const result = await client.query({
+        text: exports.getTeamQuery,
+        values: [args.tenantId, args.id],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        tenantId: row[1],
+        parentId: row[2],
+        name: row[3],
+        teamType: row[4],
+        metadata: row[5],
+        createdAt: row[6],
+        updatedAt: row[7]
+    };
+}
+exports.listChildTeamsQuery = `-- name: ListChildTeams :many
+SELECT id, tenant_id, parent_id, name, team_type, metadata, created_at, updated_at
+FROM atc.teams
+WHERE tenant_id = $1
+  AND parent_id = $2
+ORDER BY name ASC`;
+async function listChildTeams(client, args) {
+    const result = await client.query({
+        text: exports.listChildTeamsQuery,
+        values: [args.tenantId, args.parentId],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            id: row[0],
+            tenantId: row[1],
+            parentId: row[2],
+            name: row[3],
+            teamType: row[4],
+            metadata: row[5],
+            createdAt: row[6],
+            updatedAt: row[7]
+        };
+    });
+}
+exports.listTeamAncestorsQuery = `-- name: ListTeamAncestors :many
+WITH RECURSIVE ancestors AS (
+    SELECT
+        id,
+        tenant_id,
+        parent_id,
+        name,
+        team_type,
+        metadata,
+        created_at,
+        updated_at,
+        0 AS depth
+    FROM atc.teams AS team
+    WHERE team.tenant_id = $1
+      AND team.id = $2
+
+    UNION ALL
+
+    SELECT
+        team.id,
+        team.tenant_id,
+        team.parent_id,
+        team.name,
+        team.team_type,
+        team.metadata,
+        team.created_at,
+        team.updated_at,
+        ancestors.depth + 1
+    FROM atc.teams AS team
+    JOIN ancestors
+      ON team.tenant_id = ancestors.tenant_id
+     AND team.id = ancestors.parent_id
+)
+SELECT
+    id,
+    tenant_id,
+    parent_id,
+    name,
+    team_type,
+    metadata,
+    created_at,
+    updated_at
+FROM ancestors
+ORDER BY depth ASC`;
+async function listTeamAncestors(client, args) {
+    const result = await client.query({
+        text: exports.listTeamAncestorsQuery,
+        values: [args.tenantId, args.teamId],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            id: row[0],
+            tenantId: row[1],
+            parentId: row[2],
+            name: row[3],
+            teamType: row[4],
+            metadata: row[5],
+            createdAt: row[6],
+            updatedAt: row[7]
+        };
+    });
+}
+exports.upsertTeamBudgetQuery = `-- name: UpsertTeamBudget :one
+INSERT INTO atc.team_budgets (
+    tenant_id,
+    team_id,
+    period_start,
+    period_end,
+    max_cost_cents,
+    max_compute,
+    max_tool_calls
+) VALUES (
+    $1,
+    $2,
+    $3,
+    $4,
+    $5,
+    $6,
+    $7
+)
+ON CONFLICT (tenant_id, team_id, period_start)
+DO UPDATE SET
+    period_end = EXCLUDED.period_end,
+    max_cost_cents = EXCLUDED.max_cost_cents,
+    max_compute = EXCLUDED.max_compute,
+    max_tool_calls = EXCLUDED.max_tool_calls
+RETURNING tenant_id, team_id, period_start, period_end, max_cost_cents, max_compute, max_tool_calls, created_at`;
+async function upsertTeamBudget(client, args) {
+    const result = await client.query({
+        text: exports.upsertTeamBudgetQuery,
+        values: [args.tenantId, args.teamId, args.periodStart, args.periodEnd, args.maxCostCents, args.maxCompute, args.maxToolCalls],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        tenantId: row[0],
+        teamId: row[1],
+        periodStart: row[2],
+        periodEnd: row[3],
+        maxCostCents: row[4],
+        maxCompute: row[5],
+        maxToolCalls: row[6],
+        createdAt: row[7]
+    };
+}
+exports.getTeamBudgetQuery = `-- name: GetTeamBudget :one
+SELECT tenant_id, team_id, period_start, period_end, max_cost_cents, max_compute, max_tool_calls, created_at
+FROM atc.team_budgets
+WHERE tenant_id = $1
+  AND team_id = $2
+  AND period_start <= $3
+  AND period_end > $3`;
+async function getTeamBudget(client, args) {
+    const result = await client.query({
+        text: exports.getTeamBudgetQuery,
+        values: [args.tenantId, args.teamId, args.atTime],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        tenantId: row[0],
+        teamId: row[1],
+        periodStart: row[2],
+        periodEnd: row[3],
+        maxCostCents: row[4],
+        maxCompute: row[5],
+        maxToolCalls: row[6],
+        createdAt: row[7]
+    };
+}

package/dist/weavesql/atcdb/workflow_template_sql.d.ts

@@ -0,0 +1,122 @@
+import { QueryArrayConfig, QueryArrayResult } from "pg";
+interface Client {
+    query: (config: QueryArrayConfig) => Promise<QueryArrayResult>;
+}
+export declare const insertWorkflowTemplateQuery = "-- name: InsertWorkflowTemplate :one\nINSERT INTO atc.workflow_templates (\n    id,\n    tenant_id,\n    name,\n    version,\n    handler_type,\n    config,\n    security_context,\n    budget_context,\n    pinned_caller,\n    created_by\n) VALUES (\n    $1,\n    $2,\n    $3,\n    $4,\n    $5,\n    $6,\n    $7,\n    $8,\n    $9,\n    $10\n)\nRETURNING id, tenant_id, name, version, handler_type, config, security_context, budget_context, pinned_caller, created_by, created_at, updated_at";
+export interface InsertWorkflowTemplateArgs {
+    id: string;
+    tenantId: string;
+    name: string;
+    version: number;
+    handlerType: string;
+    config: any;
+    securityContext: any | null;
+    budgetContext: any | null;
+    pinnedCaller: any | null;
+    createdBy: string;
+}
+export interface InsertWorkflowTemplateRow {
+    id: string;
+    tenantId: string;
+    name: string;
+    version: number;
+    handlerType: string;
+    config: any;
+    securityContext: any | null;
+    budgetContext: any | null;
+    pinnedCaller: any | null;
+    createdBy: string;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function insertWorkflowTemplate(client: Client, args: InsertWorkflowTemplateArgs): Promise<InsertWorkflowTemplateRow | null>;
+export declare const getWorkflowTemplateQuery = "-- name: GetWorkflowTemplate :one\nSELECT id, tenant_id, name, version, handler_type, config, security_context, budget_context, pinned_caller, created_by, created_at, updated_at\nFROM atc.workflow_templates\nWHERE tenant_id = $1\n  AND id = $2";
+export interface GetWorkflowTemplateArgs {
+    tenantId: string;
+    id: string;
+}
+export interface GetWorkflowTemplateRow {
+    id: string;
+    tenantId: string;
+    name: string;
+    version: number;
+    handlerType: string;
+    config: any;
+    securityContext: any | null;
+    budgetContext: any | null;
+    pinnedCaller: any | null;
+    createdBy: string;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function getWorkflowTemplate(client: Client, args: GetWorkflowTemplateArgs): Promise<GetWorkflowTemplateRow | null>;
+export declare const getWorkflowTemplateByNameQuery = "-- name: GetWorkflowTemplateByName :one\nSELECT id, tenant_id, name, version, handler_type, config, security_context, budget_context, pinned_caller, created_by, created_at, updated_at\nFROM atc.workflow_templates\nWHERE tenant_id = $1\n  AND name = $2\nORDER BY version DESC\nLIMIT 1";
+export interface GetWorkflowTemplateByNameArgs {
+    tenantId: string;
+    name: string;
+}
+export interface GetWorkflowTemplateByNameRow {
+    id: string;
+    tenantId: string;
+    name: string;
+    version: number;
+    handlerType: string;
+    config: any;
+    securityContext: any | null;
+    budgetContext: any | null;
+    pinnedCaller: any | null;
+    createdBy: string;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function getWorkflowTemplateByName(client: Client, args: GetWorkflowTemplateByNameArgs): Promise<GetWorkflowTemplateByNameRow | null>;
+export declare const listWorkflowTemplatesQuery = "-- name: ListWorkflowTemplates :many\nSELECT id, tenant_id, name, version, handler_type, config, security_context, budget_context, pinned_caller, created_by, created_at, updated_at\nFROM atc.workflow_templates\nWHERE tenant_id = $1\nORDER BY name ASC, version DESC";
+export interface ListWorkflowTemplatesArgs {
+    tenantId: string;
+}
+export interface ListWorkflowTemplatesRow {
+    id: string;
+    tenantId: string;
+    name: string;
+    version: number;
+    handlerType: string;
+    config: any;
+    securityContext: any | null;
+    budgetContext: any | null;
+    pinnedCaller: any | null;
+    createdBy: string;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function listWorkflowTemplates(client: Client, args: ListWorkflowTemplatesArgs): Promise<ListWorkflowTemplatesRow[]>;
+export declare const updateWorkflowTemplateQuery = "-- name: UpdateWorkflowTemplate :one\nUPDATE atc.workflow_templates\nSET\n    config = $1,\n    security_context = $2,\n    budget_context = $3,\n    pinned_caller = $4,\n    version = version + 1,\n    updated_at = now()\nWHERE tenant_id = $5\n  AND id = $6\nRETURNING id, tenant_id, name, version, handler_type, config, security_context, budget_context, pinned_caller, created_by, created_at, updated_at";
+export interface UpdateWorkflowTemplateArgs {
+    config: any;
+    securityContext: any | null;
+    budgetContext: any | null;
+    pinnedCaller: any | null;
+    tenantId: string;
+    id: string;
+}
+export interface UpdateWorkflowTemplateRow {
+    id: string;
+    tenantId: string;
+    name: string;
+    version: number;
+    handlerType: string;
+    config: any;
+    securityContext: any | null;
+    budgetContext: any | null;
+    pinnedCaller: any | null;
+    createdBy: string;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function updateWorkflowTemplate(client: Client, args: UpdateWorkflowTemplateArgs): Promise<UpdateWorkflowTemplateRow | null>;
+export declare const deleteWorkflowTemplateQuery = "-- name: DeleteWorkflowTemplate :exec\nDELETE FROM atc.workflow_templates\nWHERE tenant_id = $1\n  AND id = $2";
+export interface DeleteWorkflowTemplateArgs {
+    tenantId: string;
+    id: string;
+}
+export declare function deleteWorkflowTemplate(client: Client, args: DeleteWorkflowTemplateArgs): Promise<void>;
+export {};