| 69 | ] |
| 70 | |
| 71 | def reset(self): |
| 72 | self.tools = self._load_tools() |
| 73 | if self.model_name in OPENAI_COMPLETION_MODELS: |
| 74 | self.agent = initialize_agent(self.tools, |
| 75 | OpenAI(temperature=0, model_name=self.model_name), |
| 76 | agent=AgentType.REACT_DOCSTORE, |
| 77 | verbose=self.verbose, |
| 78 | return_intermediate_steps=True, |
| 79 | max_iterations=self.max_iter) |
| 80 | elif self.model_name in OPENAI_CHAT_MODELS: |
| 81 | self.agent = initialize_agent(self.tools, |
| 82 | ChatOpenAI(temperature=0, model_name=self.model_name), |
| 83 | agent=AgentType.REACT_DOCSTORE, |
| 84 | verbose=self.verbose, |
| 85 | return_intermediate_steps=True, |
| 86 | max_iterations=self.max_iter) |
| 87 | self.agent.agent.llm_chain.prompt.template = self.fewshot |
| 88 | |
| 89 | def _parse_tool(self, intermediate_steps): |
| 90 | tool_usage = {"search": 0, "lookup": 0} |