feat: add should_summarize parameter to task dispatchers

- Introduced should_summarize parameter in TaskDispatcher and CeleryTaskDispatcher to control summary generation.
- Updated InlineTaskDispatcher to support the new parameter for document processing.
This commit is contained in:
DESKTOP-RTLN3BA\$punk 2026-02-26 19:12:37 -08:00
parent 6f4bf11a32
commit a4dc84d1ab
3 changed files with 13 additions and 4 deletions

View file

@ -18,6 +18,7 @@ class TaskDispatcher(Protocol):
filename: str,
search_space_id: int,
user_id: str,
should_summarize: bool = False,
) -> None: ...
@ -32,6 +33,7 @@ class CeleryTaskDispatcher:
filename: str,
search_space_id: int,
user_id: str,
should_summarize: bool = False,
) -> None:
from app.tasks.celery_tasks.document_tasks import (
process_file_upload_with_document_task,
@ -43,6 +45,7 @@ class CeleryTaskDispatcher:
filename=filename,
search_space_id=search_space_id,
user_id=user_id,
should_summarize=should_summarize,
)

View file

@ -13,10 +13,10 @@ TEST_DATABASE_URL = os.environ.get("TEST_DATABASE_URL", _DEFAULT_TEST_DB)
# DATABASE_URL in the environment (e.g. from .env or shell profile).
os.environ["DATABASE_URL"] = TEST_DATABASE_URL
import pytest
import pytest # noqa: E402
from app.db import DocumentType
from app.indexing_pipeline.connector_document import ConnectorDocument
from app.db import DocumentType # noqa: E402
from app.indexing_pipeline.connector_document import ConnectorDocument # noqa: E402
# ---------------------------------------------------------------------------
# Unit test fixtures

View file

@ -64,6 +64,7 @@ class InlineTaskDispatcher:
filename: str,
search_space_id: int,
user_id: str,
should_summarize: bool = False,
) -> None:
from app.tasks.celery_tasks.document_tasks import (
_process_file_with_document,
@ -71,7 +72,12 @@ class InlineTaskDispatcher:
with contextlib.suppress(Exception):
await _process_file_with_document(
document_id, temp_path, filename, search_space_id, user_id
document_id,
temp_path,
filename,
search_space_id,
user_id,
should_summarize=should_summarize,
)