diff --git a/src/memos/graph_dbs/polardb.py b/src/memos/graph_dbs/polardb.py index f24f1072c..1d7dc06fc 100644 --- a/src/memos/graph_dbs/polardb.py +++ b/src/memos/graph_dbs/polardb.py @@ -2389,6 +2389,8 @@ def add_node( self, id: str, memory: str, metadata: dict[str, Any], user_name: str | None = None ) -> None: """Add a memory node to the graph.""" + logger.info(f"In add node polardb: id-{id} memory-{memory}") + # user_name comes from metadata; fallback to config if missing metadata["user_name"] = user_name if user_name else self.config.user_name @@ -2481,6 +2483,7 @@ def add_node( cursor.execute(insert_query, (id, json.dumps(properties))) logger.info(f"Added node {id} to graph '{self.db_name}_graph'.") finally: + logger.info(f"In add node polardb: id-{id} memory-{memory} query-{insert_query}") self._return_connection(conn) def _build_node_from_agtype(self, node_agtype, embedding=None): diff --git a/src/memos/mem_scheduler/schemas/message_schemas.py b/src/memos/mem_scheduler/schemas/message_schemas.py index 9cdb6823d..7f328474f 100644 --- a/src/memos/mem_scheduler/schemas/message_schemas.py +++ b/src/memos/mem_scheduler/schemas/message_schemas.py @@ -46,6 +46,10 @@ class ScheduleMessageItem(BaseModel, DictConversionMixin): default=None, description="user name / display name (optional)", ) + session_id: str | None = Field( + default=None, + description="session_id (optional)", + ) # Pydantic V2 model configuration model_config = ConfigDict(