45 lines
1.6 KiB
Python
45 lines
1.6 KiB
Python
from langgraph.graph import START, END
|
|
from langgraph.graph.state import CompiledStateGraph
|
|
from langgraph.checkpoint.memory import InMemorySaver
|
|
|
|
from utils.nodes import *
|
|
from utils.state import getState
|
|
from utils.tools import getTools
|
|
|
|
def getGraph()->CompiledStateGraph:
|
|
"""
|
|
Récupérer le graphe compilé et prêt à invoquer
|
|
|
|
Returns:
|
|
CompiledStateGraph: Graphe compilé
|
|
"""
|
|
workflow = getState() # State prêt à utiliser
|
|
|
|
# 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("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"
|
|
})
|
|
|
|
#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",
|
|
"no_tools":END
|
|
})
|
|
|
|
return workflow.compile(checkpointer=InMemorySaver()) # TODO: Rempalcer par une vrai BDD de prod
|
|
|
|
if __name__ == "__main__":
|
|
# Affichage du graphe
|
|
getGraph().get_graph().draw_mermaid_png(output_file_path="imgs/agent.png") |