app

type AIDetectorCmd struct {
	User types.Account         `json:"user"`
	Lang domain.Lang           `json:"lang"`
	Text domain.AIDetectorText `json:"text"`
}

type ApiApplyRecordDTO struct {
	User      string `json:"user"`
	Name      string `json:"name"`
	Endpoint  string `json:"endpoint"`
	ApplyAt   string `json:"apply_at"`
	UpdateAt  string `json:"update_at"`
	ModelName string `json:"model_name"`
	Enabled   bool   `json:"enabled"`
}

type ApiInfoDTO struct {
	Id       string `json:"id"`
	Name     string `json:"name"`
	Endpoint string `json:"endpoint"`
	Doc      string `json:"doc"`
}

type ApplyRecordGetCmd struct {
	User      types.Account
	ModelName domain.ModelName
}

type AsyncBigModelService interface {
	WuKong(uint64, types.Account, *WuKongCmd) error
	GetIdleEndpoint(bid string) (int, error)
}

type BaiChuanCmd struct {
	User              types.Account
	Sampling          bool
	Text              domain.BaiChuanText
	TopK              domain.TopK
	TopP              domain.TopP
	Temperature       domain.Temperature
	RepetitionPenalty domain.RepetitionPenalty
}

type BaiChuanDTO struct {
	Text string `json:"text"`
}

type BigModelService interface {
	// luojia
	LuoJiaUploadPicture(io.Reader, types.Account) error
	LuoJia(types.Account) (string, error)
	ListLuoJiaRecord(types.Account) ([]LuoJiaRecordDTO, error)

	// wukong
	GenWuKongSamples(int) ([]string, error)
	WuKong(types.Account, *WuKongCmd) (map[string]string, string, error)
	WuKongHF(*WuKongHFCmd) (map[string]string, string, error)
	WuKongInferenceAsync(types.Account, *WuKongCmd) (string, error)
	GetWuKongWaitingTaskRank(types.Account) (WuKongRankDTO, error)
	GetWuKongLastTaskResp(types.Account) ([]wukongPictureDTO, string, error)
	AddLikeFromTempPicture(*WuKongAddLikeFromTempCmd) (string, string, error)
	AddLikeFromPublicPicture(*WuKongAddLikeFromPublicCmd) (string, string, error)
	AddPublicFromTempPicture(*WuKongAddPublicFromTempCmd) (string, string, error)
	AddPublicFromLikePicture(*WuKongAddPublicFromLikeCmd) (string, string, error)
	CancelPublic(types.Account, string) error
	GetPublicsGlobal(cmd *WuKongListPublicGlobalCmd) (WuKongPublicGlobalDTO, error)
	ListPublics(types.Account) ([]WuKongPublicDTO, error)
	CancelLike(types.Account, string) error
	ListLikes(types.Account) ([]WuKongLikeDTO, error)
	DiggPicture(*WuKongAddDiggCmd) (int, error)
	CancelDiggPicture(*WuKongCancelDiggCmd) (int, error)
	ReGenerateDownloadURL(types.Account, string) (string, string, error)

	//api service
	ApplyApi(types.Account, domain.ModelName, string) error
	WukongApi(types.Account, domain.ModelName, *WuKongApiCmd) (map[string]string, string, error)
	GetApplyRecordByModel(types.Account, domain.ModelName) (string, error)
	GetApplyRecordByUser(types.Account) ([]ApiApplyRecordDTO, error)
	IsApplyModel(types.Account, domain.ModelName) (bool, error)
	UpdateToken(types.Account, domain.ModelName, string) (string, error)

	//api info
	GetApiInfo(model domain.ModelName) (ApiInfoDTO, error)

	// ai detector
	AIDetector(*AIDetectorCmd) (string, bool, error)

	// baichuan
	BaiChuan(*BaiChuanCmd) (string, BaiChuanDTO, error)

	// glm2
	GLM2(*GLM2Cmd) (string, error)

	// llama2
	LLAMA2(*LLAMA2Cmd) (string, error)

	// iflytekspark
	IFlytekSpark(*IFlytekSparkCmd) (string, error)

	// skywork 13b
	SkyWork(*SkyWorkCmd) (string, error)
}

type CodeGeexCmd bigmodel.CodeGeexReq

type CodeGeexDTO = bigmodel.CodeGeexResp

type DescribePictureCmd struct {
	User    types.Account
	Picture io.Reader
	Name    string
	Length  int64
}

