Merge pull request #1474 from Jaseci-Labs/splice-orc-update

Splice orc: Support for from module import ClassName Syntax
Jaseci-Labs · Nov 29, 2024 · 816ff39 · 816ff39
2 parents feadbae + f36e878
commit 816ff39
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/jac-splice-orc/jac_splice_orc/plugin/splice_plugin.py b/jac-splice-orc/jac_splice_orc/plugin/splice_plugin.py
@@ -447,7 +447,16 @@ def jac_import(
             remote_module_proxy = proxy.get_module_proxy(
                 module_name=target, module_config=module_config[target]
             )
-            return (remote_module_proxy,)
+            if items:
+                imported_items = []
+                for item_name, item_alias in items.items():
+                    item = getattr(remote_module_proxy, item_name)
+                    if item_alias:
+                        item.__name__ = item_alias
+                    imported_items.append(item)
+                return tuple(imported_items)
+            else:
+                return (remote_module_proxy,)
         spec = ImportPathSpec(
             target,
             base_path,