如何从工具访问RunnableConfig

先决条件

本指南假定您熟悉以下概念

如果您有一个工具，它调用聊天模型、检索器或其他可运行对象，您可能希望访问这些可运行对象的内部事件或为其配置额外的属性。本指南将向您展示如何正确手动传递参数，以便您可以使用 astream_events() 方法实现此目的。

工具是可运行对象，在接口层面，您可以像对待其他任何可运行对象一样对待它们——您可以正常地在它们上调用 invoke()、batch() 和 stream()。然而，在编写自定义工具时，您可能希望调用其他可运行对象，例如聊天模型或检索器。为了正确追踪和配置这些子调用，您需要手动访问并传入工具当前的 RunnableConfig 对象。本指南将向您展示一些如何做到这一点的示例。

兼容性

本指南要求 langchain-core>=0.2.16。

通过参数类型推断

要从自定义工具中引用并访问活动的配置对象，您需要在工具的签名中添加一个类型为 RunnableConfig 的参数。当您调用工具时，LangChain 会检查工具的签名，查找类型为 RunnableConfig 的参数，如果存在，则用正确的值填充该参数。

注意：参数的实际名称不重要，只有类型才是关键。

为了说明这一点，我们定义一个自定义工具，它接受两个参数——一个类型为字符串，另一个类型为 RunnableConfig

%pip install -qU langchain_core

from langchain_core.runnables import RunnableConfig
from langchain_core.tools import tool


@tool
async def reverse_tool(text: str, special_config_param: RunnableConfig) -> str:
    """A test tool that combines input text with a configurable parameter."""
    return (text + special_config_param["configurable"]["additional_field"])[::-1]

API 参考：RunnableConfig | 工具

然后，如果我们在一个包含 configurable 字段的 config 下调用该工具，我们可以看到 additional_field 被正确地传递了过去

await reverse_tool.ainvoke(
    {"text": "abc"}, config={"configurable": {"additional_field": "123"}}
)

'321cba'

下一步

现在您已经了解了如何从工具内部配置和流式传输事件。接下来，请查看以下指南以了解更多关于使用工具的信息

你还可以查看工具调用的其他一些特定用法

构建使用工具的链和代理
从模型获取结构化输出

通过参数类型推断​

下一步​

通过参数类型推断

下一步