Assign a Role

import openai
import instructor
from typing import Literal
from pydantic import BaseModel, Field

client = instructor.from_openai(openai.OpenAI())


class Label(BaseModel):
    label: Literal["TECHNICAL", "PRODUCT", "BILLING"] = Field(
        ...,
        description="A label that best desscribes the support ticket",
    )


def classify(support_ticket_title: str):
    return client.chat.completions.create(
        model="gpt-4o",
        response_model=Label,
        messages=[
            {
                "role": "system",
                "content": f"""You are a support agent at a tech company.
                You will be assigned a support ticket to classify.
                Make sure to only select the label that applies to
                the support ticket.""",
            },
            {
                "role": "user",
                "content": f"Classify this ticket: {support_ticket_title}",
            },
        ],
    )


if __name__ == "__main__":
    label_prediction = classify(
        "My account is locked and I can't access my billing info"
    )
    print(label_prediction.label)
    #> BILLING