Nouveau node inject_preparation_prompt
Ne change rien, mais simplifie le système
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, preparation_docs, weekly_report_tools
|
||||
from utils.nodes import *
|
||||
from utils.state import getState
|
||||
from utils.tools import getTools
|
||||
|
||||
@@ -18,11 +18,13 @@ def getGraph()->CompiledStateGraph:
|
||||
# Définition des sommets du graphe
|
||||
workflow.add_node(call_to_LLM)
|
||||
workflow.add_node(preparation_docs)
|
||||
workflow.add_node(inject_preparation_prompt)
|
||||
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("preparation_docs")
|
||||
workflow.set_entry_point("inject_preparation_prompt")
|
||||
workflow.add_edge("inject_preparation_prompt", "preparation_docs")
|
||||
workflow.add_conditional_edges("preparation_docs", should_continue, {
|
||||
"tools":"weekly_report_tools",
|
||||
"no_tools":"call_to_LLM"
|
||||
|
||||
Reference in New Issue
Block a user