sqlgen nb

sqlgen-nb

Juypter Notebook에서 IBM Granite Code LLM, Ollama를 사용하여 온프레미스 세대 AI를 사용하여 자연어에서 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과 데이터베이스의 상호 작용이 포함됩니다. 코드는 다중 에이전트 워크플로우의 데이터베이스 에이전트에서 재사용될 수 있습니다.