Noeud de préparation des documents

Pas encore totalement fonctionnel, mais déjà de bons résultats là dessus
This commit is contained in:
2026-02-08 14:41:22 +01:00
parent 52bf2d5a82
commit fc7f692ba3
8 changed files with 43 additions and 9 deletions

View File

@@ -2,7 +2,7 @@ from langgraph.graph import START, END
from langgraph.graph.state import CompiledStateGraph
from langgraph.checkpoint.memory import InMemorySaver
from utils.nodes import call_to_LLM, should_continue, task_ended, BasicToolNode, tool_node
from utils.nodes import call_to_LLM, should_continue, task_ended, BasicToolNode, tool_node, preparation_docs, weekly_report_tools
from utils.state import getState
from utils.tools import getTools
@@ -17,10 +17,19 @@ def getGraph()->CompiledStateGraph:
# Définition des sommets du graphe
workflow.add_node(call_to_LLM)
workflow.add_node(preparation_docs)
workflow.add_node("tool_node", tool_node)# BasicToolNode(tools=getTools())) # N'est pas une fonction, mais une classe instanciée, je dois précisier le nom du node
workflow.add_node("weekly_report_tools", weekly_report_tools)
# Arrêtes
workflow.set_entry_point("call_to_LLM")
workflow.set_entry_point("preparation_docs")
workflow.add_conditional_edges("preparation_docs", should_continue, {
"tools":"weekly_report_tools",
"no_tools":"call_to_LLM"
})
#workflow.set_entry_point("call_to_LLM")
workflow.add_edge("weekly_report_tools", "preparation_docs")
workflow.add_edge("tool_node", "call_to_LLM")
workflow.add_conditional_edges("call_to_LLM", should_continue, {
"tools":"tool_node",