lakshmivairamani commited on
Commit
e0c6c19
1 Parent(s): 1a2596b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -123,9 +123,10 @@ def document_data_tool(query):
123
 
124
  # Initialize the agent with the tools
125
  tools = [
126
- 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.", tool_choice="required"),
127
  Tool(name="WeatherData", func=get_weather_data, description="Tool to get current weather data for a city or country. Handles questions related to current weather conditions in specific cities or countries.", tool_choice="required"),
128
  Tool(name="DocumentData", func=document_data_tool, description="Tool to search and retrieve information from the uploaded document.", tool_choice="required"),
 
129
  ]
130
 
131
  prompt_template = f"""You are an assistant that helps with database queries, weather information, and document retrieval.
 
123
 
124
  # Initialize the agent with the tools
125
  tools = [
126
+
127
  Tool(name="WeatherData", func=get_weather_data, description="Tool to get current weather data for a city or country. Handles questions related to current weather conditions in specific cities or countries.", tool_choice="required"),
128
  Tool(name="DocumentData", func=document_data_tool, description="Tool to search and retrieve information from the uploaded document.", tool_choice="required"),
129
+ 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.", tool_choice="required"),
130
  ]
131
 
132
  prompt_template = f"""You are an assistant that helps with database queries, weather information, and document retrieval.