""" title: URL Context id: gemini_url_context_toggle description: Retrieve content from URLs in the prompt author: suurt8ll author_url: https://github.com/suurt8ll funding_url: https://github.com/suurt8ll/open_webui_functions license: MIT version: 1.0.0 """ from typing import TYPE_CHECKING, cast from pydantic import BaseModel if TYPE_CHECKING: from open_webui.utils.manifold_types import * # My personal types in a separate file for more robustness. class Filter: class Valves(BaseModel): pass def __init__(self) -> None: self.valves = self.Valves() # Makes the filter toggleable in the front-end. self.toggle = True # Lamp bulb icon self.icon = "data:image/svg+xml;base64,PHN2ZyB4bWxucz0iaHR0cDovL3d3dy53My5vcmcvMjAwMC9zdmciIGZpbGw9Im5vbmUiIHZpZXdCb3g9IjAgMCAyNCAyNCIgc3Ryb2tlLXdpZHRoPSIxLjUiIHN0cm9rZT0iY3VycmVudENvbG9yIiBjbGFzcz0ic2l6ZS02Ij48cGF0aCBzdHJva2UtbGluZWNhcD0icm91bmQiIHN0cm9rZS1saW5lam9pbj0icm91bmQiIGQ9Ik0xMy4xOSA4LjY4OGE0LjUgNC41IDAgMCAxIDEuMjQyIDcuMjQ0bC00LjUgNC41YTQuNSA0LjUgMCAwIDEtNi4zNjQtNi4zNjRsMS43NTctMS43NTdtMTMuMzUtLjYyMiAxLjc1Ny0xLjc1N2E0LjUgNC41IDAgMCAwLTYuMzY0LTYuMzY0bC00LjUgNC41YTQuNSA0LjUgMCAwIDAgMS4yNDIgNy4yNDQiIC8+PC9zdmc+Cg==" async def inlet( self, body: "Body", ) -> "Body": # Signal downstream Gemini Manifold pipe that reasoning is enabled. # Ensure features field exists metadata = body.get("metadata") metadata_features = metadata.get("features") if metadata_features is None: metadata_features = cast(Features, {}) metadata["features"] = metadata_features metadata_features["url_context"] = True return body