Agent 통신 패턴

from pydantic import BaseModel
from typing import Optional, Any
from datetime import datetime

class AgentMessage(BaseModel):
    # 메시지 식별
    message_id: str
    timestamp: datetime

    # 발신/수신 정보
    sender: str          # 발신 Agent ID
    receiver: str        # 수신 Agent ID (또는 "broadcast")

    # 메시지 유형
    msg_type: str        # "request", "response", "event", "error"

    # 내용
    content: str         # 자연어 메시지
    data: Optional[Any]  # 구조화된 데이터

    # 메타데이터
    correlation_id: Optional[str]  # 관련 메시지 추적용
    priority: int = 0               # 우선순위

from langgraph.graph import StateGraph
from typing import TypedDict, Annotated
import operator

# 공유 상태 정의
class TeamState(TypedDict):
    # 메시지 히스토리 (append-only)
    messages: Annotated[list, operator.add]

    # 현재 작업 상태
    current_task: str
    task_status: str

    # 결과물
    research_result: str
    draft: str
    final_output: str

    # 다음 실행할 Agent
    next_agent: str

# 각 Agent는 state를 읽고 업데이트된 부분만 반환
def researcher(state: TeamState) -> TeamState:
    result = do_research(state["current_task"])
    return {
        "research_result": result,
        "messages": [{"agent": "researcher", "content": result}]
    }