- {#if embeddingEngine === 'ollama'}
-
{$i18n.t(
'Warning: If you update or change your embedding model, you will need to re-import all documents.'
@@ -694,71 +725,102 @@
{#if RAGConfig.ENABLE_RAG_HYBRID_SEARCH === true}
{$i18n.t('Reranking Model')}
-
-
-
-
-
-
-
{
- rerankingModelUpdateHandler();
- }}
- disabled={updateRerankingModelLoading}
- >
- {#if updateRerankingModelLoading}
-
-
-
+
+ @keyframes spinner_AtaB {
+ 100% {
+ transform: rotate(360deg);
+ }
+ }
+
+
+
+
+
+ {:else}
+
-
- {:else}
-
-
-
-
+ {/if}
+
+
+
+
+ {/if}
+
+
+
+ {$i18n.t('Select reranking model')}
+
+ {#if rerankingModel && rerankingModel.trim() !== ''}
+
+ {rerankingModel}
+ {#if !RAGConfig.DOWNLOADED_RERANKING_MODELS?.includes(embeddingModel)}
+ (custom)
+ {/if}
+
{/if}
-
+
+
+ {#if RAGConfig.DOWNLOADED_RERANKING_MODELS}
+ {#each RAGConfig.DOWNLOADED_RERANKING_MODELS as model}
+ {#if model !== rerankingModel}
+ {model}
+ {/if}
+ {/each}
+ {/if}
+
diff --git a/src/lib/components/workspace/common/RagConfigModal.svelte b/src/lib/components/workspace/common/RagConfigModal.svelte
index f97ae9188..33ed1149e 100644
--- a/src/lib/components/workspace/common/RagConfigModal.svelte
+++ b/src/lib/components/workspace/common/RagConfigModal.svelte
@@ -231,7 +231,11 @@
embedding_batch_size,
openai_config,
ollama_config,
- reranking_model,
+ reranking_model,
+ LOADED_EMBEDDING_MODELS,
+ DOWNLOADED_EMBEDDING_MODELS,
+ LOADED_RERANKING_MODELS,
+ DOWNLOADED_RERANKING_MODELS,
...filteredRAGConfig
} = localRAGConfig;
@@ -247,6 +251,16 @@
await rerankingModelUpdateHandler();
}
}
+
+ // Update embedding and reranking to show updates in UI
+ localRAGConfig.embedding_engine = embeddingEngine
+ localRAGConfig.embedding_model = embeddingModel
+ localRAGConfig.embedding_batch_size = embeddingBatchSize
+ localRAGConfig.openai_config = {"key": OpenAIKey, "url": OpenAIUrl}
+ localRAGConfig.ollama_config = {"key": OllamaKey, "url": OllamaUrl}
+ localRAGConfig.reranking_model = rerankingModel
+
+
console.log('RAGConfig:', localRAGConfig);
dispatch('update', localRAGConfig)
loading = false;
@@ -487,98 +501,136 @@
{/if}