Update code

microsoft · Dec 7, 2023 · 991134b · 991134b
1 parent a324cf0
commit 991134b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/promptflow/promptflow/batch/_base_executor_proxy.py b/src/promptflow/promptflow/batch/_base_executor_proxy.py
@@ -118,7 +118,7 @@ def _process_http_response(self, response: httpx.Response):
             try:
                 error_dict = response.json()
                 return error_dict["error"]
-            except JSONDecodeError:
+            except (JSONDecodeError, KeyError):
                 unexpected_error = UnexpectedError(
                     message_format=message_format, status_code=response.status_code, error=response.text
                 )