OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-26 18:26:48 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix: adjusted to load default rag settings for inital loading and added missing content extraction engine

Browse Source
This commit is contained in:
weberm1 2025-05-09 22:36:37 +02:00
parent 12774641a1
commit 34d4238436
1 changed files with 30 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
src/lib/components/workspace/Knowledge/CreateKnowledgeBase.svelte
View File

@ -69,33 +69,9 @@
let enableIndividualRagConfig = false; let enableIndividualRagConfig = false;
// Default RAG configuration let RAGConfig = null;
let RAGConfig = {
CONTENT_EXTRACTION_ENGINE: '',
TIKA_SERVER_URL: '',
DOCLING_SERVER_URL: '',
DOCUMENT_INTELLIGENCE_ENDPOINT: '',
DOCUMENT_INTELLIGENCE_KEY: '',
MISTRAL_OCR_API_KEY: '',
BYPASS_EMBEDDING_AND_RETRIEVAL: false,
TEXT_SPLITTER: '',
CHUNK_SIZE: 1000,
CHUNK_OVERLAP: 200,
RAG_FULL_CONTEXT: false,
ENABLE_RAG_HYBRID_SEARCH: false,
TOP_K: 4,
TOP_K_RERANKER: 4,
RELEVANCE_THRESHOLD: 0.0,
RAG_TEMPLATE: '',
PDF_EXTRACT_IMAGES: false,
DEFAULT_RAG_SETTINGS: true
};
const embeddingModelUpdateHandler = async () => { const embeddingModelUpdateHandler = async () => {
if (!enableIndividualRagConfig) return;
if (embeddingEngine === '' && embeddingModel.split('/').length - 1 > 1) { if (embeddingEngine === '' && embeddingModel.split('/').length - 1 > 1) {
toast.error( toast.error(
@ -142,7 +118,8 @@
openai_config: { openai_config: {
key: OpenAIKey, key: OpenAIKey,
url: OpenAIUrl url: OpenAIUrl
} },
collection_name: name
}).catch(async (error) => { }).catch(async (error) => {
toast.error(`${error}`); toast.error(`${error}`);
await setEmbeddingConfig(); await setEmbeddingConfig();
@ -161,13 +138,13 @@
}; };
const rerankingModelUpdateHandler = async () => { const rerankingModelUpdateHandler = async () => {
if (!enableIndividualRagConfig) return;
console.log('Update reranking model attempt:', rerankingModel); console.log('Update reranking model attempt:', rerankingModel);
updateRerankingModelLoading = true; updateRerankingModelLoading = true;
const res = await updateRerankingConfig(localStorage.token, { const res = await updateRerankingConfig(localStorage.token, {
reranking_model: rerankingModel reranking_model: rerankingModel,
collection_name: name
}).catch(async (error) => { }).catch(async (error) => {
toast.error(`${error}`); toast.error(`${error}`);
await setRerankingConfig(); await setRerankingConfig();
@ -192,9 +169,8 @@
}; };
const setEmbeddingConfig = async () => { const setEmbeddingConfig = async () => {
if (!enableIndividualRagConfig) return;
const embeddingConfig = await getEmbeddingConfig(localStorage.token, {collection_name: name}); const embeddingConfig = await getEmbeddingConfig(localStorage.token);
if (embeddingConfig) { if (embeddingConfig) {
embeddingEngine = embeddingConfig.embedding_engine; embeddingEngine = embeddingConfig.embedding_engine;
@ -210,7 +186,6 @@
}; };
const setRerankingConfig = async () => { const setRerankingConfig = async () => {
if (!enableIndividualRagConfig) return;
const rerankingConfig = await getRerankingConfig(localStorage.token); const rerankingConfig = await getRerankingConfig(localStorage.token);
@ -280,6 +255,13 @@
loading = false; loading = false;
}; };
onMount(async () => {
await setEmbeddingConfig();
await setRerankingConfig();
RAGConfig = await getRAGConfig(localStorage.token);
});
</script> </script>
<div class="w-full max-h-full"> <div class="w-full max-h-full">
@ -371,6 +353,23 @@
</div> </div>
{#if enableIndividualRagConfig} {#if enableIndividualRagConfig}
<div class="flex w-full justify-between">
<div class="self-center text-xs font-medium">
{$i18n.t('Content Extraction Engine')}
</div>
<div class="">
<select
class="dark:bg-gray-900 w-fit pr-8 rounded-sm px-2 text-xs bg-transparent outline-hidden text-right"
bind:value={RAGConfig.CONTENT_EXTRACTION_ENGINE}
>
<option value="">{$i18n.t('Default')}</option>
<option value="tika">{$i18n.t('Tika')}</option>
<option value="docling">{$i18n.t('Docling')}</option>
<option value="document_intelligence">{$i18n.t('Document Intelligence')}</option>
<option value="mistral_ocr">{$i18n.t('Mistral OCR')}</option>
</select>
</div>
</div>
<div class="space-y-4 mt-4"> <div class="space-y-4 mt-4">
{#if RAGConfig.CONTENT_EXTRACTION_ENGINE === ''} {#if RAGConfig.CONTENT_EXTRACTION_ENGINE === ''}
<div class="flex w-full mt-1"> <div class="flex w-full mt-1">
@ -699,9 +698,6 @@
<div class="flex items-center relative"> <div class="flex items-center relative">
<Switch <Switch
bind:state={RAGConfig.ENABLE_RAG_HYBRID_SEARCH} bind:state={RAGConfig.ENABLE_RAG_HYBRID_SEARCH}
on:change={() => {
submitHandler();
}}
/> />
</div> </div>
</div> </div>