comfyUIclient

type Client struct {
	ID string
	// contains filtered or unexported fields
}

type DataOutputFile struct {
	Filename  string `json:"filename"`
	SubFolder string `json:"subfolder"`
	Type      string `json:"type"`
}

type EndPoint struct {
	Protocol string
	Address  string
	Port     string
}

type GPU struct {
	Name           string `json:"name"`
	Type           string `json:"type"`
	Index          int    `json:"index"`
	VRAMTotal      int64  `json:"vram_total"`
	VRAMFree       int64  `json:"vram_free"`
	TorchVRAMTotal int64  `json:"torch_vram_total"`
	TorchVRAMFree  int64  `json:"torch_vram_free"`
}

type Handler interface {
	Handle(string) error
}

type ImageType string

type NodeInfo struct {
	Num           uint64
	PromptID      string
	Prompt        map[string]PromptNode `json:"prompt"`
	ExtraData     json.RawMessage       // extra data is just for user's custom data
	OutputNodeIDs []string
}

type NodeObject struct {
	Input        *NodeObjectInput `json:"input"`
	Output       []string         `json:"output"`
	OutputIsList []bool           `json:"output_is_list"`
	OutputName   []string         `json:"output_name"`
	Name         string           `json:"name"`
	DisplayName  string           `json:"display_name"`
	Description  string           `json:"description"`
	Category     string           `json:"category"`
	OutputNode   bool             `json:"output_node"`
}

type NodeObjectInput struct {
	Required map[string]interface{} `json:"required"`
	Optional map[string]interface{} `json:"optional,omitempty"`
}

type PromptHistoryItem struct {
	PromptID string
	PromptHistoryMember
}

type PromptHistoryMember struct {
	NodeInfo *NodeInfo                            `json:"prompt"`
	Outputs  map[string]PromptHistoryMemberImages `json:"outputs"`
}

type PromptHistoryMemberImages struct {
	Images *[]DataOutputFile `json:"images"`
}

type PromptNode struct {
	Inputs    map[string]interface{} `json:"inputs"`
	ClassType string                 `json:"class_type"`
}

type QueueExecInfo struct {
	ExecInfo struct {
		QueueRemaining uint64 `json:"queue_remaining"`
	} `json:"exec_info"`
}

type QueueInfo struct {
	QueueRunning []*NodeInfo `json:"queue_running"`
	QueuePending []*NodeInfo `json:"queue_pending"`
}

type QueuePromptResp struct {
	PromptID   string                 `json:"prompt_id"`
	Number     int                    `json:"number"`
	NodeErrors map[string]interface{} `json:"node_errors"`
}

type Router string

type System struct {
	OS             string `json:"os"`
	PythonVersion  string `json:"python_version"`
	EmbeddedPython bool   `json:"embedded_python"`
}

type SystemStats struct {
	System  *System `json:"system"`
	Devices []*GPU  `json:"devices"`
}

type TaskStatusType = WsMessageType

type UploadFile struct {
	Filename  string `json:"name"`
	SubFolder string `json:"subfolder"`
	Type      string `json:"type"`
}

type WSMessage struct {
	Type WsMessageType `json:"type"`
	Data interface{}   `json:"data"`
}

type WSMessageDataExecuted struct {
	Node     string `json:"node"`
	PromptID string `json:"prompt_id"`
	Output   map[string][]*DataOutputFile
}

type WSMessageDataExecuting struct {
	Node     string `json:"node"`
	PromptID string `json:"prompt_id"`
}

type WSMessageDataExecutionCached struct {
	Nodes    []string `json:"nodes"`
	PromptID string   `json:"prompt_id"`
}

type WSMessageDataExecutionStart struct {
	PromptID string `json:"prompt_id"`
}

type WSMessageDataProgress struct {
	Value int `json:"value"`
	Max   int `json:"max"`
}

type WSMessageDataStatus struct {
	Status struct {
		ExecInfo struct {
			QueueRemaining int `json:"queue_remaining"`
		} `json:"exec_info"`
	} `json:"status"`
	SID string `json:"sid"`
}

type WSMessageExecutionError struct {
	PromptID         string                 `json:"prompt_id"`
	Node             string                 `json:"node_id"`
	NodeType         string                 `json:"node_type"`
	Executed         []string               `json:"executed"`
	ExceptionMessage string                 `json:"exception_message"`
	ExceptionType    string                 `json:"exception_type"`
	Traceback        []string               `json:"traceback"`
	CurrentInputs    map[string]interface{} `json:"current_inputs"`
	CurrentOutputs   map[int]interface{}    `json:"current_outputs"`
}

type WSMessageExecutionInterrupted struct {
	PromptID string   `json:"prompt_id"`
	NodeID   string   `json:"node_id"`
	NodeType string   `json:"node_type"`
	Executed []string `json:"executed"`
}

type WebSocketConnection struct {
	URL  string
	Conn *websocket.Conn

	MaxRetry int
	// contains filtered or unexported fields
}

type WsMessageType string