jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

add variable to store unformatted data

Browse source
This commit is contained in:
italojohnny 2024-05-29 16:08:47 -03:00
commit b091c237b6
3 changed files with 12 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/backend/base/langflow/graph/graph/base.py
View file

@ -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) 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: if vertex.result is not None:
params = vertex._built_object_repr() params = vertex.artifacts_raw
log_type = vertex.artifacts_type log_type = vertex.artifacts_type
valid = True valid = True
result_dict = vertex.result result_dict = vertex.result

2
src/backend/base/langflow/graph/vertex/base.py
View file

@ -63,6 +63,7 @@ class Vertex:
self._built_result = None self._built_result = None
self._built = False self._built = False
self.artifacts: Dict[str, Any] = {} self.artifacts: Dict[str, Any] = {}
self.artifacts_raw: Any = None
self.artifacts_type: Optional[str] = None self.artifacts_type: Optional[str] = None
self.steps: List[Callable] = [self._build] self.steps: List[Callable] = [self._build]
self.steps_ran: List[Callable] = [] self.steps_ran: List[Callable] = []
@ -626,6 +627,7 @@ class Vertex:
self._built_object, self.artifacts = result self._built_object, self.artifacts = result
elif len(result) == 3: elif len(result) == 3:
self._custom_component, self._built_object, self.artifacts = result 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 self.artifacts_type = self.artifacts.get("type") or ArtifactType.UNKNOWN.value
else: else:

10
src/backend/base/langflow/interface/initialize/loading.py
View file

@ -126,5 +126,13 @@ async def instantiate_custom_component(params, user_id, vertex, fallback_to_env_
custom_repr = build_result custom_repr = build_result
if not isinstance(custom_repr, str): if not isinstance(custom_repr, str):
custom_repr = str(custom_repr) 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 return custom_component, build_result, artifact