weave-typescript 0.48.0 → 0.50.0

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

@@ -498,6 +498,33 @@ export interface GetAgentRunRow {
     finishedAt: Date | null;
 }
 export declare function getAgentRun(client: Client, args: GetAgentRunArgs): Promise<GetAgentRunRow | null>;
+export declare const claimAgentRunQuery = "-- name: ClaimAgentRun :one\nUPDATE weave.agent_runs\nSET\n    status = $1,\n    error_code = '',\n    safe_error_message = '',\n    started_at = COALESCE(started_at, $2::timestamptz),\n    finished_at = NULL\nWHERE organization_id = $3\n  AND id = $4\n  AND status = $5\nRETURNING id, organization_id, agent_definition_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";
+export interface ClaimAgentRunArgs {
+    status: string;
+    startedAt: Date;
+    organizationId: string;
+    agentRunId: string;
+    expectedStatus: string;
+}
+export interface ClaimAgentRunRow {
+    id: string;
+    organizationId: string;
+    agentDefinitionId: string;
+    agentVersionId: string;
+    status: string;
+    createdByUserId: string;
+    chatSessionId: string | null;
+    workflowRunId: string | null;
+    workflowStepRunId: string | null;
+    input: any;
+    output: any;
+    errorCode: string;
+    safeErrorMessage: string;
+    createdAt: Date;
+    startedAt: Date | null;
+    finishedAt: Date | null;
+}
+export declare function claimAgentRun(client: Client, args: ClaimAgentRunArgs): Promise<ClaimAgentRunRow | null>;
 export declare const listAgentRunsQuery = "-- name: ListAgentRuns :many\nSELECT id, organization_id, agent_definition_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\nWHERE organization_id = $1\n  AND (\n      $2::uuid IS NULL\n      OR agent_definition_id = $2::uuid\n  )\n  AND (\n      $3::text IS NULL\n      OR status = $3::text\n  )\nORDER BY created_at DESC\nLIMIT $5 OFFSET $4";
 export interface ListAgentRunsArgs {
     organizationId: string;

package/dist/weavesql/weavedb/agent_sql.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.listAgentRunEventsQuery = exports.insertAgentRunEventQuery = exports.updateAgentRunStatusQuery = exports.listAgentRunsQuery = exports.getAgentRunQuery = exports.createAgentRunQuery = exports.listAgentDefinitionVersionsQuery = exports.getAgentDefinitionVersionByIDQuery = exports.getAgentDefinitionVersionByNumberQuery = exports.setAgentDefinitionLatestVersionQuery = exports.insertAgentDefinitionVersionQuery = exports.listAgentDraftEventsQuery = exports.insertAgentDraftEventQuery = exports.insertAgentDraftPatchQuery = exports.updateAgentDraftQuery = exports.updateAgentDefinitionStatusQuery = exports.updateAgentDefinitionDraftMetadataQuery = exports.listAgentDefinitionsQuery = exports.getAgentDefinitionDraftQuery = exports.getAgentDefinitionBySlugQuery = exports.getAgentDefinitionByIDQuery = exports.createAgentDefinitionDraftQuery = exports.createAgentDefinitionQuery = void 0;
+exports.listAgentRunEventsQuery = exports.insertAgentRunEventQuery = exports.updateAgentRunStatusQuery = exports.listAgentRunsQuery = exports.claimAgentRunQuery = exports.getAgentRunQuery = exports.createAgentRunQuery = exports.listAgentDefinitionVersionsQuery = exports.getAgentDefinitionVersionByIDQuery = exports.getAgentDefinitionVersionByNumberQuery = exports.setAgentDefinitionLatestVersionQuery = exports.insertAgentDefinitionVersionQuery = exports.listAgentDraftEventsQuery = exports.insertAgentDraftEventQuery = exports.insertAgentDraftPatchQuery = exports.updateAgentDraftQuery = exports.updateAgentDefinitionStatusQuery = exports.updateAgentDefinitionDraftMetadataQuery = exports.listAgentDefinitionsQuery = exports.getAgentDefinitionDraftQuery = exports.getAgentDefinitionBySlugQuery = exports.getAgentDefinitionByIDQuery = exports.createAgentDefinitionDraftQuery = exports.createAgentDefinitionQuery = void 0;
 exports.createAgentDefinition = createAgentDefinition;
 exports.createAgentDefinitionDraft = createAgentDefinitionDraft;
 exports.getAgentDefinitionByID = getAgentDefinitionByID;
@@ -20,6 +20,7 @@ exports.getAgentDefinitionVersionByID = getAgentDefinitionVersionByID;
 exports.listAgentDefinitionVersions = listAgentDefinitionVersions;
 exports.createAgentRun = createAgentRun;
 exports.getAgentRun = getAgentRun;
+exports.claimAgentRun = claimAgentRun;
 exports.listAgentRuns = listAgentRuns;
 exports.updateAgentRunStatus = updateAgentRunStatus;
 exports.insertAgentRunEvent = insertAgentRunEvent;
@@ -794,6 +795,47 @@ async function getAgentRun(client, args) {
         finishedAt: row[15]
     };
 }
