don't prevent seeding

onyx-dot-app · Nov 21, 2024 · 805575e · 805575e
1 parent 7146d02
commit 805575e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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