mirror of
https://github.com/CopilotKit/CopilotKit.git
synced 2026-09-14 16:26:20 +08:00
Jordan Ritter
2482317ccc
320 files reformatted. One-time alignment to match the ruff format check added to CI in #4812.
20 lines
685 B
Python
20 lines
685 B
Python
from langchain_core.messages import AnyMessage
|
|
from langgraph.graph import add_messages
|
|
from typing import TypedDict, Dict, Union, List, Annotated
|
|
from copilotkit import CopilotKitState # extends MessagesState
|
|
|
|
|
|
class ResearchState(CopilotKitState):
|
|
title: str
|
|
proposal: Dict[
|
|
str, Union[str, bool, Dict[str, Union[str, bool]]]
|
|
] # Stores proposed structure before user approval
|
|
outline: dict
|
|
sections: List[dict] # list of dicts with 'title','content',and 'idx'
|
|
footnotes: str
|
|
sources: Dict[str, Dict[str, Union[str, float]]]
|
|
tool: str
|
|
logs: List[
|
|
dict
|
|
] # list of dicts logs to be sent to frontend with 'message', 'status'
|