Update app.py

app.py

@@ -12,7 +12,7 @@ import openai
 from openai.error import AuthenticationError, InvalidRequestError, RateLimitError
 from typing import Optional, Tuple
 
-TOOLS_DEFAULT_LIST = ['serpapi', 'pal-math']
+TOOLS_DEFAULT_LIST = ['serpapi', 'news-api', 'pal-math']
 MAX_TOKENS = 512
 PROMPT_TEMPLATE = PromptTemplate(
     input_variables=["original_words"],
@@ -22,6 +22,8 @@ PROMPT_TEMPLATE = PromptTemplate(
 BUG_FOUND_MSG = "Congratulations, you've found a bug in this application!"
 AUTH_ERR_MSG = "Please paste your OpenAI key."
 
+news_api_key = os.environ["NEWS_API_KEY"]
+
 def run_chain(chain, inp, capture_hidden_text):
     output = ""
     hidden_text = None
@@ -93,7 +95,7 @@ def load_chain(tools_list, llm):
     express_chain = None
     print("\ntools_list", tools_list)
     tool_names = tools_list
-    tools = load_tools(tool_names, llm=llm)
+    tools = load_tools(tool_names, llm=llm, news_api_key=news_api_key)
 
     memory = ConversationBufferMemory(memory_key="chat_history")