tj8DdZddlmZddlZddlZddlZddlmZmZm Z ddl m Z ddl m Z ee egeeefZ ee egeeeefZ dd ZGd d e ZddZddZddZejd dZgdZdS)!anUtilities for working with embedding functions and LangChain's Embeddings interface. This module provides tools to wrap arbitrary embedding functions (both sync and async) into LangChain's Embeddings interface. This enables using custom embedding functions with LangChain-compatible tools while maintaining support for both synchronous and asynchronous operations. ) annotationsN) AwaitableCallableSequence)Any) Embeddingsembed:Embeddings | EmbeddingsFunc | AEmbeddingsFunc | str | NonereturnrcL|tdt|trYt}|>ddlm}m} |d}d|d}n #|$rd}YnwxYwtd |d |d ||St|tr|St|S) aEnsure that an embedding function conforms to LangChain's Embeddings interface. This function wraps arbitrary embedding functions to make them compatible with LangChain's Embeddings interface. It handles both synchronous and asynchronous functions. Args: embed: Either an existing Embeddings instance, or a function that converts text to embeddings. If the function is async, it will be used for both sync and async operations. Returns: An Embeddings instance that wraps the provided function(s). ??? example "Examples" Wrap a synchronous embedding function: ```python def my_embed_fn(texts): return [[0.1, 0.2] for _ in texts] embeddings = ensure_embeddings(my_embed_fn) result = embeddings.embed_query("hello") # Returns [0.1, 0.2] ``` Wrap an asynchronous embedding function: ```python async def my_async_fn(texts): return [[0.1, 0.2] for _ in texts] embeddings = ensure_embeddings(my_async_fn) result = await embeddings.aembed_query("hello") # Returns [0.1, 0.2] ``` Initialize embeddings using a provider string: ```python # Requires langchain>=0.3.9 and langgraph-checkpoint>=2.0.11 embeddings = ensure_embeddings("openai:text-embedding-3-small") result = embeddings.embed_query("hello") ``` Nzembed must be providedr)PackageNotFoundErrorversion langchainzFound langchain version z, butzlangchain is not installed;z'Could not load embeddings from string 'z'. aF loading embeddings by provider:identifier string requires langchain>=0.3.9 as well as the provider-specific package. Install LangChain with: pip install 'langchain>=0.3.9' and the provider-specific package (e.g., 'langchain-openai>=0.3.0'). Alternatively, specify 'embed' as a compatible Embeddings object or python function.) ValueError isinstancestr_get_init_embeddingsimportlib.metadatar rrEmbeddingsLambda)r init_embeddingsr r lc_version version_infos X/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langgraph/store/base/embed.pyensure_embeddingsr"s^ }1222%&.00  " H H H H H H H H =$W[11 K*KKK ' = = =<  =g%ggLggg u%%%%$$ E " ""sAAAcHeZdZdZddZdd ZddZdfd Zdfd ZxZ S)ra&Wrapper to convert embedding functions into LangChain's Embeddings interface. This class allows arbitrary embedding functions to be used with LangChain-compatible tools. It supports both synchronous and asynchronous operations, and can handle: 1. A synchronous function for sync operations (async operations will use sync function) 2. An async function for both sync/async operations (sync operations will raise an error) The embedding functions should convert text into fixed-dimensional vectors that capture the semantic meaning of the text. Args: func: Function that converts text to embeddings. Can be sync or async. If async, it will be used for async operations, but sync operations will raise an error. If sync, it will be used for both sync and async operations. ??? example "Examples" With a sync function: ```python def my_embed_fn(texts): # Return 2D embeddings for each text return [[0.1, 0.2] for _ in texts] embeddings = EmbeddingsLambda(my_embed_fn) result = embeddings.embed_query("hello") # Returns [0.1, 0.2] await embeddings.aembed_query("hello") # Also returns [0.1, 0.2] ``` With an async function: ```python async def my_async_fn(texts): return [[0.1, 0.2] for _ in texts] embeddings = EmbeddingsLambda(my_async_fn) await embeddings.aembed_query("hello") # Returns [0.1, 0.2] # Note: embed_query() would raise an error ``` func EmbeddingsFunc | AEmbeddingsFuncr Nonecf|tdt|r ||_dS||_dS)Nzfunc must be provided)r_is_async_callableafuncr)selfrs r__init__zEmbeddingsLambda.__init__s> <455 5 d # # DJJJDIIItexts list[str]list[list[float]]c\t|dd}|td||S)aFEmbed a list of texts into vectors. Args: texts: list of texts to convert to embeddings. Returns: list of embeddings, one per input text. Each embedding is a list of floats. Raises: ValueError: If the instance was initialized with only an async function. rNzEmbeddingsLambda was initialized with an async function but no sync function. Use aembed_documents for async operation or provide a sync function.)getattrr)r"r%rs rembed_documentsz EmbeddingsLambda.embed_documentssBtVT** <W tE{{r$textr list[float]c:||gdS)a1Embed a single piece of text. Args: text: Text to convert to an embedding. Returns: Embedding vector as a list of floats. Note: This is equivalent to calling embed_documents with a single text and taking the first result. r)r*)r"r+s r embed_queryzEmbeddingsLambda.embed_querys##TF++A..r$cKt|dd}|'t|d{VS||d{VS)aYAsynchronously embed a list of texts into vectors. Args: texts: list of texts to convert to embeddings. Returns: list of embeddings, one per input text. Each embedding is a list of floats. Note: If no async function was provided, this falls back to the sync implementation. r!N)r)superaembed_documents)r"r%r! __class__s rr1z!EmbeddingsLambda.aembed_documentssigt,, =11%88888888 8U5\\!!!!!!!r$cKt|dd}|'t|d{VS||gd{VdS)aAAsynchronously embed a single piece of text. Args: text: Text to convert to an embedding. Returns: Embedding vector as a list of floats. Note: This is equivalent to calling aembed_documents with a single text and taking the first result. r!Nr)r)r0 aembed_query)r"r+r!r2s rr4zEmbeddingsLambda.aembed_queryspgt,, =--d33333333 3eTFmm######Q''r$)rrr r)r%r&r r')r+rr r,) __name__ __module__ __qualname____doc__r#r*r.r1r4 __classcell__)r2s@rrrms''R    ( / / / /"""""""((((((((((r$robjrpathstr | list[str]r&c|r|dkrtj|ddgSt|trt |n|}dfd ||d S)aExtract text from an object using a path expression or pre-tokenized path. Args: obj: The object to extract text from path: Either a path string or pre-tokenized path list. !!! info "Path types handled" - Simple paths: "field1.field2" - Array indexing: "[0]", "[*]", "[-1]" - Wildcards: "*" - Multi-field selection: "{field1,field2}" - Nested paths in multi-field: "{field1,nested.field2}" $TF sort_keys ensure_asciir:rtokensr&posintr c X|t|krrt|tttt frt|gS|gSt|t tfrtj |ddgSgS||}g}| dr| drt|t sgS|dd}|dkr*|D]%}| |||dz&n t|}|d krt||z}d |cxkrt|kr,nn)| ||||dzn.#ttf$rgcYSwxYw| d r)| d rt|tsgSd |ddd D}|D]} t!| } | r|} | D]&} t| tr | | vr | | } $d} | t| tttt fr#|t| t| t tfr*|tj | ddn|dkrt|tr;|D]%} | | ||dz&nt|t r(|D]%}| |||dz&nBt|tr-||vr)| ||||dz|S)NTFr?[]*r{}c6g|]}|S)strip).0fs r z?get_text_at_path.._extract_from_obj..s @@@Aaggii@@@r$,)lenrrrDfloatboollistdictjsondumps startswithendswithextendr IndexErrorsplit tokenize_pathappendvalues)r:rBrCtokenresultsindexitemidxfieldsfield nested_tokens current_obj nested_tokenvalue_extract_from_objs rrnz+get_text_at_path.._extract_from_objs* #f++  #S%677 MCz! C$.. M 3$UKKKLLIs    C : OU^^C%8%8: Oc4((  !B$KE||MMDNN#4#4T637#K#KLLLLMe**CQww!#hhnC****#c((*****'8'8S63QR7'S'STTT"J/III  c " "( Ou~~c':':( Oc4((  @@qt):):3)?)?@@@F   -e 4 4  /2K(5"" &{D99" , ; ;*5l*CKK*.K!".%kCeT3JKK#NN3{+;+;<<<<' dD\BB#NN $ $/4e!"!"!"# 0c\\#t$$ M ZZ\\NNENN#4#4UFC!G#L#LMMMMNC&& MMMDNN#4#4T637#K#KLLLL#t$$ O#00UVS1WMMNNNs A-E<<FFr)r:rrBr&rCrDr r&)rYrZrrr`)r:r;rBrns @rget_text_at_pathros E43;; 3$UCCCDD$.tS$9$9 C]4 tFIIIIIIV  S&! , ,,r$rc|sgSg}g}d}|t|kr||}|dkr|r*|d|g}d}dg}|dz }|t|krb|dkr\||dkr|dz }n||dkr|dz}||||dz }|t|kr|dk\|d||dkr|r*|d|g}d}dg}|dz }|t|krb|dkr\||dkr|dz }n||dkr|dz}||||dz }|t|kr|dk\|d||dkr-|r*|d|g}n|||dz }|t|k|r(|d||S) zTokenize a path into components. !!! info "Types handled" - Simple paths: "field1.field2" - Array indexing: "[0]", "[*]", "[-1]" - Wildcards: "*" - Multi-field selection: "{field1,field2}" rrFrHrGrKrL.)rTrajoin) r;rBcurrentichar bracket_count index_chars brace_count field_charss rr`r`Ms  FG A c$ii--Aw 3;;  bggg..///M%K FAc$ii--MA$5$57c>>!Q&MM!W^^!Q&M""47+++Q c$ii--MA$5$5 MM"''+.. / / /  S[[  bggg..///K%K FAc$ii--K!OO7c>>1$KK!W^^1$K""47+++Q c$ii--K!OO MM"''+.. / / /  S[[  bggg../// NN4  QW c$ii--Z( bggg&&''' Mr$rrVc|tj|p(t|dotj|jS)zCheck if a function is async. This includes both async def functions and classes with async __call__ methods. Args: func: Function or callable object to check. Returns: True if the function is async, False otherwise. __call__)asyncioiscoroutinefunctionhasattrr|)rs rr r s> #D)) 7 4 $ $ 7  ' 6 6r$"Callable[[str], Embeddings] | Nonec6 ddlm}|S#t$rYdSwxYw)Nrr)langchain.embeddingsr ImportErrorrs rrrsA888888 tts  )rEmbeddingsFuncAEmbeddingsFunc)r r r r)r:rr;r<r r&)r;rr r&)rrr rV)r r)r8 __future__rr} functoolsrYcollections.abcrrrtypingrlangchain_core.embeddingsrrrWrUrrrrror`r lru_cacher__all__rNr$rrs#""""" 99999999990000008C=/4U +<<=HSM?Id4;6G,HHI H#H#H#H#Vy(y(y(y(y(zy(y(y(x^-^-^-^-H????D(    r$