perf: 添加日志过滤器以抑制非文本部分警告信息

This commit is contained in:
Raven95676
2025-04-13 17:50:44 +08:00
parent 310ed76b18
commit fe95506db4
@@ -1,6 +1,7 @@
import asyncio
import base64
import json
import logging
import random
from typing import Dict, List, Optional, AsyncGenerator
@@ -20,6 +21,16 @@ from astrbot.core.utils.io import download_image_by_url
from ..register import register_provider_adapter
class SuppressNonTextPartsWarning(logging.Filter):
"""过滤 Gemini SDK 中的非文本部分警告"""
def filter(self, record):
return "there are non-text parts in the response" not in record.getMessage()
logging.getLogger("google_genai.types").addFilter(SuppressNonTextPartsWarning())
@register_provider_adapter(
"googlegenai_chat_completion", "Google Gemini Chat Completion 提供商适配器"
)