o :i'@s8UddlmZddlmZddlZddlmZddl m Z dZ Gddde eZ Gd d d ZGd d d ZGd ddZGdddZeeBeBZeed<de jde fddZdejdejdejfddZdejdejdejfddZdejdejdejfddZdejdejdejfddZdejdejd e jdejfd!d"Zdejdejd e jdejfd#d$Zd%ejdejfd&d'Zd%ejdejfd(d)Zdedejd e jdejfd*d+Z dedejd e jdejfd,d-Z!d.e"edejd e jdejfd/d0Z#dedejd e jdejfd1d2Z$dedejd e jdejfd3d4Z%dS)5)Enum) TypeAliasN) common_types)modelsg>c@seZdZdZdZdS) DistanceOrderbigger_is_bettersmaller_is_betterN)__name__ __module__ __qualname__BIGGER_IS_BETTERSMALLER_IS_BETTERrrj/home/aiteam/pcaa-dev/dashboard-backend/venv/lib/python3.10/site-packages/qdrant_client/local/distances.pyr src@sHeZdZ   ddeeedBdeeedBdejdBfddZdS) RecoQueryNpositivenegativestrategycCs|dusJd||_|dur|ng}|dur|ng}dd|D|_dd|D|_t|jr7Jdt|jrCJddS)Nz#Recommend strategy must be providedcSg|]}t|qSrnparray.0vectorrrr z&RecoQuery.__init__..cSrrrrrrrr rz%Positive vectors must not contain NaNz%Negative vectors must not contain NaN)rrrrisnanany)selfrrrrrr__init__szRecoQuery.__init__)NNN)r r r listfloatrRecommendStrategyr rrrrrsrc@s&eZdZdeedeefddZdS) ContextPairrrcCsLt||_t||_t|jrJdt|jr$JddS)Nz$Positive vector must not contain NaNz$Negative vector must not contain NaN)rrrrrr)rrrrrrr 's  zContextPair.__init__N)r r r r!r"r rrrrr$&r$c@s&eZdZdeedeefddZdS)DiscoveryQuerytargetcontextcCs.t||_||_t|jrJddS)Nz"Target vector must not contain NaN)rrr'r(rr)rr'r(rrrr 0s zDiscoveryQuery.__init__N)r r r r!r"r$r rrrrr&/r%r&c@seZdZdeefddZdS) ContextQuery context_pairscCs ||_dSN)r*)rr*rrrr 8s zContextQuery.__init__N)r r r r!r$r rrrrr)7sr)DenseQueryVectordistancereturncCs*|tjjkr tjS|tjjkrtjStjS)zn Convert distance to order Args: distance: distance to convert Returns: order )rDistanceEUCLIDrr MANHATTANr )r-rrrdistance_to_order?s  r2queryvectorscCstjj|ddddtjf}|t|dk|t}t|jdkr8tj|}|t|dk|t}t||Stjj|ddddtjf}|t|dk|t}t||j S)z Calculate cosine distance between query and vectors Args: query: query vector vectors: vectors to calculate distance with Returns: distances axisN) rlinalgnormnewaxiswhereEPSILONlenshapedotT)r3r4 vectors_norm query_normrrrcosine_similarityOs   rEcCs(t|jdkr t||St||jS)z Calculate dot product between query and vectors Args: query: query vector. vectors: vectors to calculate distance with Returns: distances r9)r?r@rrArBr3r4rrr dot_productes rGcCsDt|jdkrtjj||ddStjj||ddtjfddS)z Calculate euclidean distance between query and vectors Args: query: query vector. vectors: vectors to calculate distance with Returns: distances r9r5r6N)r?r@rr:r;r<rFrrreuclidean_distancets "rHcCsLt|jdkrtjt||ddStjt||ddtjfddS)z Calculate manhattan distance between query and vectors Args: query: query vector. vectors: vectors to calculate distance with Returns: distances r9r5r6N)r?r@rsumabsr<rFrrrmanhattan_distances &rK distance_typecCs|t|r Jd|tjjkrt||S|tjjkr!t||S|tjj kr,t ||S|tjj kr7t ||St d|)N!Query vector must not contain NaNzUnknown distance type )rrrrr/COSINErEDOTrGr0rHr1rK ValueErrorr3r4rLrrrcalculate_distances        rRcCs~t|r Jd|tjjkr"tj||tjdjdtjd S|tjj kr9tj ||tjdjdtjd St |||S)z` Calculate same internal distances as in core, rather than the final displayed distance rMdtyper9)r7rT) rrrrr/r0squarefloat32rIr1rJrRrQrrrcalculate_distance_cores  " " rWxcCs,t|s t|r |S|tdt|S)N?)rrisinfaddrJrXrrr fast_sigmoidsr]cCsdtt|dS)Ng?rY)rr[r]r\rrrscaled_fast_sigmoidsr^csndttjdtjffdd }||j}||j}t||ktdd|D|jtdd|D|jS)Nexamplesr.csljd}g}|D] }t|}||q t|dkr(|t|tj tj|tjdj dd}|SNrrSr6) r@rWappendr?rfullinfrrVmax)r_ vector_countscoresexamplescore best_scoresrLr4rrget_best_scoress    z8calculate_recommend_best_scores..get_best_scorescs|]}t|VqdSr+r^rxirrr z2calculate_recommend_best_scores..css|]}t| VqdSr+rmrnrrrrps) r!types NumpyArrayrrrr=fromiterrT)r3r4rLrkposnegrrjrcalculate_recommend_best_scoress   rwcs<dttjdtjffdd }||j}||j}||S)Nr_r.csfjd}g}|D] }t|}||q t|dkr%|t|tj|tjdjdd}|Sr`) r@rWrar?rzerosrrVrI)r_rerfrgrh sum_scoresrjrrget_sum_scoress    z6calculate_recommend_sum_scores..get_sum_scores)r!rrrsrr)r3r4rLrzrurvrrjrcalculate_recommend_sum_scoress   r{r(cCsjtj|jdtjd}|D]%}t|j||}t|j||}tddt||k||kD}||7}q |S)NrrScSs$g|]\}}|r dn|rdndqS)r9rr5r)r is_biggeris_equalrrrrsz-calculate_discovery_ranks..) rrxr@int32rWrrrzip)r(r4rL overall_rankspairrurv pair_ranksrrrcalculate_discovery_rankss rcCs<t|j||}t|j||}tdd|Dtj}||S)Ncsrlr+rmrnrrrrprqz-calculate_discovery_scores..)rr(rWr'rrtrV)r3r4rLranksdistances_to_targetsigmoided_distancesrrrcalculate_discovery_scores s rc Csvtj|jdtjd}|jD]*}t|j||}t|j||}||t}t ddt |dDtj}||7}q|S)NrrScsrlr+)r]rnrrrrp&rqz+calculate_context_scores..r8) rrxr@rVr*rWrrr>rtminimum) r3r4rLoverall_scoresrrurv difference pair_scoresrrrcalculate_context_scoress   r)&enumrtypingrnumpyrqdrant_client.conversionsrrrqdrant_client.httprr>strrrr$r&r)r,__annotations__r/r2rsrErGrHrKrRrWrVr]r^rwr{r!rrrrrrrs