maas

type CompletionResponse struct {
	MaasId    int64                  `json:"maas_id" binding:"required"`
	Prompt    string                 `json:"prompt" binding:"required"`
	Params    map[string]interface{} `json:"params" binding:"required"`
	SessionId string                 `json:"session_id" binding:"required"`
	Context   string                 `json:"context" binding:"required"`
}

type IMaasService interface {
	GetMaasList(userId int64) ([]*MaasBaseInfo, error)
	GetMaasDetail(id int64) *result.ErrorResult
	ImportMaas(model *MaasRequest, userId int64) (*Maas, error)
	UpdateMaas(id int, model *Maas) *result.ErrorResult
	DeleteMaas(id int) *result.ErrorResult
	InvokeMaas(ctx context.Context, req LLMRequest) (<-chan string, error)
	Completion(ctx *gin.Context, req CompletionResponse, userId int64)
}

type IModelRepo interface {
	FindMaasList(userId int64) ([]*Maas, error)
	FindMaasById(id int64, model *Maas) (*Maas, error)
	FindMaasByModel(model string) (*Maas, error)
	ImportMaas(model *Maas) error
	UpdateMaas(id int, model *Maas) error
	DeleteMaas(id int) error
	CheckDuplicateMaasBaseURL(ctx context.Context, model string, baseURL string) (bool, error)
}

type LLMRequest struct {
	ModelId int64                  `json:"model_id" binding:"required"`
	Prompt  string                 `json:"prompt" binding:"required"`
	Params  map[string]interface{} `json:"params" binding:"required"`
}

type Maas struct {
	ID          int64      `json:"id" gorm:"column:id; primaryKey; autoIncrement"`
	Name        string     `json:"name" gorm:"not null"`
	Description string     `json:"description"`
	Status      MaasStatus `json:"status" gorm:"type:ENUM('active', 'inactive', 'deprecated');default:'active'"`
	Model       string     `json:"model" gorm:"column:model;unique" binding:"required"`
	BaseUrl     string     `json:"base_url" gorm:"column:base_url;not null" binding:"required"`
	ApiKey      string     `json:"api_key" gorm:"column:api_key;not null" binding:"required"`
	ModelType   string     `json:"model_type" gorm:"column:model_type;not null" binding:"required"`
	Avatar      string     `json:"avatar" gorm:"column:avatar"`
	// 自动维护时间
	CreateTime time.Time `json:"create_time" gorm:"column:create_time;autoCreateTime;type:datetime(0);"`
	UpdateTime time.Time `json:"update_time" gorm:"column:update_time;autoCreateTime;<-:false;type:datetime(0);"`
	CreateBy   int64     `json:"create_by" gorm:"not null;default:0"`
	UpdateBy   int64     `json:"-" gorm:"column:update_by"`
}

type MaasAttrFunc func(u *Maas)

type MaasAttrFuncs []MaasAttrFunc

type MaasBaseInfo struct {
	Id       int64     `json:"id"`
	Model    string    `json:"model"`
	Name     string    `json:"name"`
	Avatar   string    `json:"avatar"`
	CreateAt time.Time `json:"create_time"`
}

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

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

type MaasRequest struct {
	Name        string `json:"name" `
	Description string `json:"description"`
	Model       string `json:"maas" binding:"required"`
	ModelType   string `json:"model_type" binding:"required"`
	BaseURL     string `json:"base_url" binding:"required,url"`
	APIKey      string `json:"api_key" binding:"required"`
	Avatar      string `json:"avatar"`
}

type MaasStatus string

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