package dto

import (
	"gitea.com/red-future/common/beans"
	"github.com/gogf/gf/v2/frame/g"
	"github.com/gogf/gf/v2/os/gtime"
)

// CreateDatasetReq 创建数据集请求
type CreateDatasetReq struct {
	g.Meta `path:"/create" method:"post" tags:"知识库（数据集）管理" summary:"创建知识库（数据集）" dc:"创建知识库（数据集）"`

	Name        string `json:"name" v:"required#名称不能为空"`
	Description string `json:"description"`
}

// CreateDatasetRes 创建数据集响应
type CreateDatasetRes struct {
	Id int64 `json:"id,string"`
}

// UpdateDatasetReq 更新数据集请求
type UpdateDatasetReq struct {
	g.Meta `path:"/update" method:"put" tags:"知识库（数据集）管理" summary:"更新知识库（数据集）" dc:"更新知识库（数据集）"`

	Id            int64  `json:"id" v:"required#ID不能为空"`
	Name          string `json:"name"`
	Description   string `json:"description"`
	DocumentCount int64  `json:"documentCount"`
	DocumentSize  int64  `json:"documentSize"`
}

// DeleteDatasetReq 删除数据集请求
type DeleteDatasetReq struct {
	g.Meta `path:"/delete" method:"delete" tags:"知识库（数据集）管理" summary:"删除知识库（数据集）" dc:"删除知识库（数据集）"`

	Id int64 `json:"id" v:"required#ID不能为空"`
}

// GetDatasetReq 获取数据集请求
type GetDatasetReq struct {
	g.Meta `path:"/get" method:"get" tags:"知识库（数据集）管理" summary:"获取知识库（数据集）详情" dc:"获取知识库（数据集）详情"`

	Id int64 `json:"id" v:"required#ID不能为空"`
}

// ListDatasetReq 数据集列表请求
type ListDatasetReq struct {
	g.Meta `path:"/list" method:"get" tags:"知识库（数据集）管理" summary:"获取知识库（数据集）列表" dc:"分页查询知识库（数据集）列表，支持多条件筛选"`

	Page    *beans.Page `json:"page"`
	Ids     []int64     `json:"ids" dc:"数据集ID列表"`
	Keyword string      `json:"keyword" dc:"关键词搜索"`
}

// ListDatasetRes 数据集列表响应
type ListDatasetRes struct {
	List  []*DatasetVO `json:"list"`
	Total int          `json:"total"`
}

type DatasetVO struct {
	Id            int64       `json:"id,string" dc:"id"`
	Name          string      `json:"name" dc:"数据集名称"`
	Description   string      `json:"description" dc:"数据集描述"`
	DocumentCount int64       `json:"documentCount" dc:"文件数量"`
	DocumentSize  int64       `json:"documentSize" dc:"文件大小(字节)"`
	CreatedAt     *gtime.Time `json:"createdAt" dc:"创建时间"`
	UpdatedAt     *gtime.Time `json:"updatedAt" dc:"更新时间"`
}