mlogclub
99 lines
3.1 KiB
Go
99 lines
3.1 KiB
Go
package request
|
|
|
|
import (
|
|
"io"
|
|
|
|
"agent-desk/internal/pkg/enums"
|
|
)
|
|
|
|
type CreateKnowledgeBaseRequest struct {
|
|
Name string `json:"name"`
|
|
Description string `json:"description"`
|
|
KnowledgeType string `json:"knowledgeType"`
|
|
DefaultTopK int `json:"defaultTopK"`
|
|
DefaultScoreThreshold float64 `json:"defaultScoreThreshold"`
|
|
DefaultRerankLimit int `json:"defaultRerankLimit"`
|
|
ChunkProvider string `json:"chunkProvider"`
|
|
ChunkTargetTokens int `json:"chunkTargetTokens"`
|
|
ChunkMaxTokens int `json:"chunkMaxTokens"`
|
|
ChunkOverlapTokens int `json:"chunkOverlapTokens"`
|
|
AnswerMode int `json:"answerMode"`
|
|
Remark string `json:"remark"`
|
|
}
|
|
|
|
type UpdateKnowledgeBaseRequest struct {
|
|
ID int64 `json:"id"`
|
|
CreateKnowledgeBaseRequest
|
|
}
|
|
|
|
type CreateKnowledgeDocumentRequest struct {
|
|
KnowledgeBaseID int64 `json:"knowledgeBaseId"`
|
|
Title string `json:"title"`
|
|
ContentType enums.KnowledgeDocumentContentType `json:"contentType"`
|
|
Content string `json:"content"`
|
|
}
|
|
|
|
type UpdateKnowledgeDocumentRequest struct {
|
|
ID int64 `json:"id"`
|
|
CreateKnowledgeDocumentRequest
|
|
}
|
|
|
|
type CreateKnowledgeFAQRequest struct {
|
|
KnowledgeBaseID int64 `json:"knowledgeBaseId"`
|
|
Question string `json:"question"`
|
|
Answer string `json:"answer"`
|
|
SimilarQuestions []string `json:"similarQuestions"`
|
|
Remark string `json:"remark"`
|
|
}
|
|
|
|
type UpdateKnowledgeFAQRequest struct {
|
|
ID int64 `json:"id"`
|
|
CreateKnowledgeFAQRequest
|
|
}
|
|
|
|
type KnowledgeFAQImportMode string
|
|
|
|
const (
|
|
KnowledgeFAQImportModeAppend KnowledgeFAQImportMode = "append"
|
|
KnowledgeFAQImportModeOverwrite KnowledgeFAQImportMode = "overwrite"
|
|
)
|
|
|
|
type ImportKnowledgeFAQRequest struct {
|
|
KnowledgeBaseID int64
|
|
Mode KnowledgeFAQImportMode
|
|
Filename string
|
|
Reader io.Reader
|
|
}
|
|
|
|
type KnowledgeSearchRequest struct {
|
|
KnowledgeBaseIDs []int64 `json:"knowledgeBaseIds"`
|
|
Question string `json:"question"`
|
|
TopK int `json:"topK"`
|
|
ScoreThreshold float64 `json:"scoreThreshold"`
|
|
RerankLimit int `json:"rerankLimit"`
|
|
Channel string `json:"channel"`
|
|
Scene string `json:"scene"`
|
|
SessionID string `json:"sessionId"`
|
|
ConversationID int64 `json:"conversationId"`
|
|
}
|
|
|
|
type KnowledgeAnswerRequest struct {
|
|
KnowledgeBaseIDs []int64 `json:"knowledgeBaseIds"`
|
|
Question string `json:"question"`
|
|
TopK int `json:"topK"`
|
|
ScoreThreshold float64 `json:"scoreThreshold"`
|
|
RerankLimit int `json:"rerankLimit"`
|
|
Channel string `json:"channel"`
|
|
Scene string `json:"scene"`
|
|
SessionID string `json:"sessionId"`
|
|
ConversationID int64 `json:"conversationId"`
|
|
AnswerMode int `json:"answerMode"`
|
|
}
|
|
|
|
type CreateKnowledgeFeedbackRequest struct {
|
|
RetrieveLogID int64 `json:"retrieveLogId"`
|
|
FeedbackType int `json:"feedbackType"`
|
|
FeedbackReason string `json:"feedbackReason"`
|
|
Remark string `json:"remark"`
|
|
}
|