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>({ resolver: zodResolver(formSchema), defaultValues: { ...initialSimilarityThresholdValue, - ...initialKeywordsSimilarityWeightValue, + ...initialVectorSimilarityWeightValue, ...initialTopKValue, + use_kg: false, }, }); @@ -82,7 +84,6 @@ export default function TestingForm({