Parallel Tool Calls with OpenAI

import asyncio
import json
from openai import AsyncOpenAI

client = AsyncOpenAI()

tools = [
    {'type':'function','function':{'name':'get_weather','description':'Get weather for a city','parameters':{'type':'object','properties':{'city':{'type':'string'}},'required':['city']}}},
    {'type':'function','function':{'name':'get_time','description':'Get current time in timezone','parameters':{'type':'object','properties':{'timezone':{'type':'string'}},'required':['timezone']}}},
]

async def execute_tool(name: str, args: dict) -> str:
    if name == 'get_weather':
        return f'Weather in {args["city"]}: 22°C, sunny'
    elif name == 'get_time':
        return f'Time in {args["timezone"]}: 14:30'
    return 'Unknown tool'

async def run_with_parallel_tools(question: str) -> str:
    resp = await client.chat.completions.create(model='gpt-4o-mini', messages=[{'role':'user','content':question}], tools=tools, tool_choice='auto')
    msg  = resp.choices[0].message
    if not msg.tool_calls:
        return msg.content
    results = await asyncio.gather(*[execute_tool(tc.function.name, json.loads(tc.function.arguments)) for tc in msg.tool_calls])
    tool_msgs = [{'role':'tool','tool_call_id':tc.id,'content':r} for tc, r in zip(msg.tool_calls, results)]
    final = await client.chat.completions.create(model='gpt-4o-mini', messages=[{'role':'user','content':question}, msg, *tool_msgs])
    return final.choices[0].message.content

print(asyncio.run(run_with_parallel_tools('What is the weather in Paris and the time in UTC?')))