github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-22 14:49:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

allow other namespaces (#14606)

Browse Source
This commit is contained in:
Harrison Chase 2023-12-12 09:09:59 -08:00 committed by GitHub
parent ce61a8ca98
commit ad8d8f71aa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 21 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -63,6 +63,8 @@ class Reviver:
if len(namespace) == 1 and namespace[0] == "langchain": if len(namespace) == 1 and namespace[0] == "langchain":
raise ValueError(f"Invalid namespace: {value}") 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 # Get the importable path
key = tuple(namespace + [name]) key = tuple(namespace + [name])
if key not in SERIALIZABLE_MAPPING: if key not in SERIALIZABLE_MAPPING:
@ -78,6 +80,10 @@ class Reviver:
mod = importlib.import_module(".".join(import_dir)) mod = importlib.import_module(".".join(import_dir))
# Import class # Import class
cls = getattr(mod, import_obj) cls = getattr(mod, import_obj)
# Otherwise, load by path
else:
mod = importlib.import_module(".".join(namespace))
cls = getattr(mod, name)
# The class must be a subclass of Serializable. # The class must be a subclass of Serializable.
if not issubclass(cls, Serializable): if not issubclass(cls, Serializable):