RmjfdZddlmZddlmZmZmZmZddlm Z m Z m Z m Z ddl mZmZddl mZGdde ZGd d ed ZGd de ZGdde Zd4dZd5dZedGddZedGddZedGddZedGd d!ZedGd"d#ZedGd$d%ZedGd&d'ZedGd(d)ZedGd*d+Z d,Z!edGd-d.Z"d6d1Z#gd2Z$d3S)7z+Model pricing plugin configuration helpers.) annotations) dataclassfieldfields is_dataclass)LiteralProtocol TypedDictcast)Json JsonObject)pluginc.eZdZUded<ded<ded<dS)_ConfigDiagnosticRequiredzLiteral['warning', 'error']levelstrcodemessageN)__name__ __module__ __qualname____annotations__`/home/thesage/.hermes/hermes-agent/venv/lib/python3.11/site-packages/nemo_relay/model_pricing.pyrrs.&&&& IIILLLLLrrc(eZdZUdZded<ded<dS)ConfigDiagnosticz(One model pricing validation diagnostic.r componentrNrrr__doc__rrrrrrs(22NNNJJJJJrrF)totalceZdZUdZded<dS) ConfigReportz2Validation report for model pricing configuration.zlist[ConfigDiagnostic] diagnosticsNrrrrr#r#s$<<''''''rr#ceZdZddZdS)_SupportsToDictreturnr cdSNrselfs rto_dictz_SupportsToDict.to_dict#srNr'r )rrrr,rrrr&r&"s((((((rr&valueobjectr'r ctdr'ttSt r1t t sfdtDSt tr dDSt trd DSttS)Nr,chi|].}t|jx|jt/Sr))getattrname _normalize).0 field_info field_valuer.s r z_normalize..*sF   &ujo>>> K OZ 44KKKrc,g|]}t|Sr)r4)r5items r z_normalize..0s 333T 4  333rc^i|]*\}}|tt|t|+Sr))r rr4)r5keyvals rr8z_normalize..2s2aaaHCQTQ`S# 3Q`Q`Q`r) hasattrr r&r,r isinstancetyperlistdictitemsr )r.r7s`@rr4r4&sui  6OU++33555E :eT#:#:      $Umm    %433U3333%baa aaaa e  rr cFttt|Sr))r r r4)r.s r_normalize_objectrF6s  Ju-- . ..rT)slotscLeZdZUdZded<ded<dZded<dZded<d d ZdS) TokenPricingRatesz7Per-token model rates expressed per one million tokens.floatinput_per_millionoutput_per_millionNz float | Nonecache_read_per_millioncache_write_per_millionr'r cRt|j|j|j|jdS)z9Serialize these rates to the canonical JSON object shape.)rKrLrMrN)rFrKrLrMrNr*s rr,zTokenPricingRates.to_dictCs6 %)%;&*&=*.*E+/+G      rr-)rrrr rrMrNr,rrrrIrI:slAA+/////,00000       rrIc*eZdZUdZdZded<d dZdS) PromptCachePricingz;Prompt-cache accounting settings for a model pricing entry.included_in_prompt_tokensz0Literal['included_in_prompt_tokens', 'separate']read_accountingr'r cd|jiS)zFSerialize this prompt-cache config to the canonical JSON object shape.rS)rSr*s rr,zPromptCachePricing.to_dictUs!4#788rNr-)rrrr rSrr,rrrrQrQOsAEEHcOcccc999999rrQcBeZdZUdZded<dZded<dZded<d d ZdS) TokenRateTierz,One threshold tier in a token rate schedule.zTokenPricingRates | JsonObjectratesNz int | Nonemin_prompt_tokensmax_prompt_tokensr'r cFt|j|j|jdS)zeZdZUdZeeZded<d dZdS) PricingConfigzDCanonical config document for the top-level model pricing component.r]z,list[InlineSource | FileSource | JsonObject]sourcesr'r c.td|jiS)zGSerialize this model pricing config to the canonical JSON object shape.r)rFrr*s rr,zPricingConfig.to_dicts )T\!:;;;rNr-) rrrr rrBrrr,rrrrrsONNrs/21"""""">>>>>>>>>>>>555555555555''''''''......  0(((((9((( )))))h)))     ////         ( 99999999         $         "         @                                  <<<<<<<<            &&&&   r