| @@ -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 { | |||
| 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 | |||