weave-typescript 0.15.0 → 0.17.0

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

@@ -0,0 +1,122 @@
+import { QueryArrayConfig, QueryArrayResult } from "pg";
+interface Client {
+    query: (config: QueryArrayConfig) => Promise<QueryArrayResult>;
+}
+export declare const insertRunTemplateQuery = "-- name: InsertRunTemplate :one\nINSERT INTO atc.run_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 InsertRunTemplateArgs {
+    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 InsertRunTemplateRow {
+    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 insertRunTemplate(client: Client, args: InsertRunTemplateArgs): Promise<InsertRunTemplateRow | null>;
+export declare const getRunTemplateQuery = "-- name: GetRunTemplate :one\nSELECT id, tenant_id, name, version, handler_type, config, security_context, budget_context, pinned_caller, created_by, created_at, updated_at\nFROM atc.run_templates\nWHERE tenant_id = $1\n  AND id = $2";
+export interface GetRunTemplateArgs {
+    tenantId: string;
+    id: string;
+}
+export interface GetRunTemplateRow {
+    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 getRunTemplate(client: Client, args: GetRunTemplateArgs): Promise<GetRunTemplateRow | null>;
+export declare const getRunTemplateByNameQuery = "-- name: GetRunTemplateByName :one\nSELECT id, tenant_id, name, version, handler_type, config, security_context, budget_context, pinned_caller, created_by, created_at, updated_at\nFROM atc.run_templates\nWHERE tenant_id = $1\n  AND name = $2\nORDER BY version DESC\nLIMIT 1";
+export interface GetRunTemplateByNameArgs {
+    tenantId: string;
+    name: string;
+}
+export interface GetRunTemplateByNameRow {
+    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 getRunTemplateByName(client: Client, args: GetRunTemplateByNameArgs): Promise<GetRunTemplateByNameRow | null>;
+export declare const listRunTemplatesQuery = "-- name: ListRunTemplates :many\nSELECT id, tenant_id, name, version, handler_type, config, security_context, budget_context, pinned_caller, created_by, created_at, updated_at\nFROM atc.run_templates\nWHERE tenant_id = $1\nORDER BY name ASC, version DESC";
+export interface ListRunTemplatesArgs {
+    tenantId: string;
+}
+export interface ListRunTemplatesRow {
+    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 listRunTemplates(client: Client, args: ListRunTemplatesArgs): Promise<ListRunTemplatesRow[]>;
+export declare const updateRunTemplateQuery = "-- name: UpdateRunTemplate :one\nUPDATE atc.run_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 UpdateRunTemplateArgs {
+    config: any;
+    securityContext: any | null;
+    budgetContext: any | null;
+    pinnedCaller: any | null;
+    tenantId: string;
+    id: string;
+}
+export interface UpdateRunTemplateRow {
+    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 updateRunTemplate(client: Client, args: UpdateRunTemplateArgs): Promise<UpdateRunTemplateRow | null>;
+export declare const deleteRunTemplateQuery = "-- name: DeleteRunTemplate :exec\nDELETE FROM atc.run_templates\nWHERE tenant_id = $1\n  AND id = $2";
+export interface DeleteRunTemplateArgs {
+    tenantId: string;
+    id: string;
+}
+export declare function deleteRunTemplate(client: Client, args: DeleteRunTemplateArgs): Promise<void>;
+export {};

package/dist/weavesql/atcdb/run_template_sql.js

@@ -0,0 +1,197 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.deleteRunTemplateQuery = exports.updateRunTemplateQuery = exports.listRunTemplatesQuery = exports.getRunTemplateByNameQuery = exports.getRunTemplateQuery = exports.insertRunTemplateQuery = void 0;
+exports.insertRunTemplate = insertRunTemplate;
+exports.getRunTemplate = getRunTemplate;
+exports.getRunTemplateByName = getRunTemplateByName;
+exports.listRunTemplates = listRunTemplates;
+exports.updateRunTemplate = updateRunTemplate;
+exports.deleteRunTemplate = deleteRunTemplate;
+exports.insertRunTemplateQuery = `-- name: InsertRunTemplate :one
+INSERT INTO atc.run_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 insertRunTemplate(client, args) {
+    const result = await client.query({
+        text: exports.insertRunTemplateQuery,
+        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.getRunTemplateQuery = `-- name: GetRunTemplate :one
+SELECT id, tenant_id, name, version, handler_type, config, security_context, budget_context, pinned_caller, created_by, created_at, updated_at
+FROM atc.run_templates
+WHERE tenant_id = $1
+  AND id = $2`;
+async function getRunTemplate(client, args) {
+    const result = await client.query({
+        text: exports.getRunTemplateQuery,
+        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.getRunTemplateByNameQuery = `-- name: GetRunTemplateByName :one
+SELECT id, tenant_id, name, version, handler_type, config, security_context, budget_context, pinned_caller, created_by, created_at, updated_at
+FROM atc.run_templates
+WHERE tenant_id = $1
+  AND name = $2
+ORDER BY version DESC
+LIMIT 1`;
+async function getRunTemplateByName(client, args) {
+    const result = await client.query({
+        text: exports.getRunTemplateByNameQuery,
+        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.listRunTemplatesQuery = `-- name: ListRunTemplates :many
+SELECT id, tenant_id, name, version, handler_type, config, security_context, budget_context, pinned_caller, created_by, created_at, updated_at
+FROM atc.run_templates
+WHERE tenant_id = $1
+ORDER BY name ASC, version DESC`;
+async function listRunTemplates(client, args) {
+    const result = await client.query({
+        text: exports.listRunTemplatesQuery,
+        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.updateRunTemplateQuery = `-- name: UpdateRunTemplate :one
+UPDATE atc.run_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 updateRunTemplate(client, args) {
+    const result = await client.query({
+        text: exports.updateRunTemplateQuery,
+        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.deleteRunTemplateQuery = `-- name: DeleteRunTemplate :exec
+DELETE FROM atc.run_templates
+WHERE tenant_id = $1
+  AND id = $2`;
+async function deleteRunTemplate(client, args) {
+    await client.query({
+        text: exports.deleteRunTemplateQuery,
+        values: [args.tenantId, args.id],
+        rowMode: "array"
+    });
+}

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

@@ -0,0 +1,58 @@
+import { QueryArrayConfig, QueryArrayResult } from "pg";
+interface Client {
+    query: (config: QueryArrayConfig) => Promise<QueryArrayResult>;
+}
+export declare const upsertSnapshotQuery = "-- name: UpsertSnapshot :one\nINSERT INTO atc.event_snapshots (\n    tenant_id,\n    run_id,\n    sequence,\n    snapshot_type,\n    data\n) VALUES (\n    $1,\n    $2,\n    $3,\n    $4,\n    $5\n)\nON CONFLICT (tenant_id, run_id, snapshot_type)\nDO UPDATE\nSET\n    sequence = CASE\n        WHEN EXCLUDED.sequence > atc.event_snapshots.sequence THEN EXCLUDED.sequence\n        ELSE atc.event_snapshots.sequence\n    END,\n    data = CASE\n        WHEN EXCLUDED.sequence > atc.event_snapshots.sequence THEN EXCLUDED.data\n        ELSE atc.event_snapshots.data\n    END\nRETURNING tenant_id, run_id, sequence, snapshot_type, data, created_at";
+export interface UpsertSnapshotArgs {
+    tenantId: string;
+    runId: string;
+    sequence: string;
+    snapshotType: string;
+    data: any;
+}
+export interface UpsertSnapshotRow {
+    tenantId: string;
+    runId: string;
+    sequence: string;
+    snapshotType: string;
+    data: any;
+    createdAt: Date;
+}
+export declare function upsertSnapshot(client: Client, args: UpsertSnapshotArgs): Promise<UpsertSnapshotRow | null>;
+export declare const loadSnapshotQuery = "-- name: LoadSnapshot :one\nSELECT\n    tenant_id,\n    run_id,\n    sequence,\n    snapshot_type,\n    data,\n    created_at\nFROM atc.event_snapshots\nWHERE tenant_id = $1\n  AND run_id = $2\n  AND snapshot_type = $3";
+export interface LoadSnapshotArgs {
+    tenantId: string;
+    runId: string;
+    snapshotType: string;
+}
+export interface LoadSnapshotRow {
+    tenantId: string;
+    runId: string;
+    sequence: string;
+    snapshotType: string;
+    data: any;
+    createdAt: Date;
+}
+export declare function loadSnapshot(client: Client, args: LoadSnapshotArgs): Promise<LoadSnapshotRow | null>;
+export declare const deleteSnapshotQuery = "-- name: DeleteSnapshot :exec\nDELETE FROM atc.event_snapshots\nWHERE tenant_id = $1\n  AND run_id = $2\n  AND snapshot_type = $3";
+export interface DeleteSnapshotArgs {
+    tenantId: string;
+    runId: string;
+    snapshotType: string;
+}
+export declare function deleteSnapshot(client: Client, args: DeleteSnapshotArgs): Promise<void>;
+export declare const listRunSnapshotsQuery = "-- name: ListRunSnapshots :many\nSELECT\n    tenant_id,\n    run_id,\n    sequence,\n    snapshot_type,\n    data,\n    created_at\nFROM atc.event_snapshots\nWHERE tenant_id = $1\n  AND run_id = $2\nORDER BY snapshot_type ASC";
+export interface ListRunSnapshotsArgs {
+    tenantId: string;
+    runId: string;
+}
+export interface ListRunSnapshotsRow {
+    tenantId: string;
+    runId: string;
+    sequence: string;
+    snapshotType: string;
+    data: any;
+    createdAt: Date;
+}
+export declare function listRunSnapshots(client: Client, args: ListRunSnapshotsArgs): Promise<ListRunSnapshotsRow[]>;
+export {};

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