From fc1a66ea7682ec29d12066ef78359bc6983cdbbc Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Sun, 16 Jun 2024 10:39:48 -0600
Subject: [PATCH] feat: current_time, current_datetime

---
 backend/utils/task.py  |  6 +++++-
 src/lib/utils/index.ts | 14 ++++++++++++++
 2 files changed, 19 insertions(+), 1 deletion(-)

diff --git a/backend/utils/task.py b/backend/utils/task.py
index 615febcdc..06787196c 100644
--- a/backend/utils/task.py
+++ b/backend/utils/task.py
@@ -13,9 +13,13 @@ def prompt_template(
 
     # Format the date to YYYY-MM-DD
     formatted_date = current_date.strftime("%Y-%m-%d")
+    formatted_time = current_date.strftime("%I:%M:%S %p")
 
-    # Replace {{CURRENT_DATE}} in the template with the formatted date
     template = template.replace("{{CURRENT_DATE}}", formatted_date)
+    template = template.replace("{{CURRENT_TIME}}", formatted_time)
+    template = template.replace(
+        "{{CURRENT_DATETIME}}", f"{formatted_date} {formatted_time}"
+    )
 
     if user_name:
         # Replace {{USER_NAME}} in the template with the user's name
diff --git a/src/lib/utils/index.ts b/src/lib/utils/index.ts
index ec9ada06d..1643a5eeb 100644
--- a/src/lib/utils/index.ts
+++ b/src/lib/utils/index.ts
@@ -487,9 +487,23 @@ export const promptTemplate = (
 		'-' +
 		String(currentDate.getDate()).padStart(2, '0');
 
+	// Format the time to HH:MM:SS AM/PM
+	const currentTime = currentDate.toLocaleTimeString('en-US', {
+		hour: 'numeric',
+		minute: 'numeric',
+		second: 'numeric',
+		hour12: true
+	});
+
+	// Replace {{CURRENT_DATETIME}} in the template with the formatted datetime
+	template = template.replace('{{CURRENT_DATETIME}}', `${formattedDate} ${currentTime}`);
+
 	// Replace {{CURRENT_DATE}} in the template with the formatted date
 	template = template.replace('{{CURRENT_DATE}}', formattedDate);
 
+	// Replace {{CURRENT_TIME}} in the template with the formatted time
+	template = template.replace('{{CURRENT_TIME}}', currentTime);
+
 	if (user_name) {
 		// Replace {{USER_NAME}} in the template with the user's name
 		template = template.replace('{{USER_NAME}}', user_name);