Prat0 commited on
Commit
beb1eae
·
verified ·
1 Parent(s): 085f1f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -14,7 +14,8 @@ from fastapi.middleware.cors import CORSMiddleware
14
  from llama_index.core.agent import ReActAgent
15
  from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
16
  from llama_index.core.llms import ChatMessage
17
-
 
18
 
19
  dotenv.load_dotenv()
20
 
@@ -41,6 +42,7 @@ state = {
41
  }
42
 
43
  os.environ["GOOGLE_API_KEY"] = os.getenv("GOOGLE_API_KEY")
 
44
 
45
  # Setup functions
46
  def embed_setup():
 
14
  from llama_index.core.agent import ReActAgent
15
  from llama_index.tools.duckduckgo import DuckDuckGoSearchToolSpec
16
  from llama_index.core.llms import ChatMessage
17
+ import nltk
18
+ nltk.download('punkt')
19
 
20
  dotenv.load_dotenv()
21
 
 
42
  }
43
 
44
  os.environ["GOOGLE_API_KEY"] = os.getenv("GOOGLE_API_KEY")
45
+ os.environ["COMPOSIO_API_KEY"] = os.getenv("COMPOSIO_API_KEY")
46
 
47
  # Setup functions
48
  def embed_setup():