Ng (ddlmZddlZddlZddlmZmZmZmZm Z m Z m Z m Z ddl Z ddlmZddlmZddlmZmZddlmZmZmZmZddlmZmZmZmZeje Z!dd Z"ddZ#ddZ$edddGdd eeZ%dS)) annotationsN)AnyCallableDictListOptionalTupleUnioncast) deprecated) Embeddings)convert_to_secret_strget_from_dict_or_env) BaseModel ConfigDict SecretStrmodel_validator)before_sleep_logretrystop_after_attemptwait_exponential embeddingsVoyageEmbeddingsreturnCallable[[Any], Any]c d}d}tdt|jtd||t t t jS)N T) multiplierminmax)reraisestopwait before_sleep)rr max_retriesrrloggerloggingWARNING)r min_seconds max_secondss c/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain_community/embeddings/voyageai.py_create_retry_decoratorr.sWKK   6 7 7   M M M%fgo>>    responsedictcVd|vr$tdtj||S)NdatazVoyage API Error. Message: ) RuntimeErrorjsondumps)r0s r-_check_responser7,s3 XOH9M9MOOPPP Or/kwargsrc Ht|}|dd}|di|S)z)Use tenacity to retry the embedding call.r8rrcftjdi|}t|S)N)requestspostr7r5)r8r0s r-_embed_with_retryz+embed_with_retry.._embed_with_retry6s-=**6**x}}///r/)r8rrrr;)r.)rr8retry_decoratorr>s r-embed_with_retryr@2sF-j99O000_0   & &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 ded,dZ d-d.d!Z d/d0d%Zd1d&Zd2d)Zdd*d3d+ZdS)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 truncationforbid)extrabefore)modevaluesrrrctt|dd|d<d|vrd|d<tdd|vrd|vr |ddvrdnd |d<|S) z?Validate that api key and python package exists in environment.rHVOYAGE_API_KEYrFz 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.rJ)zvoyage-2z voyage-02H)rrr(warning)clsrTs r-validate_environmentz%VoyageEmbeddings.validate_environmentjs$9 )9;K L L$ $   & )F7O NN=    v % %f$$&/=V*V*V <  r/input List[str] input_type Optional[str]ctt|j}|jdd|i|j|||jd|jd}|S)N AuthorizationzBearer )rFr\r^rO)urlheadersr5timeout)r rrHget_secret_valuerGrFrOrL)selfr\r^api_keyparamss r-_invocation_paramsz#VoyageEmbeddings._invocation_paramssny$"566GGII'')<7)<)<=("o  +    r/texts Optional[int]List[List[float]]c g}||j}|jrR ddlm}n"#t$r}t d|d}~wwxYw|t dt ||}nt dt ||}|r|dvrtd|d|D]T}t|fi| ||||z|} | d| d DU|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'.)r\r^c3&K|] }|dV dS) embeddingNr;).0rs r- z3VoyageEmbeddings._get_embeddings..s&GGa nGGGGGGr/r3) rJrN tqdm.autorn ImportErrorrangelen ValueErrorr@riextend) rfrjrJr^rrne_iterir0s r-_get_embeddingsz VoyageEmbeddings._get_embeddingssz )+  J  ! 5 *******   !>  Dq#e**j99::EE!SZZ44E  *,AAAj   H HA'))A N 23 *H   GGhv6FGGG G G G Gs :5:c<|||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. rprJr^rrJ)rfrjs r-embed_documentsz VoyageEmbeddings.embed_documentss*## do*$   r/text List[float]cJ||g|jddS)zCall out to Voyage Embedding endpoint for embedding query text. Args: text: The text to embed. Returns: Embedding for the text. rorrr)rfrs r- embed_queryzVoyageEmbeddings.embed_querys4## Ft7$    r/)r^c<|||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. rr)rfrjr^s r-embed_general_textsz$VoyageEmbeddings.embed_general_textss*## do*$   r/)rTrrr)N)r\r]r^r_rr)NN)rjr]rJrkr^r_rrl)rjr]rrl)rrErr)rjr]r^r_rrl)__name__ __module__ __qualname____doc____annotations__rGrHr'rLrNrOr model_configr classmethodr[rirrrrr;r/r-rr>s  JJJCOCCCC*.N....OOO?Krs""""""                     666666000000LLLLLLLLFFFFFFFFFFFF  8 $ $     ' ' ' '   > c c c c c y*c c  c c c r/