3i7SSKJrJrJrJr SSKJrJrJrJ r J r SSK r SSK r SSKJr SSKJr SSKr"SS\\5r"SS\\5r"S S \\5rg) ) Embeddings DocumentsEmbeddingFunctionSpace)ListDictAnycastOptionalNvalidate_config_schemac\rSrSrSrSS\\S\S\4SjjrS\S \ 4S jr \ S \4S j5r S \ 4S jrS \\ 4S jr\ S\\\4S S4Sj5rS \\\44SjrS\\\4S\\\4S S4Sjr\ S\\\4S S4Sj5rSrg)GooglePalmEmbeddingFunction zvTo use this EmbeddingFunction, you must have the google.generativeai Python package installed and have a PaLM API key.Napi_key model_nameapi_key_env_varc^SSKJn Ub[R "S[ 5 X0lU=(d [R"U5Ul UR(d[SUS35eX l URURS9 X@l g![a [S5ef=f)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 v/home/james-whalen/.local/lib/python3.13/site-packages/chromadb/utils/embedding_functions/google_embedding_function.py__init__$GooglePalmEmbeddingFunction.__init__ s  .   MM_"  /<"))O"< ||tO#44UVW W$ t||, ' E  s BB,inputreturnc [SU55(d [S5eUVs/sHKn[R"URR UR US9S[RS9PMM sn$s snf) Generate embeddings for the given documents. Args: input: Documents or images to generate embeddings for. Returns: Embeddings for the documents. c3B# UHn[U[5v M g7fN isinstancestr.0items r' 7GooglePalmEmbeddingFunction.__call__..>;UT:dC((Uz4Google PaLM only supports text documents, not images)modeltext embeddingdtype)allrnparrayr$generate_embeddingsrfloat32)r%r*r;s r'__call__$GooglePalmEmbeddingFunction.__call__3s;U;;;ST T  HH ..T__4.Pjj      sAA<cg)N google_palmrHr'name GooglePalmEmbeddingFunction.nameKsrIcgNcosinerHr%s r' default_space)GooglePalmEmbeddingFunction.default_spaceOrIc /SQ$N)rNl2iprHrOs r'supported_spaces,GooglePalmEmbeddingFunction.supported_spacesR%%rIconfigEmbeddingFunction[Documents]crURS5nURS5nUbUcS5e[XS9$)NrrThis code should not be reachedrr)getr)rZrrs r'build_from_config-GooglePalmEmbeddingFunction.build_from_configUsE **%67ZZ -  "j&8 ;; ;5*+  rIc4URURS.$)Nr^r^rOs r' get_config&GooglePalmEmbeddingFunction.get_configas#'#7#7tWWrI old_config new_configc&SU;a [S5eg)NrSThe model name cannot be changed after the embedding function has been initialized.rr%rerfs r'validate_config_update2GooglePalmEmbeddingFunction.validate_config_updateds! : %e  &rIc[US5 g) Validate the configuration using the JSON schema. Args: config: Configuration to validate Raises: ValidationError: If the configuration does not match the schema rGNr rZs r'validate_config+GooglePalmEmbeddingFunction.validate_configls v}5rI)r$rrr)Nzmodels/embedding-gecko-001CHROMA_GOOGLE_PALM_API_KEY__name__ __module__ __qualname____firstlineno____doc__r r2r(rrrD staticmethodrJrrPrrWrr r`rcrkrp__static_attributes__rHrIr'rr sA"&6; $#$$ $L i J 0#u&$u+&  $sCx.  5S    XDcNXsCx.6:38n  6S#X 64 6 6rIrc \rSrSrSrSS\\S\S\S\4SjjrS \S \ 4S jr \ S \4S j5r S \ 4S jrS \\ 4Sjr\ S\\\4S S4Sj5rS \\\44SjrS\\\4S\\\4S S4Sjr\ S\\\4S S4Sj5rSrg)#GoogleGenerativeAiEmbeddingFunctionzzxTo use this EmbeddingFunction, you must have the google.generativeai Python package installed and have a Google API key.Nrr task_typercjSSKJn Ub[R "S[ 5 X@lU=(d [R"U5Ul UR(d[SUS35eX l X0l URURS9 XPlg![a [S5ef=f)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)rrrrrrr rr!r"rrr~r#_genai)r%rrr~rgenais r'r(,GoogleGenerativeAiEmbeddingFunction.__init__}s&  /   MM_"  /<"))O"< ||tO#44UVW W$"  - ) E  s BB2r*r+cD[SU55(d [S5e/nUHgnURRURUUR S9nUR [R"US[RS95 Mi [[U5$)r-c3B# UHn[U[5v M g7fr/r0r3s r'r6?GoogleGenerativeAiEmbeddingFunction.__call__..r8r9z=Google Generative AI only supports text documents, not images)r:contentr~r<r=) r?rr embed_contentrr~appendr@rArCr rr%r*embeddings_listr;embedding_results r'rD,GoogleGenerativeAiEmbeddingFunction.__call__s;U;;;O :<D#{{88oo.. 9    " ")+6bjjI  J00rIcg)Ngoogle_generative_airHrHrIr'rJ(GoogleGenerativeAiEmbeddingFunction.names%rIcgrMrHrOs r'rP1GoogleGenerativeAiEmbeddingFunction.default_spacerRrIc /SQ$rTrHrOs r'rW4GoogleGenerativeAiEmbeddingFunction.supported_spacesrYrIrZr[cURS5nURS5nURS5nUbUbUcS5e[XUS9$)Nrrr~r]rrr~)r_r|)rZrrr~s r'r`5GoogleGenerativeAiEmbeddingFunction.build_from_configsZ **%67ZZ - JJ{+  "j&8IrI)rrrrr~)Nzmodels/embedding-001RETRIEVAL_DOCUMENTCHROMA_GOOGLE_GENAI_API_KEYrsrHrIr'r|r|zs&C"&0-< )#)) )  )V1i1J1<&#&&u&$u+&  $sCx.  5S     DcN  sCx. 6:38n    ?S#X ?4 ? ?rIr|c $\rSrSrSrSS\\S\S\S\S\4 S jjrS \S \ 4S jr \ S \4S j5r S \ 4SjrS \\ 4Sjr\ S\\\4S S4Sj5rS \\\44SjrS\\\4S\\\4S S4Sjr\ S\\\4S S4Sj5rSrg)GoogleVertexEmbeddingFunctionzTo use this EmbeddingFunction, you must have the vertexai Python package installed and have Google Cloud credentials configured.Nrr project_idregionrcSSKnSSKJn Ub[ R "S[5 XPlU=(d [R"U5Ul UR(d[ SUS35eX l X0l X@lURX4S9 UR!U5Ulg![a [ S5ef=f) 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_modelsrrrrrr rr!r"rrrrinitfrom_pretrained_model)r%rrrrrrrs r'r(&GoogleVertexEmbeddingFunction.__init__s*   C   MM_"  /<"))O"< ||tO#44UVW W$$  j :(88D + |  s B,,Cr*r+c2[SU55(d [S5e/nUH^nURRU/5nUR [ R "USR[ RS95 M` [[U5$)r-c3B# UHn[U[5v M g7fr/r0r3s r'r69GoogleVertexEmbeddingFunction.__call__..;r8r9z6Google Vertex only supports text documents, not imagesrr=) r?rrget_embeddingsrr@rAvaluesrCr rrs r'rD&GoogleVertexEmbeddingFunction.__call__0s;U;;;UV V9;D#{{994&A   " ")!,332::F J00rIcg)N google_vertexrHrHrIr'rJ"GoogleVertexEmbeddingFunction.nameHsrIcgrMrHrOs r'rP+GoogleVertexEmbeddingFunction.default_spaceLrRrIc /SQ$rTrHrOs r'rW.GoogleVertexEmbeddingFunction.supported_spacesOrYrIrZr[cURS5nURS5nURS5nURS5nUb UbUbUcS5e[UUUUS9$)Nrrrrr]rrrr)r_r)rZrrrrs r'r`/GoogleVertexEmbeddingFunction.build_from_configRsx **%67ZZ - ZZ - H%  #!!~ ;; ;5,+!!   rIc`URURURURS.$)NrrrOs r'rc(GoogleVertexEmbeddingFunction.get_confighs*#33////kk   rIrerfcjSU;a [S5eSU;a [S5eSU;a [S5eg)NrrhrzSThe project ID cannot be changed after the embedding function has been initialized.rzOThe region cannot be changed after the embedding function has been initialized.rirjs r'rk4GoogleVertexEmbeddingFunction.validate_config_updatepsU : %e  : %e  z !a  "rIc[US5 g)rnrNr ros r'rp-GoogleVertexEmbeddingFunction.validate_configs v7rI)rrrrrr)Nztextembedding-geckozcloud-large-language-modelsz us-central1CHROMA_GOOGLE_VERTEX_API_KEYrsrHrIr'rrs9K"&/7#= -E#-E-E -E  -E  -E^1i1J10#u&$u+& $sCx. 5S  * DcN sCx.6:38n   8S#X 84 8 8rIr)chromadb.api.typesrrrrtypingrrr r r r!numpyr@ numpy.typingnpt*chromadb.utils.embedding_functions.schemasr rrr|rrHrIr'rs^NN22 Mm6"3I">m6`A?*;I*FA?HM8$5i$@M8rI