from utils.prompts import GENERATOR_PROMPT
from utils.config import GENERATOR_MODEL
from langchain_openai import ChatOpenAI
from langchain_core.prompts import ChatPromptTemplate

# Prompt
prompt = ChatPromptTemplate.from_messages([("human", GENERATOR_PROMPT)])

prompt

# Generator LLM
llm = ChatOpenAI(model_name=GENERATOR_MODEL, temperature=0.7, streaming=True)


# Post-processing
def format_docs(docs):
    return "\n\n".join(doc.page_content for doc in docs)


# Chain
rag_chain = prompt | llm

# generation = rag_chain.invoke({"context": docs, "question": question})
# print(generation)