24 lines
687 B
Python
24 lines
687 B
Python
# Copyright (c) 2025 Bytedance Ltd. and/or its affiliates
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
from langgraph.graph import MessagesState
|
|
|
|
from src.prompts.planner_model import Plan
|
|
from src.rag import Resource
|
|
|
|
|
|
class State(MessagesState):
|
|
"""State for the agent system, extends MessagesState with next field."""
|
|
|
|
# Runtime Variables
|
|
locale: str = "en-US"
|
|
research_topic: str = ""
|
|
observations: list[str] = []
|
|
resources: list[Resource] = []
|
|
plan_iterations: int = 0
|
|
current_plan: Plan | str = None
|
|
final_report: str = ""
|
|
auto_accepted_plan: bool = False
|
|
enable_background_investigation: bool = True
|
|
background_investigation_results: str = None
|