Aabbhishekk commited on
Commit
3454a86
1 Parent(s): 0ed257c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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,human_ask_tool,google_searchtool],
 
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,