sqlgen nb

sqlgen-nb

Напишите запрос PostgreSQL на естественном языке с использованием локального искусственного интеллекта — с помощью 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 в Терминале:

ollama pull granite-code:8b-instruct 

Использование

1. Открыть sqlgen.ipynb
2. Введите схему вашей базы данных в schema =
3. Введите свой вопрос в user_prompt =
4. Нажмите Run all , чтобы запустить все ячейки в блокноте Juypter.
5. Прокрутите вниз, чтобы увидеть созданный SQL-запрос.

Проблемы и ограничения

Ген ИИ недетерминирован. При использовании одного и того же приглашения пользователя он каждый раз генерирует немного разные завершения. Проверьте перед использованием.

Что дальше?

Поскольку это предполагает взаимодействие LLM с базой данных. Коды можно повторно использовать в агенте базы данных в многоагентном рабочем процессе.