Initial plan for thinking levels and budgets support

Co-authored-by: owndev <69784886+owndev@users.noreply.github.com>

Author: Copilot

filters/vertex_ai_search_tool.py

@@ -41,4 +41,3 @@ def inlet(self, body: dict) -> dict:
                         "vertex_ai_search enabled but vertex_rag_store not provided in params or VERTEX_AI_RAG_STORE env var"
                     )
         return body
-

pipelines/google/google_gemini.py

@@ -426,7 +426,7 @@ async def _build_image_generation_contents(
                 {
                     "index": i + 1,
                     "label": (
-                        f"Image {i+1}" if self.valves.IMAGE_ADD_LABELS else str(i + 1)
+                        f"Image {i + 1}" if self.valves.IMAGE_ADD_LABELS else str(i + 1)
                     ),
                     "reused": reused_flags[i],
                     "origin": "history" if reused_flags[i] else "current",
@@ -1429,11 +1429,15 @@ def _configure_generation(
                 or os.getenv("VERTEX_AI_RAG_STORE")
             )
             if vertex_rag_store:
-                self.log.debug(f"Enabling Vertex AI Search grounding: {vertex_rag_store}")
+                self.log.debug(
+                    f"Enabling Vertex AI Search grounding: {vertex_rag_store}"
+                )
                 gen_config_params.setdefault("tools", []).append(
                     types.Tool(
                         retrieval=types.Retrieval(
-                            vertex_ai_search=types.VertexAISearch(datastore=vertex_rag_store)
+                            vertex_ai_search=types.VertexAISearch(
+                                datastore=vertex_rag_store
+                            )
                         )
                     )
                 )
@@ -1470,7 +1474,9 @@ def _format_grounding_chunks_as_sources(
                             "uri": getattr(context, "uri", None),
                         },
                         "document": [getattr(context, "chunk_text", None) or ""],
-                        "metadata": [{"source": getattr(context, "title", None) or "Document"}],
+                        "metadata": [
+                            {"source": getattr(context, "title", None) or "Document"}
+                        ],
                     }
                 )
             elif hasattr(chunk, "web") and chunk.web: