Skip to content

Commit ad34d42

Browse files
committed
Try refreshing token for 404 error.
1 parent f076405 commit ad34d42

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ads/llm/langchain/plugins/llms/oci_data_science_model_deployment_endpoint.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -260,7 +260,7 @@ def _check_response(self, response: Any) -> None:
260260
if hasattr(response, "status_code")
261261
else response.status
262262
)
263-
if status_code == 401 and self._refresh_signer():
263+
if status_code in [401, 404] and self._refresh_signer():
264264
raise TokenExpiredError() from http_err
265265

266266
raise ServerError(

0 commit comments

Comments
 (0)