From ae5d2ecf48f2d6274de5acd12ebdedfbf18b5c12 Mon Sep 17 00:00:00 2001
From: LiuBodong <liubodong2010@126.com>
Date: Sat, 22 Mar 2025 20:21:57 +0800
Subject: [PATCH] fix:weight_type missing when create document in dataset
 (#16503)

---
 api/fields/dataset_fields.py                                   | 1 +
 web/app/components/workflow/nodes/knowledge-retrieval/types.ts | 2 ++
 web/models/debug.ts                                            | 2 ++
 3 files changed, 5 insertions(+)

diff --git a/api/fields/dataset_fields.py b/api/fields/dataset_fields.py
index 397532af93..67d183c70d 100644
--- a/api/fields/dataset_fields.py
+++ b/api/fields/dataset_fields.py
@@ -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),
 }
diff --git a/web/app/components/workflow/nodes/knowledge-retrieval/types.ts b/web/app/components/workflow/nodes/knowledge-retrieval/types.ts
index 07db5dba15..5f8b39e02b 100644
--- a/web/app/components/workflow/nodes/knowledge-retrieval/types.ts
+++ b/web/app/components/workflow/nodes/knowledge-retrieval/types.ts
@@ -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
diff --git a/web/models/debug.ts b/web/models/debug.ts
index 940df9fee6..18c2c02b62 100644
--- a/web/models/debug.ts
+++ b/web/models/debug.ts
@@ -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