fffiloni commited on
Commit
7e89994
·
1 Parent(s): 0d0655a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,10 +9,10 @@ from langchain.tools import AIPluginTool
9
  def run(prompt, plugin_json, openai_api_key):
10
  os.environ["OPENAI_API_KEY"] = openai_api_key
11
  tool = AIPluginTool.from_plugin_url(plugin_json)
12
- llm = ChatOpenAI(temperature=0)
13
  tools = load_tools(["requests_all"])
14
  tools += [tool]
15
- agent_chain = initialize_agent(tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True)
16
  return agent_chain.run(prompt)
17
 
18
  title="""
 
9
  def run(prompt, plugin_json, openai_api_key):
10
  os.environ["OPENAI_API_KEY"] = openai_api_key
11
  tool = AIPluginTool.from_plugin_url(plugin_json)
12
+ llm = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0, max_tokens=1000)
13
  tools = load_tools(["requests_all"])
14
  tools += [tool]
15
+ agent_chain = initialize_agent(tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True, reduce_k_below_max_tokens=True)
16
  return agent_chain.run(prompt)
17
 
18
  title="""