
    ig                     @    S SK r S SKr S rS rS rS rS rS rS r	g)	    Nc                     [         R                  " U SS9n[        R                  S:X  a  UR	                  SS9  US   nUR                  SUS-   5      S S 2SS 24   R                  5       $ )	Nint32dtypebigTinplacer      )npfromfilesys	byteorderbyteswapreshapecopyfnameads      e/home/dmtnaga/Documents/work/airagagent/rag_env/lib/python3.13/site-packages/faiss/contrib/vecs_io.py
ivecs_readr      s^    
E)A
}}	

4
 	!A99RQ12&++--    c                 6    [        U 5      R                  S5      $ Nfloat32)r   viewr   s    r   
fvecs_readr          e!!),,r   c                     [         R                  S:w  d   e[        R                  " U SSS9nUS   nUR	                  SUS-   5      S S 2SS 24   $ )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    [        U 5      R                  S5      $ r   )r&   r   r   s    r   
fvecs_mmapr(   "   r    r   c                    [         R                  " U SSS9n[        R                  S:X  a.  US S S S S2   R	                  5       nUR                  S5      S   nOUS S R                  S5      S   nUR                  SUS-   5      S S 2SS 24   $ )	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                     UR                   u  p#[        R                  " X#S-   4SS9nX4S S 2S4'   XS S 2SS 24'   [        R                  S:X  a  UR                  SS9  UR                  U 5        g )Nr   r   r   r   r   Tr   )shaper   emptyr   r   r   tofile)r   mnr   m1s        r   ivecs_writer6   0   sa    77DA	1!e*G	,Bq!tHq!"uI
}}
D!IIer   c                 Z    UR                  S5      n[        XR                  S5      5        g )Nr   r   )astyper6   r   )r   r3   s     r   fvecs_writer9   :   s     	Avvg'r   )
r   numpyr   r   r   r&   r(   r.   r6   r9    r   r   <module>r<      s5     .-'-'(r   