Ë
    uPêjÒ  ã                   ó|   — d dl m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 d dlmZ dd	lmZmZ  G d
„ dee	«      Zy)é    )ÚFinalN)ÚLogging)ÚBaseLLMException)ÚBaseEmbeddingConfig)ÚAllEmbeddingInputValues)ÚEmbeddingResponseé   )ÚSnowflakeBaseConfigÚSnowflakeExceptionc                   óÞ   — e Zd ZdZ	 ddedz  dedz  dededededz  d	efd
„Zdedededed	ef
d„Z	dede
j                  dedededz  dededed	efd„Zdededee
j                   z  d	efd„Zy)ÚSnowflakeEmbeddingConfigzf
    source: https://docs.snowflake.com/developer-guide/snowflake-rest-api/reference/cortex-embed
    NÚapi_baseÚapi_keyÚmodelÚoptional_paramsÚlitellm_paramsÚstreamÚreturnc                 ó0   — | j                  ||«      }|› d�S )Nz/cortex/inference:embed)Ú_get_api_base)Úselfr   r   r   r   r   r   s          út/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/snowflake/embedding/transformation.pyÚget_complete_urlz)SnowflakeEmbeddingConfig.get_complete_url   s$   € ð ×%Ñ% h°Ó@ˆà�Ð2Ð3Ð3ó    ÚinputÚheadersc                 ó   — ||dœ|¥S )N)Útextr   © )r   r   r   r   r   s        r   Útransform_embedding_requestz4SnowflakeEmbeddingConfig.transform_embedding_request    s   € ð ¨ÑA°ÐAÐAr   Úraw_responseÚmodel_responseÚlogging_objÚrequest_datac	                 ó¸   — |j                  «       }	|	d   D ]  }
|
d   d   |
d<   Œ t        di |	¤Ž}d|j                  xs dz   |_        |�||j                  d<   |S )NÚdataÚ	embeddingr   z
snowflake/Ú r   r   )Újsonr   r   Ú_hidden_params)r   r   r!   r"   r#   r   r$   r   r   Úresponse_jsonÚitemÚreturned_responses               r   Útransform_embedding_responsez5SnowflakeEmbeddingConfig.transform_embedding_response)   s   € ð  ,×0Ñ0Ó2ˆà! &Ñ)ò 	5ˆDØ $ [Ñ 1°!Ñ 4ˆD�Òð	5ä#4Ñ#E°}Ñ#EÐà".Ð2C×2IÑ2IÒ2OÈRÑ"PÐÔàÐØ8=Ð×,Ñ,¨WÑ5Ø Ð r   Úerror_messageÚstatus_codec                 ó   — t        |||¬«      S )N)Úmessager0   r   )r   )r   r/   r0   r   s       r   Úget_error_classz(SnowflakeEmbeddingConfig.get_error_class@   s   € Ü!¨-À[ÐZaÔbÐbr   )N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚstrÚdictÚboolr   r   r    ÚhttpxÚResponser   ÚLiteLLMLoggingObjr.   ÚintÚHeadersr   r3   r   r   r   r   r      s4  „ ñð #ñ4à˜‘*ð4ð �t‘ð4ð ð	4ð
 ð4ð ð4ð �t‘ð4ð 
ó4ðBàðBð 'ðBð ð	Bð
 ðBð 
óBð!àð!ð —n‘nð!ð *ð	!ð
 'ð!ð �t‘ð!ð ð!ð ð!ð ð!ð 
ó!ð.c¨Sð c¸sð cÈTÐTY×TaÑTaÑMað cÐfvô cr   r   )Útypingr   r;   Ú*litellm.litellm_core_utils.litellm_loggingr   r=   Ú)litellm.llms.base_llm.chat.transformationr   Ú.litellm.llms.base_llm.embedding.transformationr   Úlitellm.types.llms.openair   Úlitellm.types.utilsr   Úutilsr
   r   r   r   r   r   ú<module>rG      s/   ðÝ ã å SÝ FÝ NÝ =Ý 1ç ;ô3cÐ2Ð4Gõ 3cr   