Skip to content

Commit

Permalink
fix: codestyle error fix (#904)
Browse files Browse the repository at this point in the history
  • Loading branch information
csunny authored Dec 7, 2023
1 parent 6a26b7a commit 1856514
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 3 deletions.
3 changes: 1 addition & 2 deletions pilot/scene/base_chat.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down Expand Up @@ -713,4 +712,4 @@ def _load_history_messages(
ModelMessage(role=message_type, content=message_content)
)

return history_text if str_message else history_messages
return history_text if str_message else history_messages
6 changes: 5 additions & 1 deletion pilot/scene/chat_db/auto_execute/out_parser.py
Original file line number Diff line number Diff line change
Expand Up @@ -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__)
Expand Down

0 comments on commit 1856514

Please sign in to comment.