Skip to content

Commit ab257f6

Browse files
authored
Merge branch 'dev' into dev_zdy_1030
2 parents 6ae7dc5 + 28cf578 commit ab257f6

File tree

2 files changed

+7
-0
lines changed

2 files changed

+7
-0
lines changed

src/memos/graph_dbs/polardb.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2392,6 +2392,8 @@ def add_node(
23922392
self, id: str, memory: str, metadata: dict[str, Any], user_name: str | None = None
23932393
) -> None:
23942394
"""Add a memory node to the graph."""
2395+
logger.info(f"In add node polardb: id-{id} memory-{memory}")
2396+
23952397
# user_name comes from metadata; fallback to config if missing
23962398
metadata["user_name"] = user_name if user_name else self.config.user_name
23972399

@@ -2484,6 +2486,7 @@ def add_node(
24842486
cursor.execute(insert_query, (id, json.dumps(properties)))
24852487
logger.info(f"Added node {id} to graph '{self.db_name}_graph'.")
24862488
finally:
2489+
logger.info(f"In add node polardb: id-{id} memory-{memory} query-{insert_query}")
24872490
self._return_connection(conn)
24882491

24892492
def _build_node_from_agtype(self, node_agtype, embedding=None):

src/memos/mem_scheduler/schemas/message_schemas.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,10 @@ class ScheduleMessageItem(BaseModel, DictConversionMixin):
4646
default=None,
4747
description="user name / display name (optional)",
4848
)
49+
session_id: str | None = Field(
50+
default=None,
51+
description="session_id (optional)",
52+
)
4953

5054
# Pydantic V2 model configuration
5155
model_config = ConfigDict(

0 commit comments

Comments
 (0)