AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 06:15:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: error type get wrong (#2403)

Browse Source
This commit is contained in:
takatost 2024-02-06 12:24:48 +08:00 committed by GitHub
parent 517f6d1a26
commit 42344795cd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/app_runner/generate_task_pipeline.py
View File

@ -477,7 +477,11 @@ class GenerateTaskPipeline:
} }
# Determine the response based on the type of exception # Determine the response based on the type of exception
data = error_responses.get(type(e)) data = None
for k, v in error_responses.items():
if isinstance(e, k):
data = v
if data: if data:
data.setdefault('message', getattr(e, 'description', str(e))) data.setdefault('message', getattr(e, 'description', str(e)))
else: else:

1
api/core/hosting_configuration.py
View File

@ -123,7 +123,6 @@ class HostingConfiguration:
restrict_models=[ restrict_models=[
RestrictModel(model="gpt-4", model_type=ModelType.LLM), RestrictModel(model="gpt-4", model_type=ModelType.LLM),
RestrictModel(model="gpt-4-turbo-preview", model_type=ModelType.LLM), RestrictModel(model="gpt-4-turbo-preview", model_type=ModelType.LLM),
RestrictModel(model="gpt-4-32k", model_type=ModelType.LLM),
RestrictModel(model="gpt-4-1106-preview", model_type=ModelType.LLM), RestrictModel(model="gpt-4-1106-preview", model_type=ModelType.LLM),
RestrictModel(model="gpt-4-0125-preview", model_type=ModelType.LLM), RestrictModel(model="gpt-4-0125-preview", model_type=ModelType.LLM),
RestrictModel(model="gpt-3.5-turbo", model_type=ModelType.LLM), RestrictModel(model="gpt-3.5-turbo", model_type=ModelType.LLM),