docs: Update documentation for Google Gemini

Author: owndev

README.md

@@ -130,6 +130,7 @@ Pipelines are processing functions that extend Open WebUI with **custom AI model
 - **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
 
 🔗 [Google Gemini Pipeline in Open WebUI](https://openwebui.com/f/owndev/google_gemini)

docs/google-gemini-integration.md

@@ -52,7 +52,10 @@ This integration enables **Open WebUI** to interact with **Google Gemini** model
 - **Grounding with Google search**  
   Improve the accuracy and recency of Gemini responses with Google search grounding.
 
-- **Native tool calling support**
+- **Ability to forward User Headers and change gemini base url**  
+  Forward user information headers (like Name, Id, Email and Role) to Google API or LiteLLM for better context and analytics. Also, change the base URL for the Google Generative AI API if needed.
+
+- **Native tool calling support**  
   Leverage Google genai native function calling to orchestrate the use of tools
 
 ## Environment Variables

pipelines/google/google_gemini.py

@@ -150,17 +150,20 @@ class Pipe:
     # Configuration valves for the pipeline
     class Valves(BaseModel):
         BASE_URL: str = Field(
-            default=os.getenv("GOOGLE_GENAI_BASE_URL", "https://generativelanguage.googleapis.com/"),
+            default=os.getenv(
+                "GOOGLE_GENAI_BASE_URL", "https://generativelanguage.googleapis.com/"
+            ),
             description="Base URL for the Google Generative AI API.",
         )
-        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.",
-        )
         GOOGLE_API_KEY: EncryptedStr = Field(
             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.",
+        )
         USE_VERTEX_AI: bool = Field(
             default=os.getenv("GOOGLE_GENAI_USE_VERTEXAI", "false").lower() == "true",
             description="Whether to use Google Cloud Vertex AI instead of the Google Generative AI API.",
@@ -450,7 +453,7 @@ def __init__(self):
         """Initializes the Pipe instance and configures the genai library."""
         self.valves = self.Valves()
         self.name: str = "Google Gemini: "
-        
+
         # Setup logging
         self.log = logging.getLogger("google_ai.pipe")
         self.log.setLevel(SRC_LOG_LEVELS.get("OPENAI", logging.INFO))
@@ -477,14 +480,20 @@ 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:
+            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)
+            options = types.HttpOptions(
+                api_version="v1alpha", base_url=self.valves.BASE_URL, headers=headers
+            )
             return genai.Client(
                 api_key=self.valves.GOOGLE_API_KEY.get_decrypted(), http_options=options
             )