o `Û·iBã@sœUddlmZddlmZmZmZddlmZddlm Z dZ dZ e e dœZ eeeeegeffed<d aiZeeeefed <Gd d „d eeƒZd S) é)ÚEnum)ÚCallableÚDictÚList)ÚRAY_METRIC_CARDINALITY_LEVEL)Ú MetricTypeÚWorkerIdÚName©ÚtasksÚactorsÚ"HIGH_CARDINALITY_GAUGE_AGGREGATIONNÚ_HIGH_CARDINALITY_LABELSc @sˆeZdZdZdZdZdZeddd„ƒZee j fde d e de e egeffd d „ƒZede e fd d „ƒZede de e fdd„ƒZdS)ÚMetricCardinalitya‚Cardinality level configuration for all Ray metrics (ray_tasks, ray_actors, etc.). This configurtion is used to determine whether to globally drop high cardinality labels. This is important for high scale clusters that might consist thousands of workers, millions of tasks. - LEGACY: Keep all labels. This is the default behavior. - RECOMMENDED: Drop high cardinality labels. The set of high cardinality labels are determined internally by Ray and not exposed to users. Currently, this includes the following labels: WorkerId - LOW: Same as RECOMMENDED, but also drop the Name label for tasks and actors. ÚlegacyÚ recommendedÚlowÚreturncCs:tdurtSz tt ¡ƒaWtStytjaYtSw©N)Ú_CARDINALITY_LEVELrrÚlowerÚ ValueErrorÚLEGACY©rrú_/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/_private/telemetry/metric_cardinality.pyÚget_cardinality_level's þþz'MetricCardinality.get_cardinality_levelÚ metric_nameÚ metric_typecCs>|tjtjfvr tS|tjkrtdƒ‚|tvrt|Sdd„S)a°Get the aggregation function for a metric when labels are dropped. This method does not currently support histogram metrics. Args: metric_name: The name of the metric. metric_type: The type of the metric. If provided, Counter and Sum metrics always use sum() aggregation. Returns: A function that takes a list of values and returns the aggregated value. z.No Aggregation function for histogram metrics.cSs|dS)Nrr)ÚvaluesrrrÚIsz.)rÚCOUNTERÚSUMÚsumÚ HISTOGRAMrr )rrrrrÚget_aggregation_function2s z*MetricCardinality.get_aggregation_functioncCs tt ¡ƒSr)Úlistr ÚkeysrrrrÚget_high_cardinality_metricsKs z.MetricCardinality.get_high_cardinality_metricscCsl|tvrt|St ¡}|tjks|t ¡vrgt|<gStgt|<|tjkr2|dvr2t| t¡t|S)z@ Get the high cardinality labels of the metric. r ) rrrrr'ÚWORKER_ID_TAG_KEYÚLOWÚappendÚTASK_OR_ACTOR_NAME_TAG_KEY)rÚcardinality_levelrrrÚ#get_high_cardinality_labels_to_dropOs   z5MetricCardinality.get_high_cardinality_labels_to_dropN)rr)Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚ RECOMMENDEDr)Ú staticmethodrrÚGAUGEÚstrrrÚfloatr$r'r-rrrrrs(   ÿÿÿ þr)ÚenumrÚtypingrrrÚray._private.ray_constantsrÚ#ray._private.telemetry.metric_typesrr(r+r"r r5r6Ú__annotations__rrrrrrrÚs  $þ