3fi"ddlmZddlZddlmZddlmZmZmZm Z m Z m Z ddl m Z ddlmZmZmZmZer ddlmZmZmZmZej0eZGdd eeZGd d eeZGd d e Zy)) annotationsN)Enum) TYPE_CHECKINGAnyDictListOptionalSequence)BaseChatMessageHistory) AIMessage BaseMessage HumanMessage SystemMessage)MemoryMemorySearchResultMessage NotFoundErrorceZdZdZdZ dZy) SearchScopez5Scope for the document search. Messages or Summaries?messagessummaryN)__name__ __module__ __qualname____doc__rrl/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_community/chat_message_histories/zep.pyrrs?H'G(rrceZdZdZdZ dZy) SearchTypez-Enumerator of the types of search to perform. similaritymmrN)rrrrr!r"rrrr r s7J CDrr c"eZdZdZ d ddZeddZeddZeddZddZ d ddZ d dd Z d dd Z dd Z dd Zdej ej"ddf dd ZddZddZy)ZepChatMessageHistorya,Chat message history that uses Zep as a backend. Recommended usage:: # Set up Zep Chat History zep_chat_history = ZepChatMessageHistory( session_id=session_id, url=ZEP_API_URL, api_key=, ) # Use a standard ConversationBufferMemory to encapsulate the Zep chat history memory = ConversationBufferMemory( memory_key="chat_history", chat_memory=zep_chat_history ) Zep provides long-term conversation storage for LLM apps. The server stores, summarizes, embeds, indexes, and enriches conversational AI chat histories, and exposes them via simple, low-latency APIs. For server installation instructions and more, see: https://docs.getzep.com/deployment/quickstart/ This class is a thin wrapper around the zep-python package. Additional Zep functionality is exposed via the `zep_summary` and `zep_messages` properties. For more information on the zep-python package, see: https://github.com/getzep/zep-python Ncn ddlm}||||_||_y#t$r tdwxYw)Nr) ZepClientzUCould not import zep-python package. Please install it with `pip install zep-python`.)base_urlapi_key) zep_pythonr& ImportError zep_client session_id)selfr,urlr(r&s r__init__zZepChatMessageHistory.__init__HsE   ,$S'B$ C  s4c<|j}|sgSg}|jrQt|jjdkDr/|j t |jj|j r|j D]}|j|j|j|jd}|jdk(r'|j t|j|h|j t|j||S)z!Retrieve messages from Zep memoryrcontent)uuid created_at token_countmetadataai)r2additional_kwargs) _get_memoryrlenr2appendrrr3r4r5r6roler r)r- zep_memoryrmsgr6s rrzZepChatMessageHistory.messagesYs(,'7'7'9 I&(   :%%--.2 j6H6H6P6P QR   !** HH"%..#&?? # " 88t#OO!#++ROO$S[[HU  rcB|j}|sgS|jS) Retrieve summary from Zep memory)r9rr-r=s r zep_messagesz"ZepChatMessageHistory.zep_messagesys&(,'7'7'9 I"""rcl|j}|r |jsy|jjS)r@N)r9rr2rAs r zep_summaryz!ZepChatMessageHistory.zep_summarys2(,'7'7'9 !3!3!!)))rcddlm} |jjj |j }|S#|$r&t jd|j dYywxYw)zRetrieve memory from Zepr)rSession z! not found in Zep. Returning NoneN)r)rr+memory get_memoryr,loggerwarning)r-rr=s rr9z!ZepChatMessageHistory._get_memorysd, !%!7!7!B!B4??!SJ    NN4??++LM   s/9(A$#A$c>|jt||y)zConvenience method for adding a human message string to the store. Args: message: The string contents of a human message. metadata: Optional metadata to attach to the message. r1r6N) add_messagerr-messager6s radd_user_messagez&ZepChatMessageHistory.add_user_messages g6Jrc>|jt||y)zConvenience method for adding an AI message string to the store. Args: message: The string contents of an AI message. metadata: Optional metadata to attach to the message. r1rLN)rMr rNs radd_ai_messagez$ZepChatMessageHistory.add_ai_messages 73hGrcddlm}m}||j|j|}||g}|j j j|j|y)z,Append the message to the Zep memory historyrrrr2r<r6rN) r)rrr2typer+rG add_memoryr,)r-rOr6rr zep_messager=s rrMz!ZepChatMessageHistory.add_messagesL /OO',, k]3  ))$//:Frc ddlm}m}|Dcgc];}||j|j|j j dd=}}||}|jjj|j|ycc}w)z-Append the messages to the Zep memory historyrrTr6NrUrV) r)rrr2rWr8getr+rGrXr,r-rrrrOrBr=s r add_messagesz"ZepChatMessageHistory.add_messagess~.$   \\ 2266z4H   \2  ))$//:F sAB c 8Kddlm}m}|Dcgc];}||j|j|j j dd=}}||}|jjj|j|d{ycc}w7 w)zB B Bcddlm}||||||}|jjj |j ||S)z1Search Zep memory for messages matching the queryr)MemorySearchPayload)textr6 search_scope search_type mmr_lambda)limit)r)rbr+rG search_memoryr,) r-queryr6rdrerfrgrbpayloads rsearchzZepChatMessageHistory.searchsO 3%%#!  %%33 OOWE4  rc |jjj|jy#t$r&t j d|jdYywxYw)zClear session memory from Zep. Note that Zep is long-term storage for memory and this is not advised unless you have specific data retention requirements. rF# not found in Zep. Skipping delete.N)r+rG delete_memoryr,rrIrJr-s rclearzZepChatMessageHistory.clearsP  OO " " 0 0 A  NN4??++NO  s/2,A! A!cK |jjj|jd{y7#t$r&t j d|jdYywxYww)zClear session memory from Zep asynchronously. Note that Zep is long-term storage for memory and this is not advised unless you have specific data retention requirements. NrFrm)r+rGadelete_memoryr,rrIrJros raclearzZepChatMessageHistory.aclearsY  //((77H H H  NN4??++NO  s1A02><>A0>,A-*A0,A--A0)zhttp://localhost:8000N)r,strr.rtr( Optional[str]returnNone)rvzList[BaseMessage])rvz List[Message])rvru)rvzOptional[Memory])N)rOrtr6Optional[Dict[str, Any]]rvrw)rOr r6rxrvrw)rzSequence[BaseMessage]rvrw)rirtr6zOptional[Dict]rdrrer rfzOptional[float]rgz Optional[int]rvzList[MemorySearchResult])rvrw)rrrrr/propertyrrBrDr9rPrRrMr]r`rr r!rkrprsrrrr$r$'spF+!% %%% %  %">##** BF K K&> K  KBF H H&> H  HJN G" G.F G  GG N&$($/$8$8","7"7&*#  ! "    $    " 0  rr$) __future__rloggingenumrtypingrrrrr r langchain_core.chat_historyr langchain_core.messagesr r rrr)rrrr getLoggerrrIrtrr r$rrrrsk"EE>MM   8 $)#t)EdEa2ar