23 lines
670 B
Python
23 lines
670 B
Python
from langgraph.graph import StateGraph, MessagesState
|
|
from typing import List, Annotated
|
|
import operator
|
|
|
|
|
|
class CustomState(MessagesState):
|
|
todo: Annotated[list, operator.add] # Les tâches en cours, au format JSON
|
|
|
|
lastSummarizedMessage: int # Index du message où l'on s'était arrêté de résumer
|
|
|
|
stop: bool # Permet d'indiquer la fin de l'exécution de l'agent
|
|
|
|
# TODO: Ajouter la source des documents sélectionnés pour la fin du rapport ?
|
|
|
|
|
|
def getState()->StateGraph:
|
|
"""
|
|
Retourne un StateGraph prêt à utiliser pour préparer un workflow
|
|
|
|
Returns:
|
|
StateGraph: prêt à utiliser
|
|
"""
|
|
return StateGraph(CustomState) |