Multi-Rag / src /Web /nodes /webSummerizer_node.py
VashuTheGreat's picture
Clean commit without images
1f725d8
from langchain_community.document_loaders import WebBaseLoader
from utils.asyncHandler import asyncHandler
from src.Web.models.web_model import State
from src.Web.llm.llm_loader import llm
from src.Web.prompts.prompt_templates import WEB_SUMERISER_PROMPT
from langchain_core.messages import SystemMessage, HumanMessage
from langchain_core.output_parsers import StrOutputParser
import logging
@asyncHandler
async def web_summ_node(state:State)->State:
logging.info("Entered in the web_summ_node node")
prompt=[
SystemMessage(
content=WEB_SUMERISER_PROMPT
),
HumanMessage(
content=state['page_content']
)
]
res=await (llm | StrOutputParser()).ainvoke(prompt)
logging.info("Exited from the web_summ_node node")
return {"llm_response":res}