import requests from pydantic import BaseModel, Field class Tools: """ A simple Open WebUI tool that provides an NSFW mode toggle via UserValves. """ class UserValves(BaseModel): nsfw_mode: bool = Field( False, description="Enable NSFW mode (bypass content moderation)" ) def __init__(self): # Initialize the user‑valves model self.user_valves = self.UserValves() def generate(self, prompt: str) -> str: """ Generates text using the standard Open WebUI inference API. :param prompt: The input prompt for the model. :return: The generated text, or an error message. """ payload = { "prompt": prompt, "nsfwMode": self.user_valves.nsfw_mode } # Call your local inference endpoint response = requests.post("http://localhost:3000/api/generate", json=payload) data = response.json() # Return either the text or an error key return data.get("text", data.get("error", "No response returned"))