
    Yj                        d dl mZ d dlmZmZ d dlmZ ddlmZ ddl	m
Z
 ddlmZ ddlmZ erdd	lmZmZ dd
lmZmZ ddZ	 	 	 	 	 	 	 	 	 	 ddZ	 	 	 	 	 	 	 	 	 	 ddZ	 	 	 	 	 	 	 	 	 	 ddZ	 	 	 	 	 	 	 	 	 	 ddZy)    )annotations)TYPE_CHECKINGMapping)assert_never   )Omit)is_given)VectorStoreFile)VectorStoreFileBatch)Files
AsyncFiles)FileBatchesAsyncFileBatchesc                @    | j                  d      }|t        |      S y)z7Use the server hint or the existing one-second default.zopenai-poll-after-ms  )getint)headersfrom_headers     X/var/www/html/vmm-quimica/venv/lib/python3.12/site-packages/openai/lib/_vector_stores.py_get_poll_interval_msr      s%    ++45K;    c                  ddi}t        |      rt        |      |d<   	 | j                  j                  |||      }|j	                         }|j
                  dk(  r5t        |      st        |j                        }| j                  |dz         nM|j
                  dk(  s|j
                  dk(  s|j
                  d	k(  r|S t        rt        |j
                         n|S )
z;Poll a vector-store file using the caller's resource hooks.X-Stainless-Poll-Helpertrue X-Stainless-Custom-Poll-Intervalvector_store_idextra_headersin_progressr   	cancelled	completedfailedr	   strwith_raw_responseretrieveparsestatusr   r   _sleepr   r   resourcefile_idr   poll_interval_msr   responsefiles          r   poll_vector_store_filer1      s      9&AG !69:J6K23
--66+! 7 
 ~~;;-',-#89I9I#J OO,t34[[K'4;;++EX`I`KT[[)' r   c                 K   ddi}t        |      rt        |      |d<   	 | j                  j                  |||       d{   }|j	                         }|j
                  dk(  r=t        |      st        |j                        }| j                  |dz         d{    nM|j
                  dk(  s|j
                  d	k(  s|j
                  d
k(  r|S t        rt        |j
                         n|S 7 7 Uw)zAPoll a vector-store file using the caller's async resource hooks.r   r   r   r   Nr    r   r!   r"   r#   r$   r+   s          r   async_poll_vector_store_filer3   :   s       9&AG !69:J6K23
!33<<+! = 
 
 ~~;;-',-#89I9I#J //"2T"9:::[[K'4;;++EX`I`KT[[)' 
 ;s&   A C5C1AC5C3AC53C5c               :   ddi}t        |      rt        |      |d<   	 | j                  j                  |||      }|j	                         }|j
                  j                  dkD  r5t        |      st        |j                        }| j                  |dz         }|S )z<Poll a vector-store batch using the caller's resource hooks.r   r   r   r   r   r   
r	   r%   r&   r'   r(   file_countsr    r   r   r*   r,   batch_idr   r.   r   r/   batchs          r   poll_vector_store_file_batchr:   \   s      9&AG !69:J6K23
--66+! 7 
  ((1,,-#89I9I#J OO,t34r   c               j  K   ddi}t        |      rt        |      |d<   	 | j                  j                  |||       d{   }|j	                         }|j
                  j                  dkD  r=t        |      st        |j                        }| j                  |dz         d{    |S 7 l7 	w)zBPoll a vector-store batch using the caller's async resource hooks.r   r   r   r   Nr   r   r5   r7   s          r   "async_poll_vector_store_file_batchr<   z   s       9&AG !69:J6K23
!33<<+! = 
 
  ((1,,-#89I9I#J //"2T"9:::
 ;s%   A B3B/A$B3'B1(B31B3N)r   zMapping[str, str]returnr   )
r,   r   r-   r%   r   r%   r.   
int | Omitr=   r
   )
r,   r   r-   r%   r   r%   r.   r>   r=   r
   )
r,   r   r8   r%   r   r%   r.   r>   r=   r   )
r,   r   r8   r%   r   r%   r.   r>   r=   r   )
__future__r   typingr   r   typing_extensionsr   _typesr   _utilsr	   %types.vector_stores.vector_store_filer
   +types.vector_stores.vector_store_file_batchr   resources.vector_stores.filesr   r   $resources.vector_stores.file_batchesr   r   r   r1   r3   r:   r<    r   r   <module>rI      s    " ) *   C NAT 	
 ! D 	
 ! D 	
 ! < 	
 ! r   