Better input/output visibility in Langchain callback (streamlit#7478)

* Better input/output visibility in Langchain callback * Update streamlit_callback_handler.py * Update streamlit_callback_handler.py * Update streamlit_callback_handler.py * Update streamlit_callback_handler.py
eric-skydio · Dec 20, 2023 · 619006e · 619006e
1 parent d99d14f
commit 619006e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/streamlit/external/langchain/streamlit_callback_handler.py b/lib/streamlit/external/langchain/streamlit_callback_handler.py
@@ -199,6 +199,9 @@ def on_tool_start(
             label=self._labeler.get_tool_label(self._last_tool, is_complete=False),
             state="running",
         )
+        if len(input_str) > MAX_TOOL_INPUT_STR_LENGTH:
+            # output is printed later in on_tool_end
+            self._container.markdown(f"**Input:**\n\n{input_str}\n\n**Output:**")
 
     def on_tool_end(
         self,