some early search adjustments

Kaveen Kumarasinghe 1 year ago
parent 827216f272
commit 3eaee2f041

@ -19,11 +19,10 @@ class SearchService(discord.Cog, name="SearchService"):
self,
bot,
gpt_model,
pinecone_service,
):
super().__init__()
self.bot = bot
self.model = Search(gpt_model, pinecone_service)
self.model = Search(gpt_model)
# Make a mapping of all the country codes and their full country names:
async def search_command(self, ctx, query, search_scope):

@ -63,22 +63,8 @@ if PINECONE_TOKEN:
metric="dotproduct",
pod_type="s1",
)
PINECONE_INDEX_SEARCH = "search-embeddings"
if (
EnvService.get_google_search_api_key()
and EnvService.get_google_search_engine_id()
):
if PINECONE_INDEX_SEARCH not in pinecone.list_indexes():
print("Creating pinecone index for searches. Please wait...")
pinecone.create_index(
PINECONE_INDEX_SEARCH,
dimension=1536,
metric="dotproduct",
pod_type="s1",
)
pinecone_service = PineconeService(pinecone.Index(PINECONE_INDEX))
pinecone_search_service = PineconeService(pinecone.Index(PINECONE_INDEX_SEARCH))
print("Got the pinecone service")
#
@ -185,7 +171,7 @@ async def main():
EnvService.get_google_search_api_key()
and EnvService.get_google_search_engine_id()
):
bot.add_cog(SearchService(bot, model, pinecone_search_service))
bot.add_cog(SearchService(bot, model))
print("The Search service is enabled.")
bot.add_cog(

@ -14,9 +14,8 @@ from services.usage_service import UsageService
class Search:
def __init__(self, gpt_model, pinecone_service):
def __init__(self, gpt_model):
self.model = gpt_model
self.pinecone_service = pinecone_service
self.google_search_api_key = EnvService.get_google_search_api_key()
self.google_search_engine_id = EnvService.get_google_search_engine_id()
self.loop = asyncio.get_running_loop()

Loading…
Cancel
Save