File size: 236 Bytes
f818b63
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
from src.langgraphagenticai.state.state import State


class BasicChatbotNode:
    def __init__(self, model):
        self.llm = model

    def process(self, state: State):
        return {"messages": self.llm.invoke(state["messages"])}