26 lines
720 B
Python
26 lines
720 B
Python
|
"""
|
||
|
|
LangGraph 状态定义模块
|
||
|
|
包含 MessagesState 和 GraphContext
|
||
|
|
"""
|
||
|
|
|
||
|
|
import operator
|
||
|
|
from typing import Annotated
|
||
|
|
from typing_extensions import TypedDict
|
||
|
|
from dataclasses import dataclass
|
||
|
|
from langchain_core.messages import AnyMessage
|
||
|
|
|
||
|
|
class MessagesState(TypedDict):
|
||
|
|
"""对话状态类型定义"""
|
||
|
|
messages: Annotated[list[AnyMessage], operator.add]
|
||
|
|
llm_calls: int
|
||
|
|
memory_context: str
|
||
|
|
last_token_usage: dict # 本次调用的 token 使用详情
|
||
|
|
last_elapsed_time: float # 本次调用耗时（秒）
|
||
|
|
turns_since_last_summary: int # 距离上次生成摘要的轮数
|
||
|
|
|
||
|
|
@dataclass
|
||
|
|
class GraphContext:
|
||
|
|
"""图执行上下文"""
|
||
|
|
user_id: str
|
||
|
|
# 可扩展更多上下文信息
|