tPj UdZddlmZmZddlmZddlmZddlm Z iZ ee e efe d<de dee fd Zde dee fd Zdd Zee j&j(e j*j(e j,j(e j.j(e j0j(e j2j(e j4j(e j6j(hZee d <de de dee fdZy )a Extract tool names from request body by route/call type. Used by auth (check_tools_allowlist) and ToolPolicyGuardrail so tool-format knowledge lives in one place. Uses guardrail translation handlers where available, with standalone extractors for generate_content and MCP. )AnyFinal)get_call_types_for_route)#load_guardrail_translation_mappings) CallTypesSTANDALONE_EXTRACTORSdatareturncg}|jdxsgD]m}t|ts|jdxsgD]B}t|ts|jds&|jt |dDo|S)z;Google generateContent: tools[].functionDeclarations[].nametoolsfunctionDeclarationsname)get isinstancedictappendstr)r namestooldecls r/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/proxy/guardrails/tool_name_extraction.py$_extract_generate_content_tool_namesrs E!'R0$% HH34: 0D$%$((6*: Sf./ 00 Lcg}|jdxs|jd}|r|jt||S)z,MCP call_tool: name or mcp_tool_name in bodyr mcp_tool_name)rrr)r rrs r_extract_mcp_tool_namesrs; E((6"?dhh&?D  SY LrNctrytttjj<tttj j<t ttjj<y)N)rrrgenerate_contentvalueagenerate_contentr call_mcp_toolrr_register_standalone_extractorsr#(sM>b)44::;?c)55;;<;R)11778rTOOL_CAPABLE_CALL_TYPESroutecZt|}|sgStt}|D]}t|ts|j t vr'|j tvrt|j |cS|j|}|g|j|}|s|cSgS)a Extract tool names from the request body for the given route. Uses guardrail translation handlers when available, else standalone extractors for generate_content and MCP. Returns [] for non-tool-capable routes or when no tools are present. ) rr#rrrrr$rrextract_request_tool_names)r%r call_typesmappings call_type handler_clsrs rr'r'?s17J  #%9;H  )Y/  ??"9 9  ??3 3(9$? ?ll9-  "M<r9sW<)02uT#s(^,1 t S $49S"+""##!!""$$**""((##))%%  " c$s)r