o $i @sUdZddlZddlmZddlmZd(dededefd d Zd)d ed edefddZ dedefddZ ed dddddddddddgdededddddddd d dd!d gde dedd"dddd#didgde dd$Z eeefe d%<deeeffd&d'Z dS)*z0 Custom example tools for the OpenRouter Agent. N)Dict)ToolUScitycountryreturnc Cszdddddddddd d dd d d dd}||vr7||}d|d|d|dd|dd|dd Sd|dS)z?Simulated weather function - in production, connect to real APIHSunnyA)temp conditionhumidity<CloudyPKRainyZUClearF)zNew YorkLondonTokyoSydneyz Weather in z, z: r u°F, r z , Humidity: r %zWeather data not available for z*. Try: New York, London, Tokyo, or Sydney.)rr weather_datadatarr7/var/www/apps/myagent/mysuperagent/core/tools/custom.py get_weather s    0 rquery num_resultscCsTd|dd|dd|dd|dd |d g}d |d d |d|S)z;Simulated web search - in production, connect to search APIz1. z - Recent news articlez2. z - Wikipedia entryz3. z - Research paperz4. z - Blog postz5. z - Official documentationzSearch results for 'z':  N)join)r!r"resultsrrr search_webs     r&rcCs*z t|}tj|ddWS|YS)Format data as JSON string)indent)jsonloadsdumps)rparsedrrr format_json(s  r.z"Get weather information for a cityobjectstringz The city name)type descriptionzThe country code (optional))r1r2default)rr)r1 propertiesrequired)namer2 parametersfunctionzSearch the web for informationzThe search queryintegerzNumber of results to return)r!r"r'zData to format as JSON)rr&r. CUSTOM_TOOLScCsttS)z,Return a copy of the custom tools dictionary)dictr:rrrrget_custom_toolslsr<)r)r )__doc__r*typingr core.modelsrstrrintr&r.r:__annotations__r<rrrrsl       ;