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

fix(llmproxy): fix openai chatgpt api response error (#871)

Browse Source
This commit is contained in:
magic.chen 2023-11-30 15:11:00 +08:00 committed by GitHub
parent 1a8616dab8
commit 711032c480
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pilot/model/proxy/llms/chatgpt.py
View File

@ -175,7 +175,7 @@ def chatgpt_generate_stream(
# logger.info(str(r)) # logger.info(str(r))
# Azure Openai reponse may have empty choices body in the first chunk # Azure Openai reponse may have empty choices body in the first chunk
# to avoid index out of range error # to avoid index out of range error
if not r.get("choices"): if len(r.choices) == 0:
continue continue
if r.choices[0].delta.content is not None: if r.choices[0].delta.content is not None:
content = r.choices[0].delta.content content = r.choices[0].delta.content
@ -191,7 +191,7 @@ def chatgpt_generate_stream(
text = "" text = ""
for r in res: for r in res:
if not r.get("choices"): if len(r.choices) == 0:
continue continue
if r["choices"][0]["delta"].get("content") is not None: if r["choices"][0]["delta"].get("content") is not None:
content = r["choices"][0]["delta"]["content"] content = r["choices"][0]["delta"]["content"]