uPj#hdZddlZddlZddlZddlmZddlmZddlm Z m Z ddl m Z GddZ y) z Router utilities for Search API integration. Handles search tool selection, load balancing, and fallback logic for search requests. N)Callable)partial)AnyFinal)verbose_router_loggerceZdZdZedeeefdeedzedzffdZ edede fdZ eded ede fd Z eded e fd Zeded ede fdZy)SearchAPIRouterz Static utility class for routing search API calls through the LiteLLM router. Provides methods for search tool selection, load balancing, and fallback handling. tool_litellm_paramsreturnNcN|jd}|jd}||fS)a Resolve search provider credentials from tool configuration ONLY. Credentials are stored only in search_tool.litellm_params, never in team/key metadata. This ensures secrets are not exposed in team/key API responses. Args: tool_litellm_params: Search tool litellm_params with credentials Returns: Tuple of (api_key, api_base) from tool configuration api_keyapi_base)get)r resolved_api_keyresolved_api_bases k/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/router_utils/search_api_router.py$_resolve_search_provider_credentialsz4SearchAPIRouter._resolve_search_provider_credentialss3"/B.E.Ei.P/B/F/Fz/R!222router_instance search_toolscK ddlm}tjdt |g}|D]W}|j d|j d|j di|j dd}|j |Y||_tjd t |y #t$r}tjd ||d }~wwxYww) a6 Update the router with search tools from the database. This method is called by a cron job to sync search tools from DB to router. Args: router_instance: The Router instance to update search_tools: List of search tool configurations from the database r)SearchToolTypedDictz Adding %s search tools to routersearch_tool_idsearch_tool_namelitellm_paramssearch_tool_info)rrrrz2Successfully updated router with %s search tool(s)z+Error updating router with search tools: %sN) litellm.types.routerrrdebuglenrappendrinfo Exception exception)rrrrouter_search_toolstoolrouter_search_tooles rupdate_router_search_toolsz*SearchAPIRouter.update_router_search_tools.s  @ ! ' '(JCP\L] ^02 $ ?'+hh/?&@(,1C(D&*hh/?&D(,1C(D ;" $**+=> ?,?O ( ! & &'[]`at]u v  ! + +,Y[\ ]G s)CB)B.-C. C7CCCrc|jDcgc]}|jd|k(s|}}|std|d|Scc}w)aV Get all search tools matching the given name. Args: router_instance: The Router instance search_tool_name: Name of the search tool to find Returns: List of matching search tool configurations Raises: ValueError: If no matching search tools are found rz Search tool 'z"' not found in router.search_tools)rr ValueError)rrr%matching_toolss rget_matching_search_toolsz)SearchAPIRouter.get_matching_search_toolsSs]&-99! TXXFX=Y]m=mD! ! }-=,>>`ab b! s AAoriginal_functionc :K |jd|jd}|s td||d<||d<ttj||d<|j ||d|j Dcgc]}|jd}}tjd ||||jdi|d {}|Scc}w7 #t$r>}d d l m }tj|||tj | |d }~wwxYww)a Helper function to make a search API call through the router with load balancing and fallbacks. Reuses the router's retry/fallback infrastructure. Args: router_instance: The Router instance original_function: The original litellm.asearch function **kwargs: Search parameters including search_tool_name, query, etc. Returns: SearchResponse from the search API rmodelz:search_tool_name or model parameter is required for searchoriginal_generic_function)rr-litellm_metadata)r/kwargsmetadata_variable_namezsInside SearchAPIRouter.async_search_with_fallbacks() - search_tool_name: %s, Available Search Tools: %s, kwargs: %sNr)send_llm_exception_alert)litellm_router_instancerequest_kwargserror_traceback_stroriginal_exception)rr*rr "async_search_with_fallbacks_helper_update_kwargs_before_fallbacksrrrasync_function_with_fallbacksr"!litellm.router_utils.handle_errorr4asyncio create_task traceback format_exc) rr-r2rr%available_search_tool_namesresponser'r4s rasync_search_with_fallbacksz+SearchAPIRouter.async_search_with_fallbacksns?$- &,jj1CVZZPWEX&Y # !]^^/F7O2CF. /*1BB /+F& '  ; ;&'9 <  \k[w[w1xSW$((;M2N1x '1x ! ' 'F +  %RO$Q$Q$[TZ$[[HO2y\  R   (,;#)(1(<(<(>'(  G sGDA8C<C /CCC D C D9DDDr/r0cK|} tj||}tj|}|j di}|j d}|st d|dtj |\} } tjd||d || | d|d {} | S7#t$r} tjd || | d } ~ wwxYww) a Helper function for search API calls - selects a search tool and calls the original function. Called by async_function_with_fallbacks for each retry attempt. Args: router_instance: The Router instance model: The search tool name (passed as model for compatibility) original_generic_function: The original litellm.asearch function **kwargs: Search parameters Returns: SearchResponse from the selected search provider )rrrsearch_providerz=search_provider not found in litellm_params for search tool '')r z&Selected search tool with provider: %s)rFr rNzFError in SearchAPIRouter.async_search_with_fallbacks_helper for %s: %sr9) r r,randomchoicerr*rrrr"error) rr/r0r2rr+ selected_toolrrFr rrCr's rr:z2SearchAPIRouter.async_search_with_fallbacks_helpers(#(% $3$M$M /!1%N%N$*==#@M%2$5$56F$KN%3%7%78I%JO" #`aq`rrs!tuu / T T$2!U! GX " ' '(PRa b%>% /!% %HO  ! ' 'XZjlm G  s;CBB-$B+%B-*C+B-- C6CCC)__name__ __module__ __qualname____doc__ staticmethoddictstrrtuplerlistr(r,rrDr:r9rrr r s 3!#s(^3 sTz3:% &33*"#"T""H 4>>#>>@:::$,::rr )rOr>rHr@collections.abcr functoolsrtypingrrlitellm._loggingrr r9rrrYs-  $2YYr