3filHdZddlZddlmZmZmZmZmZmZm Z ddl Z ddl Z ddl m Z mZddlmZddlmZddlmZmZmZmZmZdZGd d eZGd d eZGd deZy)z3Wrapper 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 eZdZUdZeed<edZeed<edZ eed<edZ eed<edZ eed<ed Z e eed <ed Ze eed <ed Zeed<edZeed<edZdededdfdZdededdfdZdededdfdZdededdfdZdeedeefdZ d+ded e ed e ededeeeff dZ d,d eed e ed e ed!e eeefdeeeff d"Z d,ded e ed e ed!e eeefdef d#Zd$edefd%Z e!d&'e"d(edefd)Z# d-ded e ed e edeeeffd*Z$y).GoogleGeocodingAPIWrapperzWrapper for Google Maps Geocoding API. Provides methods for geocoding locations with configurable options for address components, geometry, metadata, and navigation points. google_api_keyTdefaultinclude_boundsinclude_navigationinclude_metadatainclude_address_componentsenlanguageusregion max_retriestimeoutforbid)extracleanedresultreturnNc|jsygd}|jdgD]X}|jdg}|jd}|D].}||vr ||d|<|dk(s|jd}||dd <0Zy) z%Add address components if configured.N) street_numberroutelocalitycountry postal_codeaddress_componentstypes long_nameaddressadministrative_area_level_1 short_namestate)rget) selfr'r( address_types componentr1nametype_ state_names b/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_google_community/geocoding.py_add_componentsz)GoogleGeocodingAPIWrapper._add_components+s..   $8"= =IMM'2.E==-D =M)04GI&u-;;!*|! _add_geometryz'GoogleGeocodingAPIWrapper._add_geometryBse::j"-   ||J'2::2F #J/||H%080B #H-& r@c|jrJ|jdi}|jd|jdg|jdd|d<yy)zAdd metadata if configured.rBplace_idr1 location_type)rHr1rImetadataN)rr7rEs r> _add_metadataz'GoogleGeocodingAPIWrapper._add_metadataKsR  zz*b1H"JJz2GR0"*,,"?#GJ  !r@c|jrL|jdr:|jdgDcgc]}|d|jdgdc}|d<yyycc}w)z$Add navigation points if configured.navigation_pointslocationrestricted_travel_modes)rN restrictions navigationN)rr7)r8r'r(points r>_add_navigationz)GoogleGeocodingAPIWrapper._add_navigationUsk  " "vzz2E'F $ZZ(;R@ % !&j 1$)II.G$L%GL !(G "%sAresultscBg}|D]}|sd|jddid|jdijdiid}|j|||j|||j|||j |||j ||S)zClean and format results.fullformatted_addressrNrB)r3rB)r7r?rFrKrSappend)r8rTcleaned_resultsr(r's r> clean_resultsz'GoogleGeocodingAPIWrapper.clean_results`s ,F#FJJ/BB$GH :r : > >z2 NG  & 1   w /   w /  & 1  " "7 +# ,&r@query max_resultsc |jsddgdS|jdDcgc]#}|js|j%}}t||kDr|d|}|j|||}|j ddk7r&|j dd|j d d gdSdt|j d g|j |j d g||xs |j |xs |jd d Scc}w#t$r:}dt|g||xs |j |xs |jd dcYd}~Sd}~wwxYw)a/Process geocoding request and return results. Handles both single and batch geocoding requests with detailed location data. Args: query: Location(s) to geocode (e.g., `'Eiffel Tower'` or `'Times Square, Central Park'`). language: Language code for results (e.g., `'en'`, `'fr'`, `'ja'`). region: Region bias (e.g., `'us'`, `'fr'`, `'jp'`). max_results: Maximum number of results to return. Returns: status (str): Request status (`'OK'` or error status). total_results (int): Number of locations found. results (list): Location data with address, geometry, metadata, and navigation. query_info (dict): Query metadata (original query, language, region). ERROREmpty query providedstatusmessagerT,Nr\rr rbOK error_messagezNo results foundrT)original_queryrr rb total_resultsrT query_info)rbrcrTrk) stripsplitlen raw_resultsr7r[rr Exceptionstr) r8r\rr r]qqueriesroes r>rTz!GoogleGeocodingAPIWrapper.resultsysj2- ;;=%5!+0++c*:HQaggiqwwyHGH7|k)!,;/**hv+Kx(D0)ooh@*@RS!!$[__Y%C!D--kooi.LM&+ ( 9DMM$3   I4 !q6&+ ( 9DMM$3    sADDDDA"D6ADD E#/EEE locations componentsc |sddgdSg}g}|D]q} |j||||}|jddk(r|j|n3|j||jd|jdds|rdndt ||j |D cgc])} | jd s| jd gd +c} |t |t |t ||xs |j|xs |jd d S#t$r)} |j|dt | dYd } ~ 5d } ~ wwxYwcc} w)aProcess multiple locations in batch. Args: locations: List of locations (e.g., `["Eiffel Tower", "Times Square"]`). language: Language code for results. region: Region bias. components: Component filters (e.g., `{"country": "US"}`). Returns: status (str): Overall batch status. total_results (int): Number of successful results. results (list): Location data for successful queries. errors (list): Errors encountered with query, status, and message. query_info (dict): Batch metadata (total, successful, failed counts). r_zNo locations providedra)r\rr rvrbrfrg)r\rbrcNrTr) total_queries successfulfailedrr )rbrjrTerrorsrk) ror7rYrprqrnr[rr ) r8rurr rvrTr{rNr(rtrs r> batch_geocodez'GoogleGeocodingAPIWrapper.batch_geocodesh,!2  ! YH Y))"%!) *::h'4/NN6*MM%-&,jj&:'-zz/'B Y.&d7 \))29NQQUU9=My"%a(N!$Y!'lf+$5  /DKK    Y GPSTUPVWXX YOs$A.D!E 8E  E  EE c |jsddgdS|j|jj|xs |j|xs |jd}|r)dj d|j D|d<t|jD]} tjt||j}|j|j}|jd d k(r|cS||jd z k(r5|jd |j|jd gdcS ddgdS#tj j"$r5} ||jd z k(rd d t%| gdcYd} ~ cSYd} ~ d} ~ wwxYw#t&$r} ddt%| gdcYd} ~ Sd} ~ wwxYw)z-Get raw results with improved error handling.r_r`rbrgrTr3keyrr |c30K|]\}}|d|yw):N).0kvs r> z8GoogleGeocodingAPIWrapper.raw_results.."s#0#'1aqc1#J0srvparamsr$rbrf REQUEST_ERRORzRequest failed: NzProcessing error: zAll retry attempts failed)rlrget_secret_valuerr joinitemsranger"requestsr7GOOGLE_MAPS_API_URLr$raise_for_statusjson_get_error_message exceptionsRequestExceptionrqrp) r8r\rr rvrattemptresponsedatarts r>roz%GoogleGeocodingAPIWrapper.raw_resultss6 ;;=%%;!!;;=**;;=$5  /DKK F'*xx0+5+;+;+=0(|$!!1!12 '||+FDLL H--/#==?Dxx)T1#  D$4$4q$88&*hhx&8-1-D-D $ 2.(*  9 F8  ! **;;$"2"2Q"66&5/?Ax-H') 7 !#5c!fX!>  smF(B F(%AEF(AEF( F(F%0#F F%F(F( F%%F(( G 1GG G rbc@dddddddd}|j|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) ZERO_RESULTSOVER_DAILY_LIMITOVER_QUERY_LIMITREQUEST_DENIEDINVALID_REQUESTMAX_ELEMENTS_EXCEEDED UNKNOWN_ERRORz API Error: )r7)r8rberror_messagess r>rz,GoogleGeocodingAPIWrapper._get_error_messageNs<> 8 6A;%D= !!&Kx*@AAr@before)modevaluesc*t|dd}||d<|S)z,Validate that api key exists in environment.rGOOGLE_MAPS_API_KEYr )clsrrs r>validate_environmentz.GoogleGeocodingAPIWrapper.validate_environment[s). $&; $2  r@cK |j|jj|xs|jxsd|xs|jxsdd}t j |j}t j4d{}|jt||4d{}|jd{}|jddk(r5|j|||cdddd{cdddd{S|jdd|jd d gd cdddd{cdddd{S7777d7U7 7#1d{7swYnxYwdddd{7y#1d{7swYyxYw#t$r} dd t| gd cYd} ~ Sd} ~ wwxYww) aNGeocode query asynchronously. Args: query: Location(s) to geocode. language: Language code for results. region: Region bias. Returns: status (str): Request status. results (list): Geocoding results. query_info (dict): Request metadata. rXr)totalNrrbrfr_rgzRequest failedrzAsync request failed: )rlrrrr aiohttp ClientTimeoutr$ ClientSessionr7rrrTrprq) r8r\rr r timeout_objsessionrrrts r> geocode_asyncz'GoogleGeocodingAPIWrapper.geocode_asynces$  ;;=**;;=$; ; 5DKK52 &F "//dllCK,,.  '";;' '  !)0Dxx)T1#||E8VD      #'((8W"=)-/CS)T#%       1             !#9#a&!B  s7GB F& EF&F.E/F2E*E *E*1 F=E">F F&E$F&G%E*9 FE&F F&E(F&GF&F E*"F$F&&F(F&*E< 0E31E< 8F? F& F  F&GF#F F#F&"G#F&& G /GG GG  G)NN )NNN)NN)%__name__ __module__ __qualname____doc__r__annotations__rrboolrrrrrrqr r"intr$r model_configrr?rFrKrSrr[rrTr}rorr classmethodrrrr@r>rrs!.ND.$T22"40d0',T'::$D1Hhsm1!$/FHSM/Q'K'$GS$H-L=t=T=d=.CTC4CDCT4D t T d T$ZDJ8#' $ FF3-F F  F c3h FV#' $/3 D 9D 3-D  D T#s(^, D c3h D R#' $/3 E E 3-E  E T#s(^, E  E N B B B(#$3$#' $ ,,3-, , c3h ,r@rc.eZdZUdZedZeed<y)GoogleGeocodeInputz'Input schema for `GoogleGeocodingTool`.zLocations for query.) descriptionr\N)rrrrrr\rqrrr@r>rrs1#9:E3:r@rc eZdZUdZdZeed<dZeed<eZ e e ed<dZ e ed<ed Zeed <ed Zeed <ed Zeed <ed Zeeed<ed Zeeed<edZeed<dZeded< ddedeedeeeeefeeefffdZ ddedee deeeeefeeefffdZ!y)GoogleGeocodingToolzTool for geocoding locations using Google Maps Geocoding API. Inherits from [`BaseTool`][langchain_core.tools.BaseTool]. Supports batch location lookups with detailed address and coordinate information. google_geocoder;zbA geocoding tool for multiple locations. Input: comma-separated locations. Returns: location data.r args_schemar]TrrrrrrNr cTtttjddS)NrrX)r)rrosgetenvrr@r>zGoogleGeocodingTool.s 9$RYY/Db%IJ! r@)default_factory api_wrappercontent_and_artifactresponse_formatr\ run_managerr)cL |jdDcgc]#}|js|j%}}t||jkDr|d|j}t|dkDr3|jj ||j |j}n|jj|d|j |j}|jdd|jj|jdgd|jdd k(rdnd|jdd k(rdndd d }|jdd k7rgd |jd difS|d|fScc}w#t$r}gd t|ifcYd}~Sd}~wwxYw)a"Geocode locations. Args: query: Comma-separated locations to geocode. run_manager: Callback manager. Returns: results: Location data with coordinates and addresses. response: Raw response with status and query info. rdNr)rurr rrerbrTrf)rxryrzrierrorrgGeocoding failed) rmrlrnr]rr}rr ror7r[rprq)r8r\rlocrurTrorts r>_runzGoogleGeocodingTool._runs )05 C0@PCIIKPIP9~ 0 00%&8(8(89 9~!**88'$-- 9#..::#A,t{{; *ooh7%&#//==# 26 *++6??8+D+LaRS'2x'@D'H!a# {{8$,GW[[BT%UVVV9%w. .9Q< )Q(( ( )s9FE<E<D7F5F<F F# FF#F#cJK |jj||j|jd{}|j ddk7rgd|j ddifS|j dg|fS7B#t $r}gdt |ifcYd}~Sd}~wwxYww) a7Geocode locations asynchronously. Args: query: Comma-separated locations to geocode. run_manager: Async callback manager. Returns: results: Location data with coordinates and addresses. response: Raw response with status and query info. reNrbrfrrgrrT)rrrr r7rprq)r8r\rr(rts r>_arunzGoogleGeocodingTool._aruns )++99dmmDKK:Fzz(#t+GVZZAS%TUUU::i,f4 4   )Q(( ( )sPB#5A>A<-A>'B#(A>;B#<A>> B BB B#B  B#)N)"rrrrr;rqrrrrr rr]rrrrrrrrr rrrrr rrrrrr rrr@r>rrsY!D#  " $6Ki5K .ND.$T22"40d0#D1Hhsm1!$/FHSM/-2 .K* 8NOW34M <@/)/)78/) tDcN#T#s(^3 4 /)hAE))<=) tDcN#T#s(^3 4 )r@r)rrtypingrrrrrrr rrlangchain_core.callbacksr r langchain_core.toolsr langchain_core.utilsrpydanticrrrrrrrrrrr@r>rs\9 BBB*5MMI~ ~B ;; j)(j)r@