Update app.py

app.py

@@ -151,20 +151,20 @@ tools = [
     
     Tool(name="ASNData", func=get_ASN_data, description="Tool to get the status of ASN with ASN id given as input. Handles questions related to ASN id which starts with ASN followed by 11 numeric digits. For example, ASN24070100015 ", tool_choice="required"),
     Tool(name="DocumentData", func=document_data_tool, description="Tool to search and retrieve information from the uploaded document. Provide responses with the maximum of 150 words.", tool_choice="required"),
-    #Tool(name="DatabaseQuery", func=database_tool, description="Tool to query the database based on the user's question. Only handles questions related to the collegedb schema, including tables such as buildings, classrooms, college, course, faculty, interns, person, section, student, and textbook. Ensure to use only the available fields in these tables.Provide responses with the maximum of 150 words.", tool_choice="required"),
+    Tool(name="DatabaseQuery", func=database_tool, description="Tool to query the database based on the user's question. Only handles questions related to the collegedb schema, including tables such as buildings, classrooms, college, course, faculty, interns, person, section, student, and textbook. Ensure to use only the available fields in these tables.Provide responses with the maximum of 150 words.", tool_choice="required"),
 ]
 
 prompt_template = f"""You are an assistant that helps with database queries, weather information, and document retrieval.
 
 For ASN-related questions, if the user specifies ASN id. Provide the information like ASN status, expected Receiving Date etc.
 For document-related questions, search and retrieve information from the uploaded document.
-
+For SQL database-related questions, only use the fields available in the collegedb schema, which includes tables such as buildings, classrooms, college, course, faculty, interns, person, section, student, and textbook.
 {{agent_scratchpad}}
 Question: {{input}}
 
 """
 #{{memory.buffer}}
-#For SQL database-related questions, only use the fields available in the collegedb schema, which includes tables such as buildings, classrooms, college, course, faculty, interns, person, section, student, and textbook.
+
 prompt = ChatPromptTemplate.from_template(prompt_template)
 
 # Initialize the agent with memory