From b74546a458672643934bc3842cd9e8075d68cd11 Mon Sep 17 00:00:00 2001 From: ccurme Date: Fri, 6 Sep 2024 12:54:16 -0400 Subject: [PATCH] core[patch]: add google genai to serialization (#26154) --- libs/core/langchain_core/load/load.py | 1 + libs/core/langchain_core/load/mapping.py | 5 +++++ 2 files changed, 6 insertions(+) diff --git a/libs/core/langchain_core/load/load.py b/libs/core/langchain_core/load/load.py index fb6e3ec2a1d..e1c6e48ef1e 100644 --- a/libs/core/langchain_core/load/load.py +++ b/libs/core/langchain_core/load/load.py @@ -18,6 +18,7 @@ DEFAULT_NAMESPACES = [ "langchain_community", "langchain_anthropic", "langchain_groq", + "langchain_google_genai", ] ALL_SERIALIZABLE_MAPPINGS = { diff --git a/libs/core/langchain_core/load/mapping.py b/libs/core/langchain_core/load/mapping.py index dccabf3dbff..0b63bf247d8 100644 --- a/libs/core/langchain_core/load/mapping.py +++ b/libs/core/langchain_core/load/mapping.py @@ -303,6 +303,11 @@ SERIALIZABLE_MAPPING: Dict[Tuple[str, ...], Tuple[str, ...]] = { "bedrock", "ChatBedrock", ), + ("langchain_google_genai", "chat_models", "ChatGoogleGenerativeAI"): ( + "langchain_google_genai", + "chat_models", + "ChatGoogleGenerativeAI", + ), ("langchain", "schema", "output", "ChatGenerationChunk"): ( "langchain_core", "outputs",