OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-25 09:47:41 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Fix: fixed to handle default and individual rag config properly

Browse Source
This commit is contained in:
Maytown 2025-05-14 17:35:23 +02:00
parent ad47e77974
commit f2e9e9a12e
1 changed files with 7 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -11,8 +11,6 @@
import AccessControl from '../common/AccessControl.svelte'; import AccessControl from '../common/AccessControl.svelte';
const dispatch = createEventDispatcher();
import { import {
getQuerySettings, getQuerySettings,
updateQuerySettings, updateQuerySettings,
@ -25,14 +23,8 @@
updateRAGConfig updateRAGConfig
} from '$lib/apis/retrieval'; } from '$lib/apis/retrieval';
import { reindexKnowledgeFiles } from '$lib/apis/knowledge';
import { deleteAllFiles } from '$lib/apis/files';
import ResetUploadDirConfirmDialog from '$lib/components/common/ConfirmDialog.svelte';
import ResetVectorDBConfirmDialog from '$lib/components/common/ConfirmDialog.svelte';
import ReindexKnowledgeFilesConfirmDialog from '$lib/components/common/ConfirmDialog.svelte';
import Textarea from '$lib/components/common/Textarea.svelte'; import Textarea from '$lib/components/common/Textarea.svelte';
import Spinner from '$lib/components/common/Spinner.svelte'; import resourcesToBackend from 'i18next-resources-to-backend';
const i18n = getContext('i18n'); const i18n = getContext('i18n');
@ -44,10 +36,6 @@
let updateEmbeddingModelLoading = false; let updateEmbeddingModelLoading = false;
let updateRerankingModelLoading = false; let updateRerankingModelLoading = false;
let showResetConfirm = false;
let showResetUploadDirConfirm = false;
let showReindexConfirm = false;
let embeddingEngine = ''; let embeddingEngine = '';
let embeddingModel = ''; let embeddingModel = '';
let embeddingBatchSize = 1; let embeddingBatchSize = 1;
@ -71,7 +59,7 @@
let RAGConfig = null; let RAGConfig = null;
const embeddingModelUpdateHandler = async () => { const embeddingModelUpdateHandler = async (knowledgeId) => {
if (embeddingEngine === '' && embeddingModel.split('/').length - 1 > 1) { if (embeddingEngine === '' && embeddingModel.split('/').length - 1 > 1) {
toast.error( toast.error(
@ -119,7 +107,7 @@
key: OpenAIKey, key: OpenAIKey,
url: OpenAIUrl url: OpenAIUrl
}, },
collection_name: name knowledge_id: knowledgeId
}).catch(async (error) => { }).catch(async (error) => {
toast.error(`${error}`); toast.error(`${error}`);
await setEmbeddingConfig(); await setEmbeddingConfig();
@ -137,14 +125,14 @@
} }
}; };
const rerankingModelUpdateHandler = async () => { const rerankingModelUpdateHandler = async (knowledgeId) => {
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 knowledge_id: knowledgeId
}).catch(async (error) => { }).catch(async (error) => {
toast.error(`${error}`); toast.error(`${error}`);
await setRerankingConfig(); await setRerankingConfig();
@ -247,10 +235,10 @@
} }
if (enableIndividualRagConfig) { if (enableIndividualRagConfig) {
if (!RAGConfig.BYPASS_EMBEDDING_AND_RETRIEVAL) { if (!RAGConfig.BYPASS_EMBEDDING_AND_RETRIEVAL) {
await embeddingModelUpdateHandler(); await embeddingModelUpdateHandler(res.id);
if (RAGConfig.ENABLE_RAG_HYBRID_SEARCH) { if (RAGConfig.ENABLE_RAG_HYBRID_SEARCH) {
await rerankingModelUpdateHandler(); await rerankingModelUpdateHandler(res.id);
} }
} }
} }