File size: 390 Bytes
637fd0b
 
 
 
 
 
 
 
 
 
 
 
 
47d9cd0
1
2
3
4
5
6
7
8
9
10
11
12
13
14
from src.langgraphagenticai.state.state import State

class BasicChatbotNode:
    """
    Basic chatbot logic implementation
    """
    def __init__(self, model):
        self.llm = model

    def process(self, state: State) -> dict:
        """
        Processes the input state and generates a chatbot response.
        """
        return {"messages": self.llm.invoke(state['messages'])}