You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

33 lines
870 B

2 years ago
import traceback
import aiohttp
import discord
from models.deepl_model import TranslationModel
from models.search_model import Search
from services.environment_service import EnvService
ALLOWED_GUILDS = EnvService.get_allowed_guilds()
2 years ago
class SearchService(discord.Cog, name="SearchService"):
"""Cog containing translation commands and retrieval of translation services"""
def __init__(
self,
bot,
gpt_model,
pinecone_service,
):
super().__init__()
self.bot = bot
self.model = Search(gpt_model, pinecone_service)
# Make a mapping of all the country codes and their full country names:
async def search_command(self, ctx, query):
"""Command handler for the translation command"""
await ctx.defer()
await self.model.search(query)
await ctx.respond("ok")