7|h ddlmZddlZddlZddlmZmZmZmZm Z m Z m Z m Z ddl Z ddlmZddlmZddlmZmZddlmZmZmZmZddlmZmZmZmZej>e Z!dd Z"dd Z#dd Z$ed ddGdd eeZ%y)) annotationsN)AnyCallableDictListOptionalTupleUnioncast) deprecated) Embeddings)convert_to_secret_strget_from_dict_or_env) BaseModel ConfigDict SecretStrmodel_validator)before_sleep_logretrystop_after_attemptwait_exponentialVoyageEmbeddingsc d}d}tdt|jtd||t t t jS)N T) multiplierminmax)reraisestopwait before_sleep)rr max_retriesrrloggerloggingWARNING) embeddings min_seconds max_secondss f/var/www/html/test/engine/venv/lib/python3.12/site-packages/langchain_community/embeddings/voyageai.py_create_retry_decoratorr,sFKK   6 6 7   M%fgoo>  cPd|vr!tdtj||S)NdatazVoyage API Error. Message: ) RuntimeErrorjsondumps)responses r+_check_responser4,s, X8H9M8NOPP Or-c :t|}|dd}|di|S)z)Use tenacity to retry the embedding call.c^tjdi|}t|jS)N)requestspostr4r1)kwargsr3s r+_embed_with_retryz+embed_with_retry.._embed_with_retry6s$==*6*x}}//r-)r:rreturnrr7)r,)r(r:retry_decoratorr;s r+embed_with_retryr>2s.-j9O00  &v &&r-z0.0.29z1.0z%langchain_voyageai.VoyageAIEmbeddings)sinceremovalalternative_importceZdZUdZded<dZded<dZded<d ed < d Zd ed < dZd ed< dZ ded< dZ ded< e dZ e deddZ d d!dZ d" d#dZd$dZd%dZdd d&dZy)'raVoyage 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 truncationforbid)extrabefore)modectt|dd|d<d|vrd|d<tjdd|vrd|vr |ddvrdnd |d<|S) z?Validate that api key and python package exists in environment.rFVOYAGE_API_KEYrDz 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.rH)zvoyage-2z voyage-02H)rrr%warning)clsvaluess r+validate_environmentz%VoyageEmbeddings.validate_environmentjs$9 )9;K L$   & )F7O NN=  v %f$&/=V*V <  r-ctt|jj}|jdd|i|j |||j d|jd}|S)N AuthorizationzBearer )rDinput input_typerM)urlheadersr1timeout)r rrFget_secret_valuerErDrMrJ)selfr\r]api_keyparamss r+_invocation_paramsz#VoyageEmbeddings._invocation_paramssiy$"5"56GGI'''77))<=("oo  ++   r-c g}| |j}|jr$ ddlm}|t dt ||}nt dt ||}|r|dvrtd|d|D]@}t|fi|j||||z|} |jd| d DB|S#t$r}t d|d}~wwxYw) 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'.)r\r]c3&K|] }|d yw) embeddingNr7).0rs r+ z3VoyageEmbeddings._get_embeddings..sGa nGsr/) rHrL tqdm.autorg ImportErrorrangelen ValueErrorr>reextend) rbtextsrHr]r(rge_iterir3s r+_get_embeddingsz VoyageEmbeddings._get_embeddingss )+  J  ! ! *q#e*j9:E!SZ4E *,AAj\*   HA'))A N3 *H   Ghv6FG G H3 !> sB33 C < CC c>|j||jdS)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. rirHr]ryrH)rbrus r+embed_documentsz VoyageEmbeddings.embed_documentss'## doo*$  r-cF|j|g|jddS)zCall out to Voyage Embedding endpoint for embedding query text. Args: text: The text to embed. Returns: Embedding for the text. rhr{rr|)rbtexts r+ embed_queryzVoyageEmbeddings.embed_querys1## Ft7$   r-)r]c>|j||j|S)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. r{r|)rbrur]s r+embed_general_textsz$VoyageEmbeddings.embed_general_textss'## doo*$  r-)rXrr<r)N)r\ List[str]r] Optional[str]r<r)NN)rurrHz Optional[int]r]rr<List[List[float]])rurr<r)rrCr<z List[float])rurr]rr<r)__name__ __module__ __qualname____doc____annotations__rErFr$rJrLrMr model_configr classmethodrYreryr}rrr7r-r+rr>s  JCOSC*.N'.O?Krr7r-r+rs"    60LFF   8 $  '  > c y*c  c r-