diff --git a/pilot/scene/base_chat.py b/pilot/scene/base_chat.py index 5837fee1b..c70c1d2ea 100644 --- a/pilot/scene/base_chat.py +++ b/pilot/scene/base_chat.py @@ -506,7 +506,6 @@ def _generate_numbered_list(self) -> str: ) - def _build_model_operator( is_stream: bool = False, dag_name: str = "llm_model_dag" ) -> BaseOperator: @@ -713,4 +712,4 @@ def _load_history_messages( ModelMessage(role=message_type, content=message_content) ) - return history_text if str_message else history_messages \ No newline at end of file + return history_text if str_message else history_messages diff --git a/pilot/scene/chat_db/auto_execute/out_parser.py b/pilot/scene/chat_db/auto_execute/out_parser.py index c57ea1419..a05dd0bea 100644 --- a/pilot/scene/chat_db/auto_execute/out_parser.py +++ b/pilot/scene/chat_db/auto_execute/out_parser.py @@ -17,7 +17,11 @@ class SqlAction(NamedTuple): display: str def to_dict(self) -> Dict[str, Dict]: - return {"sql": self.sql, "thoughts": self.thoughts, "display": self.display, } + return { + "sql": self.sql, + "thoughts": self.thoughts, + "display": self.display, + } logger = logging.getLogger(__name__)