OpenAI Responses API 入門 (8) - Reasoningモデル

# パッケージのインストール
!pip install -U openai openai-agents

import os
from google.colab import userdata

# 環境変数の準備 (左端の鍵アイコンでOPENAI_API_KEYを設定)
os.environ["OPENAI_API_KEY"] = userdata.get("OPENAI_API_KEY")

from openai import OpenAI

# クライアントの準備
client = OpenAI()

# メッセージリストの準備
messages = [
    {
        "role": "user", 
        "content": "[1,2],[3,4],[5,6]' という形式の文字列として表される行列を受け取り、同じ形式で転置を出力するbashスクリプトを作成してください。"
    }    
]

# 推論の実行
response = client.responses.create(
    model="o3-mini",
    reasoning={"effort": "medium"},
    input=messages
)
print(response.output_text)

次のような bash スクリプト例をご参考ください。このスクリプトは、引数として
  "[1,2],[3,4],[5,6]"
のような文字列を受け取り、行列の転置（この場合は "[1,3,5],[2,4,6]"）を同じ形式で出力します。

内容は以下の通りです。

------------------------------------------------
#!/bin/bash
# 使い方:
#   ./transpose.sh "[1,2],[3,4],[5,6]"
#
# 入力は "[数字,数字],...[数字,数字]" という形式を期待しています。

if [ "$#" -eq 0 ]; then
    echo "Usage: $0 '[1,2],[3,4],[5,6]'"
    exit 1
fi

# 引数から入力行列の文字列を取得
input="$1"

# 余計な空白やシングル／ダブルクォートがあれば除去
matrix=$(echo "$input" | sed 's/^[[:space:]]*//; s/[[:space:]]*$//; s/^[\'"]*//; s/[\'"]*$//')

# 行の区切り "],[" を改行に変換し、先頭の "[" と末尾の "]" を除去して
# 行ごとに「1,2」などの形式にします。
rows=$(echo "$matrix" | sed 's/],\[/\n/g; s/^\[//; s/\]$//')

# awk を用いて転置を実施
# ・各行をコンマ区切りで分割し、配列 a[行番号, 列番号] に格納
# ・全体の行数と各行の列数（max 列数）を記録
# ・転置では元の列数を新しい行数、元の行数を新しい列数として出力
transposed=$(echo "$rows" | awk -F, '
{
    for (i = 1; i <= NF; i++) {
        a[NR, i] = $i
    }
    if (NF > max) {
        max = NF
    }
    rows = NR
}
END {
    output = ""
    for (i = 1; i <= max; i++) {
        line = "["
        for (j = 1; j <= rows; j++) {
            line = line a[j, i] (j < rows ? "," : "")
        }
        line = line "]"
        output = output (i < max ? line "," : line)
    }
    print output
}')

# 結果を表示
echo "$transposed"

------------------------------------------------

＜スクリプトの解説＞
1. sed によって入力文字列の前後の不要な空白やクォートを除去し、
   また "],[” を改行に置換して各行に「1,2」のような文字列が入るように整形します。
2. awk はフィールド区切り文字をコンマ (,) にして各行を分解し、
   2 次元配列 a[行,列] に値を格納します。
3. END ブロックで転置を行い、各新しい行（＝元の列）の数値を "[" と "]" で囲み、
   行同士はコンマで連結して最終的な文字列として出力します。

実行例:
$ ./transpose.sh "[1,2],[3,4],[5,6]"
[1,3,5],[2,4,6]

これで、入力された行列の転置を同じ形式で出力することができます。

# 使用状況の確認
print(response.usage)

ResponseUsage(
    input_tokens=115, 
    output_tokens=2359, 
    output_tokens_details=OutputTokensDetails(
        reasoning_tokens=1536
    ), 
    total_tokens=2474, 
    input_tokens_details={'cached_tokens': 0}
)

# 推論の実行
response = client.responses.create(
    model="o3-mini",
    reasoning={"effort": "medium"},
    input=messages,
    max_output_tokens=300,　# 出力トークン数の制限
)

if response.status == "incomplete" and response.incomplete_details.reason == "max_output_tokens":
    print("トークンがなくなった。")
    if response.output_text:
        print("部分出力:", response.output_text)
    else: 
        print("Reasoning中にトークンがなくなった。")