sqlgen nb

sqlgen-nb

Escriba consultas PostgreSQL desde lenguaje natural utilizando IA de generación local, con IBM Granite Code LLM, Ollama en Juypter Notebook.

Inspirado en un caso de uso mencionado por Anthropic, pero es una implementación local con una configuración mínima.

Instalación

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

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

Ollama https://ollama.com/download

Extraiga el modelo IBM Granite Code en la Terminal:

ollama pull granite-code:8b-instruct 

Uso

1. Abra sqlgen.ipynb
2. Ingrese el esquema de su base de datos en schema =
3. Ingrese su pregunta en user_prompt =
4. Haga clic en Run all para ejecutar todas las celdas en Juypter Notebook.
5. Desplácese hasta el final para ver la consulta SQL generada.

Desafíos y limitaciones

La generación de IA no es determinista. Con el mismo mensaje de usuario, genera finalizaciones ligeramente diferentes cada vez. Verificar antes de usar.

¿Qué sigue?

Ya que esto implica interactuar LLM con la base de datos. Los códigos podrían reutilizarse en el agente de base de datos en un flujo de trabajo multiagente.