3fiddlmZddlZddlmZddlmZmZmZddl m Z ddl m Z m Z mZddlmZerddlZd dZGd d e Zy) ) annotationsN) timedelta) TYPE_CHECKINGAnyOptional)BaseChatMessageHistory) BaseMessagemessage_to_dictmessages_from_dict) get_from_envcddlm}|j|}t||jst||j ryt||j r |jtd|)zCreate cache if it doesn't exist. Raises: SdkException: Momento service or network error Exception: Unexpected response r) CreateCacheNz$Unexpected response cache creation: ) momento.responsesr create_cache isinstanceSuccessCacheAlreadyExistsErrorinner_exception Exception) cache_client cache_namercreate_cache_responses p/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_community/chat_message_histories/momento.py_ensure_cache_existsrss.(55jA')<)<={==B );+<+< =#333>?T>UVWWceZdZdZdddd d dZedddd d dZedd Zdd Z dd Z y)MomentoChatMessageHistoryzZChat message history cache that uses Momento as a backend. See https://gomomento.com/zmessage_store:NT) key_prefixttlensure_cache_existsc$ ddlm}ddlm}t ||s t d|r t||||z|_||_ ||_ ||j||_ y|j|_ y#t$r t dwxYw)aInstantiate a chat message history cache that uses Momento as a backend. Note: to instantiate the cache client passed to MomentoChatMessageHistory, you must have a Momento account at https://gomomento.com/. Args: session_id (str): The session ID to use for this chat session. cache_client (CacheClient): The Momento cache client. cache_name (str): The name of the cache to use to store the messages. key_prefix (str, optional): The prefix to apply to the cache key. Defaults to "message_store:". ttl (Optional[timedelta], optional): The TTL to use for the messages. Defaults to None, ie the default TTL of the cache will be used. ensure_cache_exists (bool, optional): Create the cache if it doesn't exist. Defaults to True. Raises: ImportError: Momento python package is not installed. TypeError: cache_client is not of type momento.CacheClientObject r) CacheClient) CollectionTtlVCould not import momento python package. Please install it with `pip install momento`.z2cache_client must be a momento.CacheClient object.N)momentor#momento.requestsr$ ImportErrorr TypeErrorrkeyrrofr from_cache_ttl) self session_idrrrr r!r#r$s r__init__z"MomentoChatMessageHistory.__init__,s<  + 6 , 4PQ Q  z : *($ ?$'',DH$335DH @  s A::B) configurationapi_key auth_tokenc 2 ddlm}m} m} || j j } |xs tdd}| j|} ||| |} ||| |fd |i|S#t$r t dwxYw#t$r|xs tdd}Y\wxYw) z,Construct cache from CacheClient parameters.r)r#ConfigurationsCredentialProviderr%r2MOMENTO_AUTH_TOKENr1MOMENTO_API_KEY) default_ttlr ) r&r#r4r5r(Laptopv1r ValueError from_string) clsr.rr r0r1r2kwargsr#r4r5 credentialsrs rfrom_client_paramsz,MomentoChatMessageHistory.from_client_params^s  O O  *11446M L TL?S$TG)44W= "=+3O :|ZKSKFKK @   LKi9J!KG Ls A"A:"A7:BBcddlm}|jj|j|j }t ||jr7|jDcgc]}tj|}}t|St ||jrgSt ||jr |jtd|cc}w)zRetrieve the messages from Momento. Raises: SdkException: Momento service or network error Exception: Unexpected response Returns: list[BaseMessage]: List of cached messages r)CacheListFetchUnexpected response: )rrBr list_fetchrr*rHitvalue_list_stringjsonloadsr Missrrr)r-rBfetch_responsemitemss rmessagesz"MomentoChatMessageHistory.messages~s 5**55dootxxP nn&8&8 9,:,L,LMqTZZ]MEM%e, , (;(; <I (<(< = 00 03N3CDE ENsC cRddlm}tjt |}|j j |j|j||j}t||jryt||jr |jtd|)zStore a message in the cache. Args: message (BaseMessage): The message object to store. Raises: SdkException: Momento service or network error. Exception: Unexpected response. r)CacheListPushBack)r NrC)rrOrGdumpsr rlist_push_backrr*r rrrrr)r-messagerOitem push_responses r add_messagez%MomentoChatMessageHistory.add_messages 8zz/'23))88 OOTXXt9  m%6%>%> ?  '8'>'> ?// /3M?CD Drcddlm}|jj|j|j }t ||jryt ||jr |jtd|)zRemove the session's messages from the cache. Raises: SdkException: Momento service or network error. Exception: Unexpected response. r) CacheDeleteNrC) rrWrdeleterr*rrrrr)r-rWdelete_responses rclearzMomentoChatMessageHistory.clearsj 2++224??DHHM o{':': ; ):): ;!11 13O3DEF Fr) r.strrmomento.CacheClientrr[rr[r zOptional[timedelta]r!bool)r.r[rr[r rr0z&Optional[momento.config.Configuration]r1 Optional[str]r2r^r>rreturnr)r_zlist[BaseMessage])rRr r_None)r_r`) __name__ __module__ __qualname____doc__r/ classmethodr@propertyrMrUrZrrrr's"+#'$(0606*06 06  06!06"06dAE!%$(LLL L > LL"LL #LL>FF0E.Grr)rr\rr[r_r`) __future__rrGdatetimertypingrrrlangchain_core.chat_historyrlangchain_core.messagesr r r langchain_core.utilsr r&rrrgrrrnsC" //> .X(VG 6VGr