AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-13 20:05:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: multiple retrieval default weighted score (#6897)

Browse Source
This commit is contained in:
zxhlyh 2024-08-02 14:05:27 +08:00 committed by GitHub
parent 8166a8caf5
commit 33dab4fe54
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
web/app/components/datasets/common/retrieval-param-config/index.tsx
View File

@ -16,6 +16,7 @@ import ModelSelector from '@/app/components/header/account-setting/model-provide
import { useModelListAndDefaultModel } from '@/app/components/header/account-setting/model-provider-page/hooks' import { useModelListAndDefaultModel } from '@/app/components/header/account-setting/model-provider-page/hooks'
import { ModelTypeEnum } from '@/app/components/header/account-setting/model-provider-page/declarations' import { ModelTypeEnum } from '@/app/components/header/account-setting/model-provider-page/declarations'
import { import {
DEFAULT_WEIGHTED_SCORE,
RerankingModeEnum, RerankingModeEnum,
WeightedScoreEnum, WeightedScoreEnum,
} from '@/models/datasets' } from '@/models/datasets'
@ -69,12 +70,12 @@ const RetrievalParamConfig: FC<Props> = ({
result.weights = { result.weights = {
weight_type: WeightedScoreEnum.Customized, weight_type: WeightedScoreEnum.Customized,
vector_setting: { vector_setting: {
vector_weight: 0.5, vector_weight: DEFAULT_WEIGHTED_SCORE.other.semantic,
embedding_provider_name: '', embedding_provider_name: '',
embedding_model_name: '', embedding_model_name: '',
}, },
keyword_setting: { keyword_setting: {
keyword_weight: 0.5, keyword_weight: DEFAULT_WEIGHTED_SCORE.other.keyword,
}, },
} }
} }
@ -202,7 +203,6 @@ const RetrievalParamConfig: FC<Props> = ({
value.reranking_mode === RerankingModeEnum.WeightedScore && ( value.reranking_mode === RerankingModeEnum.WeightedScore && (
<WeightedScore <WeightedScore
value={{ value={{
type: value.weights!.weight_type,
value: [ value: [
value.weights!.vector_setting.vector_weight, value.weights!.vector_setting.vector_weight,
value.weights!.keyword_setting.keyword_weight, value.weights!.keyword_setting.keyword_weight,
@ -213,7 +213,6 @@ const RetrievalParamConfig: FC<Props> = ({
...value, ...value,
weights: { weights: {
...value.weights!, ...value.weights!,
weight_type: v.type,
vector_setting: { vector_setting: {
...value.weights!.vector_setting, ...value.weights!.vector_setting,
vector_weight: v.value[0], vector_weight: v.value[0],

2
web/app/components/workflow/nodes/knowledge-retrieval/utils.ts
View File

@ -88,7 +88,7 @@ export const getMultipleRetrievalConfig = (multipleRetrievalConfig: MultipleRetr
reranking_mode, reranking_mode,
reranking_model, reranking_model,
weights, weights,
reranking_enable, reranking_enable: allEconomic ? reranking_enable : true,
} }
if (allEconomic || mixtureHighQualityAndEconomic || inconsistentEmbeddingModel) if (allEconomic || mixtureHighQualityAndEconomic || inconsistentEmbeddingModel)