sqlgen nb

sqlgen-nb

Schreiben Sie PostgreSQL-Abfragen aus natürlicher Sprache mit lokaler KI der Generation – mit IBM Granite Code LLM, Ollama auf Juypter Notebook.

Inspiriert von einem von Anthropic erwähnten Anwendungsfall – handelt es sich jedoch um eine lokale Implementierung mit minimalem Setup.

Installation

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

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

Ollama https://ollama.com/download

IBM Granite Code -Modell im Terminal abrufen:

ollama pull granite-code:8b-instruct 

Verwendung

1. Öffnen Sie sqlgen.ipynb
2. Geben Sie Ihr Datenbankschema in schema = ein
3. Geben Sie Ihre Frage in user_prompt =
4. Klicken Sie auf Run all um alle Zellen im Juypter-Notizbuch auszuführen
5. Scrollen Sie nach unten, um die generierte SQL-Abfrage anzuzeigen

Herausforderungen und Einschränkungen

Gen AI ist nicht deterministisch. Bei derselben Benutzeraufforderung werden jedes Mal leicht unterschiedliche Vervollständigungen generiert. Vor Gebrauch überprüfen.

Was kommt als nächstes?

Da dies die Interaktion von LLM mit der Datenbank beinhaltet. Die Codes könnten im Datenbankagenten im Multiagenten-Workflow wiederverwendet werden.