JLW commited on
Commit
9a4d91b
·
1 Parent(s): 0d2a710

Continue improving conversational agent

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -52,12 +52,12 @@ def load_chain():
52
  """Logic for loading the chain you want to use should go here."""
53
  llm = OpenAI(temperature=0)
54
 
55
- tool_names = ['serpapi', 'pal-math', 'pal-colored-objects', 'tmdb-api']
56
  # tool_names = ['serpapi', 'pal-math', 'pal-colored-objects', 'news-api', 'tmdb-api', 'open-meteo-api']
57
 
58
  memory = ConversationBufferMemory(memory_key="chat_history")
59
 
60
- tools = load_tools(tool_names, llm=llm, tmdb_bearer_token=tmdb_bearer_token)
61
  # tools = load_tools(tool_names, llm=llm, news_api_key=news_api_key, tmdb_bearer_token=tmdb_bearer_token)
62
 
63
  chain = initialize_agent(tools, llm, agent="conversational-react-description", verbose=True, memory=memory)
 
52
  """Logic for loading the chain you want to use should go here."""
53
  llm = OpenAI(temperature=0)
54
 
55
+ tool_names = ['serpapi', 'pal-math', 'pal-colored-objects']
56
  # tool_names = ['serpapi', 'pal-math', 'pal-colored-objects', 'news-api', 'tmdb-api', 'open-meteo-api']
57
 
58
  memory = ConversationBufferMemory(memory_key="chat_history")
59
 
60
+ tools = load_tools(tool_names, llm=llm)
61
  # tools = load_tools(tool_names, llm=llm, news_api_key=news_api_key, tmdb_bearer_token=tmdb_bearer_token)
62
 
63
  chain = initialize_agent(tools, llm, agent="conversational-react-description", verbose=True, memory=memory)