Spaces:
Runtime error
Runtime error
lakshmivairamani
commited on
Commit
•
e0c6c19
1
Parent(s):
1a2596b
Update app.py
Browse files
app.py
CHANGED
@@ -123,9 +123,10 @@ def document_data_tool(query):
|
|
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 |
]
|
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.
|