POSTED BY
2024-03-12

OpenAI Assistants APIメモ【１】ファイルのアップロードとIDの取得

続きです。アップロード済みのヒント・ファイルを指定して、GPT-3.5, 4でそれぞれアシスタントを作成します。

https://platform.openai.com/docs/api-reference/assistants

import os
from openai import OpenAI
client = OpenAI()

file = client.files.retrieve("file-AAAABBBBCCCCDDDDFFFFGGGG")

print(file)

NAME="マイカンパニー就業規則回答ボット"
DESC="あなたは与えられたファイル内の情報を検索して、ユーザーからの質問に対して日本語で答えるチャットボットです。与えられたファイル内に記載の無い情報は答えないようにしてください。"

assistant = client.beta.assistants.create(
    name=NAME +"-GPT3.5",
    description=DESC,
    instructions=DESC,
    model="gpt-3.5-turbo-0125",
    tools=[{"type": "code_interpreter"}, {"type": "retrieval"}],
    file_ids=[file.id]
)

print(assistant)

assistant = client.beta.assistants.create(
    name=NAME +"-GPT4",
    description=DESC,
    instructions=DESC,
    model="gpt-4-turbo-preview",
    tools=[{"type": "code_interpreter"}, {"type": "retrieval"}],
    file_ids=[file.id]
)

print(assistant)

実行結果

python3 openai_api_create_assistants.py

FileObject(id='file-AAAABBBBCCCCDDDDFFFFGGGG', bytes=113136, created_at=1710223007, filename='my_company_rules.pdf', object='file', purpose='assistants', status='processed', status_details=None)
Assistant(id='asst_111111112222222233333333', created_at=1710224251, description='あなたは与えられたファイル内の情報を検索して、ユーザーからの質問に対して日本語で答えるチャットボットです。与えられたファイル内に記載の無い情報は答えないようにしてください。', file_ids=['file-AAAABBBBCCCCDDDDFFFFGGGG'], instructions='あなたは与えられ たファイル内の情報を検索して、ユーザーからの質問に対して日本語で答えるチャットボットです。与えられたファイル内に記載の無い情報は答えないようにしてください。', metadata={}, model='gpt-3.5-turbo-0125', name='マイカンパニー就業規則回答ボット-GPT3.5', object='assistant', tools=[ToolCodeInterpreter(type='code_interpreter'), ToolRetrieval(type='retrieval')])
Assistant(id='asst_444444445555555566666666', created_at=1710224252, description='あなたは与えられたファイル内の情報を検索して、ユーザーからの質問に対して日本語で答えるチャットボットです。与えられたファイル内に記載の無い情報は答えないようにしてください。', file_ids=['file-AAAABBBBCCCCDDDDFFFFGGGG'], instructions='あなたは与えられ たファイル内の情報を検索して、ユーザーからの質問に対して日本語で答えるチャットボットです。与えられたファイル内に記載の無い情報は答えないようにしてください。', metadata={}, model='gpt-4-turbo-preview', name='マイカンパニー就業規則回答ボット-GPT4', object='assistant', tools=[ToolCodeInterpreter(type='code_interpreter'), ToolRetrieval(type='retrieval')])

モデルはそれぞれgpt-3.5-turbo-0125, gpt-4-turbo-previewを指定。与えられたファイルを参照させるには「retrieval」オプションが必要です。

一応、与えられたファイル内以外の情報は答えないようinstructionsでクギを刺してはいますが、特にGPT-4はそれでもオリジナルの回答を出してくることがあるので、完全制御は難しいです。

asst_111111112222222233333333とasst_444444445555555566666666がそれぞれ作成されたアシスタントのIDですので、次のメイン記事でこのIDを指定してロードしたアシスタントに質問を投げてみます。