diff --git a/src/backend/base/langflow/graph/graph/base.py b/src/backend/base/langflow/graph/graph/base.py index 781b76141..789beb80d 100644 --- a/src/backend/base/langflow/graph/graph/base.py +++ b/src/backend/base/langflow/graph/graph/base.py @@ -734,7 +734,7 @@ class Graph: await vertex.build(user_id=user_id, inputs=inputs_dict,files=files, fallback_to_env_vars=fallback_to_env_vars) if vertex.result is not None: - params = vertex._built_object_repr() + params = vertex.artifacts_raw log_type = vertex.artifacts_type valid = True result_dict = vertex.result diff --git a/src/backend/base/langflow/graph/vertex/base.py b/src/backend/base/langflow/graph/vertex/base.py index 6ad5bc826..1498eddff 100644 --- a/src/backend/base/langflow/graph/vertex/base.py +++ b/src/backend/base/langflow/graph/vertex/base.py @@ -63,6 +63,7 @@ class Vertex: self._built_result = None self._built = False self.artifacts: Dict[str, Any] = {} + self.artifacts_raw: Any = None self.artifacts_type: Optional[str] = None self.steps: List[Callable] = [self._build] self.steps_ran: List[Callable] = [] @@ -626,6 +627,7 @@ class Vertex: self._built_object, self.artifacts = result elif len(result) == 3: self._custom_component, self._built_object, self.artifacts = result + self.artifacts_raw = self.artifacts.get("raw") self.artifacts_type = self.artifacts.get("type") or ArtifactType.UNKNOWN.value else: diff --git a/src/backend/base/langflow/interface/initialize/loading.py b/src/backend/base/langflow/interface/initialize/loading.py index e5f05c595..8eb2a4d9a 100644 --- a/src/backend/base/langflow/interface/initialize/loading.py +++ b/src/backend/base/langflow/interface/initialize/loading.py @@ -126,5 +126,13 @@ async def instantiate_custom_component(params, user_id, vertex, fallback_to_env_ custom_repr = build_result if not isinstance(custom_repr, str): custom_repr = str(custom_repr) - artifact = {"repr": custom_repr, "type": get_artifact_type(build_result)} + + raw = custom_component.repr_value + if hasattr(raw, "data"): + raw = raw.data + + elif hasattr(raw, "model_dump"): + raw = raw.model_dump() + + artifact = {"repr": custom_repr, "raw": raw, "type": get_artifact_type(custom_component.repr_value)} return custom_component, build_result, artifact