Skip to content

Commit fe49a7e

Browse files
committed
refresh supervisor state
1 parent cd0af7a commit fe49a7e

File tree

2 files changed

+12
-9
lines changed

2 files changed

+12
-9
lines changed

src/open_deep_research/deep_researcher.py

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -81,13 +81,16 @@ async def write_research_brief(state: AgentState, config: RunnableConfig)-> Comm
8181
goto="research_supervisor",
8282
update={
8383
"research_brief": response.research_brief,
84-
"supervisor_messages": [
85-
SystemMessage(content=lead_researcher_prompt.format(
86-
date=get_today_str(),
87-
max_concurrent_research_units=configurable.max_concurrent_research_units
88-
)),
89-
HumanMessage(content=response.research_brief)
90-
]
84+
"supervisor_messages": {
85+
"type": "override",
86+
"value": [
87+
SystemMessage(content=lead_researcher_prompt.format(
88+
date=get_today_str(),
89+
max_concurrent_research_units=configurable.max_concurrent_research_units
90+
)),
91+
HumanMessage(content=response.research_brief)
92+
]
93+
}
9194
}
9295
)
9396

src/open_deep_research/state.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,14 +52,14 @@ class AgentInputState(MessagesState):
5252
"""InputState is only 'messages'"""
5353

5454
class AgentState(MessagesState):
55-
supervisor_messages: Annotated[list[MessageLikeRepresentation], operator.add]
55+
supervisor_messages: Annotated[list[MessageLikeRepresentation], override_reducer]
5656
research_brief: Optional[str]
5757
raw_notes: Annotated[list[str], override_reducer] = []
5858
notes: Annotated[list[str], override_reducer] = []
5959
final_report: str
6060

6161
class SupervisorState(TypedDict):
62-
supervisor_messages: Annotated[list[MessageLikeRepresentation], operator.add]
62+
supervisor_messages: Annotated[list[MessageLikeRepresentation], override_reducer]
6363
research_brief: str
6464
notes: Annotated[list[str], override_reducer] = []
6565
research_iterations: int = 0

0 commit comments

Comments
 (0)