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 GitHub
parent e3bd521654
commit 718c9cbe3a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -315,8 +315,7 @@ class ChatMistralAI(BaseChatModel):
max_retries: int = 5 max_retries: int = 5
timeout: int = 120 timeout: int = 120
max_concurrent_requests: int = 64 max_concurrent_requests: int = 64
model: str = Field(default="mistral-small", alias="model_name")
model: str = "mistral-small"
temperature: float = 0.7 temperature: float = 0.7
max_tokens: Optional[int] = None max_tokens: Optional[int] = None
top_p: float = 1 top_p: float = 1