import { InternalStreamFailure, ModelStreamError } from "./errors";
export interface InvokeEndpointInput {
  EndpointName: string | undefined;
  Body: Uint8Array | undefined;
  ContentType?: string | undefined;
  Accept?: string | undefined;
  CustomAttributes?: string | undefined;
  TargetModel?: string | undefined;
  TargetVariant?: string | undefined;
  TargetContainerHostname?: string | undefined;
  InferenceId?: string | undefined;
  EnableExplanations?: string | undefined;
  InferenceComponentName?: string | undefined;
  SessionId?: string | undefined;
}
export interface InvokeEndpointOutput {
  Body: Uint8Array | undefined;
  ContentType?: string | undefined;
  InvokedProductionVariant?: string | undefined;
  CustomAttributes?: string | undefined;
  NewSessionId?: string | undefined;
  ClosedSessionId?: string | undefined;
}
export interface InvokeEndpointAsyncInput {
  EndpointName: string | undefined;
  ContentType?: string | undefined;
  Accept?: string | undefined;
  CustomAttributes?: string | undefined;
  InferenceId?: string | undefined;
  InputLocation: string | undefined;
  S3OutputPathExtension?: string | undefined;
  Filename?: string | undefined;
  RequestTTLSeconds?: number | undefined;
  InvocationTimeoutSeconds?: number | undefined;
}
export interface InvokeEndpointAsyncOutput {
  InferenceId?: string | undefined;
  OutputLocation?: string | undefined;
  FailureLocation?: string | undefined;
}
export interface InvokeEndpointWithResponseStreamInput {
  EndpointName: string | undefined;
  Body: Uint8Array | undefined;
  ContentType?: string | undefined;
  Accept?: string | undefined;
  CustomAttributes?: string | undefined;
  TargetVariant?: string | undefined;
  TargetContainerHostname?: string | undefined;
  InferenceId?: string | undefined;
  InferenceComponentName?: string | undefined;
  SessionId?: string | undefined;
}
export interface PayloadPart {
  Bytes?: Uint8Array | undefined;
}
export type ResponseStream =
  | ResponseStream.InternalStreamFailureMember
  | ResponseStream.ModelStreamErrorMember
  | ResponseStream.PayloadPartMember
  | ResponseStream.$UnknownMember;
export declare namespace ResponseStream {
  interface PayloadPartMember {
    PayloadPart: PayloadPart;
    ModelStreamError?: never;
    InternalStreamFailure?: never;
    $unknown?: never;
  }
  interface ModelStreamErrorMember {
    PayloadPart?: never;
    ModelStreamError: ModelStreamError;
    InternalStreamFailure?: never;
    $unknown?: never;
  }
  interface InternalStreamFailureMember {
    PayloadPart?: never;
    ModelStreamError?: never;
    InternalStreamFailure: InternalStreamFailure;
    $unknown?: never;
  }
  interface $UnknownMember {
    PayloadPart?: never;
    ModelStreamError?: never;
    InternalStreamFailure?: never;
    $unknown: [string, any];
  }
  interface Visitor<T> {
    PayloadPart: (value: PayloadPart) => T;
    ModelStreamError: (value: ModelStreamError) => T;
    InternalStreamFailure: (value: InternalStreamFailure) => T;
    _: (name: string, value: any) => T;
  }
}
export interface InvokeEndpointWithResponseStreamOutput {
  Body: AsyncIterable<ResponseStream> | undefined;
  ContentType?: string | undefined;
  InvokedProductionVariant?: string | undefined;
  CustomAttributes?: string | undefined;
}