datasette-enrichments-gpt by datasette

153 downloads this week        Star

README source code

datasette-enrichments-gpt

PyPI Changelog Tests License

Datasette enrichment for analyzing row data using OpenAI's GPT models

Installation

Install this plugin in the same environment as Datasette.

datasette install datasette-enrichments-gpt

Configuration

This plugin needs an OpenAI API key. Configure that in metadata.yml like so

plugins:
  datasette-enrichments-gpt:
    api_key: sk-..

Or to avoid that key being visible on /-/metadata set it as an environment variable and use this:

plugins:
  datasette-enrichments-gpt:
    api_key:
      $env: OPENAI_API_KEY

Usage

Once installed, this plugin will allow users to select rows to enrich and run them through prompts using gpt-3.5-turbo or gpt-4-turbo, saving the result of the prompt in the specified column.

The plugin also provides gpt-4-turbo vision, which can run prompts against an image identified by a URL.

Development

To set up this plugin locally, first checkout the code. Then create a new virtual environment:

cd datasette-enrichments-gpt
python3 -m venv venv
source venv/bin/activate

Now install the dependencies and test dependencies:

pip install -e '.[test]'

To run the tests:

pytest