weave-typescript 0.15.0 → 0.17.0

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 {};

package/dist/weavesql/atcdb/workflow_template_sql.js

@@ -0,0 +1,197 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.deleteWorkflowTemplateQuery = exports.updateWorkflowTemplateQuery = exports.listWorkflowTemplatesQuery = exports.getWorkflowTemplateByNameQuery = exports.getWorkflowTemplateQuery = exports.insertWorkflowTemplateQuery = void 0;
+exports.insertWorkflowTemplate = insertWorkflowTemplate;
+exports.getWorkflowTemplate = getWorkflowTemplate;
+exports.getWorkflowTemplateByName = getWorkflowTemplateByName;
+exports.listWorkflowTemplates = listWorkflowTemplates;
+exports.updateWorkflowTemplate = updateWorkflowTemplate;
+exports.deleteWorkflowTemplate = deleteWorkflowTemplate;
+exports.insertWorkflowTemplateQuery = `-- name: InsertWorkflowTemplate :one
+INSERT INTO atc.workflow_templates (
+    id,
+    tenant_id,
+    name,
+    version,
+    handler_type,
+    config,
+    security_context,
+    budget_context,
+    pinned_caller,
+    created_by
+) VALUES (
+    $1,
+    $2,
+    $3,
+    $4,
+    $5,
+    $6,
+    $7,
+    $8,
+    $9,
+    $10
+)
+RETURNING id, tenant_id, name, version, handler_type, config, security_context, budget_context, pinned_caller, created_by, created_at, updated_at`;
+async function insertWorkflowTemplate(client, args) {
+    const result = await client.query({
+        text: exports.insertWorkflowTemplateQuery,
+        values: [args.id, args.tenantId, args.name, args.version, args.handlerType, args.config, args.securityContext, args.budgetContext, args.pinnedCaller, args.createdBy],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        tenantId: row[1],
+        name: row[2],
+        version: row[3],
+        handlerType: row[4],
+        config: row[5],
+        securityContext: row[6],
+        budgetContext: row[7],
+        pinnedCaller: row[8],
+        createdBy: row[9],
+        createdAt: row[10],
+        updatedAt: row[11]
+    };
+}
+exports.getWorkflowTemplateQuery = `-- name: GetWorkflowTemplate :one
+SELECT id, tenant_id, name, version, handler_type, config, security_context, budget_context, pinned_caller, created_by, created_at, updated_at
+FROM atc.workflow_templates
+WHERE tenant_id = $1
+  AND id = $2`;
+async function getWorkflowTemplate(client, args) {
+    const result = await client.query({
+        text: exports.getWorkflowTemplateQuery,
+        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],
+        name: row[2],
+        version: row[3],
+        handlerType: row[4],
+        config: row[5],
+        securityContext: row[6],
+        budgetContext: row[7],
+        pinnedCaller: row[8],
+        createdBy: row[9],
+        createdAt: row[10],
+        updatedAt: row[11]
+    };
+}
+exports.getWorkflowTemplateByNameQuery = `-- name: GetWorkflowTemplateByName :one
+SELECT id, tenant_id, name, version, handler_type, config, security_context, budget_context, pinned_caller, created_by, created_at, updated_at
+FROM atc.workflow_templates
+WHERE tenant_id = $1
+  AND name = $2
+ORDER BY version DESC
+LIMIT 1`;
+async function getWorkflowTemplateByName(client, args) {
+    const result = await client.query({
+        text: exports.getWorkflowTemplateByNameQuery,
+        values: [args.tenantId, args.name],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        tenantId: row[1],
+        name: row[2],
+        version: row[3],
+        handlerType: row[4],
+        config: row[5],
+        securityContext: row[6],
+        budgetContext: row[7],
+        pinnedCaller: row[8],
+        createdBy: row[9],
+        createdAt: row[10],
+        updatedAt: row[11]
+    };
+}
+exports.listWorkflowTemplatesQuery = `-- name: ListWorkflowTemplates :many
+SELECT id, tenant_id, name, version, handler_type, config, security_context, budget_context, pinned_caller, created_by, created_at, updated_at
+FROM atc.workflow_templates
+WHERE tenant_id = $1
+ORDER BY name ASC, version DESC`;
+async function listWorkflowTemplates(client, args) {
+    const result = await client.query({
+        text: exports.listWorkflowTemplatesQuery,
+        values: [args.tenantId],
+        rowMode: "array"
+    });
+    return result.rows.map(row => {
+        return {
+            id: row[0],
+            tenantId: row[1],
+            name: row[2],
+            version: row[3],
+            handlerType: row[4],
+            config: row[5],
+            securityContext: row[6],
+            budgetContext: row[7],
+            pinnedCaller: row[8],
+            createdBy: row[9],
+            createdAt: row[10],
+            updatedAt: row[11]
+        };
+    });
+}
+exports.updateWorkflowTemplateQuery = `-- name: UpdateWorkflowTemplate :one
+UPDATE atc.workflow_templates
+SET
+    config = $1,
+    security_context = $2,
+    budget_context = $3,
+    pinned_caller = $4,
+    version = version + 1,
+    updated_at = now()
+WHERE tenant_id = $5
+  AND id = $6
+RETURNING id, tenant_id, name, version, handler_type, config, security_context, budget_context, pinned_caller, created_by, created_at, updated_at`;
+async function updateWorkflowTemplate(client, args) {
+    const result = await client.query({
+        text: exports.updateWorkflowTemplateQuery,
+        values: [args.config, args.securityContext, args.budgetContext, args.pinnedCaller, 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],
+        name: row[2],
+        version: row[3],
+        handlerType: row[4],
+        config: row[5],
+        securityContext: row[6],
+        budgetContext: row[7],
+        pinnedCaller: row[8],
+        createdBy: row[9],
+        createdAt: row[10],
+        updatedAt: row[11]
+    };
+}
+exports.deleteWorkflowTemplateQuery = `-- name: DeleteWorkflowTemplate :exec
+DELETE FROM atc.workflow_templates
+WHERE tenant_id = $1
+  AND id = $2`;
+async function deleteWorkflowTemplate(client, args) {
+    await client.query({
+        text: exports.deleteWorkflowTemplateQuery,
+        values: [args.tenantId, args.id],
+        rowMode: "array"
+    });
+}