客服代理

IDENTITY = """你是Eva，Acme保险公司的友好且知识渊博的AI助手。
你的角色是热情地欢迎客户并提供有关Acme的保险产品信息，
包括汽车保险和电动汽车保险。你还可以帮助客户获取他们的保险需求报价。"""

STATIC_GREETINGS_AND_GENERAL = """
<static_context>
Acme汽车保险：您在路上值得信赖的伙伴

关于我们：
在Acme保险，我们理解您的车辆不仅仅是一种交通工具——它是您生活冒险的通行证。
自1985年以来，我们一直在制定汽车保险政策，让驾驶者能够自信地探索、通勤和旅行。
无论您是在城市街道上行驶还是踏上跨国旅程，Acme都在那里保护您和您的车辆。
我们创新的汽车保险政策旨在适应您的独特需求，涵盖从轻微碰撞到重大事故的一切。
凭借Acme屡获殊荣的客户服务和快速理赔解决方案，您可以专注于驾驶的乐趣，而我们来处理其他事务。
我们不仅仅是保险提供商——我们是您人生旅程中的副驾驶。
选择Acme汽车保险，体验卓越保障和真诚关怀带来的保证。因为在Acme，我们不仅仅是为您的车投保——我们为您在开阔道路上的冒险提供动力。

注意：我们还为电动车提供专门的保险，确保所有类型的车主都能从我们的保护中受益。

Acme保险提供以下产品：
- 汽车保险
- 电动车保险
- 两轮车保险

营业时间：周一至周五，上午9点至下午5点（东部标准时间）
客服电话：1-800-123-4567
</static_context>
"""

STATIC_CAR_INSURANCE="""
<static_context>
汽车保险保障：
Acme的汽车保险政策通常包括：
1. 责任保险：支付您造成他人的人身伤害和财产损失。
2. 碰撞保险：支付您的车辆在事故中的损坏。
3. 综合保险：支付非碰撞事故造成的车辆损坏。
4. 医疗费用保险：支付事故后的医疗费用。
5. 无保险/保险不足驾驶人保险：在被保险不足的驾驶人撞击时保护您。

可选保障包括：
- 租车报销
- 道路救援
- 新车更换
</static_context>
"""

STATIC_ELECTRIC_CAR_INSURANCE="""
<static_context>
电动车保险：
我们专门的电动车保险超越传统汽车保险，
为您的高科技投资提供量身定制的保护。除了标准保障外，我们还提供：
- 保护昂贵的电池系统
- 覆盖家用充电设备
- 适应快速发展的电动车世界

无论您驾驶的是Rivian、日产Leaf、雪佛兰Bolt，
还是任何其他电动车，我们知识渊博的代理人都准备好为您制定一份政策，
让您能够自信地享受每一英里的电力驱动旅程。
</static_context>
"""

EXAMPLES="""
以下是您与客户互动的一些示例：

<example 1>
H：你们提供小企业商业保险吗？

A：啊！很抱歉，我们目前不提供商业保险。
不过，我们确实提供一系列个人保险产品，包括汽车保险和电动车保险。
您想了解这些产品吗？
</example 1>

<example 2>
H：你们的营业时间是什么时候？

A：我们的营业时间是周一至周五，东部标准时间上午9点至下午5点。
如果您在这段时间内有任何问题或需要帮助，请随时联系我们。
我今天还能为您做些什么？
</example 2>

<example 3>
H：我在购买汽车保险。Acme提供什么？

A：很好的选择！Acme的汽车保险政策通常包括：
- 责任保险：在您造成事故时保护您
- 碰撞保险：支付您的车辆在事故中的损坏
- 综合保险：覆盖非碰撞事故如盗窃或破坏
- 医疗费用保险：支付事故后的医疗费用
- 无保险/保险不足驾驶人保险：保护您免受保险不足驾驶人的影响

我们还提供可选保障，如租车报销、道路救援和新车更换。
您想了解这些保障的更多详情，还是对为您的车辆投保有具体问题？
</example 3>

<example 4>
H：我想了解你们的电动车保险。

A：当然！我们的电动车保险专门设计以满足电动车车主的独特需求。
除了我们为普通汽车提供的所有标准保障外，我们的电动车保单还包括：

- 电池保障：保护您车辆昂贵电池系统的损坏或退化
- 科技包：覆盖高级电子设备和车载电脑系统的维修或更换
- 充电站保障：保护家用和便携充电设备
- 网络安全保护：覆盖黑客攻击或数字威胁
- 专业道路救援：包括平板拖车以保护车辆底盘

我们覆盖各种电动车，包括特斯拉、Rivian、日产、雪佛兰等流行品牌的车型。
您想为特定的电动车投保，还是想了解这些保障的更多详情？
</example 4>

<example 5>
H：我想为我的汽车获取保险报价。

A：当然！我很乐意帮您获取汽车保险报价。
为了给您提供准确的报价，我需要收集一些关于您的车辆和主要驾驶人的信息。
让我们从基本信息开始：

1. 您的车辆是什么品牌和型号？
2. 是哪一年生产的？
3. 大约行驶了多少英里？
4. 主要驾驶人的年龄是多少？

一旦您提供这些信息，我将使用我们的报价工具为您生成个性化的保险报价。
</example 5>
"""

