dhSrSSKrSSKJr SSKJrJrJr SSKJ r SSK J r J r J r SSKJr S/r\"\5r"S S\ \ 5rg) z-written under MIT Licence, Michael Feil 2023.N) getLogger)AnyListOptional) Embeddings) BaseModel ConfigDictmodel_validator)SelfInfinityEmbeddingsLocalcP\rSrSr%Sr\\S'Sr\\\S'Sr \ \S'Sr \\S 'S r \\S 'S r \\S 'Sr\\S'\"SSS9r\"SS9S\4Sj5rS SjrS\SS4SjrS\\S\\\4SjrS\S\\4SjrS\\S\\\4SjrS\S\\4SjrSrg)!r amOptimized Infinity embedding models. https://github.com/michaelfeil/infinity This class deploys a local Infinity instance to embed text. The class requires async usage. Infinity is a class to interact with Embedding Models on https://github.com/michaelfeil/infinity Example: .. code-block:: python from langchain_community.embeddings import InfinityEmbeddingsLocal async with InfinityEmbeddingsLocal( model="BAAI/bge-small-en-v1.5", revision=None, device="cpu", ) as embedder: embeddings = await engine.aembed_documents(["text1", "text2"]) modelNrevision batch_sizeautodevicetorchbackendT model_warmupengineforbid)extraprotected_namespacesafter)modereturnc SSKJn U"URURUR UR URURS9Ul U$![a [S5ef=f)z?Validate that api key and python package exists in environment.r)AsyncEmbeddingEnginezrPlease install the `pip install 'infinity_emb[optimum,torch]>=0.0.24'` package to use the InfinityEmbeddingsLocal.)model_name_or_pathrrrrr) infinity_embr! ImportErrorrrrrrrr)selfr!s e/var/www/html/shao/venv/lib/python3.13/site-packages/langchain_community/embeddings/infinity_local.pyvalidate_environment,InfinityEmbeddingsLocal.validate_environmentBsn  9+#zz;;]]**<<     >  s AA,cT# URR5IShvN gN7f)astart the background worker. recommended usage is with the async with statement. async with InfinityEmbeddingsLocal( model="BAAI/bge-small-en-v1.5", revision=None, device="cpu", ) as embedder: embeddings = await engine.aembed_documents(["text1", "text2"]) N)r __aenter__)r%s r&r*"InfinityEmbeddingsLocal.__aenter__Xskk$$&&&s (&(argscR# URR"U6IShvN gN7f)zMstop the background worker, required to free references to the pytorch model.N)r __aexit__)r%r,s r&r.!InfinityEmbeddingsLocal.__aexit__eskk##T***s '%'textsc# URR(dl[RS5 UIShvN URR U5IShvN up#SSS5IShvN [RS5 W$URR U5IShvN up#U$NwNVNF!,IShvN (df  N[=fN(7f)zAsync call out to Infinity's embedding endpoint. Args: texts: The list of texts to embed. Returns: List of embeddings, one for each text. zeStarting Infinity engine on the fly. This is not recommended.Please start the engine before using it.Nz$Stopped infinity engine after usage.)rrunningloggerwarningembed)r%r0 embeddings_s r&aembed_documents(InfinityEmbeddingsLocal.aembed_documentsjs{{"" NN; t&*kk&7&7&> > t NNA B#'++"3"3E"::MJ >ttt ;sb7CB1CB7B3B7# C.B5/9C(C) C3B75C7C=C > C CtextcL# URU/5IShvN nUS$N 7f)z|Async call out to Infinity's embedding endpoint. Args: text: The text to embed. Returns: Embeddings for the text. Nr)r8)r%r:r6s r& aembed_query$InfinityEmbeddingsLocal.aembed_querys+ 00$88 !}9s $" $cv[RS5 [R"UR U55$)z This method is async only. zQThis method is async only. Please use the async version `await aembed_documents`.)r3r4asynciorunr8)r%r0s r&embed_documents'InfinityEmbeddingsLocal.embed_documentss2  E {{400788cv[RS5 [R"UR U55$)zMThis method is async only. Please use the async version `await aembed_query`.)r3r4r?r@r<)r%r:s r& embed_query#InfinityEmbeddingsLocal.embed_querys0 B {{4,,T233rC)r)rN)__name__ __module__ __qualname____firstlineno____doc__str__annotations__rrrintrrrboolrrr model_configr r r'r*r.rfloatr8r<rArF__static_attributes__rrCr&r r s* JG"Hhsm"5J0FCAGSG%L$/FC*L '"d#* '+S+T+ DI$tE{:K. s tE{ 9T#Y94U 3D944U 4rC)rLr?loggingrtypingrrrlangchain_core.embeddingsrpydanticrr r typing_extensionsr __all__rHr3r rrCr&rZsB3&&0;;" $ % 8 M4iM4rC