diff --git a/cogs/search_service_cog.py b/cogs/search_service_cog.py index 48fc950..ab12fbf 100644 --- a/cogs/search_service_cog.py +++ b/cogs/search_service_cog.py @@ -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): diff --git a/gpt3discord.py b/gpt3discord.py index 7ac09c6..c6b9e67 100644 --- a/gpt3discord.py +++ b/gpt3discord.py @@ -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( diff --git a/models/search_model.py b/models/search_model.py index 03ea3f7..660cf66 100644 --- a/models/search_model.py +++ b/models/search_model.py @@ -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()