U h@sddlmZddlZddlZddlmZmZmZmZm Z m Z m Z m Z ddl Z ddlmZddlmZddlmZmZmZddlmZmZddlmZmZmZmZeeZ d d d d d Z!dddddZ"d dddddZ#eddddGdd d eeZ$dS)) annotationsN)AnyCallableDictListOptionalTupleUnioncast) deprecated) Embeddings) BaseModel SecretStrroot_validator)convert_to_secret_strget_from_dict_or_env)before_sleep_logretrystop_after_attemptwait_exponentialVoyageEmbeddingszCallable[[Any], Any]) embeddingsreturncCs0d}d}tdt|jtd||dtttjdS)N T)Z multiplierminmax)reraisestopwaitZ before_sleep)rr max_retriesrrloggerloggingWARNING)rZ min_secondsZ max_secondsr%K/tmp/pip-unpacked-wheel-9gdii04g/langchain_community/embeddings/voyageai.py_create_retry_decorators  r'dict)responsercCs d|krtdt||S)NdatazVoyage API Error. Message: ) RuntimeErrorjsondumps)r)r%r%r&_check_response,sr.r)rkwargsrcKs&t|}|ddddd}|f|S)z)Use tenacity to retry the embedding call.r)r/rc[stjf|}t|S)N)requestspostr.r,)r/r)r%r%r&_embed_with_retry6s z+embed_with_retry.._embed_with_retry)r')rr/Zretry_decoratorr2r%r%r&embed_with_retry2sr3z0.0.29z1.0z%langchain_voyageai.VoyageAIEmbeddings)ZsinceZremovalZalternative_importc@seZdZUdZded<dZded<dZded<d ed <d Zd ed <dZd ed<dZ ded<dZ ded<GdddZ e dddddddZ d0ddddddZd1dd dd!d"d#d$Zdd!d%d&d'Zdd(d)d*d+Zdd,ddd!d-d.d/ZdS)2raVoyage embedding models. To use, you should have the environment variable ``VOYAGE_API_KEY`` set with your API key or pass it as a named parameter to the constructor. Example: .. code-block:: python from langchain_community.embeddings import VoyageEmbeddings voyage = VoyageEmbeddings(voyage_api_key="your-api-key", model="voyage-2") text = "This is a test query." query_result = voyage.embed_query(text) strmodelz&https://api.voyageai.com/v1/embeddingsvoyage_api_baseNzOptional[SecretStr]voyage_api_keyint batch_sizer!z+Optional[Union[float, Tuple[float, float]]]request_timeoutFboolshow_progress_barT truncationc@seZdZdZdS)zVoyageEmbeddings.ConfigZforbidN)__name__ __module__ __qualname__extrar%r%r%r&ConfigfsrC)prer)valuesrcCsZtt|dd|d<d|kr.d|d<tdd|krVd|krN|ddkrNdnd |d<|S) z?Validate that api key and python package exists in environment.r7ZVOYAGE_API_KEYr5z voyage-01zmodel will become a required arg for VoyageAIEmbeddings, we recommend to specify it when using this class. Currently the default is set to voyage-01.r9)zvoyage-2z voyage-02H)rrr"warning)clsrEr%r%r&validate_environmentis   z%VoyageEmbeddings.validate_environmentz List[str]z Optional[str])input input_typercCs>tt|j}|jdd|i|j|||jd|jd}|S)N AuthorizationzBearer )r5rKrLr>)urlheadersr,timeout)r rr7Zget_secret_valuer6r5r>r;)selfrKrLZapi_keyparamsr%r%r&_invocation_paramss  z#VoyageEmbeddings._invocation_paramsz Optional[int]zList[List[float]])textsr9rLrc Csg}|dkr|j}|jrlzddlm}Wn,tk rT}ztd|W5d}~XYnX|tdt||}ntdt||}|r|dkrtd|d|D]>}t|f|j |||||d} | dd | d Dq|S) Nr)tqdmzgMust have tqdm installed if `show_progress_bar` is set to True. Please install with `pip install tqdm`.)querydocumentz input_type z0 is invalid. Options: None, 'query', 'document'.)rKrLcss|]}|dVqdS)Z embeddingNr%).0rr%r%r& sz3VoyageEmbeddings._get_embeddings..r*) r9r=Z tqdm.autorU ImportErrorrangelen ValueErrorr3rSextend) rQrTr9rLrrUeZ_iterir)r%r%r&_get_embeddingss:  z VoyageEmbeddings._get_embeddings)rTrcCs|j||jddS)zCall out to Voyage Embedding endpoint for embedding search docs. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. rWr9rLrbr9)rQrTr%r%r&embed_documentss z VoyageEmbeddings.embed_documentsz List[float])textrcCs|j|g|jdddS)zCall out to Voyage Embedding endpoint for embedding query text. Args: text: The text to embed. Returns: Embedding for the text. rVrcrrd)rQrfr%r%r& embed_querys zVoyageEmbeddings.embed_query)rL)rTrLrcCs|j||j|dS)aPCall out to Voyage Embedding endpoint for embedding general text. Args: texts: The list of texts to embed. input_type: Type of the input text. Default to None, meaning the type is unspecified. Other options: query, document. Returns: Embedding for the text. rcrd)rQrTrLr%r%r&embed_general_textss z$VoyageEmbeddings.embed_general_texts)N)NN)r?r@rA__doc____annotations__r6r7r!r;r=r>rCrrJrSrbrergrhr%r%r%r&r>s*       ) )% __future__rr,r#typingrrrrrrr r r0Zlangchain_core._api.deprecationr Zlangchain_core.embeddingsr Zlangchain_core.pydantic_v1r rrZlangchain_core.utilsrrZtenacityrrrr getLoggerr?r"r'r.r3rr%r%r%r&s& (