refactor: Update logs to outputs in CustomComponent and TracingService classes
This commit is contained in:
parent
b99f292f1c
commit
3ba8a19545
2 changed files with 4 additions and 7 deletions
|
|
@ -83,7 +83,7 @@ class CustomComponent(BaseComponent):
|
|||
_flows_data: Optional[List[Data]] = None
|
||||
_outputs: List[OutputLog] = []
|
||||
_logs: List[Log] = []
|
||||
_tracing_service: "TracingService"
|
||||
tracing_service: Optional["TracingService"] = None
|
||||
|
||||
def update_state(self, name: str, value: Any):
|
||||
if not self.vertex:
|
||||
|
|
@ -489,13 +489,10 @@ class CustomComponent(BaseComponent):
|
|||
message (LoggableType | list[LoggableType]): The message to log.
|
||||
"""
|
||||
if name is None:
|
||||
name = self.display_name if self.display_name else self.__class__.__name__
|
||||
if hasattr(message, "model_dump") and isinstance(message, BaseModel):
|
||||
message = message.model_dump()
|
||||
name = self.display_name
|
||||
log = Log(message=message, type=get_artifact_type(message), name=name)
|
||||
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):
|
||||
"""
|
||||
|
|
|
|||
|
|
@ -12,8 +12,8 @@ from loguru import logger
|
|||
|
||||
from langflow.schema.data import Data
|
||||
from langflow.services.base import Service
|
||||
from langflow.services.tracing.schema import Log
|
||||
from langflow.services.tracing.base import BaseTrace
|
||||
from langflow.services.tracing.schema import Log
|
||||
|
||||
if TYPE_CHECKING:
|
||||
from langflow.services.monitor.service import MonitorService
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue