mistral[patch]: Support both model and model_name (#20557)

6 months ago · 718c9cbe3a
parent e3bd521654
commit 718c9cbe3a
1 changed files with 1 additions and 2 deletions
--- a/libs/partners/mistralai/langchain_mistralai/chat_models.py
+++ b/libs/partners/mistralai/langchain_mistralai/chat_models.py
@ -315,8 +315,7 @@ class ChatMistralAI(BaseChatModel):
    max_retries: int = 5
    timeout: int = 120
    max_concurrent_requests: int = 64
-
-    model: str = "mistral-small"
+    model: str = Field(default="mistral-small", alias="model_name")
    temperature: float = 0.7
    max_tokens: Optional[int] = None
    top_p: float = 1