uPj,!rdZddlmZmZmZddlZddlmZddlm Z ddl m Z m Z mZddlmZGdd e Zy) z Calls DataForSEO SERP API to search the web. DataForSEO API Reference: https://docs.dataforseo.com/v3/serp/google/organic/live/advanced/?bash )AnyFinalLiteralN) DEFAULT_DATAFORSEO_LOCATION_CODE)Logging)BaseSearchConfigSearchResponse SearchResult)get_secret_strc eZdZdZdZedefdZdedfdZ dde d edzd edzde fd Z dd edzd e d e e e zdzdefdZ ddee ezd e d edzde e e zfdZdej dedefdZy)DataForSEOSearchConfigz Configuration for DataForSEO SERP API search. DataForSEO uses HTTP Basic Auth with login:password credentials. API endpoint: https://api.dataforseo.com/v3/serp/google/organic/live/advanced z?https://api.dataforseo.com/v3/serp/google/organic/live/advancedreturncy)N DataForSEOrr/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/dataforseo/search/transformation.pyui_friendly_namez'DataForSEOSearchConfig.ui_friendly_namesr)GETPOSTcy)z? DataForSEO uses POST requests with JSON body. rr)selfs rget_http_methodz&DataForSEOSearchConfig.get_http_method#srNheadersapi_keyapi_basec ddl}td}td}t|xrd|v}|rd|vr|jdd\}}|s"|r |r|j ||j dd|s t d|s t d |d|} |j| jj} d | |d <d |d <|S)a" Validate DataForSEO environment and set up authentication. DataForSEO uses HTTP Basic Auth with login:password format. The credentials should be in DATAFORSEO_LOGIN and DATAFORSEO_PASSWORD env vars, or passed as api_key in "login:password" format. rNDATAFORSEO_LOGINDATAFORSEO_PASSWORD:DATAFORSEO_API_BASEzrDATAFORSEO_LOGIN is not set. Set `DATAFORSEO_LOGIN` environment variable or pass credentials in api_key parameter.zxDATAFORSEO_PASSWORD is not set. Set `DATAFORSEO_PASSWORD` environment variable or pass credentials in api_key parameter.zBasic Authorizationzapplication/jsonz Content-Type) base64r boolsplit._assert_trusted_api_base_for_server_credentialr" ValueError b64encodeencodedecode) rrrrkwargsr$loginpasswordcaller_supplied_credentials credentialsencoded_credentialss rvalidate_environmentz+DataForSEOSearchConfig.validate_environment)s 12!"78.2'2LcWn-M# sg~%mmC3OE8*u  ? ?((%"  E K  !&waz2 %+%5%5k6H6H6J%K%R%R%T%+,?+@#A "4roptional_paramsdatac <|xstdxs |jS)z Get complete URL for DataForSEO SERP API endpoint. DataForSEO uses POST requests, so no query parameters in URL. r")r r")rrr3r4r,s rget_complete_urlz'DataForSEOSearchConfig.get_complete_url\s \>*?@\DD\D\\rqueryc ~i}t|tr |r|dnd}||d<d|vr"|drtt|dd}||d<d|vr |dr|d|d<d |vr |d r|d |d <d |vr d |vrd |d <d|vr d|vr t|d<|j D]"\}}||j vs||vs|||<$|gS)u Transform Search request to DataForSEO SERP API format. Args: query: Search query (string or list of strings). DataForSEO supports single string queries. optional_params: Optional parameters for the request - max_results: Maximum number of search results → maps to `depth` (max 700) - country: Country name → maps to `location_name` - search_domain_filter: Domain to filter results → maps to `domain` - Plus any DataForSEO-specific parameters (location_code, language_code, device, os, etc.) api_key: DataForSEO credentials (login:password format) Returns: List[Dict]: Request body for DataForSEO API (array of task objects as required by API) rkeyword max_resultsidepthcountry location_namesearch_domain_filterdomain language_code language_nameen location_code) isinstancelistminintritems(get_supported_perplexity_optional_params) rr7r3rr,taskr<paramvalues rtransform_search_requestz/DataForSEOSearchConfig.transform_search_requestjs.') eT " %E!H2E Y O + 0Ns?=#ABCHE!DM  'OI,F%4I$>D ! !_ 4I_9`,-CDDN $ &?$+F$(D ! $ &?$+F$DD !,113 $LE5DIIKKPU]aPa#U  $ v r raw_response logging_objc |j}g}d|vrt|ddkDr|dd}|jddk(rd|vrt|ddkDr|dd}|jdgD]h}|jddk(st|jd d |jd d |jd d d d } |j | jt |dS)um Transform DataForSEO SERP API response to LiteLLM unified SearchResponse format. DataForSEO → LiteLLM mappings: - tasks[0].result[*].items[*].title → SearchResult.title - tasks[0].result[*].items[*].url → SearchResult.url - tasks[0].result[*].items[*].description → SearchResult.snippet - No date/last_updated fields in standard response (set to None) Args: raw_response: Raw httpx response from DataForSEO API logging_obj: Logging object for tracking Returns: SearchResponse with standardized format tasksr status_codei NresultrItypeorganictitler9url descriptionN)rWrXsnippetdate last_updatedsearch)resultsobject)jsonlengetr appendr ) rrOrPr, response_jsonr^rKrTitem search_results rtransform_search_responsez0DataForSEOSearchConfig.transform_search_responses, ,002  m #M',B(Ca(G'03Dxx &%/H4DtH~&*$(N1$5F!' 7B 7 :88F+y8,8&*hhw&;$(HHUB$7(,(C%)-1 -M$NN=9 :  r)NN)N)__name__ __module__ __qualname____doc__r" staticmethodstrrrrdictr2rFr6rNhttpxResponseLiteLLMLoggingObjr rgrrrr r s(\c!7## 11t1* 1  1n*. ]* ] ]T$Z$& ] ]$# =T#Y==t = T  =~5 nn5 '5  5 rr )rktypingrrrrolitellm.constantsr*litellm.litellm_core_utils.litellm_loggingrrq+litellm.llms.base_llm.search.transformationrr r litellm.secret_managers.mainr r rrrrws9 '& >S 8I -I r