update greeting

agents.py

@@ -70,12 +70,13 @@ class Relevant_Documents_Agent:
     self.embeddings = embeddings
     # Prompts
     self.prompt = """[INST]Answer STRICTLY with "Yes" or "No". 
+    This is important: If the user querie or the question below is a general greeting such as "hello" then you must reply with "Yes".
     Use the following context to check if the query is relevant or not.
     If the context is even slightly relevant then reply with "Yes" and 
     if the context and query are poles apart then reply with "No".
-    If the user queries with a general greeting such as "hello", mark this as relevant and return "Yes".
     Context: {context}
     Question: {query} [/INST]
+    This is important: If the user querie or the question below is a general greeting such as "hello" then you must reply with "Yes".
     """
     # Data
     self.query = ""
@@ -100,7 +101,7 @@ class Answering_Agent:
                                                             Utilize the insights from these texts to formulate a comprehensive and accurate response.
                                                             Your goal is to synthesize the information, highlight key points, and ensure the answer is informative and directly addresses the user's question.
                                                             You will also be given the previous history of chat as Context use it to influence the answer.
-                                                            If the user queries with any kind of general greeting such as "hello", respond with a general greeting as well.
+                                                            This is important: If the user queries with any kind of general greeting such as "hello", respond with a general greeting.
                                                             Relevant Text : {relevant_text}
                                                             User's Query: {query}
                                                             """)
@@ -124,3 +125,4 @@ class Answering_Agent:
 
   def generate_response(self):
     return self.chain.invoke(self.prompt)
+

app.py

@@ -97,6 +97,10 @@ class Head_Agent:
         # TODO: Get Query from Streamlit App
         # print("Query:", query)
 
+        # Query for greeting
+        if query.lower() in ["hello", "hi", "hey"]:
+            return "Hello! How can I help you today?"
+
         ## 1. Obnoxious Agent
         # Return yes/True = end
         # Return no/False = continue