File size: 766 Bytes
93a5bf9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from langchain_core.messages import SystemMessage
from .prompts import introduction_prompt, details_extract_prompt
from .state import State , DetailsFormatter
from utils.models_loader import llm
# This node generates chatbot responses using the LLM
def introduction_node(state, llm):
template = introduction_prompt
messages = [SystemMessage(content=template)] + state["messages"]
response = llm.invoke(messages)
return {"messages": [response]}
# This function extracts structured business details
def extract_business_details(interactions):
template = details_extract_prompt(interactions)
messages = [SystemMessage(content=template)]
response=llm.with_structured_output(DetailsFormatter).invoke(messages)
return response
|