Selaa lähdekoodia

fix: error type get wrong (#2403)

takatost 1 vuosi sitten
vanhempi
commit
42344795cd

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

+ 0 - 1
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),