From e8dcdfb9f0af0f44393d84d52bf423d16762bfd3 Mon Sep 17 00:00:00 2001
From: balibabu <cike8899@users.noreply.github.com>
Date: Tue, 9 Sep 2025 12:32:22 +0800
Subject: [PATCH] Fix: Issue of ineffective weight adjustment for
 retrieval_test API-related functions #9854 (#9989)

### What problem does this PR solve?

Fix: Issue of ineffective weight adjustment for retrieval_test
API-related functions #9854

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
---
 web/src/interfaces/request/knowledge.ts        |  2 +-
 web/src/pages/dataset/testing/testing-form.tsx | 11 ++++++-----
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/web/src/interfaces/request/knowledge.ts b/web/src/interfaces/request/knowledge.ts
index c43e5b2d1..de1b00b43 100644
--- a/web/src/interfaces/request/knowledge.ts
+++ b/web/src/interfaces/request/knowledge.ts
@@ -1,7 +1,7 @@
 export interface ITestRetrievalRequestBody {
   question: string;
   similarity_threshold: number;
-  keywords_similarity_weight: number;
+  vector_similarity_weight: number;
   rerank_id?: string;
   top_k?: number;
   use_kg?: boolean;
diff --git a/web/src/pages/dataset/testing/testing-form.tsx b/web/src/pages/dataset/testing/testing-form.tsx
index ba614ed67..fb9f3189b 100644
--- a/web/src/pages/dataset/testing/testing-form.tsx
+++ b/web/src/pages/dataset/testing/testing-form.tsx
@@ -12,11 +12,11 @@ import {
   topKSchema,
 } from '@/components/rerank';
 import {
-  initialKeywordsSimilarityWeightValue,
   initialSimilarityThresholdValue,
-  keywordsSimilarityWeightSchema,
+  initialVectorSimilarityWeightValue,
   SimilaritySliderFormField,
   similarityThresholdSchema,
+  vectorSimilarityWeightSchema,
 } from '@/components/similarity-slider';
 import { ButtonLoading } from '@/components/ui/button';
 import {
@@ -52,16 +52,18 @@ export default function TestingForm({
       message: t('knowledgeDetails.testTextPlaceholder'),
     }),
     ...similarityThresholdSchema,
-    ...keywordsSimilarityWeightSchema,
+    ...vectorSimilarityWeightSchema,
     ...topKSchema,
+    use_kg: z.boolean().optional(),
   });
 
   const form = useForm<z.infer<typeof formSchema>>({
     resolver: zodResolver(formSchema),
     defaultValues: {
       ...initialSimilarityThresholdValue,
-      ...initialKeywordsSimilarityWeightValue,
+      ...initialVectorSimilarityWeightValue,
       ...initialTopKValue,
+      use_kg: false,
     },
   });
 
@@ -82,7 +84,6 @@ export default function TestingForm({
       <form onSubmit={form.handleSubmit(onSubmit)} className="space-y-8">
         <FormContainer className="p-10">
           <SimilaritySliderFormField
-            vectorSimilarityWeightName="keywords_similarity_weight"
             isTooltipShown={true}
           ></SimilaritySliderFormField>
           <RerankFormFields></RerankFormFields>