Aabbhishekk
commited on
Commit
•
3454a86
1
Parent(s):
0ed257c
Update app.py
Browse files
app.py
CHANGED
@@ -115,11 +115,13 @@ def main():
|
|
115 |
human_ask_tool = CustomAskHumanTool()
|
116 |
# agent prompt
|
117 |
prefix, format_instructions, suffix = create_agent_prompt()
|
|
|
118 |
|
119 |
# initialize agent
|
120 |
agent = initialize_agent(
|
121 |
agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
|
122 |
-
tools=[db_search_tool,
|
|
|
123 |
llm=llm,
|
124 |
verbose=True,
|
125 |
max_iterations=5,
|
|
|
115 |
human_ask_tool = CustomAskHumanTool()
|
116 |
# agent prompt
|
117 |
prefix, format_instructions, suffix = create_agent_prompt()
|
118 |
+
mode = "Agent with AskHuman tool"
|
119 |
|
120 |
# initialize agent
|
121 |
agent = initialize_agent(
|
122 |
agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
|
123 |
+
tools=[human_ask_tool,db_search_tool,google_searchtool]
|
124 |
+
if mode == "Agent with AskHuman tool"
|
125 |
llm=llm,
|
126 |
verbose=True,
|
127 |
max_iterations=5,
|