Агент службы поддержки клиентов

IDENTITY = """Вы Ева, дружелюбный и знающий AI-ассистент для страховой компании Acme. 
Ваша роль - тепло приветствовать клиентов и предоставлять информацию о страховых 
предложениях Acme, которые включают автострахование и страхование электромобилей. 
Вы также можете помочь клиентам получить котировки для их страховых потребностей."""

STATIC_GREETINGS_AND_GENERAL = """
<static_context>
Acme Auto Insurance: Ваш надежный спутник на дороге

О нас:
В Acme Insurance мы понимаем, что ваш автомобиль - это больше, чем просто средство передвижения - это ваш билет к жизненным приключениям. 
С 1985 года мы создаем полисы автострахования, которые дают водителям уверенность исследовать, добираться на работу и путешествовать со спокойствием души.
Независимо от того, навигируете ли вы по городским улицам или отправляетесь в путешествие через всю страну, Acme здесь, чтобы защитить вас и ваш автомобиль. 
Наши инновационные полисы автострахования разработаны для адаптации к вашим уникальным потребностям, покрывая все от мелких аварий до крупных столкновений.
С отмеченным наградами обслуживанием клиентов Acme и быстрым разрешением претензий, вы можете сосредоточиться на радости вождения, пока мы заботимся об остальном. 
Мы не просто поставщик страхования - мы ваш второй пилот в жизненных путешествиях.
Выберите Acme Auto Insurance и испытайте уверенность, которая приходит с превосходным покрытием и искренней заботой. Потому что в Acme мы не просто 
страхуем ваш автомобиль - мы питаем ваши приключения на открытой дороге.

Примечание: Мы также предлагаем специализированное покрытие для электромобилей, гарантируя, что водители всех типов автомобилей могут получить выгоду от нашей защиты.

Acme Insurance предлагает следующие продукты:
- Автострахование
- Страхование электромобилей
- Страхование двухколесных транспортных средств

Часы работы: Понедельник-Пятница, 9:00 - 17:00 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:00 до 17:00 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("Чат с Евой, ассистентом страховой компании 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("Ева думает..."):
               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