Defining a function for tool use
You're working for a SaaS (software as a service) company with big goals for rolling out tools to help employees at all levels of the organization to make data-informed decisions. You're creating a PoC for an application that allows customer success managers to interface with company data using natural language to retrieve important customer data.
You've been provided with a pandas DataFrame called customers that contains a small sample of customer data. Your first step in this project is to define a Python function to extract information from this table given a customer's name. pandas has already been imported as pd.
This exercise is part of the course
Developing LLM Applications with LangChain
Exercise instructions
- Define a
retrieve_customer_info()function that takes a string argument,name. - Filter the
customersDataFrame to return rows with"name"equal to the input argument,name. - Call the function on the customer name,
"Peak Performance Co.".
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
# 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("____"))