diff --git a/enable_reasoning.py b/enable_reasoning.py new file mode 100644 index 0000000..40df268 --- /dev/null +++ b/enable_reasoning.py @@ -0,0 +1,29 @@ +from pydantic import BaseModel, Field +from typing import Optional + +class Filter: + class Valves(BaseModel): + pass + + def __init__(self): + self.valves = self.Valves() + self.toggle = True # IMPORTANT: This creates a switch UI in Open WebUI + self.icon = """data:image/svg+xml;base64,PHN2ZyB4bWxucz0iaHR0cDovL3d3dy53My5vcmcvMjAwMC9zdmciIGZpbGw9Im5vbmUiIHZpZXdCb3g9IjAgMCAyNCAyNCIgc3Ryb2tlLXdpZHRoPSIxLjUiIHN0cm9rZT0iY3VycmVudENvbG9yIiBjbGFzcz0ic2l6ZS02Ij4KICA8cGF0aCBzdHJva2UtbGluZWNhcD0icm91bmQiIHN0cm9rZS1saW5lam9pbj0icm91bmQiIGQ9Ik0xMiAxOHYtNS4yNW0wIDBhNi4wMSA2LjAxIDAgMCAwIDEuNS0uMTg5bS0xLjUuMTg5YTYuMDEgNi4wMSAwIDAgMS0xLjUtLjE4OW0zLjc1IDcuNDc4YTEyLjA2IDEyLjA2IDAgMCAxLTQuNSAwbTMuNzUgMi4zODNhMTQuNDA2IDE0LjQwNiAwIDAgMS0zIDBNMTQuMjUgMTh2LS4xOTJjMC0uOTgzLjY1OC0xLjgyMyAxLjUwOC0yLjMxNmE3LjUgNy41IDAgMSAwLTcuNTE3IDBjLjg1LjQ5MyAxLjUwOSAxLjMzMyAxLjUwOSAyLjMxNlYxOCIgLz4KPC9zdmc+Cg==""" + pass + + async def inlet( + self, body: dict, __event_emitter__, __user__: Optional[dict] = None + ) -> dict: + await __event_emitter__( + { + "type": "status", + "data": { + "description": "Reasoning Enabled!", + "done": True, + "hidden": False, + }, + } + ) + print("Reasoning Enabled!") + body["reasoning"] = {"enabled": True, "effort": "high"} + return body \ No newline at end of file