From 805575ef183348ce55a7d8749db477422d0b30de Mon Sep 17 00:00:00 2001 From: pablodanswer Date: Sat, 9 Nov 2024 10:47:41 -0800 Subject: [PATCH] don't prevent seeding --- backend/danswer/background/celery/apps/app_base.py | 2 +- backend/danswer/setup.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/backend/danswer/background/celery/apps/app_base.py b/backend/danswer/background/celery/apps/app_base.py index 38b91b4270c..5e5b24d2d50 100644 --- a/backend/danswer/background/celery/apps/app_base.py +++ b/backend/danswer/background/celery/apps/app_base.py @@ -457,4 +457,4 @@ def reset_tenant_id( **other_kwargs: Any, ) -> None: """Signal handler to reset tenant ID in context var after task ends.""" - CURRENT_TENANT_ID_CONTEXTVAR.set(None) + CURRENT_TENANT_ID_CONTEXTVAR.reset() diff --git a/backend/danswer/setup.py b/backend/danswer/setup.py index d7c3e82ffbe..d4af6656884 100644 --- a/backend/danswer/setup.py +++ b/backend/danswer/setup.py @@ -39,6 +39,7 @@ from danswer.natural_language_processing.search_nlp_models import warm_up_cross_encoder from danswer.search.models import SavedSearchSettings from danswer.search.retrieval.search_runner import download_nltk_data +from danswer.seeding.load_docs import seed_initial_documents from danswer.server.manage.llm.models import LLMProviderUpsertRequest from danswer.server.settings.store import load_settings from danswer.server.settings.store import store_settings