src/agents/base/flow.py

from langgraph.graph import StateGraph, START, END
from .func import State
from langgraph.graph.state import CompiledStateGraph
from langgraph.store.memory import InMemoryStore
class PrimaryChatBot:
    def __init__(self):
        self.builder = StateGraph(State)

    @staticmethod
    def routing(state: State):
        pass

    def node(self):
        pass

    def edge(self):
        pass

    def __call__(self) -> CompiledStateGraph:
        self.node()
        self.edge()
        return self.builder.compile(checkpointer=InMemoryStore())