Minimal Viable Product
C'est moche, bancal, et mal foutu, mais ça compile et ça crache un rapport de stage dans un fichier
This commit is contained in:
@@ -1,8 +1,9 @@
|
||||
from langgraph.graph import START, END
|
||||
from langgraph.graph.state import CompiledStateGraph
|
||||
|
||||
from utils.nodes import reponse_question, tool_node, should_continue
|
||||
from utils.nodes import call_to_LLM, should_continue, task_ended, BasicToolNode, tool_node
|
||||
from utils.state import getState
|
||||
from utils.tools import getTools
|
||||
|
||||
def getGraph()->CompiledStateGraph:
|
||||
"""
|
||||
@@ -14,15 +15,15 @@ def getGraph()->CompiledStateGraph:
|
||||
workflow = getState() # State prêt à utiliser
|
||||
|
||||
# Définition des sommets du graphe
|
||||
workflow.add_node(reponse_question)
|
||||
workflow.add_node("tool_node", tool_node) # N'est pas une fonction, mais une classe instanciée, je dois précisier le nom du node
|
||||
workflow.add_node(call_to_LLM)
|
||||
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
|
||||
|
||||
# Arrêtes
|
||||
workflow.set_entry_point("reponse_question")
|
||||
workflow.add_edge("tool_node", "reponse_question")
|
||||
workflow.add_conditional_edges("reponse_question", should_continue, {
|
||||
workflow.set_entry_point("call_to_LLM")
|
||||
workflow.add_edge("tool_node", "call_to_LLM")
|
||||
workflow.add_conditional_edges("call_to_LLM", should_continue, {
|
||||
"tools":"tool_node",
|
||||
END:END
|
||||
"no_tools":END
|
||||
})
|
||||
|
||||
return workflow.compile()
|
||||
|
||||
Reference in New Issue
Block a user