`gjdZddlmZddlmZddlmZmZddlm Z ddl m Z edZ edZ d Zd Zed Gd dZddZddZ ddZddddd ddZy)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)frozencNeZdZUdZded<ded<ded<ded<ded<ded <y ) ExpensiveModelWarningz=Confirmation payload for models above Hermes' cost guardrail.strmodelproviderOptional[Decimal]input_cost_per_millionoutput_cost_per_millionsourcemessageN)__name__ __module__ __qualname____doc____annotations__N/root/.hermes/venv/lib/python3.12/site-packages/hermes_cli/model_cost_guard.pyr r s%G JM--.. K Lrr c\|y tt|S#ttf$rYywxYw)N)rrr ValueErrorvalues r _to_decimalr"s5 }s5z"" j )s ++c|yd|ddS)Nunknown$z.2fz/Mrr s r _format_moneyr&'s } uSk rc~||jsyt|jt|jdfS)N)NNz models.dev) has_cost_datar" cost_input cost_output) model_infos r_pricing_from_model_infor--sA!9!9!;J))*J**+ rN)rbase_urlapi_keyr,c |xsdj}|syt|\}}}|!||r ddlm} t| ||\}}}|;|9 ddlm} | ||||} | $| j}| j}| j}|duxr |tkD} |duxr |tkD} | s| sydd|dd t|d t|d g}|r|jd |d |jtk(r|jt |jdt#||xsdj|||xsddj%|S#t$rYwxYw#t$rd} YwxYw)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)rr.r/z!!! 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)stripr-agent.models_devr1 Exceptionagent.usage_pricingr2rrrINPUT_COST_WARNING_THRESHOLDOUTPUT_COST_WARNING_THRESHOLDr&appendlowerGPT55_PRO_OPENROUTER_IDGPT55_SUGGESTIONr join) model_namerr.r/r,r input_cost output_costrr1r2entry over_input over_outputliness rexpensive_model_warningrG9s 2 $ $ &E &>z&J#J Vk1h  7.Fx// +J V k1  =%!! E  55J77K\\F $L:0L#L 4OK2O$O k * 'CD z234 - 456 $ E 'xq12 {{}// %& LL@A .b'')) +" %  [    E s#EE' E$#E$' E65E6)r!objectreturnr)r!rrIr)r,Optional[ModelInfo]rIz0tuple[Optional[Decimal], Optional[Decimal], str]) r@rr Optional[str]r.rKr/rKr,rJrIzOptional[ExpensiveModelWarning])r __future__r dataclassesrdecimalrrtypingrr6rr9r:r=r>r r"r&r-rGrrrrPsH"!-& 't} '.; $ # 5 #"!&* MMM M  M $ M%Mr