
    g                     P    d Z ddlmZ ddlmZ ddlmZ ddlmZ  G d d	ee      Z	y
)z)
Calling logic for Databricks embeddings
    )Optional)EmbeddingResponse   )OpenAILikeEmbeddingHandler   )DatabricksBasec                   r     e Zd Z	 	 	 	 	 ddedededee   dee   dedee   dee	   d	ee   d
ef fdZ
 xZS )DatabricksEmbeddingHandlermodelinputtimeoutapi_keyapi_baseoptional_paramsmodel_responsecustom_endpointheadersreturnc                 l    | j                  ||d||      \  }}t        | 	  |||||||||	|
d|      S )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                Z/var/www/openai/venv/lib/python3.12/site-packages/litellm/llms/databricks/embed/handler.pyr   z$DatabricksEmbeddingHandler.embedding   sl     !@@&+ A 
' w #+)!  ! 
 	
    )NNNNN)__name__
__module____qualname__strlistfloatr   dictr   boolr   __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    <module>r1      s'     + G )$
!;^ $
r!   