diff --git a/pyproject.toml b/pyproject.toml index d6eff74a3..f09efe38d 100644 --- a/pyproject.toml +++ b/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 diff --git a/src/memos/api/routers/product_router.py b/src/memos/api/routers/product_router.py index c5a5b6e73..bca8f2416 100644 --- a/src/memos/api/routers/product_router.py +++ b/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( diff --git a/src/memos/mem_os/product.py b/src/memos/mem_os/product.py index 7ec538fcf..45eb07d68 100644 --- a/src/memos/mem_os/product.py +++ b/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,