Tool Calling with the SDK

In this guide we will demonstrate how to take advantage of OpenAI function calling in our Python SDK.

The Humanloop SDK provides an easy way for you to integrate the functionality of OpenAI function calling, which we refer to as JSON Schema tools, into your existing projects. Tools follow the same universal JSON Schema syntax definition as OpenAI function calling. In this guide, we’ll walk you through the process of using tools with the Humanloop SDK via the chat endpoint.


Creating a Tool

Prerequisites

  • A Humanloop account - you can create one by going to our sign up page.
  • Python installed - you can download and install Python by following the steps on the Python download page.
Using other model providers

This guide assumes you’re using OpenAI with the gpt-4 model. Only specific models from OpenAI are supported for function calling.

First you need to install and initialize the SDK. If you have already done this, skip to the next section.

Open up your terminal and follow these steps:

  1. Install the Humanloop SDK:
pip install humanloop
  1. Initialize the SDK with your Humanloop API key (you can get it from the Organization Settings page).
from humanloop import Humanloop
humanloop = Humanloop(api_key="<YOUR HUMANLOOP KEY>")
# Check that the authentication was successful
print(humanloop.prompts.list())

Install and initialize the SDK

The SDK requires Python 3.9 or greater.
1

Import the Humanloop SDK: If you haven’t done so already, you’ll need to install and import the Humanloop SDK into your Python environment. You can do this using pip:

pip install humanloop

Note, this guide was built with Humanloop==0.5.18.

Then import the SDK in your script:

from humanloop import Humanloop
2

Initialize the SDK: Initialize the Humanloop SDK with your API key:

from humanloop import Humanloop
hl = Humanloop(api_key="<YOUR_HUMANLOOP_API_KEY>")
3

Create a chat with the tool: We’ll start with the general chat endpoint format.

from humanloop import Humanloop
hl = Humanloop(api_key="<YOUR_HUMANLOOP_API_KEY>")
def run_conversation():
# Step 1: send the conversation and available functions to GPT
messages = [{"role": "user", "content": "What's the weather like in Boston?"}]
# TODO - Add tools definition here
response = hl.chat(
project="Assistant",
model_config={"model": "gpt-4", "max_tokens": 100},
messages=messages,
)
response = response.data[0]
4

Define the tool: Define a tool using the universal JSON Schema syntax syntax. Let’s assume we’ve defined a get_current_weather tool, which returns the current weather for a specified location. We’ll add it in via a "tools": tools, field. We’ve also defined a dummy get_current_weather method at the top. This can be replaced by your own function to fetch real values, for now we’re hardcoding it to return a random temperature and cloudy for this example.

from humanloop import Humanloop
import random
import json
hl = Humanloop(api_key="<YOUR_HUMANLOOP_API_KEY>")
def get_current_weather(location, unit):
# Your own function call logic
# We will return dummy values in this example
# Generate random temperature between 0 and 20
temperature = random.randint(0, 20)
return {"temperature": temperature, "other": "cloudy"}
def run_conversation():
# Step 1: send the conversation and available functions to GPT
messages = [
{
"role": "user",
"content": "What's the weather like in both Boston AND London tonight?",
}
]
tools = [
{
"name": "get_current_weather",
"description": "Get the current weather in a given location",
"parameters": {
"type": "object",
"properties": {
"location": {
"type": "string",
"description": "The city and state, e.g. San Francisco, CA",
},
"unit": {"type": "string", "enum": ["celsius", "fahrenheit"]},
},
"required": ["location"],
},
},
]
response = hl.chat(
project="Assistant",
model_config={"model": "gpt-3.5-turbo-1106", "tools": tools, "max_tokens": 100},
messages=messages,
)
response = response.body
output_message = response["data"][0]["output_message"]
# Remove the deprecated tool_call field (not nessecary for SDK rc verions >0.6)
del output_message["tool_call"]
# Add the output messge from the previous chat to the messages
messages.append(output_message)
# TODO - Add assistant response logic
5

Check assistant response

The code above will make the call to OpenAI with the tool but it does nothing to handle the assistant response. When responding with a tool response the response should have a tool_calls field. Fetch that value and pass it to your own function. An example of this can be seen below. Replace the TODO - Add assistant handling logic in your code from above with the following. Multiple tool calls can be returned with the latest OpenAI models gpt-4-1106-preview and gpt-3.5-turbo-1106, so below we loop through the tool_calls and populate the response accordingly.

