From 66fb0b1f35469611316d2395248a35e5476453c2 Mon Sep 17 00:00:00 2001 From: Erick Friis Date: Thu, 18 Apr 2024 11:08:40 -0700 Subject: [PATCH] core: fix fireworks mapping (#20613) --- libs/core/langchain_core/load/mapping.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/libs/core/langchain_core/load/mapping.py b/libs/core/langchain_core/load/mapping.py index 8708331d652..417d54e35e5 100644 --- a/libs/core/langchain_core/load/mapping.py +++ b/libs/core/langchain_core/load/mapping.py @@ -246,9 +246,8 @@ SERIALIZABLE_MAPPING: Dict[Tuple[str, ...], Tuple[str, ...]] = { "ChatAnthropic", ), ("langchain", "chat_models", "fireworks", "ChatFireworks"): ( - "langchain", + "langchain_fireworks", "chat_models", - "fireworks", "ChatFireworks", ), ("langchain", "chat_models", "google_palm", "ChatGooglePalm"): ( @@ -317,9 +316,8 @@ SERIALIZABLE_MAPPING: Dict[Tuple[str, ...], Tuple[str, ...]] = { "BedrockLLM", ), ("langchain", "llms", "fireworks", "Fireworks"): ( - "langchain", + "langchain_fireworks", "llms", - "fireworks", "Fireworks", ), ("langchain", "llms", "google_palm", "GooglePalm"): (