@almadar/integrations 2.34.0 → 2.36.0

package/dist/index.d.ts

@@ -1,7 +1,7 @@
 import { e as IntegrationLogger, c as IntegrationErrorCode, a as IntegrationConfig, B as BaseIntegration, d as IntegrationFactory, g as IntegrationParams, h as IntegrationResult, I as IntegrationCallContext } from './BaseIntegration-BYlbMFlf.js';
 export { b as IntegrationError, f as IntegrationParamValue, U as UnknownActionError, V as ValidationError, i as ValidationResult, j as getIntegrationFactory, r as resetIntegrationFactory, v as validateParams } from './BaseIntegration-BYlbMFlf.js';
-import { e as CredentialStore, d as CredentialPersistence, H as HookDeclaration } from './store-DaPuMedl.js';
-export { A as ArxivActions, C as CLIActions, a as CREDENTIAL_ENTITY_TYPE, b as CREDENTIAL_MASTER_KEY_ENV, c as CredentialEntry, D as DatabaseActions, f as DatabaseDriver, g as DatabaseQueryParamValue, h as DatabaseQueryParams, i as DatabaseQueryResult, j as DatabaseRow, k as DeepAgentActions, l as DockerActions, E as EmailActions, G as GitHubActions, I as IconifyActions, m as IntegrationActionName, n as IntegrationContracts, o as IntegrationName, L as LLMIntegrationActions, M as MLActions, O as OAuthActions, p as OtelActions, Q as QueueActions, R as RedisActions, S as StorageActions, q as StripeActions, T as TwilioActions, W as WebhookActions, r as WikimediaActions, Y as YouTubeActions, s as allCredentialEnvVars, t as declaredCredentialEnvVars, u as serviceHooks } from './store-DaPuMedl.js';
+import { e as CredentialStore, d as CredentialPersistence, H as HookDeclaration } from './store-OiFX0Tkn.js';
+export { A as ArxivActions, C as CLIActions, a as CREDENTIAL_ENTITY_TYPE, b as CREDENTIAL_MASTER_KEY_ENV, c as CredentialEntry, D as DatabaseActions, f as DatabaseDriver, g as DatabaseQueryParamValue, h as DatabaseQueryParams, i as DatabaseQueryResult, j as DatabaseRow, k as DeepAgentActions, l as DockerActions, E as EmailActions, G as GitHubActions, I as IconifyActions, m as IntegrationActionName, n as IntegrationContracts, o as IntegrationName, L as LLMIntegrationActions, M as MLActions, O as OAuthActions, p as OtelActions, Q as QueueActions, R as RedisActions, S as StorageActions, q as StripeActions, T as TwilioActions, W as WebhookActions, r as WikimediaActions, Y as YouTubeActions, s as allCredentialEnvVars, t as declaredCredentialEnvVars, u as serviceHooks } from './store-OiFX0Tkn.js';
 import { LogMeta, EntityRow, EventPayload } from '@almadar/core';
 export { GitHubIntegration } from './integrations/github/index.js';
 
