From 1a8e9023de310a6a78e64fdee60203ebc5da1cfc Mon Sep 17 00:00:00 2001 From: ZhangShenao <15201440436@163.com> Date: Wed, 18 Sep 2024 05:54:37 +0800 Subject: [PATCH] Improvement[Community] Improve `streamlit_callback_handler` (#26373) - add decorator for static methods Co-authored-by: Erick Friis --- .../streamlit/streamlit_callback_handler.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/libs/community/langchain_community/callbacks/streamlit/streamlit_callback_handler.py b/libs/community/langchain_community/callbacks/streamlit/streamlit_callback_handler.py index 0d065e95dd..5d5985c039 100644 --- a/libs/community/langchain_community/callbacks/streamlit/streamlit_callback_handler.py +++ b/libs/community/langchain_community/callbacks/streamlit/streamlit_callback_handler.py @@ -53,13 +53,15 @@ class LLMThoughtLabeler: labeling logic. """ - def get_initial_label(self) -> str: + @staticmethod + def get_initial_label() -> str: """Return the markdown label for a new LLMThought that doesn't have an associated tool yet. """ return f"{THINKING_EMOJI} **Thinking...**" - def get_tool_label(self, tool: ToolRecord, is_complete: bool) -> str: + @staticmethod + def get_tool_label(tool: ToolRecord, is_complete: bool) -> str: """Return the label for an LLMThought that has an associated tool. @@ -91,13 +93,15 @@ class LLMThoughtLabeler: label = f"{emoji} **{name}:** {input}" return label - def get_history_label(self) -> str: + @staticmethod + def get_history_label() -> str: """Return a markdown label for the special 'history' container that contains overflow thoughts. """ return f"{HISTORY_EMOJI} **History**" - def get_final_agent_thought_label(self) -> str: + @staticmethod + def get_final_agent_thought_label() -> str: """Return the markdown label for the agent's final thought - the "Now I have the answer" thought, that doesn't involve a tool.