Skip to content

Commit

Permalink
add minimal retries to confluence probe (#3222)
Browse files Browse the repository at this point in the history
* add minimal retries to confluence probe

* name variable correctly
  • Loading branch information
rkuo-danswer authored Nov 23, 2024
1 parent 85cbd9c commit 3a466a4
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions backend/danswer/connectors/confluence/onyx_confluence.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down

0 comments on commit 3a466a4

Please sign in to comment.