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]: allow known namespace load from path

Browse Source
This commit is contained in:
Bagatur
2024-09-11 17:37:24 -07:00
parent 28594567de
commit e1d194160a
1 changed files with 6 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -19,6 +19,7 @@ DEFAULT_NAMESPACES = [
"langchain_anthropic",
"langchain_groq",
"langchain_google_genai",
"langchain_aws",
]
ALL_SERIALIZABLE_MAPPINGS = {
@@ -96,16 +97,11 @@ class Reviver:
if len(namespace) == 1 and namespace[0] == "langchain":
raise ValueError(f"Invalid namespace: {value}")
# If namespace is in known namespaces, try to use mapping
if namespace[0] in DEFAULT_NAMESPACES:
# Get the importable path
key = tuple(namespace + [name])
if key not in ALL_SERIALIZABLE_MAPPINGS:
raise ValueError(
"Trying to deserialize something that cannot "
"be deserialized in current version of langchain-core: "
f"{key}"
)
# If namespace is in mapping, used custom path
if (
namespace[0] in DEFAULT_NAMESPACES
and (key := tuple(namespace + [name])) in ALL_SERIALIZABLE_MAPPINGS
):
import_path = ALL_SERIALIZABLE_MAPPINGS[key]
# Split into module and name
import_dir, import_obj = import_path[:-1], import_path[-1]