sqlgen nb

sqlgen-nb

使用本地 gen AI 透過 IBM Granite Code LLM、Ollama on Juypter Notebook 從自然語言編寫 PostgreSQL 查詢。

受到 Anthropic 提到的一個用例的啟發——但它是一個具有最少設定的本地實作。

安裝

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

浪鏈 https://python.langchain.com/docs/how_to/installation/

奧拉瑪 https://ollama.com/download

在終端機中拉取IBM Granite Code模型：

ollama pull granite-code:8b-instruct 

用法

1. 打開sqlgen.ipynb
2. 在schema =中輸入您的資料庫架構
3. 在user_prompt =中輸入您的問題
4. 按Run all以執行 Juypter Notebook 中的所有儲存格
5. 捲動到底部以查看生成的 SQL 查詢

挑戰與限制

Gen AI 是不確定性的。使用相同的使用者提示，每次都會產生略有不同的完成結果。使用前請確認。

接下來怎麼辦？

因為這涉及到 LLM 與資料庫的交互。這些程式碼可以在多代理程式工作流程中的資料庫代理程式中重複使用。