core: fix fireworks mapping (#20613)

6 months ago · 66fb0b1f35
parent e786da7774
commit 66fb0b1f35
1 changed files with 2 additions and 4 deletions
--- 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"): (