
    ^h                     P    S r SSKJr  SSKJr  SSKJr  SSKJr   " S S	\\5      r	g
)z)
Calling logic for Databricks embeddings
    )Optional)EmbeddingResponse   )OpenAILikeEmbeddingHandler   )DatabricksBasec                      ^  \ rS rSr     SS\S\S\S\\   S\\   S\S\\	   S	\\
   S
\\   S\	4U 4S jjjrSrU =r$ )DatabricksEmbeddingHandler   modelinputtimeoutapi_keyapi_baseoptional_paramsmodel_responsecustom_endpointheadersreturnc                 ^   > U R                  UUSUUS9u  pl[        TU ]	  UUUUUUUUU	U
SUS9$ )N
embeddings)r   r   endpoint_typer   r   T)r   r   r   logging_objr   r   r   r   client
aembeddingr   r   )databricks_validate_environmentsuper	embedding)selfr   r   r   r   r   r   r   r   r   r   r   r   	__class__s                _/home/james-whalen/.local/lib/python3.13/site-packages/litellm/llms/databricks/embed/handler.pyr   $DatabricksEmbeddingHandler.embedding   sj     !@@&+ A 
 w #+)!  ! 
 	
     )NNNNN)__name__
__module____qualname____firstlineno__strlistfloatr   dictr   boolr   __static_attributes____classcell__)r    s   @r!   r
   r
      s     7;*."&#
#
 #
 	#
 ##
 3-#
 #
 !!23#
 "$#
 $#
 
#
 #
r#   r
   N)
__doc__typingr   litellm.utilsr   openai_like.embedding.handlerr   common_utilsr   r
   r$   r#   r!   <module>r5      s'     + G )$
!;^ $
r#   