Tool Calling

1
from openai import OpenAI
2
import json
3
4
client = OpenAI(base_url="http://localhost:8081/v1", api_key="dummy")
5
6
def get_weather(location: str, unit: str):
7
    return f"Getting the weather for {location} in {unit}..."
8
tool_functions = {"get_weather": get_weather}
9
10
tools = [{
11
    "type": "function",
12
    "function": {
13
        "name": "get_weather",
14
        "description": "Get the current weather in a given location",
15
        "parameters": {
16
            "type": "object",
17
            "properties": {
18
                "location": {"type": "string", "description": "City and state, e.g., 'San Francisco, CA'"},
19
                "unit": {"type": "string", "enum": ["celsius", "fahrenheit"]}
20
            },
21
            "required": ["location", "unit"]
22
        }
23
    }
24
}]
25
26
response = client.chat.completions.create(
27
    model="openai/gpt-oss-20b",
28
    messages=[{"role": "user", "content": "What's the weather like in San Francisco in Celsius?"}],
29
    tools=tools,
30
    tool_choice="auto",
31
    max_tokens=10000
32
)
33
print(f"{response}")
34
tool_call = response.choices[0].message.tool_calls[0].function
35
print(f"Function called: {tool_call.name}")
36
print(f"Arguments: {tool_call.arguments}")
37
print(f"Result: {tool_functions[tool_call.name](**json.loads(tool_call.arguments))}")

1
# Use tools defined in example 1
2
3
time_tool = {
4
    "type": "function",
5
    "function": {
6
        "name": "get_current_time_nyc",
7
        "description": "Get the current time in NYC.",
8
        "parameters": {}
9
    }
10
}
11
12
13
tools.append(time_tool)
14
15
messages = [
16
    {"role": "user", "content": "What's the current time in New York?"}
17
]
18
19
20
response = client.chat.completions.create(
21
    model="openai/gpt-oss-20b", #client.models.list().data[1].id,
22
    messages=messages,
23
    tools=tools,
24
    tool_choice="auto",
25
    max_tokens=100,
26
)
27
print(f"{response}")
28
tool_call = response.choices[0].message.tool_calls[0].function
29
print(f"Function called: {tool_call.name}")
30
print(f"Arguments: {tool_call.arguments}")

1
tools = [
2
    {
3
        "type": "function",
4
        "function": {
5
            "name": "get_tourist_attractions",
6
            "description": "Get a list of top tourist attractions for a given city.",
7
            "parameters": {
8
                "type": "object",
9
                "properties": {
10
                    "city": {
11
                        "type": "string",
12
                        "description": "The name of the city to find attractions for.",
13
                    }
14
                },
15
                "required": ["city"],
16
            },
17
        },
18
    },
19
]
20
21
def get_messages():
22
    return [
23
        {
24
            "role": "user",
25
            "content": (
26
                "I'm planning a trip to Tokyo next week. what are some top tourist attractions in Tokyo? "
27
            ),
28
        },
29
    ]
30
31
32
messages = get_messages()
33
34
response = client.chat.completions.create(
35
    model="openai/gpt-oss-20b",
36
    messages=messages,
37
    tools=tools,
38
    tool_choice="auto",
39
    max_tokens=100,
40
)
41
print(f"{response}")
42
tool_call = response.choices[0].message.tool_calls[0].function
43
print(f"Function called: {tool_call.name}")
44
print(f"Arguments: {tool_call.arguments}")