OuroborosM commited on
Commit
d8a62af
·
1 Parent(s): 4a9eea6

add structured agent

Browse files
Files changed (1) hide show
  1. app.py +32 -2
app.py CHANGED
@@ -542,7 +542,7 @@ def UpdateDb():
542
 
543
  ListAgentWithRemoteGPT = ['Zero Short React 2','Zero Short Agent 2',
544
  'OpenAI Multi 2', 'Conversation Agent',
545
- 'Code Interpreter']
546
 
547
  def SummarizeDoc():
548
  global vectordb_p
@@ -1011,6 +1011,30 @@ agent_ZEROSHOT_REACT_2 = initialize_agent(tools_remote, GPTfake,
1011
 
1012
  )
1013
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1014
  agent_CODE_INTERPRETER = initialize_agent(tools_remote, GPTfake,
1015
  # agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
1016
  agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
@@ -1143,6 +1167,12 @@ def SetAgent(Choice):
1143
  elif Choice =='Code Interpreter':
1144
  agent = agent_CODE_INTERPRETER
1145
  print("Set to:", Choice)
 
 
 
 
 
 
1146
 
1147
 
1148
 
@@ -1702,7 +1732,7 @@ with gr.Blocks() as demo:
1702
  agentchoice = gr.Dropdown(
1703
  choices=['Zero Short Agent','Zero Short React','OpenAI Multi',
1704
  'Zero Short React 2','Zero Short Agent 2','None','Conversation Agent',
1705
- 'Code Interpreter'],
1706
  label="SELECT AI AGENT",
1707
  scale= 2,
1708
  show_label = True,
 
542
 
543
  ListAgentWithRemoteGPT = ['Zero Short React 2','Zero Short Agent 2',
544
  'OpenAI Multi 2', 'Conversation Agent',
545
+ 'Code Interpreter', 'Structured Zero Short Agent']
546
 
547
  def SummarizeDoc():
548
  global vectordb_p
 
1011
 
1012
  )
1013
 
1014
+ agent_STRUCTURED_ZEROSHOT_REACT = initialize_agent(tools_remote, GPTfake,
1015
+ # agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
1016
+ agent=AgentType.STRUCTURED_CHAT_ZERO_SHOT_REACT_DESCRIPTION,
1017
+ verbose = True,
1018
+ handle_parsing_errors = True,
1019
+ max_iterations = int(os.environ["max_iterations"]),
1020
+ early_stopping_method="generate",
1021
+ memory = memory2,
1022
+ agent_kwargs={
1023
+ 'prefix': PREFIX_2,
1024
+ 'format_instructions': FORMAT_INSTRUCTIONS_3,
1025
+ 'suffix': SUFFIX2,
1026
+ 'input_variables': input_variables,
1027
+
1028
+ },
1029
+ # input_variables = input_variables,
1030
+ # agent_kwargs={
1031
+ # 'prompt': prompt,
1032
+ # }
1033
+
1034
+ )
1035
+
1036
+
1037
+
1038
  agent_CODE_INTERPRETER = initialize_agent(tools_remote, GPTfake,
1039
  # agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
1040
  agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
 
1167
  elif Choice =='Code Interpreter':
1168
  agent = agent_CODE_INTERPRETER
1169
  print("Set to:", Choice)
1170
+ elif Choice =='Structured Zero Short Agent':
1171
+ agent = agent_STRUCTURED_ZEROSHOT_REACT
1172
+ print("Set to:", Choice)
1173
+
1174
+
1175
+
1176
 
1177
 
1178
 
 
1732
  agentchoice = gr.Dropdown(
1733
  choices=['Zero Short Agent','Zero Short React','OpenAI Multi',
1734
  'Zero Short React 2','Zero Short Agent 2','None','Conversation Agent',
1735
+ 'Code Interpreter', 'Structured Zero Short Agent'],
1736
  label="SELECT AI AGENT",
1737
  scale= 2,
1738
  show_label = True,