ChatGPTで用いられるGPT-3.5系のモデルをAPIから利用してみた

pip install openai

import os, argparse
import openai

HISTORY_PATH='history.log'
openai.api_key = 'OPENAI_API_KEY'

parser = argparse.ArgumentParser(description='会話内容を入力してください')
parser.add_argument('input')
parser.add_argument('-r', '--reset', action='store_true',help='会話履歴(history.log)を初期化して実行する')
args = parser.parse_args()

history=''
# リセットフラグが立っていない&ファイルが存在する場合履歴を読み込む
if os.path.isfile(HISTORY_PATH) and not args.reset:
  with open(HISTORY_PATH, 'r') as f:
    history = f.read()

print(args.input)
# 会話として推論するため、会話ログを付け足してリクエストする
prompt = history + args.input

response = openai.Completion.create(
  model='text-davinci-003',
  prompt=prompt,
  temperature=0, # ランダム性の制御[0-1]
  max_tokens=1000, # 返ってくるレスポンストークンの最大数
  top_p=1.0, # 多様性の制御[0-1]
  frequency_penalty=0.0, # 周波数制御[0-2]：高いと同じ話題を繰り返さなくなる
  presence_penalty=0.0 # 新規トピック制御[0-2]：高いと新規のトピックが出現しやすくなる
)

texts = ''.join([choice['text'] for choice in response.choices])
print(texts)

# 会話履歴の保存
with open(HISTORY_PATH, 'w') as f:
  f.write(prompt)
  f.write(texts)
  # 文の切れ目に改行を追加
  f.write('\n\n')

python chat.py エラトステネスの篩について教えて。

エラトステネスの篩とは、素数を列挙するアルゴリズムのことです。

アルゴリズムの手順は以下の通りです。

1. 2から始めて、数列を作成します。

2. 2を素数としてマークします。

3. 2の倍数をすべてマークします。

4. 次に素数としてマークされていない最小の数を素数としてマークします。

5. その数の倍数をすべてマークします。

6. 4から5を繰り返します。

7. 繰り返しが終わったら、マークされていない数が素数です。

python chat.py pythonのプログラムに変換して。

def eratosthenes(n):
    prime_list = []
    for i in range(2, n+1):
        if all(i % j != 0 for j in range(2, i)):
            prime_list.append(i)
    return prime_list

print(eratosthenes(100))

python chat.py nをコマンドラインから渡せるようにして。

import sys

def eratosthenes(n):
    prime_list = []
    for i in range(2, n+1):
        if all(i % j != 0 for j in range(2, i)):
            prime_list.append(i)
    return prime_list

n = int(sys.argv[1])
print(eratosthenes(n))

python chat.py わかりやすいようにコメントを追加して。

import sys

# エラトステネスの篩を実装する関数
def eratosthenes(n):
    # 素数を格納するリスト
    prime_list = []
    # 2からnまでの数を調べる
    for i in range(2, n+1):
        # iが素数かどうかを判定する
        # 2からi-1までの数で割り切れなければ素数
        if all(i % j != 0 for j in range(2, i)):
            # 素数ならリストに追加
            prime_list.append(i)
    # 素数のリストを返す
    return prime_list

# コマンドライン引数からnを取得
n = int(sys.argv[1])
# 素数のリストを表示
print(eratosthenes(n))

python eratosthenes.py 100

[2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61, 67, 71, 73, 79, 83, 89, 97]