
    ig                     F   d Z ddlZddlZddlmZ ddlmZmZmZm	Z	m
Z
 ddlZddl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 dd
lmZmZmZmZmZmZmZ ddl m!Z!m"Z" ddl#m$Z$  e       Z%e
e&e'e(e)f   Z*ee&e*f   Z+de	e&   de&fdZ,de&dee&ef   ddfdZ-e"ddddddddde&de&de	e+   de	ee&ef      de	ee&ef      de	ee&ef      de	ee&ef      de	e
e(ej\                  f      de	e&   defd       Z/e"ddddddddde&de&de	e+   de	ee&ef      de	ee&ef      de	ee&ef      de	ee&ef      de	e
e(ej\                  f      de	e&   de
eeeeef   f   fd       Z0e"dddddddddd	de&de	e&   de	e&   d e	e&   d!e	e'   d"e	e&   de	ee&ef      de	ee&ef      de	e
e(ej\                  f      de	e&   defd#       Z1e"dddddddddd	de&de	e&   de	e&   d e	e&   d!e	e'   d"e	e&   de	ee&ef      de	ee&ef      de	e
e(ej\                  f      de	e&   de
eeeeef   f   fd$       Z2e"dddd%de&de&de	ee&ef      de	e
e(ej\                  f      de	e&   defd&       Z3e"dddd%de&de&de	ee&ef      de	e
e(ej\                  f      de	e&   de
eeeeef   f   fd'       Z4e"dddd%de&de&de	ee&ef      de	e
e(ej\                  f      de	e&   defd(       Z5e"dddd%de&de&de	ee&ef      de	e
e(ej\                  f      de	e&   de
eeeeef   f   fd)       Z6e"ddddd*de&de&de+de	ee&ef      de	ee&ef      de	e
e(ej\                  f      de	e&   defd+       Z7e"ddddd*de&de&de+de	ee&ef      de	ee&ef      de	e
e(ej\                  f      de	e&   de
eeeeef   f   fd,       Z8e"dddd%de&de&de	ee&ef      de	e
e(ej\                  f      de	e&   defd-       Z9e"dddd%de&de&de	ee&ef      de	e
e(ej\                  f      de	e&   de
eeeeef   f   fd.       Z:y)/z6LiteLLM SDK functions for managing vector store files.    N)partial)Any	CoroutineDictOptionalUnion)request_timeout)Logging)BaseLLMHTTPHandler)GenericLiteLLMParams)LlmProviders)VectorStoreFileContentResponseVectorStoreFileCreateRequestVectorStoreFileDeleteResponseVectorStoreFileListQueryParamsVectorStoreFileListResponseVectorStoreFileObjectVectorStoreFileUpdateRequest)ProviderConfigManagerclient)VectorStoreFileRequestUtilscustom_llm_providerreturnc                     | xs dS )Nopenai )r   s    p/Users/manta/Documents/Projects/TheRoad-I1/.venv/lib/python3.12/site-packages/litellm/vector_store_files/main.py_ensure_providerr   "   s    *(*    vector_store_idkwargsc                     t         j                  y 	 t         j                  j                  |       }|r|j                  |       y y # t        $ r Y y w xY w)N)litellmvector_store_registry get_credentials_for_vector_storeupdate	Exception)r    r!   registry_credentialss      r   _prepare_registry_credentialsr)   &   s]    
 $$,	))JJ 	
  MM./   s   2A 	AA)
attributeschunking_strategyextra_headersextra_query
extra_bodytimeoutr   file_idr*   r+   r,   r-   r.   r/   c        	           K   t               }
	 t        j                         }d|	d<   t        t        f| ||||||||d	|	}t        j                         }t        |j                  |      }|j                  d |       d {   }t        j                  |      r| d {   }|S |}|S 7 )7 # t        $ r}t        j                  d |||
|	      d }~ww xY ww)NTacreate)	r    r0   r*   r+   r,   r-   r.   r/   r   modelr   original_exceptioncompletion_kwargsextra_kwargs)localsasyncioget_event_loopr   createcontextvarscopy_contextrunrun_in_executoriscoroutiner'   r#   exception_type)r    r0   r*   r+   r,   r-   r.   r/   r   r!   
local_varsloopfuncctxfunc_with_contextinit_responseresponsees                     r   r2   r2   9   s     J"
%%' y
+!/'#! 3
 
 &&(#CGGT2"2249JKK}-**H  %H L +  
$$ 3 (
 	

sY   CA2B.  B*B. B, B. %C&B. )C*B. ,B. .	C7CCCc        	            t               }
	 |	j                  d      }|	j                  d      }|	j                  dd      du }t        |      }t	        | |	       t        dd| i|	}t        j                  t        |            }|t        d	|       |
