)jdZddlmZddlmZddlmZmZddlm Z ddl m Z edZ edZ d Zd Zed Gd dZd#dZd$dZd%dZdddddd&d"ZdS)'zBExpensive-model confirmation helpers for model selection surfaces.) annotations) dataclass)DecimalInvalidOperation)Optional) ModelInfo20100zopenai/gpt-5.5-proz&did you mean to select openai/gpt-5.5?T)frozencPeZdZUdZded<ded<ded<ded<ded<ded <d S) ExpensiveModelWarningz=Confirmation payload for models above Hermes' cost guardrail.strmodelproviderOptional[Decimal]input_cost_per_millionoutput_cost_per_millionsourcemessageN)__name__ __module__ __qualname____doc____annotations__B/home/wildlama/.hermes/hermes-agent/hermes_cli/model_cost_guard.pyr r sRGGJJJMMM----....KKKLLLLLrr valueobjectreturnrct|dS tt|S#ttf$rYdSwxYw)N)rrr ValueErrorrs r _to_decimalr$sL }ts5zz""" j )tts "77rc|dSd|ddS)Nunknown$z.2fz/Mrr#s r _format_moneyr('s  }y u    r model_infoOptional[ModelInfo]0tuple[Optional[Decimal], Optional[Decimal], str]c||sdSt|jt|jdfS)N)NNz models.dev) has_cost_datar$ cost_input cost_output)r)s r_pricing_from_model_infor1-sI!9!9!;!;~J)**J*++ rN)rbase_urlapi_keyr) model_namer Optional[str]r2r3Optional[ExpensiveModelWarning]c B|pd}|sdSt|\}}}|9|7|r5 ddlm} t| ||\}}}n#t$rYnwxYw|B|@ ddlm} | ||||} n#t$rd} YnwxYw| | j}| j}| j }|duo |tk} |duo |tk} | s| sdSdd|dd t|d t|d g}|r| d |d |tkr| t | dt#||pd|||pdd|S)zReturn a warning payload when known pricing exceeds safety thresholds. The guard only triggers when pricing is known. Callers should use this after model resolution so aliases and provider-specific model IDs have settled. r-Nr)get_model_info)get_pricing_entry)rr2r3z!!! EXPENSIVE MODEL WARNING !!!z2 has known pricing above Hermes' safety threshold.zInput tokens: zOutput tokens: zJThreshold: more than $20/M input tokens or more than $100/M output tokens.zPricing source: .z-Confirm only if you intend to use this model.r& )rrrrrr)stripr1agent.models_devr8 Exceptionagent.usage_pricingr9rrrINPUT_COST_WARNING_THRESHOLDOUTPUT_COST_WARNING_THRESHOLDr(appendlowerGPT55_PRO_OPENROUTER_IDGPT55_SUGGESTIONr join)r4rr2r3r)r input_cost output_costrr8r9entry over_input over_outputliness rexpensive_model_warningrM9sq 2 $ $ & &E t&>z&J&J#J Vk1h1  7 7 7 7 7 7.Fx//// +J VV    D k1  = = = = = =%%!! EE    EEE   5J7K\F $L:0L#L 4OK2O$O kt * DDD4z22446- 4466 $ E3 1111222 {{}}/// %&&& LL@AAA .b''))) +" %     s##A A&%A&.B BB)rrr r)rrr r)r)r*r r+) r4rrr5r2r5r3r5r)r*r r6)r __future__r dataclassesrdecimalrrtypingrr=rr@rArDrEr r$r(r1rMrrrrRsQHH""""""!!!!!!--------&&&&&& 'wt}} '.; $    #"!&* MMMMMMMMr