Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-16 11:57:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

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
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:
if hasattr(user, "model_dump"):
user = user.model_dump()
USER_VARIABLES = {}
if user:
if hasattr(user, "model_dump"):
user = user.model_dump()
if isinstance(user, dict):
birth_date = user.get("date_of_birth")
age = None
@@ -58,7 +60,10 @@ def prompt_template(template: str, user: Optional[Any] = None) -> str:
age = (
today.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:
pass
@@ -72,8 +77,6 @@ def prompt_template(template: str, user: Optional[Any] = None) -> str:
"age": str(age),
}
print(USER_VARIABLES)
# Get the current date
current_date = datetime.now()