
    ^h                     0    S r SSKrSSKJr   " S S5      rg)zH
Translates from OpenAI's `/v1/embeddings` to Databricks' `/embeddings`
    N)Optionalc                   r    \ rS rSr% SrSr\\   \S'   SS\\   SS4S jjr	\
S 5       rS rS	\S
\4S jrSrg)DatabricksEmbeddingConfig	   z
Reference: https://learn.microsoft.com/en-us/azure/databricks/machine-learning/foundation-models/api-reference#--embedding-task
Ninstructionreturnc                     [        5       nUR                  5        H(  u  p4US:w  d  M  Uc  M  [        U R                  X45        M*     g )Nself)localsitemssetattr	__class__)r
   r   locals_keyvalues        f/home/james-whalen/.local/lib/python3.13/site-packages/litellm/llms/databricks/embed/transformation.py__init__"DatabricksEmbeddingConfig.__init__   s5    (!--/JCf}!23 *    c                    U R                   R                  5        VVs0 s H_  u  pUR                  S5      (       a  M  [        U[        R
                  [        R                  [        [        45      (       a  MX  Uc  M]  X_Ma     snn$ s  snnf )N__)	__dict__r   
startswith
isinstancetypesFunctionTypeBuiltinFunctionTypeclassmethodstaticmethod)clskvs      r   
get_config$DatabricksEmbeddingConfig.get_config   s     **,
,<<%  &&-- 	   AD,
 	
 
s   B7B9B>Bc                     / $ N )r
   s    r   get_supported_openai_params5DatabricksEmbeddingConfig.get_supported_openai_params*   s	     	r   non_default_paramsoptional_paramsc                     U$ r&   r'   )r
   r*   r+   s      r   map_openai_params+DatabricksEmbeddingConfig.map_openai_params/   s    r   r'   r&   )__name__
__module____qualname____firstlineno____doc__r   r   str__annotations__r   r   r#   r(   dictr-   __static_attributes__r'   r   r   r   r   	   s_    
 	 # 4HSM 4T 4 
 
"
D 4 r   r   )r3   r   typingr   r   r'   r   r   <module>r9      s     ' 'r   