worker

type APIError struct {
	Msg string `json:"msg"`
}

type AudioToTextMultipartRequestBody = BodyAudioToTextAudioToTextPost

type AudioToTextResponse struct {
	Body         []byte
	HTTPResponse *http.Response
	JSON200      *TextResponse
	JSON400      *HTTPError
	JSON401      *HTTPError
	JSON413      *HTTPError
	JSON422      *HTTPValidationError
	JSON500      *HTTPError
}

type BodyAudioToTextAudioToTextPost struct {
	Audio   openapi_types.File `json:"audio"`
	ModelId *string            `json:"model_id,omitempty"`
}

type BodyImageToImageImageToImagePost struct {
	GuidanceScale      *float32           `json:"guidance_scale,omitempty"`
	Image              openapi_types.File `json:"image"`
	ImageGuidanceScale *float32           `json:"image_guidance_scale,omitempty"`
	ModelId            *string            `json:"model_id,omitempty"`
	NegativePrompt     *string            `json:"negative_prompt,omitempty"`
	NumImagesPerPrompt *int               `json:"num_images_per_prompt,omitempty"`
	NumInferenceSteps  *int               `json:"num_inference_steps,omitempty"`
	Prompt             string             `json:"prompt"`
	SafetyCheck        *bool              `json:"safety_check,omitempty"`
	Seed               *int               `json:"seed,omitempty"`
	Strength           *float32           `json:"strength,omitempty"`
}

type BodyImageToVideoImageToVideoPost struct {
	Fps               *int               `json:"fps,omitempty"`
	Height            *int               `json:"height,omitempty"`
	Image             openapi_types.File `json:"image"`
	ModelId           *string            `json:"model_id,omitempty"`
	MotionBucketId    *int               `json:"motion_bucket_id,omitempty"`
	NoiseAugStrength  *float32           `json:"noise_aug_strength,omitempty"`
	NumInferenceSteps *int               `json:"num_inference_steps,omitempty"`
	SafetyCheck       *bool              `json:"safety_check,omitempty"`
	Seed              *int               `json:"seed,omitempty"`
	Width             *int               `json:"width,omitempty"`
}

type BodyUpscaleUpscalePost struct {
	Image             openapi_types.File `json:"image"`
	ModelId           *string            `json:"model_id,omitempty"`
	NumInferenceSteps *int               `json:"num_inference_steps,omitempty"`
	Prompt            string             `json:"prompt"`
	SafetyCheck       *bool              `json:"safety_check,omitempty"`
	Seed              *int               `json:"seed,omitempty"`
}

type ChiServerOptions struct {
	BaseURL          string
	BaseRouter       chi.Router
	Middlewares      []MiddlewareFunc
	ErrorHandlerFunc func(w http.ResponseWriter, r *http.Request, err error)
}

type Chunk struct {
	Text      string        `json:"text"`
	Timestamp []interface{} `json:"timestamp"`
}

type Client struct {
	// The endpoint of the server conforming to this interface, with scheme,
	// https://api.deepmap.com for example. This can contain a path relative
	// to the server, such as https://api.deepmap.com/dev-test, and all the
	// paths in the swagger spec will be appended to the server.
	Server string

	// Doer for performing requests, typically a *http.Client with any
	// customized settings, such as certificate chains.
	Client HttpRequestDoer

	// A list of callbacks for modifying requests which are generated before sending over
	// the network.
	RequestEditors []RequestEditorFn
}

