/**
 * @license
 * Copyright 2025 Steven Roussey <sroussey@gmail.com>
 * SPDX-License-Identifier: Apache-2.0
 */
import type { IRunConfig, TaskConfig } from "@workglow/task-graph";
import { CreateWorkflow } from "@workglow/task-graph";
import type { ImageValue } from "@workglow/util/media";
import type { DataPortSchema } from "@workglow/util/schema";
import type { ModelConfig } from "../model/ModelSchema";
import { AiVisionTask } from "./base/AiVisionTask";
export declare const ImageSegmentationInputSchema: {
    readonly type: "object";
    readonly properties: {
        readonly image: {
            readonly oneOf: readonly [{
                readonly type: "string";
                readonly format: "image:data-uri";
            }, {
                readonly type: "object";
                readonly format: "image";
            }];
            readonly title: "Image";
            readonly description: "Image";
            readonly format: "image";
        };
        readonly model: {
            readonly oneOf: readonly [{
                readonly title: "Model";
                readonly description: `The model ${string}`;
            } & Record<string, never> & {
                readonly format: import(".").TypeModelSemantic;
                readonly type: "string";
            }, {
                readonly type: "object";
                readonly properties: {
                    readonly model_id: {
                        readonly type: "string";
                    };
                    readonly capabilities: {
                        readonly type: "array";
                        readonly items: {
                            readonly type: "string";
                        };
                        readonly "x-ui-editor": "multiselect";
                    };
                    readonly title: {
                        readonly type: "string";
                    };
                    readonly description: {
                        readonly type: "string";
                        readonly "x-ui-editor": "textarea";
                    };
                    readonly provider: {
                        readonly type: "string";
                    };
                    readonly provider_config: {
                        readonly type: "object";
                        readonly properties: {
                            readonly credential_key: {
                                readonly type: "string";
                                readonly format: "credential";
                                readonly "x-ui-hidden": true;
                            };
                            readonly native_dimensions: {
                                readonly type: "integer";
                                readonly description: "Native output vector dimensions for embedding models";
                            };
                            readonly mrl: {
                                readonly type: "boolean";
                                readonly description: "Whether the model supports Matryoshka Representation Learning";
                            };
                        };
                        readonly additionalProperties: true;
                        readonly default: {};
                    };
                    readonly metadata: {
                        readonly type: "object";
                        readonly default: {};
                        readonly "x-ui-hidden": true;
                    };
                };
                readonly required: readonly ["provider", "provider_config"];
                readonly format: "model";
                readonly additionalProperties: true;
            } & Record<string, never> & {
                readonly format: import(".").TypeModelSemantic;
            }];
        } & Record<string, never> & {
            readonly format: import(".").TypeModelSemantic;
        };
        readonly threshold: {
            readonly type: "number";
            readonly title: "Threshold";
            readonly description: "The threshold for filtering masks by score";
            readonly minimum: 0;
            readonly maximum: 1;
            readonly default: 0.5;
            readonly "x-ui-group": "Configuration";
        };
        readonly maskThreshold: {
            readonly type: "number";
            readonly title: "Mask Threshold";
            readonly description: "Threshold to use when turning predicted masks into binary values";
            readonly minimum: 0;
            readonly maximum: 1;
            readonly default: 0.5;
            readonly "x-ui-group": "Configuration";
        };
    };
    readonly required: readonly ["image", "model"];
    readonly additionalProperties: false;
};
export declare const ImageSegmentationOutputSchema: {
    readonly type: "object";
    readonly properties: {
        readonly masks: {
            readonly type: "array";
            readonly items: {
                readonly type: "object";
                readonly properties: {
                    readonly label: {
                        readonly type: "string";
                        readonly title: "Label";
                        readonly description: "The label of the segmented region";
                    };
                    readonly score: {
                        readonly type: "number";
                        readonly title: "Score";
                        readonly description: "The confidence score for this segmentation";
                        readonly minimum: 0;
                        readonly maximum: 1;
                    };
                    readonly mask: {
                        readonly oneOf: readonly [{
                            readonly type: "string";
                            readonly format: "image:data-uri";
                        }, {
                            readonly type: "object";
                            readonly format: "image";
                        }];
                        readonly title: "Image";
                        readonly description: "Image";
                        readonly format: "image";
                    };
                };
                readonly required: readonly ["label", "score", "mask"];
                readonly additionalProperties: false;
            };
            readonly title: "Segmentation Masks";
            readonly description: "The segmented regions with their labels, scores, and masks";
        };
    };
    readonly required: readonly ["masks"];
    readonly additionalProperties: false;
};
export type ImageSegmentationTaskInput = Omit<{
    threshold?: number | undefined;
    maskThreshold?: number | undefined;
    model: string | ModelConfig;
    image: string | {
        [x: string]: unknown;
    };
}, "image"> & {
    image: ImageValue;
};
export type ImageSegmentationTaskOutput = {
    masks: {
        label: string;
        score: number;
        mask: ImageValue;
    }[];
};
export type ImageSegmentationTaskConfig = TaskConfig<ImageSegmentationTaskInput>;
export declare class ImageSegmentationTask extends AiVisionTask<ImageSegmentationTaskInput, ImageSegmentationTaskOutput, ImageSegmentationTaskConfig> {
    static type: string;
    /** Capabilities required of the model; gated in {@link AiTask.execute}. */
    static readonly requires: ["image.segmentation"];
    static category: string;
    static title: string;
    static description: string;
    static inputSchema(): DataPortSchema;
    static outputSchema(): DataPortSchema;
}
export declare const imageSegmentation: (input: ImageSegmentationTaskInput, config?: ImageSegmentationTaskConfig, runConfig?: Partial<IRunConfig>) => Promise<ImageSegmentationTaskOutput>;
declare module "@workglow/task-graph" {
    interface Workflow {
        imageSegmentation: CreateWorkflow<ImageSegmentationTaskInput, ImageSegmentationTaskOutput, ImageSegmentationTaskConfig>;
    }
}
//# sourceMappingURL=ImageSegmentationTask.d.ts.map