o Zh @sddlmZddlZddlZddlmZmZmZmZm Z m Z m Z m Z ddl Z ddlmZddlmZddlmZmZddlmZmZmZmZddlmZmZmZmZee Z!dd dZ"dddZ#dddZ$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_exponential embeddingsVoyageEmbeddingsreturnCallable[[Any], Any]cCs0d}d}tdt|jtd||dtttjdS)N T) multiplierminmax)reraisestopwait before_sleep)rr max_retriesrrloggerloggingWARNING)rZ min_secondsZ max_secondsr)^/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/embeddings/voyageai.py_create_retry_decorators  r+responsedictcCs d|vrtdt||S)NdatazVoyage API Error. Message: ) RuntimeErrorjsondumps)r,r)r)r*_check_response,sr2kwargsrcKs$t|}|ddd}|di|S) z)Use tenacity to retry the embedding call.r3rrc[stjdi|}t|S)Nr))requestspostr2r0)r3r,r)r)r*_embed_with_retry6s z+embed_with_retry.._embed_with_retryN)r3rrrr))r+)rr3Zretry_decoratorr6r)r)r*embed_with_retry2s r7z0.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 dded2ddZ d3d4d"d#Z  d5d6d'd(Zd7d)d*Zd8d-d.Zdd/d9d0d1ZdS):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 truncationZforbid)extrabefore)modevaluesrrrcCsZtt|dd|d<d|vrd|d<tdd|vr+d|vr'|ddvr'dnd |d<|S) z?Validate that api key and python package exists in environment.r;ZVOYAGE_API_KEYr9z 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)clsrFr)r)r*validate_environmentjs z%VoyageEmbeddings.validate_environmentinput List[str] input_type Optional[str]cCs>tt|j}|jdd|i|j|||jd|jd}|S)N AuthorizationzBearer )r9rLrNrB)urlheadersr0timeout)r rr;Zget_secret_valuer:r9rBr?)selfrLrNZapi_keyparamsr)r)r*_invocation_paramss  z#VoyageEmbeddings._invocation_paramstexts Optional[int]List[List[float]]c Csg}|dur |j}|jr1zddlm}Wnty%}ztd|d}~ww|tdt||}ntdt||}|rG|dvrGtd|d|D]!}t|fi|j |||||d} | dd | d DqI|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'.)rLrNcss|]}|dVqdS)Z embeddingNr)).0rr)r)r* sz3VoyageEmbeddings._get_embeddings..r.) r=rAZ tqdm.autorZ ImportErrorrangelen ValueErrorr7rVextend) rTrWr=rNrrZeZ_iterir,r)r)r*_get_embeddingss<  z VoyageEmbeddings._get_embeddingscCs|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. r\r=rNrgr=)rTrWr)r)r*embed_documentss z VoyageEmbeddings.embed_documentstext List[float]cCs|j|g|jdddS)zCall out to Voyage Embedding endpoint for embedding query text. Args: text: The text to embed. Returns: Embedding for the text. r[rhrri)rTrkr)r)r* embed_querys  zVoyageEmbeddings.embed_query)rNcCs|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. rhri)rTrWrNr)r)r*embed_general_textss z$VoyageEmbeddings.embed_general_texts)rFrrr)N)rLrMrNrOrr)NN)rWrMr=rXrNrOrrY)rWrMrrY)rkr8rrl)rWrMrNrOrrY)__name__ __module__ __qualname____doc____annotations__r:r;r%r?rArBrZ model_configr classmethodrKrVrgrjrmrnr)r)r)r*r>s<         ) )rrrr)r,r-rr-)rrr3rrr)& __future__rr0r'typingrrrrrrr r r4Zlangchain_core._api.deprecationr Zlangchain_core.embeddingsr Zlangchain_core.utilsr rZpydanticrrrrtenacityrrrr getLoggerror&r+r2r7rr)r)r)r*s( (