+exports.claimAgentRunQuery = `-- name: ClaimAgentRun :one
+UPDATE weave.agent_runs
+SET
+    status = $1,
+    error_code = '',
+    safe_error_message = '',
+    started_at = COALESCE(started_at, $2::timestamptz),
+    finished_at = NULL
+WHERE organization_id = $3
+  AND id = $4
+  AND status = $5
+RETURNING id, organization_id, agent_definition_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 claimAgentRun(client, args) {
+    const result = await client.query({
+        text: exports.claimAgentRunQuery,
+        values: [args.status, args.startedAt, args.organizationId, args.agentRunId, args.expectedStatus],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        organizationId: row[1],
+        agentDefinitionId: 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_definition_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

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

@@ -13,6 +13,18 @@ export interface CreateUserRow {
     updatedAt: Date;
 }
 export declare function createUser(client: Client, args: CreateUserArgs): Promise<CreateUserRow | null>;
+export declare const createUserWithIDQuery = "-- name: CreateUserWithID :one\nINSERT INTO weave.users (\n    id,\n    status\n) VALUES (\n    $1,\n    $2\n)\nON CONFLICT (id) DO NOTHING\nRETURNING id, status, created_at, updated_at";
+export interface CreateUserWithIDArgs {
+    id: string;
+    status: string;
+}
+export interface CreateUserWithIDRow {
+    id: string;
+    status: string;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function createUserWithID(client: Client, args: CreateUserWithIDArgs): Promise<CreateUserWithIDRow | null>;
 export declare const getUserByIDQuery = "-- name: GetUserByID :one\nSELECT id, status, created_at, updated_at\nFROM weave.users\nWHERE id = $1";
 export interface GetUserByIDArgs {
     id: string;

package/dist/weavesql/weavedb/auth_identity_sql.js

@@ -1,8 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.listOrgTeamsQuery = exports.getOrgTeamQuery = exports.createOrgTeamQuery = exports.deleteExpiredAuthInviteSessionsQuery = exports.revokeAuthInviteSessionsByMemberQuery = exports.markAuthInviteSessionRecoveryRequiredQuery = exports.markAuthInviteSessionCompletedQuery = exports.setAuthInviteSessionOAuthStateQuery = exports.getAuthInviteSessionQuery = exports.createAuthInviteSessionQuery = exports.deleteExpiredApplicationSessionsQuery = exports.markApplicationSessionOrganizationSessionStateQuery = exports.listApplicationSessionOrganizationSessionStatesQuery = exports.getApplicationSessionOrganizationSessionQuery = exports.upsertApplicationSessionOrganizationSessionQuery = exports.revokeApplicationSessionQuery = exports.setApplicationSessionActiveOrganizationQuery = exports.getApplicationSessionQuery = exports.createApplicationSessionQuery = exports.listAuthorizationDecisionEventsByOrganizationQuery = exports.insertAuthorizationDecisionEventQuery = exports.insertAuthAuditEventQuery = exports.endSupportAccessSessionQuery = exports.createSupportAccessSessionQuery = exports.createPlatformAdminGrantQuery = exports.listActivePlatformAdminGrantsForUserQuery = exports.getOrganizationMemberAdminRecordQuery = exports.listOrganizationMembersQuery = exports.listOrganizationMembershipsForUserQuery = exports.getOrganizationMembershipQuery = exports.upsertOrganizationMembershipQuery = exports.upsertOrganizationAuthLinkQuery = exports.getOrganizationAuthLinkByOrganizationIDQuery = exports.getOrganizationAuthLinkByStytchIDQuery = exports.upsertIdentityLinkQuery = exports.getIdentityLinkByStytchMemberQuery = exports.getUserProfileQuery = exports.upsertUserProfileQuery = exports.deleteOrganizationAuthDomainQuery = exports.listOrganizationAuthDomainsQuery = exports.upsertOrganizationAuthDomainQuery = exports.resolvePrimaryOrganizationAuthLinkByDomainQuery = exports.updateUserProfileThemeQuery = exports.setUserOrganizationPagePositionQuery = exports.setUserLastOrganizationQuery = exports.getUserAuthPreferencesQuery = exports.getUserPrimaryOrganizationAuthLinkQuery = exports.setUserPrimaryOrganizationQuery = exports.getUserByIDQuery = exports.createUserQuery = void 0;
-exports.deleteOrgTeamMembershipsForOrganizationMemberQuery = exports.listEffectiveOrganizationRoleIDsQuery = exports.listOrgTeamRoleGrantsQuery = exports.listOrgTeamMembersQuery = exports.addOrgTeamRoleGrantQuery = exports.deleteOrgTeamRoleGrantsQuery = exports.addOrgTeamMembershipQuery = exports.deleteOrgTeamMembershipsQuery = exports.deleteOrgTeamQuery = exports.updateOrgTeamQuery = void 0;
+exports.getOrgTeamQuery = exports.createOrgTeamQuery = exports.deleteExpiredAuthInviteSessionsQuery = exports.revokeAuthInviteSessionsByMemberQuery = exports.markAuthInviteSessionRecoveryRequiredQuery = exports.markAuthInviteSessionCompletedQuery = exports.setAuthInviteSessionOAuthStateQuery = exports.getAuthInviteSessionQuery = exports.createAuthInviteSessionQuery = exports.deleteExpiredApplicationSessionsQuery = exports.markApplicationSessionOrganizationSessionStateQuery = exports.listApplicationSessionOrganizationSessionStatesQuery = exports.getApplicationSessionOrganizationSessionQuery = exports.upsertApplicationSessionOrganizationSessionQuery = exports.revokeApplicationSessionQuery = exports.setApplicationSessionActiveOrganizationQuery = exports.getApplicationSessionQuery = exports.createApplicationSessionQuery = exports.listAuthorizationDecisionEventsByOrganizationQuery = exports.insertAuthorizationDecisionEventQuery = exports.insertAuthAuditEventQuery = exports.endSupportAccessSessionQuery = exports.createSupportAccessSessionQuery = exports.createPlatformAdminGrantQuery = exports.listActivePlatformAdminGrantsForUserQuery = exports.getOrganizationMemberAdminRecordQuery = exports.listOrganizationMembersQuery = exports.listOrganizationMembershipsForUserQuery = exports.getOrganizationMembershipQuery = exports.upsertOrganizationMembershipQuery = exports.upsertOrganizationAuthLinkQuery = exports.getOrganizationAuthLinkByOrganizationIDQuery = exports.getOrganizationAuthLinkByStytchIDQuery = exports.upsertIdentityLinkQuery = exports.getIdentityLinkByStytchMemberQuery = exports.getUserProfileQuery = exports.upsertUserProfileQuery = exports.deleteOrganizationAuthDomainQuery = exports.listOrganizationAuthDomainsQuery = exports.upsertOrganizationAuthDomainQuery = exports.resolvePrimaryOrganizationAuthLinkByDomainQuery = exports.updateUserProfileThemeQuery = exports.setUserOrganizationPagePositionQuery = exports.setUserLastOrganizationQuery = exports.getUserAuthPreferencesQuery = exports.getUserPrimaryOrganizationAuthLinkQuery = exports.setUserPrimaryOrganizationQuery = exports.getUserByIDQuery = exports.createUserWithIDQuery = exports.createUserQuery = void 0;
+exports.deleteOrgTeamMembershipsForOrganizationMemberQuery = exports.listEffectiveOrganizationRoleIDsQuery = exports.listOrgTeamRoleGrantsQuery = exports.listOrgTeamMembersQuery = exports.addOrgTeamRoleGrantQuery = exports.deleteOrgTeamRoleGrantsQuery = exports.addOrgTeamMembershipQuery = exports.deleteOrgTeamMembershipsQuery = exports.deleteOrgTeamQuery = exports.updateOrgTeamQuery = exports.listOrgTeamsQuery = void 0;
 exports.createUser = createUser;
+exports.createUserWithID = createUserWithID;
 exports.getUserByID = getUserByID;
 exports.setUserPrimaryOrganization = setUserPrimaryOrganization;
 exports.getUserPrimaryOrganizationAuthLink = getUserPrimaryOrganizationAuthLink;
@@ -73,6 +74,33 @@ async function createUser(client, args) {
         updatedAt: row[3]
     };
 }
+exports.createUserWithIDQuery = `-- name: CreateUserWithID :one
+INSERT INTO weave.users (
+    id,
+    status
+) VALUES (
+    $1,
+    $2
+)
+ON CONFLICT (id) DO NOTHING
+RETURNING id, status, created_at, updated_at`;
+async function createUserWithID(client, args) {
+    const result = await client.query({
+        text: exports.createUserWithIDQuery,
+        values: [args.id, args.status],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        status: row[1],
+        createdAt: row[2],
+        updatedAt: row[3]
+    };
+}
 exports.getUserByIDQuery = `-- name: GetUserByID :one
 SELECT id, status, created_at, updated_at
 FROM weave.users

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

