From 13caea2dbac289e794efc3ca3aa0aa2f6f979683 Mon Sep 17 00:00:00 2001 From: rkuo-danswer Date: Sat, 23 Nov 2024 09:11:15 -0800 Subject: [PATCH] add minimal retries to confluence probe (#3222) * add minimal retries to confluence probe * name variable correctly --- backend/danswer/connectors/confluence/onyx_confluence.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/backend/danswer/connectors/confluence/onyx_confluence.py b/backend/danswer/connectors/confluence/onyx_confluence.py index 739e4aef21d..8b4ec81ef8b 100644 --- a/backend/danswer/connectors/confluence/onyx_confluence.py +++ b/backend/danswer/connectors/confluence/onyx_confluence.py @@ -294,14 +294,17 @@ def _validate_connector_configuration( wiki_base: str, ) -> None: # test connection with direct client, no retries - confluence_client_without_retries = Confluence( + confluence_client_with_minimal_retries = Confluence( api_version="cloud" if is_cloud else "latest", url=wiki_base.rstrip("/"), username=credentials["confluence_username"] if is_cloud else None, password=credentials["confluence_access_token"] if is_cloud else None, token=credentials["confluence_access_token"] if not is_cloud else None, + backoff_and_retry=True, + max_backoff_retries=6, + max_backoff_seconds=10, ) - spaces = confluence_client_without_retries.get_all_spaces(limit=1) + spaces = confluence_client_with_minimal_retries.get_all_spaces(limit=1) if not spaces: raise RuntimeError(