The Vanna library uses a prompt function to present the user with visualized results, it is possible to alter the prompt using prompt injection and run arbitrary Python code instead of the intended visualization code. Specifically - allowing external input to the library’s “ask” method with "visualize" set to True (default behavior) leads to remote code execution.

References

• https://nvd.nist.gov/vuln/detail/CVE-2024-5565
• https://research.jfrog.com/vulnerabilities/vanna-prompt-injection-rce-jfsa-2024-001034449