3fi.ddlmZddlmZmZddlmZmZmZm Z ddl m Z ddl m Z de eeefded e eeeffd Z dd e e d e e d edededed e eeeffdZ dd e e d e e dededed e eeeff dZeGddeZy))ABC) dataclassfield)AnyCallableOptionalSequence) RowMapping)DistanceStrategyresultsis_distance_metricreturnc r|sgS|Dcgc]'}tt|jd)}}t|t |}}||z }|dk(r|D]}d|d< t|S|D]6}t|jd}||z |z }|r d|z |d<2||d<8t|Scc}w)z2Normalizes scores to a 0-1 scale, where 1 is best.r?normalized_score)floatlistvaluesminmax) r ritemscores min_score max_score score_rangescore normalizeds h/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_postgres/v2/hybrid_search_config.py_normalize_scoresr! s  :A AeD'+, AF Av;F yIi'Ka +D'*D# $ +G} 2T[[]#B'i';6 '*Z'7D# $(2D# $ 2 =+Bs,B4primary_search_resultssecondary_search_resultsprimary_results_weightsecondary_results_weight fetch_top_kkwargsc |jdtj}|tjk7}t |Dcgc] }t |c}|} t |Dcgc] }t |c}d} i} | D]7} t t| jd} | d|z| d<| | | <9| D]N} t t| jd} | d|z}| | vr| | dxx|z cc<E|| d<| | | <Pt| jdd }|D]}|jdd |d |Scc}wcc}w) a Ranks documents using a weighted sum of scores from two sources. Args: primary_search_results: A list of (document, distance) tuples from the primary search. secondary_search_results: A list of (document, distance) tuples from the secondary search. primary_results_weight: The weight for the primary source's scores. Defaults to 0.5. secondary_results_weight: The weight for the secondary source's scores. Defaults to 0.5. fetch_top_k: The number of documents to fetch after merging the results. Defaults to 4. Returns: A list of (document, distance) tuples, sorted by weighted_score in descending order. distance_strategy)rFrrdistancec |dSNr*rs r z&weighted_sum_ranking..ns 4 3CTkeyreverseN) getr COSINE_DISTANCE INNER_PRODUCTr!dictstrrrsortedpop)r"r#r$r%r&r'r)is_primary_distancerownormalized_primarynormalized_secondaryweighted_scoresrdoc_idsecondary_weighted_scoreranked_resultsresults r weighted_sum_rankingrD*s8 -==,/?/M/MM+45sc5. -67sc7E 24O#'T$++-(+, 236LLZ"& '% +T$++-(+,#'(:#;>V#V _ $ F #J /3K K / 8D &*OF # + &CTN!- %t,- ,; ''O 6 8s E!E rrf_kc |jdtj}i}|tjk(}t |d|}t |D]Z\} } t t| jd} | |vrt| || <d|| d<|| dxxd| |zz z cc<\t |dd } t | D]Z\} } t t| jd} | |vrt| || <d|| d<|| dxxd| |zz z cc<\t |jd d } | d |S) a[ Ranks documents using Reciprocal Rank Fusion (RRF) of scores from two sources. Args: primary_search_results: A list of (document, distance) tuples from the primary search. secondary_search_results: A list of (document, distance) tuples from the secondary search. rrf_k: The RRF parameter k. Defaults to 60. fetch_top_k: The number of documents to fetch after merging the results. Defaults to 4. Returns: A list of (document_id, rrf_score) tuples, sorted by rrf_score in descending order. r)c |dSr,r-r.s r r/z(reciprocal_rank_fusion.. j)r0r1rgr*rc |dSr,r-r.s r r/z(reciprocal_rank_fusion..rHr0Tc |dSr,r-r.s r r/z(reciprocal_rank_fusion..s d:.>r0N) r4r r5r6r9 enumerater8rrr7)r"r#rEr&r'r) rrf_scoresis_similarity_metricsorted_primaryrankr<r@sorted_secondaryrBs r reciprocal_rank_fusionrQws~0 -==-/J -0@0N0NN )$N ~.? cT#**,'*+  #!%cJv -0Jv z *6:&#*>>& ? ) /0? cT#**,'*+  #!%cJv -0Jv z *6:&#*>>& ?!>N ,; ''r0ceZdZUdZdZeeed<dZeeed<dZ eeed<e Z e e ee eege efed<eeZeeefed <d Zeed <d Zeed <d Zeed<dZeed<y)HybridSearchConfigz AlloyDB Vector Store Hybrid Search Config. Queries might be slow if the hybrid search column does not exist. For best hybrid search performance, consider creating a TSV column and adding GIN index. tsv_columnzpg_catalog.englishtsv_lang fts_queryfusion_function)default_factoryfusion_function_parameters primary_top_ksecondary_top_klangchain_tsv_index index_nameGIN index_typeN)__name__ __module__ __qualname____doc__rUrr8__annotations__rVrWrDrXrr r rrr7rZr\intr]r_rar-r0r rSrSs!#J "2Hhsm2!Ix}! X * x 3S98C=H27t1LS#XLM3OS+J+Jr0rSN)?rhr[)<r[)abcr dataclassesrrtypingrrrr sqlalchemyr indexesr r7r8boolr!rrgrDrQrSr-r0r rps@(44!% d38n %;? d38nF%(&) J($Z0J(&z2J("J($ J(  J(  J(d38nJ(` F($Z0F(&z2F( F( F(  F( d38n F(R  r0