state_models.py 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637
  1. # -*- coding: utf-8 -*-
  2. """文档对话 LangGraph 状态定义。"""
  3. from typing import Any, Dict, List, Optional, TypedDict
  4. from langchain_core.messages import BaseMessage
  5. class DocumentChatState(TypedDict, total=False):
  6. callback_task_id: str
  7. user_id: str
  8. conversation_id: Optional[str]
  9. task_id: Optional[str]
  10. project_info: Dict[str, Any]
  11. selected_section: Dict[str, Any]
  12. document_context: Dict[str, Any]
  13. conversation_history: List[Dict[str, Any]]
  14. user_message: str
  15. skill_registry: List[Dict[str, Any]]
  16. retrieval_query: Optional[str]
  17. retrieval_keywords: List[str]
  18. retrieval_steps: List[Dict[str, Any]]
  19. retrieval_method: Optional[str]
  20. retrieval_candidates: List[Dict[str, Any]]
  21. reranked_references: List[Dict[str, Any]]
  22. approved_references: List[Dict[str, Any]]
  23. retrieval_status: Optional[str]
  24. retrieval_metrics: Dict[str, Any]
  25. intent_result: Optional[Dict[str, Any]]
  26. skill_result: Optional[Dict[str, Any]]
  27. diff_result: Optional[Dict[str, Any]]
  28. response_type: Optional[str]
  29. current_stage: str
  30. overall_task_status: str
  31. error_message: Optional[str]
  32. warnings: List[str]
  33. messages: List[BaseMessage]