j                  |	       t        j                  |
      }||d
<   |j                  |	d d| i|| |d|j                  d      |       t        j!                  | ||||||||xs t"        |	j                  d      |      }|S # t$        $ r}t'        j(                  d |||
|	      d }~ww xY w)Nlitellm_logging_objlitellm_call_idr2   FTr    r!   r    providerz.Vector store file create is not supported for r0   r    rL   exclude_noner!   r4   optional_paramslitellm_paramsr   r   )r    create_request"vector_store_files_provider_configr   rU   logging_objr,   r.   r/   r   	_is_asyncr3   r   )r8   getpopr   r)   r   r   &get_provider_vector_store_files_configr   
ValueErrorr&   r   get_create_request_paramsupdate_from_kwargs
model_dumpbase_llm_http_handler vector_store_file_create_handlerr	   r'   r#   rA   )r    r0   r*   r+   r,   r-   r.   r/   r   r!   rB   rK   rL   rY   rU   provider_configrV   rH   rI   s                      r   r;   r;   m   s    J=
17<Q1R)/4E)FJJy%0D8	./BC%ofU-XoXQWX/VV!"56
 "@AT@UV  	&!'AA*M 	 %,y!..!? 
 $3#2 !+++>
 !4 	/ 	
 )II+)/> 3)+'!.::h' J 
  
$$ 3 (
 	

s   DD) )	E2EE)	afterbeforefilterlimitorderr,   r-   r/   r   rd   re   rf   rg   rh   c        
           K   t               }	 t        j                         }d|
d<   t        t        f| |||||||||	d
|
}t        j                         }t        |j                  |      }|j                  d |       d {   }t        j                  |      r| d {   }|S |}|S 7 )7 # t        $ r}t        j                  d |	|||
      d }~ww xY ww)NTalist)
r    rd   re   rf   rg   rh   r,   r-   r/   r   r3   )r8   r9   r:   r   listr<   r=   r>   r?   r@   r'   r#   rA   )r    rd   re   rf   rg   rh   r,   r-   r/   r   r!   rB   rC   rD   rE   rF   rG   rH   rI   s                      r   rj   rj      s     J"
%%'w
+'# 3
 
 &&(#CGGT2"2249JKK}-**H  %H L*  
$$ 3 (
 	

sY   CA3B/ B+B/  B-!B/ &C'B/ *C+B/ -B/ /	C8CCCc        
            t               }	 |
j                  d      }|
j                  d      }|
j                  dd      du }t        |	      }	t	        | |
       t        dd| i|
}t        j                  t        |	            }|t        d	|	       |j                  |
       t        j                  |      }|j                  |
d d| i|| |d
|j                  d      |	       t        j!                  | |||	|||||xs t"        |
j                  d      |      }|S # t$        $ r}t'        j(                  d |	|||
      d }~ww xY w)NrK   rL   rj   FTrM   r    rN   z,Vector store file list is not supported for rP   rQ   rS   r   )r    query_paramsrW   r   rU   rX   r,   r-   r/   r   rY   r3   r   )r8   rZ   r[   r   r)   r   r   r\   r   r]   r&   r   get_list_query_paramsr_   r`   ra   vector_store_file_list_handlerr	   r'   r#   rA   )r    rd   re   rf   rg   rh   r,   r-   r/   r   r!   rB   rK   rL   rY   rU   rc   
list_queryrH   rI   s                       r   rk   rk      s   " J9
17<Q1R)/4E)FJJw.$6	./BC%ofU-XoXQWX/VV!"56
 ">?R>ST  	&!'==jI 	 	...N:N#2#2 !+++>
 !4 	/ 
	
 )GG+#/> 3)+'#.::h' H 
  
$$ 3 (
 	

s   DD$ $	E-EE)r,   r/   r   c           	        K   t               }	 t        j                         }d|d<   t        t        f| ||||d|}t        j                         }	t        |	j                  |      }
|j                  d |
       d {   }t        j                  |      r| d {   }|S |}|S 7 )7 # t        $ r}t        j                  d ||||      d }~ww xY ww)NT	aretriever    r0   r,   r/   r   r3   )r8   r9   r:   r   retriever<   r=   r>   r?   r@   r'   r#   rA   r    r0   r,   r/   r   r!   rB   rC   rD   rE   rF   rG   rH   rI   s                 r   rr   rr   ?  s      J
%%'"{
+' 3
 
 &&(#CGGT2"2249JKK}-**H  %H L*  
$$ 3 (
 	

Y   CA.B* <B&=B* B(B* !C"B* %C&B* (B* *	C3CCCc                 N   t               }	 |j                  d      }|j                  d      }|j                  dd      du }	t        |      }t	        | |       t        dd| i|}
t        j                  t        |            }|t        d	|       |j                  |d | |d
| |d|
j                  d      |       t        j                  | ||||
|||xs t        |j                  d      |	
      }|S # t        $ r}t!        j"                  d ||||      d }~ww xY w)NrK   rL   rr   FTrM   r    rN   z0Vector store file retrieve is not supported for r    r0   rP   rQ   rS   r   
r    r0   rW   r   rU   rX   r,   r/   r   rY   r3   r   )r8   rZ   r[   r   r)   r   r   r\   r   r]   r_   r`   ra   "vector_store_file_retrieve_handlerr	   r'   r#   rA   r    r0   r,   r/   r   r!   rB   rK   rL   rY   rU   rc   rH   rI   s                 r   rt   rt   j  st    J6
17<Q1R)/4E)FJJ{E2d:	./BC%ofU-XoXQWX/VV!"56
 "BCVBWX  	..#2"
 $3#2 !+++>
 !4 	/ 	
 )KK+/> 3)+'.::h' L 
  
