4ab6400a87
- Add streaming schemas and events - Implement run_review_stream in ReviewerAgent - Update task_worker to broadcast streaming events via WebSocket - Create ReviewStream component for real-time progress visualization - Integrate ReviewStream into ReviewDetail page - Show agent steps, LLM messages, and progress in real-time
56 lines
1.3 KiB
Python
56 lines
1.3 KiB
Python
"""Streaming events schemas"""
|
|
|
|
from typing import Optional, Any, Dict, Literal
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class StreamEventType:
|
|
"""Stream event types"""
|
|
AGENT_START = "agent_start"
|
|
AGENT_UPDATE = "agent_update"
|
|
AGENT_STEP = "agent_step"
|
|
LLM_START = "llm_start"
|
|
LLM_STREAM = "llm_stream"
|
|
LLM_END = "llm_end"
|
|
AGENT_ERROR = "agent_error"
|
|
AGENT_COMPLETE = "agent_complete"
|
|
|
|
|
|
class StreamEvent(BaseModel):
|
|
"""Base streaming event"""
|
|
type: str
|
|
review_id: int
|
|
timestamp: str
|
|
data: Dict[str, Any]
|
|
|
|
|
|
class AgentStepEvent(BaseModel):
|
|
"""Agent step event"""
|
|
type: Literal["agent_step"] = "agent_step"
|
|
review_id: int
|
|
step: str # fetch_pr_info, fetch_files, analyze_files, post_comments
|
|
status: str # started, completed, failed
|
|
message: str
|
|
data: Optional[Dict[str, Any]] = None
|
|
|
|
|
|
class LLMStreamEvent(BaseModel):
|
|
"""LLM streaming event"""
|
|
type: Literal["llm_stream"] = "llm_stream"
|
|
review_id: int
|
|
file_path: Optional[str] = None
|
|
chunk: str # Часть ответа от LLM
|
|
is_complete: bool = False
|
|
|
|
|
|
class ReviewProgressEvent(BaseModel):
|
|
"""Review progress event"""
|
|
type: Literal["review_progress"] = "review_progress"
|
|
review_id: int
|
|
total_files: int
|
|
analyzed_files: int
|
|
total_comments: int
|
|
current_step: str
|
|
message: str
|
|
|