From 42344795cd002731b728a1fac5fcd43003f99091 Mon Sep 17 00:00:00 2001 From: takatost Date: Tue, 6 Feb 2024 12:24:48 +0800 Subject: [PATCH] fix: error type get wrong (#2403) --- api/core/app_runner/generate_task_pipeline.py | 6 +++++- api/core/hosting_configuration.py | 1 - 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/api/core/app_runner/generate_task_pipeline.py b/api/core/app_runner/generate_task_pipeline.py index 8a7e6f457a..ea54e709e1 100644 --- a/api/core/app_runner/generate_task_pipeline.py +++ b/api/core/app_runner/generate_task_pipeline.py @@ -477,7 +477,11 @@ class GenerateTaskPipeline: } # 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: data.setdefault('message', getattr(e, 'description', str(e))) else: diff --git a/api/core/hosting_configuration.py b/api/core/hosting_configuration.py index b2917682dc..ea93c5336c 100644 --- a/api/core/hosting_configuration.py +++ b/api/core/hosting_configuration.py @@ -123,7 +123,6 @@ class HostingConfiguration: restrict_models=[ RestrictModel(model="gpt-4", 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-0125-preview", model_type=ModelType.LLM), RestrictModel(model="gpt-3.5-turbo", model_type=ModelType.LLM),