chore: Use thread.get API to fetch most recent thread messages in memory component (#416)

Author: paul-paliychuk

integrations/python/zep_autogen/examples/autogen_basic.py

@@ -63,7 +63,7 @@ async def add_message(message: str, role: str, name: str | None = None):
         # First interaction
         user_msg1 = "My name is Alice and I love hiking in the mountains."
         print(f"User: {user_msg1}")
-        await add_message(user_msg1, "user")
+        await add_message(user_msg1, "user", "Alice")
         response1 = await agent.run(task=user_msg1)
         agent_msg1 = response1.messages[-1].content
         print(f"Agent: {agent_msg1}")

integrations/python/zep_autogen/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "zep-autogen"
-version = "0.1.1"
+version = "1.0.0"
 description = "Autogen integration for Zep"
 readme = "README.md"
 requires-python = ">=3.10"
@@ -12,7 +12,7 @@ dependencies = [
     "python-dotenv>=1.0.0",
     "python-slugify>=8.0.4",
     "rich>=14.0.0",
-    "zep-cloud>=3.0.0rc1",
+    "zep-cloud>=3.2.0",
 ]
 
 [project.urls]

integrations/python/zep_autogen/src/zep_autogen/memory.py

@@ -240,6 +240,7 @@ async def update_context(self, model_context: ChatCompletionContext) -> UpdateCo
             if not self._thread_id:
                 return UpdateContextResult(memories=MemoryQueryResult(results=[]))
             memory_result = await self._client.thread.get_user_context(thread_id=self._thread_id)
+            thread = await self._client.thread.get(thread_id=self._thread_id, lastn=10)
 
             memory_contents = []
             memory_parts = []
@@ -255,20 +256,18 @@ async def update_context(self, model_context: ChatCompletionContext) -> UpdateCo
                 )
                 memory_parts.append(f"Memory context: {memory_result.context}")
 
-                # Only include recent messages if we have memory
-                if memory_result.messages:
-                    recent_messages = memory_result.messages[-10:]  # Get last 10 messages
-                    if recent_messages:
-                        message_history = []
-                        for msg in recent_messages:
-                            message_history.append(f"{msg.role}: {msg.content}")
-                        memory_parts.append("Recent conversation:\n" + "\n".join(message_history))
+            # Only include recent messages if we have memory
+            if thread.messages:
+                message_history = []
+                for msg in thread.messages:
+                    name_prefix = f"{msg.name} " if msg.name else ""
+                    message_history.append(f"{name_prefix}{msg.role}: {msg.content}")
+                memory_parts.append("Recent conversation:\n" + "\n".join(message_history))
 
             # If we have memory parts, add them to the context as a system message
             if memory_parts:
                 memory_context = "\n\n".join(memory_parts)
                 await model_context.add_message(SystemMessage(content=memory_context))
-
             return UpdateContextResult(memories=MemoryQueryResult(results=memory_contents))
 
         except Exception as e: