feat(gemini): add thought signature validation support (#2120)

Author: QIN2DIM

models/gemini/manifest.yaml

@@ -34,4 +34,4 @@ resource:
     tool:
       enabled: true
 type: plugin
-version: 0.6.4
+version: 0.7.0

models/gemini/models/llm/llm.py

@@ -50,6 +50,9 @@
     "gemini-3-pro-image-preview",
 }
 
+# https://ai.google.dev/gemini-api/docs/thought-signatures#faqs
+DEFAULT_THOUGHT_SIGNATURE: bytes = b"skip_thought_signature_validator"
+
 
 class GoogleLargeLanguageModel(LargeLanguageModel):
     is_thinking = None
@@ -506,25 +509,32 @@ def _format_message_to_gemini_content(
         :return: Gemini Content representation of message
         """
 
-        def _build_text_parts(_content: str | TextPromptMessageContent) -> List[types.Part]:
+        def _build_text_parts(
+            _content: str | TextPromptMessageContent, *, is_assistant_tree: bool = False
+        ) -> List[types.Part]:
             text_parts = []
             if isinstance(_content, TextPromptMessageContent):
                 _content = _content.data
             if message.role == PromptMessageRole.ASSISTANT:
                 _content = re.sub(r"^<think>.*?</think>\s*", "", _content, count=1, flags=re.DOTALL)
             if _content:
-                text_parts.append(types.Part.from_text(text=_content))
+                _unverified_part = types.Part.from_text(text=_content)
+                if is_assistant_tree:
+                    _unverified_part.thought_signature = DEFAULT_THOUGHT_SIGNATURE
+                text_parts.append(_unverified_part)
             return text_parts
 
         # Helper function to build parts from content
-        def build_parts(content: str | List[PromptMessageContentUnionTypes]) -> List[types.Part]:
+        def build_parts(
+            content: str | List[PromptMessageContentUnionTypes], *, is_assistant_tree: bool = False
+        ) -> List[types.Part]:
             if isinstance(content, str):
-                return _build_text_parts(content)
+                return _build_text_parts(content, is_assistant_tree=is_assistant_tree)
 
             parts_ = []
             for obj in content:
                 if obj.type == PromptMessageContentType.TEXT:
-                    parts_.extend(_build_text_parts(obj))
+                    parts_.extend(_build_text_parts(obj, is_assistant_tree=is_assistant_tree))
                 else:
                     # Filter files based on type and supported formats
                     should_upload = True
@@ -542,7 +552,10 @@ def build_parts(content: str | List[PromptMessageContentUnionTypes]) -> List[typ
                         uri, mime_type = self._upload_file_content_to_google(
                             obj, genai_client, file_server_url_prefix
                         )
-                        parts_.append(types.Part.from_uri(file_uri=uri, mime_type=mime_type))
+                        _unverified_part = types.Part.from_uri(file_uri=uri, mime_type=mime_type)
+                        if is_assistant_tree:
+                            _unverified_part.thought_signature = DEFAULT_THOUGHT_SIGNATURE
+                        parts_.append(_unverified_part)
                     else:
                         # Log skipped files for debugging
                         logging.debug(
@@ -560,17 +573,17 @@ def build_parts(content: str | List[PromptMessageContentUnionTypes]) -> List[typ
 
             # Handle text content (remove thinking tags)
             if message.content:
-                parts.extend(build_parts(message.content))
+                parts.extend(build_parts(message.content, is_assistant_tree=True))
 
             # Handle tool calls
             # https://ai.google.dev/gemini-api/docs/function-calling?hl=zh-cn&example=chart#how-it-works
             if message.tool_calls:
                 call = message.tool_calls[0]
-                parts.append(
-                    types.Part.from_function_call(
-                        name=call.function.name, args=json.loads(call.function.arguments)
-                    )
+                _unsafe_part = types.Part.from_function_call(
+                    name=call.function.name, args=json.loads(call.function.arguments)
                 )
+                _unsafe_part.thought_signature = DEFAULT_THOUGHT_SIGNATURE
+                parts.append(_unsafe_part)
 
             return types.Content(role="model", parts=parts)
 

models/gemini/pyproject.toml

@@ -8,7 +8,7 @@ requires-python = ">=3.12"
 # uv pip compile pyproject.toml -o ./requirements.txt
 dependencies = [
     "dify-plugin>=0.5.1,<0.6.0",
-    "google-genai>=1.51.0,<2.0.0",
+    "google-genai>=1.52.0,<2.0.0",
     "google-generativeai>=0.8.5",
     "numpy>=2.3.2",
 ]

models/gemini/requirements.txt

@@ -52,7 +52,7 @@ google-auth==2.41.1
     #   google-generativeai
 google-auth-httplib2==0.2.0
     # via google-api-python-client
-google-genai==1.51.0
+google-genai==1.52.0
     # via gemini-g9cie8 (pyproject.toml)
 google-generativeai==0.8.5
     # via gemini-g9cie8 (pyproject.toml)