coco-framework 0.1.0__py3-none-any.whl

coco_core/load/mapping.py

@@ -0,0 +1,1097 @@
+"""Serialization mapping.
+
+This file contains a mapping between the `lc_namespace` path for a given
+subclass that implements from `Serializable` to the namespace
+where that class is actually located.
+
+This mapping helps maintain the ability to serialize and deserialize
+well-known Coco objects even if they are moved around in the codebase
+across different Coco versions.
+
+For example, the code for the `AIMessage` class is located in
+`coco_core.messages.ai.AIMessage`. This message is associated with the
+`lc_namespace` of `["coco", "schema", "messages", "AIMessage"]`,
+because this code was originally in `coco.schema.messages.AIMessage`.
+
+The mapping allows us to deserialize an `AIMessage` created with an older
+version of Coco where the code was in a different location.
+"""
+
+# First value is the value that it is serialized as
+# Second value is the path to load it from
+SERIALIZABLE_MAPPING: dict[tuple[str, ...], tuple[str, ...]] = {
+    ("coco", "schema", "messages", "AIMessage"): (
+        "coco_core",
+        "messages",
+        "ai",
+        "AIMessage",
+    ),
+    ("coco", "schema", "messages", "AIMessageChunk"): (
+        "coco_core",
+        "messages",
+        "ai",
+        "AIMessageChunk",
+    ),
+    ("coco", "schema", "messages", "BaseMessage"): (
+        "coco_core",
+        "messages",
+        "base",
+        "BaseMessage",
+    ),
+    ("coco", "schema", "messages", "BaseMessageChunk"): (
+        "coco_core",
+        "messages",
+        "base",
+        "BaseMessageChunk",
+    ),
+    ("coco", "schema", "messages", "ChatMessage"): (
+        "coco_core",
+        "messages",
+        "chat",
+        "ChatMessage",
+    ),
+    ("coco", "schema", "messages", "FunctionMessage"): (
+        "coco_core",
+        "messages",
+        "function",
+        "FunctionMessage",
+    ),
+    ("coco", "schema", "messages", "HumanMessage"): (
+        "coco_core",
+        "messages",
+        "human",
+        "HumanMessage",
+    ),
+    ("coco", "schema", "messages", "SystemMessage"): (
+        "coco_core",
+        "messages",
+        "system",
+        "SystemMessage",
+    ),
+    ("coco", "schema", "messages", "ToolMessage"): (
+        "coco_core",
+        "messages",
+        "tool",
+        "ToolMessage",
+    ),
+    ("coco", "schema", "messages", "RemoveMessage"): (
+        "coco_core",
+        "messages",
+        "modifier",
+        "RemoveMessage",
+    ),
+    ("coco", "schema", "agent", "AgentAction"): (
+        "coco_core",
+        "agents",
+        "AgentAction",
+    ),
+    ("coco", "schema", "agent", "AgentFinish"): (
+        "coco_core",
+        "agents",
+        "AgentFinish",
+    ),
+    ("coco", "schema", "prompt_template", "BasePromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "base",
+        "BasePromptTemplate",
+    ),
+    ("coco", "chains", "llm", "LLMChain"): (
+        "coco",
+        "chains",
+        "llm",
+        "LLMChain",
+    ),
+    ("coco", "prompts", "prompt", "PromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "prompt",
+        "PromptTemplate",
+    ),
+    ("coco", "prompts", "chat", "MessagesPlaceholder"): (
+        "coco_core",
+        "prompts",
+        "chat",
+        "MessagesPlaceholder",
+    ),
+    ("coco", "llms", "openai", "OpenAI"): (
+        "coco_openai",
+        "llms",
+        "base",
+        "OpenAI",
+    ),
+    ("coco", "prompts", "chat", "ChatPromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "chat",
+        "ChatPromptTemplate",
+    ),
+    ("coco", "prompts", "chat", "HumanMessagePromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "chat",
+        "HumanMessagePromptTemplate",
+    ),
+    ("coco", "prompts", "chat", "SystemMessagePromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "chat",
+        "SystemMessagePromptTemplate",
+    ),
+    ("coco", "prompts", "image", "ImagePromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "image",
+        "ImagePromptTemplate",
+    ),
+    ("coco", "schema", "agent", "AgentActionMessageLog"): (
+        "coco_core",
+        "agents",
+        "AgentActionMessageLog",
+    ),
+    ("coco", "schema", "agent", "ToolAgentAction"): (
+        "coco",
+        "agents",
+        "output_parsers",
+        "tools",
+        "ToolAgentAction",
+    ),
+    ("coco", "prompts", "chat", "BaseMessagePromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "chat",
+        "BaseMessagePromptTemplate",
+    ),
+    ("coco", "schema", "output", "ChatGeneration"): (
+        "coco_core",
+        "outputs",
+        "chat_generation",
+        "ChatGeneration",
+    ),
+    ("coco", "schema", "output", "Generation"): (
+        "coco_core",
+        "outputs",
+        "generation",
+        "Generation",
+    ),
+    ("coco", "schema", "document", "Document"): (
+        "coco_core",
+        "documents",
+        "base",
+        "Document",
+    ),
+    ("coco", "output_parsers", "fix", "OutputFixingParser"): (
+        "coco",
+        "output_parsers",
+        "fix",
+        "OutputFixingParser",
+    ),
+    ("coco", "prompts", "chat", "AIMessagePromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "chat",
+        "AIMessagePromptTemplate",
+    ),
+    ("coco", "output_parsers", "regex", "RegexParser"): (
+        "coco",
+        "output_parsers",
+        "regex",
+        "RegexParser",
+    ),
+    ("coco", "schema", "runnable", "DynamicRunnable"): (
+        "coco_core",
+        "runnables",
+        "configurable",
+        "DynamicRunnable",
+    ),
+    ("coco", "schema", "prompt", "PromptValue"): (
+        "coco_core",
+        "prompt_values",
+        "PromptValue",
+    ),
+    ("coco", "schema", "runnable", "RunnableBinding"): (
+        "coco_core",
+        "runnables",
+        "base",
+        "RunnableBinding",
+    ),
+    ("coco", "schema", "runnable", "RunnableBranch"): (
+        "coco_core",
+        "runnables",
+        "branch",
+        "RunnableBranch",
+    ),
+    ("coco", "schema", "runnable", "RunnableWithFallbacks"): (
+        "coco_core",
+        "runnables",
+        "fallbacks",
+        "RunnableWithFallbacks",
+    ),
+    ("coco", "schema", "output_parser", "StrOutputParser"): (
+        "coco_core",
+        "output_parsers",
+        "string",
+        "StrOutputParser",
+    ),
+    ("coco", "chat_models", "openai", "ChatOpenAI"): (
+        "coco_openai",
+        "chat_models",
+        "base",
+        "ChatOpenAI",
+    ),
+    ("coco", "output_parsers", "list", "CommaSeparatedListOutputParser"): (
+        "coco_core",
+        "output_parsers",
+        "list",
+        "CommaSeparatedListOutputParser",
+    ),
+    ("coco", "schema", "runnable", "RunnableParallel"): (
+        "coco_core",
+        "runnables",
+        "base",
+        "RunnableParallel",
+    ),
+    ("coco", "chat_models", "azure_openai", "AzureChatOpenAI"): (
+        "coco_openai",
+        "chat_models",
+        "azure",
+        "AzureChatOpenAI",
+    ),
+    ("coco", "chat_models", "bedrock", "BedrockChat"): (
+        "coco_aws",
+        "chat_models",
+        "bedrock",
+        "ChatBedrock",
+    ),
+    ("coco", "chat_models", "anthropic", "ChatAnthropic"): (
+        "coco_anthropic",
+        "chat_models",
+        "ChatAnthropic",
+    ),
+    ("coco_groq", "chat_models", "ChatGroq"): (
+        "coco_groq",
+        "chat_models",
+        "ChatGroq",
+    ),
+    ("coco_openrouter", "chat_models", "ChatOpenRouter"): (
+        "coco_openrouter",
+        "chat_models",
+        "ChatOpenRouter",
+    ),
+    ("coco_xai", "chat_models", "ChatXAI"): (
+        "coco_xai",
+        "chat_models",
+        "ChatXAI",
+    ),
+    ("coco_baseten", "chat_models", "ChatBaseten"): (
+        "coco_baseten",
+        "chat_models",
+        "ChatBaseten",
+    ),
+    ("coco", "chat_models", "fireworks", "ChatFireworks"): (
+        "coco_fireworks",
+        "chat_models",
+        "ChatFireworks",
+    ),
+    ("coco", "chat_models", "google_palm", "ChatGooglePalm"): (
+        "coco",
+        "chat_models",
+        "google_palm",
+        "ChatGooglePalm",
+    ),
+    ("coco", "chat_models", "vertexai", "ChatVertexAI"): (
+        "coco_google_vertexai",
+        "chat_models",
+        "ChatVertexAI",
+    ),
+    ("coco", "chat_models", "mistralai", "ChatMistralAI"): (
+        "coco_mistralai",
+        "chat_models",
+        "ChatMistralAI",
+    ),
+    ("coco", "chat_models", "anthropic_bedrock", "ChatAnthropicBedrock"): (
+        "coco_aws",
+        "chat_models",
+        "anthropic",
+        "ChatAnthropicBedrock",
+    ),
+    ("coco", "chat_models", "bedrock", "ChatBedrock"): (
+        "coco_aws",
+        "chat_models",
+        "bedrock",
+        "ChatBedrock",
+    ),
+    ("coco_aws", "chat_models", "ChatBedrockConverse"): (
+        "coco_aws",
+        "chat_models",
+        "bedrock_converse",
+        "ChatBedrockConverse",
+    ),
+    ("coco_google_genai", "chat_models", "ChatGoogleGenerativeAI"): (
+        "coco_google_genai",
+        "chat_models",
+        "ChatGoogleGenerativeAI",
+    ),
+    ("coco", "schema", "output", "ChatGenerationChunk"): (
+        "coco_core",
+        "outputs",
+        "chat_generation",
+        "ChatGenerationChunk",
+    ),
+    ("coco", "schema", "messages", "ChatMessageChunk"): (
+        "coco_core",
+        "messages",
+        "chat",
+        "ChatMessageChunk",
+    ),
+    ("coco", "schema", "messages", "HumanMessageChunk"): (
+        "coco_core",
+        "messages",
+        "human",
+        "HumanMessageChunk",
+    ),
+    ("coco", "schema", "messages", "FunctionMessageChunk"): (
+        "coco_core",
+        "messages",
+        "function",
+        "FunctionMessageChunk",
+    ),
+    ("coco", "schema", "messages", "SystemMessageChunk"): (
+        "coco_core",
+        "messages",
+        "system",
+        "SystemMessageChunk",
+    ),
+    ("coco", "schema", "messages", "ToolMessageChunk"): (
+        "coco_core",
+        "messages",
+        "tool",
+        "ToolMessageChunk",
+    ),
+    ("coco", "schema", "output", "GenerationChunk"): (
+        "coco_core",
+        "outputs",
+        "generation",
+        "GenerationChunk",
+    ),
+    ("coco", "llms", "openai", "BaseOpenAI"): (
+        "coco",
+        "llms",
+        "openai",
+        "BaseOpenAI",
+    ),
+    ("coco", "llms", "bedrock", "Bedrock"): (
+        "coco_aws",
+        "llms",
+        "bedrock",
+        "BedrockLLM",
+    ),
+    ("coco", "llms", "bedrock", "BedrockLLM"): (
+        "coco_aws",
+        "llms",
+        "bedrock",
+        "BedrockLLM",
+    ),
+    ("coco", "llms", "fireworks", "Fireworks"): (
+        "coco_fireworks",
+        "llms",
+        "Fireworks",
+    ),
+    ("coco", "llms", "google_palm", "GooglePalm"): (
+        "coco",
+        "llms",
+        "google_palm",
+        "GooglePalm",
+    ),
+    ("coco", "llms", "openai", "AzureOpenAI"): (
+        "coco_openai",
+        "llms",
+        "azure",
+        "AzureOpenAI",
+    ),
+    ("coco", "llms", "replicate", "Replicate"): (
+        "coco",
+        "llms",
+        "replicate",
+        "Replicate",
+    ),
+    ("coco", "llms", "vertexai", "VertexAI"): (
+        "coco_vertexai",
+        "llms",
+        "VertexAI",
+    ),
+    ("coco", "output_parsers", "combining", "CombiningOutputParser"): (
+        "coco",
+        "output_parsers",
+        "combining",
+        "CombiningOutputParser",
+    ),
+    ("coco", "schema", "prompt_template", "BaseChatPromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "chat",
+        "BaseChatPromptTemplate",
+    ),
+    ("coco", "prompts", "chat", "ChatMessagePromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "chat",
+        "ChatMessagePromptTemplate",
+    ),
+    ("coco", "prompts", "few_shot_with_templates", "FewShotPromptWithTemplates"): (
+        "coco_core",
+        "prompts",
+        "few_shot_with_templates",
+        "FewShotPromptWithTemplates",
+    ),
+    ("coco", "prompts", "pipeline"): (
+        "coco_core",
+        "prompts",
+        "pipeline",
+    ),
+    ("coco", "prompts", "base", "StringPromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "string",
+        "StringPromptTemplate",
+    ),
+    ("coco", "prompts", "base", "StringPromptValue"): (
+        "coco_core",
+        "prompt_values",
+        "StringPromptValue",
+    ),
+    ("coco", "prompts", "chat", "BaseStringMessagePromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "chat",
+        "BaseStringMessagePromptTemplate",
+    ),
+    ("coco", "prompts", "chat", "ChatPromptValue"): (
+        "coco_core",
+        "prompt_values",
+        "ChatPromptValue",
+    ),
+    ("coco", "prompts", "chat", "ChatPromptValueConcrete"): (
+        "coco_core",
+        "prompt_values",
+        "ChatPromptValueConcrete",
+    ),
+    ("coco", "schema", "runnable", "HubRunnable"): (
+        "coco",
+        "runnables",
+        "hub",
+        "HubRunnable",
+    ),
+    ("coco", "schema", "runnable", "RunnableBindingBase"): (
+        "coco_core",
+        "runnables",
+        "base",
+        "RunnableBindingBase",
+    ),
+    ("coco", "schema", "runnable", "OpenAIFunctionsRouter"): (
+        "coco",
+        "runnables",
+        "openai_functions",
+        "OpenAIFunctionsRouter",
+    ),
+    ("coco", "schema", "runnable", "RouterRunnable"): (
+        "coco_core",
+        "runnables",
+        "router",
+        "RouterRunnable",
+    ),
+    ("coco", "schema", "runnable", "RunnablePassthrough"): (
+        "coco_core",
+        "runnables",
+        "passthrough",
+        "RunnablePassthrough",
+    ),
+    ("coco", "schema", "runnable", "RunnableSequence"): (
+        "coco_core",
+        "runnables",
+        "base",
+        "RunnableSequence",
+    ),
+    ("coco", "schema", "runnable", "RunnableEach"): (
+        "coco_core",
+        "runnables",
+        "base",
+        "RunnableEach",
+    ),
+    ("coco", "schema", "runnable", "RunnableEachBase"): (
+        "coco_core",
+        "runnables",
+        "base",
+        "RunnableEachBase",
+    ),
+    ("coco", "schema", "runnable", "RunnableConfigurableAlternatives"): (
+        "coco_core",
+        "runnables",
+        "configurable",
+        "RunnableConfigurableAlternatives",
+    ),
+    ("coco", "schema", "runnable", "RunnableConfigurableFields"): (
+        "coco_core",
+        "runnables",
+        "configurable",
+        "RunnableConfigurableFields",
+    ),
+    ("coco", "schema", "runnable", "RunnableWithMessageHistory"): (
+        "coco_core",
+        "runnables",
+        "history",
+        "RunnableWithMessageHistory",
+    ),
+    ("coco", "schema", "runnable", "RunnableAssign"): (
+        "coco_core",
+        "runnables",
+        "passthrough",
+        "RunnableAssign",
+    ),
+    ("coco", "schema", "runnable", "RunnablePick"): (
+        "coco_core",
+        "runnables",
+        "passthrough",
+        "RunnablePick",
+    ),
+    ("coco", "schema", "runnable", "RunnableRetry"): (
+        "coco_core",
+        "runnables",
+        "retry",
+        "RunnableRetry",
+    ),
+    ("coco_core", "prompts", "structured", "StructuredPrompt"): (
+        "coco_core",
+        "prompts",
+        "structured",
+        "StructuredPrompt",
+    ),
+    ("coco_core", "prompts", "message", "_DictMessagePromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "dict",
+        "DictPromptTemplate",
+    ),
+}
+
+# Needed for backwards compatibility for old versions of Coco where things
+# Were in different place
+_OG_SERIALIZABLE_MAPPING: dict[tuple[str, ...], tuple[str, ...]] = {
+    ("coco", "schema", "AIMessage"): (
+        "coco_core",
+        "messages",
+        "ai",
+        "AIMessage",
+    ),
+    ("coco", "schema", "ChatMessage"): (
+        "coco_core",
+        "messages",
+        "chat",
+        "ChatMessage",
+    ),
+    ("coco", "schema", "FunctionMessage"): (
+        "coco_core",
+        "messages",
+        "function",
+        "FunctionMessage",
+    ),
+    ("coco", "schema", "HumanMessage"): (
+        "coco_core",
+        "messages",
+        "human",
+        "HumanMessage",
+    ),
+    ("coco", "schema", "SystemMessage"): (
+        "coco_core",
+        "messages",
+        "system",
+        "SystemMessage",
+    ),
+    ("coco", "schema", "prompt_template", "ImagePromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "image",
+        "ImagePromptTemplate",
+    ),
+    ("coco", "schema", "agent", "OpenAIToolAgentAction"): (
+        "coco",
+        "agents",
+        "output_parsers",
+        "openai_tools",
+        "OpenAIToolAgentAction",
+    ),
+}
+
+# Needed for backwards compatibility for a few versions where we serialized
+# with coco_core paths.
+OLD_CORE_NAMESPACES_MAPPING: dict[tuple[str, ...], tuple[str, ...]] = {
+    ("coco_core", "messages", "ai", "AIMessage"): (
+        "coco_core",
+        "messages",
+        "ai",
+        "AIMessage",
+    ),
+    ("coco_core", "messages", "ai", "AIMessageChunk"): (
+        "coco_core",
+        "messages",
+        "ai",
+        "AIMessageChunk",
+    ),
+    ("coco_core", "messages", "base", "BaseMessage"): (
+        "coco_core",
+        "messages",
+        "base",
+        "BaseMessage",
+    ),
+    ("coco_core", "messages", "base", "BaseMessageChunk"): (
+        "coco_core",
+        "messages",
+        "base",
+        "BaseMessageChunk",
+    ),
+    ("coco_core", "messages", "chat", "ChatMessage"): (
+        "coco_core",
+        "messages",
+        "chat",
+        "ChatMessage",
+    ),
+    ("coco_core", "messages", "function", "FunctionMessage"): (
+        "coco_core",
+        "messages",
+        "function",
+        "FunctionMessage",
+    ),
+    ("coco_core", "messages", "human", "HumanMessage"): (
+        "coco_core",
+        "messages",
+        "human",
+        "HumanMessage",
+    ),
+    ("coco_core", "messages", "system", "SystemMessage"): (
+        "coco_core",
+        "messages",
+        "system",
+        "SystemMessage",
+    ),
+    ("coco_core", "messages", "tool", "ToolMessage"): (
+        "coco_core",
+        "messages",
+        "tool",
+        "ToolMessage",
+    ),
+    ("coco_core", "agents", "AgentAction"): (
+        "coco_core",
+        "agents",
+        "AgentAction",
+    ),
+    ("coco_core", "agents", "AgentFinish"): (
+        "coco_core",
+        "agents",
+        "AgentFinish",
+    ),
+    ("coco_core", "prompts", "base", "BasePromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "base",
+        "BasePromptTemplate",
+    ),
+    ("coco_core", "prompts", "prompt", "PromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "prompt",
+        "PromptTemplate",
+    ),
+    ("coco_core", "prompts", "chat", "MessagesPlaceholder"): (
+        "coco_core",
+        "prompts",
+        "chat",
+        "MessagesPlaceholder",
+    ),
+    ("coco_core", "prompts", "chat", "ChatPromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "chat",
+        "ChatPromptTemplate",
+    ),
+    ("coco_core", "prompts", "chat", "HumanMessagePromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "chat",
+        "HumanMessagePromptTemplate",
+    ),
+    ("coco_core", "prompts", "chat", "SystemMessagePromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "chat",
+        "SystemMessagePromptTemplate",
+    ),
+    ("coco_core", "agents", "AgentActionMessageLog"): (
+        "coco_core",
+        "agents",
+        "AgentActionMessageLog",
+    ),
+    ("coco_core", "prompts", "chat", "BaseMessagePromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "chat",
+        "BaseMessagePromptTemplate",
+    ),
+    ("coco_core", "outputs", "chat_generation", "ChatGeneration"): (
+        "coco_core",
+        "outputs",
+        "chat_generation",
+        "ChatGeneration",
+    ),
+    ("coco_core", "outputs", "generation", "Generation"): (
+        "coco_core",
+        "outputs",
+        "generation",
+        "Generation",
+    ),
+    ("coco_core", "documents", "base", "Document"): (
+        "coco_core",
+        "documents",
+        "base",
+        "Document",
+    ),
+    ("coco_core", "prompts", "chat", "AIMessagePromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "chat",
+        "AIMessagePromptTemplate",
+    ),
+    ("coco_core", "runnables", "configurable", "DynamicRunnable"): (
+        "coco_core",
+        "runnables",
+        "configurable",
+        "DynamicRunnable",
+    ),
+    ("coco_core", "prompt_values", "PromptValue"): (
+        "coco_core",
+        "prompt_values",
+        "PromptValue",
+    ),
+    ("coco_core", "runnables", "base", "RunnableBinding"): (
+        "coco_core",
+        "runnables",
+        "base",
+        "RunnableBinding",
+    ),
+    ("coco_core", "runnables", "branch", "RunnableBranch"): (
+        "coco_core",
+        "runnables",
+        "branch",
+        "RunnableBranch",
+    ),
+    ("coco_core", "runnables", "fallbacks", "RunnableWithFallbacks"): (
+        "coco_core",
+        "runnables",
+        "fallbacks",
+        "RunnableWithFallbacks",
+    ),
+    ("coco_core", "output_parsers", "string", "StrOutputParser"): (
+        "coco_core",
+        "output_parsers",
+        "string",
+        "StrOutputParser",
+    ),
+    ("coco_core", "output_parsers", "list", "CommaSeparatedListOutputParser"): (
+        "coco_core",
+        "output_parsers",
+        "list",
+        "CommaSeparatedListOutputParser",
+    ),
+    ("coco_core", "runnables", "base", "RunnableParallel"): (
+        "coco_core",
+        "runnables",
+        "base",
+        "RunnableParallel",
+    ),
+    ("coco_core", "outputs", "chat_generation", "ChatGenerationChunk"): (
+        "coco_core",
+        "outputs",
+        "chat_generation",
+        "ChatGenerationChunk",
+    ),
+    ("coco_core", "messages", "chat", "ChatMessageChunk"): (
+        "coco_core",
+        "messages",
+        "chat",
+        "ChatMessageChunk",
+    ),
+    ("coco_core", "messages", "human", "HumanMessageChunk"): (
+        "coco_core",
+        "messages",
+        "human",
+        "HumanMessageChunk",
+    ),
+    ("coco_core", "messages", "function", "FunctionMessageChunk"): (
+        "coco_core",
+        "messages",
+        "function",
+        "FunctionMessageChunk",
+    ),
+    ("coco_core", "messages", "system", "SystemMessageChunk"): (
+        "coco_core",
+        "messages",
+        "system",
+        "SystemMessageChunk",
+    ),
+    ("coco_core", "messages", "tool", "ToolMessageChunk"): (
+        "coco_core",
+        "messages",
+        "tool",
+        "ToolMessageChunk",
+    ),
+    ("coco_core", "outputs", "generation", "GenerationChunk"): (
+        "coco_core",
+        "outputs",
+        "generation",
+        "GenerationChunk",
+    ),
+    ("coco_core", "prompts", "chat", "BaseChatPromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "chat",
+        "BaseChatPromptTemplate",
+    ),
+    ("coco_core", "prompts", "chat", "ChatMessagePromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "chat",
+        "ChatMessagePromptTemplate",
+    ),
+    (
+        "coco_core",
+        "prompts",
+        "few_shot_with_templates",
+        "FewShotPromptWithTemplates",
+    ): (
+        "coco_core",
+        "prompts",
+        "few_shot_with_templates",
+        "FewShotPromptWithTemplates",
+    ),
+    ("coco_core", "prompts", "pipeline"): (
+        "coco_core",
+        "prompts",
+        "pipeline",
+    ),
+    ("coco_core", "prompts", "string", "StringPromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "string",
+        "StringPromptTemplate",
+    ),
+    ("coco_core", "prompt_values", "StringPromptValue"): (
+        "coco_core",
+        "prompt_values",
+        "StringPromptValue",
+    ),
+    ("coco_core", "prompts", "chat", "BaseStringMessagePromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "chat",
+        "BaseStringMessagePromptTemplate",
+    ),
+    ("coco_core", "prompt_values", "ChatPromptValue"): (
+        "coco_core",
+        "prompt_values",
+        "ChatPromptValue",
+    ),
+    ("coco_core", "prompt_values", "ChatPromptValueConcrete"): (
+        "coco_core",
+        "prompt_values",
+        "ChatPromptValueConcrete",
+    ),
+    ("coco_core", "runnables", "base", "RunnableBindingBase"): (
+        "coco_core",
+        "runnables",
+        "base",
+        "RunnableBindingBase",
+    ),
+    ("coco_core", "runnables", "router", "RouterRunnable"): (
+        "coco_core",
+        "runnables",
+        "router",
+        "RouterRunnable",
+    ),
+    ("coco_core", "runnables", "passthrough", "RunnablePassthrough"): (
+        "coco_core",
+        "runnables",
+        "passthrough",
+        "RunnablePassthrough",
+    ),
+    ("coco_core", "runnables", "base", "RunnableSequence"): (
+        "coco_core",
+        "runnables",
+        "base",
+        "RunnableSequence",
+    ),
+    ("coco_core", "runnables", "base", "RunnableEach"): (
+        "coco_core",
+        "runnables",
+        "base",
+        "RunnableEach",
+    ),
+    ("coco_core", "runnables", "base", "RunnableEachBase"): (
+        "coco_core",
+        "runnables",
+        "base",
+        "RunnableEachBase",
+    ),
+    (
+        "coco_core",
+        "runnables",
+        "configurable",
+        "RunnableConfigurableAlternatives",
+    ): (
+        "coco_core",
+        "runnables",
+        "configurable",
+        "RunnableConfigurableAlternatives",
+    ),
+    ("coco_core", "runnables", "configurable", "RunnableConfigurableFields"): (
+        "coco_core",
+        "runnables",
+        "configurable",
+        "RunnableConfigurableFields",
+    ),
+    ("coco_core", "runnables", "history", "RunnableWithMessageHistory"): (
+        "coco_core",
+        "runnables",
+        "history",
+        "RunnableWithMessageHistory",
+    ),
+    ("coco_core", "runnables", "passthrough", "RunnableAssign"): (
+        "coco_core",
+        "runnables",
+        "passthrough",
+        "RunnableAssign",
+    ),
+    ("coco_core", "runnables", "passthrough", "RunnablePick"): (
+        "coco_core",
+        "runnables",
+        "passthrough",
+        "RunnablePick",
+    ),
+    ("coco_core", "runnables", "retry", "RunnableRetry"): (
+        "coco_core",
+        "runnables",
+        "retry",
+        "RunnableRetry",
+    ),
+}
+
+_JS_SERIALIZABLE_MAPPING: dict[tuple[str, ...], tuple[str, ...]] = {
+    ("coco_core", "messages", "AIMessage"): (
+        "coco_core",
+        "messages",
+        "ai",
+        "AIMessage",
+    ),
+    ("coco_core", "messages", "AIMessageChunk"): (
+        "coco_core",
+        "messages",
+        "ai",
+        "AIMessageChunk",
+    ),
+    ("coco_core", "messages", "BaseMessage"): (
+        "coco_core",
+        "messages",
+        "base",
+        "BaseMessage",
+    ),
+    ("coco_core", "messages", "BaseMessageChunk"): (
+        "coco_core",
+        "messages",
+        "base",
+        "BaseMessageChunk",
+    ),
+    ("coco_core", "messages", "ChatMessage"): (
+        "coco_core",
+        "messages",
+        "chat",
+        "ChatMessage",
+    ),
+    ("coco_core", "messages", "ChatMessageChunk"): (
+        "coco_core",
+        "messages",
+        "chat",
+        "ChatMessageChunk",
+    ),
+    ("coco_core", "messages", "FunctionMessage"): (
+        "coco_core",
+        "messages",
+        "function",
+        "FunctionMessage",
+    ),
+    ("coco_core", "messages", "FunctionMessageChunk"): (
+        "coco_core",
+        "messages",
+        "function",
+        "FunctionMessageChunk",
+    ),
+    ("coco_core", "messages", "HumanMessage"): (
+        "coco_core",
+        "messages",
+        "human",
+        "HumanMessage",
+    ),
+    ("coco_core", "messages", "HumanMessageChunk"): (
+        "coco_core",
+        "messages",
+        "human",
+        "HumanMessageChunk",
+    ),
+    ("coco_core", "messages", "SystemMessage"): (
+        "coco_core",
+        "messages",
+        "system",
+        "SystemMessage",
+    ),
+    ("coco_core", "messages", "SystemMessageChunk"): (
+        "coco_core",
+        "messages",
+        "system",
+        "SystemMessageChunk",
+    ),
+    ("coco_core", "messages", "ToolMessage"): (
+        "coco_core",
+        "messages",
+        "tool",
+        "ToolMessage",
+    ),
+    ("coco_core", "messages", "ToolMessageChunk"): (
+        "coco_core",
+        "messages",
+        "tool",
+        "ToolMessageChunk",
+    ),
+    ("coco_core", "prompts", "image", "ImagePromptTemplate"): (
+        "coco_core",
+        "prompts",
+        "image",
+        "ImagePromptTemplate",
+    ),
+    ("coco", "chat_models", "bedrock", "ChatBedrock"): (
+        "coco_aws",
+        "chat_models",
+        "ChatBedrock",
+    ),
+    ("coco", "chat_models", "google_genai", "ChatGoogleGenerativeAI"): (
+        "coco_google_genai",
+        "chat_models",
+        "ChatGoogleGenerativeAI",
+    ),
+    ("coco", "chat_models", "groq", "ChatGroq"): (
+        "coco_groq",
+        "chat_models",
+        "ChatGroq",
+    ),
+    ("coco", "chat_models", "bedrock", "BedrockChat"): (
+        "coco_aws",
+        "chat_models",
+        "ChatBedrock",
+    ),
+}