| 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 |
|
|
