yujunjun/LightRAG
1
0
Fork 0
mirror of https://github.com/HKUDS/LightRAG.git synced 2025-10-29 17:00:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix stream

Browse Source
This commit is contained in:
zrguo 2025-03-17 11:41:55 +08:00
parent a9bd318d98
commit adba09f6c2
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
lightrag/llm/openai.py
View File

@ -123,21 +123,18 @@ async def openai_complete_if_cache(
async def inner():
try:
_content = ""
async for chunk in response:
content = chunk.choices[0].delta.content
if content is None:
continue
if r"\u" in content:
content = safe_unicode_decode(content.encode("utf-8"))
_content += content
return _content
yield content
except Exception as e:
logger.error(f"Error in stream response: {str(e)}")
raise
response_content = await inner()
return response_content
return inner()
else:
if (