Advertisement
Cabaca

Untitled

Aug 1st, 2024
193
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.48 KB | None | 0 0
  1. def run_agent(user_input):
  2.     intermediate_steps = []
  3.     while True:
  4.         result = agent_chain.invoke({
  5.             "input": user_input,
  6.             "intermediate_steps": intermediate_steps
  7.         })
  8.         if isinstance(result, AgentFinish):
  9.             return result.log
  10.  
  11.         tool = {
  12.             "get_web_results": get_web_results,
  13.         }[result.tool]
  14.  
  15.         observation = tool.run(result.tool_input)
  16.         intermediate_steps.append((result, observation))
  17.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement