3i SSKJrJrJrJr SSKJrJrJrJ r SSK r SSK r SSK Jr SSKr"SS\\5r"SS\\5rg) ) Embeddings DocumentsEmbeddingFunctionSpace)ListDictAnyOptionalNvalidate_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)HuggingFaceEmbeddingFunction z This class is used to get embeddings for a list of texts using the HuggingFace API. It requires an API key and a model name. The default model name is "sentence-transformers/all-MiniLM-L6-v2". Napi_key model_nameapi_key_env_varcSSKnUb[R"S[ 5 X0lU=(d [R"U5Ul UR(d[SUS35eX l SU3Ul UR5Ul URRRSS UR305 g![a [S5ef=f) aj Initialize the HuggingFaceEmbeddingFunction. Args: api_key_env_var (str, optional): Environment variable name that contains your API key for the HuggingFace API. Defaults to "CHROMA_HUGGINGFACE_API_KEY". model_name (str, optional): The name of the model to use for text embeddings. Defaults to "sentence-transformers/all-MiniLM-L6-v2". rNUThe httpx python package is not installed. Please install it with `pip install httpx`Direct 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.zAhttps://api-inference.huggingface.co/pipeline/feature-extraction/ AuthorizationBearer )httpx ImportError ValueErrorwarningswarnDeprecationWarningrosgetenvrr_api_urlClient_sessionheadersupdate)selfrrrrs {/home/james-whalen/.local/lib/python3.13/site-packages/chromadb/utils/embedding_functions/huggingface_embedding_function.py__init__%HuggingFaceEmbeddingFunction.__init__s     MM_"  /<"))O"< ||tO#44UVW W$[\f[gh    $$o7O%PQ) g  s CCinputreturncURRURUSS0S.S9R5nUVs/sH&n[R "U[R S9PM( sn$s snf)am Get the embeddings for a list of texts. Args: input (Documents): A list of texts to get embeddings for. Returns: Embeddings: The embeddings for the texts. Example: >>> hugging_face = HuggingFaceEmbeddingFunction(api_key_env_var="CHROMA_HUGGINGFACE_API_KEY") >>> texts = ["Hello, world!", "How are you?"] >>> embeddings = hugging_face(texts) wait_for_modelT)inputsoptionsjsondtyper"postr r0nparrayfloat32r%r)response embeddings r&__call__%HuggingFaceEmbeddingFunction.__call__6si ==%% MM!/?.FG&  $&  HPPx)"**5xPPPs-A,cg)N huggingfacer?r&name!HuggingFaceEmbeddingFunction.nameNsr@cgNcosiner?r%s r& default_space*HuggingFaceEmbeddingFunction.default_spaceRr@c /SQ$N)rEl2ipr?rFs r&supported_spaces-HuggingFaceEmbeddingFunction.supported_spacesU%%r@configEmbeddingFunction[Documents]crURS5nURS5nUbUcS5e[XS9$)NrrzThis code should not be reachedrr)getr)rQrrs r&build_from_config.HuggingFaceEmbeddingFunction.build_from_configXsE **%67ZZ -  "j&8 ;; ;5++  r@c4URURS.$)NrTrTrFs r& get_config'HuggingFaceEmbeddingFunction.get_configds#'#7#7tWWr@ old_config new_configc&SU;a [S5eg)NrzSThe model name cannot be changed after the embedding function has been initialized.)rr%r[r\s r&validate_config_update3HuggingFaceEmbeddingFunction.validate_config_updategs! : %e  &r@c[US5 g) Validate the configuration using the JSON schema. Args: config: Configuration to validate Raises: ValidationError: If the configuration does not match the schema r>Nr rQs r&validate_config,HuggingFaceEmbeddingFunction.validate_configos v}5r@)r r"rrr)Nz&sentence-transformers/all-MiniLM-L6-v2CHROMA_HUGGINGFACE_API_KEY)__name__ __module__ __qualname____firstlineno____doc__r strr'rrr; staticmethodrArrGrrNrr rVrYr_rd__static_attributes__r?r@r&rr s!"&B; %R#%R%R %RNQiQJQ0#u&$u+&  $sCx.  5S    XDcNXsCx.6:38n  6S#X 64 6 6r@rc\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)HuggingFaceEmbeddingServer}z This class is used to get embeddings for a list of texts using the HuggingFace Embedding server (https://github.com/huggingface/text-embeddings-inference). The embedding model is configured in the server. NurlrrcSSKnUb[R"S[ 5 XlX lURb/U=(d [R"UR5Ul OX0l UUl UR5Ul URb5URRRSSUR305 gg![a [S5ef=f)a1 Initialize the HuggingFaceEmbeddingServer. Args: url (str): The URL of the HuggingFace Embedding Server. api_key (Optional[str]): The API key for the HuggingFace Embedding Server. api_key_env_var (str, optional): Environment variable name that contains your API key for the HuggingFace API. rNrrrr)rrrrrrrrrrrrr r!r"r#r$)r%rrrrrs r&r'#HuggingFaceEmbeddingServer.__init__s     MM_"  .    +"Ebii0D0D&EDL"L%   << # MM ! ! ( (/WT\\N;S)T U $/ g  s CC(r)r*cURRURSU0S9R5nUVs/sH&n[R "U[R S9PM( sn$s snf)a` Get the embeddings for a list of texts. Args: input (Documents): A list of texts to get embeddings for. Returns: Embeddings: The embeddings for the texts. Example: >>> hugging_face = HuggingFaceEmbeddingServer(url="http://localhost:8080/embed") >>> texts = ["Hello, world!", "How are you?"] >>> embeddings = hugging_face(texts) r-r/r1r3r8s r&r;#HuggingFaceEmbeddingServer.__call__sZ ==%%dmm8U:K%LQQSHPPx)"**5xPPPs-A)cg)Nhuggingface_serverr?r?r@r&rAHuggingFaceEmbeddingServer.names#r@cgrDr?rFs r&rG(HuggingFaceEmbeddingServer.default_spacerIr@c /SQ$rKr?rFs r&rN+HuggingFaceEmbeddingServer.supported_spacesrPr@rQrRctURS5nURS5nUc [S5e[XS9$)Nrrrz3URL must be provided for HuggingFaceEmbeddingServerrrr)rUrrp)rQrrrs r&rV,HuggingFaceEmbeddingServer.build_from_configs;jj **%67 ;RS S)cSSr@c4URURS.$)NrrrFs r&rY%HuggingFaceEmbeddingServer.get_configsxxD4H4HIIr@r[r\cNSU;aUSUR:wa [S5egg)NrrzLThe URL cannot be changed after the embedding function has been initialized.)rrrr^s r&r_1HuggingFaceEmbeddingServer.validate_config_updates5 J :e#4#@^ $A r@c[US5 g)rbrxNr rcs r&rd*HuggingFaceEmbeddingServer.validate_configs v';rsFNN,, Mq6#4Y#?q6ho=!29!=o=r@