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

refactor: Update logs to outputs in CustomComponent and TracingService classes

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-21 19:57:49 -03:00
commit 4212b27800
2 changed files with 4 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

9
src/backend/base/langflow/custom/custom_component/custom_component.py
View file

@ -83,7 +83,7 @@ class CustomComponent(BaseComponent):
_flows_data: Optional[List[Data]] = None _flows_data: Optional[List[Data]] = None
_outputs: List[OutputLog] = [] _outputs: List[OutputLog] = []
_logs: List[Log] = [] _logs: List[Log] = []
_tracing_service: "TracingService" tracing_service: Optional["TracingService"] = None
def update_state(self, name: str, value: Any): def update_state(self, name: str, value: Any):
if not self.vertex: if not self.vertex:
@ -489,13 +489,10 @@ class CustomComponent(BaseComponent):
message (LoggableType | list[LoggableType]): The message to log. message (LoggableType | list[LoggableType]): The message to log.
""" """
if name is None: if name is None:
name = self.display_name if self.display_name else self.__class__.__name__ name = self.display_name
if hasattr(message, "model_dump") and isinstance(message, BaseModel):
message = message.model_dump()
log = Log(message=message, type=get_artifact_type(message), name=name) log = Log(message=message, type=get_artifact_type(message), name=name)
self._logs.append(log) self._logs.append(log)
if self.vertex: self.tracing_service.add_log(trace_name=self.vertex.id, log=log)
self._tracing_service.add_log(trace_name=self.vertex.id, log=log)
def post_code_processing(self, new_build_config: dict, current_build_config: dict): def post_code_processing(self, new_build_config: dict, current_build_config: dict):
""" """

2
src/backend/base/langflow/services/tracing/service.py
View file

@ -12,8 +12,8 @@ from loguru import logger
from langflow.schema.data import Data from langflow.schema.data import Data
from langflow.services.base import Service from langflow.services.base import Service
from langflow.services.tracing.schema import Log
from langflow.services.tracing.base import BaseTrace from langflow.services.tracing.base import BaseTrace
from langflow.services.tracing.schema import Log
if TYPE_CHECKING: if TYPE_CHECKING:
from langflow.services.monitor.service import MonitorService from langflow.services.monitor.service import MonitorService