llm-sql-prompt


Namellm-sql-prompt JSON
Version 0.3.0 PyPI version JSON
download
home_page
SummaryUtility to generate ChatGPT prompts for SQL writing, offering table structure snapshots and sample row data from Postgres and sqlite databases.
upload_time2024-01-31 18:39:08
maintainer
docs_urlNone
authorMichael Bianco
requires_python>=3.11,<4.0
license
keywords chatgpt sql prompt llms database postgres sqlite
VCS
bugtrack_url
requirements No requirements were recorded.
Travis-CI No Travis.
coveralls test coverage No coveralls.
            # 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
- [ ] multiple tables
- [ ] prompt tweaking
- [ ] understand prompt size limits and sample records until one fits
            

Raw data

            {
    "_id": null,
    "home_page": "",
    "name": "llm-sql-prompt",
    "maintainer": "",
    "docs_url": null,
    "requires_python": ">=3.11,<4.0",
    "maintainer_email": "",
    "keywords": "chatgpt,sql,prompt,llms,database,postgres,sqlite",
    "author": "Michael Bianco",
    "author_email": "mike@mikebian.co",
    "download_url": "https://files.pythonhosted.org/packages/aa/48/068aeb1008b78808dcf97e7c2355d6b9fb40d01e490620dac2df0002a8b1/llm_sql_prompt-0.3.0.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- [ ] multiple tables\n- [ ] prompt tweaking\n- [ ] understand prompt size limits and sample records until one fits",
    "bugtrack_url": null,
    "license": "",
    "summary": "Utility to generate ChatGPT prompts for SQL writing, offering table structure snapshots and sample row data from Postgres and sqlite databases.",
    "version": "0.3.0",
    "project_urls": null,
    "split_keywords": [
        "chatgpt",
        "sql",
        "prompt",
        "llms",
        "database",
        "postgres",
        "sqlite"
    ],
    "urls": [
        {
            "comment_text": "",
            "digests": {
                "blake2b_256": "94de5d3841d9655ca17963b2e3c21332ccdaff190493d94163f989f62946b2d8",
                "md5": "bb44e702a813450f6cc879fd8f04edaf",
                "sha256": "0a21700e45b38268d821efb25dd3ce6a5c85ba6bd4e73eaeb187f0b7b0cb8313"
            },
            "downloads": -1,
            "filename": "llm_sql_prompt-0.3.0-py3-none-any.whl",
            "has_sig": false,
            "md5_digest": "bb44e702a813450f6cc879fd8f04edaf",
            "packagetype": "bdist_wheel",
            "python_version": "py3",
            "requires_python": ">=3.11,<4.0",
            "size": 6042,
            "upload_time": "2024-01-31T18:39:07",
            "upload_time_iso_8601": "2024-01-31T18:39:07.867781Z",
            "url": "https://files.pythonhosted.org/packages/94/de/5d3841d9655ca17963b2e3c21332ccdaff190493d94163f989f62946b2d8/llm_sql_prompt-0.3.0-py3-none-any.whl",
            "yanked": false,
            "yanked_reason": null
        },
        {
            "comment_text": "",
            "digests": {
                "blake2b_256": "aa48068aeb1008b78808dcf97e7c2355d6b9fb40d01e490620dac2df0002a8b1",
                "md5": "4c5d28f5aef9e1e1aa68dd2b4a6b7c10",
                "sha256": "b9a06df9e40a3d11b71cc980e075d8bae7bdb9f30976066ab893138a665b9a16"
            },
            "downloads": -1,
            "filename": "llm_sql_prompt-0.3.0.tar.gz",
            "has_sig": false,
            "md5_digest": "4c5d28f5aef9e1e1aa68dd2b4a6b7c10",
            "packagetype": "sdist",
            "python_version": "source",
            "requires_python": ">=3.11,<4.0",
            "size": 4272,
            "upload_time": "2024-01-31T18:39:08",
            "upload_time_iso_8601": "2024-01-31T18:39:08.844196Z",
            "url": "https://files.pythonhosted.org/packages/aa/48/068aeb1008b78808dcf97e7c2355d6b9fb40d01e490620dac2df0002a8b1/llm_sql_prompt-0.3.0.tar.gz",
            "yanked": false,
            "yanked_reason": null
        }
    ],
    "upload_time": "2024-01-31 18:39:08",
    "github": false,
    "gitlab": false,
    "bitbucket": false,
    "codeberg": false,
    "lcname": "llm-sql-prompt"
}
        
Elapsed time: 0.17695s