File size: 629 Bytes
cfd07be
 
 
8c16f30
 
 
cfd07be
 
 
 
8c16f30
cfd07be
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
from typing import TypedDict, List, Optional, Any, Dict, Annotated
import operator

from langchain_core.messages import BaseMessage
from langgraph.graph.message import add_messages

class AgentState(TypedDict):
    """
    Common state for agents.
    """
    messages: Annotated[List[BaseMessage], add_messages]
    # Add other common fields if needed

class ExpirySweepState(TypedDict):
    """
    State for the expiry sweep graph.
    """
    providers: List[Dict[str, Any]]
    alerts_created: int
    errors: List[str]
    summary: str
    window_days: int

def merge_dicts(a: Dict, b: Dict) -> Dict:
    return {**a, **b}