""" title: Reasoning Filter for Deepseek author: shuhao version: 0.1 """ from pydantic import BaseModel, Field from typing import Optional # This filter function needs the Open WebUI >= 0.5.17 to use the stream hook. class Filter: class Valves(BaseModel): priority: int = Field( default=0, description="Priority level for the filter operations." ) def __init__(self): self.thinking_tag_created = False pass def inlet(self, body: dict) -> dict: pass def stream(self, event: dict) -> dict: # If the model is not deepseek r1, no change to the event. if event.get("model") != "deepseek-reasoner": return event choices = event.get("choices") if not choices: return event delta = choices[-1]["delta"] if delta["reasoning_content"] is not None: if not self.thinking_tag_created: delta["content"] = "<think>" + delta["reasoning_content"] self.thinking_tag_created = True delta["content"] = delta["reasoning_content"] else: if self.thinking_tag_created: delta["content"] = "</think>" + delta["content"] self.thinking_tag_created = False return event def outlet(self, body: dict) -> dict: return body