File size: 617 Bytes
7a3f093
b55b8d4
7a3f093
ef9fa4b
a6a0614
b55b8d4
5c271a3
7a3f093
b55b8d4
 
7a3f093
 
 
 
 
 
 
 
 
 
 
5c271a3
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
from langgraph.graph import StateGraph, START, END
from langgraph.checkpoint.memory import MemorySaver
from .utils.state import State
from .utils.nodes import IntroductionNode
import os


class ContextAnalysisAgent:
    def __init__(self):
        self.memory = MemorySaver()

    def context_analysis_graph(self):
        graph_builder= StateGraph(State)
        graph_builder.add_node("details_collector", IntroductionNode().run)
        graph_builder.add_edge(START, "details_collector")
        graph_builder.add_edge("details_collector", END)



        return graph_builder.compile(checkpointer=self.memory)