浏览代码

fix: update retrieval configuration to correctly handle reranking mod… (#16641)

tags/1.1.3
Wu Tianwei 7 个月前
父节点
当前提交
6e26ed2bb7
没有帐户链接到提交者的电子邮件
共有 1 个文件被更改,包括 9 次插入3 次删除
  1. 9
    3
      web/app/components/app/configuration/index.tsx

+ 9
- 3
web/app/components/app/configuration/index.tsx 查看文件

@@ -651,7 +651,13 @@ const Configuration: FC = () => {

syncToPublishedConfig(config)
setPublishedConfig(config)
const retrievalConfig = getMultipleRetrievalConfig(modelConfig.dataset_configs, datasets, datasets, {
const retrievalConfig = getMultipleRetrievalConfig({
...modelConfig.dataset_configs,
reranking_model: modelConfig.dataset_configs.reranking_model && {
provider: modelConfig.dataset_configs.reranking_model.reranking_provider_name,
model: modelConfig.dataset_configs.reranking_model.reranking_model_name,
},
}, datasets, datasets, {
provider: currentRerankProvider?.provider,
model: currentRerankModel?.model,
})
@@ -661,8 +667,8 @@ const Configuration: FC = () => {
...retrievalConfig,
...(retrievalConfig.reranking_model ? {
reranking_model: {
...retrievalConfig.reranking_model,
reranking_provider_name: correctModelProvider(modelConfig.dataset_configs.reranking_model.reranking_provider_name),
reranking_model_name: retrievalConfig.reranking_model.model,
reranking_provider_name: correctModelProvider(retrievalConfig.reranking_model.provider),
},
} : {}),
})

正在加载...
取消
保存