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

pull/14262/head
Wu Tianwei 1 year ago committed by Bharat Ramanathan
parent 148bc3f9a1
commit 353df8afd7

@ -651,7 +651,13 @@ const Configuration: FC = () => {
syncToPublishedConfig(config) syncToPublishedConfig(config)
setPublishedConfig(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, provider: currentRerankProvider?.provider,
model: currentRerankModel?.model, model: currentRerankModel?.model,
}) })
@ -661,8 +667,8 @@ const Configuration: FC = () => {
...retrievalConfig, ...retrievalConfig,
...(retrievalConfig.reranking_model ? { ...(retrievalConfig.reranking_model ? {
reranking_model: { reranking_model: {
...retrievalConfig.reranking_model, reranking_model_name: retrievalConfig.reranking_model.model,
reranking_provider_name: correctModelProvider(modelConfig.dataset_configs.reranking_model.reranking_provider_name), reranking_provider_name: correctModelProvider(retrievalConfig.reranking_model.provider),
}, },
} : {}), } : {}),
}) })

Loading…
Cancel
Save