chore: Add trace_type attribute to FlowToolComponent

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-06-19 19:01:44 -03:00
commit e1a0de8133

View file

@ -15,6 +15,7 @@ class FlowToolComponent(CustomComponent):
display_name = "Flow as Tool"
description = "Construct a Tool from a function that runs the loaded Flow."
field_order = ["flow_name", "name", "description", "return_direct"]
trace_type = "tool"
def get_flow_names(self) -> List[str]:
flow_datas = self.list_flows()