devchavda11 commited on
Commit
fcd4add
·
verified ·
1 Parent(s): a426480

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. 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
- 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):
 
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):