このコミットが含まれているのは:
2023-11-30 22:32:13 +09:00
コミット 8af98eb452
2個のファイルの変更21行の追加2行の削除
+1
ファイルの表示
@@ -1,2 +1,3 @@
/connection.py /connection.py
/__pycache__
+20 -2
ファイルの表示
@@ -5,12 +5,30 @@ import openai
class Talk: class Talk:
max_tokens_count: int = 100
responses_count: int = 5
temperature: float = .7
top_p: int = 1
@classmethod @classmethod
def main (message: str) -> None: def main (cls, message: str) -> None:
openai.organization = OPENAI_ORGANISATION openai.organization = OPENAI_ORGANISATION
openai.api_key = OPENAI_API_KEY openai.api_key = OPENAI_API_KEY
print (cls.__get_answers (message)[0])
@classmethod
def __get_answers (cls, message: str) -> list[str]:
return openai.Completion.create (
model = 'text-davinci-002',
prompt = message,
max_tokens = cls.max_tokens_count,
n = cls.temperature,
stop = None,
temperature = cls.temperature,
top_p = cls.top_p).choices
if __name__ == '__main__': if __name__ == '__main__':
Talk.main (sys.argv[1]) Talk.main (sys.argv[1] if len (sys.argv) > 1 else '')