o ưi @sUddlZddlZddlmZmZmZmZddlmZddl m Z ddl Ter5ddl m Z ddl mZmZd ed d d d deefddZdee edffddZdaeee edfed<ddZde dedfddZdS)N) TYPE_CHECKINGDictOptionalType)verbose_logger) CallTypes)*)BaseTranslation) ModelInfoUsagecustom_llm_providerusager model_infor returncCs|dkrddlm}|||dS|dkr ddlm}|||dS|dr1ddlm}|||dS|d kr7d S|d krGddlm}|||dSd S) 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. geminir)cost_per_web_search_request)rrZ anthropic)!get_cost_for_anthropic_web_search)rrZ vertex_aiZ perplexitygZxaiN)Zgemini.cost_calculatorrZanthropic.cost_calculationr startswithZ vertex_ai.gemini.cost_calculatorZxai.cost_calculator)r rrrrZ%cost_per_web_search_request_vertex_airL/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/__init__.pyget_cost_for_web_search_requests         rr c Cs"i}ztjt}|}tj|std|WSt|D]\}}}dd|D|dd<tj|dkrd|vrtj |tj|}d| tj d}z2td |t |}t|d rt|d } t| tr|| td |d t| Wqty} ztd |d | WYd} ~ qd} ~ wty} ztd|d | WYd} ~ qd} ~ wwqzddlm} || tdt| WntytdYnwtdt|dt|W|Sty} ztd| WYd} ~ |Sd} ~ ww)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 foundcSs"g|] }|ds|dkr|qS)__Zbase_llm)r).0drrr Os"z;discover_guardrail_translation_mappings..NZguardrail_translationz __init__.pyzlitellm..z'Discovering guardrail translations in: guardrail_translation_mappingsz(Found guardrail_translation_mappings in z: zCould not import zError processing r)rz-Loaded MCP guardrail translation mappings: %sz:MCP guardrail translation mappings not available; skippingz Discovered z! guardrail translation mappings: z2Error discovering guardrail translation mappings: )ospathdirname__file__existsrdebugwalkbasenamerelpathreplacesep importlib import_modulehasattrgetattr isinstancedictupdatelistkeys ImportErrorerror ExceptionZr=rrrr#load_guardrail_translation_mappingssr? call_typecCs:tdurta|tvrtd|dttt|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 Nz6No guardrail translation mapping found for call_type: z. Available mappings: )r>r= ValueErrorr0r1)r@rrr!get_guardrail_translation_mappings rB)r)rtypingrrrrZlitellm._loggingrZlitellm.types.utilsrZ__annotations__r?rBrrrrs6     % X