9i" dZddlZddlmZddlmZddlmZdddd d d d dd Zde fdZ de de e e ffdZ ddde de dzde e e ffdZ ejeede ddfdZddde de dzdedefdZddgZy)z!Factory functions for embeddings.N)util)Any) Embeddingslangchain_openai langchain_awslangchain_coherelangchain_google_vertexailangchain_huggingfacelangchain_mistralailangchain_ollama) azure_openaibedrockcoheregoogle_vertexai huggingface mistralaiollamaopenaireturncVdjdtjDS)z3Get formatted list of providers and their packages. c3RK|]\}}d|d|jdd!yw)z - z: _-N)replace).0ppkgs c/var/www/html/backtest/airagagent/rag_env/lib/python3.12/site-packages/langchain/embeddings/base.py z%_get_provider_list..s-dVQtA3bS#!6 78ds%')join_SUPPORTED_PROVIDERSitemsr_get_provider_listr&s! 99dG[GaGaGcd ddr% model_namec.d|vrt}d|d|}t||jdd\}}|jj }|j }|tvrd|dt }t||s d}t|||fS)aParse a model string into provider and model name components. The model string should be in the format 'provider:model-name', where provider is one of the supported providers. Args: model_name: A model string in the format 'provider:model-name' Returns: A tuple of (provider, model_name) ```python _parse_model_string("openai:text-embedding-3-small") # Returns: ("openai", "text-embedding-3-small") _parse_model_string("bedrock:amazon.titan-embed-text-v1") # Returns: ("bedrock", "amazon.titan-embed-text-v1") ``` Raises: ValueError: If the model string is not in the correct format or the provider is unsupported :zInvalid model format 'z'. Model name must be in format 'provider:model-name' Example valid model strings: - openai:text-embedding-3-small - bedrock:amazon.titan-embed-text-v1 - cohere:embed-english-v3.0 Supported providers:  Provider 'E' is not supported. Supported providers and their required packages: Model name cannot be empty)r" ValueErrorsplitlowerstripr&)r' providersmsgprovidermodels r_parse_model_stringr6s2 *( $ZL1$ %.;  0 o &&sA.OHe~~%%'H KKME++ #A!#$ & o *o U?r%r4r5r4c|js d}t||d|vrt|\}}n|}|st}d|}t||tvrd|dt }t|||fS)Nr-r)zMust specify either: 1. A model string in format 'provider:model-name' Example: 'openai:text-embedding-3-small' 2. Or explicitly set provider from: r+r,)r1r.r6r"r&)r5r4r3r'r2s r_infer_model_and_providerr9Qs ;;=*oC5L259* (  3k   o++ #A!#$ & o Z r%)maxsizercVtj|sd|d|d}t|y)z Check if a package is installed.zCould not import z5 python package. Please install it with `pip install `N)r find_spec ImportError)rr3s r _check_pkgr?ss8 >># !#&[\_[``ab# r%kwargsc :|s3tj}ddj|}t|t ||\}}t|}t ||dk(rddlm}|dd|i|S|dk(rdd lm}|dd|i|S|d k(rdd l m } | dd|i|S|d k(rdd l m } | dd|i|S|dk(rddl m} | dd|i|S|dk(rddlm} | dd|i|S|dk(rddlm} | dd|i|S|dk(rddlm}|dd|i|Sd|dt+}t|)ap Initialize an embedding model from a model name and optional provider. !!! note Requires the integration package for the chosen model provider to be installed. See the `model_provider` parameter below for specific package names (e.g., `pip install langchain-openai`). Refer to the [provider integration's API reference](https://docs.langchain.com/oss/python/integrations/providers) for supported model parameters to use as `**kwargs`. Args: model: The name of the model, e.g. `'openai:text-embedding-3-small'`. You can also specify model and model provider in a single argument using `'{model_provider}:{model}'` format, e.g. `'openai:text-embedding-3-small'`. provider: The model provider if not specified as part of the model arg (see above). Supported `provider` values and the corresponding integration package are: - `openai` -> [`langchain-openai`](https://docs.langchain.com/oss/python/integrations/providers/openai) - `azure_openai` -> [`langchain-openai`](https://docs.langchain.com/oss/python/integrations/providers/openai) - `bedrock` -> [`langchain-aws`](https://docs.langchain.com/oss/python/integrations/providers/aws) - `cohere` -> [`langchain-cohere`](https://docs.langchain.com/oss/python/integrations/providers/cohere) - `google_vertexai` -> [`langchain-google-vertexai`](https://docs.langchain.com/oss/python/integrations/providers/google) - `huggingface` -> [`langchain-huggingface`](https://docs.langchain.com/oss/python/integrations/providers/huggingface) - `mistraiai` -> [`langchain-mistralai`](https://docs.langchain.com/oss/python/integrations/providers/mistralai) - `ollama` -> [`langchain-ollama`](https://docs.langchain.com/oss/python/integrations/providers/ollama) **kwargs: Additional model-specific parameters passed to the embedding model. These vary by provider. Refer to the specific model provider's [integration reference](https://reference.langchain.com/python/integrations/) for all available parameters. Returns: An `Embeddings` instance that can generate embeddings for text. Raises: ValueError: If the model provider is not supported or cannot be determined ImportError: If the required provider package is not installed ???+ example ```python # pip install langchain langchain-openai # Using a model string model = init_embeddings("openai:text-embedding-3-small") model.embed_query("Hello, world!") # Using explicit provider model = init_embeddings(model="text-embedding-3-small", provider="openai") model.embed_documents(["Hello, world!", "Goodbye, world!"]) # With additional parameters model = init_embeddings("openai:text-embedding-3-small", api_key="sk-...") ``` !!! version-added "Added in `langchain` 0.3.9" z2Must specify model name. Supported providers are: z, r7rr)OpenAIEmbeddingsr5r )AzureOpenAIEmbeddingsr)VertexAIEmbeddingsr)BedrockEmbeddingsmodel_idr)CohereEmbeddingsr)MistralAIEmbeddingsr)HuggingFaceEmbeddingsr'r)OllamaEmbeddingsr+r,r$)r"keysr!r.r9r?rrBrCr rDrrErrGr rHr rIr rJr&)r5r4r@r2r3r'rrBrCrDrErGrHrIrJs rinit_embeddingsrL{stL (--/ B499YCWBXYo4UXNHj x (CsO85;j;F;;>!:$@:@@@$$@!= =f==93 ?*???85;j;F;;;;">>v>>= ?$E EfEE85;j;F;; XJ=    " S/r%rrL)__doc__ functools importlibrtypingrlangchain_core.embeddingsrr"strr&tupler6r9 lru_cachelenr?rL__all__r$r%rrWs'0' 2*&   eCe 4C4E#s(O4t   Dj  38_  DS!567CD8 t tDjt t  tp r%