import { type MswarmCapabilityRequestOptions, type MswarmGenericJobArtifactUploadInput, type MswarmGenericJobArtifactUploadResult, type MswarmGenericJobArtifactsResult, type MswarmGenericJobEventsResult, type MswarmGenericJobLifecycleSnapshot, type MswarmGenericJobLogsResult, type MswarmGenericJobOpsRequestOptions, type MswarmGenericJobOpsSummary, type MswarmGenericJobReference, type MswarmGenericNodeJobEnvelope } from "@mcoda/core";
type MswarmGenericJobRequestInput = MswarmGenericNodeJobEnvelope["job"];
type McodaGpuJobCreateInput = MswarmGenericNodeJobEnvelope | MswarmGenericJobRequestInput;
type McodaGpuJobEvent = MswarmGenericJobEventsResult["events"][number];
export interface McodaGpuJobEventStreamOptions extends Partial<MswarmGenericJobReference> {
    intervalMs?: number;
    stopOnTerminal?: boolean;
}
export interface McodaGpuJobsNamespace {
    create(job: McodaGpuJobCreateInput, input?: Partial<MswarmGenericJobReference>): Promise<MswarmGenericJobLifecycleSnapshot>;
    run(job: McodaGpuJobCreateInput, input?: Partial<MswarmGenericJobReference>): Promise<MswarmGenericJobLifecycleSnapshot>;
    status(input: Partial<MswarmGenericJobReference> & {
        jobId: string;
    }): Promise<MswarmGenericJobLifecycleSnapshot>;
    logs(input: Partial<MswarmGenericJobReference> & {
        jobId: string;
    }): Promise<MswarmGenericJobLogsResult>;
    events(jobIdOrInput: string | (Partial<MswarmGenericJobReference> & {
        jobId: string;
    }), input?: McodaGpuJobEventStreamOptions): AsyncIterable<McodaGpuJobEvent>;
    eventsSnapshot(input: Partial<MswarmGenericJobReference> & {
        jobId: string;
    }): Promise<MswarmGenericJobEventsResult>;
    artifacts(input: Partial<MswarmGenericJobReference> & {
        jobId: string;
    }): Promise<MswarmGenericJobArtifactsResult>;
    cancel(input: Partial<MswarmGenericJobReference> & {
        jobId: string;
    }): Promise<MswarmGenericJobLifecycleSnapshot>;
    retry(input: Partial<MswarmGenericJobReference> & {
        jobId: string;
    }): Promise<MswarmGenericJobLifecycleSnapshot>;
    ops(input?: Partial<MswarmGenericJobOpsRequestOptions>): Promise<MswarmGenericJobOpsSummary>;
}
export interface CreateMcodaGpuJobClientInput {
    baseUrl?: string;
    apiKey?: string;
    timeoutMs?: number;
    nodeBaseUrl?: string;
    nodeId?: string;
    signingSecret?: string;
    token?: string;
    tokenTtlSeconds?: number;
}
export interface McodaGpuJobClient {
    close(): Promise<void>;
    jobs: McodaGpuJobsNamespace;
    listGpus(input?: Partial<MswarmCapabilityRequestOptions>): Promise<Record<string, unknown>>;
    uploadArtifact(input: Omit<MswarmGenericJobArtifactUploadInput, keyof MswarmGenericJobReference> & Partial<MswarmGenericJobReference> & {
        jobId: string;
    }): Promise<MswarmGenericJobArtifactUploadResult>;
    runJob(job: McodaGpuJobCreateInput, input?: Partial<MswarmGenericJobReference>): Promise<MswarmGenericJobLifecycleSnapshot>;
    create(job: McodaGpuJobCreateInput, input?: Partial<MswarmGenericJobReference>): Promise<MswarmGenericJobLifecycleSnapshot>;
    status(input: Partial<MswarmGenericJobReference> & {
        jobId: string;
    }): Promise<MswarmGenericJobLifecycleSnapshot>;
    logs(input: Partial<MswarmGenericJobReference> & {
        jobId: string;
    }): Promise<MswarmGenericJobLogsResult>;
    events(input: Partial<MswarmGenericJobReference> & {
        jobId: string;
    }): Promise<MswarmGenericJobEventsResult>;
    artifacts(input: Partial<MswarmGenericJobReference> & {
        jobId: string;
    }): Promise<MswarmGenericJobArtifactsResult>;
    cancel(input: Partial<MswarmGenericJobReference> & {
        jobId: string;
    }): Promise<MswarmGenericJobLifecycleSnapshot>;
    retry(input: Partial<MswarmGenericJobReference> & {
        jobId: string;
    }): Promise<MswarmGenericJobLifecycleSnapshot>;
    ops(input?: Partial<MswarmGenericJobOpsRequestOptions>): Promise<MswarmGenericJobOpsSummary>;
}
export declare function createMcodaGpuJobClient(input?: CreateMcodaGpuJobClientInput): Promise<McodaGpuJobClient>;
export {};
//# sourceMappingURL=jobClient.d.ts.map