File size: 849 Bytes
e23fefd
52b993d
5e988c3
 
52b993d
afe6838
e23fefd
3b5f033
 
 
233d8ee
 
3b5f033
e670011
3b5f033
e887897
fac69f9
e887897
fac69f9
 
24e3e87
 
 
 
 
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
# type: ignore
from typing_extensions import TypedDict, Annotated
from langchain_core.messages import AnyMessage
from langgraph.graph.message import add_messages
from typing import Optional


class AppState(TypedDict):
    messages: Annotated[list[AnyMessage], add_messages]
    question: str
    pk_hash: str
    sitecode: str
    rag_result: str
    rag_sources: Optional[str]  # Added to store retrieved sources
    answer: str
    last_answer: Optional[str] = None
    last_user_message: Optional[str] = None
    last_tool: Optional[str] = None
    idsr_disclaimer_shown: bool = False
    summary: Optional[str] = None
    context: Optional[str] = None
    context_versions: dict[str, int] = {}
    last_context_injected_versions: dict[str, int] = {}
    context_version_ready_for_injection: int = 0
    context_first_response_sent: bool = True