From 10f544b3d0673c0025d7c9e0919725fb7adda800 Mon Sep 17 00:00:00 2001
From: n4ze3m <n4ze3m@gmail.com>
Date: Sat, 10 Aug 2024 20:49:16 +0530
Subject: [PATCH] chore: Update question and response templates with current
 date, time, and day

---
 server/src/chain/index.ts | 18 +++++++++++++++++-
 1 file changed, 17 insertions(+), 1 deletion(-)

diff --git a/server/src/chain/index.ts b/server/src/chain/index.ts
index cf63f0c1..c8626695 100644
--- a/server/src/chain/index.ts
+++ b/server/src/chain/index.ts
@@ -21,6 +21,17 @@ type RetrievalChainInput = {
   question: string;
 };
 
+const updateTemplateVariables = (template: string) => {
+  // replace template {time} with current time
+  template = template.replace("{time}", new Date().toLocaleTimeString());
+  // replace template {date} with current date
+  template = template.replace("{date}", new Date().toLocaleDateString());
+  // replace template {day} with current day
+  template = template.replace("{day}", new Date().toLocaleString('en-us', { weekday: 'long' }));
+
+  return template;
+}
+
 export function groupMessagesByConversation(messages: any[]) {
   // check if messages are in even numbers if not remove the last message
   if (messages.length % 2 !== 0) {
@@ -109,12 +120,17 @@ export const createChain = ({
   retriever,
   response_template,
 }: {
-  llm: BaseLanguageModel<any> | BaseChatModel<any> ;
+  llm: BaseLanguageModel<any> | BaseChatModel<any>;
   question_llm: BaseLanguageModel<any> | BaseChatModel<any>;
   retriever: Runnable;
   question_template: string;
   response_template: string;
 }) => {
+
+  question_template = updateTemplateVariables(question_template);
+  
+  response_template = updateTemplateVariables(response_template);
+
   const retrieverChain = createRetrieverChain(
     question_llm,
     retriever,