为工具使用定义函数
您在一家 SaaS(软件即服务)公司工作,目标是为组织内各层级的员工推出工具,帮助他们基于数据做出决策。您正在为一个应用的 PoC(概念验证)做原型设计,使客户成功经理能够通过自然语言与公司数据交互,以检索重要的客户信息。
系统已为您提供一个名为 customers 的 pandas DataFrame,包含一小部分客户数据。此项目的第一步是定义一个 Python 函数,根据客户名称从该表中提取信息。pandas 已经以 pd 的名称导入。
本练习是课程的一部分
使用 LangChain 开发 LLM 应用
练习说明
- 定义一个
retrieve_customer_info()函数,接收一个字符串参数name。 - 过滤
customersDataFrame,返回"name"等于输入参数name的行。 - 在客户名称
"Peak Performance Co."上调用该函数。
交互式实操练习
通过完成这段示例代码来试试这个练习。
# Define a function to retrieve customer info by-name
def retrieve_customer_info(____: ____) -> str:
"""Retrieve customer information based on their name."""
# Filter customers for the customer's name
customer_info = customers[customers['name'] == ____]
return customer_info.to_string()
# Call the function on Peak Performance Co.
print(retrieve_customer_info("____"))