now commences tool clean up

requirements.txt

@@ -53,4 +53,5 @@ O365
 # whisperx
 pytube
 pydub
-git+https://github.com/m-bain/whisperx.git
+git+https://github.com/m-bain/whisperx.git
+llama-index

setup.py

@@ -37,6 +37,7 @@ setup(
         "controlnet-aux",
         "diffusers",
         "einops",
+        'llama-index',
         "gradio",
         "imageio",
         "imageio-ffmpeg",

swarms/tools/main.py

@@ -15,7 +15,7 @@ from langchain.llms.base import BaseLLM
 
 import requests
 from bs4 import BeautifulSoup
-from llama_index import GPTSimpleVectorIndex
+from llama_index import GPTVectorStoreIndex
 from llama_index.readers.database import DatabaseReader
 
 from env import settings
@@ -135,7 +135,7 @@ class WineDB(BaseToolSet):
             FROM wine
         """
         documents = db.load_data(query=query)
-        self.index = GPTSimpleVectorIndex(documents)
+        self.index = GPTVectorStoreIndex(documents)
 
     @tool(
         name="Wine Recommendation",