OuroborosM commited on
Commit
bcfb892
·
1 Parent(s): 9e7afb7

add conversation agenet

Browse files
Files changed (1) hide show
  1. app.py +28 -5
app.py CHANGED
@@ -461,7 +461,8 @@ def UpdateDb():
461
  print("Pinecone Updated Done")
462
  print(index.describe_index_stats())
463
 
464
- ListAgentWithRemoteGPT = ['Zero Short React 2', 'Zero Short Agent 2', 'OpenAI Multi 2']
 
465
 
466
  def SummarizeDoc():
467
  global vectordb_p
@@ -916,6 +917,28 @@ agent_ZEROSHOT_REACT_2 = initialize_agent(tools_remote, GPTfake,
916
 
917
  )
918
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
919
 
920
  llm_chain = LLMChain(llm=llm, prompt=prompt)
921
  llm_chain_2 = LLMChain(llm=GPTfake, prompt=prompt)
@@ -997,9 +1020,9 @@ def SetAgent(Choice):
997
  elif Choice == "None":
998
  agent = None
999
  print("Set to:", Choice)
1000
- # elif Choice =='OpenAI Multi 2':
1001
- # agent = agent_OPENAI_MULTI_2
1002
- # print("Set to:", Choice)
1003
 
1004
  global agent
1005
  Choice = os.environ["agent_type"]
@@ -1489,7 +1512,7 @@ with gr.Blocks() as demo:
1489
 
1490
  agentchoice = gr.Dropdown(
1491
  choices=['Zero Short Agent','Zero Short React','OpenAI Multi',
1492
- 'Zero Short React 2','Zero Short Agent 2','None','OpenAI Multi 2'],
1493
  label="SELECT AI AGENT",
1494
  scale= 2,
1495
  show_label = True,
 
461
  print("Pinecone Updated Done")
462
  print(index.describe_index_stats())
463
 
464
+ ListAgentWithRemoteGPT = ['Zero Short React 2','Zero Short Agent 2',
465
+ 'OpenAI Multi 2', 'Conversation Agent']
466
 
467
  def SummarizeDoc():
468
  global vectordb_p
 
917
 
918
  )
919
 
920
+ agent_CONVERSATION = initialize_agent(tools_remote, GPTfake,
921
+ # agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
922
+ agent=AgentType.CONVERSATIONAL_REACT_DESCRIPTION,
923
+ verbose = True,
924
+ handle_parsing_errors = True,
925
+ max_iterations = int(os.environ["max_iterations"]),
926
+ early_stopping_method="generate",
927
+ memory = memory2,
928
+ # agent_kwargs={
929
+ # 'prefix': PREFIX_2,
930
+ # 'format_instructions': FORMAT_INSTRUCTIONS_3,
931
+ # 'suffix': SUFFIX2,
932
+ # 'input_variables': input_variables,
933
+
934
+ # },
935
+ # input_variables = input_variables,
936
+ # agent_kwargs={
937
+ # 'prompt': prompt,
938
+ # }
939
+
940
+ )
941
+
942
 
943
  llm_chain = LLMChain(llm=llm, prompt=prompt)
944
  llm_chain_2 = LLMChain(llm=GPTfake, prompt=prompt)
 
1020
  elif Choice == "None":
1021
  agent = None
1022
  print("Set to:", Choice)
1023
+ elif Choice =='Conversation Agent':
1024
+ agent = agent_CONVERSATION
1025
+ print("Set to:", Choice)
1026
 
1027
  global agent
1028
  Choice = os.environ["agent_type"]
 
1512
 
1513
  agentchoice = gr.Dropdown(
1514
  choices=['Zero Short Agent','Zero Short React','OpenAI Multi',
1515
+ 'Zero Short React 2','Zero Short Agent 2','None','Conversation Agent'],
1516
  label="SELECT AI AGENT",
1517
  scale= 2,
1518
  show_label = True,