【Assistants API】の実装方法について解説

!pip install openai==1.8.0

import os
import time

from openai import OpenAI

os.environ['OPENAI_API_KEY'] = 'OpenAIのAPIキー'
client = OpenAI()

# ファイルアップロード
file = client.files.create(
  file=open("/content/kotowaza.txt", "rb"),
  purpose='assistants'
)

# Assistant作成
assistant = client.beta.assistants.create(
  name="ことわざ大魔神",
  instructions="あなたは日本のことわざを知りつくしている人です。Knowledgeのファイルを使用して質問に答えます。",
  model="gpt-4-1106-preview",
  tools=[{"type": "retrieval"}],
  file_ids=[file.id]
)

# Thread作成
thread = client.beta.threads.create()

# Thread内にMessageを作成し追加
message = client.beta.threads.messages.create(
  thread_id=thread.id,
  role="user",
  content="「お」から始まることわざを教えて下さい"
)

## Thread作成と同時にMessageを作成して追加することもできます
# thread = client.beta.threads.create(
#   messages=[
#     {
#       "role": "user",
#       "content": "「お」から始まることわざを教えて下さい"
#     }
#   ]
# )

# スレッドを指定し実行
run = client.beta.threads.runs.create(
  thread_id=thread.id,
  assistant_id=assistant.id,
)

# 実行が完了するまで待つ
while True:
    # 実行状態の取得
    run = client.beta.threads.runs.retrieve(thread_id=thread.id, run_id=run.id)
    print("run.status:", run.status)

    # completedになればループを抜ける
    if run.status == 'completed':
        break

    time.sleep(5)

# 結果受け取り
messages = client.beta.threads.messages.list(thread.id)
print(messages)

SyncCursorPage[ThreadMessage](
    data=[
        ThreadMessage(
            id='msg_1JMGdLoDYpCOjpTtq8SrjA1o',
            assistant_id='asst_XOiL8dw2kd7Gc6dvkDHyq3eG',
            content=[
                MessageContentText(
                    text=Text(
                        annotations=[
                            TextAnnotationFileCitation(
                                end_index=272,
                                file_citation=TextAnnotationFileCitationFileCitation(
                                    file_id='file-oz8fUl9g19WR3EcnLIj5zAYy',
                                    quote='老いては子に従え\r\n負うた子に教えられて浅瀬を渡る\r\n鸚鵡よく言えども飛鳥を離れず\r\n大男総身に知恵が回りかね\r\n岡目八目\r\n落ち武者は薄の穂にも怖ず\r\n同じ穴の狢\r\n鬼に金棒\r\n鬼の居ぬ間の洗濯\r\n鬼の目にも涙\r\n帯に短く襷に長し\r\n溺れる者は藁をも掴む\r\n思い立ったが吉日\r\n親の意見と茄子の花は千に一つも無駄はない\r\n親の意見と冷や酒は後で効く\r\n親は無くとも子は育つ\r\n女は下げて育てよ\r\n女は乱の基\r\n陰陽師身の上知らず'
                                ),
                                start_index=262,
                                text='【9†source】',
                                type='file_citation'
                            )
                        ],
                        value='「お」から始まる日本のことわざには以下があります：\n\n- 老いては子に従え\n- 負うた子に教えられて浅瀬を渡る\n- 鸚鵡よく言えども飛鳥を離れず\n- 大男総身に知恵が回りかね\n- 岡目八目\n- 落ち武者は薄の穂にも怖ず\n- 同じ穴の狢\n- 鬼に金棒\n- 鬼の居ぬ間の洗濯\n- 鬼の目にも涙\n- 帯に短く襷に長し\n- 溺れる者は藁をも掴む\n- 思い立ったが吉日\n- 親の意見と茄子の花は千に一つも無駄はない\n- 親の意見と冷や酒は後で効く\n- 親は無くとも子は育つ\n- 女は下げて育てよ\n- 女は乱の基\n- 陰陽師身の上知らず【9†source】。'
                    ),
                    type='text'
                )
            ],
            created_at=1705655777,
            file_ids=[],
            metadata={},
            object='thread.message',
            role='assistant',
            run_id='run_0l6C7bFI6AN94WnAZCR0j18l',
            thread_id='thread_ASNMsHow7uoRpKfNK16QXLCN'
        ),
        ThreadMessage(
            id='msg_ZHC5Z2xfcOkuqHVMmKEBpfHZ',
            assistant_id=None,
            content=[
                MessageContentText(
                    text=Text(
                        annotations=[],
                        value='「お」から始まることわざを教えて下さい'
                    ),
                    type='text'
                )
            ],
            created_at=1705655766,
            file_ids=[],
            metadata={},
            object='thread.message',
            role='user',
            run_id=None,
            thread_id='thread_ASNMsHow7uoRpKfNK16QXLCN'
        )
    ],
    object='list',
    first_id='msg_1JMGdLoDYpCOjpTtq8SrjA1o',
    last_id='msg_ZHC5Z2xfcOkuqHVMmKEBpfHZ',
    has_more=False
)

client.beta.threads.delete(thread.id)  # Thread削除
client.beta.assistants.delete(assistant.id)  # Assistant削除
client.files.delete(file.id)  # アップロードしたファイル削除