mistralai[patch]: standardize model params (#20163)

Related to #20085

libs/partners/mistralai/langchain_mistralai/chat_models.py

@@ -186,7 +186,7 @@ class ChatMistralAI(BaseChatModel):
 
     client: httpx.Client = Field(default=None)  #: :meta private:
     async_client: httpx.AsyncClient = Field(default=None)  #: :meta private:
-    mistral_api_key: Optional[SecretStr] = None
+    mistral_api_key: Optional[SecretStr] = Field(default=None, alias="api_key")
     endpoint: str = "https://api.mistral.ai/v1"
     max_retries: int = 5
     timeout: int = 120
@@ -202,6 +202,12 @@ class ChatMistralAI(BaseChatModel):
     safe_mode: bool = False
     streaming: bool = False
 
+    class Config:
+        """Configuration for this pydantic object."""
+
+        allow_population_by_field_name = True
+        arbitrary_types_allowed = True
+
     @property
     def _default_params(self) -> Dict[str, Any]:
         """Get the default parameters for calling the API."""

libs/partners/mistralai/langchain_mistralai/embeddings.py

@@ -29,15 +29,16 @@ class MistralAIEmbeddings(BaseModel, Embeddings):
         .. code-block:: python
 
             from langchain_mistralai import MistralAIEmbeddings
+
             mistral = MistralAIEmbeddings(
                 model="mistral-embed",
-                mistral_api_key="my-api-key"
+                api_key="my-api-key"
             )
     """
 
     client: httpx.Client = Field(default=None)  #: :meta private:
     async_client: httpx.AsyncClient = Field(default=None)  #: :meta private:
-    mistral_api_key: Optional[SecretStr] = None
+    mistral_api_key: Optional[SecretStr] = Field(default=None, alias="api_key")
     endpoint: str = "https://api.mistral.ai/v1/"
     max_retries: int = 5
     timeout: int = 120
@@ -49,6 +50,7 @@ class MistralAIEmbeddings(BaseModel, Embeddings):
     class Config:
         extra = Extra.forbid
         arbitrary_types_allowed = True
+        allow_population_by_field_name = True
 
     @root_validator()
     def validate_environment(cls, values: Dict) -> Dict:

libs/partners/mistralai/tests/unit_tests/test_chat_models.py

@@ -1,7 +1,7 @@
 """Test MistralAI Chat API wrapper."""
 
 import os
-from typing import Any, AsyncGenerator, Dict, Generator
+from typing import Any, AsyncGenerator, Dict, Generator, cast
 from unittest.mock import patch
 
 import pytest
@@ -13,6 +13,7 @@ from langchain_core.messages import (
     HumanMessage,
     SystemMessage,
 )
+from langchain_core.pydantic_v1 import SecretStr
 
 from langchain_mistralai.chat_models import (  # type: ignore[import]
     ChatMistralAI,
@@ -31,7 +32,11 @@ def test_mistralai_initialization() -> None:
     """Test ChatMistralAI initialization."""
     # Verify that ChatMistralAI can be initialized using a secret key provided
     # as a parameter rather than an environment variable.
-    ChatMistralAI(model="test", mistral_api_key="test")
+    for model in [
+        ChatMistralAI(model="test", mistral_api_key="test"),
+        ChatMistralAI(model="test", api_key="test"),
+    ]:
+        assert cast(SecretStr, model.mistral_api_key).get_secret_value() == "test"
 
 
 @pytest.mark.parametrize(

libs/partners/mistralai/tests/unit_tests/test_embeddings.py

@@ -1,4 +1,7 @@
 import os
+from typing import cast
+
+from langchain_core.pydantic_v1 import SecretStr
 
 from langchain_mistralai import MistralAIEmbeddings
 
@@ -6,5 +9,9 @@ os.environ["MISTRAL_API_KEY"] = "foo"
 
 
 def test_mistral_init() -> None:
-    embeddings = MistralAIEmbeddings()
-    assert embeddings.model == "mistral-embed"
+    for model in [
+        MistralAIEmbeddings(model="mistral-embed", mistral_api_key="test"),
+        MistralAIEmbeddings(model="mistral-embed", api_key="test"),
+    ]:
+        assert model.model == "mistral-embed"
+        assert cast(SecretStr, model.mistral_api_key).get_secret_value() == "test"