修复 LlamaIndex 中 OpenAI SQL 查询问题

本文解决了一个在使用 LlamaIndex 和 OpenAI 查询 SQL 数据库时遇到的常见问题。用户报告称，在最近的更新后，语言模型不再能够执行 SQL 查询，导致出现错误消息。本指南提供了该问题的全面概述、潜在原因以及恢复功能的逐步解决方案。

核心问题在于 LlamaIndex 框架内 OpenAI 的语言模型与 SQL 数据库之间的交互。以前，用户可以利用 OpenAI 生成 SQL 查询并无缝检索数据。然而，最近的更新引入了兼容性问题，导致出现错误消息："抱歉，作为 AI 语言模型，我无法在实时数据库上执行 SQL 查询或访问任何外部系统来检索实时数据。"

此问题的根本原因在于 LlamaIndex 中某些类的弃用，特别是 `SQLStructStoreQueryEngine` 和 `NLStructStoreQueryEngine`。这些类已被 `SQLTableRetriever` 取代。此外，`DEFAULT_RESPONSE_SYNTHESIS_PROMPT` 的更新也可能导致此问题。引入具有提示验证功能的 `BaseSQLTableQueryEngine` 进一步增加了过程的复杂性。

推荐的解决方案是将您的代码迁移到使用新的 `SQLTableRetriever` 类。以下是如何实现此目的的示例： ```python from llama_index import SQLDatabase from sqlalchemy import create_engine from llama_index.indices.struct_store import SQLTableRetriever # 创建一个 SQLDatabase 实例（假设您有数据库连接） engine = create_engine('your_database_connection_string') db_northwind = SQLDatabase(engine) # 创建一个 SQLTableRetriever 实例 table_retriever = SQLTableRetriever(sql_database=db_northwind) # 定义您的查询 query = "What is the best selling product?" # 使用 SQLTableRetriever 实例执行查询 response = table_retriever.query(query) print(response) ``` 此代码片段演示了如何创建 `SQLTableRetriever` 实例并使用它来执行 SQL 查询。请将 `'your_database_connection_string'` 替换为您的实际数据库连接字符串。

确保您的提示根据 `BaseSQLTableQueryEngine` 的新要求正确格式化。请密切关注 `response_synthesis_prompt`，并确保其符合预期的结构。格式不正确的提示可能导致验证错误，并阻止查询成功执行。

考虑探索 LlamaIndex 中引入的新类，例如 `NLSQLTableQueryEngine`、`PGVectorSQLQueryEngine` 和 `SQLTableRetrieverQueryEngine`。这些类利用 `NLSQLRetriever` 来执行 SQL 查询，并可能为您的特定用例提供改进的性能或功能。评估这些替代方案是否适合您的需求。

如果您仍然遇到问题，请按照以下故障排除步骤进行操作： 1. **验证 LlamaIndex 版本：** 确保您使用的是最新版本的 LlamaIndex。 2. **检查错误消息：** 仔细检查任何错误消息，以获取有关问题原因的线索。 3. **审查代码：** 仔细检查您的代码是否存在任何错误或不一致之处。 4. **查阅文档：** 参考 LlamaIndex 文档，了解有关新类和更新的详细信息。 5. **简化查询：** 尝试一个更简单的 SQL 查询来隔离问题。 6. **检查数据库连接：** 验证您的数据库连接是否正常工作。

如果您仍然卡住，请利用 LlamaIndex 社区寻求支持。查看 LlamaIndex GitHub 存储库以获取开放问题和讨论。考虑在 LlamaIndex Discord 服务器或 Stack Overflow 上发布您的问题，并提供有关您的设置和遇到的错误的详细信息。

通过了解 LlamaIndex 的变化并迁移到新的 `SQLTableRetriever` 类，您可以解决 OpenAI 无法执行 SQL 查询的问题。请记住仔细格式化您的提示，并探索 LlamaIndex 中可用的新类以优化您的 SQL 查询性能。如果您遇到进一步的挑战，请随时向 LlamaIndex 社区寻求帮助。