Q i7vddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl m Z ddlmZmZmZmZmZmZmZmZddlmZddlmZdd lmZdd lmZed Zd efd Z d e!fdZ"ded eeeffdZ#de$de$d dfdZ%ded efdZ&ded efdZ'd)dedeed efdZ(edZ)d*deedeefdZ*d*deedeefdZ+degeeeeffd egdffdZ,d e!d e!fd!Z-d e!d e!fd"Z.d e!d e!fd#Z/d$ed%ed eefd&Z0d'ed efd(Z1dS)+N)contextmanager)Enum)wrapstime)AnyCallable CoroutineDictOptionalSequenceTypeVarcast)warn) BaseModel)Redis)ULIDTreturnc8ttS)z>Generate a unique identifier to group related Redis documents.)strrC:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\redisvl/utils/utils.py create_ulidrs tvv;;rctS)z=Generate a unix epoch timestamp to assign to Redis documents.rrrrcurrent_timestamprs 66Mrmodelcfd|dd}|D]\}}|||<|S)z Custom serialization function that converts a Pydantic model to a dict, serializing Enum fields to their values, and handling nested models and lists. ct|tr|jSt|tr7|dd}fd|DSt|tr fd|DSt|trfd|DS|S)NTF exclude_noneexclude_defaultsc.i|]\}}||Srr.0keyvalueserialize_items r z9model_to_dict..serialize_item..,s)UUU:3C..UUUrc.i|]\}}||Srrr%s rr*z9model_to_dict..serialize_item...s)NNN:3C..NNNrc&g|] }|Srr)r&elementr)s r z9model_to_dict..serialize_item..0s#@@@NN7++@@@r) isinstancerr(lowerr model_dumpitemsdictlist)item nested_datar)s rr)z%model_to_dict..serialize_item&s dD ! ! :##%% % i ( ( //te/TTKUUUUARARATATUUU U d # # NNNNNNN N d # # @@@@4@@@ @KrTFr!)r1r2)rserialized_datar'r(r)s @r model_to_dictr8 sr     &&D5&QQO%++--55 U-~e44 rv1v2c@||krtd|d|ddS)z(Check the equality of vector dimensions.z6Invalid vector dimensions! Vector has dims defined as z!Vector field has dims defined as z@Vector dims must be equal in order to perform similarity search.N) ValueError)r9r:s rvalidate_vector_dimsr=;s= Rxx L L L 4 4 4 N   xrdatac*tj|S)z Serlize the input into a string.)jsondumpsr>s r serializerCE :d  rc*tj|S)z$Deserialize the input from a string.)r@loadsrBs r deserializerGJrDrargument replacementc8dd|r d|dz fd}|S)a` Decorator to warn if a deprecated argument is passed. When the wrapped function is called, the decorator will warn if the deprecated argument is passed as an argument or keyword argument. NOTE: The @deprecated_argument decorator should not fall "outside" of the @classmethod decorator, but instead should come between it and the method definition. For example: class MyClass: @classmethod @deprecated_argument("old_arg", "new_arg") @other_decorator def test_method(cls, old_arg=None, new_arg=None): pass z Argument z= is deprecated and will be removed in the next major release.z Use z instead.c(tttfrVjt fd}ttrt|St|St fd}|S)Nc|vrttdnAtj}|j|i|}|jvrttd|i|SN) stacklevelrDeprecationWarninginspect signaturebind arguments)argskwargssig bound_argsrHmessage underlyings r inner_wrappedz=deprecated_argument..decorator..inner_wrappedjsv%%"4CCCCC!+J77C!)4!:6!:!:J:#777W&8QGGGG!z426222rc|vrttdnAtj}|j|i|}|jvrttd|i|SrMrP)rVrWrXrYrHfuncrZs r inner_normalz.decorator..inner_normal{sv%%"4CCCCC!+D11C!)4!:6!:!:J:#777W&8QGGGGtT,V,,,r)r/ classmethod staticmethod__func__r)r^r\r_r[rHrZs` @r decoratorz&deprecated_argument..decoratores d[,7 8 8 J :   3 3 3 3 3 3  3$ ,, 3"=111#M2224[[ - - - - - -[ - rr)rHrIrcrZs` @rdeprecated_argumentrdOs\$b(aaaG21;1111! ! ! ! ! ! F rc#Ktj5tjddVddddS#1swxYwYdS)zH Assert that a function does not emit any warnings when called. errorN)warningscatch_warnings simplefilterrrrassert_no_warningsrjs  " "g&&& s<AAnamecfd}|S)z Decorator to mark a function as deprecated. When the wrapped function is called, the decorator will log a deprecation warning. cnpj}d|drz tfd}|S)Nz Function > is deprecated and will be removed in the next major release. cBttd|i|S)NcategoryrOrrQ)rVrWr^warning_messages rwrapperz7deprecated_function..decorator..wrappers. +=! L L L L4((( (r)__name__r)r^fn_namerurtrkrIs` @rrcz&deprecated_function..decoratorsu'$- 2 2 2 2   + { *O t ) ) ) ) )  )rrrkrIrcs`` rdeprecated_functionrys* rcfd}|S)a Decorator to mark a class as deprecated. When the decorated class is instantiated, the decorator will emit a deprecation warning. Args: name: Optional custom name for the class in the warning message. If not provided, uses the class's __name__. replacement: Optional message describing what to use instead. Example: @deprecated_class(replacement="Use NewClass instead.") class OldClass: pass cp|j}d|drz |jtfd}||_|S)NzClass rncNttd|g|Ri|dS)NrNrqrs)selfrVrW original_initrts rnew_initz5deprecated_class..decorator..new_inits? +=! L L L L M$ 0 0 0 0 0 0 0 0 0r)rv__init__r)cls class_namerr~rtrkrIs @@rrcz#deprecated_class..decorators)S\  2Z 2 2 2   + { *O }   1 1 1 1 1   1   rrrxs`` rdeprecated_classrs*$& rfncfd}|S)NctttdddS tj}n#t$rd}Ynt $rYdSwxYw ||s0tdStj}tj|| }| |dS#tttf$r }Yd}~dSd}~wt $rYdSwxYw)N _getframe) sysgetattrasyncioget_running_loop RuntimeError Exception is_runningnew_event_loopset_event_loop create_taskrun_until_completeAttributeError TypeError)looptaskers rruzsync_wrapper..wrappers2 ;'#{D99A F +--DD   DDD    FF  |4??#4#4|?F-//&t,,,##BBDD))D  # #D ) ) ) ) )ni8    FFFFF    FF s46 A AAC4ACC=, C=<C=r)rrus` r sync_wrapperrs$"""""H Nrr(c.td|z dz dS)zL Normalize a cosine distance to a similarity score between 0 and 1. rNrmaxr(s rnorm_cosine_distancers E Q " ""rc.tdd|zz dS)zb Denormalize a similarity score between 0 and 1 to a cosine distance between 0 and 2. rNrrrs rdenorm_cosine_distancers q1u9}a  rcdd|zz S)z$ Normalize the L2 distance. rrs rnorm_l2_distancers E ?r redis_clientpatterncfddlm}|t||S)a Scan the Redis database for keys matching a specific pattern. Args: redis (Redis): The Redis client instance. pattern (str): The pattern to match keys against. Returns: List[str]: A dictionary containing the keys and their values. r) convert_bytes)match)redisvl.redis.utilsrr4 scan_iter)rrrs rscan_by_patternrs?211111 =l4474CCDD E EEr module_pathc|d}|d}|tjvr%t|dkrtj|SGdd}||S)a Lazily import a module or object from a module only when it's actually used. This function helps reduce startup time and avoid unnecessary dependencies by only importing modules when they are actually needed. Args: module_path (str): The import path, e.g., "numpy" or "numpy.array" Returns: Any: The imported module or object, or a proxy that will import it when used Examples: >>> np = lazy_import("numpy") >>> # numpy is not imported yet >>> array = np.array([1, 2, 3]) # numpy is imported here >>> array_func = lazy_import("numpy.array") >>> # numpy is not imported yet >>> arr = array_func([1, 2, 3]) # numpy is imported here .rrcDeZdZdefdZdZdedefdZdededefd Zd S) lazy_import..LazyModulercV||_d|_|d|_dS)Nr) _module_path_modulesplit_parts)r}rs rrz(lazy_import..LazyModule.__init__Hs* +D DL%++C00DKKKrcb|j|jS |jd}tj|}t |jdkr ||_|S|}|jddD]%} t ||}#t $rYdSwxYw||_|S#t$r}||_Yd}~dSd}~wwxYw)z+Import the module or attribute on first useNrr) rr importlib import_modulelenrr ImportError _import_error)r}base_module_namemoduleobjpartrs r_import_modulez.lazy_import..LazyModule._import_moduleMs|'|# #';q> "01ABBt{##q((#)DL!M KO$$D$%c400)$$$#ttt$ #     %&"ttttt sBABB&A76B7 BBB B B.B))B.rkrcn|jj|}|Td|jvrtd|jd|jt d|jdd|jddt|j|rt|j|St d|jd|d) NrFailed to lazily import : zmodule 'r' has no attribute 'r') rr__dict__rrrrrhasattrr)r}rkrs r __getattr__z+lazy_import..LazyModule.__getattr__ls|#,,..>&$-77)`t7H``DL^`` -\t{1~\\4;WX>\\\ t|T** 3t|T222!I4,II$III rrVrWc d|jr|}|\d|jvrtd|jd|jtd|jd|jdd|jdd t|jr |j|i|St|jd ) NrrrzFailed to find z : module 'rrrrz is not callable) rrrrrrrcallabler)r}rVrWrs r__call__z(lazy_import..LazyModule.__call__s|#,,..>&$-77)`t7H``DL^`` *Ad.?AA4;WX>AAosoz{|o}AAA  %% 5#t|T4V444t0BBBCC CrN) rv __module__ __qualname__rrrrrrrrr LazyModulerGs 1 1 1 1 1    > C C    6 D# D D D D D D D Drr)rrmodulesr)rpartstop_module_namers r lazy_importr)s,   c " "EAhO#+%%#e**//{?++WDWDWDWDWDWDWDWDr :k " ""r)N)NN)2rrrRr@loggingrrg contextlibrenumr functoolsrrtypingrr r r r r rrrpydanticrredisrulidrrrrfloatrr8intr=rCrGrdrjryrrrrrrrrrrrs  %%%%%%TTTTTTTTTTTTTTTTTTTT GCLLS 5 tCH~6 S c d    CC cc 99#9HSM9X9999xhsm#6%%8C=%hsm%%%%P%Xb)CcM"::;%T@R%%%%P##%####!%!E!!!!EeFF Fc]FFFF&w#Sw#Sw#w#w#w#w#w#r