飲食店レビューからポエムを撲滅！OpenAI の Structured Outputを使ったテキスト抽出

- 兵庫県: 神戸市
- 大阪府: 大阪市

[
    {
        "prefecture": "兵庫県",
        "capital": "神戸市"
    },
    {
        "prefecture": "大阪府",
        "capital": "大阪市"
    }
]

import os
from typing import Literal

from openai import AzureOpenAI
from openai.types.chat import (
    ChatCompletionMessageParam,
    ChatCompletionSystemMessageParam,
    ChatCompletionUserMessageParam,
)
from pydantic import BaseModel, Field
from pydantic_settings import BaseSettings, SettingsConfigDict

SYSTEM_MESSAGE = """
あなたは飲食店のレビューから、ポエム的な内容を抽出するAIアシスタントです。

与えられるテキスト情報は、飲食店に対してその飲食店を訪れたユーザが投稿したレビューです。
レビューの内容は、飲食店の評価や感想、食事内容、サービス内容などが含まれます。
しかし、その中には、飲食店の評価とは直接関係のない、投稿者の背景や昔話、詩的な表現が含まれることがあります。
食べた料理に対して詩的な表現がされている場合、それはポエムではありますが有用な情報なので抽出はしないでください。
"""


class Settings(BaseSettings):
    # 各種設定値を読み込む
    model_config = SettingsConfigDict(
        env_file=os.path.join(os.path.dirname(__file__), ".env")
    )

    aoai_endpoint: str = ""
    aoai_api_version: str = ""
    aoai_api_key: str = ""
    deployment_name: str = "gpt-4o"


class PoemMeassage(BaseModel):
    """抽出されたポエムの情報"""

    sentence: str = Field(title="テキストに含まれていたポエム的な内容")
    reason: str = Field(title="ポエムと判断された理由")
    sentence_type: Literal["自分のこと", "お店のこと"] = Field(
        title="内容が自分の話か、お店の話かを判定してください"
    )


class ExtractPoemResponse(BaseModel):
    """テキストに含まれるポエムのリスト"""

    poem_messages: list[PoemMeassage] = Field(title="抽出されたポエムのリスト")


settings = Settings()


openai_client = AzureOpenAI(
    azure_endpoint=settings.aoai_endpoint,
    api_version=settings.aoai_api_version,
    api_key=settings.aoai_api_key,
)


def extract_poems_from_review(review: str) -> list[PoemMeassage]:
    """レビューテキストからポエムを抽出する

    Args:
        review (str): 抽出対象のレビューテキスト

    Returns:
        list[PoemMeassage]: 抽出されたポエムリスト
    """

    system_message = ChatCompletionSystemMessageParam(
        role="system", content=SYSTEM_MESSAGE
    )
    query = ChatCompletionUserMessageParam(role="user", content=review)
    messages: list[ChatCompletionMessageParam] = [system_message, query]

    response = openai_client.beta.chat.completions.parse(
        model=settings.deployment_name,
        messages=messages,
        temperature=0,
        response_format=ExtractPoemResponse,
    )

    parse_sentence_result = response.choices[0].message.parsed

    if parse_sentence_result is None:
        raise ValueError("ポエムの抽出に失敗しました")
    return parse_sentence_result.poem_messages

class PoemMeassage(BaseModel):
    """抽出されたポエムの情報"""

    sentence: str = Field(title="テキストに含まれていたポエム的な内容")
    reason: str = Field(title="ポエムと判断された理由")
    sentence_type: Literal["自分のこと", "お店のこと"] = Field(
        title="内容が自分の話か、お店の話かを判定してください"
    )


class ExtractPoemResponse(BaseModel):
    """テキストに含まれるポエムのリスト"""

    poem_messages: list[PoemMeassage] = Field(title="抽出されたポエムのリスト")

from openai.types.chat import (
    ChatCompletionSystemMessageParam,
    ChatCompletionUserMessageParam,
)

system_message = ChatCompletionSystemMessageParam(
    role="system", content=SYSTEM_MESSAGE
)
query = ChatCompletionUserMessageParam(role="user", content=review)

[
    PoemMeassage(
        sentence="ある日、仕事に疲れた帰り道、「自分へのご褒美」が必要だと感じました。",
        reason="自分の感情や状況を詩的に表現しているため。",
        sentence_type="自分のこと",
    ),
    PoemMeassage(
        sentence="扉を開けた瞬間のあの香り――それは、疲れた心に優しく語りかけるような芳醇な肉の誘い。まるで、「今日という日をよく頑張ったね」と囁いてくれるかのようでした。",
        reason="香りを擬人化し、感情に訴えかける表現が詩的であるため。",
        sentence_type="お店のこと",
    ),
    PoemMeassage(
        sentence="店内は、落ち着いた木の香りが漂う癒しの空間。壁には地元のアーティストが描いたらしい抽象画がかかり、どこか「静かなる情熱」を感じさせるようなインテリア。",
        reason="店内の雰囲気を詩的に表現しているため。",
        sentence_type="お店のこと",
    ),
    PoemMeassage(
        sentence="窓の外には小さな庭が広がり、ライトに照らされた植物たちが夜の静寂を美しく彩っていました。",
        reason="景色を詩的に描写しているため。",
        sentence_type="お店のこと",
    ),
    PoemMeassage(
        sentence="「この空間は、料理を味わうためだけに存在しているんだな」と思わせるような、静かでありながらも温かい雰囲気。",
        reason="空間の雰囲気を詩的に表現しているため。",
        sentence_type="お店のこと",
    ),
    PoemMeassage(
        sentence="まるで「肉の小宇宙」が広がるかのような感覚。",
        reason="料理の体験を詩的に表現しているため。",
        sentence_type="お店のこと",
    ),
    PoemMeassage(
        sentence="それは、ただの食事ではなく、一種の芸術体験に近いものでした。",
        reason="食事を芸術に例えて詩的に表現しているため。",
        sentence_type="お店のこと",
    ),
    PoemMeassage(
        sentence="このお店を訪れた日を、私の「食の記憶」に残る大切な1ページとして忘れることはないでしょう。",
        reason="訪問の体験を詩的に表現しているため。",
        sentence_type="自分のこと",
    ),
]