From ac53977dbc43adb4a8b005164d127c05a5e5e8dc Mon Sep 17 00:00:00 2001
From: ZhangShenao <15201440436@163.com>
Date: Thu, 6 Feb 2025 10:31:54 +0800
Subject: [PATCH] [MistralAI] Improve MistralAIEmbeddings (#29242)

- Add static method decorator for method.
- Add expected exception for retry decorator

#29125
---
 libs/partners/mistralai/langchain_mistralai/embeddings.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

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),
             )