,i۫ddlZddlZddlmZmZmZmZmZddlm Z ddl m Z m Z ddl mZmZmZmZddlmZmZddlmZdd lmZd d lmZd d lmZmZmZd d lm Z d dl!m"Z"d dl#m$Z$d dl%m&Z&d dl'm(Z(d dl)m*Z*d dl+m,Z,dZ-dZ.dZ/dZ0dZ1dZ2dZ3dZ4dZ5dZ6dZ7dZ8d Z9d!Z:d"Z;d#Zd&Z?d'Z@d(ZAd)ZBd*ZCd+ZDd,ZEd-ZFd.ZGd/ZHd0ZId1ZJd2ZKd3ZLd4ZMd5ZNd6ZOd7ZPd8ZQd9ZRd:ZSd;ZTdd?ZWGd@dAeWZXy)BN)AnyDictListOptionalUnion) pairs_to_dict) NEVER_DECODEPipeline)CombineResultsMethodHybridCursorQueryHybridPostProcessingConfig HybridQuery)HybridCursorResult HybridResult)deprecated_function)get_protocol_version) to_string)AggregateRequestAggregateResultCursor)Document)Field)IndexDefinition)ProfileInformation)Query)Result)SuggestionParserNUMERICz FT.CREATEzFT.ALTERz FT.SEARCHzFT.ADDz FT.ADDHASHz FT.DROPINDEXz FT.EXPLAINz FT.EXPLAINCLIzFT.DELz FT.AGGREGATE FT.PROFILEz FT.CURSORz FT.SPELLCHECKz FT.DICTADDz FT.DICTDELz FT.DICTDUMPzFT.MGETz FT.CONFIGz FT.TAGVALSz FT.ALIASADDzFT.ALIASUPDATEz FT.ALIASDELzFT.INFOz FT.SUGADDz FT.SUGDELz FT.SUGLENz FT.SUGGETz FT.SYNUPDATEz FT.SYNDUMPz FT.HYBRID NOOFFSETSNOFIELDSNOHLNOFREQS MAXTEXTFIELDS TEMPORARY STOPWORDSSKIPINITIALSCAN WITHSCORESFUZZY WITHPAYLOADSceZdZdZdZdZdZdZdZdZ dZ d Z d Z dcd Z ddd eedededeeedeedededefdZd eeeeffdZdedefdZ dfdZ dgdZedd dhdeded ed!eed"ed#ed$eed%ed eefd&Zedddid'Zedd(djd)Zd*Zedd(d+Z d,Z!d-ee"eeee#ee$fffd.Z%d-ee"eeee#ee$fffd/Z& dkd0eee'fd-ee"eeee#ee$ffd ffd1Z( dld0e)d2ee*d3ee+d4ee"eeee#ee$ffd5ee#d6ee,d7ee-e.e/ffd8Z0 dkd0eee'fd-ee"eeee#ee$fffd9Z1d0eee'ffd:Z2 dkd0ee3e4fd-ee"eeee#ee$fffd;Z5dZ6 dmd0ee'e3fd?ed-ee"eeee#ee$fffd@Z7dndAZ8dBedCeefdDZ9dBedCeefdEZ:dBefdFZ;edGdHdIedJed7efdKZ<edGdLdIed7efdMZ=dNefdOZ>dPefdQZ?dPefdRZ@dPefdSZAdTZBdUed7e#fdVZCdUedWed7e#fdXZD dodUedYedZed[e#d\ed]ed7eeEfd^ZFded_ed`edCeefdaZGdbZHy )pSearchCommandszSearch commands.c t|jdvr|dk(r t|S|S|j||fi|S)N3r!)rclientr_RESP2_MODULE_CALLBACKS)selfcmdreskwargss c/var/www/html/langgraph-service/venv/lib/python3.12/site-packages/redis/commands/search/commands.py_parse_resultszSearchCommands._parse_resultsPsH  , 8.1\.A%c* Js J44//4SCFC Cc Ltt|}tt||SN)maprdictzip)r5r7r8its r9 _parse_infozSearchCommands._parse_infoVs C CBK  r;c t||dj |d|dj|dj|djS)Nqueryduration)rE has_payload with_scoresfield_encodings)r _no_content_with_payloads _with_scores_return_fields_decode_asr5r7r8s r9 _parse_searchzSearchCommands._parse_searchZsQ w++ +J'w66w44"7ODD   r;c t|d}d|vr%tt|dt|dSg}|dD] }t|d}|j|"t t|d||d t |d  S) NT) decode_keyscursorSEARCHVSIM)search_cursor_idvsim_cursor_idresults total_resultswarningsexecution_time)rWrVrXrY)rrintappendrfloat)r5r7r8res_dictrVres_item item_dicts r9_parse_hybrid_searchz#SearchCommands._parse_hybrid_searchds $7 v %!$Xh%7!8"8F#34  )+!+ &H%hDAI NN9 % &h78j) *:!;<   r;c 4|j||d|dS)NrD has_cursor)_get_aggregate_resultrMs r9_parse_aggregatezSearchCommands._parse_aggregatezs ))#vg|@TUUr;c |d}t|tr!|j|d||j}n5t |d|j |d|j |j}|t|dfS)NrDrrE)rErFrGr) isinstancerrc_cursorrrIrJrKr)r5r7r8rDresults r9_parse_profilezSearchCommands._parse_profile}sw e- .//Au}}MFA%%% +!00!.. F)#a&111r;c i}|dk(r|S|D]X}t|tr|dk(rt|dk7r(|ds.|dds7|dDcgc] }|d|ddc}||d<Z|Scc}w)Nrr2rr)score suggestion)rfrZlen)r5r7r8 corrections _correction_items r9_parse_spellcheckz SearchCommands._parse_spellchecks !8  K+s+ q0@;1$q>q>!$$JUUV+@E%(%(;+K A '5 < +s A)c B|r|Dcic] }|d|d c}SiScc}w)Nrr)r5r7r8kvss r9_parse_config_getz SearchCommands._parse_config_gets)25#.3AA.=2=.sc htdt|dDcic]}||||dzc}Scc}w)Nrrr)rangerm)r5r7r8is r9_parse_syndumpzSearchCommands._parse_syndumps4,1!SXq,ABqAAE "BBBs/c(|j||S)zT Create a new batch indexer from the client with a given chunk size ) chunk_size) BatchIndexer)r5r{s r9 batch_indexerzSearchCommands.batch_indexers  * ==r;Nfieldsno_term_offsetsno_field_flags stopwords definition no_highlightno_term_frequenciesskip_initial_scanc 6t|jg} || |jz } |r| jt|6t |t r&| jt| j||r| jt|r| jt|r| jt| r| jt| r| jt|Kt |tttfr0| t t#|gz } t#|dkDr| t|z } | jd | tt%j&d|Dz } |j,| S#t($r| |j+z } Y-wxYw)a Creates the search index. The index must not already exist. For more information, see https://redis.io/commands/ft.create/ Args: fields: A list of Field objects. no_term_offsets: If `true`, term offsets will not be saved in the index. no_field_flags: If true, field flags that allow searching in specific fields will not be saved. stopwords: If provided, the index will be created with this custom stopword list. The list can be empty. definition: If provided, the index will be created with this custom index definition. max_text_fields: If true, indexes will be encoded as if there were more than 32 text fields, allowing for additional fields beyond 32. temporary: Creates a lightweight temporary index which will expire after the specified period of inactivity. The internal idle timer is reset whenever the index is searched or added to. no_highlight: If true, disables highlighting support. Also implied by `no_term_offsets`. no_term_frequencies: If true, term frequencies will not be saved in the index. skip_initial_scan: If true, the initial scan and indexing will be skipped. rSCHEMAc3<K|]}|jywr= redis_args.0fs r9 z.SearchCommands.create_index..*Ja1<<>*J) CREATE_CMD index_nameargsr[r&rfrZr'r"r$r#r%r)listtuplesetr(rm itertoolschain TypeErrorrexecute_command) r5r~rrrrmax_text_fields temporaryrrrrs r9 create_indexzSearchCommands.create_indexsSNDOO,  ! JOO #D  KK &  Z 3%? KK " KK "  KK "  KK   KK !  KK  KK (  Z D%;M%N YI/ /D9~!Y' H ( D*J6*JKL LD$t##T** ( F%%' 'D (s%E99FFct|jddg} |ttjd|Dz }|j|S#t $r||j z }Y-wxYw)a Alter the existing search index by adding new fields. The index must already exist. ### Parameters: - **fields**: a list of Field objects to add for the index For more information see `FT.ALTER `_. rADDc3<K|]}|jywr=rrs r9rz2SearchCommands.alter_schema_add..rr) ALTER_CMDrrrrrrr)r5r~rs r9alter_schema_addzSearchCommands.alter_schema_addsr4??He< ( D*J6*JKL LD$t##T** ( F%%' 'D (s%A A)(A)delete_documentsct|jg}t|tr|durdnd}|r|j ||j |S)ai Drop the index if it exists. Replaced `drop_index` in RediSearch 2.0. Default behavior was changed to not delete the indexed documents. ### Parameters: - **delete_documents**: If `True`, all documents will be deleted. For more information see `FT.DROPINDEX `_. TDD) DROPINDEX_CMDrrfboolr[r)r5rr delete_strs r9 dropindexzSearchCommands.dropindexsXt/*D16F$6N    KK ##t##T**r;c  |s| rd}t|j||g} |r| jd|"| jd| j||r7| jd|r| jd| r| jd|r| d|gz } | jd| tt j | j z } ||j| S|j| S) zS Internal add_document used for both batch and single doc indexing TNOSAVEPAYLOADREPLACEPARTIALNOCREATELANGUAGEFIELDS)ADD_CMDrr[rrritemsr) r5doc_idconnnosaverkpayloadreplacepartiallanguage no_creater~rs r9 _add_documentzSearchCommands._add_document0s" iG&%8  KK !   KK " KK  KK " I& J'  Z* *D H Y__flln566  '4''. .#t##T**r;ct|j||g}|r|jd|r|d|gz }||j|S|j|S)zX Internal add_document_hash used for both batch and single doc indexing rr) ADDHASH_CMDrr[r)r5rrrkrrrs r9_add_document_hashz!SearchCommands._add_document_hashZscT__fe<  KK "  Z* *D  '4''. .#t##T**r;z2.0.0z2deprecated since redisearch 2.0, call hset insteadversionreasonrrrkrrrrrc :|j|fd|||||||d| S)a Add a single document to the index. Args: doc_id: the id of the saved document. nosave: if set to true, we just index the document, and don't save a copy of it. This means that searches will just return ids. score: the document ranking, between 0.0 and 1.0 payload: optional inner-index payload we can save for fast access in scoring functions replace: if True, and the document already is in the index, we perform an update and reindex the document partial: if True, the fields specified will be added to the existing document. This has the added benefit that any fields specified with `no_index` will not be reindexed again. Implies `replace` language: Specify the language used for document tokenization. no_create: if True, the document is only updated and reindexed if it already exists. If the document does not exist, an error will be returned. Implies `replace` fields: kwargs dictionary of the document fields to be saved and/or indexed. NOTE: Geo points shoule be encoded as strings of "lon,lat" N)rrrkrrrrr)r) r5rrrkrrrrrr~s r9 add_documentzSearchCommands.add_documentnsDV"t!!       r;c.|j|d|||S)a Add a hash document to the index. ### Parameters - **doc_id**: the document's id. This has to be an existing HASH key in Redis that will hold the fields the index needs. - **score**: the document ranking, between 0.0 and 1.0 - **replace**: if True, and the document already is in the index, we perform an update and reindex the document - **language**: Specify the language used for document tokenization. N)rrkrr)r)r5rrkrrs r9add_document_hashz SearchCommands.add_document_hashs' && UXw'  r;zdeprecated since redisearch 2.0ct|j|g}|r|jd||j|S|j|S)a Delete a document from index Returns 1 if the document was deleted, 0 if not ### Parameters - **delete_actual_document**: if set to True, RediSearch also delete the actual document if it is in the index r)DEL_CMDrr[r)r5rrdelete_actual_documentrs r9delete_documentzSearchCommands.delete_documentsO&1 ! KK   '4''. .#t##T**r;c|jj|}|jDcic]\}}t|t|}}}|} |d=t dd|i|Scc}}w#t$rYwxYw). Load a single document by id idrsr3hgetallrrKeyErrorrr5rr~kvf2s r9 load_documentzSearchCommands.load_documents$$R(5;\\^ DTQilIaL( D D t (2(((E    sA$A** A65A6cF|jt|jg|S)z Returns the full contents of multiple documents. ### Parameters - **ids**: the ids of the saved documents. )rMGET_CMDr)r5idss r9getzSearchCommands.gets"$t##HdooDDDr;cn|jt|j}|jt|S) Get info an stats about the the current index, including the number of documents, memory consumption, etc For more information see `FT.INFO `_. rINFO_CMDrr:r5r7s r9infozSearchCommands.infos-""8T__=""8S11r; query_paramsc|gSg}t|dkDrh|jd|jt|dz|jD]'\}}|j||j|)|S)NrPARAMSr)rmr[r)r5rrkeyvalues r9get_params_argszSearchCommands.get_params_argss~  I | q KK ! KKL)A- .*002 # U C  E" # r;c|jg}t|tr t|}t|tst dt |||j z }||j|z }||fS)NzBad query type )rrfstrr ValueErrortypeget_argsr)r5rDrrs r9_mk_query_argszSearchCommands._mk_query_argsss  eS !%LE%'tE{m<= =    $$\22U{r;rDcT|j||\}}tj}i}t|jdvr d|t <|j tg|i|}t|tr|S|jt||tj|z dzS) Search the index for a given query, and return a result of documents ### Parameters - **query**: the search query. Either a text for simple queries with default parameters, or a Query object for complex queries. See RediSearch's documentation on query format For more information see `FT.SEARCH `_. rr0T@@rDrE rtime monotonicrr3r r SEARCH_CMDrfr r:r5rDrrstoptionsr7s r9searchzSearchCommands.searchs ))%l)K e ^^   ,H <$(GL !"d"":@@@ c8 $J"" 5DNN4Dr4IV3S#  r;combine_methodpost_processingparams_substitutiontimeoutrQreturncb|j}i}t|g} | j|j|r| j|j|r| j|j |r | j|j ||r| jd|f|r$d|d<| j|j t |jdvr d|t<|j| i|} t| tr| S|jt| fi|S)M Execute a hybrid search using both text and vector queries Args: - **query**: HybridQuery object Contains the text and vector queries - **combine_method**: CombineResultsMethod object Contains the combine method and parameters - **post_processing**: HybridPostProcessingConfig object Contains the post processing configuration - **params_substitution**: Dict[str, Union[str, int, float, bytes]] Contains the parameters substitution - **timeout**: int - contains the timeout in milliseconds - **cursor**: HybridCursorQuery object - contains the cursor configuration For more information see `FT.SEARCH `. TIMEOUTTrQr0 r HYBRID_CMDextendr build_argsrrr3r rrfr r: r5rDrrrrrQindexrpiecesr7s r9 hybrid_searchzSearchCommands.hybrid_search3s 6e$ enn&'  MM.113 4  MM/446 7  MM$../BC D  MM9g. /  $GH  MM&++- .  ,H <$(GL !"d""F6g6 c8 $J"t"":s>g>>r;c\|j||\}}|jtg|S)zReturns the execution plan for a complex query. For more information see `FT.EXPLAIN `_. r)rr EXPLAIN_CMD)r5rDrr query_texts r9explainzSearchCommands.explainhs6 ..u<.Pj#t##K7$77r;ctd)Nz#EXPLAINCLI will not be implemented.)NotImplementedError)r5rDs r9 explain_clizSearchCommands.explain_clits!"GHHr;ct|tr9t|j}t|j g|j z}nCt|tr'd}td|j g|j z}n td|||j|z }|j|}|jt|||S) Issue an aggregation query. ### Parameters **query**: This can be either an `AggregateRequest`, or a `Cursor` An `AggregateResult` object is returned. You can access the rows from its `rows` property, which will always yield the rows of the result. For more information see `FT.AGGREGATE `_. TREAD Bad queryrDrb rfrrrg AGGREGATE_CMDrrr CURSOR_CMDrrrr:r5rDrrbr6raws r9 aggregatezSearchCommands.aggregatews" e- .emm,J $//2U5E5E5GGC v &Jvt7%:J:J:LLC[%0 0 t##L11"d""C("" 3e #  r;rrbc|r1t|tr |d|_|}nt|d}|d}nd}t|tr|jr |d}|dd}nd}|dd}t |||S)Nrrr)rfrcidr _with_schemar)r5rrDrbrQschemarowss r9rcz$SearchCommands._get_aggregate_results %(F Aa&CF e- .53E3EVFqr7DFqr7DtVV44r;limitedctj}t|jdg}|r|j d|j dt |t rd|d<||jz }nHt |tr-d|d<||jz }||j|z }n td|j|}|jt||tj|z dz S) a Performs a search or aggregate command and collects performance information. ### Parameters **query**: This can be either an `AggregateRequest` or `Query`. **limited**: If set to True, removes details of reader iterator. **query_params**: Define one or more value parameters. Each parameter has a name and a value. rLIMITEDQUERY AGGREGATErrRz5Must provide AggregateRequest object or Query object.rr)rr PROFILE_CMDrr[rfrrrrrrrr:)r5rDr!rrr6r7s r9profilezSearchCommands.profiles$^^ DOOR0  JJy ! 7 e- . CF 5##% %C u %CF 5>># #C 4'' 5 5CTU U"d""C("" ET^^5E5Jf4T#  r;ct|j|g}|r|jd|g|r|jdd|g|r|jdd|g|j|}|j t|S)a Issue a spellcheck query Args: query: search query. distance: the maximal Levenshtein distance for spelling suggestions (default: 1, max: 4). include: specifies an inclusion custom dictionary. exclude: specifies an exclusion custom dictionary. For more information see `FT.SPELLCHECK `_. DISTANCETERMSINCLUDEEXCLUDESPELLCHECK_CMDrrrr:r5rDdistanceincludeexcluder6r7s r9 spellcheckzSearchCommands.spellchecksyt6  JJ H- .  JJG4 5  JJG4 5"d""C("">377r;nametermscRt|g}|j||j|S)zAdds terms to a dictionary. ### Parameters - **name**: Dictionary name. - **terms**: List of items for adding to the dictionary. For more information see `FT.DICTADD `_. ) DICT_ADD_CMDrrr5r4r5r6s r9dict_addzSearchCommands.dict_add-T" 5#t##S))r;cRt|g}|j||j|S)aDeletes terms from a dictionary. ### Parameters - **name**: Dictionary name. - **terms**: List of items for removing from the dictionary. For more information see `FT.DICTDEL `_. ) DICT_DEL_CMDrrr8s r9dict_delzSearchCommands.dict_delr:r;c0t|g}|j|S)zDumps all terms in the given dictionary. ### Parameters - **name**: Dictionary name. For more information see `FT.DICTDUMP `_. ) DICT_DUMP_CMDr)r5r4r6s r9 dict_dumpzSearchCommands.dict_dump s!d##t##S))r;8.0.0Ddeprecated since Redis 8.0, call config_set from core module insteadoptionrc>td||g}|j|}|dk(S)Set runtime configuration option. ### Parameters - **option**: the name of the configuration option. - **value**: a value for the configuration option. For more information see `FT.CONFIG SET `_. SETOK CONFIG_CMDrr5rCrr6rs r9 config_setzSearchCommands.config_sets.5&%0"d""C(d{r;Ddeprecated since Redis 8.0, call config_get from core module insteadc^td|g}|j|}|jt|S)Get runtime configuration option value. ### Parameters - **option**: the name of the configuration option. For more information see `FT.CONFIG GET `_. GETrIrr:r5rCr6r7s r9 config_getzSearchCommands.config_get+s55&)"d""C("":s33r;tagfieldcD|jt|j|S)z Return a list of all possible tag values ### Parameters - **tagfield**: Tag field name For more information see `FT.TAGVALS `_. )r TAGVALS_CMDr)r5rSs r9tagvalszSearchCommands.tagvals<s##K(KKr;aliascD|jt||jS)z Alias a search index - will fail if alias already exists ### Parameters - **alias**: Name of the alias to create For more information see `FT.ALIASADD `_. )r ALIAS_ADD_CMDrr5rWs r9aliasaddzSearchCommands.aliasaddIs##M5$//JJr;cD|jt||jS)z Updates an alias - will fail if alias does not already exist ### Parameters - **alias**: Name of the alias to create For more information see `FT.ALIASUPDATE `_. )rALIAS_UPDATE_CMDrrZs r9 aliasupdatezSearchCommands.aliasupdateVs##$4eT__MMr;c.|jt|S)z Removes an alias to a search index ### Parameters - **alias**: Name of the alias to delete For more information see `FT.ALIASDEL `_. )r ALIAS_DEL_CMDrZs r9aliasdelzSearchCommands.aliasdelcs##M599r;ch|jd}|D]}t||j|jg}|j dr|j d|j r,|j d|j |j |j||jdS)aJ Add suggestion terms to the AutoCompleter engine. Each suggestion has a score and string. If kwargs["increment"] is true and the terms are already in the server's dictionary, we increment their scores. For more information see `FT.SUGADD `_. F transaction incrementINCRr pipelineSUGADD_COMMANDstringrkrr[rrexecuter5r suggestionsr8pipesugrs r9sugaddzSearchCommands.sugaddos}}}/ (C"CSYY?Dzz+& F#{{ I& CKK( D $ ' (||~b!!r;rc.|jt|S)z Return the number of entries in the AutoCompleter index. For more information see `FT.SUGLEN `_. )rSUGLEN_COMMAND)r5rs r9suglenzSearchCommands.suglens ##NC88r;rkc0|jt||S)z Delete a string from the AutoCompleter index. Returns 1 if the string was found and deleted, 0 otherwise. For more information see `FT.SUGDEL `_. )rSUGDEL_COMMAND)r5rrks r9sugdelzSearchCommands.sugdels##NC@@r;prefixfuzzynumrG with_payloadsct||d|g}|r|jt|r|jt|r|jt|j |}g} |s| St |||} | D cgc]} | c} Scc} w)t Get a list of suggestions from the AutoCompleter, for a given prefix. Parameters: prefix : str The prefix we are searching. **Must be valid ascii or utf-8** fuzzy : bool If set to true, the prefix search is done in fuzzy mode. **NOTE**: Running fuzzy searches on short (<3 letters) prefixes can be very slow, and even scan the entire index. with_scores : bool If set to true, we also return the (refactored) score of each suggestion. This is normally not needed, and is NOT the original score inserted into the index. with_payloads : bool Return suggestion payloads num : int The maximum number of results we return. Note that we might return less. The algorithm trims irrelevant suggestions. Returns: list: A list of Suggestion objects. If with_scores was False, the score of all suggestions is 1. For more information see `FT.SUGGET `_. MAXSUGGET_COMMANDr[r+r*r,rr) r5rrxryrzrGr{rr7rVparserss r9suggetzSearchCommands.suggetsPVUC8  KK   KK #  KK %"d""D)N!+}cB!"a"""s7 Bgroupid skipinitialct|j|g}|r|jdg|j||j|S)a Updates a synonym group. The command is used to create or update a synonym group with additional terms. Only documents which were indexed after the update will be affected. Parameters: groupid : Synonym group id. skipinitial : bool If set to true, we do not scan and index. terms : The terms. For more information see `FT.SYNUPDATE `_. r)) SYNUPDATE_CMDrrr)r5rrr5r6s r9 synupdatezSearchCommands.synupdatesF$doow7  JJ)* + 5#t##S))r;cn|jt|j}|jt|S)a Dumps the contents of a synonym group. The command is used to dump the synonyms data structure. Returns a list of synonym terms and their synonym group ids. For more information see `FT.SYNDUMP `_. )r SYNDUMP_CMDrr:rs r9syndumpzSearchCommands.syndumps-"";@"";44r;)d) FFNNFNFFF)F)NF?NFFNF)NrNF)FrNFFNF)rNF)NFr=NNNNN)FNNNNF FF)I__name__ __module__ __qualname____doc__r:rBrNr`rdrirqruryr}rrrrrrrrrrrrrr\rrrrrrrrZbytesrrrrrr r r rrr r r rrrrrcr'r3r9r=r@rKrRrVr[r^rarqrtrwrrrrrsr;r9r.r.MsD !  ,V 2#J>C>!&$)-04"$)"'D+U D+D+ D+ DI& D+ _- D+D+"D+ D+L+uUDK-?'@+(+$+8(+VDI+( T "&"&3 3 3  3 $ 3  3 3 3-3 3 s)3 3 j T  "1RS+T+& )1RS ET E 2 $T#uS#ue5K/L*L%MN #+DeCeU@DRV!%.23?3?!!563?""<= 3? &d3c3u6L0M+M&NO 3? # 3?*+3? |/9 :3?pLP 8S%Z  8tCsC/E)F$FGH 8IsEz!2I LP %v-. tCsC/E)F$FGH @55 %&6&> ?5MQ52KO & U,,-& & tCsC/E)F$FGH & P88 *S *$s) * *S *$s) * *c *U  S T   U 4 4 4  4 L L Kc K N N :c :".9#9#9A#AsAsA!#6# 6#6# 6#  6#  6#6#  6#p**4*c*0 5r;r.ceZdZdZ d deeefdeeeeee e e fffdZ d!de deedeedeeeeee e e ffd ee d eed eeeeffd Z d deeefdeeeeee e e fffd Zd"dZedddeded efdZeddded efdZdZdZ d#dededede deded ee fdZ!y)$AsyncSearchCommandscK|jt|jd{}|jt|S7w)rNrrs r9rzAsyncSearchCommands.infos:((4??CC""8S11Ds$AAANrDrcpK|j||\}}tj}i}t|jdvr d|t <|j tg|i|d{}t|tr|S|jt||tj|z dzS7Gw)rrr0TNrrrrs r9rzAsyncSearchCommands.searchs ))%l)K e ^^   ,H <$(GL !(D((FdFgFF c8 $J"" 5DNN4Dr4IV3S#  GsA*B6,B4-AB6rrrrrQrc~K|j}i}t|g} | j|j|r| j|j|r| j|j |r | j|j ||r| jd|f|r$d|d<| j|j t |jdvr d|t<|j| i|d{} t| tr| S|jt| fi|S7.w)rrTrQr0Nrrs r9r z!AsyncSearchCommands.hybrid_search s6e$ enn&'  MM.113 4  MM/446 7  MM$../BC D  MM9g. /  $GH  MM&++- .  ,H <$(GL !(D((&g>> =sD D= D; /D=cKt|tr9t|j}t|j g|j z}nCt|tr'd}td|j g|j z}n td|||j|z }|j|d{}|jt|||S7w)rTrrNrrrs r9rzAsyncSearchCommands.aggregateUs" e- .emm,J $//2U5E5E5GGC v &Jvt7%:J:J:LLC[%0 0 t##L11(D((#.."" 3e #  /sB3C5C6CcKt|j|g}|r|jd|g|r|jdd|g|r|jdd|g|j|d{}|j t|S7w)a Issue a spellcheck query ### Parameters **query**: search query. **distance***: the maximal Levenshtein distance for spelling suggestions (default: 1, max: 4). **include**: specifies an inclusion custom dictionary. **exclude**: specifies an exclusion custom dictionary. For more information see `FT.SPELLCHECK `_. r)r*r+r,Nr-r/s r9r3zAsyncSearchCommands.spellcheckust6  JJ H- .  JJG4 5  JJG4 5(D((#.."">377/sA'B)B*BrArBrrCrcZKtd||g}|j|d{}|dk(S7 w)rErFNrGrHrJs r9rKzAsyncSearchCommands.config_sets:5&%0(D((#..d{/s +) +rLc~Ktd|g}i}|j|d{}|jt|S7w)rNrONrPrQs r9rRzAsyncSearchCommands.config_getsF5&)(D((#.."":s33/s =;=cK|jj|d{}|jDcic]\}}t|t|}}}|} |d=t dd|i|S7Lcc}}w#t$rY wxYww)rNrrsrrs r9rz!AsyncSearchCommands.load_documents{{**2..5;\\^ DTQilIaL( D D t (2(((/ D    sDBA.BA0BA6!B0B6 B?BBBcK|jd}|D]}t||j|jg}|j dr|j d|j r,|j d|j |j |j||jd{dS7w)aI Add suggestion terms to the AutoCompleter engine. Each suggestion has a score and string. If kwargs["increment"] is true and the terms are already in the server's dictionary, we increment their scores. For more information see `FT.SUGADD `_. FrcrerfrNrgrhrms r9rqzAsyncSearchCommands.sugadds}}}/ (C"CSYY?Dzz+& F#{{ I& CKK( D $ ' (lln$b))$sB4C6B>7CrrxryrzrGr{c,Kt||d|g}|r|jt|r|jt|r|jt|j |d{}g} |s| St |||} | D cgc]} | c} S7'cc} ww)r}r~Nr) r5rrxryrzrGr{rretrVrrs r9rzAsyncSearchCommands.suggetsPVUC8  KK   KK #  KK %(D(($//N!+}cB!"a"" 0 #s$A#B%B &B B BBr=rrr)"rrrrrrrrrrZr\rrrr r r rrr r rrrr3rrrKrRrrqrrrrsr;r9rrs 2LP S%Z  tCsC/E)F$FGH F:>@DRV!%.23?3?!!563?""<= 3? &d3c3u6L0M+M&NO 3? # 3?*+3? |/9 :3?pLP _f,- tCsC/E)F$FGH @88U s 3 4   U 4s 4s 4  4 )*6!#6# 6#6# 6#  6#  6#6#  6#r;r)Yrrtypingrrrrrredis._parsers.helpersr redis.clientr r "redis.commands.search.hybrid_queryr r r r#redis.commands.search.hybrid_resultrr redis.utilsrhelpersr_utilr aggregationrrrdocumentrfieldrindex_definitionrprofile_informationrrDrrhrrlrr rrrrrrr EXPLAINCLI_CMDrrr&rr.r7r<r?rrIrUrYr]r`rrjrvrsrrrrr"r#r$r%r&r'r(r)r*r+r,r.rrsr;r9rs? 330/ Q+* -3(                     #          #   d5d5N]#.]#r;