Skip to content

Commit

Permalink
chore: lint
Browse files Browse the repository at this point in the history
  • Loading branch information
davidx33 committed Oct 21, 2024
1 parent 40b0cad commit 767496a
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion backend/retrieval_graph/graph.py
Original file line number Diff line number Diff line change
Expand Up @@ -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", "query": state.messages[-1].content}
return {
"steps": response["steps"],
"documents": "delete",
"query": state.messages[-1].content,
}


async def conduct_research(state: AgentState) -> dict[str, Any]:
Expand Down

0 comments on commit 767496a

Please sign in to comment.