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

Fix: Update display time for very short operations and fix outputs and titles for tool calling executions (#8718)

Browse source 
* update display time for very short operations and fix outputs and titles

* [autofix.ci] apply automated fixes

* [autofix.ci] apply automated fixes (attempt 2/3)

* cleanup

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
Co-authored-by: Edwin Jose <edwin.jose@datastax.com>
This commit is contained in:
Mike Fortman 2025-06-24 18:07:19 -05:00 • committed by GitHub
commit 927cdd0837
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 35 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

40
src/backend/base/langflow/base/agents/events.py
View file

@ -161,7 +161,7 @@ async def handle_on_tool_start(
tool_content = ToolContent(
type="tool_use",
name=tool_name,
input=tool_input,
tool_input=tool_input,
output=None,
error=None,
header={"title": f"Accessing **{tool_name}**", "icon": "Hammer"},
@ -191,13 +191,39 @@ async def handle_on_tool_end(
tool_content = tool_blocks_map.get(tool_key)
if tool_content and isinstance(tool_content, ToolContent):
tool_content.output = event["data"].get("output")
duration = _calculate_duration(start_time)
tool_content.duration = duration
tool_content.header = {"title": f"Executed **{tool_content.name}**", "icon": "Hammer"}
# Call send_message_method first to get the updated message structure
agent_message = await send_message_method(message=agent_message)
new_start_time = perf_counter() # Get new start time for next operation
new_start_time = perf_counter()
# Now find and update the tool content in the current message
duration = _calculate_duration(start_time)
tool_key = f"{tool_name}_{run_id}"
# Find the corresponding tool content in the updated message
updated_tool_content = None
if agent_message.content_blocks and agent_message.content_blocks[0].contents:
for content in agent_message.content_blocks[0].contents:
if (
isinstance(content, ToolContent)
and content.name == tool_name
and content.tool_input == tool_content.tool_input
):
updated_tool_content = content
break
# Update the tool content that's actually in the message
if updated_tool_content:
updated_tool_content.duration = duration
updated_tool_content.header = {"title": f"Executed **{updated_tool_content.name}**", "icon": "Hammer"}
updated_tool_content.output = event["data"].get("output")
# Update the map reference
tool_blocks_map[tool_key] = updated_tool_content
for content in agent_message.content_blocks[0].contents:
if isinstance(content, ToolContent):
header_title = content.header.get("title", "N/A") if content.header else "None"
return agent_message, new_start_time
return agent_message, start_time

3
src/frontend/src/components/core/chatComponents/DurationDisplay.tsx
View file

@ -41,7 +41,8 @@ export default function DurationDisplay({
const displayTime = duration ?? durations[chatId] ?? 0;
const secondsValue = displayTime / 1000;
const humanizedTime = `${secondsValue.toFixed(1)}s`;
const humanizedTime =
secondsValue < 0.05 ? "< 0.1s" : `${secondsValue.toFixed(1)}s`;
return (
<div