From 23862cf1ac47f810557f20410a9773fc4555872f Mon Sep 17 00:00:00 2001 From: vbarda Date: Thu, 17 Oct 2024 17:52:25 -0400 Subject: [PATCH] update --- src/retrieval_graph/graph.py | 4 ++-- src/shared/state.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/retrieval_graph/graph.py b/src/retrieval_graph/graph.py index bec7555..48a2605 100644 --- a/src/retrieval_graph/graph.py +++ b/src/retrieval_graph/graph.py @@ -119,7 +119,7 @@ async def respond_to_general_query( async def create_research_plan( state: AgentState, *, config: RunnableConfig -) -> dict[str, list[str]]: +) -> dict[str, list[str] | str]: """Create a step-by-step research plan for answering a LangChain-related query. Args: @@ -141,7 +141,7 @@ class Plan(TypedDict): {"role": "system", "content": configuration.research_plan_system_prompt} ] + state.messages response = cast(Plan, await model.ainvoke(messages)) - return {"steps": response["steps"]} + return {"steps": response["steps"], "documents": "delete"} async def conduct_research(state: AgentState) -> dict[str, Any]: diff --git a/src/shared/state.py b/src/shared/state.py index d41e31c..2ef6beb 100644 --- a/src/shared/state.py +++ b/src/shared/state.py @@ -61,7 +61,7 @@ def reduce_docs( if item_id not in existing_ids: new_list.append( - Document(**item, metadata={**metadata, "uuid": item_id}) + Document(**{**item, "metadata": {**metadata, "uuid": item_id}}) ) existing_ids.add(item_id)