1
0
forked from github/onyx

Compare commits

...

2 Commits

Author SHA1 Message Date
pablodanswer
5452bc8a36 quick nit 2024-12-28 14:15:24 -05:00
pablodanswer
6ba082807a skip 401 active session 2024-12-28 14:15:24 -05:00

View File

@@ -113,6 +113,21 @@ def execute_paginated_retrieval(
results = _execute_with_retry(
lambda: retrieval_function(**request_kwargs).execute()
)
elif e.resp.status == 401:
# Skip only error code 4 (invalid session) but still raise for other 401s
error_content = e.content.decode("utf-8", errors="ignore")
if "Active session is invalid. Error code: 4" in error_content:
logger.warning(
f"Skipping 'Active session invalid (code 4)' "
f"401 error for request_kwargs={request_kwargs}: {e}"
)
results = {}
else:
logger.error(
"Reraising 401 HttpError. Possibly invalid/missing credentials."
)
raise e
else:
logger.exception("Error executing request:")
raise e