import z from 'zod';
import { Tool } from 'fastmcp';
interface SignatureOptions {
    url: string;
    secretKey: string;
}
interface SignatureResult {
    signature: string;
    timestamp: number;
    signatureNonce: string;
}
type ParamsInput = {
    class_type: string;
    inputs: Record<string, string | number | boolean>;
};
export type GenerateParams = {
    workflowUuid: string;
    [nodeId: string]: ParamsInput | string;
};
interface ComfyUIRequest<T extends GenerateParams> {
    templateUuid: string;
    generateParams: T;
}
export interface ImageInfo {
    /**
     * 1：待审核
     * 2：审核中
     * 3：审核通过
     * 4：审核拦截
     * 5：审核失败
     */
    auditStatus: number;
    imageUrl: string;
    nodeId: string;
    outputName: string;
}
export interface GenerateStatusResult {
    accountBalance: number;
    /**
     * 1：等待执行
     * 2：执行中
     * 3：已生图
     * 4：审核中
     * 5：任务成功
     * 6：任务失败
     */
    generateStatus: number;
    generateUuid: string;
    images: ImageInfo[];
    percentCompleted: number;
    pointsCost: number;
    videos: any[];
    generateMsg?: string;
}
/**
 * 生成签名
 * @param options 签名选项
 * @returns 签名结果
 */
export declare function generateSignature(options: SignatureOptions): SignatureResult;
/**
 * 调用ComfyUI工作流
 * @param request 请求参数
 * @returns Promise<string>
 */
export declare function invokeComfyUIWorkflow<T extends GenerateParams>(request: ComfyUIRequest<T>): Promise<string>;
/**
 * 查询生图任务状态
 * @param generateUuid 生图任务UUID
 * @returns Promise<GenerateStatusResult>
 */
export declare function queryComfyUIImageStatus(generateUuid: string): Promise<GenerateStatusResult | string>;
export interface AdditionalNetwork {
    modelId: string;
    weight: number;
}
type PromptAdapter<T> = (args: T, defaultArgs?: T) => string;
export declare function qwenToolGenerator<T extends z.AnyZodObject>(options: {
    name: string;
    description: string;
    parameters: T;
    defaultHeight?: number;
    defaultWidth?: number;
    promptAdapter: PromptAdapter<z.infer<T>>;
    defaultParams?: z.infer<T>;
    additionalNetwork?: AdditionalNetwork[];
    enable8Step?: boolean;
}): Tool<any, T>;
export interface TextToImageParams {
    prompt: string;
    negativePrompt?: string;
    width?: number;
    height?: number;
    additionalNetwork?: AdditionalNetwork[];
    enable8Steps?: boolean;
}
/**
 *  使用千问模型调用文生图接口
 * @param request 请求参数
 * @returns Promise<string> 返回生成任务的UUID
 */
export declare function invokeTextToImageByQwen(params: TextToImageParams): Promise<string>;
export {};