github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2026-01-25 14:54:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(ChatData): ChatData Strean response

Browse Source 
1.ChatData Stream response
This commit is contained in:
yhjun1026
2023-10-23 15:40:40 +08:00
parent bb882a8d73
commit bf75af520c
3 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -104,10 +104,11 @@ async def get_editor_sql(con_uid: str, round: int):
f'history ai json resp:{element["data"]["content"]}'
)
api_call = ApiCall()
api_call.update_from_context(element["data"]["content"])
result = {}
result['thoughts'] = element["data"]["content"]
result['thoughts'] = element["data"]["content"].replace("\n", " ").replace("\\n", " ")
if api_call.check_last_plugin_call_ready(element["data"]["content"]):
api_call.update_from_context(element["data"]["content"])
if len(api_call.plugin_status_map) > 0:
first_item = next(iter(api_call.plugin_status_map.items()))[1]
result['sql'] = first_item.args["sql"]