src/genai/context_analysis_agent/utils/nodes.py

from langchain_core.messages import SystemMessage
from src.genai.utils.models_loader import llm_gpt
from .prompts import introduction_prompt
from .state import ConversationFormatter, State


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

    def run(self, state:State):
        template = introduction_prompt
        messages = [SystemMessage(content=template)] + state["messages"]
        result = self.llm.with_structured_output(ConversationFormatter).invoke(messages)
        print('Response:', result)
        return {
        "messages": [{"role": "assistant", "content": result.response}],
        "response": result.response,
        "completion": result.completion,
    }