service

type ChunkSearchList struct {
	Title    string      `json:"title"`
	Snippet  string      `json:"snippet"`
	KbName   string      `json:"kb_name"`
	MetaData interface{} `json:"meta_data"`
}

type ContentListResp struct {
	List          []FileSplitContent `json:"content_list"`
	ChunkTotalNum int                `json:"chunk_total_num"`
}

type ContentMetaData struct {
	FileName        string `json:"file_name"`
	ChunkLen        int    `json:"chunk_len"`
	ChunkCurrentNum int    `json:"chunk_current_num"`
	ChunkTotalNum   int    `json:"chunk_total_num"`
}

type DocSegmentStatusUpdateAllParams struct {
	DocSegmentStatusUpdateParams
	All bool `json:"on_off_switch"`
}

type DocSegmentStatusUpdateParams struct {
	UserId        string `json:"userId"`
	KnowledgeName string `json:"knowledgeBase"`
	FileName      string `json:"fileName"`
	ContentId     string `json:"content_id"`
	Status        bool   `json:"status"`
}

type DocUrlParams struct {
	Url string `json:"url"`
}

type DocUrlResp struct {
	Url          string        `json:"url"`
	OldName      string        `json:"old_name"`
	FileName     string        `json:"file_name"`
	FileSize     float64       `json:"file_size"`
	ResponseInfo RagCommonResp `json:"response_info"`
}

type DocUrlRespSafeArray struct {
	// contains filtered or unexported fields
}

type FileSplitContent struct {
	Content   string          `json:"content"`
	Order     int             `json:"order"`
	Status    bool            `json:"status"`
	MetaData  ContentMetaData `json:"meta_data"`
	ContentId string          `json:"content_id"`
	UserId    string          `json:"userId"`
	KbName    string          `json:"kb_name"`
	FileName  string          `json:"file_name"`
	Labels    []string        `json:"labels"`
}

type KnowledgeHitData struct {
	Prompt     string             `json:"prompt"`
	SearchList []*ChunkSearchList `json:"searchList"`
	Score      []float64          `json:"score"`
}

type KnowledgeHitParams struct {
	UserId               string                `json:"userId"`
	Question             string                `json:"question" validate:"required"`
	KnowledgeBase        []string              `json:"knowledgeBase" validate:"required"`
	Threshold            float64               `json:"threshold"`
	TopK                 int32                 `json:"topK"`
	RerankModelId        string                `json:"rerank_model_id"`               // rerankId
	RerankMod            string                `json:"rerank_mod"`                    // rerank_model:重排序模式，weighted_score：权重搜索
	RetrieveMethod       string                `json:"retrieve_method"`               // hybrid_search:混合搜索， semantic_search:向量搜索， full_text_search：文本搜索
	Weight               *WeightParams         `json:"weights"`                       // 权重搜索下的权重配置
	TermWeight           float32               `json:"term_weight_coefficient"`       // 关键词系数
	MetaFilter           bool                  `json:"metadata_filtering"`            // 元数据过滤开关
	MetaFilterConditions []*MetadataFilterItem `json:"metadata_filtering_conditions"` // 元数据过滤条件
}

type MetaData struct {
	Key       string      `json:"key"`
	Value     interface{} `json:"value"`
	ValueType string      `json:"value_type"`
}

type MetaItem struct {
	MetaName           string      `json:"meta_name"`           // 元数据名称
	MetaType           string      `json:"meta_type"`           // 元数据类型
	ComparisonOperator string      `json:"comparison_operator"` // 比较运算符
	Value              interface{} `json:"value,omitempty"`     // 用于过滤的条件值
}

type MetadataFilterItem struct {
	FilterKnowledgeName string      `json:"filtering_kb_name"`
	LogicalOperator     string      `json:"logical_operator"`
	Conditions          []*MetaItem `json:"conditions"`
}

type NewChunkItem struct {
	Content string   `json:"content"`
	Labels  []string `json:"labels"`
}

type RagCommonResp struct {
	Code    int    `json:"code"`
	Message string `json:"message"`
}

type RagCreateDocSegmentParams struct {
	UserId          string          `json:"userId"`            // 发起请求的用户ID
	KnowledgeBase   string          `json:"knowledgeBase"`     // 知识库的名称
	KnowledgeId     string          `json:"kb_id"`             // 知识库的唯一ID
	FileName        string          `json:"fileName"`          // 与chunk关联的文件名
	MaxSentenceSize int             `json:"max_sentence_size"` // 最大分段长度限制
	Chunks          []*NewChunkItem `json:"chunks"`            // 分段数据列表
}

type RagCreateParams struct {
	UserId           string `json:"userId"`
	Name             string `json:"knowledgeBase"`
	KnowledgeBaseId  string `json:"kb_id"`
	EmbeddingModelId string `json:"embedding_model_id"`
}

type RagDeleteDocParams struct {
	UserId        string `json:"userId"`
	KnowledgeBase string `json:"knowledgeBase"`
	FileName      string `json:"fileName"`
}

type RagDeleteDocSegmentParams struct {
	UserId        string   `json:"userId"`        // 发起请求的用户ID
	KnowledgeBase string   `json:"knowledgeBase"` // 知识库的名称
	KnowledgeId   string   `json:"kb_id"`         // 知识库的唯一ID
	FileName      string   `json:"fileName"`      // 与chunk关联的文件名
	ChunkIds      []string `json:"chunk_ids"`     // 分段数据列表
}