$$ 3 (
 	

   C/C< <	D$DD$c           	        K   t               }	 t        j                         }d|d<   t        t        f| ||||d|}t        j                         }	t        |	j                  |      }
|j                  d |
       d {   }t        j                  |      r| d {   }|S |}|S 7 )7 # t        $ r}t        j                  d ||||      d }~ww xY ww)NTaretrieve_contentrs   r3   )r8   r9   r:   r   retrieve_contentr<   r=   r>   r?   r@   r'   r#   rA   ru   s                 r   r~   r~     s      J
%%'&*"#
+' 3
 
 &&(#CGGT2"2249JKK}-**H  %H L*  
$$ 3 (
 	

rv   c                 N   t               }	 |j                  d      }|j                  d      }|j                  dd      du }	t        |      }t	        | |       t        dd| i|}
t        j                  t        |            }|t        d	|       |j                  |d | |d
| |d|
j                  d      |       t        j                  | ||||
|||xs t        |j                  d      |	
      }|S # t        $ r}t!        j"                  d ||||      d }~ww xY w)NrK   rL   r~   FTrM   r    rN   z8Vector store file content retrieve is not supported for rx   rP   rQ   rS   r   ry   r3   r   )r8   rZ   r[   r   r)   r   r   r\   r   r]   r_   r`   ra   !vector_store_file_content_handlerr	   r'   r#   rA   r{   s                 r   r   r     su    J6
17<Q1R)/4E)FJJ2E:dB	./BC%ofU-XoXQWX/VV!"56
 "JK^J_`  	..#2"
 $3#2 !+++>
 !4 	/ 	
 )JJ+/> 3)+'.::h' K 
  
$$ 3 (
 	

r|   )r,   r.   r/   r   c                   K   t               }	 t        j                         }	d|d<   t        t        f| ||||||d|}
t        j                         }t        |j                  |
      }|	j                  d |       d {   }t        j                  |      r| d {   }|S |}|S 7 )7 # t        $ r}t        j                  d ||||      d }~ww xY ww)NTaupdate)r    r0   r*   r,   r.   r/   r   r3   )r8   r9   r:   r   r&   r<   r=   r>   r?   r@   r'   r#   rA   )r    r0   r*   r,   r.   r/   r   r!   rB   rC   rD   rE   rF   rG   rH   rI   s                   r   r   r     s      J
%%' y

+!'! 3

 

 &&(#CGGT2"2249JKK}-**H  %H L*  
$$ 3 (
 	

sY   CA0B, >B(?B, B*B, #C$B, 'C(B, *B, ,	C5CCCc                    t               }	 |j                  d      }	|j                  d      }
|j                  dd      du }t        |      }t	        | |       t        dd| i|}t        j                  t        |            }|t        d	|       |j                  |       t        j                  |      }||d
<   |	j                  |d | |d|| |
d|j                  d      |       t        j!                  | ||||||	|||xs t"        |j                  d      |      }|S # t$        $ r}t'        j(                  d ||||      d }~ww xY w)NrK   rL   r   FTrM   r    rN   z.Vector store file update is not supported for r*   rx   rP   rQ   rS   r   )r    r0   update_requestrW   r   rU   rX   r,   r.   r/   r   rY   r3   r   )r8   rZ   r[   r   r)   r   r   r\   r   r]   r&   r   get_update_request_paramsr_   r`   ra    vector_store_file_update_handlerr	   r'   r#   rA   )r    r0   r*   r,   r.   r/   r   r!   rB   rK   rL   rY   rU   rc   r   rH   rI   s                    r   r&   r&   L  s    J?
17<Q1R)/4E)FJJy%0D8	./BC%ofU-XoXQWX/VV!"56
 "@AT@UV  	&!'AA*M 	 (2|$..#2" ! $3#2 !+++>
 !4 	/ 	
  )II+)/> 3)+'!.::h' J 
  