@@ -26,6 +26,31 @@ export interface CreateOrganizationRow {
     updatedAt: Date;
 }
 export declare function createOrganization(client: Client, args: CreateOrganizationArgs): Promise<CreateOrganizationRow | null>;
+export declare const createOrganizationWithIDQuery = "-- name: CreateOrganizationWithID :one\nINSERT INTO weave.organizations (\n    id,\n    slug,\n    name,\n    description,\n    primary_color,\n    logo,\n    logo_content_type,\n    timezone,\n    location\n) VALUES (\n    $1,\n    $2,\n    $3,\n    $4,\n    $5,\n    $6,\n    $7,\n    $8,\n    $9\n)\nON CONFLICT (slug) DO NOTHING\nRETURNING\n    id,\n    slug,\n    name,\n    description,\n    primary_color,\n    (logo IS NOT NULL) AS has_logo,\n    timezone,\n    location,\n    created_at,\n    updated_at";
+export interface CreateOrganizationWithIDArgs {
+    id: string;
+    slug: string;
+    name: string;
+    description: string;
+    primaryColor: string;
+    logo: Buffer | null;
+    logoContentType: string | null;
+    timezone: string;
+    location: string;
+}
+export interface CreateOrganizationWithIDRow {
+    id: string;
+    slug: string;
+    name: string;
+    description: string;
+    primaryColor: string;
+    hasLogo: string | null;
+    timezone: string;
+    location: string;
+    createdAt: Date;
+    updatedAt: Date;
+}
+export declare function createOrganizationWithID(client: Client, args: CreateOrganizationWithIDArgs): Promise<CreateOrganizationWithIDRow | null>;
 export declare const getOrganizationBySlugQuery = "-- name: GetOrganizationBySlug :one\nSELECT\n    id,\n    slug,\n    name,\n    description,\n    primary_color,\n    (logo IS NOT NULL) AS has_logo,\n    timezone,\n    location,\n    created_at,\n    updated_at\nFROM weave.organizations\nWHERE slug = $1";
 export interface GetOrganizationBySlugArgs {
     slug: string;

package/dist/weavesql/weavedb/organization_sql.js

@@ -1,7 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.clearOrganizationLogoQuery = exports.setOrganizationLogoQuery = exports.getOrganizationLogoQuery = exports.deleteOrganizationQuery = exports.updateOrganizationQuery = exports.listOrganizationsQuery = exports.getOrganizationByIDQuery = exports.getOrganizationBySlugQuery = exports.createOrganizationQuery = void 0;
+exports.clearOrganizationLogoQuery = exports.setOrganizationLogoQuery = exports.getOrganizationLogoQuery = exports.deleteOrganizationQuery = exports.updateOrganizationQuery = exports.listOrganizationsQuery = exports.getOrganizationByIDQuery = exports.getOrganizationBySlugQuery = exports.createOrganizationWithIDQuery = exports.createOrganizationQuery = void 0;
 exports.createOrganization = createOrganization;
+exports.createOrganizationWithID = createOrganizationWithID;
 exports.getOrganizationBySlug = getOrganizationBySlug;
 exports.getOrganizationByID = getOrganizationByID;
 exports.listOrganizations = listOrganizations;
@@ -62,6 +63,63 @@ async function createOrganization(client, args) {
         updatedAt: row[9]
     };
 }
