From fe95506db416a42c9b2b21e2ed0a2927fff9d079 Mon Sep 17 00:00:00 2001 From: Raven95676 Date: Sun, 13 Apr 2025 17:50:44 +0800 Subject: [PATCH] =?UTF-8?q?perf:=20=E6=B7=BB=E5=8A=A0=E6=97=A5=E5=BF=97?= =?UTF-8?q?=E8=BF=87=E6=BB=A4=E5=99=A8=E4=BB=A5=E6=8A=91=E5=88=B6=E9=9D=9E?= =?UTF-8?q?=E6=96=87=E6=9C=AC=E9=83=A8=E5=88=86=E8=AD=A6=E5=91=8A=E4=BF=A1?= =?UTF-8?q?=E6=81=AF?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- astrbot/core/provider/sources/gemini_source.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/astrbot/core/provider/sources/gemini_source.py b/astrbot/core/provider/sources/gemini_source.py index ab312e502..9b58646f5 100644 --- a/astrbot/core/provider/sources/gemini_source.py +++ b/astrbot/core/provider/sources/gemini_source.py @@ -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 提供商适配器" )