system prompt

app.py

@@ -1,3 +1,4 @@
+from langchain.prompts import ChatPromptTemplate, SystemMessagePromptTemplate, HumanMessagePromptTemplate
 import os
 import logging
 from dotenv import load_dotenv
@@ -22,11 +23,16 @@ I am Akilesh, your assistant for question-answering tasks! I'll respond as if yo
 """
 
 # Create a PromptTemplate with the custom system prompt
-prompt_template = PromptTemplate(
-    template=f"{
-        CUSTOM_SYSTEM_PROMPT}\n\nContext: {{context}}\n\nQuestion: {{question}}\n\nAnswer:",
-    input_variables=["context", "question"]
-)
+
+system_template = SystemMessagePromptTemplate.from_template(
+    CUSTOM_SYSTEM_PROMPT)
+human_template = HumanMessagePromptTemplate.from_template(
+    "Context: {context}\n\nQuestion: {question}")
+
+prompt_template = ChatPromptTemplate.from_messages([
+    system_template,
+    human_template
+])
 
 
 # Initialize the ChatOpenAI model