如何将一个字典列表作为输入传递给Pydantic的Langchain工具函数？

Question

你好，我正在尝试创建一个代理工具，该工具接受结构化输入数据（作为包含'price'键的字典列表），并计算这些价格的总和。

我的输入数据是这样的：

input_data=[
    {'Junk_Food_name': 'Meat Lovers Pizza', 'price': 800},
    {'Junk_Food_name': 'Chicken Alfredo Pizza', 'price': 800},
    {'Junk_Food_name': 'Cheeseburger Pizza', 'price': 800},
    {'Junk_Food_name': 'Chicken Bacon Ranch Pizza', 'price': 800},
    {'Junk_Food_name': 'Chicken Pesto Pizza', 'price': 800}
]

为了处理这个任务，我定义了一个函数，并尝试使用 Pydantic 来确保输入格式正确。这是我的函数定义：

from typing import List, Dict
from pydantic import BaseModel, Field
from langchain.agents import tool

class InputData(BaseModel):
    input_data: List[Dict[str, int]] = Field(description="List of input data with 'Junk_Food_name' and 'price' fields")

@tool(args_schema=InputData)
def calc_price(input_data:List[Dict[str, int]])-> int:
    """ 计算订单的最终总价 """
    sum_total = 0
    for item in input_data:
        sum_total += int(item['price'])
    
    return sum_total

然而，当我尝试将上述输入数据传递给这个函数时，我遇到了以下错误：

ValidationError: 1 validation error for InputData
input_data
  field required (type=value_error.missing)

我希望能得到输出结果为 4000（因为5个披萨每个800元，总共是4000元）。请问有人能帮我解决这个错误吗？

tim_yates · Answer

运行以下命令来检查你的 Pydantic 版本：

pip show pydantic

你可能正在使用 Pydantic v2，而 langchain 目前仍在使用基于 Pydantic v1 的 BaseModel。因此，请将导入语句从这样：

from pydantic import BaseModel, Field

改为：

from pydantic.v1 import BaseModel, Field

apsillers · Answer

在你的工具函数定义中，你可以添加一个参数来处理验证错误，并将其作为观察结果传递给 LangChain，以便根据该错误生成响应。在你的代码中，修改如下：

# 定义工具函数
@tool(args_schema=InputData, handle_validation_error=True)
def calc_price(input_data: List[Dict[str, int]]) -> int:

这样，当遇到输入验证错误时，错误会被适当地管理，并反馈给LangChain以进行后续处理。相关参考代码可在此链接中找到：

LangChain Tools 源代码