update llm

app.py

@@ -521,9 +521,9 @@ chat = AzureChatOpenAI(
 )
 
 
-# llm = chat
+llm = chat
 
-llm = GPTfake
+# llm = GPTfake
 
 llm_math = LLMMathChain.from_llm(llm)
 
@@ -654,6 +654,7 @@ prompt_openai = OpenAIMultiFunctionsAgent.create_prompt(
 
 input_variables=["input", "chat_history", "agent_scratchpad"]
 
+
 agent_ZEROSHOT_REACT = initialize_agent(tools2, llm, 
                         #  agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
                          agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
@@ -674,6 +675,27 @@ agent_ZEROSHOT_REACT = initialize_agent(tools2, llm,
                          
                         )
 
+agent_ZEROSHOT_REACT_2 = initialize_agent(tools2, GPTfake, 
+                        #  agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
+                         agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
+                         verbose = True,
+                         handle_parsing_errors = True,
+                         max_iterations = int(os.environ["max_iterations"]),
+                         early_stopping_method="generate",
+                         agent_kwargs={
+                            'prefix': PREFIX,
+                            'format_instructions': FORMAT_INSTRUCTIONS,
+                            'suffix': SUFFIX,
+                            # 'input_variables': input_variables,
+                         },
+                        #  input_variables = input_variables,
+                        #  agent_kwargs={
+                        #     'prompt': prompt,
+                        #  }
+                         
+                        )
+
+
 llm_chain = LLMChain(llm=llm, prompt=prompt)
 
 # llm_chain_openai = LLMChain(llm=llm, prompt=prompt_openai, verbose=True)
@@ -718,6 +740,10 @@ elif Choice =='Zero Short React':
 elif Choice =='OpenAI Multi':
     agent = agent_OPENAI_MULTI
     print("Set to:", Choice)
+elif Choice =='Zero Short React 2':
+    agent = agent_ZEROSHOT_REACT_2
+    print("Set to:", Choice)
+
 # agent = agent_ZEROSHOT_AGENT
 
 
@@ -1185,7 +1211,7 @@ with gr.Blocks() as demo:
         with gr.Row():
 
             agentchoice = gr.Dropdown(
-                choices=['Zero Short Agent','Zero Short React','OpenAI Multi'],
+                choices=['Zero Short Agent','Zero Short React','OpenAI Multi','Zero Short React 2'],
                 label="SELECT AI AGENT",
                 scale= 2,
                 show_label = True,