JLW commited on
Commit
0d2a710
·
1 Parent(s): 43760cc

Continue improving conversational agent

Browse files
Files changed (1) hide show
  1. app.py +11 -4
app.py CHANGED
@@ -4,6 +4,10 @@ from typing import Optional, Tuple
4
  import datetime
5
  import gradio as gr
6
  import requests
 
 
 
 
7
  from langchain import ConversationChain
8
 
9
  from langchain.agents import load_tools, initialize_agent
@@ -13,8 +17,7 @@ from langchain.llms import OpenAI
13
  news_api_key = os.environ["NEWS_API_KEY"]
14
  tmdb_bearer_token = os.environ["TMDB_BEARER_TOKEN"]
15
 
16
- import whisper
17
-
18
  WHISPER_MODEL = whisper.load_model("tiny")
19
  print("WHISPER_MODEL", WHISPER_MODEL)
20
 
@@ -48,11 +51,15 @@ def transcribe(aud_inp):
48
  def load_chain():
49
  """Logic for loading the chain you want to use should go here."""
50
  llm = OpenAI(temperature=0)
51
- tool_names = ['serpapi', 'pal-math', 'pal-colored-objects', 'news-api', 'tmdb-api', 'open-meteo-api']
 
 
52
 
53
  memory = ConversationBufferMemory(memory_key="chat_history")
54
 
55
- tools = load_tools(tool_names, llm=llm, news_api_key=news_api_key, tmdb_bearer_token=tmdb_bearer_token)
 
 
56
  chain = initialize_agent(tools, llm, agent="conversational-react-description", verbose=True, memory=memory)
57
  return chain
58
 
 
4
  import datetime
5
  import gradio as gr
6
  import requests
7
+
8
+ import warnings
9
+ import whisper
10
+
11
  from langchain import ConversationChain
12
 
13
  from langchain.agents import load_tools, initialize_agent
 
17
  news_api_key = os.environ["NEWS_API_KEY"]
18
  tmdb_bearer_token = os.environ["TMDB_BEARER_TOKEN"]
19
 
20
+ warnings.filterwarnings("ignore")
 
21
  WHISPER_MODEL = whisper.load_model("tiny")
22
  print("WHISPER_MODEL", WHISPER_MODEL)
23
 
 
51
  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)
64
  return chain
65