mirror of
https://github.com/eigent-ai/eigent.git
synced 2026-04-28 11:40:25 +00:00
33829f2c1e
Co-authored-by: 4pmtong <web_chentong@163.com> Co-authored-by: Wendong-Fan <133094783+Wendong-Fan@users.noreply.github.com> Co-authored-by: Wendong-Fan <w3ndong.fan@gmail.com>
86 lines
3.2 KiB
Python
86 lines
3.2 KiB
Python
# ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
# ========= Copyright 2025-2026 @ Eigent.ai All Rights Reserved. =========
|
|
|
|
import json
|
|
import re
|
|
|
|
|
|
def normalize_error_to_openai_format(exception: Exception) -> tuple[str, str | None, dict | None]:
|
|
"""
|
|
Normalize error to OpenAI-style error structure.
|
|
|
|
Args:
|
|
exception: The exception to normalize
|
|
|
|
Returns:
|
|
tuple: (message, error_code, error_object)
|
|
"""
|
|
raw_msg = str(exception)
|
|
error_obj = None
|
|
error_code = None
|
|
message = raw_msg
|
|
|
|
# Match "Error code: <code> - {json}"
|
|
m = re.search(r"Error code:\s*(\d+)\s*-\s*(\{.*\})", raw_msg, re.DOTALL)
|
|
if m:
|
|
error_code = m.group(1)
|
|
try:
|
|
parsed = json.loads(m.group(2))
|
|
err = parsed.get("error") or parsed
|
|
if isinstance(err, dict):
|
|
error_obj = {
|
|
"message": err.get("message"),
|
|
"type": err.get("type"),
|
|
"param": err.get("param"),
|
|
"code": err.get("code"),
|
|
}
|
|
if err.get("message"):
|
|
message = err.get("message")
|
|
if err.get("code"):
|
|
error_code = err.get("code")
|
|
except Exception:
|
|
pass
|
|
|
|
# Heuristics if not parsed
|
|
if error_obj is None:
|
|
lower = raw_msg.lower()
|
|
if "invalid_api_key" in lower or "incorrect api key" in lower or "unauthorized" in lower or " 401" in lower:
|
|
error_code = "invalid_api_key"
|
|
message = "Invalid key. Validation failed."
|
|
error_obj = {
|
|
"message": message,
|
|
"type": "invalid_request_error",
|
|
"param": None,
|
|
"code": "invalid_api_key",
|
|
}
|
|
elif "model_not_found" in lower or "does not exist" in lower or " 404" in lower:
|
|
error_code = "model_not_found"
|
|
message = "Invalid model name. Validation failed."
|
|
error_obj = {
|
|
"message": message,
|
|
"type": "invalid_request_error",
|
|
"param": None,
|
|
"code": "model_not_found",
|
|
}
|
|
elif "insufficient_quota" in lower or "quota" in lower or " 429" in lower:
|
|
error_code = "insufficient_quota"
|
|
message = "You exceeded your current quota, please check your plan and billing details."
|
|
error_obj = {
|
|
"message": message,
|
|
"type": "insufficient_quota",
|
|
"param": None,
|
|
"code": "insufficient_quota",
|
|
}
|
|
|
|
return message, error_code, error_obj
|