OuroborosM commited on
Commit
e4cdf4d
·
1 Parent(s): 36b2f88

correct prompt

Browse files
Files changed (1) hide show
  1. app.py +9 -7
app.py CHANGED
@@ -399,6 +399,7 @@ prompt = ZeroShotAgent.create_prompt(
399
  input_variables=["input", "chat_history", "agent_scratchpad"]
400
  )
401
 
 
402
 
403
  agent = initialize_agent(tools, llm,
404
  # agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
@@ -407,14 +408,15 @@ agent = initialize_agent(tools, llm,
407
  handle_parsing_errors = True,
408
  max_iterations = int(os.getenv("max_iterations")),
409
  early_stopping_method="generate",
410
- # agent_kwargs={
411
- # 'prefix': PREFIX,
412
- # 'format_instructions': FORMAT_INSTRUCTIONS,
413
- # 'suffix': SUFFIX
414
- # },
415
  agent_kwargs={
416
- 'prompt': prompt,
417
- }
 
 
 
 
 
 
418
 
419
  )
420
 
 
399
  input_variables=["input", "chat_history", "agent_scratchpad"]
400
  )
401
 
402
+ input_variables=["input", "chat_history", "agent_scratchpad"]
403
 
404
  agent = initialize_agent(tools, llm,
405
  # agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
 
408
  handle_parsing_errors = True,
409
  max_iterations = int(os.getenv("max_iterations")),
410
  early_stopping_method="generate",
 
 
 
 
 
411
  agent_kwargs={
412
+ 'prefix': PREFIX,
413
+ 'format_instructions': FORMAT_INSTRUCTIONS,
414
+ 'suffix': SUFFIX,
415
+ 'inputinput_variables': input_variables
416
+ },
417
+ # agent_kwargs={
418
+ # 'prompt': prompt,
419
+ # }
420
 
421
  )
422