fix:weight_type missing when create document in dataset (#16503)

This commit is contained in:
LiuBodong
2025-03-22 20:21:57 +08:00
committed by GitHub
parent 1907d2a90a
commit ae5d2ecf48
3 changed files with 5 additions and 0 deletions

View File

@@ -24,6 +24,7 @@ vector_setting_fields = {
}
weighted_score_fields = {
"weight_type": fields.String,
"keyword_setting": fields.Nested(keyword_setting_fields),
"vector_setting": fields.Nested(vector_setting_fields),
}

View File

@@ -10,6 +10,7 @@ import type {
DataSet,
MetadataInDoc,
RerankingModeEnum,
WeightedScoreEnum,
} from '@/models/datasets'
export type MultipleRetrievalConfig = {
@@ -21,6 +22,7 @@ export type MultipleRetrievalConfig = {
}
reranking_mode?: RerankingModeEnum
weights?: {
weight_type: WeightedScoreEnum
vector_setting: {
vector_weight: number
embedding_provider_name: string

View File

@@ -1,6 +1,7 @@
import type { AgentStrategy, ModelModeType, RETRIEVE_TYPE, ToolItem, TtsAutoPlay } from '@/types/app'
import type {
RerankingModeEnum,
WeightedScoreEnum,
} from '@/models/datasets'
import type { FileUpload } from '@/app/components/base/features/types'
import type {
@@ -165,6 +166,7 @@ export type DatasetConfigs = {
}
reranking_mode?: RerankingModeEnum
weights?: {
weight_type: WeightedScoreEnum
vector_setting: {
vector_weight: number
embedding_provider_name: string