uPj*tUdZddlmZddlmZddlmZddlmZm Z ddl Z ddl m Z m Z mZmZddlmZdd lmZmZmZdd lmZerdd lmZd Zd ed<Gdde ZGdde ZGdde ZGdde Z ee!e"dfZ#d ed<eiZ$ded<d!dZ%GddeZ&d"dZ'd#dZ(d$d Z)y)%z Calls Nimble's /v2/search endpoint to search the web. Nimble API Reference: https://docs.nimbleway.com/api-reference/search/search ) annotations)Mapping)MappingProxyType) TYPE_CHECKINGFinalN) BaseModel ConfigDict TypeAdapterValidationError)BaseLLMException)BaseSearchConfigSearchResponse SearchResult)get_secret_str)Loggingz6https://docs.nimbleway.com/api-reference/search/searchr_NIMBLE_DOCS_URLcleZdZUdZeddZdZded<dZded<dZ ded <dZ ded <dZ d ed <y) _NimbleResultzOne entry of Nimble's `results` array. Every field is optional so a single degraded result degrades to empty strings instead of failing the whole call.ignoreTextrafrozenN str | Nonetitleurlcontent descriptionobjectadditional_data) __name__ __module__ __qualname____doc__r model_configr__annotations__rrrrn/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/nimble/search/transformation.pyrrsHKHT:LE:CGZ"K""OV"r'rc0eZdZUdZeddZded<y)_NimbleSearchResponsez&Nimble's /v2/search response envelope.rTrztuple[_NimbleResult, ...]resultsN)r r!r"r#r r$r%r&r'r(r*r*,s0HT:L'&r'r*c4eZdZUdZeddZdZded<y)_AdditionalDatazPThe slice of a result's free-form `additional_data` that maps onto SearchResult.rTrNr publish_date)r r!r"r#r r$r.r%r&r'r(r-r-6sZHT:L#L*#r'r-cBeZdZUdZeddZdZded<dZded<y) _ErrorEnvelopezNimble reports errors as either `{"detail": ...}` (validation) or `{"success": "false", "task_id": ..., "message": ...}` (collection).rTrNrdetailmessage) r r!r"r#r r$r1r%r2r&r'r(r0r0>s*LHT:LFJGZr'r0._DomainListAdapterzFinal[Mapping[str, object]]_NOTHINGc,| t||iStS)zRA one-entry mapping to spread into a payload, or nothing when the value is absent.)rr4)keyvalues r( _optionalr8Ms-2-> S%L )LHLr'ceZdZdZed dZ d d dZ d d dZ ddZ ddZ ddZ y)NimbleSearchConfigzhttps://sdk.nimbleway.com/v2cy)NNimbler&r&r'r(ui_friendly_namez#NimbleSearchConfig.ui_friendly_nameUsr'Nc v|j||dd|j}|s tdi|d|dddS) z Validate environment and return headers. Returns a new dict rather than mutating ``headers``: the http handler calls this a second time after ``litellm/search/main.py`` already did, so it has to be idempotent. )NIMBLE_API_KEYNIMBLE_API_BASE)caller_api_keycaller_api_base key_env_vars base_env_vardefault_api_basezENIMBLE_API_KEY is not set. Set `NIMBLE_API_KEY` environment variable.zBearer zapplication/jsonlitellm) Authorizationz Content-TypezX-Client-Source)resolve_server_api_keyr@ ValueError)selfheadersapi_keyapi_basekwargsresolved_api_keys r(validate_environmentz'NimbleSearchConfig.validate_environmentYsi#'"="="$,*!11 #>#  de e  &'7&89.(   r'c |xstdxs |jjd}|jdr|S|dS)Nr@/z/search)rr@rstripendswith)rJrMoptional_paramsdatarN resolved_bases r(get_complete_urlz#NimbleSearchConfig.get_complete_urlwsM!) eN;L,M eQUQeQemmnqr  ! !) , ((r'c |j}|jd}t|jDcic] \}}||vs ||c}}}it |jd|dt |t rdj|n|itd|jdtdt |tr|jSdScc}}w)a Transform Search request to Nimble API format. Nimble already uses the Perplexity unified spec's names, so this is close to a pass-through: - query -> query (a list is joined with spaces; Nimble takes a single string) - max_results -> max_results (sent unclamped so Nimble's own 1-100 validation reports the error) - country -> country, upper-cased to the ISO form Nimble documents - search_domain_filter -> include_domains, with `-`-prefixed entries going to exclude_domains - max_tokens_per_page -> dropped (no Nimble equivalent) Everything else is forwarded as-is, so the rest of Nimble's surface stays reachable without LiteLLM tracking it. countrysearch_domain_filterquery max_resultsN) (get_supported_perplexity_optional_paramsgetritems_domain_filters isinstancelistjoinr8strupper) rJr\rUrNunified_paramsrZparamr7 passthroughs r(transform_search_requestz+NimbleSearchConfig.transform_search_requests&!% M M O(,,Y7..=.C.C.E eleUVdIdUE\ e  o112HIJ    5$(?SXXe_U  ':':='IJ   j#6N7==?Y   UYY    fs C# C# c  tj|j}t|jDcgc]r}td |jxsd|jxsd|jxs|jxsdt|jddt!d|jtc}dS#t$r9}|j d||j t |jd}~wwxYwcc}w) a Transform Nimble API response to LiteLLM unified SearchResponse format. `date` carries only the absolute `publish_date`. News results often carry a relative `publish_date_raw` ("1 day ago") instead, which is not a date, so the whole `additional_data` object rides through as an extra on `SearchResult` and nothing is lost. Nimble ranks results itself via metadata.position, so the order is preserved as received. A body that does not match the documented schema raises an attributed error rather than being reported as a successful empty search. Parsing the response bytes rather than `.json()` covers the non-JSON case through that same path. z:response does not match the documented /v2/search schema: ) error_message status_coderKN)rrsnippetdate last_updatedrsearch)r+rr&)r*model_validate_jsonrr get_error_classrndictrKrr+rrrr _publish_daterr8)rJ raw_response logging_objrNparsederesults r(transform_search_responsez,NimbleSearchConfig.transform_search_responses$ 1EElFZFZ[F%nn  ,,," (b"NNFf.@.@FB&v'='=>!%   163I3IJ     && Z[\Z]^(44\112'   sB4A7C94 C6=4C11C6cht|jd}t|d|dtd|S)Nz. zNimble Search: z. See z for details.)rnr2rK)_unwrap_error_detailrSr r)rJrmrnrKr1s r(ruz"NimbleSearchConfig.get_error_classs@ -];BB4H#%fXV4D3E]S  r')returnrf)NN) rKdict[str, str]rLrrMrrNrrrN) rMrrUdict[str, object]rVz2dict[str, object] | list[dict[str, object]] | NonerNrrrf)r\zstr | list[str]rUrrNrrr)rxzhttpx.ResponseryLiteLLMLoggingObjrNrrr)rmrfrnintrKrr Exception) r r!r"r@ staticmethodr=rPrXrkr}rur&r'r(r:r:Rs4O ##          DDH ) )+ )A )  ) )" " +"  "  " H( $( '(  (  ( T          r'r:c tj|}|jxs|jxs|S#t$r|cYSwxYw)z Surface the human-readable message inside Nimble's error envelopes. Falls back to the raw body for anything else (CDN HTML pages, plain text, other shapes). )r0rtr r1r2)rmbodys r(rrsH $88G ;; 7$,, 7-7 s5 AAc tj|}t dt d|Dfdt d|DffDcic] \}}|r|| c}}S#t$r tcYSwxYwcc}}w)a= Split the unified `search_domain_filter` into Nimble's include/exclude lists. Follows the Perplexity unified spec, where a `-` prefix means "exclude this domain". Anything that is not a list of strings is ignored rather than raising, since it only ever narrows a search that is otherwise valid. include_domainsc3JK|]}|s|jdr|yw)-N) startswith.0ds r( z"_domain_filters..s)\!,,WZJ[!)\s ###exclude_domainsc3hK|]*}|jdst|dkDs$|dd,yw)rN)rlenrs r(rz"_domain_filters..s1)eAALLQTDUZ]^_Z`cdZd!AB%)es 22 2)r3validate_pythonr r4rtuple)r[domainsr6r7s r(rbrbs+;;KKK s ! --)r6rfr7rrMapping[str, object])rmrfrrf)r[rrr)rrrr)*r# __future__rcollections.abcrtypesrtypingrrhttpxpydanticrr r r )litellm.llms.base_llm.chat.transformationr +litellm.llms.base_llm.search.transformationr rrlitellm.secret_managers.mainr*litellm.litellm_core_utils.litellm_loggingrrrr%rr*r-r0rrfr3r4r8r:rrbrwr&r'r(rs ##"' HHF 8WR%R #I #'I'$i$Y(c3h8E8(8(< %<M J )J Z 80r'