[D] limiting LLM output to certain words

Suppose I want to do a multi-class classification on text. One approach is to prompt engineer, however, this can output labels different to what I want. Here is an example:

Extract the following labels from the text. Labels: Apples, Oranges.
Text: I ate an apple and then a few oranges.
Answer: Apples, Oranges

The answer shown above being simply the expected answer. If we were to use prompts, some other possibilities would be \[Apple, Orange\], \[Oranges, Apples\] etc.

In my …

apple class classification engineer example extract however labels llm machinelearning prompt prompt engineer text words