type ClientInterface interface {
	// AudioToTextWithBody request with any body
	AudioToTextWithBody(ctx context.Context, contentType string, body io.Reader, reqEditors ...RequestEditorFn) (*http.Response, error)

	// Health request
	Health(ctx context.Context, reqEditors ...RequestEditorFn) (*http.Response, error)

	// ImageToImageWithBody request with any body
	ImageToImageWithBody(ctx context.Context, contentType string, body io.Reader, reqEditors ...RequestEditorFn) (*http.Response, error)

	// ImageToVideoWithBody request with any body
	ImageToVideoWithBody(ctx context.Context, contentType string, body io.Reader, reqEditors ...RequestEditorFn) (*http.Response, error)

	// TextToImageWithBody request with any body
	TextToImageWithBody(ctx context.Context, contentType string, body io.Reader, reqEditors ...RequestEditorFn) (*http.Response, error)

	TextToImage(ctx context.Context, body TextToImageJSONRequestBody, reqEditors ...RequestEditorFn) (*http.Response, error)

	// UpscaleWithBody request with any body
	UpscaleWithBody(ctx context.Context, contentType string, body io.Reader, reqEditors ...RequestEditorFn) (*http.Response, error)
}

type ClientOption func(*Client) error

type ClientWithResponses struct {
	ClientInterface
}

type ClientWithResponsesInterface interface {
	// AudioToTextWithBodyWithResponse request with any body
	AudioToTextWithBodyWithResponse(ctx context.Context, contentType string, body io.Reader, reqEditors ...RequestEditorFn) (*AudioToTextResponse, error)

	// HealthWithResponse request
	HealthWithResponse(ctx context.Context, reqEditors ...RequestEditorFn) (*HealthResponse, error)

	// ImageToImageWithBodyWithResponse request with any body
	ImageToImageWithBodyWithResponse(ctx context.Context, contentType string, body io.Reader, reqEditors ...RequestEditorFn) (*ImageToImageResponse, error)

	// ImageToVideoWithBodyWithResponse request with any body
	ImageToVideoWithBodyWithResponse(ctx context.Context, contentType string, body io.Reader, reqEditors ...RequestEditorFn) (*ImageToVideoResponse, error)

	// TextToImageWithBodyWithResponse request with any body
	TextToImageWithBodyWithResponse(ctx context.Context, contentType string, body io.Reader, reqEditors ...RequestEditorFn) (*TextToImageResponse, error)

	TextToImageWithResponse(ctx context.Context, body TextToImageJSONRequestBody, reqEditors ...RequestEditorFn) (*TextToImageResponse, error)

	// UpscaleWithBodyWithResponse request with any body
	UpscaleWithBodyWithResponse(ctx context.Context, contentType string, body io.Reader, reqEditors ...RequestEditorFn) (*UpscaleResponse, error)
}

type DockerManager struct {
	// contains filtered or unexported fields
}

type EnvValue string

type HTTPError struct {
	Detail APIError `json:"detail"`
}

type HTTPValidationError struct {
	Detail *[]ValidationError `json:"detail,omitempty"`
}

type HealthCheck struct {
	Status *string `json:"status,omitempty"`
}

type HealthResponse struct {
	Body         []byte
	HTTPResponse *http.Response
	JSON200      *HealthCheck
}

type HttpRequestDoer interface {
	Do(req *http.Request) (*http.Response, error)
}

type ImageResponse struct {
	Images []Media `json:"images"`
}

type ImageToImageMultipartRequestBody = BodyImageToImageImageToImagePost

type ImageToImageResponse struct {
	Body         []byte
	HTTPResponse *http.Response
	JSON200      *ImageResponse
	JSON400      *HTTPError
	JSON401      *HTTPError
	JSON422      *HTTPValidationError
	JSON500      *HTTPError
}

type ImageToVideoMultipartRequestBody = BodyImageToVideoImageToVideoPost

type ImageToVideoResponse struct {
	Body         []byte
	HTTPResponse *http.Response
	JSON200      *VideoResponse
	JSON400      *HTTPError
	JSON401      *HTTPError
	JSON422      *HTTPValidationError
	JSON500      *HTTPError
}

type InvalidParamFormatError struct {
	ParamName string
	Err       error
}

type Media struct {
	Nsfw bool   `json:"nsfw"`
	Seed int    `json:"seed"`
	Url  string `json:"url"`
}

