
    Oj                    ,   d dl mZ d dlZd dlZd dlmZmZ 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 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ed	 	 	 	 	 	 	 	 	 	 	 ddZed	 	 	 	 	 	 	 	 	 	 	 ddZ	 	 	 	 	 	 	 	 	 	 ddZ	 	 	 	 	 	 	 	 	 	 ddZy)    )annotationsN)TYPE_CHECKINGMapping)assert_never   )Omit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     p/var/www/origus_pro_usr/data/www/origus.pro/core/.venv/lib/python3.12/site-packages/openai/lib/_vector_stores.py_get_poll_interval_msr      s%    ++45K;    c                f    t        |       r&t        j                  |       r| dk  rt        d      y y )Nr   z<Expected a finite, non-negative value for `max_wait_seconds`)r
   mathisfinite
ValueErrormax_wait_secondss    r   validate_max_wait_secondsr       s6     !4==9I+JN^abNbWXX Oc!r   c                \    | t        j                         z
  }|dk  rt        d|d      |S )Nr   z(Timed out waiting for vector store file z to finish processing)time	monotonicTimeoutError)deadlinefile_id	remainings      r   _remaining_poll_timer(      s6    4>>++IA~Eg[Pefggr   r   c               l   t        |       t        |      rt        j                         |z   nd}ddi}t        |      rt	        |      |d<   	 |t        ||       | j                  j                  |||      }|j                         }|j                  dk(  rOt        |      st        |j                        }|dz  }	|t        |	t        ||            }	| j                  |	       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.NX-Stainless-Poll-Helpertrue X-Stainless-Custom-Poll-Intervalvector_store_idextra_headersin_progressr   	cancelled	completedfailedr    r
   r"   r#   strr(   with_raw_responseretrieveparsestatusr   r   min_sleepr   r   
resourcer&   r.   poll_interval_msr   r%   r   responsefilesleep_secondss
             r   poll_vector_store_filerB   &   s+    ./6>?O6Pt~~"22VZH8&AG !69:J6K23
 73--66+! 7 
 ~~;;-',-#89I9I#J ,t3M# #M3GRY3Z [OOM*[[K'4;;++EX`I`KT[[)3 r   c                 K   t        |       t        |      rt        j                         |z   nd}ddi}t        |      rt	        |      |d<   	 |t        ||       | j                  j                  |||       d{   }|j                         }|j                  dk(  rWt        |      st        |j                        }|dz  }	|t        |	t        ||            }	| j                  |	       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.Nr*   r+   r,   r-   r0   r   r1   r2   r3   r4   r<   s
             r   async_poll_vector_store_filerD   Q   sE     ./6>?O6Pt~~"22VZH8&AG !69:J6K23
 73!33<<+! = 
 
 ~~;;-',-#89I9I#J ,t3M# #M3GRY3Z [//-000[[K'4;;++EX`I`KT[[)3 
 1s&   A=E?E A4E4E
5AE
E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
   r5   r6   r7   r8   file_countsr0   r   r   r;   r=   batch_idr.   r>   r   r?   batchs          r   poll_vector_store_file_batchrK   |   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   rF   rH   s          r   "async_poll_vector_store_file_batchrM      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)r   zMapping[str, str]returnr   )r   float | OmitrN   None)r%   floatr&   r5   rN   rQ   )r=   r   r&   r5   r.   r5   r>   
int | Omitr   rO   rN   r   )r=   r   r&   r5   r.   r5   r>   rR   r   rO   rN   r   )
r=   r   rI   r5   r.   r5   r>   rR   rN   r   )
r=   r   rI   r5   r.   r5   r>   rR   rN   r   )
__future__r   r   r"   typingr   r   typing_extensionsr   _typesr   r	   _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   r    r(   rB   rD   rK   rM    r   r   <module>r]      s%   "   ) *   C NATY
 &*((( 	(
 !( #( (b &*((( 	(
 !( #( (V 	
 ! < 	
 ! r   