notebook

type AscendInfo struct {
	Npu       int    `json:"npu"`
	NpuMemory string `json:"npu_memory"`
	Type      string `json:"type"`
}

type BillingInfo struct {
	Code    string `json:"code"`
	UnitNum int    `json:"unit_num"`
}

type CreateOpts struct {
	Name        string         `json:"name" required:"true"`
	Flavor      string         `json:"flavor" required:"true"`
	ImageId     string         `json:"image_id" required:"true"`
	Volume      VolumeReq      `json:"volume" required:"true"`
	Description string         `json:"description,omitempty"`
	Duration    *int           `json:"duration,omitempty"`
	Endpoints   []EndpointsReq `json:"endpoints,omitempty"`
	Feature     string         `json:"feature,omitempty"`
	PoolId      string         `json:"pool_id,omitempty"`
	WorkspaceId string         `json:"workspace_id,omitempty"`
}

type Endpoints struct {
	AllowedAccessIps []string `json:"allowed_access_ips"`
	Service          string   `json:"service"`
	KeyPairNames     []string `json:"key_pair_names"`
	Uri              string   `json:"uri"`
}

type EndpointsReq struct {
	AllowedAccessIps []string `json:"allowed_access_ips,omitempty"`
	Service          string   `json:"service,omitempty"`
	KeyPairNames     []string `json:"key_pair_names,omitempty"`
}

type Flavor struct {
	Arch        string      `json:"arch"`
	Ascend      AscendInfo  `json:"ascend"`
	Billing     BillingInfo `json:"billing"`
	Category    string      `json:"category"`
	Description string      `json:"description"`
	Feature     string      `json:"feature"`
	Free        bool        `json:"free"`
	Gpu         GPUInfo     `json:"gpu"`
	Id          string      `json:"id"`
	Memory      int         `json:"memory"`
	Name        string      `json:"name"`
	SoldOut     bool        `json:"sold_out"`
	Storages    []string    `json:"storages"`
	Vcpus       int         `json:"vcpus"`
}

type GPUInfo struct {
	Gpu       int    `json:"gpu"`
	GpuMemory string `json:"gpu_memory"`
	Type      string `json:"type"`
}

type Image struct {
	Id      string `json:"id"`
	Name    string `json:"name"`
	SwrPath string `json:"swr_path"`
	Type    string `json:"type"`
}

type ImageDetail struct {
	Id          string `json:"id"`
	Name        string `json:"name"`
	SwrPath     string `json:"swr_path"`
	Type        string `json:"type"`
	Description string `json:"description"`
	Arch        string `json:"arch"`
}

type ImagePage struct {
	pagination.OffsetPageBase
}

type ImagesResp struct {
	Current int           `json:"current"`
	Data    []ImageDetail `json:"data"`
	Pages   int           `json:"pages"`
	Size    int           `json:"size"`
	Total   int           `json:"total"`
}

type JobProgress struct {
	NotebookId      string `json:"notebook_id"`
	Status          string `json:"status"`
	Step            int    `json:"step"`
	StepDescription string `json:"step_description"`
}

type Lease struct {
	CreateAt int  `json:"create_at"`
	Duration int  `json:"duration"`
	Enable   bool `json:"enable"`
	UpdateAt int  `json:"update_at"`
}

type ListImageOpts struct {
	Limit       int    `q:"limit,omitempty"`
	Name        string `q:"name,omitempty"`
	Namespace   string `q:"namespace,omitempty"`
	Offset      int    `q:"offset,omitempty"`
	SortDir     string `q:"sort_dir,omitempty"`
	SortKey     string `q:"sort_key,omitempty"`
	Type        string `q:"type,omitempty"`
	WorkspaceId string `q:"workspace_id,omitempty"`
}

type ListNotebooks struct {
	Current int        `json:"current"`
	Data    []Notebook `json:"data"`
	Pages   int        `json:"pages"`
	Size    int        `json:"size"`
	Total   int        `json:"total"`
}

type ListOpts struct {
	Feature     string `q:"feature,omitempty"`
	Limit       int    `q:"limit,omitempty"`
	Name        string `q:"name,omitempty"`
	Offset      int    `q:"offset,omitempty"`
	Owner       string `q:"owner,omitempty"`
	SortDir     string `q:"sort_dir,omitempty"`
	SortKey     string `q:"sort_key,omitempty"`
	Status      string `q:"status,omitempty"`
	WorkspaceId string `q:"workspaceId,omitempty"`
}

type ModelartsError struct {
	ErrorCode string `json:"error_code"`
	ErrorMsg  string `json:"error_msg"`
}

type MountStorage struct {
	Category  string `json:"category"`
	Id        string `json:"id"`
	MountPath string `json:"mount_path"`
	Status    string `json:"status"`
	Uri       string `json:"uri"`
}

type MountStorageListResp struct {
	Current int            `json:"current"`
	Data    []MountStorage `json:"data"`
	Pages   int            `json:"pages"`
	Size    int            `json:"size"`
	Total   int            `json:"total"`
}

type MountStorageOpts struct {
	Category  string `json:"category"`
	MountPath string `json:"mount_path"`
	Uri       string `json:"uri"`
}

type Notebook struct {
	ActionProgress []JobProgress `json:"action_progress"`
	Description    string        `json:"description"`
	Endpoints      []Endpoints   `json:"endpoints"`
	FailReason     string        `json:"fail_reason"`
	Feature        string        `json:"feature"`
	Flavor         string        `json:"flavor"`
	Id             string        `json:"id"`
	Image          Image         `json:"image"`
	Lease          Lease         `json:"lease"`
	Name           string        `json:"name"`
	Pool           PoolRes       `json:"pool"`
	Status         string        `json:"status"`
	Token          string        `json:"token"`
	Url            string        `json:"url"`
	Volume         VolumeRes     `json:"volume"`
	WorkspaceId    string        `json:"workspace_id"`
	CreateAt       int           `json:"create_at"`
	UpdateAt       int           `json:"update_at"`
}

type PoolRes struct {
	Id   string `json:"id"`
	Name string `json:"name"`
}

type UpdateOpts struct {
	Description    *string        `json:"description,omitempty"`
	Endpoints      []EndpointsReq `json:"endpoints,omitempty"`
	Flavor         string         `json:"flavor,omitempty"`
	ImageId        string         `json:"image_id,omitempty"`
	Name           string         `json:"name,omitempty"`
	StorageNewSize *int           `json:"storage_new_size,omitempty"`
}

type VolumeReq struct {
	Category  string `json:"category" required:"true"`
	Ownership string `json:"ownership" required:"true"`
	Capacity  *int   `json:"capacity,omitempty"`
	Uri       string `json:"uri,omitempty"`
}

type VolumeRes struct {
	Capacity  int    `json:"capacity"`
	Category  string `json:"category"`
	MountPath string `json:"mount_path"`
	Ownership string `json:"ownership"`
	Status    string `json:"status"`
}