sqlgen nb

sqlgen-nb

เขียนแบบสอบถาม PostgreSQL จากภาษาธรรมชาติโดยใช้ AI เจนเนอเรชั่นภายในองค์กร -- ด้วย IBM Granite Code LLM, Ollama บน Juypter Notebook

แรงบันดาลใจจากกรณีการใช้งานที่ Anthropic กล่าวถึง แต่เป็นการใช้งานในท้องถิ่นที่มีการตั้งค่าเพียงเล็กน้อย

การติดตั้ง

สมุดบันทึก Jupyter https://marketplace.visualstudio.com/items?itemName=ms-toolsai.jupyter

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

โอลามะ https://ollama.com/download

ดึงโมเดล IBM Granite Code ใน Terminal:

ollama pull granite-code:8b-instruct 

การใช้งาน

1. เปิด sqlgen.ipynb
2. ป้อนสคีมาฐานข้อมูลของคุณใน schema =
3. ป้อนคำถามของคุณใน user_prompt =
4. คลิก Run all เพื่อเรียกใช้เซลล์ทั้งหมดใน Juypter Notebook
5. เลื่อนไปที่ด้านล่างเพื่อดูแบบสอบถาม SQL ที่สร้างขึ้น

ความท้าทายและข้อจำกัด

Gen AI ไม่สามารถกำหนดได้ ด้วยข้อความแจ้งผู้ใช้คนเดียวกัน มันสร้างความสำเร็จที่แตกต่างกันเล็กน้อยทุกครั้ง ตรวจสอบก่อนใช้งาน

อะไรต่อไป?

เนื่องจากเกี่ยวข้องกับการโต้ตอบ LLM กับฐานข้อมูล รหัสสามารถนำมาใช้ซ้ำได้ในเอเจนต์ฐานข้อมูลในเวิร์กโฟลว์แบบหลายเอเจนต์