如何使用LangChain查询PostgreSQL非public模式下的表？

我有一个PostgreSQL数据库，想用LangChain查询默认public模式以外的表和视图，但langchain_community.utilities.sql_database.SQLDatabase.get_table_info()方法执行的SQL仅返回public模式下的表列表：

SELECT pg_catalog.pg_class.relname 
    FROM pg_catalog.pg_class JOIN pg_catalog.pg_namespace ON pg_catalog.pg_namespace.oid = pg_catalog.pg_class.relnamespace 
    WHERE pg_catalog.pg_class.relkind = ANY (ARRAY['r', 'p']) AND pg_catalog.pg_class.relpersistence != 't' AND pg_catalog.pg_table_is_visible(pg_catalog.pg_class.oid) AND pg_catalog.pg_namespace.nspname != 'pg_catalog'

问题1：如何让LangChain识别public模式以外的表？ #

• 方法一：初始化时指定目标模式
  创建SQLDatabase实例时，通过schemas参数显式指定需要访问的模式列表，示例代码：

  from langchain_community.utilities.sql_database import SQLDatabase
  
  db = SQLDatabase.from_uri(
      "postgresql://user:password@host:port/dbname",
      schemas=["public", "your_custom_schema"]
  )
  

  这样LangChain会自动遍历指定的模式，获取其中的表和视图信息。

• 方法二：修改表信息查询逻辑
  原SQL中的pg_catalog.pg_table_is_visible会过滤掉当前会话不可见的模式表，你可以自定义查询语句，去掉该限制并指定目标模式：

  SELECT pg_catalog.pg_namespace.nspname || '.' || pg_catalog.pg_class.relname AS full_table_name
  FROM pg_catalog.pg_class 
  JOIN pg_catalog.pg_namespace ON pg_catalog.pg_namespace.oid = pg_catalog.pg_class.relnamespace 
  WHERE pg_catalog.pg_class.relkind = ANY (ARRAY['r', 'p', 'v']) -- 包含视图v
    AND pg_catalog.pg_class.relpersistence != 't' 
    AND pg_catalog.pg_namespace.nspname IN ('public', 'your_custom_schema')
    AND pg_catalog.pg_namespace.nspname != 'pg_catalog'
  

  之后通过自定义SQLDatabase子类或直接替换原查询逻辑来生效。

问题2：如何让LangChain使用完整表名（schema_name.table_basename）查询表？ #

• 方法一：初始化时开启模式包含
  创建SQLDatabase实例时设置include_schemas=True，这样LangChain生成的表信息会自动带上模式前缀，示例：

  db = SQLDatabase.from_uri(
      "postgresql://user:password@host:port/dbname",
      schemas=["public", "your_custom_schema"],
      include_schemas=True
  )
  

  此时LangChain生成SQL查询时会自动使用schema.table的完整格式。

• 方法二：自定义表信息输出
  若需要更精细的控制，可以重写get_table_info方法，确保返回的表名都是完整的schema.table格式，让LangChain在生成查询时直接沿用该格式。

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