Bee Agent Framework 中 SQLTool 工具对 PostgreSQL 模式(schema)支持的技术解析

在数据库应用开发中，PostgreSQL 的模式(schema)管理是一个重要特性。本文将深入分析 Bee Agent Framework 中 SQLTool 工具对 PostgreSQL 模式支持的技术实现和改进方案。

问题背景

PostgreSQL 数据库使用模式(schema)作为数据库对象的逻辑分组机制。默认情况下，PostgreSQL 的 search_path 设置为 public 模式。当应用程序需要访问非 public 模式下的表时，必须在查询中显式指定模式名称，否则会导致查询失败。

在 Bee Agent Framework 的 SQLTool 工具中，当 LLM 生成 SQL 查询时，经常会出现因缺少模式前缀而导致查询失败的情况。这主要是因为 LLM 生成的查询默认使用 public 模式，而实际表可能位于其他模式中。

技术解决方案

开发团队提出了三种可能的解决方案：

1. 数据库层面解决方案：在 EDB(EnterpriseDB)环境中创建 public 同义词
2. 配置层面解决方案：修改数据库的 search_path 参数
3. 工具层面解决方案：增强 SQLTool 工具，使其能够正确处理模式名称

最终，团队选择了第三种方案，对 SQLTool 工具进行了以下改进：

实现细节

改进后的 SQLTool 工具采用了更智能的查询生成流程：

1. 接收用户查询请求
2. 首先获取数据库元数据(包括模式信息)
3. 基于完整元数据生成查询(此时可包含正确的模式前缀)

这种改进显著减少了与 LLM 的往返交互次数，提高了查询生成的准确率。在实现上，工具现在能够在第二次尝试时自动添加正确的模式名称前缀。

技术影响

这一改进带来了以下技术优势：

1. 更高的查询成功率：减少了因模式缺失导致的查询失败
2. 更好的性能：减少了与 LLM 的交互次数
3. 更智能的查询生成：基于完整元数据生成更准确的查询

最佳实践

对于使用 Bee Agent Framework 的开发人员，建议：

1. 确保在初始化 SQLTool 时提供完整的数据库连接信息
2. 对于复杂查询场景，考虑在提示词中加入模式名称提示
3. 及时更新到最新版本以获取最佳的模式支持体验

这一改进已在 Bee Agent Framework 0.0.41 版本中发布，显著提升了框架在 PostgreSQL 环境中的可用性和可靠性。