如何使用 pg_gpt 项目进行自然语言查询 PostgreSQL 数据库

如何使用 pg_gpt 项目进行自然语言查询 PostgreSQL 数据库

pg_gpt Experimental extension that brings OpenAI API to your PostgreSQL to run queries in human language. 项目地址: https://gitcode.com/gh_mirrors/pg/pg_gpt

项目基础介绍和主要编程语言

pg_gpt 是一个实验性的 PostgreSQL 扩展，它将 OpenAI API 集成到 PostgreSQL 中，使用户能够使用自然语言来编写查询。此项目利用 OpenAI GPT 模型的能力，通过用户提供的输入来生成相应的 SQL 查询语句。

主要编程语言

• Rust: pg_gpt 扩展是用 Rust 语言编写的。Rust 是一种系统编程语言，它能够提供高性能、并发性和安全性，非常适用于构建系统级扩展。

新手使用项目时需要注意的三个问题及解决方案

问题 1：如何安装 pg_gpt 扩展？

解决步骤：

1. 确保你安装了 Rust 的 cargo 工具和 pgx。可以通过以下命令安装 cargo-pgx：

   cargo install --locked cargo-pgx
   cargo pgx init
   

2. 克隆 pg_gpt 项目到本地：

   git clone ***
   ***
   

3. 设置 OpenAI API 密钥。在运行命令之前，请确保你拥有有效的 OpenAI API 密钥，并将其赋值给环境变量 OPENAI_KEY：

   export OPENAI_KEY=<YOUR_KEY>
   

4. 安装扩展到 PostgreSQL：

   cargo pgx run # 将会进入到 psql shell
   

5. 在 psql shell 中创建 pg_gpt 扩展：

   create extension pg_gpt;
   set openai_key = '<YOUR OPENAPI API KEY HERE>';
   

6. 现在，你可以使用 pg_gpt 扩展进行自然语言查询了。

问题 2：如何使用 pg_gpt 执行自然语言查询？

解决步骤：

1. 使用 gpt 函数，传入自然语言描述的查询语句。例如：

   select gpt('show me all open aws s3 buckets');
   

2. 扩展会调用 OpenAI API 来生成相应的 SQL 查询语句。
3. 你可以根据需要执行返回的查询语句来获取结果。

问题 3：如何处理 pg_gpt 的使用限制？

解决步骤：

1. 记住 pg_gpt 目前是实验性项目，不建议在生产数据库上使用。
2. 由于使用了 gpt-3.5-turbo，查询受到 4096 tokens 的限制。在使用时，请尽量提供简明扼要的查询描述。
3. 如果面对大型数据库模式，考虑使用 gpt_tables 函数来仅针对特定的表格模式生成查询。
4. 在使用该插件时，请确保只发送数据库模式（不包含数据），以保护数据安全。

请注意，由于 pg_gpt 是一个实验性项目，用户应谨慎使用，并密切关注 CloudQuery 的官方公告和更新，以便及时了解该项目的最新进展和支持情况。

pg_gpt Experimental extension that brings OpenAI API to your PostgreSQL to run queries in human language. 项目地址: https://gitcode.com/gh_mirrors/pg/pg_gpt