File size: 820 Bytes
1f725d8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
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}