Spring/spring-ai
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

feat(mistral-schema): allow custom structured output with json schema (#2211)

Browse Source 
Signed-off-by: krsamuel <samuel.t.kerr@gmail.com>
This commit is contained in:
samuel-taleez
2025-02-11 10:56:38 +01:00
committed by Christian Tzolov
parent 5127662f2c
commit fbd03b7f55
1 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
models/spring-ai-mistral-ai/src/main/java/org/springframework/ai/mistralai/api/MistralAiApi.java
View File

@@ -627,9 +627,11 @@ public class MistralAiApi {
* @param stop A list of tokens that the model should stop generating after. If set,
* @param randomSeed The seed to use for random sampling. If set, different calls will
* generate deterministic results.
* @param responseFormat An object 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.
* @param responseFormat An object specifying the format or schema that the model must
* output. Setting to { "type": "json_object" } enables JSON mode, which guarantees
* the message the model generates is valid JSON. Setting to { "type": "json_object" ,
* "json_schema": schema} allows you to ensure the model provides an answer in a very
* specific JSON format by supplying a clear JSON schema.
*/
@JsonInclude(Include.NON_NULL)
public record ChatCompletionRequest(
@@ -718,12 +720,9 @@ public class MistralAiApi {
public enum ToolChoice {
// @formatter:off
@JsonProperty("auto")
AUTO,
@JsonProperty("any")
ANY,
@JsonProperty("none")
NONE
@JsonProperty("auto") AUTO,
@JsonProperty("any") ANY,
@JsonProperty("none") NONE
// @formatter:on
}
@@ -732,10 +731,11 @@ public class MistralAiApi {
* An object specifying the format that the model must output.
*
* @param type Must be one of 'text' or 'json_object'.
* @param jsonSchema A specific JSON schema to match, if 'type' is 'json_object'.
*/
@JsonInclude(Include.NON_NULL)
public record ResponseFormat(@JsonProperty("type") String type) {
public record ResponseFormat(@JsonProperty("type") String type,
@JsonProperty("json_schema") Map<String, Object> jsonSchema) {
}
}