package model

import (
	"synoth.com/janus/internal/core"
	"synoth.com/janus/internal/enum"
)

// Type 模型类型
type Type string

// JanusModel 供应商下可用的AI模型
type JanusModel struct {
	core.BaseModel
	// ProviderID 供应商ID
	ProviderID int64 `gorm:"not null" json:"provider_id"`
	// Name 模型名称
	Name string `gorm:"not null" json:"name"`
	// 模型类型
	ModelType Type `gorm:"not null" json:"model_type"`
	// 模型状态
	Status enum.Status `gorm:"default:1;not null" json:"status"`
}

const (
	Chat       Type = "chat"
	Completion Type = "completion"
	Embedding  Type = "embedding"
	Image      Type = "image"
)