type GLM2Cmd struct {
	CH                chan string
	User              types.Account
	History           []domain.History
	Sampling          bool
	Text              domain.GLM2Text
	TopK              domain.TopK
	TopP              domain.TopP
	Temperature       domain.Temperature
	RepetitionPenalty domain.RepetitionPenalty
}

type GLM2DTO struct {
	Reply        string `json:"reply"`
	StreamStatus string `json:"stream_status"`
}

type GenPictureCmd struct {
	User types.Account `json:"user"`
	Desc domain.Desc   `json:"desc"`
}

type IFlytekSparkCmd struct {
	CH                chan string
	User              types.Account
	Sampling          bool
	Text              domain.IFlytekSparkText
	TopK              domain.TopK
	Temperature       domain.Temperature
	RepetitionPenalty domain.RepetitionPenalty
}

type IFlytekSparkDTO struct {
	Reply        string `json:"reply"`
	StreamStatus string `json:"stream_status"`
}

type LLAMA2Cmd struct {
	CH                chan string
	User              types.Account
	History           []domain.History
	Sampling          bool
	Text              domain.LLAMA2Text
	TopK              domain.TopK
	TopP              domain.TopP
	Temperature       domain.Temperature
	RepetitionPenalty domain.RepetitionPenalty
}

type LLAMA2DTO struct {
	Reply        string `json:"reply"`
	StreamStatus string `json:"stream_status"`
}

type LuoJiaHFCmd struct {
	User    userdomain.Account
	Picture io.Reader
}

type LuoJiaRecordDTO struct {
	CreatedAt string `json:"created_at"`
	Id        string `json:"id"`
}

type SkyWorkCmd struct {
	CH                chan string
	User              types.Account
	History           []domain.History
	Sampling          bool
	Text              domain.SkyWorkText
	TopK              domain.TopK
	TopP              domain.TopP
	Temperature       domain.Temperature
	RepetitionPenalty domain.RepetitionPenalty
}

type SkyWorkDTO struct {
	Reply        string `json:"reply"`
	StreamStatus string `json:"stream_status"`
}

type VQAHFCmd struct {
	User    types.Account
	Picture io.Reader
	Ask     string
}

type WuKongAddDiggCmd struct {
	User  types.Account
	Owner types.Account
	Id    string
}

type WuKongAddLikeFromPublicCmd struct {
	User  types.Account
	Owner types.Account
	Id    string
}

type WuKongAddLikeFromTempCmd struct {
	User    types.Account
	OBSPath domain.OBSPath
}

type WuKongAddPublicFromLikeCmd struct {
	User types.Account
	Id   string
}

type WuKongAddPublicFromTempCmd = WuKongAddLikeFromTempCmd

type WuKongApiCmd struct {
	WuKongCmd

	EndPointType string
	User         types.Account
}

type WuKongCancelDiggCmd WuKongAddDiggCmd

type WuKongCmd struct {
	domain.WuKongPictureMeta

	EsType string
}

type WuKongHFCmd struct {
	WuKongCmd

	EndPointType string
	User         types.Account
}

type WuKongIsLikeDTO struct {
	IsLike bool
	LikeID string
}

type WuKongLikeDTO struct {
	IsPublic bool   `json:"is_public"`
	Avatar   string `json:"avatar"`

	WuKongPictureBaseDTO
}

type WuKongListPublicGlobalCmd struct {
	User  types.Account
	Level domain.WuKongPictureLevel
	WuKongPictureListOption
}

type WuKongPictureBaseDTO struct {
	Id        string `json:"id"`
	Owner     string `json:"owner"` // owner of picture
	Desc      string `json:"desc"`
	Style     string `json:"style"`
	Link      string `json:"link"`
	CreatedAt string `json:"created_at"`
}

type WuKongPictureListOption struct {
	CountPerPage int
	PageNum      int
}

type WuKongPicturesListCmd = repository.WuKongPictureListOption

type WuKongPublicDTO struct {
	Avatar    string `json:"avatar"`
	IsLike    bool   `json:"is_like"`
	LikeID    string `json:"like_id"`
	IsDigg    bool   `json:"is_digg"`
	DiggCount int    `json:"digg_count"`

	WuKongPictureBaseDTO
}

type WuKongPublicGlobalDTO struct {
	Pictures []WuKongPublicDTO `json:"pictures"`
	Total    int               `json:"total"`
}

type WuKongRankDTO struct {
	Rank int `json:"rank"`
}