使用 LangChain 构建您自己的数据库机器人

几天前，在我的一个小组中有一个想法讨论，其中一个让我着迷的想法是将 OpenAI 的功能与我们的数据库连接起来，并以自然语言与之交互以获取数据。这就是激发我写这篇博客的全部想法。

介绍：

故事非常简单，我们使用 OpenAI 嵌入来理解自然语言来制定 SQL 查询并在数据库上运行生成的 SQL 查询，获取结果，然后将结果翻译回自然语言。这个想法可以进一步扩展，但我们将看到它的基本实现。

没有进一步的延迟让我们跳到代码。

实施：

安装 postgresql 并将其与您喜欢的编辑器连接。然后运行下面的sql脚本。

-- Create the orders table if it doesn't exist
CREATE TABLE Orders (
    OrderID INT PRIMARY KEY,
    CustomerID INT,
    OrderDate DATE,
    TotalAmount DECIMAL(10, 2),
    ShippingAddress VARCHAR(255)
);

--Insert sample orders into the orders table
INSERT INTO Orders (OrderID, CustomerID, OrderDate, TotalAmount, ShippingAddress)
VALUES (1, 1001, '2023-06-01', 150.99, '123 Main St, City A'),
       (2, 1002, '2023-06-02', 78.50, '456 Elm St, City