BvhqtddlmZmZmZmZddlmZddlmZm Z m Z m Z ddl Z ddl ZddlZGddeeZy))EmbeddingFunctionSpace Embeddings Documentsvalidate_config_schema)ListDictAnyOptionalNc eZdZdZ ddeedededeedef dZd ed e fd Z e d efd Z d e fd Zd ee fdZe deeefd dfdZd eeeffdZdeeefdeeefd dfdZe deeefd dfdZy)VoyageAIEmbeddingFunctionz_ This class is used to generate embeddings for a list of texts using the VoyageAI API. Napi_key model_nameapi_key_env_var input_type truncationcb ddl}|tjdt ||_|xstj||_ |jstd|d||_ ||_ ||_ |j|j|_y#t$r tdwxYw)a Initialize the VoyageAIEmbeddingFunction. Args: api_key_env_var (str, optional): Environment variable name that contains your API key for the VoyageAI API. Defaults to "CHROMA_VOYAGE_API_KEY". model_name (str, optional): The name of the model to use for text embeddings. Defaults to "voyage-large-2". api_key (str, optional): API key for the VoyageAI API. If not provided, will look for it in the environment variable. input_type (str, optional): The type of input to use for the VoyageAI API. Defaults to None. truncation (bool): Whether to truncate the input text. Defaults to True. rNz[The voyageai python package is not installed. Please install it with `pip install voyageai`zDirect api_key configuration will not be persisted. Please use environment variables via api_key_env_var for persistent storage.zThe z! environment variable is not set.)r)voyageai ImportError ValueErrorwarningswarnDeprecationWarningrosgetenvrrrrClient_client)selfrrrrrrs h/RAG/venv/lib/python3.12/site-packages/chromadb/utils/embedding_functions/voyageai_embedding_function.py__init__z"VoyageAIEmbeddingFunction.__init__s,     MM_"  /<"))O"< ||tO#44UVW W$$$t||< ) m  s BB.inputreturnc|jj||j|j|j}|j Dcgc]'}t j|t j)c}Scc}w)z Generate embeddings for the given documents. Args: input: Documents to generate embeddings for. Returns: Embeddings for the documents. )textsmodelrr)dtype) rembedrrr embeddingsnparrayfloat32)rr"r) embeddings r __call__z"VoyageAIEmbeddingFunction.__call__<sp\\''// ( DNCXCX CXiBHHYbjj 1CX   s ,A;cy)Nrr0r namezVoyageAIEmbeddingFunction.nameRsr1cy)Ncosiner0rs r default_spacez'VoyageAIEmbeddingFunction.default_spaceVsr1c gdS)N)r4l2ipr0r5s r supported_spacesz*VoyageAIEmbeddingFunction.supported_spacesYs%%r1configzEmbeddingFunction[Documents]c|jd}|jd}|jd}|jd}||Jdt|||||SdS)NrrrrzThis code should not be reachedTrrrr)getr)r;rrrrs r build_from_configz+VoyageAIEmbeddingFunction.build_from_config\s **%67ZZ - ZZ - ZZ -  "j&8 ;; ;5(+!!%/%;z   BF   r1c`|j|j|j|jdS)Nr=r=r5s r get_configz$VoyageAIEmbeddingFunction.get_configms*#33//////   r1 old_config new_configc"d|vr tdy)NrzSThe model name cannot be changed after the embedding function has been initialized.)r)rrBrCs r validate_config_updatez0VoyageAIEmbeddingFunction.validate_config_updateus! : %e  &r1ct|dy)z Validate the configuration using the JSON schema. Args: config: Configuration to validate Raises: ValidationError: If the configuration does not match the schema rNr)r;s r validate_configz)VoyageAIEmbeddingFunction.validate_config}s vz2r1)Nzvoyage-large-2CHROMA_VOYAGE_API_KEYNT)__name__ __module__ __qualname____doc__r strboolr!rrr. staticmethodr2rr6r r:r r r?rArErGr0r1r rr s6 "&*6$( ,=#,=,= ,= SM ,=  ,=\ i J ,#u&$u+& $sCx. 5S   DcN sCx.6:38n  3S#X 34 3 3r1r)chromadb.api.typesrrrr*chromadb.utils.embedding_functions.schemasrtypingr r r r rnumpyr*rrr0r1r rTs1NNM,, 3 1) <3r1