now commences tool clean up

main
Kye 2 years ago
parent f7e722e928
commit 171809a135

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

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

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

Loading…
Cancel
Save