@@ -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),
@@ -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
@@ -1,6 +1,7 @@
import type { AgentStrategy, ModelModeType, RETRIEVE_TYPE, ToolItem, TtsAutoPlay } from '@/types/app'
import type {
import type { FileUpload } from '@/app/components/base/features/types'
@@ -165,6 +166,7 @@ export type DatasetConfigs = {