fmt

libs/community/tests/unit_tests/load/test_serializable.py

@@ -102,6 +102,16 @@ def test_serializable_mapping() -> None:
             "modifier",
             "RemoveMessage",
         ),
+        ("langchain", "chat_models", "mistralai", "MistralAI"): (
+            "langchain_mistralai",
+            "chat_models",
+            "ChatMistralAI",
+        ),
+        ("langchain_groq", "chat_models", "ChatGroq"): (
+            "langchain_groq",
+            "chat_models",
+            "ChatGroq",
+        ),
     }
     serializable_modules = import_all_modules("langchain")
 

libs/core/langchain_core/load/mapping.py

@@ -274,7 +274,7 @@ SERIALIZABLE_MAPPING: Dict[Tuple[str, ...], Tuple[str, ...]] = {
     ("langchain_groq", "chat_models", "ChatGroq"): (
         "langchain_groq",
         "chat_models",
-        "ChatGroq"
+        "ChatGroq",
     ),
     ("langchain", "chat_models", "fireworks", "ChatFireworks"): (
         "langchain_fireworks",

libs/partners/mistralai/langchain_mistralai/chat_models.py

@@ -362,7 +362,9 @@ class ChatMistralAI(BaseChatModel):
     """A chat model that uses the MistralAI API."""
 
     client: httpx.Client = Field(default=None, exclude=True)  #: :meta private:
-    async_client: httpx.AsyncClient = Field(default=None, exclude=True)  #: :meta private:
+    async_client: httpx.AsyncClient = Field(
+        default=None, exclude=True
+    )  #: :meta private:
     mistral_api_key: Optional[SecretStr] = Field(
         alias="api_key",
         default_factory=secret_from_env("MISTRAL_API_KEY", default=None),

libs/partners/openai/tests/unit_tests/chat_models/__snapshots__/test_azure_standard.ambr

@@ -45,6 +45,13 @@
       'AzureChatOpenAI',
     ]),
     'kwargs': dict({
+      'azure_ad_token': dict({
+        'id': list([
+          'AZURE_OPENAI_AD_TOKEN',
+        ]),
+        'lc': 1,
+        'type': 'secret',
+      }),
       'azure_endpoint': 'https://test.azure.com',
       'deployment_name': 'test',
       'max_retries': 2,