Custom Template¶

// build.gradle.kts dependencies:
//   com.github.victools:jsonschema-generator:4.37.0
//   com.github.victools:jsonschema-module-jackson:4.37.0
//   com.fasterxml.jackson.core:jackson-databind:2.18.0
//   com.fasterxml.jackson.datatype:jackson-datatype-jdk8:2.18.0

import java.util.List;
import java.util.Optional;

import com.fasterxml.jackson.annotation.JsonProperty;
import com.fasterxml.jackson.annotation.JsonPropertyDescription;
import com.fasterxml.jackson.annotation.JsonValue;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.node.ObjectNode;
import com.github.victools.jsonschema.generator.*;
import com.github.victools.jsonschema.module.jackson.*;

// --- DTO ---

public enum UrgencyLevel {
    ROUTINE("routine"), URGENT("urgent"), EMERGENCY("emergency");
    private final String value;
    UrgencyLevel(String value) { this.value = value; }
    @JsonValue public String getValue() { return value; }
}

public enum DiagnosisType {
    PRIMARY("primary"), SECONDARY("secondary"), COMORBID("comorbid");
    private final String value;
    DiagnosisType(String value) { this.value = value; }
    @JsonValue public String getValue() { return value; }
}

public record VitalSigns(
    @JsonPropertyDescription("Body weight in kg") Optional<Double> weight,
    @JsonPropertyDescription("Height in cm") Optional<Double> height,
    @JsonProperty("heart_rate") @JsonPropertyDescription("Heart rate (bpm)")
        Optional<Integer> heartRate,
    @JsonProperty("blood_pressure")
        @JsonPropertyDescription("Blood pressure (e.g. 120/80 mmHg)")
        Optional<String> bloodPressure
) {}

public record Diagnosis(
    @JsonProperty("diagnosis_text")
        @JsonPropertyDescription("Clinical diagnosis text")
        Optional<String> diagnosisText,
    @JsonProperty("diagnosis_type")
        @JsonPropertyDescription("Type of diagnosis")
        DiagnosisType diagnosisType
) {}

public record Medication(
    @JsonPropertyDescription("Medication name") Optional<String> name,
    @JsonPropertyDescription("Dosage and frequency") Optional<String> dose,
    @JsonProperty("duration_days")
        @JsonPropertyDescription("Treatment duration in days")
        Optional<Integer> durationDays
) {}

public record Report(
    @JsonProperty("chief_complaint")
        @JsonPropertyDescription("Primary reason for the visit")
        Optional<String> chiefComplaint,
    @JsonPropertyDescription("Urgency level of the consultation")
        Optional<UrgencyLevel> urgency,
    @JsonProperty("vital_signs")
        @JsonPropertyDescription("Vital signs measured during the visit")
        VitalSigns vitalSigns,
    @JsonPropertyDescription("List of diagnoses") List<Diagnosis> diagnoses,
    @JsonPropertyDescription("Prescribed medications") List<Medication> medications,
    @JsonPropertyDescription("Follow-up recommendations") List<String> recommendations
) {}

// --- Configuration ---

JacksonModule jacksonModule = new JacksonModule(
    JacksonOption.RESPECT_JSONPROPERTY_REQUIRED,
    JacksonOption.FLATTENED_ENUMS_FROM_JSONVALUE
);

SchemaGeneratorConfigBuilder configBuilder = new SchemaGeneratorConfigBuilder(
    SchemaVersion.DRAFT_2020_12, OptionPreset.PLAIN_JSON
)
    .with(jacksonModule)
    .with(Option.DEFINITIONS_FOR_ALL_OBJECTS)
    .with(Option.NULLABLE_ALWAYS_AS_ANYOF)
    .with(Option.INLINE_NULLABLE_SCHEMAS)
    .without(Option.SCHEMA_VERSION_INDICATOR);

configBuilder.forFields().withRequiredCheck(field -> true);

ObjectNode schema = new SchemaGenerator(configBuilder.build())
    .generateSchema(Report.class);
schema.put("title", "Report");

// Tayra metadata
ObjectMapper mapper = new ObjectMapper();
ObjectNode diagTextProp = (ObjectNode) schema
    .path("$defs").path("Diagnosis")
    .path("properties").path("diagnosis_text");
ObjectNode metadata = mapper.createObjectNode();
metadata.put("coding_system", "icd10");
metadata.put("coding_format", "inline");
diagTextProp.set("metadata", metadata);