Bi6\UddlmZmZddlZddlmZddlmZ ddl m Z m Z m Z mZGddZGdd ZGd d ZGd d ZeezezZeed<de j*dee j*dej.de j*fdZde j*dee j*dej.de j*fdZdedee j*dej.de j*fdZdedee j*dej.de j*fdZdeedee j*dej.de j*fdZdedee j*dej.de j*fdZdedee j*dej.de j*fdZy))Any TypeAliasN)models) common_types)calculate_distancescaled_fast_sigmoidEPSILON fast_sigmoidcreZdZ ddeeeedzdeeeedzdej dzfdZy)MultiRecoQueryNpositivenegativestrategyc|Jd||_||ng}||ng}|D],}tj|js'Jd|D],}tj|js'Jd|Dcgc]}tj|c}|_|Dcgc]}tj|c}|_ycc}wcc}w)Nz#Recommend strategy must be providedz%Positive vectors must not contain NaNz%Negative vectors must not contain NaN)rnpisnananyarrayr r)selfr rrvectors e/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/qdrant_client/local/multi_distances.py__init__zMultiRecoQuery.__init__s #J%JJ#  '38'38 WFxx'++- V/V V- W WFxx'++- V/V V- WQY0Yf&1A0Y PX0Yf&1A0Y 1Z0Ys C+C)NNN)__name__ __module__ __qualname__listfloatrRecommendStrategyrrr r sc483748 ZtDK()D0ZtDK()D0Z**T1 Zr r c4eZdZdeeedeeefdZy)MultiContextPairr rc<tj||_tj||_tj|jj rJdtj|jj rJdy)Nz$Positive vector must not contain NaNz$Negative vector must not contain NaN)rrr rrr)rr rs rrzMultiContextPair.__init__'so*,((8*< *,((8*< 88DMM*..0X2XX088DMM*..0X2XX00r N)rrrrrrrr rr"r"&s*Yd5k!2Yd4;>OYr r"c.eZdZdeeedeefdZy)MultiDiscoveryQuerytargetcontextctj||_||_tj|jj rJdy)Nz"Target vector must not contain NaN)rrr&r'rr)rr&r's rrzMultiDiscoveryQuery.__init__0sA(*(8  88DKK(,,.T0TT..r N)rrrrrr"rrr rr%r%/s'UtDK0U4@P;QUr r%ceZdZdeefdZy)MultiContextQuery context_pairsc||_yN)r+)rr+s rrzMultiContextQuery.__init__8s *r N)rrrrr"rrr rr*r*7s+d+;&<+r r*MultiQueryVector query_matrixmatrices distance_typereturnctj|jrJdt|jdk(sJdt |||}|t jjk(r*tjtj|}|S|t jjk(rtj|}|S)Nz!Query matrix must not contain NaNzQuery must be a matrix) rrrlenshapecalculate_multi_distance_corerDistanceEUCLIDsqrtabs MANHATTAN)r/r0r1 distancess rcalculate_multi_distancer>?s xx %))+P-PP + |!! "a 'A)AA '-lHmTI...GGBFF9-.   &//33 3FF9% r c dtjdtjdtdtjfd}dtjdtjdtdtjfd}tj|j rJdg}|t jjt jjfvr9|ddtjf}|t jjk(r|n|}nt}|D]P}||||}ttjtj|d } |j| Rtj |S) Nqm_r2ctj||z tjjdtj SNdtype)axisrF)rsquarefloat32sumr@rArBs r euclideanz0calculate_multi_distance_core..euclideanUs2 !a%rzz266Bbjj6QQQr ctj||z tjjdtj SrD)rr;rJrKrLs r manhattanz0calculate_multi_distance_core..manhattanXs2q1uBJJ/332::3NNNr z!Query vector must not contain NaNrGrH)types NumpyArrayrrrrrr8r9r<newaxisrrrKmaxappendr) r/r0r1rMrO similarities dist_funcmatrix sim_matrix similaritys rr7r7PsF RU%%R%*:*:RRHXHXROU%%O%*:*:OOHXHXOxx %))+P-PP + "L//1J1JKK#ArzzM2 !.&//2H2H!HIi & (|V]C 266"&&""=>? J'( 88L !!r querycrdttjdtjffd }||j}||j}t j ||kDt jd|D|jt jd|D|jS)Nexamplesr2cZt}g}|D] }t|}|j|"t|dk(r4|jtj|tj tj |tjjd}|SNrrErP) r5r7rUrfullinfrrJrT)r] matrix_countscoresexamplescore best_scoresr1r0s rget_best_scoresz>calculate_multi_recommend_best_scores..get_best_scoresps8} *, !G1'8]SE MM%  ! v;!  MM"'',8 9hhvRZZ8<z8calculate_multi_recommend_best_scores..s;(,;c34K|]}t| ywr-rirjs rrmz8calculate_multi_recommend_best_scores..s<")"--.get_sum_scoress8} )+ !G1'8]SE MM%  ! v;!  MM"((<0 1XXfBJJ7;;;C r )rrQrRr r)r[r0r1ryrrrss `` r$calculate_multi_recommend_sum_scoresrzsO e&6&6!7 E.s?R R ?rn)rr'r7r&rrqrJ)r[r0r1ranksdistances_to_targetsigmoided_distancess r calculate_multi_discovery_scoresrsX ,EMM8] SE8 hP]^++?+>? & &&r ctjt|tj}|jD]}t |j ||}t |j||}||z tz }tjdtj|dDtj}||z }|S)NrEc32K|]}t|ywr-)r rjs rrmz1calculate_multi_context_scores..s D"\"  Drng) rrwr5rJr+r7r rr rqminimum) r[r0r1overall_scoresrrrrs difference pair_scoress rcalculate_multi_context_scoresrs(*xxH RZZ'PN## &+DMM8]S+DMM8]S3Y( kk D :s(C Dbjj  +% & r )typingrrnumpyrqdrant_client.httprqdrant_client.conversionsrrQqdrant_client.local.distancesrrr r r r"r%r*r.__annotations__rRrr8r>r7rtrzrrrrr rrs!!%;ZZ.YYUU++ 24EEV)V""5##$??  """""5##$"??"  ": %)%*:*:%;LROO > %)%*:*:%;LROO . " #5##$??  . '  '*.u/?/?*@ 'QWQ`Q` '  ' (,U-=-=(>OU r