使用Langchain时db.get_usable_table_names()未返回PostgreSQL全部表的原因及解决

用户使用LangChain连接PostgreSQL数据库时，代码可正常运行，但调用db.get_usable_table_names()无法获取数据库中的全部表，相关代码如下：

import getpass
import os
from langchain_community.utilities import SQLDatabase
from langchain.chains import create_sql_query_chain
from langchain_openai import ChatOpenAI

os.environ["OPENAI_API_KEY"] = getpass.getpass()
db = SQLDatabase.from_uri("postgresql://xxx:xxx:5432/xxx")
print("db.dialect : ", db.dialect)
print("db.get_usable_table_names() : " , db.get_usable_table_names())

可能的原因 #

• 账号权限不足：连接数据库的账号未拥有部分表的访问权限（如SELECT权限），LangChain会自动过滤无权限的表。
• Schema未指定：PostgreSQL中表若属于非public schema，默认不会被LangChain检索到，默认仅扫描public schema。
• 表被显式过滤：初始化SQLDatabase时若设置了exclude_tables参数，或未指定include_tables，会导致部分表被排除。
• 系统表过滤规则：LangChain默认过滤PostgreSQL系统表（如pg_开头的表），若业务表命名符合该规则，会被误过滤。

解决方法 #

1. 检查并提升账号权限 #

确保连接账号拥有目标表的SELECT权限，执行以下SQL授权：

-- 授权public schema下所有表的SELECT权限
GRANT SELECT ON ALL TABLES IN SCHEMA public TO your_username;
-- 若有其他schema，需对应授权
GRANT SELECT ON ALL TABLES IN SCHEMA your_target_schema TO your_username;

2. 指定扫描的Schema #

初始化SQLDatabase时，通过参数指定需要包含的Schema：

# 方式1：指定单个Schema
db = SQLDatabase.from_uri("postgresql://xxx:xxx@xxx:5432/xxx", schema="your_target_schema")

# 方式2：包含所有Schema
db = SQLDatabase.from_uri("postgresql://xxx:xxx@xxx:5432/xxx", include_schemas=True)

3. 显式指定要包含的表 #

通过include_tables参数精确指定需要加载的表：

db = SQLDatabase.from_uri(
    "postgresql://xxx:xxx@xxx:5432/xxx",
    include_tables=["table_a", "table_b", "table_c"]
)

4. 关闭系统表过滤（按需使用） #

若业务表被误当作系统表过滤，可关闭默认过滤规则：

db = SQLDatabase.from_uri(
    "postgresql://xxx:xxx@xxx:5432/xxx",
    ignore_system_tables=False
)

5. 验证表的存在性 #

通过PostgreSQL客户端执行以下SQL，确认表存在且当前账号可访问：

SELECT table_schema, table_name FROM information_schema.tables WHERE table_type = 'BASE TABLE';

内容的提问来源于stack exchange，提问作者Yatan