github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-27 06:18:05 +00:00
Code Issues Packages Projects Releases Wiki Activity

core[patch]: add google genai to serialization (#26154)

Browse Source
This commit is contained in:
ccurme
2024-09-06 12:54:16 -04:00
committed by GitHub
parent 8a3a9c8968
commit b74546a458
2 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
libs/core/langchain_core/load/load.py
View File

@@ -18,6 +18,7 @@ DEFAULT_NAMESPACES = [
"langchain_community", "langchain_community",
"langchain_anthropic", "langchain_anthropic",
"langchain_groq", "langchain_groq",
"langchain_google_genai",
] ]
ALL_SERIALIZABLE_MAPPINGS = { ALL_SERIALIZABLE_MAPPINGS = {

5
libs/core/langchain_core/load/mapping.py
View File

@@ -303,6 +303,11 @@ SERIALIZABLE_MAPPING: Dict[Tuple[str, ...], Tuple[str, ...]] = {
"bedrock", "bedrock",
"ChatBedrock", "ChatBedrock",
), ),
("langchain_google_genai", "chat_models", "ChatGoogleGenerativeAI"): (
"langchain_google_genai",
"chat_models",
"ChatGoogleGenerativeAI",
),
("langchain", "schema", "output", "ChatGenerationChunk"): ( ("langchain", "schema", "output", "ChatGenerationChunk"): (
"langchain_core", "langchain_core",
"outputs", "outputs",