Агента продаж на основе ИИ

  """
  Agent продаж используя AgentInbox WebSocket

  Это простой пример, показывающий, как:
  - Подключиться к AgentInbox через WebSocket для обработки писем в реальном времени
  - Использовать OpenAI для обработки разговоров о продажах
  - Отправлять письма клиентам и отвечать на ответы
  """

  import asyncio
  import os
  import re
  from dotenv import load_dotenv
  from agentinbox import AsyncAgentInbox, Subscribe, Subscribed, MessageReceivedEvent
  from openai import AsyncOpenAI

  # Загрузите переменные окружения
  load_dotenv()

  # Инициализируйте клиентов
  agentinbox = AsyncAgentInbox(api_key=os.getenv("AGENTINBOX_API_KEY"))
  openai = AsyncOpenAI(api_key=os.getenv("OPENAI_API_KEY"))

  # Простая история беседы (thread_id -> сообщения)
  conversations = {}

  # Сохраните адрес электронной почты менеджера для уведомлений
  manager_email = None


  def extract_email(from_field):
      """Извлеките адрес электронной почты из формата 'Name <email@example.com>'"""
      match = re.search(r'<(.+?)>', from_field)
      return match.group(1) if match else from_field


  def is_from_manager(email_body):
      """Простая проверка, если письмо от менеджера по продажам (содержит информацию о клиенте)"""
      keywords = ['customer', 'lead', 'contact', 'reach out', 'email']
      return any(keyword in email_body.lower() for keyword in keywords)


  def extract_customer_info(email_body):
      """Извлеките адрес электронной почты клиента из сообщения менеджера"""
      email_pattern = r'\b[A-Za-z0-9._%+-]+@[A-Za-z0-9.-]+\.[A-Z|a-z]{2,}\b'
      emails = re.findall(email_pattern, email_body)

      # Верните первый найденный адрес электронной почты (должен быть адрес электронной почты клиента в теле сообщения)
      if emails:
          return emails[0]
      return None


  async def get_ai_response(messages, system_prompt):
      """Получите ответ от OpenAI"""
      try:
          response = await openai.chat.completions.create(
              model="gpt-4o-mini",
              messages=[
                  {"role": "system", "content": system_prompt},
                  *messages
              ],
              temperature=0.7,
          )
          return response.choices[0].message.content
      except Exception as e:
          print(f"Ошибка получения ответа ИИ: {e}")
          return "Извините, я столкнулся с ошибкой. Пожалуйста, попробуйте еще раз."


  async def send_email(inbox_id, to_email, subject, body):
      """Отправьте новое письмо"""
      try:
          await agentinbox.inboxes.messages.send(
              inbox_id=inbox_id,
              to=[to_email],
              subject=subject,
              text=body
          )
          print(f"✓ Письмо отправлено на {to_email}")
      except Exception as e:
          print(f"Ошибка при отправке письма: {e}")


  async def reply_to_email(inbox_id, message_id, to_email, body):
      """Ответьте на письмо"""
      try:
          await agentinbox.inboxes.messages.reply(
              inbox_id=inbox_id,
              message_id=message_id,
              to=[to_email],  # Требуемый параметр для ответов
              text=body
          )
          print(f"✓ Ответ отправлен на {to_email}")
      except Exception as e:
          print(f"Ошибка при ответе: {e}")


  async def handle_manager_email(inbox_id, message_id, from_email, subject, body):
      """Обработайте письмо от менеджера по продажам - извлеките клиента и отправьте торговое предложение"""
      global manager_email
      manager_email = from_email  # Запомните менеджера для будущих уведомлений

      print(f"\n📧 Письмо от МЕНЕДЖЕРА: {from_email}")

      # Извлеките адрес электронной почты клиента
      customer_email = extract_customer_info(body)
      print(f"→ Извлеченный адрес электронной почты клиента: {customer_email}")

      if not customer_email:
          await reply_to_email(
              inbox_id,
              message_id,
              from_email,  # Ответьте менеджеру
              "Я не смог найти адрес электронной почты клиента. Пожалуйста, включите его в ваше сообщение."
          )
          return

      # Создайте торговое предложение используя ИИ
      system_prompt = """Вы полезный агент по продажам. Создайте краткое профессиональное письмо о продажах
      на основе запроса менеджера. Держите его менее 150 слов. Будьте дружелюбны и профессиональны."""

      messages = [{"role": "user", "content": f"Создайте письмо о продажах на основе этого: {body}"}]
      sales_pitch = await get_ai_response(messages, system_prompt)

      # Отправьте письмо клиенту
      await send_email(
          inbox_id,
          customer_email,
          f"Введение: {subject}" if subject else "Быстрое введение",
          sales_pitch
      )

      # Подтвердите менеджеру
      await reply_to_email(
          inbox_id,
          message_id,
          from_email,  # Ответьте менеджеру
          f"✓ Я отправил письмо с введением на адрес {customer_email}.\n\nВот что я отправил:\n\n{sales_pitch}"
      )


  async def handle_customer_email(inbox_id, message_id, thread_id, from_email, subject, body):
      """Обработайте письмо от клиента - отслеживайте беседу, обнаруживайте намерение и уведомляйте менеджера"""
      print(f"\n📧 Письмо от КЛИЕНТА: {from_email}")

      # Отслеживайте историю беседы
      if thread_id not in conversations:
          conversations[thread_id] = []
      conversations[thread_id].append({"role": "user", "content": body})

      # Обнаружите намерение клиента
      intent_keywords = {
          'interested': ['interested', 'demo', 'meeting', 'tell me more', 'sounds good'],
          'not_interested': ['not interested', 'no thank', 'not right now', 'maybe later'],
          'question': ['?', 'how', 'what', 'when', 'why', 'can you']
      }

      body_lower = body.lower()
      intent = 'question'  # по умолчанию
      for key, keywords in intent_keywords.items():
          if any(keyword in body_lower for keyword in keywords):
              intent = key
              break

      # Создайте ответ на основе ИИ
      system_prompt = """Вы полезный агент по продажам. Отвечайте на вопросы клиентов профессионально
      и полезно. Держите ответы краткими (менее 100 слов). Будьте дружелюбны, но профессиональны."""

      response = await get_ai_response(conversations[thread_id], system_prompt)

      # Ответьте клиенту
      await reply_to_email(inbox_id, message_id, from_email, response)

      # Уведомите менеджера при сильном сигнале намерения
      if manager_email and intent in ['interested', 'not_interested']:
          status = "проявляет интерес" if intent == 'interested' else "не заинтересован в данный момент"
          await send_email(
              inbox_id,
              manager_email,
              f"Обновление: {from_email}",
              f"Клиент {from_email} {status}.\n\nИх сообщение:\n{body}\n\nМой ответ:\n{response}"
          )
          print(f"→ Уведомлен менеджер о намерении {intent} клиента")

      # Обновите историю беседы
      conversations[thread_id].append({"role": "assistant", "content": response})


  async def handle_new_email(message):
      """Обработайте входящее письмо из WebSocket"""
      try:
          # Извлеките данные сообщения, используя атрибуты объекта
          inbox_id = message.inbox_id
          message_id = message.message_id
          thread_id = message.thread_id
          from_field = message.from_ or ""  # SDK использует from_
          from_email = extract_email(from_field)
          subject = message.subject or ""
          body = message.text or ""  # SDK использует text для тела

          print(f"\n{'='*60}")
          print(f"Новое письмо от: {from_email}")
          print(f"Тема: {subject}")
          print(f"{'='*60}")

          # Маршрутизируйте на основе содержимого письма
          if is_from_manager(body):
              await handle_manager_email(inbox_id, message_id, from_email, subject, body)
          else:
              await handle_customer_email(inbox_id, message_id, thread_id, from_email, subject, body)

      except Exception as e:
          print(f"Ошибка при обработке письма: {e}")


  async def main():
      """Основной цикл WebSocket"""
      inbox_username = os.getenv("INBOX_USERNAME", "sales-agent")
      inbox_id = f"{inbox_username}@agentinbox.ru"

      print(f"\nAgent продаж начинает работу...")
      print(f"Входящий ящик: {inbox_id}")
      print(f"✓ Подключение к AgentInbox WebSocket...")

      # Подключитесь к WebSocket
      try:
          async with agentinbox.websockets.connect() as socket:
              print(f"✓ Подключено! Ожидание писем...\n")

              # Подпишитесь на входящий ящик
              await socket.send_subscribe(Subscribe(inbox_ids=[inbox_id]))

              # Слушайте события
              async for event in socket:
                  if isinstance(event, Subscribed):
                      print(f"✓ Подписано на: {event.inbox_ids}\n")

                  elif isinstance(event, MessageReceivedEvent):
                      print(f"📨 Новое письмо получено!")
                      await handle_new_email(event.message)

      except (KeyboardInterrupt, asyncio.CancelledError):
          print("\n\nЗавершение работы корректно...")
      except Exception as e:
          print(f"\nОшибка: {e}")


  def run():
      """Запустите основную функцию"""
      try:
          asyncio.run(main())
      except KeyboardInterrupt:
          print("\n✓ Завершение завершено")


  if __name__ == "__main__":
      run()

  client = AsyncAgentInbox(api_key="your-key")
  print(await client.inboxes.list())  # Должна успешно выполниться

  pip show agentmail

  client = AsyncAgentInbox()
  print(await client.inboxes.get("sales-agent@agentinbox.ru"))

  python --version