|
@@ -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),
|
|
|
},
|
|
|
} : {}),
|
|
|
})
|