feat: + error handler

app.py

@@ -5,17 +5,15 @@ from __future__ import annotations
 import asyncio
 import contextlib
 import pathlib
+import re
 import shutil
 import traceback
 import uuid
 from collections import deque
 from functools import partial
-from json import JSONDecodeError
 from typing import Dict
 
 import fire
-import httpx
-import openai
 import tenacity
 import uvicorn
 from fastapi import FastAPI, Request
@@ -25,6 +23,7 @@ from loguru import logger
 from metagpt.config import CONFIG
 from metagpt.logs import set_llm_stream_logfunc
 from metagpt.schema import Message
+from metagpt.utils.common import any_to_name, any_to_str
 from openai import OpenAI
 
 from data_model import (
@@ -154,32 +153,42 @@ class Service:
             while isinstance(original_exception, tenacity.RetryError):
                 original_exception = original_exception.last_attempt.exception()
 
-            if isinstance(original_exception, openai.AuthenticationError):
-                answer = original_exception.message
-                title = "OpenAI AuthenticationError"
-                think_act_prompt = cls.create_error_think_act_prompt(tc_id, title, title, answer)
-                return think_act_prompt.prompt + "\n\n"
-            elif isinstance(original_exception, openai.APITimeoutError):
-                answer = original_exception.message
-                title = "OpenAI APITimeoutError"
-                think_act_prompt = cls.create_error_think_act_prompt(tc_id, title, title, answer)
-                return think_act_prompt.prompt + "\n\n"
-            elif isinstance(original_exception, httpx.ReadTimeout):
-                answer = f"{original_exception}. {original_exception.request}"
-                title = "httpx ReadTimeout"
-                think_act_prompt = cls.create_error_think_act_prompt(tc_id, title, title, answer)
-                return think_act_prompt.prompt + "\n\n"
-            elif isinstance(original_exception, JSONDecodeError):
-                answer = str(original_exception)
-                title = "MetaGPT Error"
-                description = "LLM return result parsing error"
-                think_act_prompt = cls.create_error_think_act_prompt(tc_id, title, description, answer)
-                return think_act_prompt.prompt + "\n\n"
-            else:
-                return cls.handle_unexpected_error(tc_id, error)
+            name = any_to_str(original_exception)
+            if re.match(r"^openai\.", name):
+                return cls._handle_openai_error(tc_id, original_exception)
+
+            if re.match(r"^httpx\.", name):
+                return cls._handle_httpx_error(tc_id, original_exception)
+
+            if re.match(r"^json\.", name):
+                return cls._handle_json_error(tc_id, original_exception)
+
+            return cls.handle_unexpected_error(tc_id, error)
         except Exception:
             return cls.handle_unexpected_error(tc_id, error)
 
+    @classmethod
+    def _handle_openai_error(cls, tc_id, original_exception):
+        answer = original_exception.message
+        title = f"OpenAI {any_to_name(original_exception)}"
+        think_act_prompt = cls.create_error_think_act_prompt(tc_id, title, title, answer)
+        return think_act_prompt.prompt + "\n\n"
+
+    @classmethod
+    def _handle_httpx_error(cls, tc_id, original_exception):
+        answer = f"{original_exception}. {original_exception.request}"
+        title = f"httpx {any_to_name(original_exception)}"
+        think_act_prompt = cls.create_error_think_act_prompt(tc_id, title, title, answer)
+        return think_act_prompt.prompt + "\n\n"
+
+    @classmethod
+    def _handle_json_error(cls, tc_id, original_exception):
+        answer = str(original_exception)
+        title = "MetaGPT Action Node Error"
+        description = f"LLM response parse error. {any_to_str(original_exception)}: {original_exception}"
+        think_act_prompt = cls.create_error_think_act_prompt(tc_id, title, description, answer)
+        return think_act_prompt.prompt + "\n\n"
+
     @classmethod
     def handle_unexpected_error(cls, tc_id, error):
         description = str(error)