o Æ°i¸ã@s*dZddlZddlmZGdd„dƒZdS)zH Translates from OpenAI's `/v1/embeddings` to Databricks' `/embeddings` éN)ÚOptionalc@s`eZdZUdZdZeeed<ddeeddfdd„Ze dd„ƒZ d d „Z d e d e fd d„Z dS)ÚDatabricksEmbeddingConfigz‰ Reference: https://learn.microsoft.com/en-us/azure/databricks/machine-learning/foundation-models/api-reference#--embedding-task NÚ instructionÚreturncCs>tƒ ¡}| ¡D]\}}|dkr|durt|j||ƒq dS)NÚself)ÚlocalsÚcopyÚitemsÚsetattrÚ __class__)rrZlocals_ÚkeyÚvalue©rúc/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/databricks/embed/transformation.pyÚ__init__s €þz"DatabricksEmbeddingConfig.__init__cCsdd„|j ¡DƒS)NcSs>i|]\}}| d¡st|tjtjttfƒs|dur||“qS)Ú__N)Ú startswithÚ isinstanceÚtypesÚ FunctionTypeÚBuiltinFunctionTypeÚ classmethodÚ staticmethod)Ú.0ÚkÚvrrrÚ s ýüþü ô z8DatabricksEmbeddingConfig.get_config..)Ú__dict__r )ÚclsrrrÚ get_configsþz$DatabricksEmbeddingConfig.get_configcCsgS©Nr)rrrrÚget_supported_openai_params*sz5DatabricksEmbeddingConfig.get_supported_openai_paramsÚnon_default_paramsÚoptional_paramscCs|Sr r)rr"r#rrrÚmap_openai_params/sz+DatabricksEmbeddingConfig.map_openai_paramsr )Ú__name__Ú __module__Ú __qualname__Ú__doc__rrÚstrÚ__annotations__rrrr!Údictr$rrrrr s þÿ r)r(rÚtypingrrrrrrÚs