v2.4.30

cookbook/assistants/clear_memory.py

@@ -0,0 +1,17 @@
+from phi.assistant import Assistant
+from phi.llm.openai import OpenAIChat
+from phi.utils.log import logger
+
+assistant = Assistant(llm=OpenAIChat(model="gpt-4o"))
+# -*- Print a response to the cli
+assistant.print_response("Share a 1 line joke")
+
+# -*- Print the assistant memory
+logger.info("*** Assistant Memory ***")
+logger.info(assistant.memory.to_dict())
+
+# -*- Clear the assistant memory
+logger.info("Clearing the assistant memory...")
+assistant.memory.clear()
+logger.info("*** Assistant Memory ***")
+logger.info(assistant.memory.to_dict())

phi/memory/assistant.py

@@ -41,7 +41,7 @@ class AssistantMemory(BaseModel):
 
  def to_dict(self) -> Dict[str, Any]:
  _memory_dict = self.model_dump(
- exclude_none=True, exclude={"db", "updating", "memories", "classifier", "manager"}
+ exclude_none=True, exclude={"db", "updating", "memories", "classifier", "manager", "retrieval"}
  )
  if self.memories:
  _memory_dict["memories"] = [memory.to_dict() for memory in self.memories]
@@ -220,3 +220,11 @@ def get_memories_for_system_prompt(self) -> Optional[str]:
  memory_str += "\n</memory_from_previous_interactions>"
 
  return memory_str
+
+ def clear(self) -> None:
+ """Clears the assistant memory"""
+ self.chat_history = []
+ self.llm_messages = []
+ self.references = []
+ self.memories = None
+ logger.debug("Assistant Memory cleared")

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "phidata"
-version = "2.4.29"
+version = "2.4.30"
 description = "Memory, knowledge and tools for LLMs."
 requires-python = ">=3.7"
 readme = "README.md"