diff --git a/backend/open_webui/routers/knowledge.py b/backend/open_webui/routers/knowledge.py index ad67cc31f..e58442c50 100644 --- a/backend/open_webui/routers/knowledge.py +++ b/backend/open_webui/routers/knowledge.py @@ -25,6 +25,7 @@ from open_webui.utils.access_control import has_access, has_permission from open_webui.env import SRC_LOG_LEVELS +from open_webui.models.models import Models, ModelForm log = logging.getLogger(__name__) @@ -473,6 +474,36 @@ async def delete_knowledge_by_id(id: str, user=Depends(get_verified_user)): detail=ERROR_MESSAGES.ACCESS_PROHIBITED, ) + log.info(f"Deleting knowledge base: {id} (name: {knowledge.name})") + + # Get all models + models = Models.get_all_models() + log.info(f"Found {len(models)} models to check for knowledge base {id}") + + # Update models that reference this knowledge base + for model in models: + if model.meta and hasattr(model.meta, "knowledge"): + knowledge_list = model.meta.knowledge or [] + # Filter out the deleted knowledge base + updated_knowledge = [k for k in knowledge_list if k.get("id") != id] + + # If the knowledge list changed, update the model + if len(updated_knowledge) != len(knowledge_list): + log.info(f"Updating model {model.id} to remove knowledge base {id}") + model.meta.knowledge = updated_knowledge + # Create a ModelForm for the update + model_form = ModelForm( + id=model.id, + name=model.name, + base_model_id=model.base_model_id, + meta=model.meta, + params=model.params, + access_control=model.access_control, + is_active=model.is_active, + ) + Models.update_model_by_id(model.id, model_form) + + # Clean up vector DB try: VECTOR_DB_CLIENT.delete_collection(collection_name=id) except Exception as e: diff --git a/src/lib/utils/index.ts b/src/lib/utils/index.ts index 1d2095780..82b30b2d6 100644 --- a/src/lib/utils/index.ts +++ b/src/lib/utils/index.ts @@ -74,7 +74,6 @@ export const processResponseContent = (content: string) => { return content.trim(); }; - export function unescapeHtml(html: string) { const doc = new DOMParser().parseFromString(html, 'text/html'); return doc.documentElement.textContent;