qi>)"dZddlmZmZmZmZddlmZddlm Z m Z ddl m Z m Z mZddlmZddlmZmZddlmZmZ dd lmZmZGddeZy #e$r7Zej8d eej8d ed ed Z[wwxYw)zMem0 memory service integration for Pipecat. This module provides a memory service that integrates with Mem0 to store and retrieve conversational memories, enhancing LLM context with relevant historical information. )AnyDictListOptional)logger) BaseModelField)FrameLLMContextFrameLLMMessagesFrame) LLMContext)OpenAILLMContextOpenAILLMContextFrame)FrameDirectionFrameProcessor)Memory MemoryClientz Exception: zhIn order to use Mem0, you need to `pip install mem0ai`. Also, set the environment variable MEM0_API_KEY.zMissing module: NceZdZdZGddeZdddddddddeedeeee fdeed eed eed eed eeffd Z de eee ffdZ dede eee ffdZ deezdefdZdedeffd ZxZS)Mem0MemoryServiceaA standalone memory service that integrates with Mem0. This service intercepts message frames in the pipeline, stores them in Mem0, and enhances context with relevant memories before passing them downstream. Supports both local and cloud-based Mem0 configurations. ceZdZUdZeddZeed<eddd Ze ed <ed Z e ed <ed Z e ed<ed Z eed<ed Zeed<y)Mem0MemoryService.InputParamsa Configuration parameters for Mem0 memory service. Parameters: search_limit: Maximum number of memories to retrieve per query. search_threshold: Minimum similarity threshold for memory retrieval. api_version: API version to use for Mem0 client operations. system_prompt: Prefix text for memory context messages. add_as_system_message: Whether to add memories as system messages. position: Position to insert memory messages in context. )defaultge search_limitg?gg?)rrlesearch_thresholdv2)r api_versionz-Based on previous conversations, I recall: system_promptTadd_as_system_messagepositionN)__name__ __module__ __qualname____doc__r rint__annotations__rfloatr strr!r"boolr#N/opt/pipecat/venv/lib/python3.12/site-packages/pipecat/services/mem0/memory.py InputParamsr-sl ""3 c3"'"D%D . S."+\] s]&+D&9t9a(#(r.r0N)api_key local_configuser_idagent_idrun_idparamshostr1r2r3r4r5r6r7c.t||xsi}|xstj}|rt j ||_nt|||_t|||gs td||_ ||_ ||_ |j|_ |j|_|j|_|j |_|j"|_|j$|_d|_t)j*d|d|d|y)aInitialize the Mem0 memory service. Args: api_key: The API key for accessing Mem0's cloud API. local_config: Local configuration for Mem0 client (alternative to cloud API). user_id: The user ID to associate with memories in Mem0. agent_id: The agent ID to associate with memories in Mem0. run_id: The run ID to associate with memories in Mem0. params: Configuration parameters for memory retrieval and storage. host: The host of the Mem0 server. Raises: ValueError: If none of user_id, agent_id, or run_id are provided. )r1r7z=At least one of user_id, agent_id, or run_id must be providedNz+Initialized Mem0MemoryService with user_id=z , agent_id=z , run_id=)super__init__rr0r from_config memory_clientrany ValueErrorr3r4r5rrr r!r"r# last_queryrinfo) selfr1r2r3r4r5r6r7 __class__s r/r:zMem0MemoryService.__init__@s4 #)r :,88: !'!3!3L!AD !-gD!ID GXv./\] ]    "// & 7 7!--#11%+%A%A"  B'K41aQ]!Q$KK3$,,33=f= -/t{{+ =EZ[T5HYD%=ZZ-44/",,33# ,,++"33"( 4 LL:c'l^3FG HN)L[  LL@D EI sJA5E 7 EEAE  E E&A-E  E F )FF F contextc|j|k(ry||_|j|}|sy|j}t|ddD] \}}||d|j dddz }"|j r|j d|d n|j d |d tjd t|d y) zEnhance the LLM context with relevant memories. Args: context: The LLM context to enhance with memory information. query: The query to search for relevant memories. Nrarz. memoryz system)rolecontentuserzEnhanced context with z memories) r?rfr! enumerategetr" add_messagerrJrK)rArgrTmemories memory_textiris r/_enhance_context_with_memoriesz0Mem0MemoryService._enhance_context_with_memoriess ??e # **51 (( "8I#6: BIAv aS6::h#;"7&#t $rL}|j#dt||d{7|j|d{7Yd}~yd}~wwxYw7dw)zProcess incoming frames, intercept context frames for memory integration. Args: frame: The incoming frame to process. direction: The direction of frame flow in the pipeline. NrlrnrmzError processing with Mem0: ) error_msg exception)r9 process_framerMr rrgr rCr get_messagesreversedrpr+rurS push_framerO push_error) rArvrwrgrCcontext_messageslatest_user_messagemessagerRrBs r/r{zMem0MemoryService.process_framesg#E9555 eo/DE FmmG / 0~~H *G  -#*#7#7#9 &*#'(89G{{6*f4GKKPYDZ\_9`.5kk).D+ '77ATU(()9:'//*:7;O;O;Q*RSSS//%000T//%3 3 3S 6<T1 -oo 0404r.r)r'typingrrrrlogururpydanticrr pipecat.frames.framesr r r *pipecat.processors.aggregators.llm_contextr 1pipecat.processors.aggregators.openai_llm_contextrr"pipecat.processors.frame_processorrrmem0rrModuleNotFoundErrorrRrPrOrr-r.r/rs-,%JJAN,)`4`4,FLL;qc"#FLLr &qc* ++ ,sAB2B  B