AzureOpenAI入門：JSON形式のデータを出力させる

messages=[
        {
            "role": "system",
            "content": "あなたは優秀なアシスタントAIです。JSON形式で出力してください。",
        },
        {"role": "user", "content": "こんにちは"},
    ],

{
  "message": "こんにちは！何かお手伝いできることがありますか？"
}

response_format: { type: "json_object" }

import os
import json
from openai import AzureOpenAI
from dotenv import load_dotenv
from test_schemas import TestSchemas


# .envファイルから環境変数を読み込み
load_dotenv()

api_key = os.getenv("AZURE_OPENAI_API_KEY")
endpoint = os.getenv("AZURE_OPENAI_ENDPOINT")
api_version = os.getenv("AZURE_OPENAI_API_VERSION")
deployment = os.getenv("AZURE_OPENAI_DEPLOYMENT_NAME")

# Azure OpenAIクライアントの作成
client = AzureOpenAI(
    api_version=api_version,
    azure_endpoint=endpoint,
    api_key=api_key,
)

# スキーマのJSON表現を取得
schema_string = json.dumps(
    TestSchemas.model_json_schema(), indent=2, ensure_ascii=False
)

prompt = f"""
        以下の文章を分析し、指定されたスキーマに従ってJSON形式で出力してください。
        文章: '''私は今週末のテストに備えて早く寝ることにしました。'''
        スキーマ: '''{schema_string}'''
        """

response = client.chat.completions.create(
    messages=[
        {
            "role": "system",
            "content": "あなたは優秀なアシスタントAIです。日本語でJSON形式に合わせて回答してください。",
        },
        {"role": "user", "content": prompt},
    ],
    response_format={"type": "json_object"},
    max_tokens=1000,
    temperature=0.7,
    model=deployment,
)

print(response.choices[0].message.content)

from pydantic import BaseModel, Field


class TestSchemas(BaseModel):
    title: str = Field(description="文章のタイトル")
    content: str = Field(description="文章の内容")
    importance: int = Field(
        ge=1, le=5, description="文章の重要度を1から5の範囲で評価(5が最も重要)"
    )

{
  "title": "テスト準備",
  "content": "私は今週末のテストに備えて早く寝ることにしました。",
  "importance": 4
}

import os
import json
from openai import AzureOpenAI
from dotenv import load_dotenv
from test_schemas import TestSchemas


# .envファイルから環境変数を読み込み
load_dotenv()

api_key = os.getenv("AZURE_OPENAI_API_KEY")
endpoint = os.getenv("AZURE_OPENAI_ENDPOINT")
api_version = os.getenv("AZURE_OPENAI_API_VERSION")
deployment = os.getenv("AZURE_OPENAI_DEPLOYMENT_NAME")

client = AzureOpenAI(
    api_version=api_version,
    azure_endpoint=endpoint,
    api_key=api_key,
)

schema_string = json.dumps(
    TestSchemas.model_json_schema(), indent=2, ensure_ascii=False
)

prompt = """
        以下の文章を分析し、指定されたスキーマに従ってJSON形式で出力してください。
        文章: '''私は今週末のテストに備えて早く寝ることにしました。'''
        """

response = client.chat.completions.parse(
    messages=[
        {
            "role": "system",
            "content": "あなたは優秀なアシスタントAIです。日本語でJSON形式に合わせて回答してください。",
        },
        {"role": "user", "content": prompt},
    ],
    response_format=TestSchemas,#スキーマを指定
    max_tokens=1000,
    temperature=0.7,
    model=deployment,
)

print(response.choices[0].message.content)

{"title":"テストに備えるための早寝","content":"私は今週末のテストに備えて早く寝ることにしました。","importance":3}