Google Colab で OpenAI API の Retrieval を試す

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

import os
os.environ["OPENAI_API_KEY"] = "<OpenAI_APIキー>"

from openai import OpenAI

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

# ファイルをOpenAIのサーバにアップロード
file = client.files.create(
    file=open("bocchi.txt", "rb"),
    purpose="assistants"
)

# アシスタントの作成
assistant = client.beta.assistants.create(
    instructions="あなたはQAチャットボットです。 ナレッジベースを使用して、ユーザーの質問に最大限に対応します。",
    model="gpt-4-1106-preview",
    tools=[{"type": "retrieval"}],
    file_ids=[file.id]
)

# スレッドの準備
thread = client.beta.threads.create()

# ユーザーメッセージの追加
message = client.beta.threads.messages.create(
    thread_id=thread.id,
    role="user",
    content="ぼっちちゃんの得意な楽器は？"
)

# スレッドのメッセージリストの確認
messages = client.beta.threads.messages.list(
    thread_id=thread.id,
    order="asc"
)
for message in messages:
    print(message.role, ":", message.content[0].text.value)

user : ぼっちちゃんの得意な楽器は？

# アシスタントにリクエスト
run = client.beta.threads.runs.create(
    thread_id=thread.id,
    assistant_id=assistant.id,
)

# 実行状況の確認
run = client.beta.threads.runs.retrieve(
    thread_id=thread.id,
    run_id=run.id
)
print(run.status)

completed

# スレッドのメッセージリストの確認
messages = client.beta.threads.messages.list(
    thread_id=thread.id,
    order="asc"
)
for message in messages:
    print(message.role, ":", message.content[0].text.value)

user : ぼっちちゃんの得意な楽器は？
assistant : ぼっちちゃん、正式には後藤ひとり、の得意な楽器はギターです。彼はギターヒーローとしてインターネット上でそこそこの人気を集めており、その腕前はかなりのものとなっています。

user : ぼっちちゃんの得意な楽器は？
assistant : ぼっちちゃん、正式には後藤ひとり、の得意な楽器はギターです。彼はギターヒーローとしてインターネット上でそこそこの人気を集めており、その腕前はかなりのものとなっています。
user : ライブハウス「STARRY」の店長は？
assistant : ライブハウス「STARRY」の店長に関する情報は見つかりませんでした。文書の中をさらに検索してみる必要があります。もう少しお待ちください。
assistant : ライブハウス「STARRY」の店長は伊地知星歌（いじちせいか）です。彼女は30歳になったばかりで、ぱっと見は真面目な清楚系に見えるが、かなりいい加減な性格の持ち主です。また、伊地知星歌は伊地知虹夏の姉であり、かわいいもの好きで乙女な一面も持っています。過去にはバンドでギターを担当しており、「御茶ノ水の魔王」と呼ばれていた経験もあるようです。