
    rhg                     @    d dl Z d dlZ	 d Zd Zd Zd Zd Zd Zd Z	y)	    Nc                     t        j                  | d      }t        j                  dk(  r|j	                  d       |d   }|j                  d|dz         d d dd f   j                         S )	Nint32dtypebigTinplacer      )npfromfilesys	byteorderbyteswapreshapecopyfnameads      h/var/www/html/ai-insurance-compliance-backend/venv/lib/python3.12/site-packages/faiss/contrib/vecs_io.py
ivecs_readr      s^    
E)A
}}	

4
 	!A99RQ12&++--    c                 6    t        |       j                  d      S Nfloat32)r   viewr   s    r   
fvecs_readr          e!!),,r   c                     t         j                  dk7  sJ t        j                  | dd      }|d   }|j	                  d|dz         d d dd f   S )Nr   r   rr   moder   r
   r   )r   r   r   memmapr   r   s      r   
ivecs_mmapr&      sO    ==E!!!
		%wS1A	!A99RQ12&&r   c                 6    t        |       j                  d      S r   )r&   r   r   s    r   
fvecs_mmapr(   "   r    r   c                    t        j                  | dd      }t        j                  dk(  r.|d d d d d   j	                         }|j                  d      d   }n|d d j                  d      d   }|j                  d|dz         d d dd f   S )	Nuint8r"   r#   r      r
   r   r   )r   r%   r   r   r   r   r   )r   xdar   s       r   
bvecs_mmapr.   &   s    
		%wS1A
}}rU4R4[GGGQbqEJJw"99RQ12&&r   c                     |j                   \  }}t        j                  ||dz   fd      }||d d df<   ||d d dd f<   t        j                  dk(  r|j                  d       |j                  |        y )Nr   r   r   r   r   Tr   )shaper   emptyr   r   r   tofile)r   mnr   m1s        r   ivecs_writer6   0   si    77DAq	1a!e*G	,BBq!tHBq!"uI
}}
D!IIer   c                 \    |j                  d      }t        | |j                  d             y )Nr   r   )astyper6   r   )r   r3   s     r   fvecs_writer9   :   s"    	Aqvvg'r   )
r   numpyr   r   r   r&   r(   r.   r6   r9    r   r   <module>r<      s5     .-'-'(r   