j2 UddlmZmZddlZddlmZddlmZ ddl m Z m Z m Z mZGddZGdd ZGd d ZGd d ZeezezZeed<de jdee jdejde jfdZde jdee jdejde jfdZdedee jdejde jfdZdedee jdejde jfdZdeedee jdejde jfdZdedee jdejde jfdZdedee jdejde jfdZdS))Any TypeAliasN)models) common_types)calculate_distancescaled_fast_sigmoidEPSILON fast_sigmoidceZdZ ddeeeedzdeeeedzdejdzfdZdS)MultiRecoQueryNpositivenegativestrategyc\| Jd||_||ng}||ng}|D]2}tj|r Jd3|D]2}tj|r Jd3d|D|_d|D|_dS)Nz#Recommend strategy must be providedz%Positive vectors must not contain NaNz%Negative vectors must not contain NaNc6g|]}tj|Snparray.0vectors QC:\Projects\DSCons\.venv\Lib\site-packages\qdrant_client\local\multi_distances.py z+MultiRecoQuery.__init__..""0Y0Y0Yf&1A1A0Y0Y0Yc6g|]}tj|Srrrs rrz+MultiRecoQuery.__init__..#rr)rrisnananyr r)selfr rrrs r__init__zMultiRecoQuery.__init__s ##%J###  '388'388 W WFx''++-- V V/V V V- V W WFx''++-- V V/V V V- V0Y0YPX0Y0Y0Y 0Y0YPX0Y0Y0Y r)NNN)__name__ __module__ __qualname__listfloatrRecommendStrategyr!rrrr r s483748 ZZtDK()D0ZtDK()D0Z*T1 ZZZZZZrr cNeZdZdeeedeeefdZdS)MultiContextPairr rc>tj||_tj||_tj|jr Jdtj|jr JddS)Nz$Positive vector must not contain NaNz$Negative vector must not contain NaN)rrr rrr)r r rs rr!zMultiContextPair.__init__'s*,(8*<*< *,(8*<*< 8DM**..00XX2XXX08DM**..00XX2XXX0XXrN)r"r#r$r%r&r!rrrr)r)&sPYd5k!2Yd4;>OYYYYYYrr)cBeZdZdeeedeefdZdS)MultiDiscoveryQuerytargetcontextctj||_||_tj|jr JddS)Nz"Target vector must not contain NaN)rrr-r.rr)r r-r.s rr!zMultiDiscoveryQuery.__init__0sM(*(8(8  8DK((,,..TT0TTT.TTrN)r"r#r$r%r&r)r!rrrr,r,/sMUtDK0U4@P;QUUUUUUrr,c&eZdZdeefdZdS)MultiContextQuery context_pairsc||_dSN)r2)r r2s rr!zMultiContextQuery.__init__8s*rN)r"r#r$r%r)r!rrrr1r17s5+d+;&<++++++rr1MultiQueryVector query_matrixmatrices distance_typereturnctj|r Jdt|jdks Jdt |||}|t jjkr'tj tj |}n)|t jj krtj |}|S)Nz!Query matrix must not contain NaNzQuery must be a matrix) rrrlenshapecalculate_multi_distance_corerDistanceEUCLIDsqrtabs MANHATTAN)r6r7r8 distancess rcalculate_multi_distancerE?s x %%))++PP-PPP + |! " "a ' ' ')A ' ' '-lHmTTI...GBF9--.. &/3 3 3F9%% rc dtjdtjdtdtjfd}dtjdtjdtdtjfd}tj|r Jdg}|t jjt jj fvr0|ddtj f}|t jjkr|n|}nt}|D]Y}||||}ttj tj|d } || Ztj|S) Nqm_r9ctj||z tjdtj SNdtype)axisrM)rsquarefloat32sumrGrHrIs r euclideanz0calculate_multi_distance_core..euclideanUs5 !a%rz22266Bbj6QQQQrctj||z tjdtj SrK)rrBrQrRrSs r manhattanz0calculate_multi_distance_core..manhattanXs5q1uBJ///332:3NNNNrz!Query vector must not contain NaNrNrO)types NumpyArrayrrrrrr?r@rCnewaxisrr&rRmaxappendr) r6r7r8rTrV similarities dist_funcmatrix sim_matrix similaritys rr>r>Pst RU%R%*:RRHXRRRROU%O%*:OOHXOOOOx %%))++PP-PPP + "L/1JKKK#AAArzM2 !.&/2H!H!HIIi & ((Y|V]CC 26"&""="="=>>?? J'''' 8L ! !!rquerycVdttjdtjffd }||j}||j}t j||kt jd|D|jt jd|D|jS)Nexamplesr9ctt}g}|D](}t|}||)t|dkr3|tj|tj tj|tjd}|SNrrLrW) r<r>r\rfullinfrrQr[)rd matrix_countscoresexamplescore best_scoresr8r7s rget_best_scoresz>calculate_multi_recommend_best_scores..get_best_scoresps8}} *, ! !G1'8]SSE MM%  v;;!   MM"',88 9 9 9hvRZ888<z8calculate_multi_recommend_best_scores..s+;;(,,;;;;;;rc36K|]}t| VdSr4rprqs rrsz8calculate_multi_recommend_best_scores..s.<<")"---<<<<<r\rzerosrrQrR)rdrirjrkrl sum_scoresr8r7s rget_sum_scoresz.get_sum_scoress8}} )+ ! !G1'8]SSE MM% v;;!   MM"(<00 1 1 1XfBJ777;;;CC r)r%rXrYr r)rbr7r8r~rwrxs `` r$calculate_multi_recommend_sum_scoresrsn e&6!7 E.s>   'Ix9$9AAr   r) rr|r<int32r>r rrzip)r.r7r8 overall_rankspairrwrx pair_rankss rcalculate_multi_discovery_ranksrs ')hs8}}BH&M&M&MM $ $+DM8]SS+DM8]SSX  +.sSy#*+E+E      # rct|j||}t|j||}t jd|Dtj}||zS)Nc34K|]}t|VdSr4rprqs rrsz3calculate_multi_discovery_scores..s+??R R ??????r)rr.r>r-rrvrQ)rbr7r8ranksdistances_to_targetsigmoided_distancess r calculate_multi_discovery_scoresrsd ,EM8] S SE8 hP]^^,.K??+>???-- & &&rchtjt|tj}|jD]|}t |j||}t |j||}||z tz }tj dtj |dDtj}||z }}|S)NrLc34K|]}t|VdSr4)r rqs rrsz1calculate_multi_context_scores..s* D D"\"   D D D D D Drg) rr|r<rQr2r>r rr rvminimum) rbr7r8overall_scoresrrwrx difference pair_scoress rcalculate_multi_context_scoresrs(*xH RZ'P'P'PN# & &+DM8]SS+DM8]SS3Y( (* D D :s(C(C D D Dbj) )  +% r)typingrrnumpyrqdrant_client.httprqdrant_client.conversionsrrXqdrant_client.local.distancesrrr r r r)r,r1r5__annotations__rYr%r?rEr>ryrrrrrrrrsL!!!!!!!!!%%%%%%;;;;;;ZZZZZZZZ.YYYYYYYYUUUUUUUU++++++++ 24EEV)VVV"5#$?  """"5#$"?"  """": %)%*:%;LRO > %)%*:%;LRO . " #5#$?  . '  '*.u/?*@ 'QWQ` '  ' ' ' ' (,U-=(>OU r