o Lj@sddlmZmZddlmZmZmZmZdedefddZ Gdd d eZ Gd d d eZ Gd d d e Z Gddde Z Gddde ZGddde ZGddde ZGddde ZGddde ZGdddeZGddde ZGdddeZGd d!d!eZGd"d#d#eZd$S)%)IterableUnion)AscDescReducer SortDirectionnamereturncCs|dr|Sd|S)z3Return ``name`` with a single leading ``@`` prefix.@) startswith)r r \/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/redis/commands/search/reducers.py_ensure_at_prefixsrcs*eZdZdZdeddffdd ZZS)FieldOnlyReducerzHSee https://redis.io/docs/interact/search-and-query/search/aggregations/fieldr Ncst|||_dSN)super__init___fieldselfr __class__r rrs  zFieldOnlyReducer.__init__)__name__ __module__ __qualname____doc__strr __classcell__r r rrr srcs&eZdZdZdZdfdd ZZS)countz3 Counts the number of results in the group COUNTr NcstdSrrr)rrr rrszcount.__init__)r N)rrrrNAMErrr r rrr sr c.eZdZdZdZdeddffdd ZZS)sumzS Calculates the sum of all the values in the given fields within the group SUMrr Nct|dSrr"rrr rr%z sum.__init__rrrrr#rrrr r rrr%r%cr$)minzK Calculates the smallest value in the given field within the group MINrr Ncr'rr"rrr rr0r(z min.__init__r)r r rrr+)r*r+cr$)maxzJ Calculates the largest value in the given field within the group MAXrr Ncr'rr"rrr rr;r(z max.__init__r)r r rrr-4r*r-cr$)avgzG Calculates the mean value in the given field within the group AVGrr Ncr'rr"rrr rrFr(z avg.__init__r)r r rrr/?r*r/cr$)tolistz6 Returns all the matched properties in a list TOLISTrr Ncr'rr"rrr rrQr(ztolist.__init__r)r r rrr1Jr*r1cr$)count_distinctzs Calculate the number of distinct values contained in all the results in the group for the given field COUNT_DISTINCTrr Ncr'rr"rrr rr]r(zcount_distinct.__init__r)r r rrr3Usr3c@seZdZdZdZdS)count_distinctishz Calculate the number of distinct values contained in all the results in the group for the given field. This uses a faster algorithm than `count_distinct` but is less accurate COUNT_DISTINCTISHN)rrrrr#r r r rr5asr5c2eZdZdZdZdededdffdd ZZS) quantilezp Return the value for the nth percentile within the range of values for the field within the group. QUANTILErpctr Ncst|t|||_dSr)rrrr)rrr:rr rrss zquantile.__init__) rrrrr#rfloatrrr r rrr8ks"r8cr$)stddevzG Return the standard deviation for the values within the group STDDEVrr Ncr'rr"rrr rrr(zstddev.__init__r)r r rrr<xr*r<cs:eZdZdZdZdedeeefddffdd Z Z S) first_valuezR Selects the first value within the group according to sorting parameters FIRST_VALUErbyfieldsr Ncsg}t|dkrt|dtrt|dtr|d|g}|D] }||j|jg7}q|g}|r6|dg|7}tj|||_ dS)a Selects the first value of the given field within the group. ### Parameter - **field**: Source field used for the value - **byfields**: How to sort the results. This can be either the *class* of `aggregation.Asc` or `aggregation.Desc` in which case the field `field` is also used as the sort input. `byfields` can also be one or more *instances* of `Asc` or `Desc` indicating the sort order for these fields rrBYN) len isinstancetype issubclassrr DIRSTRINGrrr)rrr@ fieldstrsfargsrr rrs     zfirst_value.__init__) rrrrr#rrrrrrr r rrr>s*r>cr7) random_samplezT Returns a random sample of items from the dataset, from the given property RANDOM_SAMPLErsizer Ncs"|t|g}tj|||_dS)z ### Parameter **field**: Field to sample from **size**: Return this many items (can be less) N)rrrr)rrrLrIrr rrs   zrandom_sample.__init__) rrrrr#rintrrr r rrrJs"rJc sleZdZdZdZ    d deeeBdedee Beee BBdBd e e e fdBd df fd d Z Z S)collecta Gathers the rows of each ``GROUPBY`` group, projects a chosen set of fields from every row, optionally deduplicates, sorts, and limits them, and returns them as an array of per-entry maps under the reducer alias. ``COLLECT`` is a preview feature gated behind ``search-enable-unstable-features``. See `FT.AGGREGATE `_. COLLECT*FNfieldsdistinctsort_bylimitr c s g}|dkr |ddg7}n.t|tr|gnt|}|r$tdd|Dr(tddd|D}|dtt|g|7}|rB|dg7}|d urut|ttfrP|gn|}g}|D] } |t| j | j g7}qV|sitd |d tt|g|7}|d ur|\} } |d t| t| g7}t j |d S) a ### Parameters - **fields**: The fields to project from each collected row. Either the literal ``"*"`` (project every field present in the pipeline at this stage) or a field name / iterable of field names. Names are normalized to a single leading ``@`` on the wire; output map keys are the bare names. - **distinct**: When ``True``, emit ``DISTINCT`` to deduplicate entries with identical projected fields. Forward-compatible: this option is not yet implemented by the server and currently produces a server error when sent. - **sort_by**: An ``Asc``/``Desc`` instance or an iterable of them, used to order the collected entries within each group. - **limit**: An ``(offset, count)`` pair. Returns at most ``count`` entries per group after skipping ``offset``. With ``sort_by`` this acts as a top-N selection. rPFIELDScss|]}| VqdSr)strip.0nr r r sz#collect.__init__..z7collect fields must be '*' or a non-empty list of namescSsg|]}t|qSr )rrWr r r sz$collect.__init__..DISTINCTNz/collect sort_by must contain at least one fieldSORTBYLIMIT) rCrlistany ValueErrorrBrrrrrFrr) rrQrRrSrTrInames sort_fields sort_argsrHoffsetr rr rrs0 zcollect.__init__)rPFNN)rrrrr#rrboolrrtuplerMrrr r rrrNs$  rNN)typingrr aggregationrrrrrrrr r%r+r-r/r1r3r5r8r<r>rJrNr r r rs"          (