客户支持代理

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

STATIC_GREETINGS_AND_GENERAL = """
<static_context>
Acme汽车保险：您在路上的可信伴侣

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

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

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

营业时间：周一至周五，上午9点 - 下午5点EST
客户服务电话：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点EST。
如果您有任何问题或需要帮助，请随时在这些时间联系我们。
今天我还能如何帮助您？
</example 2>

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

A: 很好的选择！Acme的汽车保险政策通常包括：
- 责任保险：如果您在事故中有过错，保护您
- 碰撞保险：为事故中您汽车的损坏付费
- 综合保险：涵盖非碰撞事件，如盗窃或故意破坏
- 医疗费用保险：为事故后的医疗费用付费
- 无保险/保险不足的驾驶员保险：保护您免受保险不足的司机的伤害

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

<example 4>
H: 我想了解您的电动汽车保险。

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

- 电池保险：保护您车辆昂贵电池系统的损坏或退化
- 技术套餐：涵盖先进电子设备和车载计算机系统的维修或更换
- 充电站保险：保护家庭和便携式充电设备
- 网络安全保护：防范黑客攻击或数字威胁的保险
- 专门的道路救援：包括平板拖车以保护车辆底盘

我们涵盖广泛的电动汽车，包括Tesla、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