kaiyuan/chatgpt-on-wechat
1
0
Fork 0
mirror of https://github.com/zhayujie/chatgpt-on-wechat.git synced 2026-07-21 06:07:13 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: memory loss issue caused by scheduler

Browse Source
This commit is contained in:
saboteur7
2026-02-03 20:45:22 +08:00
parent 3d63459c25
commit a77a8741b5
1 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
agent/tools/scheduler/integration.py
View File

@@ -112,11 +112,15 @@ def _execute_agent_task(task: dict, agent_bridge):
logger.info(f"[Scheduler] Task {task['id']}: Executing agent task '{task_description}'") logger.info(f"[Scheduler] Task {task['id']}: Executing agent task '{task_description}'")
# Create a unique session_id for this scheduled task to avoid polluting user's conversation
# Format: scheduler_<receiver>_<task_id> to ensure isolation
scheduler_session_id = f"scheduler_{receiver}_{task['id']}"
# Create context for Agent # Create context for Agent
context = Context(ContextType.TEXT, task_description) context = Context(ContextType.TEXT, task_description)
context["receiver"] = receiver context["receiver"] = receiver
context["isgroup"] = is_group context["isgroup"] = is_group
context["session_id"] = receiver context["session_id"] = scheduler_session_id
# Channel-specific setup # Channel-specific setup
if channel_type == "web": if channel_type == "web":
@@ -140,7 +144,8 @@ def _execute_agent_task(task: dict, agent_bridge):
context["is_scheduled_task"] = True context["is_scheduled_task"] = True
try: try:
reply = agent_bridge.agent_reply(task_description, context=context, on_event=None, clear_history=True) # Don't clear history - scheduler tasks use isolated session_id so they won't pollute user conversations
reply = agent_bridge.agent_reply(task_description, context=context, on_event=None, clear_history=False)
if reply and reply.content: if reply and reply.content:
# Send the reply via channel # Send the reply via channel
@@ -378,6 +383,10 @@ def _execute_skill_call(task: dict, agent_bridge):
logger.info(f"[Scheduler] Task {task['id']}: Executing skill '{skill_name}' with params {skill_params}") logger.info(f"[Scheduler] Task {task['id']}: Executing skill '{skill_name}' with params {skill_params}")
# Create a unique session_id for this scheduled task to avoid polluting user's conversation
# Format: scheduler_<receiver>_<task_id> to ensure isolation
scheduler_session_id = f"scheduler_{receiver}_{task['id']}"
# Build a natural language query for the Agent to execute the skill # Build a natural language query for the Agent to execute the skill
# Format: "Use skill-name to do something with params" # Format: "Use skill-name to do something with params"
param_str = ", ".join([f"{k}={v}" for k, v in skill_params.items()]) param_str = ", ".join([f"{k}={v}" for k, v in skill_params.items()])
@@ -389,7 +398,7 @@ def _execute_skill_call(task: dict, agent_bridge):
context = Context(ContextType.TEXT, query) context = Context(ContextType.TEXT, query)
context["receiver"] = receiver context["receiver"] = receiver
context["isgroup"] = is_group context["isgroup"] = is_group
context["session_id"] = receiver context["session_id"] = scheduler_session_id
# Channel-specific setup # Channel-specific setup
if channel_type == "web": if channel_type == "web":
@@ -402,7 +411,8 @@ def _execute_skill_call(task: dict, agent_bridge):
# Use Agent to execute the skill # Use Agent to execute the skill
try: try:
reply = agent_bridge.agent_reply(query, context=context, on_event=None, clear_history=True) # Don't clear history - scheduler tasks use isolated session_id so they won't pollute user conversations
reply = agent_bridge.agent_reply(query, context=context, on_event=None, clear_history=False)
if reply and reply.content: if reply and reply.content:
content = reply.content content = reply.content