ADD initial agent param

app.py

@@ -537,9 +537,22 @@ agent_OPENAI_MULTI = AgentExecutor.from_agent_and_tools(
 # agent.max_execution_time = int(os.getenv("max_iterations"))
 # agent.handle_parsing_errors = True
 # agent.early_stopping_method = "generate"
+
 global agent
+Choice = os.getenv("agent_type")
+
+if Choice =='Zero Short Agent':
+    agent = agent_ZEROSHOT_AGENT
+    print("Set to:", Choice)
+elif Choice =='Zero Short React':
+    agent = agent_ZEROSHOT_REACT
+    print("Set to:", Choice)
+elif Choice =='OpenAI Multi':
+    agent = agent_OPENAI_MULTI
+    print("Set to:", Choice)
 agent = agent_ZEROSHOT_AGENT
 
+
 print(agent.agent.llm_chain.prompt.template)
 # print(agent.agent.llm_chain.prompt)