askui 0.7.1 → 0.8.0

package/dist/cjs/core/inference-response/inference-response.d.ts

@@ -1,15 +1,11 @@
 import { ControlCommand } from '../ui-control-commands/control-command';
 import { Annotation } from '../annotation/annotation';
-export declare class InferenceResponse {
-    type: string;
-    data: ControlCommand | Annotation;
-    constructor(type: string, data: ControlCommand | Annotation);
-    static fromJson(json: unknown, resizeRatio?: number, image?: string): ControlCommand | Annotation;
-    static createModels(type: string, data: ControlCommand | Annotation, resizeRatio: number, image?: string): ControlCommand | Annotation;
-    static models: Models;
+import { ModelType } from './model-type';
+export interface InferenceResponseBody {
+    type: ModelType;
+    data: ModelType extends 'COMMANDS' ? ControlCommand : Annotation;
 }
-interface Models {
-    DETECTED_ELEMENTS: CallableFunction;
-    COMMANDS: CallableFunction;
+export declare class InferenceResponse {
+    static fromJson(json: InferenceResponseBody, resizeRatio?: number, image?: string): ControlCommand | Annotation;
+    static createModels(type: ModelType, data: ModelType extends 'COMMANDS' ? ControlCommand : Annotation, resizeRatio: number, image?: string): ControlCommand | Annotation;
 }
-export {};

package/dist/cjs/core/inference-response/inference-response.js

@@ -3,23 +3,18 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.InferenceResponse = void 0;
 const control_command_1 = require("../ui-control-commands/control-command");
 const annotation_1 = require("../annotation/annotation");
+const invalid_model_type_error_1 = require("./invalid-model-type-error");
 class InferenceResponse {
-    constructor(type, data) {
-        this.type = type;
-        this.data = data;
-    }
     static fromJson(json, resizeRatio = 1, image) {
-        const inferenceResponse = json;
-        return this.createModels(inferenceResponse.type, inferenceResponse.data, resizeRatio, image);
+        return this.createModels(json.type, json.data, resizeRatio, image);
     }
     static createModels(type, data, resizeRatio, image) {
-        return this.models[type](data, resizeRatio, image);
+        if (type === 'COMMANDS')
+            return control_command_1.ControlCommand.fromJson(data, resizeRatio);
+        if (type === 'DETECTED_ELEMENTS') {
+            return annotation_1.Annotation.fromJson({ image, detected_elements: data.detected_elements }, resizeRatio);
+        }
+        throw new invalid_model_type_error_1.InvalidModelTypeError(type);
     }
 }
 exports.InferenceResponse = InferenceResponse;
-InferenceResponse.models = {
-    DETECTED_ELEMENTS: (data, resizeRatio, image) => annotation_1.Annotation
-        .fromJson({ image, detected_elements: data.detected_elements }, resizeRatio),
-    COMMANDS: (data, resizeRatio) => control_command_1.ControlCommand
-        .fromJson(data, resizeRatio),
-};

package/dist/cjs/core/inference-response/invalid-model-type-error.d.ts

@@ -0,0 +1,4 @@
+import { ModelType } from './model-type';
+export declare class InvalidModelTypeError extends Error {
+    constructor(type: ModelType);
+}

package/dist/cjs/core/inference-response/invalid-model-type-error.js

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.InvalidModelTypeError = void 0;
+class InvalidModelTypeError extends Error {
+    constructor(type) {
+        super(`Invalid model type: ${type}`);
+    }
+}
+exports.InvalidModelTypeError = InvalidModelTypeError;

package/dist/cjs/core/inference-response/model-type.d.ts

@@ -0,0 +1 @@
+export declare type ModelType = 'DETECTED_ELEMENTS' | 'COMMANDS';

package/dist/cjs/core/inference-response/model-type.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });