o Ci @sLddlZddlZddZddZddZdd Zd d Zd d ZddZ dS)Nc Cs2t|}||}tj|dd}d}}zm||}|dkr*tt|||j| |}|j tj j krAtj||j fdd}n|j }|j}||d|} tj| |||fdd}|dkrjtt|||jW|duru||||dur|||||fS|dur||||dur|||ww)zy returns the inverted lists content as a pair of (list_ids, list_codes). The codes are reshaped to a proper size int64dtypeNruint8)faissdowncast_InvertedLists list_sizenpzerosget_idsmemcpyswig_ptrnbytes get_codes code_size InvertedListsINVALID_CODE_SIZE n_per_block block_size release_ids release_codes) invlistsllslist_idsidscodes list_codesnpbbsls_roundr"s/var/www/html/fyndo/python/python_agents/rag_suite/venv/lib/python3.10/site-packages/faiss/contrib/inspect_tools.py get_invlist s8       r$cs"tjfddtjDddS)z1 return the array of sizes of the inverted lists csg|]}|qSr")r .0irr"r# +sz%get_invlist_sizes..rr)r arrayrangenlistr(r"r(r#get_invlist_sizes)s r-cCs*|jjD]}t|dt||qdS)z3 list values all fields of an object known to SWIG z = N) __class____swig_getmethods__printgetattr)objnamer"r"r#print_object_fields1s r4cCs t|j}||j|j|jS)z% return the PQ centroids as an array )rvector_to_array centroidsreshapeMksubdsub)pqcenr"r"r#get_pq_centroids8s r=cCs,t|j}t|j|j|j}||fS)zo extract matrix + bias from the PCA object works for any linear transform (OPQ, random rotation, etc.) )rr5bAr7d_outd_in)pcar>r?r"r"r#get_LinearTransform_matrix>s rCcs<t|jd|jt|jfddt|jDS)z. return to codebooks of an additive quantizer cs$g|]}||dqS)rr"r%co codebooksr"r#r)Ksz4get_additive_quantizer_codebooks..)rr5rGr7dcodebook_offsetsr+r8)aqr"rEr# get_additive_quantizer_codebooksGs   rKcCs"t|jd}||j|jS)z1 copy and return the data matrix in an IndexFlat float32)rr5rviewr7ntotalrH)indexxbr"r"r# get_flat_dataQsrQ) numpyr rr$r-r4r=rCrKrQr"r"r"r#s