fix: auto-assign tasks to PM agent when moved to in_progress, fix datetime serialization in WebSocket, add executor logging
This commit is contained in:
@@ -217,6 +217,17 @@ async def move_task(
|
|||||||
actor=user.username
|
actor=user.username
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# Auto-assign to default agent (PM) if not already assigned
|
||||||
|
if not current_task.get("assigned_to"):
|
||||||
|
updates["assigned_to"] = "pm"
|
||||||
|
updates["assigned_type"] = "agent"
|
||||||
|
# Create agent execution record
|
||||||
|
await opc_db.create_agent_execution(
|
||||||
|
task_id=task_id,
|
||||||
|
agent_id="pm",
|
||||||
|
actor=user.username
|
||||||
|
)
|
||||||
|
|
||||||
if old_status == "in_progress" and move.status != "in_progress":
|
if old_status == "in_progress" and move.status != "in_progress":
|
||||||
await opc_db.stop_time_entry(task_id=task_id)
|
await opc_db.stop_time_entry(task_id=task_id)
|
||||||
|
|
||||||
|
|||||||
@@ -74,24 +74,44 @@ async def websocket_endpoint(websocket: WebSocket):
|
|||||||
|
|
||||||
async def broadcast_task_update(task_id: int, action: str, task_data: dict):
|
async def broadcast_task_update(task_id: int, action: str, task_data: dict):
|
||||||
"""Broadcast task update to all connected clients"""
|
"""Broadcast task update to all connected clients"""
|
||||||
|
# Serialize datetime objects to ISO format
|
||||||
|
serialized_data = task_data.copy()
|
||||||
|
for key in ["created_at", "updated_at", "completed_at", "due_date"]:
|
||||||
|
if serialized_data.get(key):
|
||||||
|
serialized_data[key] = serialized_data[key].isoformat() if hasattr(serialized_data[key], 'isoformat') else serialized_data[key]
|
||||||
|
|
||||||
|
timestamp = task_data.get("updated_at")
|
||||||
|
if timestamp and hasattr(timestamp, 'isoformat'):
|
||||||
|
timestamp = timestamp.isoformat()
|
||||||
|
|
||||||
message = {
|
message = {
|
||||||
"type": "task_update",
|
"type": "task_update",
|
||||||
"task_id": task_id,
|
"task_id": task_id,
|
||||||
"action": action, # created, updated, moved, deleted
|
"action": action, # created, updated, moved, deleted
|
||||||
"data": task_data,
|
"data": serialized_data,
|
||||||
"timestamp": task_data.get("updated_at")
|
"timestamp": timestamp
|
||||||
}
|
}
|
||||||
await manager.broadcast(message)
|
await manager.broadcast(message)
|
||||||
|
|
||||||
|
|
||||||
async def broadcast_agent_execution(execution_id: int, status: str, execution_data: dict):
|
async def broadcast_agent_execution(execution_id: int, status: str, execution_data: dict):
|
||||||
"""Broadcast agent execution update"""
|
"""Broadcast agent execution update"""
|
||||||
|
# Serialize datetime objects to ISO format
|
||||||
|
serialized_data = execution_data.copy()
|
||||||
|
for key in ["started_at", "completed_at"]:
|
||||||
|
if serialized_data.get(key):
|
||||||
|
serialized_data[key] = serialized_data[key].isoformat() if hasattr(serialized_data[key], 'isoformat') else serialized_data[key]
|
||||||
|
|
||||||
|
timestamp = execution_data.get("started_at") or execution_data.get("completed_at")
|
||||||
|
if timestamp and hasattr(timestamp, 'isoformat'):
|
||||||
|
timestamp = timestamp.isoformat()
|
||||||
|
|
||||||
message = {
|
message = {
|
||||||
"type": "agent_execution",
|
"type": "agent_execution",
|
||||||
"execution_id": execution_id,
|
"execution_id": execution_id,
|
||||||
"status": status, # pending, running, completed, failed, pending_approval
|
"status": status, # pending, running, completed, failed, pending_approval
|
||||||
"data": execution_data,
|
"data": serialized_data,
|
||||||
"timestamp": execution_data.get("started_at") or execution_data.get("completed_at")
|
"timestamp": timestamp
|
||||||
}
|
}
|
||||||
await manager.broadcast(message)
|
await manager.broadcast(message)
|
||||||
|
|
||||||
|
|||||||
@@ -43,13 +43,18 @@ class AgentExecutor:
|
|||||||
async def start(self):
|
async def start(self):
|
||||||
"""Start the executor service"""
|
"""Start the executor service"""
|
||||||
self.running = True
|
self.running = True
|
||||||
|
print("=== Agent Executor Loop Started ===")
|
||||||
logger.info("Agent executor service started")
|
logger.info("Agent executor service started")
|
||||||
|
|
||||||
while self.running:
|
while self.running:
|
||||||
try:
|
try:
|
||||||
|
print(f"[Executor] Checking for pending executions... (running={self.running})")
|
||||||
await self._process_pending_executions()
|
await self._process_pending_executions()
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
print(f"[Executor] Error in executor loop: {e}")
|
||||||
logger.error(f"Error in executor loop: {e}")
|
logger.error(f"Error in executor loop: {e}")
|
||||||
|
import traceback
|
||||||
|
traceback.print_exc()
|
||||||
|
|
||||||
await asyncio.sleep(5) # Check every 5 seconds
|
await asyncio.sleep(5) # Check every 5 seconds
|
||||||
|
|
||||||
@@ -61,11 +66,14 @@ class AgentExecutor:
|
|||||||
async def _process_pending_executions(self):
|
async def _process_pending_executions(self):
|
||||||
"""Process pending agent executions"""
|
"""Process pending agent executions"""
|
||||||
executions = await opc_db.get_agent_executions(status="pending", limit=10)
|
executions = await opc_db.get_agent_executions(status="pending", limit=10)
|
||||||
|
print(f"[Executor] Found {len(executions)} pending executions")
|
||||||
|
|
||||||
for execution in executions:
|
for execution in executions:
|
||||||
try:
|
try:
|
||||||
|
print(f"[Executor] Processing execution {execution['id']} for agent {execution['agent_id']}")
|
||||||
await self._execute_agent(execution)
|
await self._execute_agent(execution)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
print(f"[Executor] Failed to execute agent {execution['agent_id']}: {e}")
|
||||||
logger.error(f"Failed to execute agent {execution['agent_id']}: {e}")
|
logger.error(f"Failed to execute agent {execution['agent_id']}: {e}")
|
||||||
await self._mark_execution_failed(execution["id"], str(e))
|
await self._mark_execution_failed(execution["id"], str(e))
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user