package dto

import (
	"github.com/gogf/gf/v2/frame/g"
)

// RAGQueryReq RAG查询请求
type RAGQueryReq struct {
	g.Meta `path:"/ragQuery" method:"post" tags:"RAG查询" summary:"执行RAG查询" dc:"执行RAG查询"`

	Content    string     `json:"content" v:"required#查询内容不能为空" dc:"用户问题"`
	DatasetIds []int64    `json:"datasetIds" dc:"数据集ID"`
	History    []*Message `json:"history" dc:"历史对话"`
	TopK       int        `json:"topK" d:"5" dc:"检索topK，默认5"`
}

type Message struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

// RAGQueryRes RAG查询响应
type RAGQueryRes struct {
	Answer string `json:"answer" dc:"生成的答案"`
}