mirror of
https://github.com/arc53/DocsGPT
synced 2024-11-02 03:40:17 +00:00
69 lines
1.6 KiB
Python
69 lines
1.6 KiB
Python
import os
|
|
import re
|
|
|
|
import discord
|
|
import requests
|
|
from discord.ext import commands
|
|
import dotenv
|
|
|
|
dotenv.load_dotenv()
|
|
|
|
# Replace 'YOUR_BOT_TOKEN' with your bot's token
|
|
TOKEN = os.getenv("DISCORD_TOKEN")
|
|
PREFIX = '@DocsGPT'
|
|
BASE_API_URL = 'http://localhost:5001'
|
|
|
|
intents = discord.Intents.default()
|
|
intents.message_content = True
|
|
|
|
bot = commands.Bot(command_prefix=PREFIX, intents=intents)
|
|
|
|
|
|
def split_string(input_str):
|
|
pattern = r'^<@!?{0}>\s*'.format(bot.user.id)
|
|
match = re.match(pattern, input_str)
|
|
if match:
|
|
content = input_str[match.end():].strip()
|
|
return str(bot.user.id), content
|
|
return None, input_str
|
|
|
|
|
|
@bot.event
|
|
async def on_ready():
|
|
print(f'{bot.user.name} has connected to Discord!')
|
|
|
|
|
|
async def fetch_answer(question):
|
|
data = {
|
|
'sender': 'discord',
|
|
'question': question,
|
|
'history': ''
|
|
}
|
|
headers = {"Content-Type": "application/json",
|
|
"Accept": "application/json"}
|
|
response = requests.post(BASE_API_URL + '/api/answer', json=data, headers=headers)
|
|
if response.status_code == 200:
|
|
return response.json()['answer']
|
|
return 'Sorry, I could not fetch the answer.'
|
|
|
|
|
|
@bot.event
|
|
async def on_message(message):
|
|
if message.author == bot.user:
|
|
return
|
|
|
|
content = message.content.strip()
|
|
prefix, content = split_string(content)
|
|
if prefix is None:
|
|
return
|
|
|
|
part_prefix = str(bot.user.id)
|
|
if part_prefix == prefix:
|
|
answer = await fetch_answer(content)
|
|
await message.channel.send(answer)
|
|
|
|
await bot.process_commands(message)
|
|
|
|
|
|
bot.run(TOKEN)
|