ИИ-агент квалифицирует лиды в amoCRM: горячие, тёплые, холодные — автоматически за 30 секунд

BANT_PROMPT = """Ты помощник по квалификации лидов. 
Проанализируй данные лида и оцени по критериям BANT.

КРИТЕРИИ ОЦЕНКИ:

Budget (Бюджет):
- Высокий: упомянул бюджет, должность директор/владелец/руководитель, B2B-сегмент
- Средний: корпоративная почта, пришёл с коммерческого запроса
- Низкий: частное лицо, нет упоминания бюджета, учебный/информационный запрос

Authority (Полномочия):
- Высокий: должность ЛПР в форме, пишет от имени компании, принимает решение
- Средний: не указал должность, но корпоративная почта
- Низкий: явно не ЛПР, «согласовываю с руководством», студент

Need (Потребность):
- Высокий: описал конкретную задачу с деталями, использует профессиональные термины
- Средний: общий интерес без конкретики
- Низкий: «просто интересуюсь», нет описания задачи

Timeline (Срочность):
- Высокий: «срочно», «до конца месяца», «уже выбираем», «переходим от конкурента»
- Средний: «планируем», «рассматриваем»
- Низкий: «когда-нибудь», «на будущее», нет указания срока

ДАННЫЕ ЛИДА:
{lead_data}

Верни JSON строго в таком формате:
{{
"status": "горячий" | "тёплый" | "холодный",
"score": {{
  "budget": "высокий" | "средний" | "низкий",
  "authority": "высокий" | "средний" | "низкий", 
  "need": "высокий" | "средний" | "низкий",
  "timeline": "высокий" | "средний" | "низкий"
}},
"reasoning": "2-3 предложения почему такой статус",
"next_action": "конкретное действие менеджера"
}}

Правило: если данных мало — ставь "тёплый", не "холодный". 
Лучше перепроверить тёплого, чем потерять горячего."""

import json
import httpx
import anthropic
from fastapi import FastAPI, Request

app = FastAPI()

# Настройки
AMOCRM_DOMAIN = "yourcompany.amocrm.ru"
AMOCRM_TOKEN = "YOUR_ACCESS_TOKEN"      # OAuth 2.0 токен amoCRM
CLAUDE_API_KEY = "YOUR_CLAUDE_KEY"

AMO_HEADERS = {
  "Authorization": f"Bearer {AMOCRM_TOKEN}",
  "Content-Type": "application/json",
}

BANT_PROMPT = """Ты помощник по квалификации лидов.
Проанализируй данные лида и оцени по критериям BANT.
[... полный промпт из раздела выше ...]
Верни JSON строго в описанном формате."""


def parse_amo_webhook(form_data: dict) -> dict:
  """amoCRM шлёт bracket-нотацию: leads[add][0][id] = 12345"""
  lead = {}
  for key, value in form_data.items():
      # Берём первый добавленный лид
      if key.startswith("leads[add][0]["):
          field = key.replace("leads[add][0][", "").rstrip("]")
          lead[field] = value
  return lead


async def get_lead_details(lead_id: str) -> dict:
  """Получаем полные данные лида через amoCRM API v4."""
  async with httpx.AsyncClient() as client:
      # Лид с кастомными полями и контактами
      r = await client.get(
          f"https://{AMOCRM_DOMAIN}/api/v4/leads/{lead_id}",
          headers=AMO_HEADERS,
          params={"with": "contacts,custom_fields_values"},
      )
      r.raise_for_status()
      return r.json()


async def qualify_with_claude(lead_details: dict) -> dict:
  """Отправляем данные в Claude, получаем BANT-оценку."""
  # Собираем данные для анализа
  custom_fields = {
      cf["field_name"]: cf.get("values", [{}])[0].get("value", "")
      for cf in lead_details.get("custom_fields_values") or []
  }

  lead_data = {
      "name": lead_details.get("name", ""),
      "pipeline": lead_details.get("_embedded", {}).get("pipeline", {}).get("name", ""),
      "source": custom_fields.get("Источник", ""),
      "utm_source": custom_fields.get("UTM Source", ""),
      "utm_medium": custom_fields.get("UTM Medium", ""),
      "message": custom_fields.get("Сообщение", ""),
      "company": custom_fields.get("Компания", ""),
      "position": custom_fields.get("Должность", ""),
  }

  client = anthropic.Anthropic(api_key=CLAUDE_API_KEY)
  message = client.messages.create(
      model="claude-sonnet-4-6",
      max_tokens=512,
      messages=[{
          "role": "user",
          "content": BANT_PROMPT.replace("{lead_data}", json.dumps(lead_data, ensure_ascii=False, indent=2))
      }]
  )

  # Claude возвращает JSON — парсим
  response_text = message.content[0].text
  # Ищем JSON в ответе
  start = response_text.find("{")
  end = response_text.rfind("}") + 1
  return json.loads(response_text[start:end])


async def update_lead_in_amo(lead_id: str, qualification: dict):
  """Обновляем карточку лида: теги + примечание."""
  status = qualification["status"]
  reasoning = qualification.get("reasoning", "")
  next_action = qualification.get("next_action", "")

  async with httpx.AsyncClient() as client:
      # 1. Добавляем тег к лиду через PATCH /api/v4/leads/{id}
      tag_data = {
          "_embedded": {
              "tags": [{"name": f"квалификация:{status}"}]
          }
      }
      await client.patch(
          f"https://{AMOCRM_DOMAIN}/api/v4/leads/{lead_id}",
          headers=AMO_HEADERS,
          json=tag_data,
      )

      # 2. Добавляем примечание с объяснением
      note_text = (
          f"🤖 Автоквалификация: {status.upper()}

"
          f"BANT: B={qualification['score']['budget']} | "
          f"A={qualification['score']['authority']} | "
          f"N={qualification['score']['need']} | "
          f"T={qualification['score']['timeline']}

"
          f"Вывод: {reasoning}

"
          f"Следующий шаг: {next_action}"
      )
      note_data = {
          "note_type": "common",
          "params": {"text": note_text},
      }
      await client.post(
          f"https://{AMOCRM_DOMAIN}/api/v4/leads/{lead_id}/notes",
          headers=AMO_HEADERS,
          json=[note_data],
      )


@app.post("/webhook/amo-lead")
async def handle_new_lead(request: Request):
  """Принимаем webhook от amoCRM о новом лиде."""
  form_data = dict(await request.form())
  lead_basic = parse_amo_webhook(form_data)

  lead_id = lead_basic.get("id")
  if not lead_id:
      return {"status": "skipped", "reason": "no lead id"}

  # Полные данные, оценка, обновление карточки
  lead_details = await get_lead_details(lead_id)
  qualification = await qualify_with_claude(lead_details)
  await update_lead_in_amo(lead_id, qualification)

  return {"status": "ok", "lead_id": lead_id, "qualification": qualification["status"]}

# Запуск: uvicorn qualify_leads:app --host 0.0.0.0 --port 8000