fix: access query value in output for online evals (#389)

* fix: access query value in output for online evals * chore: lint
langchain-ai · Oct 21, 2024 · 83a1470 · 83a1470
1 parent c0de6a3
commit 83a1470
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/backend/retrieval_graph/graph.py b/backend/retrieval_graph/graph.py
@@ -145,7 +145,11 @@ class Plan(TypedDict):
         {"role": "system", "content": configuration.research_plan_system_prompt}
     ] + state.messages
     response = cast(Plan, await model.ainvoke(messages))
-    return {"steps": response["steps"], "documents": "delete"}
+    return {
+        "steps": response["steps"],
+        "documents": "delete",
+        "query": state.messages[-1].content,
+    }
 
 
 async def conduct_research(state: AgentState) -> dict[str, Any]:

diff --git a/backend/retrieval_graph/state.py b/backend/retrieval_graph/state.py
@@ -81,3 +81,4 @@ class AgentState(InputState):
     """Populated by the retriever. This is a list of documents that the agent can reference."""
     answer: str = field(default="")
     """Final answer. Useful for evaluations"""
+    query: str = field(default="")