JeffJiang
462752b462
* feat: add rag provider and retriever * feat: retriever tool * feat: add retriever tool to the researcher node * feat: add rag http apis * feat: new message input supports resource mentions * feat: new message input component support resource mentions * refactor: need_web_search to need_search * chore: RAG integration docs * chore: change example api host * fix: user message color in dark mode * fix: mentions style * feat: add local_search_tool to researcher prompt * chore: research prompt * fix: ragflow page size and reporter with * docs: ragflow integration and add acknowledgment projects * chore: format
23 lines
658 B
Python
23 lines
658 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"
|
|
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
|