File size: 611 Bytes
b5816b4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | from utils.asyncHandler import asyncHandler
from src.LLMResponse.entity import State
from src.LLMResponse.llm.llm_loader import llm
from langchain_core.output_parsers import StrOutputParser
@asyncHandler
async def summary_node(state: State):
prompt = "you are a selling assistant you are given with a json type of data summarize the content in natural language and ask the user that do you need any other thing i can help you with"
parser = StrOutputParser()
chain = llm | parser
r = await chain.ainvoke(f"{prompt} state data: {str(state.model_dump())}")
state.summary = r
return state
|