| 12345678910111213141516171819202122232425262728 |
- # -*- coding: utf-8 -*-
- """LangGraph state definitions for document chat."""
- from typing import Any, Dict, List, Optional, TypedDict
- from langchain_core.messages import BaseMessage
- class DocumentChatState(TypedDict, total=False):
- callback_task_id: str
- user_id: str
- conversation_id: Optional[str]
- task_id: Optional[str]
- project_info: Dict[str, Any]
- selected_section: Dict[str, Any]
- document_context: Dict[str, Any]
- conversation_history: List[Dict[str, Any]]
- user_message: str
- skill_registry: List[Dict[str, Any]]
- intent_result: Optional[Dict[str, Any]]
- skill_result: Optional[Dict[str, Any]]
- diff_result: Optional[Dict[str, Any]]
- response_type: Optional[str]
- current_stage: str
- overall_task_status: str
- error_message: Optional[str]
- warnings: List[str]
- messages: List[BaseMessage]
|
