@@ -118,7 +118,7 @@ def initialize(self) -> bool:
118118
119119 # Create session associated with the user
120120 try :
121- self .zep_client .memory . add_session ( session_id = self .session_id , user_id = self .user_id )
121+ self .zep_client .thread . create ( thread_id = self .session_id , user_id = self .user_id )
122122 log_debug (f"Created session { self .session_id } for user { self .user_id } " )
123123 except Exception as e :
124124 log_debug (f"Session may already exist: { e } " )
@@ -148,16 +148,16 @@ def add_zep_message(self, role: str, content: str) -> str:
148148
149149 try :
150150 zep_message = ZepMessage (
151- role_type = role ,
151+ role = role ,
152152 content = content ,
153153 )
154154
155155 # Prepare ignore_roles if needed
156156 ignore_roles_list = ["assistant" ] if self .ignore_assistant_messages else None
157157
158158 # Add message to Zep memory
159- self .zep_client .memory . add ( # type: ignore
160- session_id = self .session_id ,
159+ self .zep_client .thread . add_messages ( # type: ignore
160+ thread_id = self .session_id ,
161161 messages = [zep_message ],
162162 ignore_roles = ignore_roles_list ,
163163 )
@@ -171,7 +171,7 @@ def get_zep_memory(self, memory_type: str = "context") -> str:
171171 """
172172 Retrieves the memory for the current Zep session.
173173 Args:
174- memory_type: The type of memory to retrieve ('context', 'messages', 'relevant_facts' ).
174+ memory_type: The type of memory to retrieve ('context', 'messages').
175175 Returns:
176176 The requested memory content as a string, or an error string.
177177 """
@@ -181,21 +181,16 @@ def get_zep_memory(self, memory_type: str = "context") -> str:
181181
182182 try :
183183 log_debug (f"Getting Zep memory for session { self .session_id } " )
184- memory_data = self .zep_client .memory .get (session_id = self .session_id ) # type: ignore
185184
186185 if memory_type == "context" :
187186 # Ensure context is a string
188- return memory_data .context or "No context available."
187+ user_context = self .zep_client .thread .get_user_context (thread_id = self .session_id , mode = "basic" ) # type: ignore
188+ log_debug (f"Memory data: { user_context } " )
189+ return user_context .context or "No context available."
189190 elif memory_type == "messages" :
191+ messages_list = self .zep_client .thread .get (thread_id = self .session_id ) # type: ignore
190192 # Ensure messages string representation is returned
191- return str (memory_data .messages ) if memory_data .messages else "No messages available."
192- elif memory_type == "relevant_facts" :
193- # Return all relevant facts from memory
194- if memory_data .relevant_facts :
195- facts_str = "\n " .join ([f"- { fact .fact } " for fact in memory_data .relevant_facts ])
196- return f"Relevant facts:\n { facts_str } "
197- else :
198- return "No relevant facts available."
193+ return str (messages_list .messages ) if messages_list .messages else "No messages available."
199194 else :
200195 warning_msg = f"Unsupported memory_type requested: { memory_type } . Returning empty string."
201196 log_warning (warning_msg )
@@ -328,7 +323,7 @@ async def initialize(self) -> bool:
328323
329324 # Create session associated with the user
330325 try :
331- await self .zep_client .memory . add_session ( session_id = self .session_id , user_id = self .user_id ) # type: ignore
326+ await self .zep_client .thread . create ( thread_id = self .session_id , user_id = self .user_id ) # type: ignore
332327 log_debug (f"Created session { self .session_id } for user { self .user_id } " )
333328 except Exception as e :
334329 log_debug (f"Session may already exist: { e } " )
@@ -361,16 +356,16 @@ async def add_zep_message(self, role: str, content: str) -> str:
361356
362357 try :
363358 zep_message = ZepMessage (
364- role_type = role ,
359+ role = role ,
365360 content = content ,
366361 )
367362
368363 # Prepare ignore_roles if needed
369364 ignore_roles_list = ["assistant" ] if self .ignore_assistant_messages else None
370365
371366 # Add message to Zep memory
372- await self .zep_client .memory . add ( # type: ignore
373- session_id = self .session_id ,
367+ await self .zep_client .thread . add_messages ( # type: ignore
368+ thread_id = self .session_id ,
374369 messages = [zep_message ],
375370 ignore_roles = ignore_roles_list ,
376371 )
@@ -384,7 +379,7 @@ async def get_zep_memory(self, memory_type: str = "context") -> str:
384379 """
385380 Retrieves the memory for the current Zep session.
386381 Args:
387- memory_type: The type of memory to retrieve ('context', 'messages', 'relevant_facts' ).
382+ memory_type: The type of memory to retrieve ('context', 'messages').
388383 Returns:
389384 The requested memory content as a string, or an error string.
390385 """
@@ -396,25 +391,19 @@ async def get_zep_memory(self, memory_type: str = "context") -> str:
396391 return "Error: Zep client/session not initialized."
397392
398393 try :
399- memory_data = await self .zep_client .memory .get (session_id = self .session_id ) # type: ignore
400-
401394 if memory_type == "context" :
402395 # Ensure context is a string
403- return memory_data .context or "No context available."
396+ user_context = await self .zep_client .thread .get_user_context (thread_id = self .session_id , mode = "basic" ) # type: ignore
397+ log_debug (f"Memory data: { user_context } " )
398+ return user_context .context or "No context available."
404399 elif memory_type == "messages" :
405400 # Ensure messages string representation is returned
406- return str (memory_data .messages ) if memory_data .messages else "No messages available."
407- elif memory_type == "relevant_facts" :
408- # Return relevant facts from memory
409- if memory_data .relevant_facts :
410- facts_str = "\n " .join ([f"- { fact .fact } " for fact in memory_data .relevant_facts ])
411- return f"Relevant facts:\n { facts_str } "
412- else :
413- return "No relevant facts available."
401+ messages_list = await self .zep_client .thread .get (thread_id = self .session_id ) # type: ignore
402+ return str (messages_list .messages ) if messages_list .messages else "No messages available."
414403 else :
415404 warning_msg = f"Unsupported memory_type requested: { memory_type } . Returning context."
416405 log_warning (warning_msg )
417- return memory_data . context or "No context available."
406+ return "No context available."
418407
419408 except Exception as e :
420409 error_msg = f"Failed to get Zep memory for session { self .session_id } : { e } "
0 commit comments