$$ 3 (
 	

s   DD+ +	E4EEc           	        K   t               }	 t        j                         }d|d<   t        t        f| ||||d|}t        j                         }	t        |	j                  |      }
|j                  d |
       d {   }t        j                  |      r| d {   }|S |}|S 7 )7 # t        $ r}t        j                  d ||||      d }~ww xY ww)NTadeleters   r3   )r8   r9   r:   r   deleter<   r=   r>   r?   r@   r'   r#   rA   ru   s                 r   r   r     s      J
%%' y
+' 3
 
 &&(#CGGT2"2249JKK}-**H  %H L*  
$$ 3 (
 	

rv   c                 N   t               }	 |j                  d      }|j                  d      }|j                  dd      du }	t        |      }t	        | |       t        dd| i|}
t        j                  t        |            }|t        d	|       |j                  |d | |d
| |d|
j                  d      |       t        j                  | ||||
|||xs t        |j                  d      |	
      }|S # t        $ r}t!        j"                  d ||||      d }~ww xY w)NrK   rL   r   FTrM   r    rN   z.Vector store file delete is not supported for rx   rP   rQ   rS   r   ry   r3   r   )r8   rZ   r[   r   r)   r   r   r\   r   r]   r_   r`   ra    vector_store_file_delete_handlerr	   r'   r#   rA   r{   s                 r   r   r     st    J6
17<Q1R)/4E)FJJy%0D8	./BC%ofU-XoXQWX/VV!"56
 "@AT@UV  	..#2"
 $3#2 !+++>
 !4 	/ 	
 )II+/> 3)+'.::h' J 
  
$$ 3 (
 	

r|   );__doc__r9   r<   	functoolsr   typingr   r   r   r   r   httpxr#   litellm.constantsr	   *litellm.litellm_core_utils.litellm_loggingr
   LiteLLMLoggingObj*litellm.llms.custom_httpx.llm_http_handlerr   litellm.types.routerr   litellm.types.utilsr    litellm.types.vector_store_filesr   r   r   r   r   r   r   litellm.utilsr   r    litellm.vector_store_files.utilsr   ra   strintfloatboolVectorStoreFileAttributeValueVectorStoreFileAttributesr   r)   Timeoutr2   r;   rj   rk   rr   rt   r~   r   r   r&   r   r   r   r   r   <module>r      s=	   <    8 8   - S I 5 ,   8 H*,  %c3t&; <  &C!CD +(3- +C + cN 
	& 
 7;26.2,0+/59)-0
0
 0
 23	0

  S#X/0
 DcN+0
 $sCx.)0
 c3h(0
 eE5==0120
 "#0
 0
 0
f 
 7;26.2,0+/59)-K
K
 K
 23	K

  S#X/K
 DcN+K
 $sCx.)K
 c3h(K
 eE5==012K
 "#K
  )C6K,K"LLMK
 K
\     .2,059)-1
1
 C=1
 SM	1

 SM1
 C=1
 C=1
 DcN+1
 $sCx.)1
 eE5==0121
 "#1
 !1
 1
h     .2,059)-J
J
 C=J
 SM	J

 SMJ
 C=J
 C=J
 DcN+J
 $sCx.)J
 eE5==012J
 "#J
 35P+P!QQJ
 J
Z 
 /359)-'
'
 '
 DcN+	'

 eE5==012'
 "#'
 '
 '
T 
 /359)-@
@
 @
 DcN+	@

 eE5==012@
 "#@
  )C6K,K"LLM@
 @
F 
 /359)-&
&
 &
 DcN+	&

 eE5==012&
 "#&
 $&
 &
R 
 /359)-B
B
 B
 DcN+	B

 eE5==012B
 "#B
 "Ic38V.V$WWB
 B
J  /3+/59)-*
*
 *
 *	*

 DcN+*
 c3h(*
 eE5==012*
 "#*
 *
 *
Z  /3+/59)-K
K
 K
 *	K

 DcN+K
 c3h(K
 eE5==012K
 "#K
  )C6K,K"LLMK
 K
\ 
 /359)-&
&
 &
 DcN+	&

 eE5==012&
 "#&
 #&
 &
R 
 /359)-B
B
 B
 DcN+	B

 eE5==012B
 "#B
 !9S#7T-T#UUB
 B
r   