jdZddlZddlmZddlmZmZddlmZm Z ddl m Z m Z ddl mZmZddlmZmZdd lmZmZmZmZej2eZd Zd ed ed edefdZdedefdZ dedefdZ!de"deddfdZ#de$de"ddfdZ%dde"defdZ&y)zBuilds a tenant's pairing graph. The job owns strategist_product_neighbors and rewrites it freely. It never writes to strategist_pairing_decisions -- a merchant who rejects a pair and then re-syncs must not see that pair return. N)sql)Jsonexecute_values)get_db_connectionmigrate_pairing_tables)candidate_pairs load_eligible) is_servableload)cosine vectors_for) PAIR_TYPESscore_complement score_similar score_upsellabvectorsreturnc$|j|d|j|d}}|r|r t||nd}g}dt|||fdt|||fdt ||ffD]%\}}|s |j |d|d|d|'|S)N product_keygsimilar complementupsell) anchor_key neighbor_key pair_type)getr rrrappend) rrrleftright similarityresultsrscoreds 7/var/www/html/strategist-ai/app/services/pairing/job.py score_pairr's++a ./Q}=M1N%D)-e$CJG M!Q 34 '1j9: <1%&? 6  NN!M*:,-m,<)2>6<> ? ? Nproductsc |Dchc]}|d }}g}|D]S}|jdxsijdxsg}|D]$}||vr|j|d|dddddgd&U|Scc}w) zEA merchant's own declarations outrank every inference in this module.rtenant_relations complementaryrg?merchantzdeclared by the merchant)rrrscore confidencesourcereasons)rr )r)pknownpairsproductdeclaredneighbors r&merchant_pairsr8,s'/ 0!Q}  0E 0 EDKK 239r>>OUSU  DHu$ LL (>*2#&cZ&@%AC D DD L 1s A/r4ci}|D]+}|j|d|dfgj|-g}|jD].}|jdd|j |dt 0|S)Nrrc |dS)Nr.)r2s r&z*_top_per_anchor_and_type..Cs 1W:r(T)keyreverse) setdefaultr valuessortextendMAX_NEIGHBORS_PER_TYPE)r4groupedpairkeptmemberss r&_top_per_anchor_and_typerH<sGUD.[0ABBGNNtTU D>>#6 -t < G33456 Kr( tenant_idct} |j5}|jtjdj tj ||rut|tjdj tj ||Dcgc])}|d|d|d|d|d|dt|d f+c}d d d |j |jy cc}w#1swY0xYw#t$r*|jtjd |d wxYw#|jwxYw)zReplaces this tenant's whole graph in one transaction. Delete-then-insert rather than upsert: a pair that no longer scores must disappear, and an upsert would leave last run's edges behind forever. z+DELETE FROM {}.strategist_product_neighborszINSERT INTO {}.strategist_product_neighbors (anchor_key, neighbor_key, pair_type, score, confidence, source, reasons) VALUES %srrrr.r/r0r1Nz(Could not write the pairing graph for %sTexc_info)rcursorexecuterSQLformat Identifierrrcommit Exceptionrollbackloggererrorclose)rIr4conncurr2s r&replace_neighborsrZHs=  D [[] ;c KK=fS^^I./ 1 sCGG1%& 2349:./ oq'8!K.j!L/1X;1Y<(*: ; ;"  : ; ;$   ?"  $  sADB D %.DD DD  DD3EE Epercentstepc||y |t||y#t$rtjddYywxYw)ztProgress is best-effort. The job is spending real money on model calls; a reporting failure must never abort it.NzProgress callback failedTrK)intrSrUwarning)progressr[r\s r&_reportraksCBWt$ B1DABs  ;;c t|t|}t|ddt|dkrt|ddt|dddidSt ||}t|dd t |}t|}g}d}t |d D]J\}\} } |jt| | |dd|z|zz} | |k7s6| }t|| d |d |L|s t|ddt|} | D chc]} | d| d| df}} |D cgc]} | d| d| df|vr| c} | z}t|}t||t|ddt|tfd|D}tDcic]tfd|D}}t|t||t||z |d}tj!d|||Scc} wcc} wcc}w)Nzloading eligible productsdzno eligible pairs to scorer)r)r4servablequeuedby_type-zembedding products)startuscoring candidate pairs — z of Zzscoring candidate pairsrrrzwriting pairing graphc 3rK|].}t|j|d|d|dfs+d0yw)rrrrjN)r r).0r2 decisionss r& zpair_tenant..sAQk 9==!L/1^+.s=A;1)rs 0QJN?   8 $$4$4& T d  D T   T d FBsB#B$B434$4r(