LangChain集成PowerBI遇ConfigError问题求助

问题描述 #

使用LangChain的PowerBI集成功能时，持续触发错误：

ConfigError: field "credential" not yet prepared so type is still a ForwardRef, you might need to call PowerBIDataset.update_forward_refs()

现有代码 #

load_dotenv('credentials.env')

os.environ["AZURE_TENANT_ID"] = os.environ.get("AZURE_TENANT_ID")
os.environ["AZURE_CLIENT_ID"] = os.environ.get("AZURE_CLIENT_ID")
os.environ["AZURE_CLIENT_SECRET"] = os.environ.get("AZURE_CLIENT_SECRET")

username=<my_username>
password=<my_password>
bearer_token=<my_token>

llm = AzureChatOpenAI(
        deployment_name='gpt-4',
        temperature=0.5,
        max_tokens=1000,
        streaming=True,
        api_version="2023-12-01-preview"
)

credential = DefaultAzureCredential()

powerbi = PowerBIDataset(
    dataset_id=dataset_id,
    table_names=["Product", "Project"],
    token = bearer_token
)

toolkit = PowerBIToolkit(
    powerbi=powerbi,
    llm=llm,
    output_token_limit = 1000
)

agent_executor = create_pbi_agent(
    llm=llm,
    toolkit=toolkit,
    verbose=True
)

已尝试的无效方案 #

• 调用PowerBIDataset.update_forward_refs()，报错：name 'TokenCredential' is not defined
• 尝试token = TokenCredential(bearer_token)，报错：Protocols cannot be instantiated
• 仅传入token不使用credential字段，Agent仅能识别指定表名，无法获取表结构，执行日志如下：

  Now that I know the Product table exists, I can retrieve the schema for it to identify the columns.
  Action: schema_powerbi
  Action Input: Product
  Error while getting table info for Product: 'results'
  Observation: unknown
  Thought:

有效解决方案 #

1. 修复ForwardRef错误：正确导入并处理TokenCredential #

首先导入依赖并显式更新ForwardRefs，解决TokenCredential未定义问题：

from azure.core.credentials import TokenCredential
from langchain.tools import PowerBIDataset

# 显式更新类型引用
PowerBIDataset.update_forward_refs(TokenCredential=TokenCredential)

2. 正确传递凭证：二选一方案 #

方案A：使用DefaultAzureCredential推荐方式

删除手动传入的token参数，改用已初始化的credential实例，确保Azure环境变量配置正确：

powerbi = PowerBIDataset(
    dataset_id=dataset_id,
    table_names=["Product", "Project"],
    credential=credential  # 使用DefaultAzureCredential实例
)

此方式依赖已正确加载的AZURE_TENANT_ID、AZURE_CLIENT_ID、AZURE_CLIENT_SECRET环境变量。

方案B：手动包装Bearer Token为TokenCredential兼容对象

若必须使用手动获取的bearer_token，需创建符合TokenCredential协议的自定义类：

from azure.core.credentials import TokenCredential, AccessToken

class StaticTokenCredential(TokenCredential):
    def __init__(self, token):
        self.token = token
    def get_token(self, *scopes, **kwargs):
        # 按需设置合理的过期时间，示例设为永久有效
        return AccessToken(self.token, expires_on=float("inf"))

# 创建静态凭证实例
static_cred = StaticTokenCredential(bearer_token)

# 初始化PowerBIDataset
powerbi = PowerBIDataset(
    dataset_id=dataset_id,
    table_names=["Product", "Project"],
    credential=static_cred
)

3. 验证权限配置 #

确保使用的凭证（无论是DefaultAzureCredential还是静态token）拥有PowerBI数据集的读取权限，包括获取表元数据的权限。可在PowerBI门户中检查数据集的共享设置，确认当前身份有权访问该数据集的结构信息。

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