Bvh7ddlmZmZmZmZddlmZmZmZm Z m Z ddl Z ddl Z ddlmZddlmZddlZGddeeZGddeeZGd d eeZy) ) Embeddings DocumentsEmbeddingFunctionSpace)ListDictAnycastOptionalNvalidate_config_schemaceZdZdZ ddeededefdZdedefd Z e defd Z de fd Z dee fd Ze d eeefddfdZdeeeffdZdeeefdeeefddfdZe d eeefddfdZy)GooglePalmEmbeddingFunctionzvTo use this EmbeddingFunction, you must have the google.generativeai Python package installed and have a PaLM API key.Napi_key model_nameapi_key_env_varcN ddlm}|t j dt ||_|xstj||_ |jstd|d||_ |j|j||_ y#t$r tdwxYw)a Initialize the GooglePalmEmbeddingFunction. Args: api_key_env_var (str, optional): Environment variable name that contains your API key for the Google PaLM API. Defaults to "CHROMA_GOOGLE_PALM_API_KEY". model_name (str, optional): The name of the model to use for text embeddings. Defaults to "models/embedding-gecko-001". rNrThe Google Generative AI python package is not installed. Please install it with `pip install google-generativeai`Direct api_key configuration will not be persisted. Please use environment variables via api_key_env_var for persistent storage.The ! environment variable is not set.r)google.generativeai generativeai ImportError ValueErrorwarningswarnDeprecationWarningrosgetenvrr configure_palm)selfrrrpalms f/RAG/venv/lib/python3.12/site-packages/chromadb/utils/embedding_functions/google_embedding_function.py__init__z$GooglePalmEmbeddingFunction.__init__ s  .   MM_"  /<"))O"< ||tO#44UVW W$ t||, ' E  s BB$inputreturnc td|Ds td|Dcgc]O}tj|jj |j |dtjQc}Scc}w) Generate embeddings for the given documents. Args: input: Documents or images to generate embeddings for. Returns: Embeddings for the documents. c3<K|]}t|tywN isinstancestr.0items r& z7GooglePalmEmbeddingFunction.__call__..>;UT:dC(Uz4Google PaLM only supports text documents, not images)modeltext embeddingdtype)allrnparrayr#generate_embeddingsrfloat32)r$r(r8s r&__call__z$GooglePalmEmbeddingFunction.__call__3s~;U;;ST T  HH ..T__4.Pjj      sAA9cy)N google_palmrDr&namez GooglePalmEmbeddingFunction.nameKsrEcyNcosinerDr$s r& default_spacez)GooglePalmEmbeddingFunction.default_spaceOrEc gdSN)rIl2iprDrJs r&supported_spacesz,GooglePalmEmbeddingFunction.supported_spacesR%%rEconfigEmbeddingFunction[Documents]cv|jd}|jd}||Jdt||S)NrrThis code should not be reachedrr)getr)rSrrs r&build_from_configz-GooglePalmEmbeddingFunction.build_from_configUsG **%67ZZ -  "j&8 ;; ;5*+   rEc4|j|jdS)NrWrWrJs r& get_configz&GooglePalmEmbeddingFunction.get_configas#'#7#7tWWrE old_config new_configc"d|vr tdy)NrSThe model name cannot be changed after the embedding function has been initialized.rr$r\r]s r&validate_config_updatez2GooglePalmEmbeddingFunction.validate_config_updateds! : %e  &rEct|dy) Validate the configuration using the JSON schema. Args: config: Configuration to validate Raises: ValidationError: If the configuration does not match the schema rCNr rSs r&validate_configz+GooglePalmEmbeddingFunction.validate_configls v}5rE)Nzmodels/embedding-gecko-001CHROMA_GOOGLE_PALM_API_KEY__name__ __module__ __qualname____doc__r r0r'rrrA staticmethodrFrrKrrQrr rYr[rbrfrDrEr&rr sA"&6; $#$$ $L i J 0#u&$u+&  $sCx.  5S    XDcNXsCx.6:38n  6S#X 64 6 6rErc eZdZdZ ddeedededefdZded efd Z e d efd Z d e fd Z d ee fd Ze deeefd dfdZd eeeffdZdeeefdeeefd dfdZe deeefd dfdZy)#GoogleGenerativeAiEmbeddingFunctionzxTo use this EmbeddingFunction, you must have the google.generativeai Python package installed and have a Google API key.Nrr task_typerc\ ddlm}|t j dt ||_|xstj||_ |jstd|d||_ ||_ |j|j||_y#t$r tdwxYw)a Initialize the GoogleGenerativeAiEmbeddingFunction. Args: api_key_env_var (str, optional): Environment variable name that contains your API key for the Google Generative AI API. Defaults to "CHROMA_GOOGLE_GENAI_API_KEY". model_name (str, optional): The name of the model to use for text embeddings. Defaults to "models/embedding-001". task_type (str, optional): The task type for the embeddings. Use "RETRIEVAL_DOCUMENT" for embedding documents and "RETRIEVAL_QUERY" for embedding queries. Defaults to "RETRIEVAL_DOCUMENT". rNrrrrr)rrrrrrrrr r!rrrpr"_genai)r$rrrprgenais r&r'z,GoogleGenerativeAiEmbeddingFunction.__init__}s&  /   MM_"  /<"))O"< ||tO#44UVW W$"  - ) E  s BB+r(r)c@td|Ds tdg}|D]k}|jj|j||j }|j tj|dtjmtt|S)r+c3<K|]}t|tywr-r.r1s r&r4z?GoogleGenerativeAiEmbeddingFunction.__call__..r5r6z=Google Generative AI only supports text documents, not images)r7contentrpr9r:) r<rrr embed_contentrrpappendr=r>r@r rr$r(embeddings_listr8embedding_results r&rAz,GoogleGenerativeAiEmbeddingFunction.__call__s;U;;O :<D#{{88oo.. 9    " ")+6bjjI  J00rEcy)Ngoogle_generative_airDrDrEr&rFz(GoogleGenerativeAiEmbeddingFunction.names%rEcyrHrDrJs r&rKz1GoogleGenerativeAiEmbeddingFunction.default_spacerLrEc gdSrNrDrJs r&rQz4GoogleGenerativeAiEmbeddingFunction.supported_spacesrRrErSrTc|jd}|jd}|jd}|||Jdt|||S)NrrrprVrrrp)rXro)rSrrrps r&rYz5GoogleGenerativeAiEmbeddingFunction.build_from_configs\ **%67ZZ - JJ{+  "j&8IrE)Nzmodels/embedding-001RETRIEVAL_DOCUMENTCHROMA_GOOGLE_GENAI_API_KEYrhrDrEr&rorozs&C"&0-< )#)) )  )V1i1J1<&#&&u&$u+&  $sCx.  5S     DcN  sCx. 6:38n    ?S#X ?4 ? ?rEroc eZdZdZ ddeededededef dZd ed efd Z e d efd Z d e fd Z d ee fdZe deeefd dfdZd eeeffdZdeeefdeeefd dfdZe deeefd dfdZy)GoogleVertexEmbeddingFunctionzTo use this EmbeddingFunction, you must have the vertexai Python package installed and have Google Cloud credentials configured.Nrr project_idregionrc~ ddl}ddlm}|t j dt||_|xstj||_ |jst d|d||_ ||_ ||_|j|||j!||_y#t$r t dwxYw) a Initialize the GoogleVertexEmbeddingFunction. Args: api_key_env_var (str, optional): Environment variable name that contains your API key for the Google Vertex AI API. Defaults to "CHROMA_GOOGLE_VERTEX_API_KEY". model_name (str, optional): The name of the model to use for text embeddings. Defaults to "textembedding-gecko". project_id (str, optional): The Google Cloud project ID. Defaults to "cloud-large-language-models". region (str, optional): The Google Cloud region. Defaults to "us-central1". rN)TextEmbeddingModelzjThe vertexai python package is not installed. Please install it with `pip install google-cloud-aiplatform`rrr)projectlocation)vertexaivertexai.language_modelsrrrrrrrr r!rrrrinitfrom_pretrained_model)r$rrrrrrrs r&r'z&GoogleVertexEmbeddingFunction.__init__s*   C   MM_"  /<"))O"< ||tO#44UVW W$$  j6 :(88D + |  s B''B<r(r)c(td|Ds tdg}|D]_}|jj|g}|j t j |djt jatt|S)r+c3<K|]}t|tywr-r.r1s r&r4z9GoogleVertexEmbeddingFunction.__call__..;r5r6z6Google Vertex only supports text documents, not imagesrr:) r<rrget_embeddingsrxr=r>valuesr@r rrys r&rAz&GoogleVertexEmbeddingFunction.__call__0s;U;;UV V9;D#{{994&A   " ")!,332::F J00rEcy)N google_vertexrDrDrEr&rFz"GoogleVertexEmbeddingFunction.nameHsrEcyrHrDrJs r&rKz+GoogleVertexEmbeddingFunction.default_spaceLrLrEc gdSrNrDrJs r&rQz.GoogleVertexEmbeddingFunction.supported_spacesOrRrErSrTc|jd}|jd}|jd}|jd}||||Jdt||||S)NrrrrrVrrrr)rXr)rSrrrrs r&rYz/GoogleVertexEmbeddingFunction.build_from_configRsx **%67ZZ - ZZ - H%  #!!~ ;; ;5,+!!   rEc`|j|j|j|jdS)NrrrJs r&r[z(GoogleVertexEmbeddingFunction.get_confighs*#33////kk   rEr\r]c^d|vr tdd|vr tdd|vr tdy)Nrr_rzSThe project ID cannot be changed after the embedding function has been initialized.rzOThe region cannot be changed after the embedding function has been initialized.r`ras r&rbz4GoogleVertexEmbeddingFunction.validate_config_updatepsU : %e  : %e  z !a  "rEct|dy)rdrNr res r&rfz-GoogleVertexEmbeddingFunction.validate_configs v7rE)Nztextembedding-geckozcloud-large-language-modelsz us-central1CHROMA_GOOGLE_VERTEX_API_KEYrhrDrEr&rrs9K"&/7#= -E#-E-E -E  -E  -E^1i1J10#u&$u+& $sCx. 5S  * DcN sCx.6:38n   8S#X 84 8 8rEr)chromadb.api.typesrrrrtypingrrr r r r numpyr= numpy.typingnpt*chromadb.utils.embedding_functions.schemasr rrrorrDrEr&rs^NN22 Mm6"3I">m6`A?*;I*FA?HM8$5i$@M8rE