Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
D
dify
Project
Project
Details
Activity
Releases
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
ai-tech
dify
Commits
c4d759df
Unverified
Commit
c4d759df
authored
Aug 17, 2023
by
takatost
Committed by
GitHub
Aug 17, 2023
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
fix: wenxin error not raise when stream mode (#884)
parent
a58f95fa
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
21 additions
and
7 deletions
+21
-7
wenxin.py
api/core/third_party/langchain/llms/wenxin.py
+21
-7
No files found.
api/core/third_party/langchain/llms/wenxin.py
View file @
c4d759df
...
...
@@ -3,6 +3,7 @@ from __future__ import annotations
import
json
import
logging
from
json
import
JSONDecodeError
from
typing
import
(
Any
,
Dict
,
...
...
@@ -223,6 +224,8 @@ class Wenxin(LLM):
for
token
in
self
.
_client
.
post
(
request
)
.
iter_lines
():
if
token
:
token
=
token
.
decode
(
"utf-8"
)
if
token
.
startswith
(
'data:'
):
completion
=
json
.
loads
(
token
[
5
:])
yield
GenerationChunk
(
text
=
completion
[
'result'
])
...
...
@@ -231,3 +234,14 @@ class Wenxin(LLM):
if
completion
[
'is_end'
]:
break
else
:
try
:
json_response
=
json
.
loads
(
token
)
except
JSONDecodeError
:
raise
ValueError
(
f
"Wenxin Response Error {token}"
)
raise
ValueError
(
f
"Wenxin API {json_response['error_code']}"
f
" error: {json_response['error_msg']}, "
f
"please confirm if the model you have chosen is already paid for."
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment