瀏覽代碼

fix: error type get wrong (#2403)

takatost 1 年之前
父節點
當前提交
42344795cd
共有 2 個文件被更改,包括 5 次插入2 次删除
  1. 5 1
      api/core/app_runner/generate_task_pipeline.py
  2. 0 1
      api/core/hosting_configuration.py

+ 5 - 1
api/core/app_runner/generate_task_pipeline.py

@@ -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:

+ 0 - 1
api/core/hosting_configuration.py

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