griptape-nodes 0.35.0__py3-none-any.whl → 0.35.1__py3-none-any.whl

griptape_nodes/retained_mode/managers/workflow_manager.py

@@ -15,6 +15,7 @@ from dataclasses import dataclass, field, fields, is_dataclass
 from datetime import UTC, datetime
 from enum import StrEnum
 from importlib import resources
+from inspect import getmodule, isclass
 from pathlib import Path
 from typing import TYPE_CHECKING, Any, ClassVar, Literal, NamedTuple, TypeVar, cast
 from urllib.parse import urljoin
@@ -1132,6 +1133,9 @@ class WorkflowManager:
 
         import_recorder.add_import("pickle")
 
+        # Get the list of manually-curated, globally available modules
+        global_modules_set = {"builtins", "__main__"}
+
         # Serialize the unique values as pickled strings.
         unique_parameter_dict = {}
         for uuid, unique_parameter_value in unique_parameter_uuid_to_values.items():
@@ -1140,6 +1144,13 @@ class WorkflowManager:
             unique_parameter_byte_str = unique_parameter_bytes.decode("latin1")
             unique_parameter_dict[uuid] = unique_parameter_byte_str
 
+            # Add import for the unique parameter value's class/module. But not globals.
+            value_type = type(unique_parameter_value)
+            if isclass(value_type):
+                module = getmodule(value_type)
+                if module and module.__name__ not in global_modules_set:
+                    import_recorder.add_from_import(module.__name__, value_type.__name__)
+
         # Generate a comment explaining what we're doing:
         comment_text = (
             "\n"

{griptape_nodes-0.35.0.dist-info → griptape_nodes-0.35.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: griptape-nodes
-Version: 0.35.0
+Version: 0.35.1
 Summary: Add your description here
 License-File: LICENSE
 Requires-Python: ~=3.12

{griptape_nodes-0.35.0.dist-info → griptape_nodes-0.35.1.dist-info}/RECORD

@@ -67,7 +67,7 @@ griptape_nodes/retained_mode/managers/os_manager.py,sha256=zOYcuuTMjlFQm-Hxri0Ta
 griptape_nodes/retained_mode/managers/secrets_manager.py,sha256=wx69EhWs4EiYSPDhYOhOpU0KE_aMMSf4FWt6AT3cLKg,5829
 griptape_nodes/retained_mode/managers/settings.py,sha256=NGRonV8do52zNglK0sD75bglBF6rGgn244wtaQwwyls,4379
 griptape_nodes/retained_mode/managers/static_files_manager.py,sha256=4KcOE5WRNKiLBxeFN_tDmdgmnrNq_wcgpmOdr-b6Fe0,6722
-griptape_nodes/retained_mode/managers/workflow_manager.py,sha256=GJTD2HAZbrl6ZToBQTREhecxaG2p7irvmV9yKIFrxK0,102386
+griptape_nodes/retained_mode/managers/workflow_manager.py,sha256=SXRFJmXnLjEOwPxLUrA8fH1YbGt7c6mLwbQO_-HN1xQ,102941
 griptape_nodes/traits/__init__.py,sha256=bTLXTiZTJz2z15RRLoPI4nvLnNW9FiLcKL_2pT4E10g,32
 griptape_nodes/traits/add_param_button.py,sha256=Xtc39SfvYY6JDjhxu2wmdeaHW2X4LkpNAQufSTCP3hY,655
 griptape_nodes/traits/button.py,sha256=rpQQ77ASKH3YJabc5WWw9DpEHQiMg1jx0i0sdkwH-SA,611
@@ -84,8 +84,8 @@ griptape_nodes/updater/__main__.py,sha256=lhMMctbBeni57ximitvkwqJ1TQzk79Jg9w0rkK
 griptape_nodes/utils/__init__.py,sha256=5-NvqVvxiwuRx2UgyzenuLFXyodvfDqJ_dzcrgZpPOk,33
 griptape_nodes/utils/dict_utils.py,sha256=kyli5MOfzSor0a8lmXfYBVS_NXGf6UK1SM6k_Qn5bs8,6889
 griptape_nodes/utils/metaclasses.py,sha256=RSKyAvZppH_hbMugmUmIr8QqOg1sYY4uRNfQybD4O8Q,287
-griptape_nodes-0.35.0.dist-info/METADATA,sha256=AEHDD5lbZAOp6_U0_HDurrni4Cp7XljAouXlU3T9Vik,3925
-griptape_nodes-0.35.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-griptape_nodes-0.35.0.dist-info/entry_points.txt,sha256=Q8V3Ydy7PNfJgTjNEVkPui25AwZ4HJ3kZCU3oh4mq-c,81
-griptape_nodes-0.35.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-griptape_nodes-0.35.0.dist-info/RECORD,,
+griptape_nodes-0.35.1.dist-info/METADATA,sha256=_ZMgtpHkpUBz91eWirZ-Irr9nm3hZgxlHfcnn-uD1ew,3925
+griptape_nodes-0.35.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+griptape_nodes-0.35.1.dist-info/entry_points.txt,sha256=Q8V3Ydy7PNfJgTjNEVkPui25AwZ4HJ3kZCU3oh4mq-c,81
+griptape_nodes-0.35.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+griptape_nodes-0.35.1.dist-info/RECORD,,