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),