3fiC%vddlmZddlmZddlmZmZmZmZddlm Z ddl m Z ddl m Z mZmZGdde Zy ) )datetime)sleep)AnyCallableListUnion)uuid4)BaseChatMessageHistory) BaseMessagemessage_to_dictmessages_from_dictc6eZdZUdZdZeed<dZeed<dZeed<de d ed e d d fd Z de de d e e fdZd)dZd efdZd efdZd efdZd)dZde d d fdZd)dZddddfde de de d e d!e d"ed#e gee effd d fd$Zed e efd%Zd&ed d fd'Zd)d(Zy )*RocksetChatMessageHistoryaUses Rockset to store chat messages. To use, ensure that the `rockset` python package installed. Example: .. code-block:: python from langchain_community.chat_message_histories import ( RocksetChatMessageHistory ) from rockset import RocksetClient history = RocksetChatMessageHistory( session_id="MySession", client=RocksetClient(), collection="langchain_demo", sync=True ) history.add_user_message("hi!") history.add_ai_message("whats up?") print(history.messages) # noqa: T201 SLEEP_INTERVAL_MSiADD_TIMEOUT_MSi NCREATE_TIMEOUT_MSmethodtimeout method_paramsreturnNc tj}|di|sdtj}||z jdz|kDrt|d|dt t j dz |di|scyy)zPSleeps until meth() evaluates to true. Passes kwargs into meth. iz timed out at z msN)rnow total_seconds TimeoutErrorrrr)selfrrrstartcurrs r/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_community/chat_message_histories/rocksetdb.py _wait_untilz%RocksetChatMessageHistory._wait_until0sx )=)<<>Du ++-4w>"fX^G9C#HII +==D E )=)query query_paramsc P|jj||jS)zExecutes an SQL statement and returns the result Args: - query: The SQL string - **query_params: Parameters to pass into the query )params)clientsqlresults)rr#r$s r _queryz RocksetChatMessageHistory._query;s! {{u\:BBBr"cz|jjj|j|jy)z-Creates a collection for this message history)name workspaceN)r' Collectionscreate_s3_collection collectionr-rs r _create_collectionz,RocksetChatMessageHistory._create_collectionCs, 44DNN 5 r"c |jjj|jy#|jj j $rYywxYw)z;Checks whether a collection exists for this message historyr0FT)r'r.getr0rockset exceptionsNotFoundExceptionr1s r _collection_existsz,RocksetChatMessageHistory._collection_existsIsO  KK # # ' '4?? ' C||&&88  s03#AAc|jjj|jjj dk(S)z^Checks whether the collection for this message history is ready to be queried r4READY)r'r.r5r0datastatusr1s r _collection_is_readyz.RocksetChatMessageHistory._collection_is_readyQs< KK # # ' '4?? ' C H H O O  r"cpt|jd|jd|jdk7S)Nz? SELECT 1 FROM zd WHERE _id=:session_id LIMIT 1  session_idr)lenr*locationrAr1s r _document_existsz*RocksetChatMessageHistory._document_existsZsK  "mm_- $   r"cLjfdtjy)z\Sleeps until the collection for this message history is ready to be queried c$jSN)r>r1sr zJRocksetChatMessageHistory._wait_until_collection_created..osD--/r"N)r!rrr1s`r _wait_until_collection_createdz8RocksetChatMessageHistory._wait_until_collection_createdjs  / % 7 7 r" message_idcPjfdtj|y)z4Sleeps until a message is added to the messages listc tjdjdjdj|dk7S)Nzl SELECT * FROM UNNEST(( SELECT z" FROM z WHERE _id = :session_id )) AS message WHERE message.data.additional_kwargs.id = :message_id LIMIT 1 )rArJr)rBr* messages_keyrCrA)rJrs r rHzERocksetChatMessageHistory._wait_until_message_added..vs^s $%)$5$5#67""&--1  $)   ! r")rJN)r!rr)rrJs` r _wait_until_message_addedz3RocksetChatMessageHistory._wait_until_message_addedss)  " & 4 4!'  r"c|jjj|j|jd|j |j gigy)zPCreates or replaces a document for this message history with no messages_idr0r-r<N)r' Documents add_documentsr0r-rArMr1s r _create_empty_docz+RocksetChatMessageHistory._create_empty_docsH ++nn$//4+<+s S\r"rAr'r0r-rMsyncmessage_uuid_methodcf ddl}t||jst dt |||_||_||_||_ d|jd|jd|_ ||_||_ ||_ ||_ |jjd|j!s1|j#|j%|j'y|j)s|j'yy#t$r tdwxYw#t$rYwxYw)aConstructs a new RocksetChatMessageHistory. Args: - session_id: The ID of the chat session - client: The RocksetClient object to use to query - collection: The name of the collection to use to store chat messages. If a collection with the given name does not exist in the workspace, it is created. - workspace: The workspace containing `collection`. Defaults to `"commons"` - messages_key: The DB column containing message history. Defaults to `"messages"` - sync: Whether to wait for messages to be added. Defaults to `False`. NOTE: setting this to `True` will slow down performance. - message_uuid_method: The method that generates message IDs. If set, all messages will have an `id` field within the `additional_kwargs` property. If this param is not set and `sync` is `False`, message IDs will not be created. If this param is not set and `sync` is `True`, the `uuid.uuid4` method will be used to create message IDs. rNz]Could not import rockset client python package. Please install it with `pip install rockset`.z;client should be an instance of rockset.RocksetClient, got "z"." langchain)r6 ImportError isinstance RocksetClient ValueErrortyperAr'r0r-rCrMrZrYset_applicationAttributeErrorr9r2rIrTrD) rrAr'r0r-rMrYrZr6s r __init__z"RocksetChatMessageHistory.__init__s7@  &'"7"78F|n&  % $"DNN+3t.?qA  (#6    KK ' ' 4 &&(  # # %  / / 1  " " $&&(  " " $)A @  0   sD D$ D!$ D0/D0c t|jd|jd|jd|jS)zMessages in this chat history.za SELECT * FROM UNNEST (( SELECT "z" FROM zX WHERE _id = :session_id )) r@)r r*rMrCrAr1s r rVz"RocksetChatMessageHistory.messagessV" KK!"&!2!2 34"mm_-  ??    r"messagecD|jr+d|jvr|j|jd<|jjj |j |j|jjjj|j|jjjjdd|jdt!|gg|jr|j#|jdyy) zpAdd a Message object to the history. Args: message: A BaseMessage object to store. idADD/z/-)oppathvalue)ripatchrQN)rYadditional_kwargsrZr'rRpatch_documentsr0r-r6modelpatch_document PatchDocumentrApatch_operationPatchOperationrMr rN)rrgs r add_messagez%RocksetChatMessageHistory.add_messages 99W%>%>>.2.F.F.HG % %d + --nn ""11?? **::II$#$T%6%6$7r!:"1'":J@   . 99  * *7+D+DT+J K r"cjjr$jfdtjyy)z*Removes all messages from the chat historycj SrG)rVr1sr rHz1RocksetChatMessageHistory.clear.. sDMM)r"N)rTrYr!rrr1s`r clearzRocksetChatMessageHistory.clears5  99   ))88  r")rN)__name__ __module__ __qualname____doc__rint__annotations__rrrrr!rXrr*r2boolr9r>rDrIrNrTrrepropertyr rVrwrzrr"r rrsp:sNC"s" F( FS F3 FSW FCCCCcC D d  $   C D 0 #&=QC%C%C% C%  C%  C%C%&b%S/&9:C% C%J ${+  L;L4L6r"rN)rtimertypingrrrruuidr langchain_core.chat_historyr langchain_core.messagesr r r rrr"r rs0-->~ 6~r"