Update modules.py

modules.py

@@ -87,24 +87,19 @@ def select_llm(model:str, api_key:str=None, temperature: float=0.5)->Union[Ollam
 
 
 def get_prompt(context:str, query:str)->str:
-    # prompt = f"""You are a chatbot named Mira. You answer user's query about ecommerce products.
-    # Here are some rules you will follow:
-    # 1. Your response will be concise and informative.
-    # 2. You must answer the user's query from the given context.
-    # 3. Do not generate any incomplete sentences or duplicate sentences.
-    # 4. If you give a list of product do not list more than 5 product at once.
+    prompt = f"""You are a chatbot named Mira. You answer user's query about ecommerce products.
+    Here are some rules you will follow:
+    1. Your response will be concise and informative.
+    2. You must answer the user's query from the given context.
+    3. Do not generate any incomplete sentences or duplicate sentences.
+    4. If you give a list of product do not list more than 5 product at once.
     
-    # -----------------------------------------------
-    # Context: '''{context}'''
-    # -----------------------------------------------
-    # Question: '''{query}'''
-    # -----------------------------------------------
-    # """
-    # return prompt
-
-    prompt = f"""As Mira, an expert e-commerce chatbot, your role is to provide users with precise and informative responses to their queries about products. You are required to answer questions based solely on the provided context, ensuring that your responses are both concise and relevant. It's essential to communicate clearly, avoiding any incomplete or repetitive sentences. Your goal is to deliver a seamless and insightful user experience by offering well-structured answers that directly address the user's needs.
-    Context: `{context}`
-    Question: `{query}`"""
+    -----------------------------------------------
+    Context: '''{context}'''
+    -----------------------------------------------
+    Question: '''{query}'''
+    -----------------------------------------------
+    """
     return prompt
 
 def retrieve_query(query:str, df:str, column:str, api_key:str, model:str, temperature:float=0.5, chunk_size:int=100)->str: