o ZhU@sdZddlZddlmZmZmZmZmZmZm Z ddl Z ddl Z ddl m Z mZddlmZddlmZddlmZmZmZmZmZdZGd d d eZGd d d eZGd ddeZdS)z4Wrapper and Tool for the Google Maps Geocoding API. N)AnyDictListLiteralOptionalTupleType)AsyncCallbackManagerForToolRunCallbackManagerForToolRun)BaseToolget_from_dict_or_env) BaseModel ConfigDictField SecretStrmodel_validatorz1https://maps.googleapis.com/maps/api/geocode/jsonc @sFeZdZUdZeed<eddZeed<eddZ eed<eddZ eed<eddZ eed<ed dZ e eed <ed dZe eed <ed dZeed<eddZeed<eddZdededdfddZdededdfddZdededdfddZdededdfddZdeedeefd d!Z   "d7d#ed e ed e ed$edeeeff d%d&Z   d8d'eed e ed e ed(e eeefdeeeff d)d*Z   d8d#ed e ed e ed(e eeefdef d+d,Zd-edefd.d/Z e!d0d1e"d2edefd3d4Z#  d9d#ed e ed e edeeeffd5d6Z$dS):GoogleGeocodingAPIWrapperz&Wrapper for Google Maps Geocoding API.google_api_keyTdefaultinclude_boundsinclude_navigationinclude_metadatainclude_address_componentsenlanguageusregion max_retriestimeoutZforbid)extracleanedresultreturnNc Cs||jsdSgd}|dgD],}|dg}|d}|D]}||vr+||d|<q|dkr:|d}||dd <qqdS) z%Add address components if configured.N)Z street_numberZrouteZlocalityZcountryZ postal_codeZaddress_componentstypesZ long_nameaddressZadministrative_area_level_1Z short_namestate)rget) selfr$r%Z address_types componentr'nametype_Z state_namer/[/var/www/html/lang_env/lib/python3.10/site-packages/langchain_google_community/geocoding.py_add_components's    z)GoogleGeocodingAPIWrapper._add_componentscCsR|di}|jr%|dr|d|dd<|dr'|d|dd<dSdSdS)z#Add geometry details if configured.geometryZviewportZboundsN)r*rr+r$r%r2r/r/r0 _add_geometry>s   z'GoogleGeocodingAPIWrapper._add_geometrycCs>|jr|di}|d|dg|dd|d<dSdS)zAdd metadata if configured.r2place_idr' location_type)r5r'r6metadataN)rr*r3r/r/r0 _add_metadataGs  z'GoogleGeocodingAPIWrapper._add_metadatacCs6|jr|drdd|dgD|d<dSdSdS)z$Add navigation points if configured.Znavigation_pointscSs"g|] }|d|dgdqS)locationZrestricted_travel_modes)r9Z restrictionsr*).0pointr/r/r0 Ts  z=GoogleGeocodingAPIWrapper._add_navigation..Z navigationN)rr*)r+r$r%r/r/r0_add_navigationQs  z)GoogleGeocodingAPIWrapper._add_navigationresultscCs|g}|D]7}|s qd|ddid|didiid}||||||||||||||q|S)zClean and format results.fullZformatted_addressr9r2)r(r2)r*r1r4r8r>append)r+r?Zcleaned_resultsr%r$r/r/r0 clean_results\s     z'GoogleGeocodingAPIWrapper.clean_results query max_resultsc Csz^|s ddgdWSdd|dD}t||kr"|d|}|j|||d}|d d kr@|d d|d d gdWSd t|d g||d g||pU|j|pY|jddWSty}zdt |g||pp|j|pt|jddWYd}~Sd}~ww)a* Process geocoding request and return comprehensive results. This method handles both single and batch geocoding requests, returning detailed location information with optional components. Args: query: Location(s) to geocode. Examples: - "Eiffel Tower" - "Times Square, Central Park" language: Optional language code for results (e.g., "en", "fr", "ja") region: Optional region bias (e.g., "us", "fr", "jp") max_results: Maximum number of results to return (default: 10) Returns: Dict containing: status: Status of the request ("OK" or error status) total_results: Number of locations found results: List of dictionaries containing location data: address: { full: Complete formatted address street_number: Building number (if available) route: Street name locality: City/Town state: State/Province country: Country postal_code: Postal/ZIP code } geometry: { location: {lat, lng} coordinates viewport: Recommended viewport bounds: Geographic bounds (if available) } metadata: { place_id: Unique Google place identifier types: Categories (e.g., ["establishment", "point_of_interest"]) location_type:(e.g., "ROOFTOP", "GEOMETRIC_CENTER") } navigation: List of navigation points with: location: {latitude, longitude} restrictions: Travel mode restrictions query_info: { original_query: Input query language: Language used region: Region bias used } Example Response: { "status": "OK", "total_results": 2, "results": [ { "address": { "full": "Street, Country..", "route": "Avenue Gustave Eiffel", "locality": "Paris", "country": "France" }, "geometry": { "location": {"lat": 48.8584, "lng": 2.2945} } }, ... ], "query_info": { "original_query": "Eiffel Tower, Big Ben", "language": "en", "region": "us" } } Raises: ValueError: If query is empty or invalid Exception: For API errors or connection issues ERROREmpty query providedstatusmessager?cSg|] }|r|qSr/strip)r;qr/r/r0r=z5GoogleGeocodingAPIWrapper.results..,NrErrrJOK error_messagezNo results foundr?)Zoriginal_queryrrrJ total_resultsr? query_info)rJrKr?rW) rNsplitlen raw_resultsr*rCrr Exceptionstr)r+rErrrFZqueriesrZer/r/r0r?usHS     z!GoogleGeocodingAPIWrapper.results locations componentsc Cs|sddgdSg}g}|D]F}z'|j||||d}|ddkr'||n|||d|ddWqtyT} z||dt| dWYd } ~ qd } ~ ww|rYdndt||d d |D|t|t|t||pt|j|px|jd d S)uProcess multiple locations in a single structured request. Efficiently handles multiple location queries, processing them as a batch while maintaining individual result integrity. Args: locations: List of location strings to geocode Examples: ["Eiffel Tower", "Times Square", "東京スカイツリー"] language: Optional language code for results region: Optional region bias components: Optional filters (e.g., {"country": "US"}) Returns: Dict containing: status: Overall batch status total_results: Number of successful geocoding results results: List of location data (same structure as single results) errors: List of any errors encountered: query: The location query that failed status: Error status code message: Detailed error message query_info: { total_queries: Total locations processed successful: Number of successful queries failed: Number of failed queries language: Language used region: Region bias used } Example: batch_geocode( locations=["Eiffel Tower", "Big Ben"], language="en", components={"country": "FR"} ) rGzNo locations providedrI)rErrr_rJrSrT)rErJrKNcSs&g|]}|dr|dgdqS)r?rr:)r;rr/r/r0r=Fs&z;GoogleGeocodingAPIWrapper.batch_geocode..) total_queries successfulfailedrr)rJrVr?errorsrW) rZr*rBr[r\rYrCrr) r+r^rrr_r?rdr9r%r]r/r/r0 batch_geocodesR+ "  z'GoogleGeocodingAPIWrapper.batch_geocodec Csz|s ddgdWS||j|p|j|p|jd}|r.ddd|D|d<t|jD]k}z:t j t ||j d }| |}| d d krT|WWS||jd krn| d || d gdWWSWq3t jjy} z!||jd krd dt| gdWYd} ~ WSWYd} ~ q3d} ~ wwWnty} zddt| gdWYd} ~ Sd} ~ wwddgdS)z-Get raw results with improved error handling.rGrHrJrTr?r(keyrr|css"|] \}}|d|VqdS):Nr/)r;kvr/r/r0 ms z8GoogleGeocodingAPIWrapper.raw_results..r_paramsr"rJrSZ REQUEST_ERRORzRequest failed: NzProcessing error: zAll retry attempts failed)rNrget_secret_valuerrjoinitemsranger requestsr*GOOGLE_MAPS_API_URLr"raise_for_statusjson_get_error_message exceptionsRequestExceptionr\r[) r+rErrr_roattemptresponsedatar]r/r/r0rZRsj       z%GoogleGeocodingAPIWrapper.raw_resultsrJcCs&dddddddd}||d |S) z0Get detailed error message based on status code.zNo results found for this queryzAPI key quota exceededzQuery limit exceededz!Request was denied, check API keyzInvalid request parameterszToo many locations in requestzServer error, please try again)Z ZERO_RESULTSZOVER_DAILY_LIMITZOVER_QUERY_LIMITZREQUEST_DENIEDZINVALID_REQUESTZMAX_ELEMENTS_EXCEEDED UNKNOWN_ERRORz API Error: r:)r+rJZerror_messagesr/r/r0rys z,GoogleGeocodingAPIWrapper._get_error_messagebefore)modevaluescCst|dd}||d<|S)z,Validate that api key exists in environment.rGOOGLE_MAPS_API_KEYr )clsrrr/r/r0validate_environments z.GoogleGeocodingAPIWrapper.validate_environmentc sz||j|p|jpd|p|jpdd}tj|jd}t4IdHr}|j t ||d4IdHN}| IdH}| ddkr_| |||WdIdHWdIdHWS| dd| d d gd WdIdHWdIdHWS1IdHswYWdIdHWdS1IdHswYWdSt y} zdd t| gd WYd} ~ Sd} ~ ww) aRun query through Google Maps Geocoding API asynchronously. Args: query: The location(s) to geocode language: Optional language code for results region: Optional region bias Returns: Dict containing: status: Status of the request results: List of geocoding results query_info: Metadata about the request rArg)totalNrnrJrSrGrTzRequest failedrfzAsync request failed: )rNrrqrraiohttpZ ClientTimeoutr"Z ClientSessionr*rvrxr?r[r\) r+rErrro timeout_objsessionr}r~r]r/r/r0 geocode_asyncsB     2  z'GoogleGeocodingAPIWrapper.geocode_async)NNrD)NNN)NN)%__name__ __module__ __qualname____doc__r__annotations__rrboolrrrrrr\rr intr"rZ model_configrr1r4r8r>rrCrr?rerZryr classmethodrrr/r/r/r0rs         ^ G   rc@s$eZdZUdZeddZeed<dS)GoogleGeocodeInputzInput for the Geocoding tool.zLocations for query.) descriptionrEN)rrrrrrEr\rr/r/r/r0rs rc @sBeZdZUdZdZeed<dZeed<eZ e e ed<dZ e ed<ed d Zeed <ed d Zeed <ed d Zeed <edd Zeeed<edd Zeeed<edddZeed<dZeded< ddedeedeeeeefeeefffddZ ddedee deeeeefeeefffddZ!dS) GoogleGeocodingToolaTool that queries the Google Maps Geocoding API for batch location lookups. Instantiate: .. code-block:: python from tools.geocoding_wrapper import GoogleGeocodingTool tool = GoogleGeocodingTool( max_results=5, include_bounds=True, include_navigation=True, include_metadata=True, language="en" ) Invoke directly: .. code-block:: python result = tool.invoke({ "query": "Eiffel Tower, Empire State Building" }) Invoke with agent: .. code-block:: python agent.invoke({ "input": "Find coordinates of Times Square and Central Park" }) Returns: Tuple containing: - List of location data with coordinates and addresses - Raw response data with query information Zgoogle_geocoder-zbA geocoding tool for multiple locations. Input: comma-separated locations. Returns: location data.r args_schemarFTrrrrrrNrcCstttdddS)NrrA)r)rrosgetenvr/r/r/r0szGoogleGeocodingTool.)default_factory api_wrapperZcontent_and_artifactresponse_formatrE run_managerr&c Cs&zxdd|dD}t||jkr|d|j}t|dkr+|jj||j|jd}n5|jj|d|j|jd}|d d|j |d gd|d d krPdnd|d d krZdndd d }|d d krrgd|ddifWS|d |fWSt y}zgdt |ifWYd}~Sd}~ww)z Use the tool.cSrLr/rM)r;locr/r/r0r=(rPz,GoogleGeocodingTool._run..rQNrp)r^rrrrRrJr?rS)rarbrcrUerrorrTGeocoding failed) rXrYrFrrerrrZr*rCr[r\)r+rErr^r?rZr]r/r/r0_run!s8    zGoogleGeocodingTool._runc sz)|jj||j|jdIdH}|ddkr"gd|ddifWS|dg|fWStyD}zgdt|ifWYd}~Sd}~ww) zUse the tool asynchronously.rRNrJrSrrTrr?)rrrrr*r[r\)r+rErr%r]r/r/r0_arunIs  zGoogleGeocodingTool._arun)N)"rrrrr-r\rrrrrrrFrrrrrrrrrrrrrr rrrrrr rr/r/r/r0rs@  #   +r)rrtypingrrrrrrrrruZlangchain_core.callbacksr r Zlangchain_core.toolsr Zlangchain_core.utilsr Zpydanticrrrrrrvrrrr/r/r/r0s $  P