From 5f535001db493024012fa4cd4ff025b0ae4be8d6 Mon Sep 17 00:00:00 2001 From: BiDuang <2267304497@qq.com> Date: Wed, 10 Dec 2025 20:27:51 +0800 Subject: [PATCH] fix: incorrect modalities enum of gemini api provider (#3993) --- astrbot/core/provider/sources/gemini_source.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/astrbot/core/provider/sources/gemini_source.py b/astrbot/core/provider/sources/gemini_source.py index 2d171cfd8..e2efc6aab 100644 --- a/astrbot/core/provider/sources/gemini_source.py +++ b/astrbot/core/provider/sources/gemini_source.py @@ -127,15 +127,15 @@ class ProviderGoogleGenAI(Provider): ) -> types.GenerateContentConfig: """准备查询配置""" if not modalities: - modalities = ["Text"] + modalities = ["TEXT"] # 流式输出不支持图片模态 if ( self.provider_settings.get("streaming_response", False) - and "Image" in modalities + and "IMAGE" in modalities ): logger.warning("流式输出不支持图片模态,已自动降级为文本模态") - modalities = ["Text"] + modalities = ["TEXT"] tool_list: list[types.Tool] | None = [] model_name = self.get_model() @@ -431,9 +431,9 @@ class ProviderGoogleGenAI(Provider): None, ) - modalities = ["Text"] + modalities = ["TEXT"] if self.provider_config.get("gm_resp_image_modal", False): - modalities.append("Image") + modalities.append("IMAGE") conversation = self._prepare_conversation(payloads) temperature = payloads.get("temperature", 0.7) @@ -490,7 +490,7 @@ class ProviderGoogleGenAI(Provider): logger.warning( f"{self.get_model()} 不支持多模态输出,降级为文本模态", ) - modalities = ["Text"] + modalities = ["TEXT"] else: raise continue