# Step 2: check if GPT wanted to call a tool
if output_message.get("tool_calls"):
# Step 3: call the function
# Note: the JSON response may not always be valid; be sure to handle errors
available_functions = {
"get_current_weather": get_current_weather,
}
for tool_call in output_message["tool_calls"]:
function_name = tool_call["function"]["name"]
function_args = json.loads(tool_call["function"]["arguments"])
function_to_call = available_functions[function_name]
function_response = function_to_call(
location=function_args.get("location"),
unit=function_args.get("unit"),
# TODO - return the tool response back to OpenAI
6

Return the tool response

We can then return the tool response to OpenAI. This can be done by formatting OpenAI tool message into the relative assistant message seen below along with a tool message with the function name and function response.

# Step 2: check if GPT wanted to call a tool
if output_message.get("tool_calls"):
# Step 3: call the function
# Note: the JSON response may not always be valid; be sure to handle errors
available_functions = {
"get_current_weather": get_current_weather,
}
for tool_call in output_message["tool_calls"]:
function_name = tool_call["function"]["name"]
function_args = json.loads(tool_call["function"]["arguments"])
function_to_call = available_functions[function_name]
function_response = function_to_call(
location=function_args.get("location"),
unit=function_args.get("unit"),
)
# Step 4: send the response back to the model per function call
messages.append(
{
"role": "tool",
"content": json.dumps(function_response),
"tool_call_id": tool_call["id"],
}
)
second_response = hl.chat(
project="Assistant",
model_config={
"model": "gpt-3.5-turbo-1106",
"tools": tools,
"max_tokens": 500,
},
messages=messages,
)
return second_response
7

Review assistant response

The assistant should respond with a message that incorporates the parameters you provided, for example: The current weather in Boston is 22 degrees and cloudy. The above can be run by adding the python handling logic at the both of your file:

if __name__ == "__main__":
response = run_conversation()
response = response.data[0].output
# Print to console the response from OpenAI with the formatted message
print(response)

The full code from this example can be seen below:

from humanloop import Humanloop
import random
import json
hl = Humanloop(
api_key="<YOUR_HUMANLOOP_API_KEY>",
)
def get_current_weather(location, unit):
# Your own function call logic
# We will return dummy values in this example
# Generate random temperature between 0 and 20
temperature = random.randint(0, 20)
return {"temperature": temperature, "other": "cloudy"}
def run_conversation():
# Step 1: send the conversation and available functions to GPT
messages = [
{
"role": "user",
"content": "What's the weather like in both Boston AND London tonight?",
}
]
tools = [
{
"name": "get_current_weather",
"description": "Get the current weather in a given location",
"parameters": {
"type": "object",
"properties": {
"location": {
"type": "string",
"description": "The city and state, e.g. San Francisco, CA",
},
"unit": {"type": "string", "enum": ["celsius", "fahrenheit"]},
},
"required": ["location"],
},
},
]
response = hl.chat(
project="Assistant",
model_config={"model": "gpt-3.5-turbo-1106", "tools": tools, "max_tokens": 100},
messages=messages,
)
response = response.body
output_message = response["data"][0]["output_message"]
# Remove the deprecated tool_call field (not nessecary for SDK rc verions >0.6)
del output_message["tool_call"]
# Add the output messge from the previous chat to the messages
messages.append(output_message)
# Step 2: check if GPT wanted to call a tool
if output_message.get("tool_calls"):
# Step 3: call the function
# Note: the JSON response may not always be valid; be sure to handle errors
available_functions = {
"get_current_weather": get_current_weather,
}
for tool_call in output_message["tool_calls"]:
function_name = tool_call["function"]["name"]
function_args = json.loads(tool_call["function"]["arguments"])
function_to_call = available_functions[function_name]
function_response = function_to_call(
location=function_args.get("location"),
unit=function_args.get("unit"),
)
# Step 4: send the response back to the model per function call
messages.append(
{
"role": "tool",
"content": json.dumps(function_response),
"tool_call_id": tool_call["id"],
}
)
second_response = hl.chat(
project="Assistant",
model_config={
"model": "gpt-3.5-turbo-1106",
"tools": tools,
"max_tokens": 500,
},
messages=messages,
)
return second_response
if __name__ == "__main__":
response = run_conversation()
response = response.data[0]output
# Print to console the response from OpenAI with the formatted message
print(response)