结构化工具代理完成

Question

我使用Langchain创建了一个结构化工具，我想在使用该工具后直接输出agent finish。在初始化该工具时，kwargs 中是否有任何选项可以传递给该工具？或者我只需使用 Base Tool 重新制作该工具即可。

 Tool(
            name="GetStock",
            func=get_stock,
            description=get_stock_description,
        )

Answer 1

您可以使用

BaseTool

和

return_redirect

标志来实现这一点。

class GetStockTool(BaseTool):

    name = "Custom tool"
    description = "Custom tool for some task"

    def __init__(self):
        super().__init__()
        self.return_direct = True

    def _run(self, query: str):
        """Return stock data"""
        # logic to return data

    def _arun(self, query: str):
        raise NotImplementedError("This tool does not support async")

BaseTool

示例