研究室の先生からChatGPTのAPIをもらったので、試しにChatGPTとしりとりができるコードを作ってみた。
ChatGPTの現状
・・・全然ダメやん！
こんな感じで、しりとりのルールを教えても全然うまくいかないのです。
そんなChatGPTのAPIを使って、きちんとしりとりができるようなものを作ってみた、というものです。
実行結果
とりあえず最終的にはこんなものができましたよ、というものです。
・名詞以外が来たらダメです
・一度ゲームに出てきた単語もダメです
・最後が「ん」ももちろんダメです
・「ヨーヨー」のように、最後に「ー」が付いてても対応しています。
また、例えば「最初」→「予防接種」というように、最後が「ょ」の場合は「よ」に変換するようにしています。
使ったライブラリ
上から、Juman++という形態素解析を行うライブラリ、ChatGPTのAPIを使うためにopenAIのインストール、jaconvというひらがなとカタカタの相互変換を行ってくれるやつです。
from pyknp import Juman
import openai
import jaconv
方針
①ユーザーが言葉を入力します。
②その言葉がしりとりのルールを満たしているかどうかをチェック→満たしていなければ負けと出力しゲーム終了。満たしていればChatGPTの回答に移る。
③ ChatGPTへのプロンプトとしては、「ユーザーが入力した言葉の最後の文字から始まる言葉を一つ言ってください」だけ。重要なのはこの後で、これだけではChatGPTは平気で全然関係ない言葉を言うので、しりとりのルールを満たす言葉を言うまでwhile文を回して聞き続ける。(だからChatGPTが負ける可能性がない)
④条件を満たす言葉を言ったら、それをChatGPTの回答とする
⑤以上の操作を繰り返す
と言った感じです。しりとりのルールを満たしているかどうかの判定が作る上で一番めんどかった。(名詞ではない、最後の文字からは始まってない、最後が「ん」で終わらない、同じ言葉を2回言う...など)
コードについて
クラスとインスタンス変数
word_listは、ゲームの実行中に出てきた言葉を管理するリスト(重複なし)です。これから書く関数は全てShiritoriGameのクラス下にあります。
class ShiritoriGame:
def __init__(self):
self.N = 100
self.word_list = set()
self.juman = Juman()
self.last_word = ""
self.openai_organization = "org-xxx"
self.openai_api_key = "sk-xxx"
APIの利用
これでChatGPTのAPIを使ってChatGPTを呼び出しています。
def ask_ChatGPT(self, message):
openai.organization = self.openai_organization
openai.api_key = self.openai_api_key
completion = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=[
{
"role": "user",
"content": message,
}
],
max_tokens=1024,
n=1,
stop=None,
temperature=0.5,
)
response = completion.choices[0].message.content
return response
しりとりの条件を満たしているかを判定する
最後の文字が「ん」でないかを判別
def check_last(self, word):
if word == "":
return False
result = self.juman.analysis(word)
mrph_list = result.mrph_list()
mrph = mrph_list[-1]
s=jaconv.kata2hira(mrph.yomi)
if s[-1] == "ん":
return False
return True
名詞かどうかを判別
def check_hinsi(self, word):
result = self.juman.analysis(word)
mrph_list = result.mrph_list()
if not mrph_list:
return False
mrph = mrph_list[-1]
if mrph.hinsi == "名詞":
return True
return False
小さいひらがなかカタカナ(「ぁ」や「ョ」など)であることを判別
def check_small(self,character):
small_list=["ぁ","ぃ","ぅ","ぇ","ぉ","ゃ","ゅ","ょ","ァ","ィ","ゥ","ェ","ォ","ャ","ュ","ョ"]
if character in small_list:
return True
return False
小さいひらがな、カタカナを、大きいひらがな、カタカナに直す
def change_small(self,s):
if s=="ぁ" or s=="ァ":
return "あ"
elif s=="ぃ" or s=="ィ":
return "い"
elif s=="ぅ" or s=="ゥ":
return "う"
elif s=="ぇ" or s=="ェ":
return "え"
elif s=="ぉ" or s=="ォ":
return "お"
elif s=="ゃ" or s=="ャ":
return "や"
elif s=="ゅ" or s=="ュ":
return "ゆ"
elif s=="ょ" or s=="ョ":
return "よ"
return s
単語の最後の文字を返します。次のターンではこの文字から始めないと負けになります。
def get_lastletter(self, current_word):
result = self.juman.analysis(current_word)
mrph_list = result.mrph_list()
mrph = mrph_list[-1]
if mrph.yomi[-1]=="ー":
return mrph.yomi[-2:-1]
return mrph.yomi[-1]
これらの関数は、以下の「最後の文字からスタートしているか」(りんご→コアラはNG)のチェックの際の補足として使っています。
具体的には、例えば
「シーシャ」だと「ャ」では始まれないので「や」からスタートする言葉を探しています。
また、「ヨーヨー」といった、最後に伸ばし棒がある単語はその一個前の「よ」からスタートするようにしています。
def check_initial(self, last_word, current_word):
if last_word == "":
return True
if current_word == "":
return False
result1 = self.juman.analysis(current_word)
mrph_list1 = result1.mrph_list()
mrph1 = mrph_list1[0]
result2 = self.juman.analysis(last_word)
mrph_list2 = result2.mrph_list()
mrph2 = mrph_list2[-1]
m1_new = jaconv.kata2hira(mrph1.yomi)
m2_new = jaconv.kata2hira(mrph2.yomi)
if m2_new[-1]=="ー":
last=m2_new[-2:-1]
if self.check_small(last):
last_new=self.change_small(last)
if m1_new[0] != last_new:
return False
return True
if m1_new[0] != last:
return False
else:
return True
if self.check_small(m2_new[-1])==False:
if m1_new[0] != m2_new[-1]:
return False
return True
if self.check_small(m2_new[-1]):
yomi_new=self.change_small(m2_new[-1])
if m1_new[0] != yomi_new:
return False
return True
上記の判定関数を組み合わせて、しりとりのルールをその単語が満たしているかどうかの結果を返します。満たしていなかったらその理由を出力するようにしています。
def win_or_lose(self, last_word, current_word):
if (
current_word not in self.word_list
and self.check_initial(last_word, current_word)
and self.check_last(current_word)
and self.check_hinsi(current_word)
):
return True
else:
if current_word in self.word_list:
print("既出です")
if self.check_initial(last_word, current_word) == False:
print("最後の文字から始まっていません")
if self.check_last(current_word) == False:
print("「ん」で終わっています")
if self.check_hinsi(current_word) == False:
print("名詞ではありません")
return False
あとは、ゲームを始めるだけです。
def play(self):
n = 0
current_word = ""
self.last_word = ""
last_letter = None
user = input("名前を教えてください: ")
while n < self.N:
word = input("単語を入力してください: ")
current_word = word
if self.win_or_lose(self.last_word, current_word) == False:
print(f"{user}の負けです")
exit()
self.word_list.add(current_word)
last_letter = self.get_lastletter(current_word)
if self.check_small(last_letter):
last_letter = self.change_small(last_letter)
self.last_word = current_word
current_word = ""
while self.win_or_lose_for_chatgpt(self.last_word, current_word) == False:
print(f"「{last_letter}」から始まる単語を探しています...")
command = f"{last_letter}から始まる名詞を一つ言ってください"
current_word = self.ask_ChatGPT(command)
print("ChatGPT:", current_word)
if self.win_or_lose(self.last_word, current_word) == False:
print("Chatgptの負けです")
exit()
self.last_word = current_word
self.word_list.add(current_word)
n += 1
最後に、コード全体を置いておきます。
from pyknp import Juman
import openai
import jaconv
class ShiritoriGame:
def __init__(self):
self.N = 100
self.word_list = set()
self.juman = Juman()
self.last_word = ""
self.openai_organization = "org-xxx"
self.openai_api_key = "sk-xxx"
def ask_ChatGPT(self, message):
openai.organization = self.openai_organization
openai.api_key = self.openai_api_key
completion = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=[
{
"role": "user",
"content": message,
}
],
max_tokens=1024,
n=1,
stop=None,
temperature=0.5,
)
response = completion.choices[0].message.content
return response
def check_last(self, word):
if word == "":
return False
result = self.juman.analysis(word)
mrph_list = result.mrph_list()
mrph = mrph_list[-1]
s=jaconv.kata2hira(mrph.yomi)
if s[-1] == "ん":
return False
return True
def check_hinsi(self, word):
result = self.juman.analysis(word)
mrph_list = result.mrph_list()
if not mrph_list:
return False
mrph = mrph_list[-1]
if mrph.hinsi == "名詞":
return True
return False
def check_small(self,character):
small_list=["ぁ","ぃ","ぅ","ぇ","ぉ","ゃ","ゅ","ょ","ァ","ィ","ゥ","ェ","ォ","ャ","ュ","ョ"]
if character in small_list:
return True
return False
def change_small(self,s):
if s=="ぁ" or s=="ァ":
return "あ"
elif s=="ぃ" or s=="ィ":
return "い"
elif s=="ぅ" or s=="ゥ":
return "う"
elif s=="ぇ" or s=="ェ":
return "え"
elif s=="ぉ" or s=="ォ":
return "お"
elif s=="ゃ" or s=="ャ":
return "や"
elif s=="ゅ" or s=="ュ":
return "ゆ"
elif s=="ょ" or s=="ョ":
return "よ"
return s
def check_initial(self, last_word, current_word):
if last_word == "":
return True
if current_word == "":
return False
result1 = self.juman.analysis(current_word)
mrph_list1 = result1.mrph_list()
mrph1 = mrph_list1[0]
result2 = self.juman.analysis(last_word)
mrph_list2 = result2.mrph_list()
mrph2 = mrph_list2[-1]
m1_new = jaconv.kata2hira(mrph1.yomi)
m2_new = jaconv.kata2hira(mrph2.yomi)
if m2_new[-1]=="ー":
last=m2_new[-2:-1]
if self.check_small(last):
last_new=self.change_small(last)
if m1_new[0] != last_new:
return False
return True
if m1_new[0] != last:
return False
else:
return True
if self.check_small(m2_new[-1])==False:
if m1_new[0] != m2_new[-1]:
return False
return True
if self.check_small(m2_new[-1]):
yomi_new=self.change_small(m2_new[-1])
if m1_new[0] != yomi_new:
return False
return True
def win_or_lose(self, last_word, current_word):
if (
current_word not in self.word_list
and self.check_initial(last_word, current_word)
and self.check_last(current_word)
and self.check_hinsi(current_word)
):
return True
else:
if current_word in self.word_list:
print("既出です")
if self.check_initial(last_word, current_word) == False:
print("最後の文字から始まっていません")
if self.check_last(current_word) == False:
print("「ん」で終わっています")
if self.check_hinsi(current_word) == False:
print("名詞ではありません")
return False
def win_or_lose_for_chatgpt(self, last_word, current_word):
if (
current_word not in self.word_list
and self.check_initial(last_word, current_word)
and self.check_last(current_word)
and self.check_hinsi(current_word)
):
return True
return False
def get_lastletter(self, current_word):
result = self.juman.analysis(current_word)
mrph_list = result.mrph_list()
mrph = mrph_list[-1]
if mrph.yomi[-1]=="ー":
return mrph.yomi[-2:-1]
return mrph.yomi[-1]
def play(self):
n = 0
current_word = ""
self.last_word = ""
last_letter = None
user = input("名前を教えてください: ")
while n < self.N:
word = input("単語を入力してください: ")
current_word = word
if self.win_or_lose(self.last_word, current_word) == False:
print(f"{user}の負けです")
exit()
self.word_list.add(current_word)
last_letter = self.get_lastletter(current_word)
if self.check_small(last_letter):
last_letter=self.change_small(last_letter)
self.last_word = current_word
current_word = ""
while self.win_or_lose_for_chatgpt(self.last_word, current_word) == False:
print(f"「{last_letter}」から始まる単語を探しています...")
command = f"{last_letter}から始まる名詞を一つ言ってください"
current_word = self.ask_ChatGPT(command)
print("ChatGPT:", current_word)
if self.win_or_lose(self.last_word, current_word) == False:
print("Chatgptの負けです")
exit()
self.last_word = current_word
self.word_list.add(current_word)
n += 1
game = ShiritoriGame()
game.play()
感想
細かい判定がめんどすぎた。
コードは改善点だらけだと思いますが読んでくださった方ありがとうございました。駄文失礼しました。