a dg@sddlmZddlmZmZmZmZmZGdddZddddd Z d d d d dZ d#dd dddddZ dd ddddZ d$dd dddddZ d%dd ddd d!d"ZdS)&) annotations)AnyDictListOptionalUnionc@s0eZdZd dddddZddd d ZeZd S) _EmbedFrboolvaluekeepcCs||_||_dSNr )selfr r ru/var/www/html/cobodadashboardai.evdpl.com/venv/lib/python3.9/site-packages/langchain_experimental/rl_chain/helpers.py__init__sz_Embed.__init__str)returncCs t|jSr )rr )rrrr__str__ sz_Embed.__str__N)F)__name__ __module__ __qualname__rr__repr__rrrrrsrrr) embeddingrcCsdddt|DS)z!Convert an embedding to a string. cSsg|]\}}|d|qS):r).0ierrr z'stringify_embedding..)join enumerate)rrrrstringify_embeddingsr#rr )itemrcCs t|tpt|tot|jtS)zCheck if an item is a string.) isinstancerrr )r$rrris_stringtype_instances r&NzUnion[str, _Embed]z Optional[str]z Dict[str, Union[str, List[str]]])r$model namespacercCsd}t|tr8t||j}|jrd|jddd}n,t|trP|dd}ntdt |d|durttd|||iS)z#Embed a string or an _Embed object.r_zUnsupported type z for embeddingNzPThe default namespace must be provided when embedding a string or _Embed object.) r%rr#encoder r replacer ValueErrortype)r$r'r(Zkeep_strencodedrrrembed_string_types  r0rzDict[str, Any])r$r'rcCshi}|D]V\}}t|trPg||<|D]"}t|||}||||q*q |t|||q |S)zEmbed a dictionary item.)itemsr%listr0appendupdate)r$r'Z inner_dictns embed_itemZembed_list_itemZembeddedrrrembed_dict_type6s  r7r2z&List[Dict[str, Union[str, List[str]]]]csg}|D]z}t|tr(|t||qt|trpt|||}tt|dfdd|Di}||q|t|||q|S)zEmbed a list item.rcsg|] }|qSrr)rr$Z first_keyrrrRr z#embed_list_type..) r%dictr3r7r2embed_list_typenextiterr0)r$r'r(Zret_listr6Zitem_embeddinggroupingrr8rr:Ds    r:zEUnion[Union[str, _Embed], Dict, List[Union[str, _Embed]], List[Dict]])to_embedr'r(rcCsft|trt|jts t|tr.t|||gSt|trDt||gSt|trZt|||St ddS)a Embed the actions or context using the SentenceTransformer model (or a model that has an `encode` function). Attributes: to_embed: (Union[Union(str, _Embed(str)), Dict, List[Union(str, _Embed(str))], List[Dict]], required) The text to be embedded, either a string, a list of strings or a dictionary or a list of dictionaries. namespace: (str, optional) The default namespace to use when dictionary or list of dictionaries not provided. model: (Any, required) The model to use for embedding Returns: List[Dict[str, str]]: A list of dictionaries where each dictionary has the namespace as the key and the embedded string as the value z"Invalid input format for embeddingN) r%rr rr0r9r7r2r:r-)r>r'r(rrrembedYs    r?)N)N)N) __future__rtypingrrrrrrr#r&r0r7r:r?rrrrs