weave-typescript 0.15.0 → 0.16.0

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"
+    });
+}

package/dist/weavesql/weavedb/provider_sql.d.ts

@@ -0,0 +1,162 @@
+import { QueryArrayConfig, QueryArrayResult } from "pg";
+interface Client {
+    query: (config: QueryArrayConfig) => Promise<QueryArrayResult>;
+}
+export declare const createProviderConfigurationQuery = "-- name: CreateProviderConfiguration :one\nINSERT INTO weave.llm_provider_configurations (\n    id,\n    organization_id,\n    provider_kind,\n    display_name,\n    base_url,\n    status,\n    created_by_user_id,\n    updated_by_user_id\n) VALUES (\n    $1,\n    $2,\n    $3,\n    $4,\n    $5,\n    $6,\n    $7,\n    $8\n)\nRETURNING\n    id,\n    organization_id,\n    provider_kind,\n    display_name,\n    base_url,\n    status,\n    created_by_user_id,\n    updated_by_user_id,\n    created_at,\n    updated_at";
+export interface CreateProviderConfigurationArgs {
+    id: string;
+    organizationId: string;
+    providerKind: string;
+    displayName: string;
+    baseUrl: string | null;
+    status: string;
+    createdByUserId: string;
+    updatedByUserId: string;
+}
+export interface CreateProviderConfigurationRow {
+    id: string;
+    organizationId: string;
+    providerKind: string;
+    displayName: string;
+    baseUrl: string | null;
+    status: string;
+    createdByUserId: string;
+    updatedByUserId: string;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function createProviderConfiguration(client: Client, args: CreateProviderConfigurationArgs): Promise<CreateProviderConfigurationRow | null>;
+export declare const getProviderConfigurationByIDQuery = "-- name: GetProviderConfigurationByID :one\nSELECT\n    id,\n    organization_id,\n    provider_kind,\n    display_name,\n    base_url,\n    status,\n    created_by_user_id,\n    updated_by_user_id,\n    created_at,\n    updated_at\nFROM weave.llm_provider_configurations\nWHERE organization_id = $1\n  AND id = $2";
+export interface GetProviderConfigurationByIDArgs {
+    organizationId: string;
+    id: string;
+}
+export interface GetProviderConfigurationByIDRow {
+    id: string;
+    organizationId: string;
+    providerKind: string;
+    displayName: string;
+    baseUrl: string | null;
+    status: string;
+    createdByUserId: string;
+    updatedByUserId: string;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function getProviderConfigurationByID(client: Client, args: GetProviderConfigurationByIDArgs): Promise<GetProviderConfigurationByIDRow | null>;
+export declare const getProviderConfigurationByKindQuery = "-- name: GetProviderConfigurationByKind :one\nSELECT\n    id,\n    organization_id,\n    provider_kind,\n    display_name,\n    base_url,\n    status,\n    created_by_user_id,\n    updated_by_user_id,\n    created_at,\n    updated_at\nFROM weave.llm_provider_configurations\nWHERE organization_id = $1\n  AND provider_kind = $2";
+export interface GetProviderConfigurationByKindArgs {
+    organizationId: string;
+    providerKind: string;
+}
+export interface GetProviderConfigurationByKindRow {
+    id: string;
+    organizationId: string;
+    providerKind: string;
+    displayName: string;
+    baseUrl: string | null;
+    status: string;
+    createdByUserId: string;
+    updatedByUserId: string;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function getProviderConfigurationByKind(client: Client, args: GetProviderConfigurationByKindArgs): Promise<GetProviderConfigurationByKindRow | null>;
+export declare const listProviderConfigurationsQuery = "-- name: ListProviderConfigurations :many\nSELECT\n    id,\n    organization_id,\n    provider_kind,\n    display_name,\n    base_url,\n    status,\n    created_by_user_id,\n    updated_by_user_id,\n    created_at,\n    updated_at\nFROM weave.llm_provider_configurations\nWHERE organization_id = $1\nORDER BY updated_at DESC, created_at DESC";
+export interface ListProviderConfigurationsArgs {
+    organizationId: string;
+}
+export interface ListProviderConfigurationsRow {
+    id: string;
+    organizationId: string;
+    providerKind: string;
+    displayName: string;
+    baseUrl: string | null;
+    status: string;
+    createdByUserId: string;
+    updatedByUserId: string;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function listProviderConfigurations(client: Client, args: ListProviderConfigurationsArgs): Promise<ListProviderConfigurationsRow[]>;
+export declare const updateProviderConfigurationByIDQuery = "-- name: UpdateProviderConfigurationByID :one\nUPDATE weave.llm_provider_configurations\nSET\n    display_name = $1,\n    base_url = $2,\n    status = $3,\n    updated_by_user_id = $4,\n    updated_at = now()\nWHERE organization_id = $5\n  AND id = $6\nRETURNING\n    id,\n    organization_id,\n    provider_kind,\n    display_name,\n    base_url,\n    status,\n    created_by_user_id,\n    updated_by_user_id,\n    created_at,\n    updated_at";
+export interface UpdateProviderConfigurationByIDArgs {
+    displayName: string;
+    baseUrl: string | null;
+    status: string;
+    updatedByUserId: string;
+    organizationId: string;
+    id: string;
+}
+export interface UpdateProviderConfigurationByIDRow {
+    id: string;
+    organizationId: string;
+    providerKind: string;
+    displayName: string;
+    baseUrl: string | null;
+    status: string;
+    createdByUserId: string;
+    updatedByUserId: string;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function updateProviderConfigurationByID(client: Client, args: UpdateProviderConfigurationByIDArgs): Promise<UpdateProviderConfigurationByIDRow | null>;
+export declare const disableProviderConfigurationByIDQuery = "-- name: DisableProviderConfigurationByID :one\nUPDATE weave.llm_provider_configurations\nSET\n    status = 'PROVIDER_CONFIGURATION_STATUS_DISABLED',\n    updated_by_user_id = $1,\n    updated_at = now()\nWHERE organization_id = $2\n  AND id = $3\nRETURNING\n    id,\n    organization_id,\n    provider_kind,\n    display_name,\n    base_url,\n    status,\n    created_by_user_id,\n    updated_by_user_id,\n    created_at,\n    updated_at";
+export interface DisableProviderConfigurationByIDArgs {
+    updatedByUserId: string;
+    organizationId: string;
+    id: string;
+}
+export interface DisableProviderConfigurationByIDRow {
+    id: string;
+    organizationId: string;
+    providerKind: string;
+    displayName: string;
+    baseUrl: string | null;
+    status: string;
+    createdByUserId: string;
+    updatedByUserId: string;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function disableProviderConfigurationByID(client: Client, args: DisableProviderConfigurationByIDArgs): Promise<DisableProviderConfigurationByIDRow | null>;
+export declare const upsertProviderCredentialQuery = "-- name: UpsertProviderCredential :one\nINSERT INTO weave.llm_provider_credentials (\n    provider_configuration_id,\n    organization_id,\n    encrypted_credentials,\n    credential_hint,\n    credential_status,\n    key_version,\n    last_rotated_at\n) VALUES (\n    $1,\n    $2,\n    $3,\n    $4,\n    $5,\n    $6,\n    $7\n)\nON CONFLICT (provider_configuration_id) DO UPDATE\nSET\n    organization_id = EXCLUDED.organization_id,\n    encrypted_credentials = EXCLUDED.encrypted_credentials,\n    credential_hint = EXCLUDED.credential_hint,\n    credential_status = EXCLUDED.credential_status,\n    key_version = EXCLUDED.key_version,\n    last_rotated_at = EXCLUDED.last_rotated_at,\n    updated_at = now()\nRETURNING\n    provider_configuration_id,\n    organization_id,\n    encrypted_credentials,\n    credential_hint,\n    credential_status,\n    key_version,\n    last_rotated_at,\n    created_at,\n    updated_at";
+export interface UpsertProviderCredentialArgs {
+    providerConfigurationId: string;
+    organizationId: string;
+    encryptedCredentials: Buffer;
+    credentialHint: string;
+    credentialStatus: string;
+    keyVersion: string | null;
+    lastRotatedAt: Date;
+}
+export interface UpsertProviderCredentialRow {
+    providerConfigurationId: string;
+    organizationId: string;
+    encryptedCredentials: Buffer;
+    credentialHint: string;
+    credentialStatus: string;
+    keyVersion: string | null;
+    lastRotatedAt: Date;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function upsertProviderCredential(client: Client, args: UpsertProviderCredentialArgs): Promise<UpsertProviderCredentialRow | null>;
+export declare const getProviderCredentialByProviderIDQuery = "-- name: GetProviderCredentialByProviderID :one\nSELECT\n    provider_configuration_id,\n    organization_id,\n    encrypted_credentials,\n    credential_hint,\n    credential_status,\n    key_version,\n    last_rotated_at,\n    created_at,\n    updated_at\nFROM weave.llm_provider_credentials\nWHERE organization_id = $1\n  AND provider_configuration_id = $2";
+export interface GetProviderCredentialByProviderIDArgs {
+    organizationId: string;
+    providerConfigurationId: string;
+}
+export interface GetProviderCredentialByProviderIDRow {
+    providerConfigurationId: string;
+    organizationId: string;
+    encryptedCredentials: Buffer;
+    credentialHint: string;
+    credentialStatus: string;
+    keyVersion: string | null;
+    lastRotatedAt: Date;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function getProviderCredentialByProviderID(client: Client, args: GetProviderCredentialByProviderIDArgs): Promise<GetProviderCredentialByProviderIDRow | null>;
+export {};