package llm

import (
	"yunion.io/x/onecloud/pkg/apis"
)

type InstantModelListInput struct {
	apis.SharableVirtualResourceListInput
	apis.EnabledResourceBaseListInput

	ModelName string `json:"model_name"`
	ModelTag  string `json:"model_tag"`
	ModelId   string `json:"model_id"`
	Image     string `json:"image"`

	Mounts string `json:"mounts"`

	AutoCache *bool `json:"auto_cache"`
}

type InstantModelImportInput struct {
	ModelName string           `json:"model_name"`
	ModelTag  string           `json:"model_tag"`
	LlmType   LLMContainerType `json:"llm_type"`
}

type InstantModelCreateInput struct {
	apis.SharableVirtualResourceCreateInput
	apis.EnabledBaseResourceCreateInput

	LlmType   LLMContainerType `json:"llm_type"`
	ModelName string           `json:"model_name"`
	ModelTag  string           `json:"model_tag"`
	ImageId   string           `json:"image_id"`
	Size      int64            `json:"size"`
	ModelId   string           `json:"model_id"`

	ActualSizeMb int32 `json:"actual_size_mb"`

	Mounts []string `json:"mounts"`

	DoNotImport *bool `json:"do_not_import,omitempty"`
}

type InstantModelUpdateInput struct {
	apis.SharableVirtualResourceBaseUpdateInput

	ImageId string `json:"image_id"`
	Size    int64  `json:"size"`

	ActualSizeMb int32 `json:"actual_size_mb"`

	Mounts []string `json:"mounts"`
}

type InstantModelDetails struct {
	apis.SharableVirtualResourceDetails

	Image string `json:"image"`

	CacheCount  int `json:"cache_count"`
	CachedCount int `json:"cached_count"`

	IconBase64 string `json:"icon_base64"`

	MountedByLLMs []MountedByLLMInfo `json:"mounted_by_llms"`

	GPUMemoryRequired int64 `json:"gpu_memory_required"`
}

type MountedByLLMInfo struct {
	LlmId   string `json:"llm_id"`
	LlmName string `json:"llm_name"`
}

type InstantModelSyncstatusInput struct {
}

type InstantAppCacheInput struct {
}

type InstantModelEnableAutoCacheInput struct {
	AutoCache bool `json:"auto_cache"`
}

type MountedModelResourceListInput struct {
	MountedModels []string `json:"mounted_models"`
}

type MountedModelResourceCreateInput struct {
	MountedModels []string `json:"mounted_models"`
}

type MountedModelResourceUpdateInput struct {
	MountedModels []string `json:"mounted_models"`
}