Add new agent openai multiple

app.py

@@ -461,6 +461,8 @@ def UpdateDb():
     print("Pinecone Updated Done")
     print(index.describe_index_stats())
 
+ListAgentWithRemoteGPT = ['Zero Short React 2', 'Zero Short Agent 2', 'OpenAI Multi 2']
+
 def SummarizeDoc():
     global vectordb_p
     global Choice
@@ -473,7 +475,7 @@ def SummarizeDoc():
     print(split_docs[tt-1])
     sum_text=""
     try:
-        if Choice == 'Zero Short React 2' or Choice == "Zero Short Agent 2":
+        if Choice in ListAgentWithRemoteGPT:
             sum_chain = load_summarize_chain(GPTfake, chain_type='refine', verbose=True)
         else:
             sum_chain = load_summarize_chain(llm, chain_type='refine', verbose=True)
@@ -925,6 +927,8 @@ agent_core_2 = ZeroShotAgent(llm_chain=llm_chain_2, tools=tools2, verbose=True)
 
 agent_core_openai = OpenAIMultiFunctionsAgent(llm=llm, tools=tools, prompt=prompt_openai, verbose=True)
 
+agent_core_openai_2 = OpenAIMultiFunctionsAgent(llm=GPTfake, tools=tools, prompt=prompt_openai, verbose=True)
+
 agent_ZEROSHOT_AGENT = AgentExecutor.from_agent_and_tools(
     agent=agent_core, 
     tools=tools2, 
@@ -956,6 +960,17 @@ agent_OPENAI_MULTI = AgentExecutor.from_agent_and_tools(
     early_stopping_method="generate",
     )
 
+agent_OPENAI_MULTI_2 = AgentExecutor.from_agent_and_tools(
+    agent=agent_core_openai_2, 
+    tools=tools, 
+    verbose=True, 
+    # memory=memory_openai,
+    handle_parsing_errors = True,
+    max_iterations = int(os.environ["max_iterations"]),
+    early_stopping_method="generate",
+    )
+
+
 # agent.max_execution_time = int(os.getenv("max_iterations"))
 # agent.handle_parsing_errors = True
 # agent.early_stopping_method = "generate"
@@ -980,6 +995,9 @@ def SetAgent(Choice):
     elif Choice == "None":
         agent = None
         print("Set to:", Choice)
+    elif Choice =='OpenAI Multi 2':
+        agent = agent_OPENAI_MULTI_2
+        print("Set to:", Choice)
 
 global agent
 Choice = os.environ["agent_type"]
@@ -1468,7 +1486,8 @@ with gr.Blocks() as demo:
         with gr.Row():
 
             agentchoice = gr.Dropdown(
-                choices=['Zero Short Agent','Zero Short React','OpenAI Multi','Zero Short React 2','Zero Short Agent 2','None'],
+                choices=['Zero Short Agent','Zero Short React','OpenAI Multi',
+                         'Zero Short React 2','Zero Short Agent 2','None','OpenAI Multi 2'],
                 label="SELECT AI AGENT",
                 scale= 2,
                 show_label = True,
@@ -1585,7 +1604,7 @@ def QAQuery_p(question: str):
     retriever.search_kwargs['k'] = int(os.environ["search_kwargs_k"])
     # retriever.search_kwargs['fetch_k'] = 100
     # if agent == agent_ZEROSHOT_REACT_2 or agent == agent_ZEROSHOT_AGENT_2:
-    if Choice == 'Zero Short React 2' or Choice == "Zero Short Agent 2":
+    if Choice in ListAgentWithRemoteGPT:
         print("--------------- QA with Remote --------------")
         qa = RetrievalQA.from_chain_type(llm=GPTfake, chain_type="stuff", 
                                         retriever=retriever, return_source_documents = True,