This commit is contained in:
Bagatur
2024-09-04 12:42:04 -07:00
parent 63ba12d8e0
commit 38674efac4
50 changed files with 3127 additions and 2088 deletions

View File

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

View File

@@ -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",