Spaces:
Running
Running
File size: 2,569 Bytes
a75702e |
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 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 |
from langgraph.graph import StateGraph, MessagesState
from langchain_core.prompts import ChatPromptTemplate
from langgraph.prebuilt import ToolNode
from src.tools.customer_support_tools import query_knowledge_base, search_for_product_reccommendations, data_protection_check, create_new_customer, place_order, retrieve_existing_customer_orders
import os
class Customer_Support_Bot:
def __init__(self,llm):
self.llm = llm
def chat_bot(self):
prompt = """#Purpose
You are a customer service chatbot for a flower shop company. You can help the customer achieve the goals listed below.
#Goals
1. Answer questions the user might have relating to serivces offered
2. Recommend products to the user based on their preferences
3. Help the customer check on an existing order, or place a new order
4. To place and manage orders, you will need a customer profile (with an associated id). If the customer already has a profile, perform a data protection check to retrieve their details. If not, create them a profile.
#Tone
Helpful and friendly. Use gen-z emojis to keep things lighthearted. You MUST always include a funny flower related pun in every response."""
chat_template = ChatPromptTemplate.from_messages(
[
('system', prompt),
('placeholder', "{messages}")
]
)
tools = [query_knowledge_base, search_for_product_reccommendations, data_protection_check, create_new_customer, place_order, retrieve_existing_customer_orders]
llm = self.llm
llm_with_prompt = chat_template | llm.bind_tools(tools)
def call_agent(message_state: MessagesState):
response = llm_with_prompt.invoke(message_state)
return {
'messages': [response]
}
def is_there_tool_calls(state: MessagesState):
last_message = state['messages'][-1]
if last_message.tool_calls:
return 'tool_node'
else:
return '__end__'
graph = StateGraph(MessagesState)
tool_node = ToolNode(tools)
graph.add_node('agent', call_agent)
graph.add_node('tool_node', tool_node)
graph.add_conditional_edges(
"agent",
is_there_tool_calls
)
graph.add_edge('tool_node', 'agent')
graph.set_entry_point('agent')
app = graph.compile()
return app
|