remove query from input

Author: albinjal

deep_security/src/open_deep_research/deep_researcher.py

@@ -146,14 +146,29 @@ async def write_research_brief(state: AgentState, config: RunnableConfig) -> Com
         .with_config(research_model_config)
     )
 
-    # Step 2: Generate structured research brief from user messages
+    # Step 2: Extract original query from messages for caching
+    messages = state.get("messages", [])
+    original_query = None
+    if not state.get("original_query"):
+        # Find the first user/human message content
+        for msg in messages:
+            if hasattr(msg, 'type') and msg.type in ['human', 'user']:
+                original_query = str(msg.content).strip()
+                break
+            elif isinstance(msg, HumanMessage):
+                original_query = str(msg.content).strip()
+                break
+    else:
+        original_query = state.get("original_query")
+
+    # Step 3: Generate structured research brief from user messages
     prompt_content = transform_messages_into_research_topic_prompt.format(
-        messages=get_buffer_string(state.get("messages", [])),
+        messages=get_buffer_string(messages),
         date=get_today_str()
     )
     response = await research_model.ainvoke([HumanMessage(content=prompt_content)])
 
-    # Step 3: Initialize supervisor with research brief and instructions
+    # Step 4: Initialize supervisor with research brief and instructions
     supervisor_system_prompt = lead_researcher_prompt.format(
         date=get_today_str(),
         max_concurrent_research_units=configurable.max_concurrent_research_units,
@@ -164,6 +179,7 @@ async def write_research_brief(state: AgentState, config: RunnableConfig) -> Com
         goto="research_supervisor",
         update={
             "research_brief": response.research_brief,
+            "original_query": original_query,  # Add original query for caching
             "supervisor_messages": {
                 "type": "override",
                 "value": [

deep_security/src/open_deep_research/state.py

@@ -255,8 +255,6 @@ def override_reducer(current_value, new_value):
 class AgentInputState(MessagesState):
     """InputState is only 'messages'."""
 
-    original_query: Optional[str] = None
-
 class AgentState(MessagesState):
     """Main agent state containing messages and research data."""