Compare commits

..

1 Commits

Author SHA1 Message Date
Rumio
d6ae312e97
Merge fea9be8c63 into 3cc882b116 2025-09-18 16:01:51 +00:00
2 changed files with 1 additions and 31 deletions

View File

@ -187,12 +187,8 @@ class LLMModel(LLMModelBase):
logger.debug(f"🔑 API密钥: {masked_key}") logger.debug(f"🔑 API密钥: {masked_key}")
logger.debug(f"📋 请求头: {dict(request_data.headers)}") logger.debug(f"📋 请求头: {dict(request_data.headers)}")
sanitizer_req_context_map = {"gemini": "gemini_request"}
sanitizer_req_context = sanitizer_req_context_map.get(
self.api_type, "openai_request"
)
sanitized_body = sanitize_for_logging( sanitized_body = sanitize_for_logging(
request_data.body, context=sanitizer_req_context request_data.body, context="gemini_request"
) )
request_body_str = json.dumps(sanitized_body, ensure_ascii=False, indent=2) request_body_str = json.dumps(sanitized_body, ensure_ascii=False, indent=2)
logger.debug(f"📦 请求体: {request_body_str}") logger.debug(f"📦 请求体: {request_body_str}")

View File

@ -50,29 +50,6 @@ def _sanitize_openai_response(response_json: dict) -> dict:
return response_json return response_json
def _sanitize_openai_request(body: dict) -> dict:
"""净化OpenAI兼容API的请求体主要截断图片base64。"""
try:
sanitized_json = copy.deepcopy(body)
if "messages" in sanitized_json and isinstance(
sanitized_json["messages"], list
):
for message in sanitized_json["messages"]:
if "content" in message and isinstance(message["content"], list):
for i, part in enumerate(message["content"]):
if part.get("type") == "image_url":
if "image_url" in part and isinstance(
part["image_url"], dict
):
url = part["image_url"].get("url", "")
message["content"][i]["image_url"]["url"] = (
_truncate_base64_string(url)
)
return sanitized_json
except Exception:
return body
def _sanitize_gemini_response(response_json: dict) -> dict: def _sanitize_gemini_response(response_json: dict) -> dict:
"""净化Gemini API的响应体处理文本和图片生成两种格式。""" """净化Gemini API的响应体处理文本和图片生成两种格式。"""
try: try:
@ -170,9 +147,6 @@ def sanitize_for_logging(data: Any, context: str | None = None) -> Any:
elif context == "gemini_request": elif context == "gemini_request":
if isinstance(data, dict): if isinstance(data, dict):
return _sanitize_gemini_request(data) return _sanitize_gemini_request(data)
elif context == "openai_request":
if isinstance(data, dict):
return _sanitize_openai_request(data)
else: else:
if isinstance(data, str): if isinstance(data, str):
return _truncate_base64_string(data) return _truncate_base64_string(data)