This version is still in development and is not considered stable yet. For the latest snapshot version, please use Spring AI 1.0.0-SNAPSHOT! |
Structured Output Converter
As of 02.05.2024 the old OutputParser , BeanOutputParser , ListOutputParser and MapOutputParser classes are deprecated in favor of the new StructuredOutputConverter , BeanOutputConverter , ListOutputConverter and MapOutputConverter implementations.
Later are drop-in replacements for the former ones and provide the same functionality. The reason for the change was primarily naming, as there isn’t any parsing being done, but also have aligned with the Spring org.springframework.core.convert.converter package brining in some improved functionality.
|
The ability of LLMs to produce structured outputs is important for downstream applications that rely on reliably parsing output values. Developers want to quickly turn results from an AI model into data types, such as JSON, XML or Java Classes, that can be passed to other application functions and methods.
The Spring AI Structured Output Converters
help to convert the LLM output into a structured format.
As shown in the following diagram, this approach operates around the LLM text completion endpoint:
![Structured Output Converter Architecture](../_images/structured-output-architecture.jpg)
Generating structured outputs from Large Language Models (LLMs) using generic completion APIs requires careful handling of inputs and outputs. The structured output converter plays a crucial role before and after the LLM call, ensuring the desired output structure is achieved.
Before the LLM call, the converter appends format instructions to the prompt, providing explicit guidance to the models on generating the desired output structure. These instructions act as a blueprint, shaping the model’s response to conform to the specified format.
After the LLM call, the converter takes the model’s output text and transforms it into instances of the structured type. This conversion process involves parsing the raw text output and mapping it to the corresponding structured data representation, such as JSON, XML, or domain-specific data structures.
The StructuredOutputConverter is a best effort to convert the model output into a structured output.
The AI Model is not guaranteed to return the structured output as requested.
The model may not understand the prompt or be unable to generate the structured output as requested.
Consider implementing a validation mechanism to ensure the model output is as expected.
|
The StructuredOutputConverter is not used for LLM Function Calling, as this feature inherently provides structured outputs by default.
|
Structured Output API
The StructuredOutputConverter
interface allows you to obtain structured output, such as mapping the output to a Java class or an array of values from the text-based AI Model output.
The interface definition is:
public interface StructuredOutputConverter<T> extends Converter<String, T>, FormatProvider {
}
It combines the Spring Converter<String, T> interface and the FormatProvider
interface
public interface FormatProvider {
String getFormat();
}
The following diagram shows the data flow when using the structured output API.
![Structured Output API](../_images/structured-output-api.jpg)
The FormatProvider
supplies specific formatting guidelines to the AI Model, enabling it to produce text outputs that can be converted into the designated target type T
using the Converter
. Here is an example of such formatting instructions:
Your response should be in JSON format. The data structure for the JSON should match this Java class: java.util.HashMap Do not include any explanations, only provide a RFC8259 compliant JSON response following this format without deviation.
The format instructions are most often appended to the end of the user input using the PromptTemplate like this:
StructuredOutputConverter outputConverter = ...
String userInputTemplate = """
... user text input ....
{format}
"""; // user input with a "format" placeholder.
Prompt prompt = new Prompt(
new PromptTemplate(
userInputTemplate,
Map.of(..., "format", outputConverter.getFormat()) // replace the "format" placeholder with the converter's format.
).createMessage());
The Converter<String, T> is responsible to transform output text from the model into instances of the specified type T
.
Available Converters
Currently, Spring AI provides AbstractConversionServiceOutputConverter
, AbstractMessageOutputConverter
, BeanOutputConverter
, MapOutputConverter
and ListOutputConverter
implementations:
![Structured Output Class Hierarchy](../_images/structured-output-hierarchy4.jpg)
-
AbstractConversionServiceOutputConverter<T>
- Offers a pre-configured GenericConversionService for transforming LLM output into the desired format. No defaultFormatProvider
implementation is provided. -
AbstractMessageOutputConverter<T>
- Supplies a pre-configured MessageConverter for converting LLM output into the desired format. No defaultFormatProvider
implementation is provided. -
BeanOutputConverter<T>
- Configured with a designated Java class (e.g., Bean) or a ParameterizedTypeReference, this converter employs aFormatProvider
implementation that directs the AI Model to produce a JSON response compliant with aDRAFT_2020_12
,JSON Schema
derived from the specified Java class. Subsequently, it utilizes anObjectMapper
to deserialize the JSON output into a Java object instance of the target class. -
MapOutputConverter
- Extends the functionality ofAbstractMessageOutputConverter
with aFormatProvider
implementation that guides the AI Model to generate an RFC8259 compliant JSON response. Additionally, it incorporates a converter implementation that utilizes the providedMessageConverter
to translate the JSON payload into ajava.util.Map<String, Object>
instance. -
ListOutputConverter
- Extends theAbstractConversionServiceOutputConverter
and includes aFormatProvider
implementation tailored for comma-delimited list output. The converter implementation employs the providedConversionService
to transform the model text output into ajava.util.List
.
Using Converters
The following sections provide guides how to use the available converters to generate structured outputs.
Bean Output Converter
The following example shows how to use BeanOutputConverter
to generate the filmography for an actor.
The target record representing actor’s filmography:
record ActorsFilms(String actor, List<String> movies) {
}
Here is how to apply the BeanOutputConverter:
BeanOutputConverter<ActorsFilms> beanOutputConverter =
new BeanOutputConverter<>(ActorsFilms.class);
String format = beanOutputConverter.getFormat();
String actor = "Tom Hanks";
String template = """
Generate the filmography of 5 movies for {actor}.
{format}
""";
Generation generation = chatModel.call(
new Prompt(new PromptTemplate(template, Map.of("actor", actor, "format", format)).createMessage())).getResult();
ActorsFilms actorsFilms = beanOutputConverter.convert(generation.getOutput().getContent());
Generic Bean Types
Use the ParameterizedTypeReference
constructor to specify a more complex target class structure.
For example, to represent a list of actors and their filmographies:
BeanOutputConverter<List<ActorsFilmsRecord>> outputConverter = new BeanOutputConverter<>(
new ParameterizedTypeReference<List<ActorsFilmsRecord>>() { });
String format = outputConverter.getFormat();
String template = """
Generate the filmography of 5 movies for Tom Hanks and Bill Murray.
{format}
""";
Prompt prompt = new Prompt(new PromptTemplate(template, Map.of("format", format)).createMessage());
Generation generation = chatModel.call(prompt).getResult();
List<ActorsFilmsRecord> actorsFilms = outputConverter.convert(generation.getOutput().getContent());
Map Output Converter
Following sniped shows how to use MapOutputConverter
to generate a list of numbers.
MapOutputConverter mapOutputConverter = new MapOutputConverter();
String format = mapOutputConverter.getFormat();
String template = """
Provide me a List of {subject}
{format}
""";
PromptTemplate promptTemplate = new PromptTemplate(template,
Map.of("subject", "an array of numbers from 1 to 9 under they key name 'numbers'", "format", format));
Prompt prompt = new Prompt(promptTemplate.createMessage());
Generation generation = chatModel.call(prompt).getResult();
Map<String, Object> result = mapOutputConverter.convert(generation.getOutput().getContent());
List Output Converter
Following snippet shows how to use ListOutputConverter
to generate a list of ice cream flavors.
ListOutputConverter listOutputConverter = new ListOutputConverter(new DefaultConversionService());
String format = listOutputConverter.getFormat();
String template = """
List five {subject}
{format}
""";
PromptTemplate promptTemplate = new PromptTemplate(template,
Map.of("subject", "ice cream flavors", "format", format));
Prompt prompt = new Prompt(promptTemplate.createMessage());
Generation generation = this.chatModel.call(prompt).getResult();
List<String> list = listOutputConverter.convert(generation.getOutput().getContent());
Supported AI Models
The following AI Models have been tested to support List, Map and Bean structured outputs.
Model |
Integration Tests / Samples |
Build-in JSON mode
Some AI Models provide dedicated configuration options to generate structured (usually JSON) output.
-
OpenAI - provides a
spring.ai.openai.chat.options.responseFormat
options specifying the format that the model must output. Setting to{ "type": "json_object" }
enables JSON mode, which guarantees the message the model generates is valid JSON. -
Azure OpenAI - provides a
spring.ai.azure.openai.chat.options.responseFormat
options specifying the format that the model must output. Setting to{ "type": "json_object" }
enables JSON mode, which guarantees the message the model generates is valid JSON. -
Ollama - provides a
spring.ai.ollama.chat.options.format
option to specify the format to return a response in. Currently the only accepted value isjson
. -
Mistral AI - provides a
spring.ai.mistralai.chat.options.responseFormat
option to specify the format to return a response in. Setting to{ "type": "json_object" }
enables JSON mode, which guarantees the message the model generates is valid JSON.