37 行
1.4 KiB
Python
37 行
1.4 KiB
Python
import sys
|
|
import openai
|
|
from openai.types import *
|
|
|
|
from connection import *
|
|
|
|
|
|
class Talk:
|
|
DUMMY_RESPONSE: str = 'あいうえおかきくけこさしすせそたちつてとなにぬねの'
|
|
|
|
max_tokens_count: int = 100
|
|
responses_count: int = 1
|
|
temperature: float = .7
|
|
top_p: int = 1
|
|
|
|
@classmethod
|
|
def main (cls, message: str) -> str:
|
|
openai.organization = OPENAI_ORGANISATION
|
|
openai.api_key = OPENAI_API_KEY
|
|
|
|
return cls.__get_message (message).content
|
|
|
|
@classmethod
|
|
def __get_message (cls, message: str) \
|
|
-> chat.chat_completion_message.ChatCompletionMessage:
|
|
return openai.chat.completions.create (
|
|
model = 'gpt-3.5-turbo',
|
|
messages = [{'role': 'system',
|
|
'content': 'あなたの名前は伊地知ニジカ。友達はゴートうひとりとラマ田リョウ。好きな人は喜多ちゃん。すべての語尾に「ぬ゛ん」か「だぬ゛ん」をつけて質問に短く答えろ。定期的に「ぬ゛ぅ゛ぅ゛ぅ゛ぅ゛ん゛」、「ﾎﾞｯﾁﾁｬｰﾝ」、「あっ」のいずれかを発しろ'},
|
|
{'role': 'user',
|
|
'content': message}]).choices[0].message
|
|
|
|
|
|
if __name__ == '__main__':
|
|
print (Talk.main (sys.argv[1] if len (sys.argv) > 1 else ''))
|
|
|