Prompt Shield con Spring AI

Qué hace

Prompt Shield ofusca datos sensibles (DNI, email, teléfonos, IBAN) y desde la versión 1.1.1 también claves de API de 16 servicios populares (OpenAI, Anthropic, AWS, GitHub, Stripe...) antes de enviarlos a la IA, y restaura los valores originales en la respuesta.

Configuración

application.yml

spring:
  ai:
    openai:
      api-key: ${OPENAI_API_KEY}
      chat:
        options:
          model: gpt-4o-mini

prompt-shield:
  service-keys-enabled: true
  advisor:
    enabled: true
    inject-system-prompt: true
    restore-on-response: true

Código

@RestController
@RequestMapping("/api/ai")
public class AiController {

    private final ChatClient chatClient;

    public AiController(ChatClient.Builder builder, PromptShieldAdvisor advisor) {
        this.chatClient = builder
                .defaultAdvisors(advisor)
                .build();
    }

    @PostMapping("/chat")
    public AiResponse chat(@RequestBody AiRequest request) {
        String response = chatClient.prompt()
                .user(request.message())
                .call()
                .content();
        return new AiResponse(response);
    }
}