uPjUdZddlmZmZmZddlmZddlZddlm Z ddl m Z m Z mZmZmZmZddlmZmZmZddlmZd Zeed <d Zeed <d Zeed<GddeZy)aN Calls APISerpent's search endpoints to search Google, Bing, Yahoo, or DuckDuckGo. Two endpoints under one provider, selected via the ``deep`` boolean param: - ``deep=False`` (default) -> quick search (/api/search/quick) - ``deep=True`` -> deep search (/api/search) APISerpent API Reference: https://apiserpent.com/docs )FinalLiteralcast) urlencodeN)Logging)DEEP_SEARCH_PATHNUM_MAXNUM_MIN NUM_MIN_DEEPQUICK_SEARCH_PATHAPISerpentSearchParams)BaseSearchConfigSearchResponse SearchResult)get_secret_strdeepDEEP_SEARCH_PARAMzhttps://apiserpent.comAPISERPENT_BASE_apiserpent_paramsAPISERPENT_PARAMS_KEYc eZdZedefdZdedfdZedede fdZ dded edzd edzdefd Z dd edzded ee ezdzdefd Z dee ezdedefdZedede edefdZdej"dedzdefdZy)APISerpentSearchConfigreturncy)N APISerpentrr/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/apiserpent/search/transformation.pyui_friendly_namez'APISerpentSearchConfig.ui_friendly_name&sr)GETPOSTcy)Nr r)selfs rget_http_methodz&APISerpentSearchConfig.get_http_method*sroptional_paramsc>t|jtSN)boolgetr)r%s r_is_deep_searchz&APISerpentSearchConfig._is_deep_search-sO''(9:;;rNheadersapi_keyapi_basec h|j||ddt}|s td||d<d|d<|S)N)APISERPENT_API_KEYAPISERPENT_API_BASE)caller_api_keycaller_api_base key_env_vars base_env_vardefault_api_basezMAPISERPENT_API_KEY is not set. Set `APISERPENT_API_KEY` environment variable.z X-API-Keyzapplication/jsonz Content-Type)resolve_server_api_keyr ValueError)r#r+r,r-kwargss rvalidate_environmentz+APISerpentSearchConfig.validate_environment1sR--"$0., . lm m& "4rdatac $|xstdxstjd}|j|rtnt }|j |s||}|r3t|tr#t|vrt|td}|d|S|S)a Build the search URL. APISerpent uses GET, so the transformed request is serialized into the query string. The endpoint path (quick vs deep) is always applied; an ``api_base`` / ``APISERPENT_API_BASE`` override only changes the host. The ``endswith`` guard keeps this idempotent, since the handler re-invokes this method with the already-resolved URL as api_base. r0/T)doseq?) rrrstripr*rr endswith isinstancedictrr)r#r-r%r:r8basepath query_strings rget_complete_urlz'APISerpentSearchConfig.get_complete_urlEsTN+@AT_\\]`a*.*>*>*O&Uf}}T"VD6?D JtT*/D/L"+D1F,Gt"TLV1\N+ + rqueryc nt|trdj|}|j|}i}d|vr.|rtnt }t |t|dt|d<d|vr$tt|dj|d<|jD]&\}}|tjvs||vs"|||<(itdi|jd|i} d|vrD|d} t| tr/t!| dkDr!|j#t| d| | d<t$| iS) a Transform a unified search request into APISerpent query params. Unified spec mappings: - query -> q - max_results -> num (clamped to the endpoint's valid range) - country -> country (lowercased) - search_domain_filter -> site: clauses appended to q All other APISerpent params (engine, language, freshness, safe, pages, format, pixel_position) pass through, defaulting via APISerpentSearchParams.  max_resultsnumcountryqsearch_domain_filterrr)rAlistjoinr*r r maxminr rstrloweritemsr field_namesto_request_paramslen_append_domain_filtersr) r#rGr%r8is_deep overridesnum_minparamvalueparamsdomainss rtransform_search_requestz/APISerpentSearchConfig.transform_search_request^s?$ eT "HHUOE--o>!# O +-4\'G"7C 0NPW,XYIe   '#'_Y-G#H#N#N#PIi +113 )LE5.::<<iAW#( %  )`1>I>PPR_TWY^_ !_ 4,-CDG'4(S\A-="99#fSk:JGTs %v..rr`cDdjd|D}d|d|dS)Nz OR c3&K|] }d| yw)zsite:Nr).0domains r z@APISerpentSearchConfig._append_domain_filters..s+SeF8,<+Ss(z) ())rP)rGr`domain_clausess rrYz-APISerpentSearchConfig._append_domain_filterss, & +S7+S S5'^,A..r raw_response logging_objc ||j}|jdxsi}t|tr|jdgn|}g}|D]`}|j t |jdd|jdd|jdd|jdd bt |d S) z Transform APISerpent response to the unified SearchResponse format. Full format nests results under ``results.organic[]``; simple format returns a flat ``results[]`` array. Both expose title/url/snippet. resultsorganictitleurlsnippetdateN)rorqrrrs last_updatedsearch)rmobject)jsonr)rArBappendrr) r#rjrkr8 response_json raw_resultsrnrmresults rtransform_search_responsez0APISerpentSearchConfig.transform_search_responses ,002 *..y9?R ;EkSW;XB7^i-/ F NN **Wb1 5"-"JJy"5F+!%     r)NNr')__name__ __module__ __qualname__ staticmethodrSrrr$rBr(r*r9rOrFrarYhttpxResponseLiteLLMLoggingObjrr|rrrrr%sLc!7<<$<< ## t*   0*. *T$Z$&  2)/T#Y)/)/  )/V/c/DI/#//  nn  '-    rr)__doc__typingrrr urllib.parserr*litellm.litellm_core_utils.litellm_loggingrr'litellm.llms.apiserpent.search.defaultsrr r r r r +litellm.llms.base_llm.search.transformationrrrlitellm.secret_managers.mainrr__annotations__rrrrrrrsc('" S 8!5!113u3I -I r