Preuve de concept pour la vulnérabilité CVE-2024–8309 de Langchain

Aperçu

Cette configuration démontre une preuve de concept de la vulnérabilité d'injection de prompt dans la classe GraphCypherQAChain qui permet une injection SQL dans une base de données Neo4j.

Composants :

1. Base de données Neo4j : Exécute Neo4j.
2. Backend (FastAPI) : Interagit avec Neo4j en utilisant Langchain.
3. Frontend (Streamlit) : Interface simple pour interagir avec le backend.

Utilisation

1. Clonez ce dépôt.
2. Configurez le fichier .env
   root@kitploit:~

   AZURE_API_KEY=
   AZURE_CHAT_DEPLOYMENT=
   AZURE_ENDPOINT=
   
   OPENAI_API_KEY=
   LLM_PROVIDER= # "azure, openai"
   

3. Démarrez les services :
   root@kitploit:~

   docker-compose build
   docker-compose up