add variable to store unformatted data
This commit is contained in:
parent
c1cf8fe564
commit
b091c237b6
3 changed files with 12 additions and 2 deletions
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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:
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue