Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +2 -2
src/streamlit_app.py
CHANGED
|
@@ -84,8 +84,8 @@ def show_file(file_path: str):
|
|
| 84 |
def render_tool_message(tool_message: ToolMessage):
|
| 85 |
"""Render tool execution based on tool name instead of message content."""
|
| 86 |
file_related_keywords = ["read", "write", "file", "save", "export", "create"]
|
| 87 |
-
|
| 88 |
-
|
| 89 |
|
| 90 |
# Check if this is a file-related tool
|
| 91 |
if any(k in tool_name for k in file_related_keywords):
|
|
|
|
| 84 |
def render_tool_message(tool_message: ToolMessage):
|
| 85 |
"""Render tool execution based on tool name instead of message content."""
|
| 86 |
file_related_keywords = ["read", "write", "file", "save", "export", "create"]
|
| 87 |
+
tool_name = getattr(tool_message, "name", "").lower()
|
| 88 |
+
st.info(f"🧰 Using tool :- {tool_name or ''}")
|
| 89 |
|
| 90 |
# Check if this is a file-related tool
|
| 91 |
if any(k in tool_name for k in file_related_keywords):
|