diff --git a/surfsense_backend/tests/e2e/run_backend.py b/surfsense_backend/tests/e2e/run_backend.py index 9a9a2b7ce..a1b4f7c8a 100644 --- a/surfsense_backend/tests/e2e/run_backend.py +++ b/surfsense_backend/tests/e2e/run_backend.py @@ -99,7 +99,8 @@ def _patch_llm_bindings() -> None: "app.tasks.connector_indexers.google_drive_indexer.get_user_long_context_llm", "app.tasks.connector_indexers.google_gmail_indexer.get_user_long_context_llm", "app.tasks.connector_indexers.local_folder_indexer.get_user_long_context_llm", - "app.tasks.document_processors.file_processors.get_user_long_context_llm", + "app.tasks.document_processors._save.get_user_long_context_llm", + "app.tasks.document_processors.markdown_processor.get_user_long_context_llm", ] for target in targets: try: diff --git a/surfsense_backend/tests/e2e/run_celery.py b/surfsense_backend/tests/e2e/run_celery.py index 2cbba966f..6e442de54 100644 --- a/surfsense_backend/tests/e2e/run_celery.py +++ b/surfsense_backend/tests/e2e/run_celery.py @@ -83,7 +83,8 @@ def _patch_llm_bindings() -> None: "app.tasks.connector_indexers.google_drive_indexer.get_user_long_context_llm", "app.tasks.connector_indexers.google_gmail_indexer.get_user_long_context_llm", "app.tasks.connector_indexers.local_folder_indexer.get_user_long_context_llm", - "app.tasks.document_processors.file_processors.get_user_long_context_llm", + "app.tasks.document_processors._save.get_user_long_context_llm", + "app.tasks.document_processors.markdown_processor.get_user_long_context_llm", ] for target in targets: try: