feat: demo product api change

pyproject.toml

@@ -28,7 +28,6 @@ sqlalchemy = "^2.0.41"
 redis = "^6.2.0"
 pika = "^1.3.2"
 schedule = "^1.2.2"
-pika = "^1.3.2"
 
 [tool.poetry.group.dev]
 optional = false

src/memos/api/routers/product_router.py

@@ -65,6 +65,8 @@ async def register_user(user_req: UserRegisterRequest):
         user_config, default_mem_cube = APIConfig.create_user_config(
             user_name=user_req.user_id, user_id=user_req.user_id
         )
+        logger.info(f"user_config: {user_config.model_dump(mode='json')}")
+        logger.info(f"default_mem_cube: {default_mem_cube.config.model_dump(mode='json')}")
         mos_product = get_mos_product_instance()
         # Register user with default config and mem cube
         result = mos_product.user_register(

src/memos/mem_os/product.py

@@ -7,7 +7,7 @@
 from typing import Any, Literal
 
 from transformers import AutoTokenizer
-
+from dotenv import load_dotenv
 from memos.configs.mem_os import MOSConfig
 from memos.log import get_logger
 from memos.mem_cube.general import GeneralMemCube
@@ -29,7 +29,9 @@
 
 logger = get_logger(__name__)
 
-CUBE_PATH = "/tmp/data"
+load_dotenv()
+
+CUBE_PATH = os.getenv("MOS_CUBE_PATH", "/tmp/data/")
 with open("./tmp/fake_data.json") as f:
     MOCK_DATA = json.loads(f.read())
 
@@ -820,13 +822,15 @@ def chat_with_references(
         self.add(
             user_id=user_id,
             messages=[
-                {"role": "user", "content": query},
-                {"role": "assistant", "content": full_response}
+                {"role": "user", "content": query, "chat_time": str(datetime.now().strftime("%Y-%m-%d %H:%M:%S"))},
+                {"role": "assistant", "content": full_response, "chat_time": str(datetime.now().strftime("%Y-%m-%d %H:%M:%S"))}
             ],
             mem_cube_id=cube_id
         )
+        # Keep chat history under 30 messages by removing oldest conversation pair
         if len(self.chat_history_manager[user_id].chat_history) > 30:
-            self.chat_history_manager[user_id].chat_history.pop(0)
+            self.chat_history_manager[user_id].chat_history.pop(0)  # Remove oldest user message
+            self.chat_history_manager[user_id].chat_history.pop(0)  # Remove oldest assistant response
 
     def get_all(
         self,