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

This commit is contained in:
Eugene Yurtsev
2024-04-18 16:12:33 -04:00
committed by William Fu-Hinthorn
parent 2f42e0e1b5
commit 386395a90d

View File

@@ -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