uPj UddlZddlZddlmZmZddlmZddlmZddl erddl m Z ddlm Z m Z d ed d d d dedzfdZd ed d d d dedzfdZdeeedffdZdaeeedfdzed<dZdededfdZy)N) TYPE_CHECKINGFinal)verbose_logger) CallTypes)*)BaseTranslation) ModelInfoUsagecustom_llm_providerusager model_infor returncP|dk7r|jdsyddlm}|||S)z Get the cost of Grounding with Google Maps for a given model. Only Gemini models on the Gemini API and Vertex AI can populate the Maps grounding counter, so every other provider returns None. gemini vertex_aiNr)&cost_per_google_maps_grounding_requestr r) startswithgemini.cost_calculatorr)r r rrs Z/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/__init__.py*get_cost_for_google_maps_grounding_requestrs-h&/B/M/Mk/ZN 1* UUc|dk(rddlm}|||S|dk(rddlm}|||S|j dr]|r|j d d nd }d |j vr%ddlm}tjd |||Sddl m}|||S|d k(ry|dk(rddl m}|||S|dk(rddl m}|||Sy)z Get the cost for a web search request for a given model. Args: custom_llm_provider: The custom LLM provider. usage: The usage object. model_info: The model info. rr)cost_per_web_search_requestr anthropic)!get_cost_for_anthropic_web_search)rr rkeyclaudeuSvertex_ai/claude model detected — routing web search cost to Anthropic calculator perplexitygxaigroqN) rranthropic.cost_calculationrrgetlowerrdebug vertex_ai.gemini.cost_calculatorxai.cost_calculatorgroq.cost_calculator)r r rrr model_key%cost_per_web_search_request_vertex_ai groq_cost_per_web_search_requests rget_cost_for_web_search_requestr. sh&G*:NN  +Q0JeTT  ' ' 4>H ub 9R y( ( U  !v w4 RWX X 55ZXX  ,  %D*:NN  & 0e SSrr c i} tjjt}|}tjj |st j d|Stj|D]M\}}}|Dcgc]}|jdr|dk7s|c}|ddtjj|dk(sVd|vs[tjj|tjj|}d|jtjdz} t j d |tj|} t| d r[t!| d } t#| t$r?|j'| t j d |t)| j+P ddlm} |j'| t j dt)| j+t j dt7|t)|j+|Scc}w#t,$r"} t j.d || Yd} ~ d} ~ wt0$r"} t j.d || Yd} ~ #d} ~ wwxYw#t,$rt j dYwxYw#t0$r!} t j.d| Yd} ~ |Sd} ~ wwxYw)a; Discover guardrail translation mappings by scanning the llms directory structure. Scans for modules with guardrail_translation_mappings dictionaries and aggregates them. Returns: Dict[CallTypes, Type[BaseTranslation]]: A dictionary mapping call types to their translation handler classes zllms directory not found__base_llmNguardrail_translationz __init__.pyzlitellm..z)Discovering guardrail translations in: %sguardrail_translation_mappingsz.Found guardrail_translation_mappings in %s: %szCould not import %s: %szError processing %s: %sr)r4z-Loaded MCP guardrail translation mappings: %sz:MCP guardrail translation mappings not available; skippingz0Discovered %s guardrail translation mappings: %sz4Error discovering guardrail translation mappings: %s)ospathdirname__file__existsrr'walkrbasenamerelpathreplacesep importlib import_modulehasattrgetattr isinstancedictupdatelistkeys ImportErrorerror ExceptionXWW__X6 %ww~~h'  !; <& &"$!2  D$"&UQall4.@Q*_qUDGww%)@@]V[E[77??41JK(8+;+;BFFC+HH "(()TVab&44[AFv'GH#*63S#T%h5/66x@*00 PR]_cdldqdqds_t+ @ _  & &'I J  ?7<<>?   > # $ $))+ ,  iV.#"(()BKQRS "(()BKQRS _  !] ^ _ XSUVWW XsAK!KIII!(K KA K0BIKAJ 7KK JI.(K. J:JKJKK=K?KK K- K((K-'endpoint_guardrail_translation_mappingsc.t tatS)N)r[rZrr#load_guardrail_translation_mappingsr^s.62Y2[/ 22r call_typec t ta|tvr,td|dttj t|S)aO Get the guardrail translation handler for a given call type. Args: call_type: The type of call (e.g., completion, acompletion, anthropic_messages) Returns: The translation handler class for the given call type Raises: ValueError: If no translation mapping exists for the given call type z6No guardrail translation mapping found for call_type: z. Available mappings: )r[rZ ValueErrorrFrG)r_s r!get_guardrail_translation_mappingrbs` /62Y2[/??DYKP##'(O(T(T(V#W"X Z  39 ==r)r?r5typingrrlitellm._loggingrlitellm.types.utilsrrrks '+)5 V V%, V:E V T\ V00W0Ze0jorvjv0fKiFWAX6X1YK^\`'i>O9P.P)QTX)X_3>>tDU?V>r