コードのエラー文を自動でChatGPTに読み込ませる

1. 何でそんなことするの？

エラーの原因がよくわからない時、エラー文をコピペしてChatGPTに投げるということを頻繁にやっているのですが、コピペがめんどくさい。

2. この記事でやること

Google Colaboratoryで実行したコードの「エラー文をChatGPTに投げる＋その結果をColab上で出力する」を全自動で行います。

3. 方法

3.1 必要なライブラリをインストール

!pip install openai
import openai
import os

# APIキーを設定します
openai.api_key = 'YOUR_API_KEY'

・「YOUR_API_KEY」：実際の自分のAPI Key
OpenAI API Keyの取得方法はコチラ（簡単!）↓

3.2 エラー文をChatGPTに送信するための関数(send_error_to_chatgpt)を定義

def send_error_to_chatgpt(error_message):
    response = openai.ChatCompletion.create(
      model="使用するモデル", 
      messages=[
        {"role": "system", "content": "You are a helpful assistant."},
        {"role": "user", "content": f"以下のエラーメッセージの原因と解決策を教えてください：\n\n{error_message}"}
      ],
      max_tokens=2048
    )
    return response.choices[0].message['content'].strip()

・「使用するモデル」：gpt-4o、gpt-3.5-turboなどの実際のモデル名
・「max_tokens」:出力する解答文のトークン数

3.3 任意のコードを実行

cell_code = """

コード

"""

execute_and_handle_error(cell_code)

・「コード」：実際のコード

3.3.1 実行例

ChatGPT-3.5 Turboを使って上記コードを実行したらこんな感じです（めっちゃ便利！）↓

4. おわりに

これで「cmd/ctrl + c/v」の回数が格段に減り、作業効率が上がりました。まだやったことない方はぜひお試しください。
他にもAPIを使ったtipsあればなんでも教えてください。