diff --git a/examples/foundational/37-mem0.py b/examples/foundational/37-mem0.py index b4a0dbed7..dfb3e7db4 100644 --- a/examples/foundational/37-mem0.py +++ b/examples/foundational/37-mem0.py @@ -44,6 +44,7 @@ The bot runs as part of a pipeline that processes audio frames and manages the c import argparse import os from typing import Union + from dotenv import load_dotenv from loguru import logger @@ -64,7 +65,7 @@ from pipecat.transports.network.webrtc_connection import SmallWebRTCConnection load_dotenv(override=True) try: - from mem0 import MemoryClient, Memory + from mem0 import Memory, MemoryClient # noqa: F401 except ModuleNotFoundError as e: logger.error(f"Exception: {e}") logger.error( diff --git a/src/pipecat/services/mem0/memory.py b/src/pipecat/services/mem0/memory.py index d1d0d5abe..a5f076b6e 100644 --- a/src/pipecat/services/mem0/memory.py +++ b/src/pipecat/services/mem0/memory.py @@ -17,7 +17,7 @@ from pipecat.processors.aggregators.openai_llm_context import ( from pipecat.processors.frame_processor import FrameDirection, FrameProcessor try: - from mem0 import MemoryClient, Memory # noqa: F401 + from mem0 import Memory, MemoryClient # noqa: F401 except ModuleNotFoundError as e: logger.error(f"Exception: {e}") logger.error(