o Ti &@sddlmZmZddlmZddlmZddlmZeGdddZ ddZ e d kr@ee Z e dZe ejejejd Sd S) ) dataclassfield)Dataset)HfArgumentParser)get_json_schemac@sZeZdZUdZedddidZeed<edddidZe ed <ed dd idZ e ed <d S)ScriptArgumentsa Arguments for the script. Args: test_size (`float`, *optional*, defaults to `0.1`): Fraction of the dataset to include in the test split. push_to_hub (`bool`, *optional*, defaults to `False`): Whether to push the dataset to the Hugging Face Hub. repo_id (`str`, *optional*, defaults to `"trl-internal-testing/zen"`): Hugging Face repository ID to push the dataset to. g?helpz5Fraction of the dataset to include in the test split.)defaultmetadata test_sizeFz4Whether to push the dataset to the Hugging Face Hub. push_to_hubztrl-internal-testing/toolcallz2Hugging Face repository ID to push the dataset to.repo_idN) __name__ __module__ __qualname____doc__rr float__annotations__r boolr strrrU/home/ubuntu/.local/lib/python3.10/site-packages/scripts/generate_toolcall_dataset.pyrs  rc Cs dtdtfdd}dtdtfdd}dtdtfdd }d td tdtfd d }dtdtdtfdd}dtdtdtfdd}dtdtdtfdd} dtdtdtttffdd} t|}t|}t|}t|}t|}t|}t| } t| } tdddd d!d"dd#id$d%gd&d'd"d(d)d d*dgdd+dd d!d,dd-id$d%gd&d'd,d.d)d d/dgdd0dd d!d1dd2id$d%gd&d'd1d3d)d d4dgdd5dd d!d6d7d8d9d$d%gd&d'd6d:d)d d;dgddd?d@d$d%gd&d'd=dAd)d dBdgddCdd d!dDdEdFdGd$d%gd&d'dDdHd)d dIdgddJdd d!dKdLdMdNd$d%gd&d'dKdOd)d dPdgddQdd d!dRdSdTdUd$d%gd&d'dRdVd)d dWdgg|| g|g||| g||g|| g|g|| g|| ggdX} | j|dYdZ} |r| |dSdS)[NdurationreturncSs|S)z Starts a timer for the specified duration in seconds. Args: duration: Duration in seconds to set the timer for. Returns: The duration set for the timer. r)rrrr start_timer4 zmain..start_timerlocationcSdS)z Returns the current time in the specified location. Args: location: The location for which to get the current time. Returns: The current time in the specified location. 06:22:48rrrrrget_current_time@rzmain..get_current_timecSr)z Returns the air quality index for the specified location. Args: location: The location for which to get the air quality index. Returns: The air quality index for the specified location. 5rrrrrget_air_quality_indexLrz#main..get_air_quality_indextitleartistcSsddiS)a Plays music by the specified title and artist. Args: title: The title of the music to play. artist: The artist of the music to play. Returns: A dictionary indicating the status of the music playback. statusPlayingrr#r$rrr play_musicX zmain..play_musiccitydatecSs dddS)a? Returns the weather forecast for the specified city and date. Args: city: The city for which to get the weather forecast. date: The date for which to get the weather forecast. Returns: A dictionary containing the temperature and weather condition. z partly cloudy) temperature conditionrr*r+rrrget_weather_forecastes z"main..get_weather_forecastroomstatecSsd|iS)a Controls the light in the specified room. Args: room: The room where the light should be controlled. state: The desired state of the light ("on" or "off"). Returns: A dictionary indicating the state of the light. r2rr1r2rrr control_lightrr)zmain..control_lighttimenotecSr)a Creates a reminder for the specified time and note. Args: time: The time for the reminder. note: The note for the reminder. Returns: A confirmation message indicating that the reminder has been set. z$I'll remind you to call mom at 7 PM.rr5r6rrrcreate_reminder zmain..create_reminderunitcSr)a0 Returns the wind conditions for the specified city. Args: city: The city for which to get the wind conditions. unit: The unit of measurement for the wind speed (e.g., "mph"). Returns: A tuple containing the wind speed and direction. )NWrr*r:rrrget_wind_conditionsr9z!main..get_wind_conditionsuserzSet a timer for 10 minutes.)rolecontent assistantfunctionriX)name arguments)typerC)r@ tool_callstool600)r@rDrAzTimer set for 10 minutes.zWhat time is it in Tokyo?r Tokyorz&The current time in Tokyo is 06:22 AM.z!Is the air clean today in Lisbon?get_air_qualityzLisbon, Portugal53zThe air quality is moderate.zPlay some music.r(z Take Fivez Dave Brubeckr'z{'status': 'Playing'}zEnjoy the jazz tunes!z+What's the weather like tomorrow in Berlin?r0Berlinz 2025-06-16r/z1{'temperature': 22, 'condition': 'partly cloudy'}u>Tomorrow in Berlin will be partly cloudy with a high of 22°C.zTurn on the living room lights.r4z living roomonr3z{'state': 'on'}z"The living room lights are now on.zRemind me to call mom at 7 PM.r8z19:00zCall momr7z Reminder setu,Okay, I’ll remind you to call mom at 7 PM.z,How strong is the wind in Chicago right now?r>Chicagomphr=z (14, 'NW')zdatasetrrrmain2s                3>rZ__main__N) dataclassesrrdatasetsr transformersrtransformers.utilsrrrZrparserparse_args_into_dataclasses script_argsr r r rrrrs   6