nizika_broadcast/talk.py

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 ''))