""" title: Message Date And Time author: benasraudys author_url: https://github.com/benasraudys funding_url: https://github.com/benasraudys description: Gives model current date and time context for each message. Don't forget to adjust the timezone in the settings. version: 0.1.1 required_open_webui_version: 0.6.4 """ import datetime import os from pydantic import BaseModel, Field from typing import Callable, Awaitable, Any, Optional class Filter: class Valves(BaseModel): timezone_hours: int = Field( default=0, description="Timezone offset hours (e.g., 5 for UTC+5:30, -4 for UTC-4:00)", ) timezone_minutes: int = Field( default=0, description="Timezone offset minutes (e.g., 30 for UTC+5:30, 45 for UTC-4:45)", ) southern_hemisphere: bool = Field( default=False, description="Enable if you're in the Southern Hemisphere (Australia, South America, etc.)", ) def __init__(self): self.valves = self.Valves( **{ "timezone_hours": int(os.getenv("DATETIME_TIMEZONE_HOURS", "0")), "timezone_minutes": int(os.getenv("DATETIME_TIMEZONE_MINUTES", "0")), "southern_hemisphere": os.getenv( "DATETIME_SOUTHERN_HEMISPHERE", "false" ).lower() == "true", } ) def get_season(self, month, southern_hemisphere=False): if not southern_hemisphere: if 3 <= month <= 5: return "Spring" elif 6 <= month <= 8: return "Summer" elif 9 <= month <= 11: return "Autumn" else: return "Winter" else: if 3 <= month <= 5: return "Autumn" elif 6 <= month <= 8: return "Winter" elif 9 <= month <= 11: return "Spring" else: return "Summer" def get_time_of_day(self, hour): if 5 <= hour < 12: return "Morning" elif 12 <= hour < 17: return "Afternoon" elif 17 <= hour < 21: return "Evening" else: return "Night" async def inlet( self, body: dict, __event_emitter__: Callable[[Any], Awaitable[None]], __request__: Any, __user__: Optional[dict] = None, __model__: Optional[dict] = None, ) -> dict: now_utc = datetime.datetime.utcnow() timezone_hours = self.valves.timezone_hours timezone_minutes = self.valves.timezone_minutes total_offset_minutes = (timezone_hours * 60) + timezone_minutes now = now_utc + datetime.timedelta(minutes=total_offset_minutes) month = now.month hour = now.hour formatted_date = now.strftime("%B %d, %Y") formatted_time = now.strftime("%H:%M:%S") day_of_week = now.strftime("%A") hours_offset = abs(total_offset_minutes) // 60 minutes_offset = abs(total_offset_minutes) % 60 if minutes_offset == 0: if total_offset_minutes >= 0: timezone_str = f"UTC+{hours_offset}" else: timezone_str = f"UTC-{hours_offset}" else: if total_offset_minutes >= 0: timezone_str = f"UTC+{hours_offset}:{minutes_offset:02d}" else: timezone_str = f"UTC-{hours_offset}:{minutes_offset:02d}" season = self.get_season(month, self.valves.southern_hemisphere) time_of_day = self.get_time_of_day(hour) context = f"Current date is {day_of_week}, {formatted_date}, {season}, {time_of_day}, the user time is {formatted_time} {timezone_str}" datetime_message = { "role": "system", "content": f"Time context: {context}. ", } if "messages" in body and isinstance(body["messages"], list): body["messages"].insert(0, datetime_message) else: body["messages"] = [datetime_message] return body