# ChatGPT Prompt for SQL Writing
Generate a prompt for writing SQL queries with LLMs like ChatGPT. Drop your database URL and table name into the script and it will generate a prompt for you to copy and paste into your favorite LLM.
## What this does
- Snapshot of Table Structure: Understand the columns, types, and organization of your table at a glance.
- Sample Rows: Includes INSERT statements to describe the data in your table.
## Usage
Install the package:
```shell
pip install llm-sql-prompt
```
Here's how to use it:
```shell
Usage: llm-sql-prompt [OPTIONS] DATABASE_URL [TABLE_NAME]
Options:
--help Show this message and exit.
```
Generate a prompt from a postgres database:
```shell
llm-sql-prompt postgresql://postgres:postgres@localhost:5555/database_name table_name | pbcopy
llm-sql-prompt $DATABASE_URL
```
### Tunneling to a remote port
If you find yourself wanting to tunnel into a remote box and work with a production database, here's some helpful commands so you don't need to remember the weird SSH tunneling syntax:
```shell
function find_random_open_port() {
local start_port=${1:-1024}
local max_attempts=100
local attempt=0
local port=$start_port
while (( attempt < max_attempts )); do
if ! nc -z localhost $port 2>/dev/null; then
echo $port
return
fi
port=$((port + 1))
attempt=$((attempt + 1))
done
echo "No open port found after $max_attempts attempts, starting from $start_port." > /dev/stderr
return 1
}
function ssh-tunnel() {
if [ $# -lt 2 ]; then
echo "Usage: ssh-tunnel remote_host remote_port [local_port]"
echo "This function sets up SSH port forwarding."
return 1
fi
local remote_host=$1
local remote_port=$2
local local_port=${3:-$(find-random-open-port $remote_port)}
if [[ -z $local_port ]]; then
echo "Failed to find an open local port."
return 1
fi
echo "Forwarding local port $local_port to remote port $remote_port on $remote_host..."
set -x
ssh $remote_host -L ${local_port}:localhost:${remote_port}
}
```
## TODO
Super basic script, needs a lot of work
- [x] pg support
- [x] one entrypoint
- [ ] use DB comments on columns + tables
- [x] multiple tables
- [ ] prompt tweaking
- [ ] understand prompt size limits and sample records until one fits
Raw data
{
"_id": null,
"home_page": "https://pypi.org/project/llm-sql-prompt/",
"name": "llm-sql-prompt",
"maintainer": null,
"docs_url": null,
"requires_python": "<4.0,>=3.11",
"maintainer_email": null,
"keywords": "chatgpt, sql, prompt, llms, database, postgres, sqlite",
"author": "Michael Bianco",
"author_email": "mike@mikebian.co",
"download_url": "https://files.pythonhosted.org/packages/f5/04/3351c4a61750b03c71d0593aa37e84cd0027a8d5b6e1130eae1d1f7c4c09/llm_sql_prompt-0.5.1.tar.gz",
"platform": null,
"description": "# ChatGPT Prompt for SQL Writing\n\nGenerate a prompt for writing SQL queries with LLMs like ChatGPT. Drop your database URL and table name into the script and it will generate a prompt for you to copy and paste into your favorite LLM.\n\n## What this does\n\n- Snapshot of Table Structure: Understand the columns, types, and organization of your table at a glance.\n- Sample Rows: Includes INSERT statements to describe the data in your table.\n\n## Usage\n\nInstall the package:\n\n```shell\npip install llm-sql-prompt\n```\n\nHere's how to use it:\n\n```shell\nUsage: llm-sql-prompt [OPTIONS] DATABASE_URL [TABLE_NAME]\n\nOptions:\n --help Show this message and exit.\n```\n\nGenerate a prompt from a postgres database:\n\n```shell\nllm-sql-prompt postgresql://postgres:postgres@localhost:5555/database_name table_name | pbcopy\nllm-sql-prompt $DATABASE_URL\n```\n\n### Tunneling to a remote port\n\nIf you find yourself wanting to tunnel into a remote box and work with a production database, here's some helpful commands so you don't need to remember the weird SSH tunneling syntax:\n\n```shell\nfunction find_random_open_port() {\n local start_port=${1:-1024}\n local max_attempts=100\n local attempt=0\n local port=$start_port\n\n while (( attempt < max_attempts )); do\n if ! nc -z localhost $port 2>/dev/null; then\n echo $port\n return\n fi\n port=$((port + 1))\n attempt=$((attempt + 1))\n done\n\n echo \"No open port found after $max_attempts attempts, starting from $start_port.\" > /dev/stderr\n return 1\n}\n\n\nfunction ssh-tunnel() {\n if [ $# -lt 2 ]; then\n echo \"Usage: ssh-tunnel remote_host remote_port [local_port]\"\n echo \"This function sets up SSH port forwarding.\"\n return 1\n fi\n\n local remote_host=$1\n local remote_port=$2\n local local_port=${3:-$(find-random-open-port $remote_port)}\n\n if [[ -z $local_port ]]; then\n echo \"Failed to find an open local port.\"\n return 1\n fi\n\n echo \"Forwarding local port $local_port to remote port $remote_port on $remote_host...\"\n set -x\n ssh $remote_host -L ${local_port}:localhost:${remote_port}\n}\n```\n\n## TODO\n\nSuper basic script, needs a lot of work\n\n- [x] pg support\n- [x] one entrypoint\n- [ ] use DB comments on columns + tables\n- [x] multiple tables\n- [ ] prompt tweaking\n- [ ] understand prompt size limits and sample records until one fits",
"bugtrack_url": null,
"license": null,
"summary": "Utility to generate ChatGPT prompts for SQL writing, offering table structure snapshots and sample row data from Postgres and sqlite databases.",
"version": "0.5.1",
"project_urls": {
"Homepage": "https://pypi.org/project/llm-sql-prompt/"
},
"split_keywords": [
"chatgpt",
" sql",
" prompt",
" llms",
" database",
" postgres",
" sqlite"
],
"urls": [
{
"comment_text": "",
"digests": {
"blake2b_256": "9a40da8e2a6ba25cb969d414e16730c27985bf91e484141febbbd42f0f17d8e1",
"md5": "608b848c834d9fa243c3f05865916cd9",
"sha256": "ee6b3829951097f6a155a00a7cdb4a3b6d2303dfea9588ef0cd97cc023441e0a"
},
"downloads": -1,
"filename": "llm_sql_prompt-0.5.1-py3-none-any.whl",
"has_sig": false,
"md5_digest": "608b848c834d9fa243c3f05865916cd9",
"packagetype": "bdist_wheel",
"python_version": "py3",
"requires_python": "<4.0,>=3.11",
"size": 6550,
"upload_time": "2024-07-30T22:53:35",
"upload_time_iso_8601": "2024-07-30T22:53:35.849835Z",
"url": "https://files.pythonhosted.org/packages/9a/40/da8e2a6ba25cb969d414e16730c27985bf91e484141febbbd42f0f17d8e1/llm_sql_prompt-0.5.1-py3-none-any.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "f5043351c4a61750b03c71d0593aa37e84cd0027a8d5b6e1130eae1d1f7c4c09",
"md5": "1497c09713f5bb6c0727d3a37f1d3e7f",
"sha256": "26c369b0a034f5feb19b099e34184bc87a380e56c3802d731a7e415c2d211db3"
},
"downloads": -1,
"filename": "llm_sql_prompt-0.5.1.tar.gz",
"has_sig": false,
"md5_digest": "1497c09713f5bb6c0727d3a37f1d3e7f",
"packagetype": "sdist",
"python_version": "source",
"requires_python": "<4.0,>=3.11",
"size": 4821,
"upload_time": "2024-07-30T22:53:36",
"upload_time_iso_8601": "2024-07-30T22:53:36.646879Z",
"url": "https://files.pythonhosted.org/packages/f5/04/3351c4a61750b03c71d0593aa37e84cd0027a8d5b6e1130eae1d1f7c4c09/llm_sql_prompt-0.5.1.tar.gz",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2024-07-30 22:53:36",
"github": false,
"gitlab": false,
"bitbucket": false,
"codeberg": false,
"lcname": "llm-sql-prompt"
}
JSON