tPjE\ UdZddlmZmZddlmZddlmZmZmZddl m Z m Z m Z ddl mZddlmZddlmZdd lmZmZmZdd lmZdd lmZdd lmZmZmZdd lm Z e Z!ee"d<eZ#ee"d<dee$zdzde$dzfdZ%ee$egeefZ&ee"d<d4dZ'de$dedefdZ(dede$dzfdZ)e(fde*edede&de*efdZ+e!jYddge egee efdefd Z-Gd!d"eZ.e!j_d#dge eg$d%e.fd&Z0Gd'd(eZ1e!jed)dge eg$d*e$d%e1fd+Z3e!jid)dge eg$d*e$fd,Z5e!jYd)dge eg$d*e$fd-Z6Gd.d/eZ7e!j_d0dge eg$d%e7fd1Z8e!jYd2dge eg$d3Z9y)5z! CRUD ENDPOINTS FOR SEARCH TOOLS ) AwaitableCallable)datetime)AnyFinal TypeAlias) APIRouterDepends HTTPException) BaseModel)verbose_proxy_logger)UI_SESSION_TOKEN_TEAM_ID)LiteLLM_TeamTableLitellmUserRolesUserAPIKeyAuth)user_api_key_auth)SearchToolRegistry)ListSearchToolsResponse SearchToolSearchToolInfoResponse)SearchProvidersrouterSEARCH_TOOL_REGISTRYvalueNreturncL|yt|tr|jS|S)z Convert datetime object to ISO format string. Args: value: datetime object, string, or None Returns: ISO format string or original value if already string or None N) isinstancer isoformat)rs z/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/search_endpoints/search_tool_management.py_convert_datetime_to_strr "s( }%"  LTeamObjectLookupcKddlm} |jd{y7#t$r }t j d|Yd}~yd}~wwxYww)zPush the search tools table into this worker's router. Best-effort: the row is already committed, so a refresh failure must not surface as a 500 and push the caller into a retry that creates duplicates. r) proxy_configNz>Search tool router refresh failed after a management write: %s)litellm.proxy.proxy_serverr$reload_search_tools_from_db Exceptionr exception)r$es r_refresh_router_search_toolsr*6sI 8l66888 l&&'gijkkls6A%#%A% AA A AAteam_iduser_api_key_dictclKddlm}ddlm}m}m}|||||j |d{S7w)Nr)get_team_object) prisma_clientproxy_logging_objuser_api_key_cache)r+r/r1parent_otel_spanr0)litellm.proxy.auth.auth_checksr.r%r/r0r1r2)r+r,r.r/r0r1s r_team_object_from_dbr4DsA> !#-*;;+   s +424c6|j}|r |tk(ry|S)a The team whose object_permission allowlist scopes this caller, or None when there is none. Every Admin UI session key is stamped with UI_SESSION_TOKEN_TEAM_ID, a reserved sentinel that never has a row in LiteLLM_TeamTable (`/team/new` rejects it as a real team id), so looking it up would raise 404 instead of resolving a team. It carries no allowlist of its own, so the caller is scoped by its key-level allowlist alone. Any other team id is looked up for real and a failed lookup still surfaces. N)r+r)r,r+s r_allowlist_team_idr6Us"'..G g!99 Nr! search_toolslookup_team_objectcBK|jtjtjfvr|Sddlm}t |}|r|||d{nd}g}|D];}|jd}|s||||d{s+|j|=|S7J7w)z Drop search tools the caller is not authorized to invoke, applying the same key/team object_permission allowlists enforced on /search. Admins see all tools. r)can_user_view_search_toolNsearch_tool_name)r; valid_token team_object) user_roler PROXY_ADMINPROXY_ADMIN_VIEW_ONLYr3r:r6getappend) r7r,r8r:allowlist_team_idr=visibletool tool_names r_filter_visible_search_toolsrGes""$$..'H12CDJ[ !24EFFFae46G!HH/0 8&)#   NN4 ! N G  s0ABB B2BBBBBz/search_tools/listz Search Tools)tags dependenciesresponse_modelc Kddlm}ddlm}m}| t dd t j|d{}|Dchc]}|jd }}g}g} |jd{} |j| } | r| }|D]{} | jd } | st| jd i}||d d }| jd}|jt!d| ||r t|ndddd}|Dcgc]}|jd |vs|}}|D]}t|jd i}||d d }|jt!|jd|jd d||jdt#|jdt#|jddt%||d{}t'|S7cc}w7#t$r!} tjd | Yd} ~ d} ~ wwxYwcc}w7M#t $rt$r1} tj(d| t dt+| d} ~ wwxYww)a List all search tools that are available in the database and config file. Example Request: ```bash curl -X GET "http://localhost:4000/search_tools/list" -H "Authorization: Bearer " ``` Example Response: ```json { "search_tools": [ { "search_tool_id": "123e4567-e89b-12d3-a456-426614174000", "search_tool_name": "litellm-search", "litellm_params": { "search_provider": "perplexity", "api_key": "sk-***", "api_base": "https://api.perplexity.ai" }, "search_tool_info": { "description": "Perplexity search tool" }, "created_at": "2023-11-09T12:34:56.789Z", "updated_at": "2023-11-09T12:34:56.789Z", "is_from_config": false }, { "search_tool_name": "config-search-tool", "litellm_params": { "search_provider": "tavily", "api_key": "tvly-***" }, "is_from_config": true } ] } ``` r_get_masked_values)r/r$NPrisma client not initialized status_codedetailr/r;z-Could not get config-defined search tools: %slitellm_paramsunmasked_lengthnumber_of_asteriskssearch_tool_infoTsearch_tool_idr;rTrY created_at updated_atis_from_configr[r\r]F)r7zError getting search tools: %s)*litellm.litellm_core_utils.litellm_loggingrMr%r/r$r rget_all_search_tools_from_dbrA get_configparse_search_toolsr'r debugdictrBrr rGrr(str)r,rMr/r$search_tools_from_dbrE db_tool_namessearch_tool_configsconfig_search_toolsconfig parsed_toolsr)config_search_toolrFlitellm_params_dictmasked_litellm_params_dictconfig_tool_infodb_search_toolvisible_search_toolss rlist_search_toolsrss`NF4STTG<%9%V%Ver%V%ssI]^); <^ ^<>  ["."9"9";;F"."A"A&"IL&2##6  *../ABI&*+=+A+ABRTV+W&X#-?'$%().* $6#9#9:L#M #***'+)2'ADT$/?*@Z^#'#''+  01 DHH=O4PXe4eD  3 N"&~'9'9:JB'O"P );# !$%* & & &#5#56F#G%3%7%78JB%O#=%3%7%78J%K78J8J<8XY78J8J<8XY#(  (-II\^o,p&p&4HIIC t^< [ & &'VXY Z Z [4 0'q  <&&'GKCF;;H=HH="H 5H6H H=(A+H=H6,H60B=H=-H;.H==J>H=H=H H3H.(H=.H33 H==J,I;;JJceZdZUeed<y)CreateSearchToolRequest search_toolN__name__ __module__ __qualname__r__annotations__r!rrurur!ruz /search_tools)rHrIrequestc|Kddlm}| tdd tj |j |d{}t d{tjd|jd |S7=7-#t$r1}tjd |tdt|d}~wwxYww) aD Create a new search tool. Example Request: ```bash curl -X POST "http://localhost:4000/search_tools" \ -H "Authorization: Bearer " \ -H "Content-Type: application/json" \ -d '{ "search_tool": { "search_tool_name": "litellm-search", "litellm_params": { "search_provider": "perplexity", "api_key": "sk-..." }, "search_tool_info": { "description": "Perplexity search tool" } } }' ``` Example Response: ```json { "search_tool_id": "123e4567-e89b-12d3-a456-426614174000", "search_tool_name": "litellm-search", "litellm_params": { "search_provider": "perplexity", "api_key": "sk-..." }, "search_tool_info": { "description": "Perplexity search tool" }, "created_at": "2023-11-09T12:34:56.789Z", "updated_at": "2023-11-09T12:34:56.789Z" } ``` rrSNrNrOrP)rvr/z0Successfully added search tool '%s' to database.r;z"Error adding search tool to db: %s) r%r/r radd_search_tool_to_dbrvr*r rdrAr'r(rf)r~r/resultr)s rcreate_search_toolr sZ94STT<2HH++=I  +,,,"" > JJ) *    - <&&'KQOCF;;" \ -H "Content-Type: application/json" \ -d '{ "search_tool": { "search_tool_name": "updated-search", "litellm_params": { "search_provider": "perplexity", "api_key": "sk-new-key" }, "search_tool_info": { "description": "Updated search tool" } } }' ``` Example Response: ```json { "search_tool_id": "123e4567-e89b-12d3-a456-426614174000", "search_tool_name": "updated-search", "litellm_params": { "search_provider": "perplexity", "api_key": "sk-new-key" }, "search_tool_info": { "description": "Updated search tool" }, "created_at": "2023-11-09T12:34:56.789Z", "updated_at": "2023-11-09T13:45:12.345Z" } ``` rrSNrNrOrPr[r/Search tool with ID not found)r[rvr/z2Successfully updated search tool '%s' in database.r;zError updating search tool: %s)r%r/r rget_search_tool_by_id_from_dbupdate_search_tool_in_dbrvr*r rdrAr'r(rf)r[r~r/ existing_toolrr)s rupdate_search_toolrSsZ94STT<%9%W%W)&X&    -n-=ZH  3KK)++'L   +,,,"" @ JJ) *  /  -  <&&'GKCF;;" ``` Example Response: ```json { "message": "Search tool 123e4567-e89b-12d3-a456-426614174000 deleted successfully", "search_tool_name": "litellm-search" } ``` rrSNrNrOrPrrrrz/Successfully deleted search tool from database.zError deleting search tool: %s) r%r/r rrdelete_search_tool_from_dbr*r rdr'r(rf)r[r/rrr)s rdelete_search_toolrs.94STT<%9%W%W)&X&    -n-=ZH  3MM)N  +,,,""#TU %   -  <&&'GKCF;;" ``` Example Response: ```json { "search_tool_id": "123e4567-e89b-12d3-a456-426614174000", "search_tool_name": "litellm-search", "litellm_params": { "search_provider": "perplexity", "api_key": "sk-***" }, "search_tool_info": { "description": "Perplexity search tool" }, "created_at": "2023-11-09T12:34:56.789Z", "updated_at": "2023-11-09T12:34:56.789Z" } ``` rrLrSNrNrOrPrrrrrTrUrVr[r;r_rYr\r]FrZz"Error getting search tool info: %s)r`rMr%r/r rrrerArr r'r r(rf)r[rMr/rrnror)s rget_search_tool_infors@@N84STT <2PP)Q   >-n-=ZH  &*&**5Er*J%K,>  !- " &!::&67#ZZ(:B?5#ZZ(:;/ <0HI/ <0HI   % 6  <&&'KQOCF;;" \ -H "Content-Type: application/json" \ -d '{ "litellm_params": { "search_provider": "perplexity", "api_key": "sk-..." } }' ``` Example Response (Success): ```json { "status": "success", "message": "Successfully connected to perplexity search provider", "test_query": "test", "results_count": 5 } ``` Example Response (Failure): ```json { "status": "error", "message": "Authentication failed: Invalid API key", "error_type": "AuthenticationError" } ``` r)asearchsearch_providerapi_keyapi_baseiz-search_provider is required in litellm_paramsrPz)Testing connection to search provider: %stestg$@)queryrrr max_resultstimeoutNz4Successfully tested connection to %s search providersuccesszSuccessfully connected to z search provider)statusmessage test_query results_countz(Failed to connect to search provider: %serror)rr error_type)litellm.searchrrTrAr r rdresultslenr'rftyperxr() r~rrTrrrrresponser) error_messagers rtest_search_tool_connectionr+sJV- *!( 6 6!/!3!34E!F'++I6(,,Z8C8gh h""#NP_`# '+ !   ""#Y[jk 3O3DDTU$7?HDTDTc("2"23   [\    $   "1v  G,, &&'QS`a$$    sTD A=CC ACD C D  C D;DDD DD z$/search_tools/ui/available_providerscK ddlm}g}tD]E} |j|}|.|j }|j |j |dGd|iS#t$r+}tjd|j |Yd}~{d}~wwxYw#t$r1}tjd|td t| d}~wwxYww) a Get the list of available search providers with their configuration fields. Auto-discovers search providers and their UI-friendly names from transformation configs. Example Request: ```bash curl -X GET "http://localhost:4000/search_tools/ui/available_providers" \ -H "Authorization: Bearer " ``` Example Response: ```json { "providers": [ { "provider_name": "perplexity", "ui_friendly_name": "Perplexity" }, { "provider_name": "tavily", "ui_friendly_name": "Tavily" } ] } ``` r)ProviderConfigManager)providerN) provider_nameui_friendly_namez/Could not get config for search provider %s: %s providersz,Error getting available search providers: %srNrP) litellm.utilsrrget_provider_search_configrrBrr'r rdr(r rf)ravailable_providersrrkui_namer)s rget_available_search_providersrsB<7%'( H .IIS[I\%$557G'..-5^^07 &011  $**+\^f^l^lnop  <&&'UWXYCF;;rs0((5516 CR 0 02e2HsNT$9cDj"'^'Z'Z[)[ lSd" . S4Z &,@-.%) ! D  +,-*  )00A(Bw<%w<  w