26 lines
706 B
Go
26 lines
706 B
Go
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:"生成的答案"`
|
|
}
|