ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-17 14:55:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

refac

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2025-08-21 04:40:08 +04:00
parent d720e027ad
commit 66f00ce4bf
1 changed files with 30 additions and 27 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
backend/open_webui/utils/task.py
View file

@ -39,11 +39,13 @@ def prompt_variables_template(template: str, variables: dict[str, str]) -> str:
def prompt_template(template: str, user: Optional[Any] = None) -> str: def prompt_template(template: str, user: Optional[Any] = None) -> str:
if hasattr(user, "model_dump"):
user = user.model_dump()
USER_VARIABLES = {} USER_VARIABLES = {}
if user:
if hasattr(user, "model_dump"):
user = user.model_dump()
if isinstance(user, dict): if isinstance(user, dict):
birth_date = user.get("date_of_birth") birth_date = user.get("date_of_birth")
age = None age = None
@ -58,7 +60,10 @@ def prompt_template(template: str, user: Optional[Any] = None) -> str:
age = ( age = (
today.year today.year
- birth_date.year - birth_date.year
- ((today.month, today.day) < (birth_date.month, birth_date.day)) - (
(today.month, today.day)
< (birth_date.month, birth_date.day)
)
) )
except Exception as e: except Exception as e:
pass pass
@ -72,8 +77,6 @@ def prompt_template(template: str, user: Optional[Any] = None) -> str:
"age": str(age), "age": str(age),
} }
print(USER_VARIABLES)
# Get the current date # Get the current date
current_date = datetime.now() current_date = datetime.now()