
    i                     X    d 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	  G d	 d
ee	      Z
y)z)
Calling logic for Databricks embeddings
    N)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                    |j                  dd       xs@ |j                  dd       xs, t        j                  d      xs t        j                  d      }| j                  ||d|||      \  }}t        |   |||||||||	|
d|      S )	N
user_agentdatabricks_user_agentLITELLM_USER_AGENTDATABRICKS_USER_AGENT
embeddings)r   r   endpoint_typer   r   custom_user_agentT)r   r   r   logging_objr   r   r   r   client
aembeddingr   r   )poposgetenvdatabricks_validate_environmentsuper	embedding)selfr   r   r   r   r   r   r   r   r   r   r   r   r   	__class__s                 v/Users/manta/Documents/Projects/TheRoad-I1/.venv/lib/python3.12/site-packages/litellm/llms/databricks/embed/handler.pyr%   z$DatabricksEmbeddingHandler.embedding   s    & d3 2""#:DA2yy-.2 yy01	 	 !@@&+/ A 
' w #+)!  ! 
 	
    )NNNNN)__name__
__module____qualname__strlistfloatr   dictr   boolr%   __classcell__)r'   s   @r(   r
   r
      s     7;*."&.
.
 .
 	.
 #.
 3-.
 .
 !!23.
 "$.
 $.
 
.
 .
r)   r
   )__doc__r!   typingr   litellm.utilsr   openai_like.embedding.handlerr   common_utilsr   r
    r)   r(   <module>r9      s*    
  + G )/
!;^ /
r)   