github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-08-14 06:26:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: fix function bard_generate_stream's output

Browse Source 
fix function bard_generate_stream's output
This commit is contained in:
xuyuan23 2023-07-27 11:11:59 +08:00
parent 572c336b81
commit c4638bfde3
2 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pilot/model/proxy/proxy_llms/bard.py
View File

@ -34,6 +34,5 @@ def bard_generate_stream(model, tokenizer, params, device, context_len=2048):
response = bardapi.core.Bard(token).get_answer(last_user_input["content"]) response = bardapi.core.Bard(token).get_answer(last_user_input["content"])
if response is not None and response.get("content") is not None: if response is not None and response.get("content") is not None:
yield str(response["content"]) yield str(response["content"])
else:
yield f"bard response error: {str(response)}" yield f"bard response error: {str(response)}"