Update app.py
Browse files
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():
|