Ë
    Ýªg³  ã                   ó0   — d Z ddlZddlmZ  G d„ d«      Zy)zH
Translates from OpenAI's `/v1/embeddings` to Databricks' `/embeddings`
é    N)ÚOptionalc                   ód   — e Zd ZU dZdZee   ed<   ddee   ddfd„Ze	d„ «       Z
d„ Zded	efd
„Zy)ÚDatabricksEmbeddingConfigz‰
    Reference: https://learn.microsoft.com/en-us/azure/databricks/machine-learning/foundation-models/api-reference#--embedding-task
    NÚinstructionÚreturnc                 óˆ   — t        «       }|j                  «       D ]%  \  }}|dk7  sŒ|€Œt        | j                  ||«       Œ' y )NÚself)ÚlocalsÚitemsÚsetattrÚ	__class__)r	   r   Úlocals_ÚkeyÚvalues        úa/var/www/openai/venv/lib/python3.12/site-packages/litellm/llms/databricks/embed/transformation.pyÚ__init__z"DatabricksEmbeddingConfig.__init__   s9   € Ü“(ˆØ!Ÿ-™-ž/‰JˆCØf‹} Ñ!2Ü˜Ÿ™¨¨UÕ3ñ *ó    c                 óø   — | j                   j                  «       D ci c]O  \  }}|j                  d«      s9t        |t        j
                  t        j                  t        t        f«      s|||“ŒQ c}}S c c}}w )NÚ__)	Ú__dict__r   Ú
startswithÚ
isinstanceÚtypesÚFunctionTypeÚBuiltinFunctionTypeÚclassmethodÚstaticmethod)ÚclsÚkÚvs      r   Ú
get_configz$DatabricksEmbeddingConfig.get_config   sy   € ð Ÿ™×*Ñ*Ô,ô
á,‘1Ø—<‘< Ô%ÜØä×&Ñ&Ü×-Ñ-ÜÜ ð	ôð ð ˆq‰DØ,ò
ð 	
ùó 
s   žAA6c                 ó   — g S ©N© )r	   s    r   Úget_supported_openai_paramsz5DatabricksEmbeddingConfig.get_supported_openai_params*   s	   € ð ˆ	r   Únon_default_paramsÚoptional_paramsc                 ó   — |S r#   r$   )r	   r&   r'   s      r   Úmap_openai_paramsz+DatabricksEmbeddingConfig.map_openai_params/   s   € ØÐr   r#   )Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   ÚstrÚ__annotations__r   r   r!   r%   Údictr)   r$   r   r   r   r   	   s_   … ñð
 	ð ˜#‘ó ñ4 H¨S¡Mð 4¸Tó 4ð ñ
ó ð
ò"ð
°Dð È4ô r   r   )r-   r   Útypingr   r   r$   r   r   Ú<module>r2      s   ðñó Ý ÷'ò 'r   