@@ -1024,11 +1024,19 @@ declare class ArxivIntegration extends BaseIntegration {
 declare class RiggerIntegration extends BaseIntegration {
     private readonly baseUrl;
     private readonly timeoutMs;
+    private readonly generateTimeoutMs;
     constructor(config: IntegrationConfig);
     execute(action: string, params: IntegrationParams): Promise<IntegrationResult>;
     private health;
     private buildMesh;
     private render;
+    private listPresets;
+    private loadRig;
+    private saveRig;
+    private ownership;
+    private buildLayers;
+    private pose;
+    private generate;
 }
 
 export { AccountingIntegration, type AlmadarCheckoutSession, type AlmadarCustomer, type AlmadarInvoiceFailure, type AlmadarInvoicePayment, type AlmadarPortalSession, type AlmadarStripeEvent, type AlmadarStripeEventOk, type AlmadarSubscription, type AlmadarSubscriptionStatus, type AlmadarTier, ArxivIntegration, type ArxivResult, BankingIntegration, BaseIntegration, CLIIntegration, CREDENTIALS_FILE_ENV, CalendarIntegration, type CancelSubscriptionInput, ConsoleLogger, type CreateCheckoutInput, type CreateCustomerInput, type CreatePortalInput, CredentialPersistence, CredentialStore, CredentialsIntegration, DatabaseIntegration, DeepAgentIntegration, DockerIntegration, DriveIntegration, EmailIntegration, EsignIntegration, FileCredentialPersistence, HookDeclaration, type HookProviderInput, type HookProviderResult, IconifyIntegration, InMemoryPendingGrantStore, IntegrationCallContext, IntegrationConfig, type IntegrationConstructor, IntegrationErrorCode, IntegrationFactory, IntegrationLogger, IntegrationParams, IntegrationResult, LLMIntegration, MLIntegration, MetaAdsIntegration, OAuthIntegration, OtelIntegration, type PendingGrantStore, PushIntegration, QueueIntegration, RedisIntegration, type RegisteredHookProvider, type RetryConfig, RiggerIntegration, type SqlGuardResult, StorageIntegration, StripeIntegration, type StripePriceMap, TwilioIntegration, type UpdateSubscriptionInput, type VerifyAndParseInput, WebhookIntegration, WikimediaIntegration, YouTubeIntegration, allHookDeclarations, assertReadOnlySelect, getActiveFactory, getInstalledCredentialStore, getIntegration, getRegisteredIntegrations, googleCalendarHookProvider, installActiveFactory, installCredentialStore, installPendingGrantStore, isKnownIntegration, parseCalendarPushNotification, registerIntegration, registeredHookProviders, resolveCredentialRef, uninstallCredentialStore, verifyAndParseStripeEvent, withRetry };

package/dist/index.js

@@ -161,7 +161,8 @@ var serviceCredentials = {
   ],
   rigger: [
     { envVar: "RIGGER_BASE_URL", required: false, description: "Local rigger service base URL (default http://127.0.0.1:8181)" },
-    { envVar: "RIGGER_TIMEOUT_MS", required: false, description: "Per-request timeout (ms)" }
+    { envVar: "RIGGER_TIMEOUT_MS", required: false, description: "Per-request timeout (ms)" },
+    { envVar: "RIGGER_GENERATE_TIMEOUT_MS", required: false, description: "Timeout (ms) for the 'generate' action; a local z-image-turbo run takes ~60-90 s and a kontext edit at 1024px several minutes (default 600000)" }
   ]
 };
 function allCredentialEnvVars() {
@@ -5836,11 +5837,36 @@ registerIntegration("arxiv", ArxivIntegration);
 // src/integrations/rigger/index.ts
 var DEFAULT_BASE_URL = "http://127.0.0.1:8181";
 var DEFAULT_TIMEOUT_MS5 = 3e4;
+var DEFAULT_GENERATE_TIMEOUT_MS = 6e5;
+var GENERATE_MODELS = ["z-image-turbo", "kontext"];
+function generateParamErrors(params) {
+  const errors = [];
+  const model = params.model;
+  if (typeof model !== "string" || !GENERATE_MODELS.includes(model)) {
+    errors.push(`model must be one of ${GENERATE_MODELS.join(", ")}`);
+  }
+  if (model === "kontext" && (typeof params.sourcePath !== "string" || params.sourcePath === "")) {
+    errors.push("sourcePath is required when model is kontext");
+  }
+  return errors;
+}
+async function riggerError(path, response) {
+  let detail = "";
+  try {
+    const body = await response.json();
+    if (typeof body.detail === "string") detail = body.detail;
+    else if (body.detail !== void 0) detail = JSON.stringify(body.detail);
+  } catch {
+    detail = "";
+  }
+  return new Error(`Rigger ${path} returned ${response.status}${detail ? `: ${detail}` : ""}`);
+}
 var RiggerIntegration = class extends BaseIntegration {
   constructor(config) {
     super(config);
     this.baseUrl = config.env.RIGGER_BASE_URL || DEFAULT_BASE_URL;
     this.timeoutMs = Number(config.env.RIGGER_TIMEOUT_MS) || DEFAULT_TIMEOUT_MS5;
+    this.generateTimeoutMs = Number(config.env.RIGGER_GENERATE_TIMEOUT_MS) || DEFAULT_GENERATE_TIMEOUT_MS;
     this.logger.info("Rigger integration initialized", { baseUrl: this.baseUrl });
   }
   async execute(action, params) {
@@ -5857,6 +5883,21 @@ var RiggerIntegration = class extends BaseIntegration {
         metadata: this.createMetadata(action, 0)
       };
     }
+    if (action === "generate") {
+      const invalid = generateParamErrors(params);
+      if (invalid.length > 0) {
+        return {
+          success: false,
+          error: {
+            name: "IntegrationError",
+            message: "Validation failed",
+            code: "VALIDATION_ERROR",
+            details: invalid
+          },
+          metadata: this.createMetadata(action, 0)
+        };
+      }
+    }
     const startTime = Date.now();
     try {
       let data;
@@ -5870,6 +5911,27 @@ var RiggerIntegration = class extends BaseIntegration {
         case "render":
           data = await this.executeWithRetry(() => this.render(params));
           break;
+        case "listPresets":
+          data = await this.executeWithRetry(() => this.listPresets());
+          break;
+        case "loadRig":
+          data = await this.executeWithRetry(() => this.loadRig(params));
+          break;
+        case "saveRig":
+          data = await this.executeWithRetry(() => this.saveRig(params));
+          break;
+        case "ownership":
+          data = await this.executeWithRetry(() => this.ownership(params));
+          break;
+        case "buildLayers":
+          data = await this.executeWithRetry(() => this.buildLayers(params));
+          break;
+        case "pose":
+          data = await this.executeWithRetry(() => this.pose(params));
+          break;
+        case "generate":
+          data = await this.executeWithRetry(() => this.generate(params));
+          break;
         default:
           throw new UnknownActionError(this.config.name, action);
       }
@@ -5897,22 +5959,109 @@ var RiggerIntegration = class extends BaseIntegration {
       signal: AbortSignal.timeout(this.timeoutMs)
     });
     if (!response.ok) {
-      throw new Error(`Rigger /mesh/build returned ${response.status}`);
+      throw await riggerError("/mesh/build", response);
     }
     return await response.json();
   }
   async render(params) {
-    const { imagePath, rig, preset, outDir } = params;
+    const { imagePath, rig, preset, outDir, mode, supersample } = params;
     const response = await fetch(`${this.baseUrl}/render`, {
       method: "POST",
       headers: { "Content-Type": "application/json" },
-      body: JSON.stringify({ imagePath, rig, preset, outDir }),
+      body: JSON.stringify({ imagePath, rig, preset, outDir, mode, supersample }),
       signal: AbortSignal.timeout(this.timeoutMs)
     });
     if (!response.ok) {
-      throw new Error(`Rigger /render returned ${response.status}`);
+      throw await riggerError("/render", response);
     }
-    return response.json();
+    return await response.json();
+  }
+  async listPresets() {
+    const response = await fetch(`${this.baseUrl}/presets`, {
+      signal: AbortSignal.timeout(this.timeoutMs)
+    });
+    if (!response.ok) {
+      throw await riggerError("/presets", response);
+    }
+    return await response.json();
+  }
+  async loadRig(params) {
+    const { rigPath } = params;
+    const response = await fetch(`${this.baseUrl}/rig/load`, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ rigPath }),
+      signal: AbortSignal.timeout(this.timeoutMs)
+    });
+    if (!response.ok) {
+      throw await riggerError("/rig/load", response);
+    }
+    return await response.json();
+  }
+  async saveRig(params) {
+    const { rigPath, rig } = params;
+    const response = await fetch(`${this.baseUrl}/rig/save`, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ rigPath, rig }),
+      signal: AbortSignal.timeout(this.timeoutMs)
+    });
+    if (!response.ok) {
+      throw await riggerError("/rig/save", response);
+    }
+    return await response.json();
+  }
+  async ownership(params) {
+    const { imagePath, rig, outPath } = params;
+    const response = await fetch(`${this.baseUrl}/ownership`, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ imagePath, rig, outPath }),
+      signal: AbortSignal.timeout(this.timeoutMs)
+    });
+    if (!response.ok) {
+      throw await riggerError("/ownership", response);
+    }
+    return await response.json();
+  }
+  async buildLayers(params) {
+    const { imagePath, rig, outDir } = params;
+    const response = await fetch(`${this.baseUrl}/layers`, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ imagePath, rig, outDir }),
+      signal: AbortSignal.timeout(this.timeoutMs)
+    });
+    if (!response.ok) {
+      throw await riggerError("/layers", response);
+    }
+    return await response.json();
+  }
+  async pose(params) {
+    const { imagePath, rig, roles, preset, overrides } = params;
+    const response = await fetch(`${this.baseUrl}/pose`, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ imagePath, rig, roles, preset, overrides }),
+      signal: AbortSignal.timeout(this.timeoutMs)
+    });
+    if (!response.ok) {
+      throw await riggerError("/pose", response);
+    }
+    return await response.json();
+  }
+  async generate(params) {
+    const { model, prompt, seed, size, steps, sourcePath, outDir } = params;
+    const response = await fetch(`${this.baseUrl}/generate`, {
+      method: "POST",
+      headers: { "Content-Type": "application/json" },
+      body: JSON.stringify({ model, prompt, seed, size, steps, sourcePath, outDir }),
+      signal: AbortSignal.timeout(this.generateTimeoutMs)
+    });
+    if (!response.ok) {
+      throw await riggerError("/generate", response);
+    }
+    return await response.json();
   }
 };
 registerIntegration("rigger", RiggerIntegration);