@hautechai/sdk 0.1.4 → 0.2.0

package/dist/sdk/pipelines/index.js

@@ -4,7 +4,9 @@ import { useAutogeneratedAPI } from '../api';
 const pipelines = (options) => {
     const api = useAutogeneratedAPI({ API: PipelinesApi, options });
     const callAPI = useAutogeneratedAPI({ API: CallApi, options });
-    const callMethod = (method) => (params) => callAPI.call({
+    const callMethod = (method) => 
+    // Type intersection is used as a type check assistance to prevent accidental passing of raw TaskOutput.
+    (params) => callAPI.call({
         run: (methods) => method(methods)(params),
     });
     const createPipeline = () => new Pipeline({
@@ -50,6 +52,61 @@ const pipelines = (options) => {
             get: callMethod((methods) => methods.callControllerCallOperationsGetV1),
             list: callMethod((methods) => methods.callControllerCallOperationsListV1),
             updateMetadata: callMethod((methods) => methods.callControllerCallOperationsMetadataUpdateV1),
+            run: {
+                haute: {
+                    linda: {
+                        v1: callMethod((methods) => methods.callControllerCallOperationsRunHauteLindaV1V1)
+                    },
+                    naomi: {
+                        v1: callMethod((methods) => methods.callControllerCallOperationsRunHauteNaomiV1V1)
+                    },
+                },
+                inpaint: {
+                    kate: {
+                        v1: callMethod((methods) => methods.callControllerCallOperationsRunInpaintKateV1V1),
+                    }
+                },
+                gpt: {
+                    v1: callMethod((methods) => methods.callControllerCallOperationsRunGptV1V1)
+                },
+                imagine: {
+                    kate: {
+                        v1: callMethod((methods) => methods.callControllerCallOperationsRunImagineKateV1V1)
+                    },
+                },
+                upscale: {
+                    v1: callMethod((methods) => methods.callControllerCallOperationsRunUpscaleV1V1),
+                },
+                objectDetection: {
+                    v1: callMethod((methods) => methods.callControllerCallOperationsRunObjectDetectionV1V1),
+                },
+                segmentAnything: {
+                    embeddings: {
+                        v1: callMethod((methods) => methods.callControllerCallOperationsRunSegmentAnythingEmbeddingsV1V1),
+                    },
+                    mask: {
+                        v1: callMethod((methods) => methods.callControllerCallOperationsRunSegmentAnythingMaskV1V1),
+                    }
+                },
+                poseEstimation: {
+                    v1: callMethod((methods) => methods.callControllerCallOperationsRunPoseEstimationV1V1),
+                },
+                cut: {
+                    v1: callMethod((methods) => methods.callControllerCallOperationsRunCutV1V1),
+                },
+                composite: {
+                    v1: callMethod((methods) => methods.callControllerCallOperationsRunCompositeV1V1),
+                },
+                vton: {
+                    gisele: {
+                        v1: callMethod((methods) => methods.callControllerCallOperationsRunVtonGiseleV1V1),
+                    }
+                },
+                negateImage: {
+                    v1: callMethod((methods) => methods.callControllerCallOperationsRunNegateImageV1V1),
+                },
+            },
+            wait: callMethod((methods) => methods.callControllerCallOperationsWaitV1),
         },
         poses: {
             get: callMethod((methods) => methods.callControllerCallPosesGetV1),
@@ -90,7 +147,7 @@ const pipelines = (options) => {
                 if (pipeline.status === 'completed')
                     return resolve(pipeline);
                 if (pipeline.status === 'failed')
-                    return reject(pipeline);
+                    return resolve(pipeline);
                 timeoutId = setTimeout(poll, delay);
             };
             timeoutId = setTimeout(poll, initialDelay);

package/dist/sdk/pipelines/pipelines.js

@@ -0,0 +1 @@
+"use strict";

package/dist/token/index.js

@@ -31,24 +31,35 @@ const serializePermissions = (permissions) => {
     traverse(permissions);
     return result;
 };
-export const createTokenSigner = (options) => ({
-    createAccountToken: async (props) => createToken({
-        appKeyId: options.appKeyId,
-        appKeySecret: options.appKeySecret,
-        expiresInSeconds: props.expiresInSeconds,
-        payload: {
-            iss: options.appId,
-            permissions: serializePermissions(props.permissions ?? {}),
-            sub: props.accountId,
-        },
-    }),
-    createRootToken: async (props) => createToken({
-        appKeyId: options.appKeyId,
-        appKeySecret: options.appKeySecret,
-        expiresInSeconds: props.expiresInSeconds ?? 3600,
-        payload: {
-            iss: options.appId,
-            permissions: ['*'],
-        },
-    }),
-});
+export const createTokenSigner = (options) => {
+    if (!options.appId || options.appId.length === 0) {
+        throw new Error('appId is required');
+    }
+    if (!options.appKeyId || options.appKeyId.length === 0) {
+        throw new Error('appKeyId is required');
+    }
+    if (!options.appKeySecret || options.appKeySecret.length === 0) {
+        throw new Error('appKeySecret is required');
+    }
+    return {
+        createAccountToken: async (props) => createToken({
+            appKeyId: options.appKeyId,
+            appKeySecret: options.appKeySecret,
+            expiresInSeconds: props.expiresInSeconds,
+            payload: {
+                iss: options.appId,
+                permissions: serializePermissions(props.permissions ?? {}),
+                sub: props.accountId,
+            },
+        }),
+        createRootToken: async (props) => createToken({
+            appKeyId: options.appKeyId,
+            appKeySecret: options.appKeySecret,
+            expiresInSeconds: props.expiresInSeconds ?? 3600,
+            payload: {
+                iss: options.appId,
+                permissions: ['*'],
+            },
+        }),
+    };
+};

package/dist/types.d.ts

@@ -1,6 +1,7 @@
 export interface SDKOptions {
     authToken: () => string | Promise<string>;
     endpoint?: string;
+    useWebsocket?: boolean;
 }
 export type ListProps = {
     cursor?: string;

package/openapitools.json

@@ -1,7 +1,7 @@
 {
-  "$schema": "./node_modules/@openapitools/openapi-generator-cli/config.schema.json",
-  "spaces": 2,
-  "generator-cli": {
-    "version": "7.10.0"
-  }
+    "$schema": "./node_modules/@openapitools/openapi-generator-cli/config.schema.json",
+    "spaces": 2,
+    "generator-cli": {
+        "version": "7.10.0"
+    }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hautechai/sdk",
-  "version": "0.1.4",
+  "version": "0.2.0",
   "license": "MIT",
   "keywords": [],
   "repository": {
@@ -10,9 +10,10 @@
   "main": "dist/index.js",
   "description": "Hautech SDK",
   "dependencies": {
-    "@hautechai/pipelines": "0.1.0",
+    "@hautechai/pipelines": "0.3.2",
     "axios": "1.6.1",
-    "jose": "5.9.6"
+    "jose": "5.9.6",
+    "ws": "^8.18.0"
   },
   "devDependencies": {
     "@jest/globals": "^29.7.0",
@@ -20,6 +21,7 @@
     "@types/jest": "29.5.12",
     "@types/jsonwebtoken": "9.0.7",
     "@types/node": "22.5.2",
+    "@types/ws": "^8.5.14",
     "dotenv": "16.4.7",
     "jest": "29.7.0",
     "jest-environment-jsdom": "29.7.0",