diff --git a/libs/core/langchain_core/load/load.py b/libs/core/langchain_core/load/load.py index f7db2ff6f02..fb6e3ec2a1d 100644 --- a/libs/core/langchain_core/load/load.py +++ b/libs/core/langchain_core/load/load.py @@ -17,6 +17,7 @@ DEFAULT_NAMESPACES = [ "langchain_core", "langchain_community", "langchain_anthropic", + "langchain_groq", ] ALL_SERIALIZABLE_MAPPINGS = { diff --git a/libs/core/langchain_core/load/mapping.py b/libs/core/langchain_core/load/mapping.py index 84c3878c5d8..c884d65ac55 100644 --- a/libs/core/langchain_core/load/mapping.py +++ b/libs/core/langchain_core/load/mapping.py @@ -271,6 +271,11 @@ SERIALIZABLE_MAPPING: Dict[Tuple[str, ...], Tuple[str, ...]] = { "chat_models", "ChatAnthropic", ), + ("langchain_groq", "chat_models", "ChatGroq"): ( + "langchain_groq", + "chat_models", + "ChatGroq" + ), ("langchain", "chat_models", "fireworks", "ChatFireworks"): ( "langchain_fireworks", "chat_models", @@ -287,6 +292,11 @@ SERIALIZABLE_MAPPING: Dict[Tuple[str, ...], Tuple[str, ...]] = { "chat_models", "ChatVertexAI", ), + ("langchain", "chat_models", "mistralai", "ChatMistralAI"): ( + "langchain_mistralai", + "chat_models", + "ChatMistralAI", + ), ("langchain", "schema", "output", "ChatGenerationChunk"): ( "langchain_core", "outputs", diff --git a/libs/partners/mistralai/langchain_mistralai/chat_models.py b/libs/partners/mistralai/langchain_mistralai/chat_models.py index 06ee230d1f6..6a497bf91f0 100644 --- a/libs/partners/mistralai/langchain_mistralai/chat_models.py +++ b/libs/partners/mistralai/langchain_mistralai/chat_models.py @@ -359,8 +359,8 @@ def _convert_message_to_mistral_chat_message( class ChatMistralAI(BaseChatModel): """A chat model that uses the MistralAI API.""" - client: httpx.Client = Field(default=None) #: :meta private: - async_client: httpx.AsyncClient = Field(default=None) #: :meta private: + client: httpx.Client = 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), diff --git a/libs/partners/mistralai/tests/unit_tests/__snapshots__/test_standard.ambr b/libs/partners/mistralai/tests/unit_tests/__snapshots__/test_standard.ambr index 16389a8d9da..75bdad2decf 100644 --- a/libs/partners/mistralai/tests/unit_tests/__snapshots__/test_standard.ambr +++ b/libs/partners/mistralai/tests/unit_tests/__snapshots__/test_standard.ambr @@ -45,24 +45,6 @@ 'ChatMistralAI', ]), 'kwargs': dict({ - 'async_client': dict({ - 'id': list([ - 'httpx', - 'AsyncClient', - ]), - 'lc': 1, - 'repr': '', - 'type': 'not_implemented', - }), - 'client': dict({ - 'id': list([ - 'httpx', - 'Client', - ]), - 'lc': 1, - 'repr': '', - 'type': 'not_implemented', - }), 'endpoint': 'boo', 'max_concurrent_requests': 64, 'max_retries': 2,