o ưik#@sLddlZddlmZmZmZmZmZddlmZddl m Z ddl Tddl m Z mZddlmZeZejdeeged gd ejd eeged gd ejd eeged gd ejd eeged gd eedfdedede deefdd Zejdeeged gd ejdeeged gd eefdedede fddZdS)N) APIRouterDepends HTTPExceptionRequestResponse)ORJSONResponseverbose_proxy_logger)*)UserAPIKeyAuthuser_api_key_auth)ProxyBaseLLMRequestProcessingz/v1/search/{search_tool_name}search) dependenciesresponse_classtagsz/search/{search_tool_name}z /v1/searchz/searchrequestfastapi_responseuser_api_key_dictsearch_tool_namec sddlm}m}m}m}m}m} m} m} m } m } m }| IdH}t |}|dur/||d<d|vr|dr|d|d<|durt|dr|dtddd d |jDd t|jfd d |jD}|r|d}|d id}|r||d<d|vri|d<|dd<t|d}z:|jd$id|d|d|ddd|d|d|d|d|ddd| d| d| d | d!| d"|IdHWSty}z |j||||d#IdHd}~ww)%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 search_toolsz0Search endpoint - Looking for search_tool_name: z$. Available search tools in router: cSsg|]}|dqSrget.0toolr)_/home/app/Keep/.python/lib/python3.10/site-packages/litellm/proxy/search_endpoints/endpoints.py szsearch..z. Total search tools: csg|] }|dkr|qSr#r$r&Zsearch_tool_name_valuer)r*r+s litellm_paramssearch_providerZcustom_llm_providermetadataZ model_group)datarrrZ route_typeZasearchrrrrrrrrrrr )errr r))litellm.proxy.proxy_serverrrrrrrrrrrr bodyorjsonloadshasattrr debugr"lenr%r Zbase_process_llm_request ExceptionZ_handle_llm_api_exception)rrrrrrrrrrrrrrr r3r0Zmatching_toolsZ search_toolr. processorr1r)r,r*rs4b             z/v1/search/toolsz /search/toolsc sddlm}z@g}|durBt|drB|jD]+}|d|didd}d |vr<|d r<|d d }|r<||d <||qd |d WStyi}zdd lm} | d|t dt |dd}~ww)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"rr-r.)rr.Zsearch_tool_info descriptionlist)objectr0rzError listing search tools: i) status_codedetail) r2rr6r"r%appendr9litellm._loggingr exceptionrstr) rrrrZsearch_tools_listr(Z tool_infor;r1r r)r)r*list_search_toolss. +   rD)r4ZfastapirrrrrZfastapi.responsesrrAr Zlitellm.proxy._typesZ$litellm.proxy.auth.user_api_key_authr r Z'litellm.proxy.common_request_processingr ZrouterpostOptionalrCrr%rDr)r)r)r*s