github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-09 20:28:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

style:fmt

Browse Source
This commit is contained in:
aries_ckt 2023-10-20 16:27:13 +08:00
parent 8db497f6c0
commit b0a9fb7810
2 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
pilot/openapi/api_v1/editor/api_editor_v1.py
View File

@ -105,12 +105,16 @@ async def get_editor_sql(con_uid: str, round: int):
) )
api_call = ApiCall() api_call = ApiCall()
result = {} result = {}
result['thoughts'] = element["data"]["content"] result["thoughts"] = element["data"]["content"]
if api_call.check_last_plugin_call_ready(element["data"]["content"]): if api_call.check_last_plugin_call_ready(
element["data"]["content"]
):
api_call.update_from_context(element["data"]["content"]) api_call.update_from_context(element["data"]["content"])
if len(api_call.plugin_status_map) > 0: if len(api_call.plugin_status_map) > 0:
first_item = next(iter(api_call.plugin_status_map.items()))[1] first_item = next(
result['sql'] = first_item.args["sql"] iter(api_call.plugin_status_map.items())
)[1]
result["sql"] = first_item.args["sql"]
return Result.succ(result) return Result.succ(result)
return Result.faild(msg="not have sql!") return Result.faild(msg="not have sql!")
@ -165,13 +169,17 @@ async def sql_editor_submit(sql_edit_context: ChatSqlEditContext = Body()):
for element in edit_round["messages"]: for element in edit_round["messages"]:
if element["type"] == "ai": if element["type"] == "ai":
new_ai_text = element["data"]["content"] new_ai_text = element["data"]["content"]
new_ai_text.replace(sql_edit_context.old_sql, sql_edit_context.new_sql) new_ai_text.replace(
sql_edit_context.old_sql, sql_edit_context.new_sql
)
element["data"]["content"] = new_ai_text element["data"]["content"] = new_ai_text
for element in edit_round["messages"]: for element in edit_round["messages"]:
if element["type"] == "view": if element["type"] == "view":
api_call = ApiCall() api_call = ApiCall()
new_view_text = api_call.run_display_sql(new_ai_text, conn.run_to_df) new_view_text = api_call.run_display_sql(
new_ai_text, conn.run_to_df
)
element["data"]["content"] = new_view_text element["data"]["content"] = new_view_text
history_mem.update(history_messages) history_mem.update(history_messages)
return Result.succ(None) return Result.succ(None)

1
pilot/scene/chat_db/auto_execute/chat.py
View File

@ -75,6 +75,7 @@ class ChatWithDbAutoExecute(BaseChat):
text = text.replace("\n", " ") text = text.replace("\n", " ")
print(f"stream_plugin_call:{text}") print(f"stream_plugin_call:{text}")
return self.api_call.run_display_sql(text, self.database.run_to_df) return self.api_call.run_display_sql(text, self.database.run_to_df)
# #
# def do_action(self, prompt_response): # def do_action(self, prompt_response):
# print(f"do_action:{prompt_response}") # print(f"do_action:{prompt_response}")