modified classification according to chatpromptTemplate

prompts.py

@@ -1,4 +1,4 @@
-from langchain.prompts import PromptTemplate
+from langchain_core.prompts import ChatPromptTemplate
 
 # Define a list of harmful or inappropriate topics to block
 HARMFUL_TOPICS = ["kill", "suicide", "harm", "death", "murder", "violence"]
@@ -10,7 +10,6 @@ You are a helpful assistant that classifies user questions into three categories
 3) "OutOfScope" if it does not fall into the above two categories or if the question involves harmful topics.
 **Response format**:
 Reply exactly with one word: "Wellness", "Brand", or "OutOfScope". Do not provide any additional explanation.
-Question: {query}
 """
 
 tailor_prompt_str = """
@@ -57,9 +56,11 @@ For more wellness-related questions or to learn more about DailyWellnessAI, feel
 
 # Define the PromptTemplate objects:
 
-classification_prompt = PromptTemplate(
-    template=classification_prompt_str,
-    input_variables=["query"]
+classification_prompt = ChatPromptTemplate(
+    [
+        ("system",classification_prompt_str),
+        ("user","{query}")
+    ]
 )
 
 tailor_prompt = PromptTemplate(