@almadar/integrations 2.35.0 → 2.36.0

package/dist/runtime/index.d.ts

@@ -1,5 +1,5 @@
 import { g as IntegrationParams, I as IntegrationCallContext, d as IntegrationFactory } from '../BaseIntegration-BYlbMFlf.js';
-import { o as IntegrationName, m as IntegrationActionName, n as IntegrationContracts, e as CredentialStore } from '../store-BGNfuoFd.js';
+import { o as IntegrationName, m as IntegrationActionName, n as IntegrationContracts, e as CredentialStore } from '../store-OiFX0Tkn.js';
 import '@almadar/core';
 
 /**

package/dist/runtime/index.js

@@ -5349,6 +5349,18 @@ registerIntegration("arxiv", ArxivIntegration);
 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 {
@@ -5382,6 +5394,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;
@@ -5407,6 +5434,12 @@ var RiggerIntegration = class extends BaseIntegration {
         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;
@@ -5502,6 +5535,32 @@ var RiggerIntegration = class extends BaseIntegration {
     }
     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`, {
@@ -5577,28 +5636,41 @@ function createCallServiceHandler(factory) {
 }
 
 // src/runtime/RuntimeIntegrationManager.ts
-function generateMockFromShape(shape) {
-  const data = {};
-  for (const [key, type] of Object.entries(shape)) {
-    switch (type) {
+function mockValueForDescriptor(key, descriptor) {
+  if (typeof descriptor === "string") {
+    if (descriptor.endsWith("[]")) return [];
+    switch (descriptor) {
       case "string":
-        data[key] = `mock_${key}`;
-        break;
+        return `mock_${key}`;
       case "number":
-        data[key] = 42;
-        break;
+        return 42;
       case "boolean":
-        data[key] = true;
-        break;
+        return true;
       case "object":
-        data[key] = {};
-        break;
-      case "string[]":
-        data[key] = [];
-        break;
+        return {};
       default:
-        data[key] = `mock_${key}`;
+        return `mock_${key}`;
+    }
+  }
+  if (descriptor.type === "array") {
+    return descriptor.items ? [mockValueForDescriptor(`${key}Item`, descriptor.items)] : [];
+  }
+  if (descriptor.properties) {
+    const obj = {};
+    for (const [propKey, propDescriptor] of Object.entries(descriptor.properties)) {
+      obj[propKey] = mockValueForDescriptor(propKey, propDescriptor);
     }
+    return obj;
+  }
+  if (descriptor.mapValue) {
+    return { sample: mockValueForDescriptor("sample", descriptor.mapValue) };
+  }
+  return {};
+}
+function generateMockFromShape(shape) {
+  const data = {};
+  for (const [key, descriptor] of Object.entries(shape)) {
+    data[key] = mockValueForDescriptor(key, descriptor);
   }
   return data;
 }