Merge branch 'master' into 64-when-using-azure-ai-search-openwebui-does-not-show-citationssources

Author: owndev

README.md

@@ -137,13 +137,16 @@ The functions include a built-in encryption mechanism for sensitive information:
 - **Advanced Image Processing**: Optimized image handling with configurable compression, resizing, and quality settings.
 - **Configurable Parameters**: Environment variables for image optimization (quality, max dimensions, format conversion).
 - Grounding with Google search with [google_search_tool.py filter](./filters/google_search_tool.py)
-- Ability to forward User Headers and change gemini base url
 - Native tool calling support
+- Configurable API version support
 
 🔗 [Google Gemini Pipeline in Open WebUI](https://openwebui.com/f/owndev/google_gemini)
 
 🔗 [Learn More About Google Gemini](https://ai.google.dev/gemini-api/docs?hl=de)
 
+> [!NOTE]
+> **For LiteLLM Users**: To use Google Gemini models through LiteLLM, configure LiteLLM directly in Open WebUI's Admin Panel → Settings → Connections → OpenAI section instead of using this pipeline. For more information about LiteLLM, visit the [official LiteLLM GitHub repository](https://github.com/BerriAI/litellm).
+
 ## Filters 🔍
 
 > [!NOTE]

pipelines/google/google_gemini.py

@@ -4,7 +4,7 @@
 author_url: https://github.com/owndev/
 project_url: https://github.com/owndev/Open-WebUI-Functions
 funding_url: https://github.com/sponsors/owndev
-version: 1.6.0
+version: 1.6.1
 license: Apache License 2.0
 description: Highly optimized Google Gemini pipeline with advanced image generation capabilities, intelligent compression, and streamlined processing workflows.
 features:
@@ -30,7 +30,6 @@
   - Optimized payload creation for image generation models
   - Configurable image processing parameters (size, quality, compression)
   - Flexible upload fallback options and optimization controls
-  - Ability to forward User Headers and change gemini base url
 """
 
 import os
@@ -159,10 +158,9 @@ class Valves(BaseModel):
             default=os.getenv("GOOGLE_API_KEY", ""),
             description="API key for Google Generative AI (used if USE_VERTEX_AI is false).",
         )
-        ENABLE_FORWARD_USER_INFO_HEADERS: bool = Field(
-            default=os.getenv("ENABLE_FORWARD_USER_INFO_HEADERS", "false").lower()
-            == "true",
-            description="Whether to forward user information headers.",
+        API_VERSION: str = Field(
+            default=os.getenv("GOOGLE_API_VERSION", "v1alpha"),
+            description="API version to use for Google Generative AI (e.g., v1alpha, v1beta, v1).",
         )
         USE_VERTEX_AI: bool = Field(
             default=os.getenv("GOOGLE_GENAI_USE_VERTEXAI", "false").lower() == "true",
@@ -479,20 +477,8 @@ def _get_client(self) -> genai.Client:
             )
         else:
             self.log.debug("Initializing Google Generative AI client with API Key")
-            headers = {}
-            if (
-                self.valves.ENABLE_FORWARD_USER_INFO_HEADERS
-                and hasattr(self, "user")
-                and self.user
-            ):
-                headers = {
-                    "X-OpenWebUI-User-Name": self.user.name,
-                    "X-OpenWebUI-User-Id": self.user.id,
-                    "X-OpenWebUI-User-Email": self.user.email,
-                    "X-OpenWebUI-User-Role": self.user.role,
-                }
             options = types.HttpOptions(
-                api_version="v1alpha", base_url=self.valves.BASE_URL, headers=headers
+                api_version=self.valves.API_VERSION, base_url=self.valves.BASE_URL
             )
             return genai.Client(
                 api_key=self.valves.GOOGLE_API_KEY.get_decrypted(), http_options=options