Spaces:
Sleeping
Sleeping
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}
|