feat: Enhance review process with streaming events and detailed logging

This commit is contained in:
Primakov Alexandr Alexandrovich
2025-10-13 17:26:41 +03:00
parent a762d09b3b
commit 2f29ccff74
10 changed files with 309 additions and 205 deletions

View File

@@ -206,6 +206,14 @@ class ReviewTaskWorker:
# Create event handler
async def on_review_event(event: dict):
"""Handle review events and broadcast to clients"""
print(f"\n{'*'*80}")
print(f"CALLBACK INVOKED!")
print(f" Event type: {event.get('type')}")
print(f" Event step: {event.get('step')}")
print(f" Event message: {event.get('message')}")
print(f" Active WS connections: {len(manager.active_connections)}")
print(f"{'*'*80}")
try:
# Prepare event data
event_data = {
@@ -216,6 +224,7 @@ class ReviewTaskWorker:
"data": event
}
print(f" Prepared event_data: {event_data}")
logger.info(f" 🔔 Broadcasting event: type={event.get('type')}, connections={len(manager.active_connections)}")
# Save event to database
@@ -229,10 +238,13 @@ class ReviewTaskWorker:
)
db.add(db_event)
await db.commit()
print(f" ✓ Event saved to DB: {db_event.id}")
logger.debug(f" 💾 Event saved to DB: {db_event.id}")
# Broadcast to all connected clients
print(f" Broadcasting to {len(manager.active_connections)} connections...")
await manager.broadcast(event_data)
print(f" ✓ Broadcast completed")
# Log the event
if event.get("type") == "agent_step":
@@ -242,6 +254,7 @@ class ReviewTaskWorker:
message = event.get("message", "")[:100]
logger.info(f" 💬 LLM: {message}...")
except Exception as e:
print(f" ❌ ERROR in callback: {e}")
logger.error(f" ❌ Ошибка broadcast события: {e}")
import traceback
traceback.print_exc()