mirror of
https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git
synced 2025-08-12 20:19:11 +08:00
fix zhipuAI stream issue (#825)
### What problem does this PR solve? ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
parent
081f922ee6
commit
f6a599461f
@ -222,7 +222,7 @@ def completion():
|
||||
resp.headers.add_header("Content-Type", "text/event-stream; charset=utf-8")
|
||||
return resp
|
||||
else:
|
||||
ans = chat(dia, msg, False, **req)
|
||||
ans = chat(dia, msg, **req)
|
||||
fillin_conv(ans)
|
||||
API4ConversationService.append_message(conv.id, conv.to_dict())
|
||||
return get_json_result(data=ans)
|
||||
|
@ -162,7 +162,7 @@ def completion():
|
||||
return resp
|
||||
|
||||
else:
|
||||
ans = chat(dia, msg, False, **req)
|
||||
ans = chat(dia, msg, **req)
|
||||
fillin_conv(ans)
|
||||
ConversationService.update_by_id(conv.id, conv.to_dict())
|
||||
return get_json_result(data=ans)
|
||||
|
@ -193,10 +193,11 @@ class ZhipuChat(Base):
|
||||
if not resp.choices[0].delta.content:continue
|
||||
delta = resp.choices[0].delta.content
|
||||
ans += delta
|
||||
tk_count = resp.usage.total_tokens if response.usage else 0
|
||||
if resp.output.choices[0].finish_reason == "length":
|
||||
if resp.choices[0].finish_reason == "length":
|
||||
ans += "...\nFor the content length reason, it stopped, continue?" if is_english(
|
||||
[ans]) else "······\n由于长度的原因,回答被截断了,要继续吗?"
|
||||
tk_count = resp.usage.total_tokens
|
||||
if resp.choices[0].finish_reason == "stop": tk_count = resp.usage.total_tokens
|
||||
yield ans
|
||||
except Exception as e:
|
||||
yield ans + "\n**ERROR**: " + str(e)
|
||||
|
Loading…
x
Reference in New Issue
Block a user