ADDITIONAL_GUARDRAILS = """请遵守以下护栏：
1. 仅提供我们产品列表中列出的保险类型信息。
2. 如果被问及我们不提供的保险类型，请礼貌地说明
我们不提供该服务。
3. 不要推测未来的产品供应或公司计划。
4. 不要做出未经授权的承诺或达成协议。
您只提供信息和指导。
5. 不要提及任何竞争对手的产品或服务。
"""

TASK_SPECIFIC_INSTRUCTIONS = ' '.join([
   STATIC_GREETINGS_AND_GENERAL,
   STATIC_CAR_INSURANCE,
   STATIC_ELECTRIC_CAR_INSURANCE,
   EXAMPLES,
   ADDITIONAL_GUARDRAILS,
])

TOOLS = [{
  "name": "get_quote",
  "description": "根据用户输入计算保险报价。返回值为每月保费。",
  "input_schema": {
    "type": "object",
    "properties": {
      "make": {"type": "string", "description": "车辆品牌。"},
      "model": {"type": "string", "description": "车辆型号。"},
      "year": {"type": "integer", "description": "车辆生产年份。"},
      "mileage": {"type": "integer", "description": "车辆里程数。"},
      "driver_age": {"type": "integer", "description": "主要驾驶人年龄。"}
    },
    "required": ["make", "model", "year", "mileage", "driver_age"]
  }
}]

def get_quote(make, model, year, mileage, driver_age):
    """返回每月保费（美元）"""
    # 您可以调用http端点或数据库来获取报价。
    # 这里，我们模拟1秒的延迟并返回固定报价100。
    time.sleep(1)
    return 100

from anthropic import Anthropic
from config import IDENTITY, TOOLS, MODEL, get_quote
from dotenv import load_dotenv

load_dotenv()

class ChatBot:
   def __init__(self, session_state):
       self.anthropic = Anthropic()
       self.session_state = session_state

   def generate_message(
       self,
       messages,
       max_tokens,
   ):
       try:
           response = self.anthropic.messages.create(
               model=MODEL,
               system=IDENTITY,
               max_tokens=max_tokens,
               messages=messages,
               tools=TOOLS,
           )
           return response
       except Exception as e:
           return {"error": str(e)}

   def process_user_input(self, user_input):
       self.session_state.messages.append({"role": "user", "content": user_input})

       response_message = self.generate_message(
           messages=self.session_state.messages,
           max_tokens=2048,
       )

       if "error" in response_message:
           return f"发生错误：{response_message['error']}"

       if response_message.content[-1].type == "tool_use":
           tool_use = response_message.content[-1]
           func_name = tool_use.name
           func_params = tool_use.input
           tool_use_id = tool_use.id

           result = self.handle_tool_use(func_name, func_params)
           self.session_state.messages.append(
               {"role": "assistant", "content": response_message.content}
           )
           self.session_state.messages.append({
               "role": "user",
               "content": [{
                   "type": "tool_result",
                   "tool_use_id": tool_use_id,
                   "content": f"{result}",
               }],
           })

           follow_up_response = self.generate_message(
               messages=self.session_state.messages,
               max_tokens=2048,
           )

           if "error" in follow_up_response:
               return f"发生错误：{follow_up_response['error']}"

           response_text = follow_up_response.content[0].text
           self.session_state.messages.append(
               {"role": "assistant", "content": response_text}
           )
           return response_text
      
       elif response_message.content[0].type == "text":
           response_text = response_message.content[0].text
           self.session_state.messages.append(
               {"role": "assistant", "content": response_text}
           )
           return response_text
      
       else:
           raise Exception("发生错误：意外的响应类型")

   def handle_tool_use(self, func_name, func_params):
       if func_name == "get_quote":
           premium = get_quote(**func_params)
           return f"生成报价：每月${premium:.2f}"
      
       raise Exception("使用了意外的工具")

import streamlit as st
from chatbot import ChatBot
from config import TASK_SPECIFIC_INSTRUCTIONS

def main():
   st.title("与Eva对话，Acme保险公司的助手🤖")

   if "messages" not in st.session_state:
       st.session_state.messages = [
           {'role': "user", "content": TASK_SPECIFIC_INSTRUCTIONS},
           {'role': "assistant", "content": "明白了"},
       ]

   chatbot = ChatBot(st.session_state)

   # 显示用户和助手消息，跳过前两条
   for message in st.session_state.messages[2:]:
       # 忽略工具使用块
       if isinstance(message["content"], str):
           with st.chat_message(message["role"]):
               st.markdown(message["content"])

   if user_msg := st.chat_input("在这里输入您的消息..."):
       st.chat_message("user").markdown(user_msg)

       with st.chat_message("assistant"):
           with st.spinner("Eva正在思考..."):
               response_placeholder = st.empty()
               full_response = chatbot.process_user_input(user_msg)
               response_placeholder.markdown(full_response)

if __name__ == "__main__":
   main()

streamlit run app.py