tPjU+`UddlmZddlZddlmZmZmZmZmZddl m Z ddl m Z ddl ddlmZmZddlmZeZeed <ej+d eege d g ej+d eege d g ej+deege d g ej+deege d g eedfdededededzfdZej1deege d g ej1deege d g eefdededefdZy))FinalN) APIRouterDepends HTTPExceptionRequestResponse)ORJSONResponseverbose_proxy_logger)*)UserAPIKeyAuthuser_api_key_auth)ProxyBaseLLMRequestProcessingrouterz/v1/search/{search_tool_name}search) dependenciesresponse_classtagsz/search/{search_tool_name}z /v1/searchz/searchrequestfastapi_responseuser_api_key_dictsearch_tool_namecfKddlm}m}m}m}m}m} m} m} m } m } m }|jd{}tj|}|||d<d|vrO|drI|d|d<|d}ddlm}m}m} |||d{|j&rEddlm}m}m}||j&|||j,| d{}||| d{|t5|d rt1j6d ||j8Dcgc]}|j;dc}t=|j8|j8Dcgc]}|j;d|k(s|}}|r>|d}|j;dij;d}|r||d<d|vri|d<||dd<d|vst?|j;dt@si|d<tC|dd|jD|dd<tC|dd|j&|dd<tG|} |jI|||d|||||d| | | | | |d{S7777x#t.$r}t1j2d ||d}~wwxYwcc}wcc}w7C#t.$r#}|jK||||d{7d}~wwxYww)a\ Search endpoint for performing web searches. Follows the Perplexity Search API spec: https://docs.perplexity.ai/api-reference/search-post The search_tool_name can be passed either: 1. In the URL path: /v1/search/{search_tool_name} 2. In the request body: {"search_tool_name": "..."} Example with search_tool_name in URL (recommended - keeps body Perplexity-compatible): ```bash curl -X POST "http://localhost:4000/v1/search/litellm-search" -H "Authorization: Bearer sk-1234" -H "Content-Type: application/json" -d '{ "query": "latest AI developments 2024", "max_results": 5, "search_domain_filter": ["arxiv.org", "nature.com"], "country": "US" }' ``` Example with search_tool_name in body: ```bash curl -X POST "http://localhost:4000/v1/search" -H "Authorization: Bearer sk-1234" -H "Content-Type: application/json" -d '{ "search_tool_name": "litellm-search", "query": "latest AI developments 2024", "max_results": 5, "search_domain_filter": ["arxiv.org", "nature.com"], "country": "US" }' ``` Request Body Parameters (when search_tool_name not in URL): - search_tool_name (str, required if not in URL): Name of the search tool configured in router - query (str or list[str], required): Search query - max_results (int, optional): Maximum number of results (1-20), default 10 - search_domain_filter (list[str], optional): List of domains to filter (max 20) - max_tokens_per_page (int, optional): Max tokens per page, default 1024 - country (str, optional): Country code filter (e.g., 'US', 'GB', 'DE') When using URL path parameter, only Perplexity-compatible parameters are needed in body: - query (str or list[str], required): Search query - max_results (int, optional): Maximum number of results (1-20), default 10 - search_domain_filter (list[str], optional): List of domains to filter (max 20) - max_tokens_per_page (int, optional): Max tokens per page, default 1024 - country (str, optional): Country code filter (e.g., 'US', 'GB', 'DE') Response follows Perplexity Search API format: ```json { "object": "search", "results": [ { "title": "Result title", "url": "https://example.com", "snippet": "Result snippet...", "date": "2024-01-01", "last_updated": "2024-01-01" } ] } ``` r) general_settings llm_router proxy_configproxy_logging_objselect_data_generator user_api_baseuser_max_tokens user_modeluser_request_timeoutuser_temperatureversionNrmodel)can_key_call_search_toolcan_team_call_search_toolget_team_object)r valid_token) prisma_clientruser_api_key_cache)team_idr*r+parent_otel_spanr)r team_objectz+Search tool authorization failed for %s: %s search_toolszpSearch endpoint - Looking for search_tool_name: %s. Available search tools in router: %s. Total search tools: %slitellm_paramssearch_providercustom_llm_providermetadata model_group team_metadatauser_api_key_team_metadatar,user_api_key_team_id)dataasearch)rrr route_typerrrrrr%r!r#r"r rr$)errr$)&litellm.proxy.proxy_serverrrrrrrr r!r"r#r$bodyorjsonloadslitellm.proxy.auth.auth_checksr&r'r(r,r*r+r- Exceptionr errorhasattrdebugr/getlen isinstancedictgetattrr5rbase_process_llm_request_handle_llm_api_exception)rrrrrrrrrrr r!r"r#r$r=r8search_tool_name_valuer&r'r(r*r+r.r;toolmatching_tools search_toolr1 processors m/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/search_endpoints/endpoints.pyrrs<D     &D,,t$D##3  T!d+=&>/0W (,-?(@   *!7-   !(( ,;-55"/'9%6%G%G&7 ,& 0%; +  !gj.&I & &C&:D:Q:QR$,-RJ++,  ",!8!8%DHHEW3D3L3LZ/05$?I 77-/ /!-%"7!-!5+'!8   i'( &  & &'TVlno p  S%8 $  77// 8     s.J1IAJ14 I I>I II I I -J1I6 )J1I;I;#B4J1$J<J=JJ1I I I I3I..I33 J1J J. J)#J&$J))J..J1z/v1/search/toolsz /search/toolscKddlm} g}|t|dry|jD]j}|j d|j dij dd}d |vr |d r|d j d }|r||d <|j |ld |d S#t $r4}dd lm} | jd|tdt|d}~wwxYww)ar List all available search tools configured in the router. This endpoint returns the search tools that are currently loaded and available for use with the /v1/search endpoint. Example: ```bash curl -X GET "http://localhost:4000/v1/search/tools" -H "Authorization: Bearer sk-1234" ``` Response: ```json { "object": "list", "data": [ { "search_tool_name": "litellm-search", "search_provider": "perplexity", "description": "Perplexity search tool" } ] } ``` r)rNr/rr0r1)rr1search_tool_info descriptionlist)objectr8r zError listing search tools: %si) status_codedetail) r<rrCr/rEappendrAlitellm._loggingr exceptionrstr) rrrrsearch_tools_listrM tool_inforTr;r s rQlist_search_toolsr_sV6<#%  !gj.&I"// 4(,1C(D'+xx0@"'E'I'IJ['\ &-$7I2J"&'9":">">}"MK"3> -0!((3 4!*;<< <9&&&'GKCF;; fastapirrrrrfastapi.responsesr rZr litellm.proxy._types$litellm.proxy.auth.user_api_key_authr r'litellm.proxy.common_request_processingrr__annotations__postr\rrEr_rQrjs HH,1"RQ #+,-!     +,-!    +,-!     +,-!   )00A(B#' E E E &E Dj E    0E P+,-!    +,-!   )00A(B8< 8<8<&8<  8