sqlgen nb

sqlgen-nb

Écrivez une requête PostgreSQL à partir d'un langage naturel à l'aide de l'IA de génération sur site - avec IBM Granite Code LLM, Ollama sur Juypter Notebook.

Inspiré d'un cas d'utilisation mentionné par Anthropic, mais il s'agit d'une implémentation locale avec une configuration minimale.

Installation

Carnet Jupyter https://marketplace.visualstudio.com/items?itemName=ms-toolsai.jupyter

LangChain https://python.langchain.com/docs/how_to/installation/

Ollama https://ollama.com/download

Extrayez le modèle IBM Granite Code dans le terminal :

ollama pull granite-code:8b-instruct 

Usage

1. Ouvrez sqlgen.ipynb
2. Entrez le schéma de votre base de données dans schema =
3. Entrez votre question dans user_prompt =
4. Cliquez sur Run all pour exécuter toutes les cellules du bloc-notes Juypter
5. Faites défiler vers le bas pour la requête SQL générée

Défis et limites

La génération AI n’est pas déterministe. Avec la même invite utilisateur, il génère à chaque fois des complétions légèrement différentes. Vérifiez avant utilisation.

Quelle est la prochaine étape ?

Comme cela implique une interaction LLM avec la base de données. Les codes pourraient être réutilisés dans l'agent de base de données dans un workflow multi-agent.