github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-30 15:21:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: codestyle error fix (#904)

Browse Source
This commit is contained in:
magic.chen 2023-12-07 16:59:35 +08:00 committed by GitHub
parent 6a26b7aefe
commit 18565149ea
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pilot/scene/base_chat.py
View File

@ -506,7 +506,6 @@ class BaseChat(ABC):
)
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
return history_text if str_message else history_messages

6
pilot/scene/chat_db/auto_execute/out_parser.py
View File

@ -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__)