a bg @sddlmZddlZddlZddlmZmZmZmZm Z m Z m Z m Z ddl Z ddlmZddlmZddlmZmZddlmZmZmZmZddlmZmZmZmZee Z!d d d d d Z"dddddZ#d dddddZ$eddddGdd d eeZ%dS)) annotationsN)AnyCallableDictListOptionalTupleUnioncast) deprecated) Embeddings)convert_to_secret_strget_from_dict_or_env) BaseModel ConfigDict SecretStrmodel_validator)before_sleep_logretrystop_after_attemptwait_exponentialVoyageEmbeddingszCallable[[Any], Any]) embeddingsreturncCs0d}d}tdt|jtd||dtttjdS)N T) multiplierminmax)reraisestopwait before_sleep)rr max_retriesrrloggerloggingWARNING)rZ min_secondsZ max_secondsr(u/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_community/embeddings/voyageai.py_create_retry_decorators  r*dict)responsercCs d|vrtdt||S)NdatazVoyage API Error. Message: ) RuntimeErrorjsondumps)r,r(r(r)_check_response,sr1r)rkwargsrcKs*t|}|ddddd}|fi|S)z)Use tenacity to retry the embedding call.r)r2rc[stjfi|}t|S)N)requestspostr1r/)r2r,r(r(r)_embed_with_retry6sz+embed_with_retry.._embed_with_retry)r*)rr2Zretry_decoratorr5r(r(r)embed_with_retry2sr6z0.0.29z1.0z%langchain_voyageai.VoyageAIEmbeddings)ZsinceZremovalZalternative_importc@seZdZUdZded<dZded<dZded<d ed <d Zd ed <dZd ed<dZ ded<dZ ded<e ddZ e ddedddddZd2ddddd d!Zd3dd"dd#d$d%d&Zdd#d'd(d)Zdd*d+d,d-Zdd.ddd#d/d0d1ZdS)4raVoyage 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 truncationZforbid)extrabefore)moderr)valuesrcCsZtt|dd|d<d|vr.d|d<tdd|vrVd|vrN|ddvrNdnd |d<|S) z?Validate that api key and python package exists in environment.r:ZVOYAGE_API_KEYr8z 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.r<)zvoyage-2z voyage-02H)r rr%warning)clsrEr(r(r)validate_environmentjs   z%VoyageEmbeddings.validate_environmentz List[str]z Optional[str])input input_typercCs>tt|j}|jdd|i|j|||jd|jd}|S)N AuthorizationzBearer )r8rKrLrA)urlheadersr/timeout)r rr:Zget_secret_valuer9r8rAr>)selfrKrLZapi_keyparamsr(r(r)_invocation_paramss  z#VoyageEmbeddings._invocation_paramsz Optional[int]zList[List[float]])textsr<rLrc Csg}|dur|j}|jrnzddlm}Wn.tyV}ztd|WYd}~n d}~00|tdt||}ntdt||}|r|dvrtd|d|D]B}t|fi|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) z3VoyageEmbeddings._get_embeddings..r-) r<r@Z tqdm.autorU ImportErrorrangelen ValueErrorr6rSextend) rQrTr<rLrrUeZ_iterir,r(r(r)_get_embeddingss8  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. rWr<rLrcr<)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. rVrdrre)rQrgr(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. rdre)rQrTrLr(r(r)embed_general_textss z$VoyageEmbeddings.embed_general_texts)N)NN)__name__ __module__ __qualname____doc____annotations__r9r:r$r>r@rArZ model_configr classmethodrJrSrcrfrhrir(r(r(r)r>s0       ) )& __future__rr/r&typingrrrrrrr r r3Zlangchain_core._api.deprecationr Zlangchain_core.embeddingsr Zlangchain_core.utilsr rZpydanticrrrrtenacityrrrr getLoggerrjr%r*r1r6rr(r(r(r)s& (