客戶支援代理

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、Nissan Leaf、Chevrolet 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、Nissan、Chevrolet 等熱門車型。
您有特定的電動車想要投保，還是想了解這些保障中任何一項的更多詳情？
</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