Skip to content

Commit 28cf578

Browse files
authored
fix: message schema bug (#426)
* feat: change MOS_SCHEDULER_THREAD_POOL_MAX_WORKERS to 10000 * feat: add user_name to schedule server router * feat: roll back to old mem-reader-prompt * feat: add moniter in schedule * feat: set default MEMRADER_MAX_TOKENS to 8000 * feat: add metric in schedule status * fix: bug * fix: base scheduler bug * fix: message schema bug
1 parent 8be2e80 commit 28cf578

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
@@ -2389,6 +2389,8 @@ def add_node(
23892389
self, id: str, memory: str, metadata: dict[str, Any], user_name: str | None = None
23902390
) -> None:
23912391
"""Add a memory node to the graph."""
2392+
logger.info(f"In add node polardb: id-{id} memory-{memory}")
2393+
23922394
# user_name comes from metadata; fallback to config if missing
23932395
metadata["user_name"] = user_name if user_name else self.config.user_name
23942396

@@ -2481,6 +2483,7 @@ def add_node(
24812483
cursor.execute(insert_query, (id, json.dumps(properties)))
24822484
logger.info(f"Added node {id} to graph '{self.db_name}_graph'.")
24832485
finally:
2486+
logger.info(f"In add node polardb: id-{id} memory-{memory} query-{insert_query}")
24842487
self._return_connection(conn)
24852488

24862489
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)