Bi`1UddlmZddlmZddlmZddlZddlm Z ddl m Z dZ dZGd d eeZGd d ZGd dZGddZGddZGddZGddZGddZGddZeezezezZeed<de j6defdZde j:d e j:de j:fd!Zde j:d e j:de j:fd"Zde j:d e j:de j:fd#Z de j:d e j:de j:fd$Z!de j:d e j:d%e j6de j:fd&Z"de j:d e j:d%e j6de j:fd'Z#d(ejHdejHfd)Z%d(ejHdejHfd*Z&ded e j:d%e j6de j:fd+Z'ded e j:d%e j6de j:fd,Z(d-e)ed e j:d%e j6de j:fd.Z*ded e j:d%e j6de j:fd/Z+ded e j:d%e j6de j:fd0Z,ded e j:d%e j6de j:fd1Z-y)2)Enum) TypeAlias) permutationsN) common_types)modelsg>ceZdZdZdZy) DistanceOrderbigger_is_bettersmaller_is_betterN)__name__ __module__ __qualname__BIGGER_IS_BETTERSMALLER_IS_BETTER_/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/qdrant_client/local/distances.pyr r s)+rr cfeZdZ ddeeedzdeeedzdej dzfdZy) RecoQueryNpositivenegativestrategyc|Jd||_||ng}||ng}|Dcgc]}tj|c}|_|Dcgc]}tj|c}|_tj |jj rJdtj |jj rJdycc}wcc}w)Nz#Recommend strategy must be providedz%Positive vectors must not contain NaNz%Negative vectors must not contain NaN)rnparrayrrisnanany)selfrrrvectors r__init__zRecoQuery.__init__s #J%JJ#  '38'38PX0Yf&1A0Y PX0Yf&1A0Y 88DMM*..0Y2YY088DMM*..0Y2YY00 1Z0Ys C C)NNN)r rrlistfloatrRecommendStrategyr!rrrrrsY.2-148 ZtE{#d*ZtE{#d*Z**T1 Zrrc(eZdZdeedeefdZy) ContextPairrrc<tj||_tj||_tj|jj rJdtj|jj rJdy)Nz$Positive vector must not contain NaNz$Negative vector must not contain NaN)rrrrrr)rrrs rr!zContextPair.__init__*so*,((8*< *,((8*< 88DMM*..0X2XX088DMM*..0X2XX00rNr rrr"r#r!rrrr&r&)s YeYU Yrr&c(eZdZdeedeefdZy)DiscoveryQuerytargetcontextctj||_||_tj|jj rJdy)N"Target vector must not contain NaN)rrr+r,rr)rr+r,s rr!zDiscoveryQuery.__init__3sA(*(8  88DKK(,,.T0TT..rN)r rrr"r#r&r!rrrr*r*2s!UtE{UT+5FUrr*ceZdZdeefdZy) ContextQuery context_pairsc||_yN)r1)rr1s rr!zContextQuery.__init__;s *rN)r rrr"r&r!rrrr0r0:s+d;&7+rr0c"eZdZdeedefdZy) FeedbackItemr scorectj||_||_tj|jj rJdy)N$Feedback vector must not contain NaN)rrr r6rr)rr r6s rr!zFeedbackItem.__init__@sAhhv&  88DKK(,,.V0VV..rNr(rrrr5r5?sWtE{W5Wrr5c eZdZdededefdZy)NaiveFeedbackCoefficientsabcc.||_||_||_yr3)r;r<r=)rr;r<r=s rr!z"NaiveFeedbackCoefficients.__init__GsrN)r rrr#r!rrrr:r:Fs%Eerr:cHeZdZdejdejdefdZy)FeedbackContextPairrrpartial_computationc.||_||_||_yr3rrrA)rrrrAs rr!zFeedbackContextPair.__init__Ns!   #6 rN)r rrtypes NumpyArrayr#r!rrrr@r@Ms+7((7494D4D7[`7rr@c,eZdZdeedeedefdZy)NaiveFeedbackQueryr+feedback coefficientscFtj||_||_||_tj |jj rJd|jD]6}tj |jj s1Jdy)Nr.r8)rrr+rHrIrrr )rr+rHrIitems rr!zNaiveFeedbackQuery.__init__Ws hhv&   (88DKK(,,.T0TT.MM [Dxx ,002 Z4Z Z2 [rN)r rrr"r#r5r:r!rrrrGrGVs0 [U  [|$ [0 [rrGDenseQueryVectordistancereturnc|tjjk(rtjS|tjj k(rtjStj S)zn Convert distance to order Args: distance: distance to convert Returns: order )rDistanceEUCLIDr r MANHATTANr)rMs rdistance_to_orderrSisL6??)))... V__.. ....  ) ))rqueryvectorscztjj|dddtjf}|tj|dk7|t z}t |jdk(rVtjj|}|tj|dk7|t z}tj||Stjj|dddtjf}|tj|dk7|t z}tj||jS)z Calculate cosine distance between query and vectors Args: query: query vector vectors: vectors to calculate distance with Returns: distances axisNr) rlinalgnormnewaxiswhereEPSILONlenshapedotT)rTrU vectors_norm query_norms rcosine_similarityrfys99>>'>3ArzzMBL rxx +\7CCG 5;;1YY^^E*  *+ZAAvvgu%%B/2:: >J RXXjC'W ==E 66% ##rct|jdk(rtj||Stj||jS)z Calculate dot product between query and vectors Args: query: query vector. vectors: vectors to calculate distance with Returns: distances rZ)r`rarrbrcrTrUs r dot_productris; 5;;1vvgu%%vveWYY''rct|jdk(r$tjj ||z dStjj ||ddtj fz dS)z Calculate euclidean distance between query and vectors Args: query: query vector. vectors: vectors to calculate distance with Returns: distances rZrWrXN)r`rarr[r\r]rhs reuclidean_distancerksY 5;;1yy~~goB~77yy~~gam(<<2~FFrct|jdk(r-tjtj||z dStjtj||ddtj fz dS)z Calculate manhattan distance between query and vectors Args: query: query vector. vectors: vectors to calculate distance with Returns: distances rZrWrXN)r`rarsumabsr]rhs rmanhattan_distanceros_ 5;;1vvbffWu_-B77vvbffWuQ ]';;<2FFr distance_typectj|jrJd|tjj k(r t ||S|tjjk(r t||S|tjjk(r t||S|tjjk(r t||Std|)N!Query vector must not contain NaNzUnknown distance type )rrrrrPCOSINErfDOTrirQrkrRro ValueErrorrTrUrps rcalculate_distancerwsxx""$I&II $... 00 &//-- -5'** &//00 0!%11 &//33 3!%111-ABBrctj|jrJd|tjj k(rHtj ||z tjjdtj S|tjjk(rHtj||z tjjdtj St|||S)z` Calculate same internal distances as in core, rather than the final displayed distance rrdtyperZ)rYrz) rrrrrPrQsquarefloat32rmrRrnrwrvs rcalculate_distance_corer}s xx""$I&II $... 'E/<@@arzz@ZZZ111wbjj9==1BJJ=WWW!%-@@rxctj|stj|r|S|tjdt |z S)N?)rrisinfaddrnr~s r fast_sigmoidrs6 xx{bhhqk rvvc3q6" ""rcFdtjt|dzS)Ng?r)rrrrs rscaled_fast_sigmoidrs "&&a#. //rcrdttjdtjffd }||j}||j}t j ||kDt jd|D|jt jd|D|jS)NexamplesrNcbjd}g}|D] }t|}|j|"t|dk(r4|jt j |tj t j|tjjd}|SNrryrX) rar}appendr`rfullinfrr|max)r vector_countscoresexampler6 best_scoresrprUs rget_best_scoresz8calculate_recommend_best_scores..get_best_scoress}}Q' *, !G+GWmLE MM%  ! v;!  MM"'',8 9hhvRZZ8<z2calculate_recommend_best_scores..s;(,;c34K|]}t| ywr3rrs rrz2calculate_recommend_best_scores..s<")"--.get_sum_scores s}}Q' )+ !G+GWmLE MM%  ! v;!  MM"((<0 1XXfBJJ7;;;C r)r"rDrErr)rTrUrprrrs `` rcalculate_recommend_sum_scoresrsO e&6&6!7 EM $%dmmWmL%dmmWmLXX,/sSy#*+E 'IxAr9   #  $  s B/ ct|j||}t|j||}t j d|Dtj }||zS)Nc32K|]}t|ywr3rrs rrz-calculate_discovery_scores..?s?R R ?r)rr,r}r+rrr|)rTrUrpranksdistances_to_targetsigmoided_distancess rcalculate_discovery_scoresr6sW &emmWm LE2%,,W++?+>? & &&rctj|jdtj}|jD]}t |j ||}t |j||}||z tz }tjdtj|dDtj}||z }|S)Nrryc32K|]}t|ywr3)rrs rrz+calculate_context_scores..Ps D"\"  Drr) rrrar|r1r}rrr_rminimum) rTrUrpoverall_scoresrrr difference pair_scoress rcalculate_context_scoresrEsXXgmmA.bjjAN## &%dmmWmL%dmmWmL3Y( kk D :s(C Dbjj  +% & rcg}t|jdk\rt|jdD]}|d|d}}|j|jz }|tkr0||j j z|j jz}|jt|j|j||j jt|j||z} |D]G} t| j||} t| j||} | | z } | | j | zz } I| S)NrrZrC)r`rHrr6 NAIVE_FEEDBACK_CONFIDENCE_MARGINrIr<r=rr@r r;r}r+rrrA)rTrUrpr1p positive_item negative_item confidencerAr6rsim_possim_negdeltas rcalculate_naive_feedback_queryrWs:02M 5>>aenna0 A+,Q41=M&,,}/B/BBJ==#-u/A/A/C/C#CuGYGYG[G["[   #*11*11(;      #:5<<R_#` `E2)$---P)$---P'! ))E11 2 Lr).enumrtypingr itertoolsrnumpyrqdrant_client.conversionsrrDqdrant_client.httprr_rstrr rr&r*r0r5r:r@rGrL__annotations__rPrSrErfrirkrorwr}r|rrrrr"rrrrrrrrsu";% $' ,C, ZZ(YYUU++ WW77 [ [ -|;iGJ\\)\ * *M * $U--$8H8H$UM]M]$, (u'' (%2B2B (uGWGW ( Ge.. G9I9I GeN^N^ G Ge.. G9I9I GeN^N^ GC   C&+&6&6CGMC C" A    A&+&6&6 AGM A  A #BJJ#2::#02::0"**0 $//@F > $//@F 0 +    ??  . '  '$)$4$4 'EK__ '  ' "'"2"2CI?? $ (-(8(8IO r