-
Notifications
You must be signed in to change notification settings - Fork 1.8k
checkpointed confluence #4473
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
checkpointed confluence #4473
Conversation
The latest updates on your projects. Learn more about Vercel for Git ↗︎
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
PR Summary
This PR integrates a new checkpointing mechanism for multiple connectors, replacing CheckpointConnector with the new CheckpointedConnector interface and standardizing checkpoint management across the codebase.
- Updated backend/onyx/connectors/factory.py to support CheckpointedConnector for POLL input types.
- Modified backend/onyx/connectors/connector_runner.py to enforce time_range and wrap outputs with CheckpointOutputWrapper.
- Refactored backend/onyx/background/indexing/checkpointing_utils.py for updated checkpoint validation and loading.
- Adjusted key connectors (Confluence, GitHub, GoogleDrive, etc.) to increment checkpoints per batch with enhanced error propagation.
- Test files now use utilities like load_all_docs_from_checkpoint_connector for thorough validation.
15 file(s) reviewed, no comment(s)
Edit PR Review Bot Settings | Greptile
9156b74
to
6df1008
Compare
6df1008
to
11854b1
Compare
if not self.continue_on_failure: | ||
if _should_propagate_error(e): | ||
raise | ||
# TODO: should we remove continue_on_failure entirely now that we have checkpointing? |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
yes, I think so!
11a806c
to
5b5c35b
Compare
* checkpointed confluence * confluence checkpointing tested * fixed integration tests * attempt to fix connector test flakiness * fix rebase
Description
Addresses https://linear.app/danswer/issue/DAN-1703/checkpointed-confluence-connector
Checkpointing for the confluence connector. Creates a new checkpoint each time the previous version of the connector would have finished a batch.
How Has This Been Tested?
tested in UI and unit tests
Backporting (check the box to trigger backport action)
Note: You have to check that the action passes, otherwise resolve the conflicts manually and tag the patches.