LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2026-04-28 13:00:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

discord bot fix

Browse Source 
Stop random answers
This commit is contained in:
Pavel
2023-03-28 01:51:54 +04:00
parent 70bc7465c9
commit 2ebe5e051c
1 changed files with 7 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
extensions/discord/bot.py
View File

@@ -10,7 +10,7 @@ dotenv.load_dotenv()
# Replace 'YOUR_BOT_TOKEN' with your bot's token # Replace 'YOUR_BOT_TOKEN' with your bot's token
TOKEN = os.getenv("DISCORD_TOKEN") TOKEN = os.getenv("DISCORD_TOKEN")
PREFIX = '@docsgpt ' PREFIX = '@DocsGPT'
BASE_API_URL = 'http://localhost:5001' BASE_API_URL = 'http://localhost:5001'
intents = discord.Intents.default() intents = discord.Intents.default()
@@ -20,13 +20,11 @@ bot = commands.Bot(command_prefix=PREFIX, intents=intents)
def split_string(input_str): def split_string(input_str):
pattern = r'<(.*?)>' pattern = r'^<@!?{0}>\s*'.format(bot.user.id)
match = re.search(pattern, input_str) match = re.match(pattern, input_str)
if match: if match:
content = match.group(1) content = input_str[match.end():].strip()
rest = input_str[:match.start()] + input_str[match.end():] return str(bot.user.id), content
return content, rest.strip()
return None, input_str return None, input_str
@@ -59,8 +57,8 @@ async def on_message(message):
if prefix is None: if prefix is None:
return return
part_prefix = "@" part_prefix = str(bot.user.id)
if part_prefix in prefix: if part_prefix == prefix:
answer = await fetch_answer(content) answer = await fetch_answer(content)
await message.channel.send(answer) await message.channel.send(answer)