move function call

This commit is contained in:
italojohnny 2024-06-18 02:54:22 -03:00
commit c8f29d0f02
2 changed files with 2 additions and 3 deletions

View file

@ -610,6 +610,7 @@ class Vertex:
fallback_to_env_vars=fallback_to_env_vars,
vertex=self,
)
self.logs = build_logs(self, result)
self._update_built_object_and_artifacts(result)
except Exception as exc:
logger.exception(exc)
@ -628,7 +629,6 @@ class Vertex:
self.artifacts_raw = self.artifacts.get("raw", None)
self.artifacts_type = self.artifacts.get("type", None) or ArtifactType.UNKNOWN.value
self.artifacts = {self.outputs[0]["name"]: self.artifacts}
self.logs = build_logs(self)
else:
self._built_object = result

View file

@ -10,7 +10,7 @@ from langflow.graph.utils import UnbuiltObject, serialize_field
from langflow.graph.vertex.base import Vertex
from langflow.schema import Data
from langflow.schema.artifact import ArtifactType
from langflow.schema.schema import INPUT_FIELD_NAME, build_logs
from langflow.schema.schema import INPUT_FIELD_NAME
from langflow.services.monitor.utils import log_transaction, log_vertex_build
from langflow.utils.schemas import ChatOutputResponse, DataOutputResponse
from langflow.utils.util import unescape_string
@ -48,7 +48,6 @@ class ComponentVertex(Vertex):
for key in self.artifacts:
self.artifacts_raw[key] = self.artifacts[key].get("raw", None)
self.artifacts_type[key] = self.artifacts[key].get("type", None) or ArtifactType.UNKNOWN.value
self.logs = build_logs(self)
else:
self._built_object = result