o
    CiJ                     @   sF   d dl Z	 dd Zdd Zdd Zdd	 Zd
d Zdd Zdd ZdS )    Nc                 C   s:   t j| dd}|d }|d|d d d dd f  S )Nint32dtyper      )npfromfilereshapecopyfnamead r   m/var/www/html/fyndo/python/python_agents/rag_suite/venv/lib/python3.10/site-packages/faiss/contrib/vecs_io.py
ivecs_read   s   $r   c                 C      t | dS Nfloat32)r   viewr   r   r   r   
fvecs_read      r   c                 C   s8   t j| ddd}|d }|d|d d d dd f S )Nr   rr   moder   r   r   )r   memmapr	   r   r   r   r   
ivecs_mmap   s    r   c                 C   r   r   )r   r   r   r   r   r   
fvecs_mmap   r   r   c                 C   sF   t j| ddd}|d d dd }|d|d d d dd f S )Nuint8r   r      r   r   r   )r   r   r   r	   )r   xr   r   r   r   
bvecs_mmap"   s    r"   c                 C   sR   |j \}}tj||d fdd}||d d df< ||d d dd f< ||  d S )Nr   r   r   r   )shaper   emptytofile)r   mnr   m1r   r   r   ivecs_write(   s
   
r)   c                 C   s   | d}t| |d d S )Nr   r   )astyper)   r   )r   r&   r   r   r   fvecs_write0   s   
r+   )	numpyr   r   r   r   r   r"   r)   r+   r   r   r   r   <module>   s   