type MiddlewareFunc func(http.Handler) http.Handler

type OptimizationFlags map[string]EnvValue

type RequestEditorFn func(ctx context.Context, req *http.Request) error

type RequiredHeaderError struct {
	ParamName string
	Err       error
}

type RequiredParamError struct {
	ParamName string
}

type RunnerContainer struct {
	RunnerContainerConfig

	Client *ClientWithResponses
}

type RunnerContainerConfig struct {
	Type     RunnerContainerType
	Pipeline string
	ModelID  string
	Endpoint RunnerEndpoint

	// For managed containers only
	ID       string
	GPU      string
	KeepWarm bool
	// contains filtered or unexported fields
}

type RunnerContainerType int

type RunnerEndpoint struct {
	URL   string
	Token string
}

type ServerInterface interface {
	// Audio To Text
	// (POST /audio-to-text)
	AudioToText(w http.ResponseWriter, r *http.Request)
	// Health
	// (GET /health)
	Health(w http.ResponseWriter, r *http.Request)
	// Image To Image
	// (POST /image-to-image)
	ImageToImage(w http.ResponseWriter, r *http.Request)
	// Image To Video
	// (POST /image-to-video)
	ImageToVideo(w http.ResponseWriter, r *http.Request)
	// Text To Image
	// (POST /text-to-image)
	TextToImage(w http.ResponseWriter, r *http.Request)
	// Upscale
	// (POST /upscale)
	Upscale(w http.ResponseWriter, r *http.Request)
}

type ServerInterfaceWrapper struct {
	Handler            ServerInterface
	HandlerMiddlewares []MiddlewareFunc
	ErrorHandlerFunc   func(w http.ResponseWriter, r *http.Request, err error)
}

type TextResponse struct {
	Chunks []Chunk `json:"chunks"`
	Text   string  `json:"text"`
}

type TextToImageJSONRequestBody = TextToImageParams

type TextToImageParams struct {
	GuidanceScale      *float32 `json:"guidance_scale,omitempty"`
	Height             *int     `json:"height,omitempty"`
	ModelId            *string  `json:"model_id,omitempty"`
	NegativePrompt     *string  `json:"negative_prompt,omitempty"`
	NumImagesPerPrompt *int     `json:"num_images_per_prompt,omitempty"`
	NumInferenceSteps  *int     `json:"num_inference_steps,omitempty"`
	Prompt             string   `json:"prompt"`
	SafetyCheck        *bool    `json:"safety_check,omitempty"`
	Seed               *int     `json:"seed,omitempty"`
	Width              *int     `json:"width,omitempty"`
}

type TextToImageResponse struct {
	Body         []byte
	HTTPResponse *http.Response
	JSON200      *ImageResponse
	JSON400      *HTTPError
	JSON401      *HTTPError
	JSON422      *HTTPValidationError
	JSON500      *HTTPError
}

type TooManyValuesForParamError struct {
	ParamName string
	Count     int
}

type UnescapedCookieParamError struct {
	ParamName string
	Err       error
}

type Unimplemented struct{}

type UnmarshalingParamError struct {
	ParamName string
	Err       error
}

type UpscaleMultipartRequestBody = BodyUpscaleUpscalePost

type UpscaleResponse struct {
	Body         []byte
	HTTPResponse *http.Response
	JSON200      *ImageResponse
	JSON400      *HTTPError
	JSON401      *HTTPError
	JSON422      *HTTPValidationError
	JSON500      *HTTPError
}

type ValidationError struct {
	Loc  []ValidationError_Loc_Item `json:"loc"`
	Msg  string                     `json:"msg"`
	Type string                     `json:"type"`
}

type ValidationErrorLoc0 = string

type ValidationErrorLoc1 = int

type ValidationError_Loc_Item struct {
	// contains filtered or unexported fields
}

type VideoResponse struct {
	Frames [][]Media `json:"frames"`
}

type Worker struct {
	// contains filtered or unexported fields
}