src/genai/context_analysis_agent/utils/nodes.py

from .state import DetailsFormatter
from langchain_core.messages import SystemMessage
from src.genai.utils.models_loader import llm_gpt
from .prompts import introduction_prompt

class IntroductionNode:
    def __init__(self):
        self.llm = llm_gpt

    def run(self, state, llm):
        template = introduction_prompt
        messages = [SystemMessage(content=template)] + state["messages"]
        response = llm.invoke(messages)
        return {"messages": [response]}