AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 07:09:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

enforce utf-8 encoding for provider response (#1973)

Browse Source
This commit is contained in:
Chenhe Gu 2024-01-09 14:22:05 +08:00 committed by GitHub
parent 126b4c332f
commit 344821ed35
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
api/core/model_runtime/model_providers/openai_api_compatible/llm/llm.py
View File

@ -168,7 +168,8 @@ class OAIAPICompatLargeLanguageModel(_CommonOAI_API_Compat, LargeLanguageModel):
type=ParameterType.FLOAT,
default=float(credentials.get('temperature', 0.7)),
min=0,
max=2
max=2,
precision=2
),
ParameterRule(
name=DefaultParameterName.TOP_P.value,
@ -176,7 +177,8 @@ class OAIAPICompatLargeLanguageModel(_CommonOAI_API_Compat, LargeLanguageModel):
type=ParameterType.FLOAT,
default=float(credentials.get('top_p', 1)),
min=0,
max=1
max=1,
precision=2
),
ParameterRule(
name="top_k",
@ -246,7 +248,8 @@ class OAIAPICompatLargeLanguageModel(_CommonOAI_API_Compat, LargeLanguageModel):
:return: full response or stream response chunk generator result
"""
headers = {
'Content-Type': 'application/json'
'Content-Type': 'application/json',
'Accept-Charset': 'utf-8',
}
api_key = credentials.get('api_key')
@ -298,6 +301,9 @@ class OAIAPICompatLargeLanguageModel(_CommonOAI_API_Compat, LargeLanguageModel):
stream=stream
)
if response.encoding is None or response.encoding == 'ISO-8859-1':
response.encoding = 'utf-8'
if response.status_code != 200:
raise InvokeError(f"API request failed with status code {response.status_code}: {response.text}")