BiY( hUddlmZmZmZddlZddlmZddl m Z ddl m Z m Z mZddlmZmZmZmZGddZGd d ZGd d ZGd dZe ezezezZeed< d$de dee dedej8fdZde de dej<dzfdZdeedee dej8fdZ dedee dej8fdZ!dedee dej8fdZ"dedee dej8fdZ#dedee dej8fdZ$de de dede fdZ%dee de fd Z&d!e d"e de fd#Z'y)%)CallableSequence TypeAliasN) common_types) SparseVector)EPSILON fast_sigmoidscaled_fast_sigmoid)empty_sparse_vector is_sortedsort_sparse_vectorvalidate_sparse_vectorcveZdZ d deedzdeedzdej dzfdZdedgdfddfd Z y) SparseRecoQueryNpositivenegativestrategyc|Jd||_||ng}||ng}t|D]\}}t|t|||< t|D]\}}t|t|||< ||_||_y)Nz#Recommend strategy must be provided)r enumeraterr rr)selfrrrivectors f/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/qdrant_client/local/sparse_distances.py__init__zSparseRecoQuery.__init__s #J%JJ#  '38'38"8, 5IAv "6 *,V4HQK 5#8, 5IAv "6 *,V4HQK 5!   foorreturnc t|jDcgc] }|| c}|jDcgc] }|| c}|jScc}wcc}w)N)rrr)rrrr)rrrs rtransform_sparsez SparseRecoQuery.transform_sparse)sJ04 >fc&k>04 >fc&k>]]  >>s AA )NNN) __name__ __module__ __qualname__listrtypesRecommendStrategyrrrrrrrso/3.237 !|$t+!|$t+!))D0 !0 ^,n<=  rrceZdZdedefdZy)SparseContextPairrrcpt|t|t||_t||_yN)rr rr)rrrs rrzSparseContextPair.__init__4s*x(x(&8&B &8&B rN)r r!r"rrr&rrr(r(3sCCCrr(c>eZdZdedeefdZdedgdfddfdZy) SparseDiscoveryQuerytargetcontextcHt|t||_||_yr*)rr r-r.)rr-r.s rrzSparseDiscoveryQuery.__init__<sv&$6v$>  rrrrc t||j|jDcgc].}t||j||j 0c}Scc}w)N)r-r.)r,r-r.r(rrrrpairs rrz%SparseDiscoveryQuery.transform_sparseAsU$t{{#VZVbVbNR!#dmm"4c$--6HI  s3A N) r r!r"rr#r(rrrr&rrr,r,;s=|d;L6M  ^,n<=  rr,c:eZdZdeefdZdedgdfddfdZy)SparseContextQuery context_pairsc||_yr*r5)rr5s rrzSparseContextQuery.__init__Ms *rrrrc t|jDcgc].}t||j||j0c}Scc}w)Nr7)r4r5r(rrr1s rrz#SparseContextQuery.transform_sparsePsM"!.."#dmm"4c$--6HI  s3AN)r r!r"r#r(rrrr&rrr4r4Ls6+d+<&=+ ^,n<=  rr4SparseQueryVectorqueryvectors empty_is_zerorc4g}|D]m}t||}||j|#|s%|jtjdJ|jtjdotj|tjS)a&Calculate distances between a query sparse vector and a list of sparse vectors. Args: query (SparseVector): The query sparse vector. vectors (list[SparseVector]): A list of sparse vectors to compare against. empty_is_zero (bool): If True, distance between vectors with no overlap is treated as zero. Otherwise, it is treated as negative infinity. Simple nearest search requires `empty_is_zero` to be False, while methods like recommend, discovery, and context search require True. z-infdtype)sparse_dot_productappendnpfloat32array)r:r;r<scoresrscores rcalculate_distance_sparserH`syF+"5&1   MM%  MM"**V, - MM"**S/ *+ 88F"** --rvector1vector2c>d}d\}}d}t|sJdt|sJd|t|jkr|t|jkr|j||j|k(r/d}||j||j|zz }|dz }|dz }n*|j||j|kr|dz }n|dz }|t|jkr|t|jkr|rt j |Sy)Nr>rrFz"Query sparse vector must be sortedz,Sparse vector to compare with must be sortedT)r lenindicesvaluesrCrD)rIrJresultrjoverlaps rrArA~s F DAqG W CCC  W MMM  c'//" "q3w+?'? ??1 !3 3G gnnQ''..*;; ;F FA FA __Q '//!"4 4 FA FA c'//" "q3w+?'?zz&!!rr.c ftjt|tj}|D]w}t |j |d}t |j |d}tjt||kD||k(Dcgc]\}}|rdn|rdndc}}}||z }y|Scc}}w)Nr?Tr<rMr) rCzerosrNint32rHrrrEzip) r.r; overall_ranksr2posneg is_biggeris_equal pair_rankss r calculate_sparse_discovery_ranksr`s')hhs7|288&LM $' wdS' wdSXX,/sSy#*+E 'IxAr9   #  $  sB- ct|j|}t|j|d}t j d|Dtj }||zS)NTrUc32K|]}t|ywr*r .0xis r z4calculate_sparse_discovery_scores..s?R R ?)r`r.rHr-rCfromiterrD)r:r;ranksdistances_to_targetsigmoided_distancess r!calculate_sparse_discovery_scoresrmsV -U]]G DE4ELL'Y]^++?+>? & &&rctjt|tj}|jD]}t |j |d}t |j|d}||z tz }tjdtj|dDtj}||z }|S)Nr?TrUc32K|]}t|ywr*)r rds rrgz2calculate_sparse_context_scores..s D"\"  Drhr>) rCrWrNrDr5rHrrrriminimum)r:r;overall_scoresr2r[r\ difference pair_scoress rcalculate_sparse_context_scoresrts(*xxG BJJ'ON## &' wdS' wdS3Y( kk D :s(C Dbjj  +% & rcZdttdtjffd }||j}||j }t j||kDt jd|D|jt jd|D|jS)Nexamplesrc\t}g}|D]!}t|d}|j|#t|dk(r4|jtj|tj tj |tjjd}|SNTrUrr?)axis) rNrHrBrCfullinfrErDmax)rv vector_countrFexamplerG best_scoresr;s rget_best_scoresz?calculate_sparse_recommend_best_scores..get_best_scoress7| *, !G-gwdSE MM%  ! v;!  MM"'',8 9hhvRZZ8<.s;(,;rhc34K|]}t| ywr*rcrds rrgz9calculate_sparse_recommend_best_scores..s<")"--.get_sum_scoress7| )+ !G-gwdSE MM%  ! v;!  MM"((<0 1XXfBJJ7;;;C r)r#rr$rrr)r:r;rr[r\s ` r%calculate_sparse_recommend_sum_scoresrsI l!3 8H8H   (C  (C 9ropct}d\}}|t|jkr|t|jkr|j||j|k(ro|jj|j||jj||j||j||dz }|dz }n|j||j|kr]|jj|j||jj||j|d|dz }n\|jj|j||jj|d|j||dz }|t|jkr|t|jkr|t|jkru|jj|j||jj||j|d|dz }|t|jkru|t|jkru|jj|j||jj|d|j||dz }|t|jkru|S)NrLrMr>)r rNrOrBrP)rIrJrrQrrRs rcombine_aggregatersN "F DAq c'//" "q3w+?'? ??1 !3 3 NN ! !'//!"4 5 MM GNN1$5w~~a7H!I J FA FA __Q '//!"4 4 NN ! !'//!"4 5 MM GNN1$5s!; < FA NN ! !'//!"4 5 MM C):!; < FA c'//" "q3w+?'? c'//" "gooa01 Rq 1378 Q c'//" " c'//" "gooa01 RW^^A%678 Q c'//" " Mrct}t|dk(r|Sd}|D]}|dz }t||d}tj|j |j |_|S)NrrMc ||zSr*r&)v1v2s rzsparse_avg..-s "r'r)r rNrrCdividerPtolist)r;rQ sparse_countrs r sparse_avgr%sp "F 7|q LK "663IJKIIfmm\:AACFM Mrrrct||dS)Nc||z|z Sr*r&)r[r\s rrz1merge_positive_and_negative_avg..7s#)c/r)r)rrs rmerge_positive_and_negative_avgr4s Xx1Q RRr)F)(typingrrrnumpyrCqdrant_client.conversionsrr$qdrant_client.http.modelsrqdrant_client.local.distancesrr r qdrant_client.local.sparser r r rrr(r,r4r9__annotations__r#boolrrHrDrAr`rmrtrrrrrr&rrrs00;2TT   FCC  "   ''*<<N9 MR. ."&|"4.EI. .< | UYHY2 # $ ,  , '  '*.|*< '  ' (,\(: $ %),%7 > %),%7 0|lUa@ . < SS&2SSr