+exports.createOrganizationWithIDQuery = `-- name: CreateOrganizationWithID :one
+INSERT INTO weave.organizations (
+    id,
+    slug,
+    name,
+    description,
+    primary_color,
+    logo,
+    logo_content_type,
+    timezone,
+    location
+) VALUES (
+    $1,
+    $2,
+    $3,
+    $4,
+    $5,
+    $6,
+    $7,
+    $8,
+    $9
+)
+ON CONFLICT (slug) DO NOTHING
+RETURNING
+    id,
+    slug,
+    name,
+    description,
+    primary_color,
+    (logo IS NOT NULL) AS has_logo,
+    timezone,
+    location,
+    created_at,
+    updated_at`;
+async function createOrganizationWithID(client, args) {
+    const result = await client.query({
+        text: exports.createOrganizationWithIDQuery,
+        values: [args.id, args.slug, args.name, args.description, args.primaryColor, args.logo, args.logoContentType, args.timezone, args.location],
+        rowMode: "array"
+    });
+    if (result.rows.length !== 1) {
+        return null;
+    }
+    const row = result.rows[0];
+    return {
+        id: row[0],
+        slug: row[1],
+        name: row[2],
+        description: row[3],
+        primaryColor: row[4],
+        hasLogo: row[5],
+        timezone: row[6],
+        location: row[7],
+        createdAt: row[8],
+        updatedAt: row[9]
+    };
+}
 exports.getOrganizationBySlugQuery = `-- name: GetOrganizationBySlug :one
 SELECT
     id,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "weave-typescript",
-  "version": "0.48.0",
+  "version": "0.50.0",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "files": [