Discord bot with chatGPT 作成日記
前置き
おはようございます、えるです❢
最近始めたdiscord bot作成方法を、自分のためにもまとめて見ることにしました
誰かの参考にもなれば嬉しいです
Discord アカウントを作成
※とりあえず省略
Discord botアカウントを作成
Botアカウント作成 (discordpy.readthedocs.io) の通りに作成します
ーえるさんの日記ー
ー完ー
…とか書くとnoteにならないので、以下は実際に作ってみたやつです❢
事前準備
discordにログインしておきます
自分用のdiscordサーバーも作っておきます
そしてdiscordのApplicationページにいきます
discord Applicationページ
右上の「New Application」をポチッと
bot名入力→規約確認→create
CreateするとBOTの設定ページに遷移します
general infomationページ
まずこのページに来ますが、そのままBotタブへ移動
Botページ
1) botアイコンにイラストを設定したい場合、ICONで画像をアップロードします
2) Reset Token ボタンを押し、表示されるトークンをコピーしておきます これは後ほど使用します
3) PUBLIC BOTをオフにします これはまあ、お好みで
4) MESSAGE CONTENT INTENTをオンにします
※3つともオンでも可
BOTページでやることは以上です
OAuth2タブに移動します
OAuth2ページ
URL Generatorに行きます
とりあえずメッセージ送受信の権限を追加します
右下のコピーを押して、URLをコピーします
これが、BOTを自サーバーに追加するURLになります
コピーしたURLを適当なWEBブラウザで開きます
自分のサーバーを選択→はい
→認証
→人間です！
追加できました！
サーバーに追加しただけではBOTが反応出来ないので、適当なチャンネルにBOTを呼び込みます
私の場合は新しくdebugチャンネルを作り、「メンバーまたはロールを追加」からBOTをメンバー追加しました
BOTを追加しただけでは何も出来ないので、次に
BOTを動かすためのコードを書いていきます
Discord bot コード：Hellow
Google Colabで作ります
1) discord.pyをインストール
!pip install discord.py
2) トークンを入力
トークンはdiscordのbotページで取得したものです
import os
token = "ここにトークンを入力"#@param {type:"string"}
os.environ['TOKEN'] = token
3) 実行
main.pyはテキストで別に作成して、colabにアップロードしておきます
!python3 main.py
main.pyの中身
#main
import discord
import os
#initial
intents = discord.Intents.default()
intents.message_content = True
client = discord.Client(intents=intents)
token = os.environ['TOKEN']
@client.event
async def on_ready():
print(f'We have logged in as {client.user}')
@client.event
async def on_message(message):
if message.author.bot == True:
return
else:
await message.channel.send("hellow!")
client.run(token)Discordで動作確認
出来ました❣
【注意事項】
その1)
discord botページの 4) MESSAGE CONTENT INTENTをオン しない場合、main.py実行後にエラーで止まります
その2)
main.pyの記述をファイル化せず、Colabのセルに記述を貼り付けて実行するとエラーで止まります
#まだ原因よくわかってないです… Python versionとか？
OpenAI APIを取得
※とりあえず省略
Discord botコード：chatGPT
相変わらず、colabで作ります
1) インストール
!pip install discord.py
!pip install openai2) DiscordトークンとOpenAIのAPI Keyを入力
import os
token = "ここにdiscordトークンを入力"#@param {type:"string"}
os.environ['TOKEN'] = token
openai_apikey = "ここにOpenAIのAPI Keyを入力"#@param {type:"string"}
os.environ['OPENAI_API'] = openai_apikey
3) 実行
前回同様にmain.pyはテキストで別に作成して、colabにアップロードしておきます
!python3 main.py今回のmain.py
# main
import discord
import os
import openai
#initial
intents = discord.Intents.default()
intents.message_content = True
client = discord.Client(intents=intents)
token = os.environ['TOKEN']
openai.api_key = os.environ['OPENAI_API']
@client.event
async def on_ready():
print(f'We have logged in as {client.user}')
@client.event
async def on_message(message):
if message.author.bot == True:
return
else:
text = message.content
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=[
{"role": "system", "content": "あなたはAIアシスタントです"},
{"role": "user", "content":text}
]
)
await message.channel.send(response['choices'][0]['message']['content'])
client.run(token)【補足】
”あなたはAIアシスタントです” の部分を変更すると、AIの性格を方向付けできます
Discordで動作確認
動きました❣
Discord botコード：短期記憶
chatGPT APIはweb版とは異なり、そのままだと会話内容を覚えていません
覚えてほしいので、短期的な記憶を実装していきます
手段はきっと色々あるのですが、とりあえずlangchainさんに頑張ってもらいます
Welcome to LangChain — 🦜🔗 LangChain 0.0.152
そして今日もcolabで作ります
1) インストール
!pip install discord.py
!pip install openai
!pip install langchain
!pip install tiktoken2) DiscordトークンとOpenAIのAPI Keyを入力
import os
token = "ここにdiscordトークンを入力"#@param {type:"string"}
os.environ['TOKEN'] = token
openai_apikey = "ここにOpenAIのAPI Keyを入力"#@param {type:"string"}
os.environ['OPENAI_API_KEY'] = openai_apikey
3) 実行
今までと同様に、main.pyは別に作成してアップロードします
!python3 main.pymain.pyの中身は以下の通り
使用するmemory.txtは空ファイルで良いのでアップロードしておきます
# main
#import
import discord
import os
import openai
#langchain import
from langchain.prompts import (
ChatPromptTemplate,
MessagesPlaceholder,
SystemMessagePromptTemplate,
HumanMessagePromptTemplate
)
from langchain.chains import ConversationChain
from langchain.chat_models import ChatOpenAI
from langchain.chains.conversation.memory import ConversationSummaryBufferMemory
#Discord initial
intents = discord.Intents.default()
intents.message_content = True
client = discord.Client(intents=intents)
#token and API seting
token = os.environ['TOKEN']
openai.api_key = os.environ['OPENAI_API_KEY']
#langchain initial
system_settings = """あなたはAIアシスタントです"""
prompt = ChatPromptTemplate.from_messages([
SystemMessagePromptTemplate.from_template(system_settings),
MessagesPlaceholder(variable_name="history"),
HumanMessagePromptTemplate.from_template("{input}")
])
use_model = "gpt-3.5-turbo"
S_conversation = ConversationChain(
memory=ConversationSummaryBufferMemory(
return_messages=True,
llm=ChatOpenAI(model_name=use_model),
max_token_limit=500
),
prompt=prompt,
llm=ChatOpenAI(model_name=use_model),
verbose=True
)
# buffer load
f = open('memory.txt', 'r')
memory_text = f.read()
f.close()
S_conversation.predict(input=memory_text)
#Discord event
@client.event
async def on_ready():
print(f'We have logged in as {client.user}')
@client.event
async def on_message(message):
if message.author.bot == True:
return
else:
text = message.content
S_text = S_conversation.predict(input=text)
await message.channel.send(S_text)
S_memory_text = S_conversation.memory.load_memory_variables({})
Sf = open('memory.txt', 'w')
Sf.write(str(S_memory_text))
Sf.close()
client.run(token)だんだん長くなってきました。。
Discordで動作確認
先ほどと同じやりとりをしてみます
覚えてくれました❣
空ファイルで作成したmemory.txtですが、会話後は会話内容が記載されます
Bot立ち上げ時も最初に前回の会話内容をinputしにいくため、会話内容をある程度保持しておくことができます
Colabにアップロードしたファイルは削除されてしまいますが、GoogleDriveをマウントしそちらに保存することで解決できるかと思います
【注意事項】
記憶のためのデータのやり取りはもっと効率の良いやり方があるはずですが、現状の方法でもある程度可能なためひとまず良いかと思っています。
Discord botコード：Streaming
chatGPT API はデフォルトだとメッセージを全文作成後に応答します
一方でchatGPT のweb版はストリーミングテキストなので、初動レスポンスが速いです
上記に憧れたのでStreaming機能を試してみました
結果
↓
これは求めていたものと違う…
と思ったのでいろいろ調べてみましたが、discordはメッセージ送信時に改行が入ってしまうので、web版風は今のところ出来ていません。。
とは言え、なんとかならないかなーと頑張ってみました
以下が現状実装です まだ試行錯誤中なのでcolabでは無いです
import asyncio
import json
import aiohttp
import os
import discord
bot_token = os.environ['KEMI_BOT_KEY']
openai_key = os.environ['GPT_KEY']
url = "https://api.openai.com/v1/chat/completions"
### discord initial
intents = discord.Intents.default()
intents.message_content = True
client = discord.Client(intents=intents)
message_chunks = []
@client.event
async def on_ready():
print(f'{client.user.name} has connected to Discord!')
@client.event
async def on_message(message):
print(message.content)
if message.author == client.user:
return
async with aiohttp.ClientSession() as session:
headers = {
"Content-Type": "application/json",
"Authorization": f"Bearer {openai_key}",
}
data = {
"model": "gpt-3.5-turbo",
"messages": [
{"role": "system","content": "あなたは優秀なアシスタントです。質問内容に対して可能な限り簡潔に回答してください",},
{"role": "user","content": message.content,},
],
"stream": True,
}
async with session.post(url, headers=headers, data=json.dumps(data)) as resp:
async for line in resp.content:
if line.strip() == b'data: [DONE]':
print("Stream finished")
response_message = "".join(message_chunks)
if response_message.strip():
await message.channel.send(response_message)
message_chunks.clear()
break
else:
try:
response = json.loads(line[6:]) # skip the 'data: ' part of the SSE
content = response["choices"][0]["delta"].get("content", "")
if content:
message_chunks.append(content)
if "\n" in content or "。" in content:
response_message = "".join(message_chunks)
await message.channel.send(response_message)
message_chunks.clear()
#if content:
# await message.channel.send(content)
except json.JSONDecodeError as e:
print(f"Failed to parse JSON: {line[6:]}"+"content:"+content)
client.run(bot_token)頑張りポイントが伝わりにくそうだけど、昨日頑張ったdiscordストリーミングテキストの成果🥳 pic.twitter.com/iogkKZHqC4
— える (@el_el_san) May 20, 2023
できました❣
文章途中で送信（＝改行）すると変なところで改行されてしまうので、
改行コードや句点で送信するようにしています
全文待ちよりは体感かなり良い感じになったかと、個人的には満足しています
でもほんとはweb版のようなStreamingをdiscordでもしたい。。
何か方法はあるのかなあ
ーーー
ひとまずはここまでです❢ 以降は考え中の内容も含むため、時間を要する気がします。
省略箇所も記載要望があれば追記を考えたいと思います
Discord botコード：長期記憶
※実装方法考え中
Discord botコード：ボイス
※作成中