Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions src/memos/graph_dbs/polardb.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down Expand Up @@ -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):
Expand Down
4 changes: 4 additions & 0 deletions src/memos/mem_scheduler/schemas/message_schemas.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down