Noeud de préparation des documents
Pas encore totalement fonctionnel, mais déjà de bons résultats là dessus
This commit is contained in:
@@ -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",
|
||||
|
||||
Reference in New Issue
Block a user