src/basic_bot/chatbot_node.py

from src.LLMS.groqllm import GroqLLM
from src.state.state import State

class ChatbotNode:
    """
    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'])}