feat: Enhance review process with streaming events and detailed logging
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user