diff --git a/libs/partners/mistralai/langchain_mistralai/embeddings.py b/libs/partners/mistralai/langchain_mistralai/embeddings.py index 9b97c8a1e3d..85e4d4289b7 100644 --- a/libs/partners/mistralai/langchain_mistralai/embeddings.py +++ b/libs/partners/mistralai/langchain_mistralai/embeddings.py @@ -224,7 +224,9 @@ class MistralAIEmbeddings(BaseModel, Embeddings): batch_responses = [] @retry( - retry=retry_if_exception_type(httpx.TimeoutException), + retry=retry_if_exception_type( + (httpx.TimeoutException, httpx.HTTPStatusError) + ), wait=wait_fixed(self.wait_time), stop=stop_after_attempt(self.max_retries), )