type RagDeleteParams struct {
	UserId            string `json:"userId"`
	KnowledgeBaseName string `json:"knowledgeBase"`
}

type RagDocMetaParams struct {
	UserId        string      `json:"userId"`
	KnowledgeBase string      `json:"knowledgeBase"`
	FileName      string      `json:"fileName"`
	MetaList      []*MetaData `json:"tags"`
}

type RagDocSegmentLabelsParams struct {
	UserId        string   `json:"userId"`        // 发起请求的用户ID
	KnowledgeBase string   `json:"knowledgeBase"` // 知识库的名称
	KnowledgeId   string   `json:"kb_id"`         // 知识库的唯一ID
	FileName      string   `json:"fileName"`      // 与chunk关联的文件名
	ContentId     string   `json:"chunk_id"`      // 要更新标签的chunk的唯一ID
	Labels        []string `json:"labels"`        // 需要为该chunk关联的标签列表
}

type RagDocSegmentResp struct {
	Code    int           `json:"code"`
	Message string        `json:"message"`
	Data    SegmentResult `json:"data"`
}

type RagGetDocSegmentParams struct {
	UserId            string `json:"userId"`
	KnowledgeBaseName string `json:"knowledgeBase"`
	FileName          string `json:"fileName"`
	PageSize          int32  `json:"page_size"`
	SearchAfter       int32  `json:"search_after"`
}

type RagGetDocSegmentResp struct {
	RagCommonResp
	Data *ContentListResp `json:"data"`
}

type RagImportDocParams struct {
	DocId             string               `json:"id"`         //文档id
	KnowledgeName     string               `json:"categoryId"` //知识库名称
	CategoryId        string               `json:"kb_id"`      //知识库id
	IsEnhanced        string               `json:"is_enhanced"`
	UserId            string               `json:"userId"`
	Overlap           float32              `json:"overlap" `
	ObjectName        string               `json:"objectName"`
	SegmentSize       int                  `json:"chunk_size"`
	OriginalName      string               `json:"originalName"`
	SegmentType       string               `json:"chunk_type"`
	Separators        []string             `json:"separators"`
	ParserChoices     []string             `json:"parser_choices"`
	OcrModelId        string               `json:"ocr_model_id"`
	PreProcess        []string             `json:"pre_process"`
	RagMetaDataParams []*RagMetaDataParams `json:"meta_data"`
}

type RagImportUrlDocParams struct {
	Url               string               `json:"url"`
	FileName          string               `json:"file_name"`
	Overlap           float32              `json:"overlap_size" `
	SegmentSize       int                  `json:"sentence_size"`
	SegmentType       string               `json:"chunk_type"`
	UserId            string               `json:"userId"`
	KnowledgeBaseName string               `json:"knowledgeBase"`
	IsEnhanced        bool                 `json:"is_enhanced"`
	Separators        []string             `json:"separators"`
	TaskId            string               `json:"task_id"`
	OcrModelId        string               `json:"ocr_model_id"`
	PreProcess        []string             `json:"pre_process"`
	RagMetaDataParams []*RagMetaDataParams `json:"meta_data"`
}

type RagKnowledgeHitResp struct {
	Code    int               `json:"code"`
	Message string            `json:"message"`
	Data    *KnowledgeHitData `json:"data"`
}

type RagMetaDataParams struct {
	MetaId    string      `json:"meta_id"`    // 元数据id
	Key       string      `json:"key"`        // key
	Value     interface{} `json:"value"`      // 常量
	ValueType string      `json:"value_type"` // 常量类型
	Rule      string      `json:"rule"`       // 正则表达式
}

type RagOperateKeywordsParams struct {
	Id               uint32   `json:"id"`
	UserId           string   `json:"user_id"`
	Action           string   `json:"action"`
	Name             string   `json:"name"`
	Alias            []string `json:"alias"`
	KnowledgeBaseIds []string `json:"knowledge_base_list"`
}

type RagOperationParams struct {
	Operation string              `json:"operation"`
	Type      string              `json:"type"`
	Doc       *RagImportDocParams `json:"doc"`
}

type RagUpdateDocSegmentParams struct {
	UserId          string           `json:"userId"`            // 发起请求的用户ID
	KnowledgeBase   string           `json:"knowledgeBase"`     // 知识库的名称
	KnowledgeId     string           `json:"kb_id"`             // 知识库的唯一ID
	FileName        string           `json:"fileName"`          // 与chunk关联的文件名
	MaxSentenceSize int              `json:"max_sentence_size"` // 最大分段长度限制
	Chunk           *UpdateChunkItem `json:"chunk"`             // 分段数据列表
}

type RagUpdateParams struct {
	UserId          string `json:"userId"`
	KnowledgeBaseId string `json:"kb_id"`
	OldKbName       string `json:"old_kb_name"`
	NewKbName       string `json:"new_kb_name"`
}

type SegmentResult struct {
	SuccessCount int `json:"success_count"` // 分段成功导入数量
}

type UpdateChunkItem struct {
	ChunkId string   `json:"chunk_id"`
	Content string   `json:"content"`
	Labels  []string `json:"labels"`
}

type WeightParams struct {
	VectorWeight float32 `json:"vector_weight"` //语义权重
	TextWeight   float32 `json:"text_weight"`   //关键字权重
}