NguddlmZmZmZmZddlmZddlmZm Z ddl m Z ddl m Z mZddlmZeddd Gd d e Zeddd Gd deZdS))AnyDictListOptional) deprecated) BaseMessageget_buffer_string)pre_init)BaseChatMemory BaseMemory)get_prompt_input_keyz0.3.1z1.0.0z_Please see the migration guide at: https://python.langchain.com/docs/versions/migrating_memory/)sinceremovalmessagecreZdZUdZdZeed<dZeed<dZeed<e de fd Z de fd Z d e edefd Ze defd ZdefdZe de efdZde efdZe de efdZdeee fdeee ffdZdeee fdeee ffdZdS)ConversationBufferMemoryaSA basic memory implementation that simply stores the conversation history. This stores the entire conversation history in memory without any additional processing. Note that additional processing may be required in some situations when the conversation history is too large to fit in the context window of the model. Human human_prefixAI ai_prefixhistory memory_keyreturnc,|jr|jn|jS)String buffer of memory.)return_messagesbuffer_as_messages buffer_as_strselfs S/var/www/html/ai-engine/env/lib/python3.11/site-packages/langchain/memory/buffer.pybufferzConversationBufferMemory.buffer!s+/*>Vt&&DDVVc|K|jr|d{Vn|d{VS)rN)rabuffer_as_messagesabuffer_as_strrs r!abufferz ConversationBufferMemory.abuffer&s\# -$**,, , , , , , , ,**,,,,,,,, r#messagesc:t||j|jS)N)rr)r rrr r(s r!_buffer_as_strz'ConversationBufferMemory._buffer_as_str.s' *n    r#c@||jjS)?Exposes the buffer as a string in case return_messages is True.)r+ chat_memoryr(rs r!rz&ConversationBufferMemory.buffer_as_str5s""4#3#<===r#cnK|jd{V}||S)r-N)r. aget_messagesr+r*s r!r&z'ConversationBufferMemory.abuffer_as_str:s@)7799999999""8,,,r#c|jjS)JExposes the buffer as a list of messages in case return_messages is False.)r.r(rs r!rz+ConversationBufferMemory.buffer_as_messages?s((r#cDK|jd{VS)r2N)r.r0rs r!r%z,ConversationBufferMemory.abuffer_as_messagesDs-%33555555555r#c|jgS)zMWill always return list of memory variables. :meta private: rrs r!memory_variablesz)ConversationBufferMemory.memory_variablesHs   r#inputsc|j|jiSzReturn history buffer.rr"r r7s r!load_memory_variablesz.ConversationBufferMemory.load_memory_variablesP--r#cLK|d{V}|j|iS)z9Return key-value pairs given the text input to the chain.N)r'r)r r7r"s r!aload_memory_variablesz/ConversationBufferMemory.aload_memory_variablesTs2||~~%%%%%%((r#N)__name__ __module__ __qualname____doc__rstr__annotations__rrpropertyrr"r'rrr+rr&rr%r6rr<r?r#r!rr s L#IsJ WWWWXW s     tK'8 S    >s>>>X>-c---- )D$5)))X)64 +<6666!$s)!!!X!.DcN.tCH~....)4S>)d3PS8n))))))r#rceZdZUdZdZeed<dZeed< dZeed<dZ e eed <dZ e eed <d Z eed <e d edefdZedeefdZdeeefdeeeffdZdeeefdeeeffdZdeeefdeeefddfdZdeeefdeeefddfdZddZddZdS)ConversationStringBufferMemoryaA basic memory implementation that simply stores the conversation history. This stores the entire conversation history in memory without any additional processing. Equivalent to ConversationBufferMemory but tailored more specifically for string-based conversations rather than chat models. Note that additional processing may be required in some situations when the conversation history is too large to fit in the context window of the model. rrrrr"N output_key input_keyrrvaluesrcP|ddrtd|S)z*Validate that return messages is not True.rFz@return_messages must be False for ConversationStringBufferMemory)get ValueError)clsrMs r!validate_chainsz.ConversationStringBufferMemory.validate_chainsws7 ::' / / R  r#c|jgS)zLWill always return list of memory variables. :meta private: r5rs r!r6z/ConversationStringBufferMemory.memory_variabless   r#r7c|j|jiSr9r:r;s r!r<z4ConversationStringBufferMemory.load_memory_variablesr=r#c0K||Sr9)r<r;s r!r?z5ConversationStringBufferMemory.aload_memory_variabless))&111r#outputsc|jt||j}n|j}|j_t |dkr$t d|t|d}n|j}|jd||z}|j d||z}|xj dd ||gzz c_ dS).Save context from this conversation to buffer.NzOne output key expected, got rz:  ) rLr r6rKlenrPkeyslistrrr"join)r r7rVprompt_input_keyrKhumanais r! save_contextz+ConversationStringBufferMemory.save_contexts > !3FD2d3PS8n222254S>5DcN5t5555 238n2/3CH~2 2222 r#rIN)typingrrrrlangchain_core._apirlangchain_core.messagesrr langchain_core.utilsr langchain.memory.chat_memoryr r langchain.memory.utilsr rrIrGr#r!rosX,,,,,,,,,,,,******BBBBBBBB))))))CCCCCCCC777777    G D)D)D)D)D)~D)D)D)N    G HHHHHZHHHHHr#