
    ^hS                     n    S SK r S SKJr  S SKrS SKrS SKJrJrJr  S SK	J
r
  SSKJrJr   " S S\5      rg)	    N)Optional)AsyncHTTPHandlerHTTPHandlerget_async_httpx_client)EmbeddingResponse   )OpenAILikeBaseOpenAILikeErrorc                       \ rS rSrS r SS\S\S\S\S\	S	\	S
\S\4S jjr
     SS\	S\S\S\\	   S	\\	   S\S\\   S\\   S
\\   S\4S jjrSrg)OpenAILikeEmbeddingHandler   c                     g N )selfkwargss     d/home/james-whalen/.local/lib/python3.13/site-packages/litellm/llms/openai_like/embedding/handler.py__init__#OpenAILikeEmbeddingHandler.__init__   s        Ninputdatamodel_responsetimeoutapi_keyapi_baseheadersreturnc
                 @  #    S n
 U	b  [        U	[        5      (       a*  [        [        R                  R
                  SU0S9U l        OXl         U R                  R                  UU[        R                  " U5      S9I S h  vN n
U
R                  5         U
R                  5       nUR)                  UUSU0US	9  [+        S0 UD6$  NC! [        R                   aU  n[        UR                  R                  UR                  (       a  UR                  R                   S9e[#        U5      S9eS nAf[        R$                   a    [        SSS9e[&         a  n[        S[#        U5      S9eS nAff = f! [&         a!  nUR)                  UU[#        U5      S
9  UeS nAff = f7f)Nr   )llm_providerparamsr   r   status_codemessage  Timeout error occurred.  complete_input_dictr   r   additional_argsoriginal_response)r   r   r,   r   )
isinstancer   r   litellmLlmProvidersOPENAIasync_clientpostjsondumpsraise_for_statushttpxHTTPStatusErrorr
   responser$   textstrTimeoutException	Exception	post_callr   )r   r   r   r   r   r   r   logging_objr   clientr8   response_jsones                r   
aembedding%OpenAILikeEmbeddingHandler.aembedding   s     .	~F4D!E!E$:!(!5!5!<!<%w/%!
 %+!G!%!2!2!7!7#D) "8 "  ))+ ( !!!6 ="/	 "  %5}559 (( % !

 6 6/0zzAJJOO ?B1v  )) % #-F   G%#s1vFFG  	!!"%a& " 
 G	sl   FAE0 2C C$C &E0 FC E-AD++*E-E((E--E0 0
F:FFFmodeloptional_paramscustom_endpointc                    U R                  UUSUUS9u  plUnXS.UEnUR                  UUXS.S9  U
SL a  U R                  XXHXeX9US9	$ U	b  [        U	[        5      (       a  [        US9U l        OXl         U R                  R                  UU[        R                  " U5      S	9nUR                  5         UR                  5       nUR)                  UUSU0US9  [*        R,                  " S0 UD6$ ! [        R                   a7  n[        UR                  R                  UR                  R                   S
9eS nAf[        R"                   a    [        SSS
9e[$         a  n[        S['        U5      S
9eS nAff = f)N
embeddings)r   r   endpoint_typer   rF   )rD   r   )r)   r   )r   r   r+   T)	r   r   r>   r   r   r   r   r?   r   )r   r"   r#   r&   r'   r(   r)   r*   r   )_validate_environmentpre_callrB   r-   r   r   r?   r2   r3   r4   r5   r6   r7   r
   r8   r$   r9   r;   r<   r:   r=   r.   r   )r   rD   r   r   r>   r   r   rE   r   r?   rB   rF   r   r   r8   r@   rA   s                    r   	embedding$OpenAILikeEmbeddingHandler.embeddingV   s    !66&+ 7 
 B/B 	48O 	 	
 ??{  }E  `g  @G?  H  H>Z0@AA%g6DK K	C{{''ZZ% ( H %%'$MMOM 	2D9+	 	 	
 ((9=99% $$ 	!JJ22

  %% 	V!c;TUU 	C!c3q6BB	Cs%   7AC0 0E82D66*E8 E33E8)r1   r?   r   )NNNNN)__name__
__module____qualname____firstlineno__r   listdictr   floatr:   rB   r   boolrL   __static_attributes__r   r   r   r   r      s     ;; ; *	;
 ; ; ; ; 
;L 7;*."&D:D: D: 	D: #D: 3-D: D: !!23D: "$D: $D: 
D: D:r   r   )r3   typingr   r6   r.   &litellm.llms.custom_httpx.http_handlerr   r   r   litellm.types.utilsr   common_utilsr	   r
   r   r   r   r   <module>r[      s3   
     
 2 :E: E:r   