add minimal retries to confluence probe (#3222)

* add minimal retries to confluence probe

* name variable correctly
This commit is contained in:
rkuo-danswer 2024-11-23 09:11:15 -08:00 committed by GitHub
parent 85cbd9caed
commit 3a466a4b08
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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(