Github/screenshot-to-code
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

null check for chunk.choices

Browse Source
This commit is contained in:
Naman 2024-05-25 07:02:02 +05:30
parent c5695975c7
commit f0ee686660
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
backend/llm.py
View File

@ -59,7 +59,7 @@ async def stream_openai_response(
full_response = "" full_response = ""
async for chunk in stream: # type: ignore async for chunk in stream: # type: ignore
assert isinstance(chunk, ChatCompletionChunk) assert isinstance(chunk, ChatCompletionChunk)
if chunk.choices and chunk.choices[0].delta.content: if chunk.choices and len(chunk.choices) > 0 and chunk.choices[0].delta and chunk.choices[0].delta.content:
content = chunk.choices[0].delta.content or "" content = chunk.choices[0].delta.content or ""
full_response += content full_response += content
await callback(content) await callback(content)