Готовые сценарии

#!/usr/bin/env bash
set -euo pipefail

: "${FORGE_URL:?укажите адрес портала}"
: "${FORGE_API_KEY:?укажите личный ключ}"

AUTH="Authorization: Bearer ${FORGE_API_KEY}"

# 1. Запуск задачи с файлом на входе
TASK_ID=$(curl -sS -X POST "${FORGE_URL}/api/tasks" \
  -H "$AUTH" \
  -F "taskTypeSlug=sales-summary" \
  -F "params={\"period\":\"2026-33\"}" \
  -F "visibility=private" \
  -F "files=@продажи-33-неделя.xlsx" \
  | jq -r '.data.id')
echo "задача: ${TASK_ID}"

# 2. Ожидание завершения: опрос раз в 5 секунд
while :; do
  STATUS=$(curl -sS "${FORGE_URL}/api/tasks/${TASK_ID}" -H "$AUTH" | jq -r '.data.status')
  case "${STATUS}" in
    done|partial|failed) break ;;
  esac
  sleep 5
done
echo "статус: ${STATUS}"

# 3. Список файлов задачи
curl -sS "${FORGE_URL}/api/tasks/${TASK_ID}/files" -H "$AUTH" \
  | jq -r '.data[] | [.filename, .direction, (.isFinalResult|tostring)] | @tsv'

# 4. Скачивание выходных файлов
for FILE_ID in $(curl -sS "${FORGE_URL}/api/tasks/${TASK_ID}/files" -H "$AUTH" \
  | jq -r '.data[] | select(.direction == "output") | .id'); do
  curl -sS -OJ "${FORGE_URL}/api/tasks/${TASK_ID}/files/${FILE_ID}/download" -H "$AUTH"
done

import json
import os
import time
from pathlib import Path

import requests

FORGE_URL = os.environ["FORGE_URL"].rstrip("/")
HEADERS = {"Authorization": f"Bearer {os.environ['FORGE_API_KEY']}"}
POLL_SECONDS = 5


def main() -> None:
    # 1. Запуск задачи с файлом на входе
    with open("продажи-33-неделя.xlsx", "rb") as fh:
        response = requests.post(
            f"{FORGE_URL}/api/tasks",
            headers=HEADERS,
            data={
                "taskTypeSlug": "sales-summary",
                "params": json.dumps({"period": "2026-33"}),
                "visibility": "private",
            },
            files={"files": ("продажи-33-неделя.xlsx", fh)},
            timeout=120,
        )
    response.raise_for_status()
    task_id = response.json()["data"]["id"]
    print("задача:", task_id)

    # 2. Ожидание завершения: опрос раз в 5 секунд
    while True:
        task = requests.get(
            f"{FORGE_URL}/api/tasks/{task_id}", headers=HEADERS, timeout=30
        ).json()["data"]
        if task["status"] in ("done", "partial", "failed"):
            break
        time.sleep(POLL_SECONDS)
    print("статус:", task["status"])

    # 3. Файлы задачи
    files = requests.get(
        f"{FORGE_URL}/api/tasks/{task_id}/files", headers=HEADERS, timeout=30
    ).json()["data"]

    # 4. Скачивание выходных файлов
    for item in files:
        if item["direction"] != "output":
            continue
        stream = requests.get(
            f"{FORGE_URL}/api/tasks/{task_id}/files/{item['id']}/download",
            headers=HEADERS,
            timeout=300,
            stream=True,
        )
        stream.raise_for_status()
        Path(item["filename"]).write_bytes(stream.content)
        print("сохранено:", item["filename"])


if __name__ == "__main__":
    main()

#!/usr/bin/env bash
set -euo pipefail

: "${FORGE_URL:?укажите адрес портала}"
: "${FORGE_API_KEY:?укажите личный ключ}"
: "${TASK_TYPE_ID:?укажите taskTypeId из ответа GET /api/tasks/:id}"

AUTH="Authorization: Bearer ${FORGE_API_KEY}"

new_uuid() { uuidgen | tr '[:upper:]' '[:lower:]'; }

# 1. Создание сессии
SESSION_ID=$(curl -sS -X POST "${FORGE_URL}/api/chat-sessions" \
  -H "$AUTH" -H "Content-Type: application/json" \
  -d "{\"taskTypeId\":\"${TASK_TYPE_ID}\"}" \
  | jq -r '.data.id')
echo "сессия: ${SESSION_ID}"

# 2. Отправка вопроса. Ответ — поток событий; соединение держится,
# пока портал не завершит ход.
send_message() {
  local text="$1"
  curl -sS -N --max-time 600 -X POST "${FORGE_URL}/api/chat-sessions/${SESSION_ID}/messages" \
    -H "$AUTH" -H "Content-Type: application/json" \
    -d "{\"text\":\"${text}\",\"attachedFileIds\":[],\"clientMessageId\":\"$(new_uuid)\"}" \
    > /dev/null
}

# 3. Ожидание ответа: опрос раз в 3 секунды, пока последнее сообщение
# не станет готовым ответом ассистента.
wait_for_answer() {
  while :; do
    RESULT=$(curl -sS "${FORGE_URL}/api/chat-sessions/${SESSION_ID}" -H "$AUTH" \
      | jq -r '.data.messages[-1] | select(. != null) | "\(.role) \(.status)"')
    case "${RESULT}" in
      "assistant completed"|"assistant stopped"|"assistant failed") break ;;
    esac
    sleep 3
  done
}

# 4. Печать последнего ответа ассистента
print_answer() {
  curl -sS "${FORGE_URL}/api/chat-sessions/${SESSION_ID}" -H "$AUTH" \
    | jq -r '[.data.messages[] | select(.role == "assistant")] | last | .content'
}

send_message "Сколько продаж закрыто на этой неделе?"
wait_for_answer
print_answer

# 5. Уточнение в ту же сессию
send_message "Из них сколько закрыто без замечаний?"
wait_for_answer
print_answer

import os
import time
import uuid

import requests

FORGE_URL = os.environ["FORGE_URL"].rstrip("/")
HEADERS = {"Authorization": f"Bearer {os.environ['FORGE_API_KEY']}"}
TASK_TYPE_ID = os.environ["TASK_TYPE_ID"]  # поле taskTypeId из ответа GET /api/tasks/:id
POLL_SECONDS = 3
TERMINAL = ("completed", "stopped", "failed")


def create_session() -> str:
    response = requests.post(
        f"{FORGE_URL}/api/chat-sessions",
        headers=HEADERS,
        json={"taskTypeId": TASK_TYPE_ID},
        timeout=30,
    )
    response.raise_for_status()
    return response.json()["data"]["id"]


def send_message(session_id: str, text: str) -> None:
    """Отправляет сообщение и держит соединение, пока портал не завершит ход."""
    body = {
        "text": text,
        "attachedFileIds": [],
        "clientMessageId": str(uuid.uuid4()),
    }
    with requests.post(
        f"{FORGE_URL}/api/chat-sessions/{session_id}/messages",
        headers=HEADERS,
        json=body,
        stream=True,
        timeout=(10, 600),
    ) as stream:
        for _ in stream.iter_lines():
            pass


def wait_for_answer(session_id: str) -> str:
    """Опрашивает сессию, пока последнее сообщение не станет готовым ответом."""
    while True:
        data = requests.get(
            f"{FORGE_URL}/api/chat-sessions/{session_id}", headers=HEADERS, timeout=30
        ).json()["data"]
        messages = data["messages"]
        if messages and messages[-1]["role"] == "assistant" and messages[-1]["status"] in TERMINAL:
            return messages[-1]["content"]
        time.sleep(POLL_SECONDS)


def main() -> None:
    session_id = create_session()
    print("сессия:", session_id)

    send_message(session_id, "Сколько продаж закрыто на этой неделе?")
    print(wait_for_answer(session_id))

    # Уточнение в ту же сессию
    send_message(session_id, "Из них сколько закрыто без замечаний?")
    print(wait_for_answer(session_id))


if __name__ == "__main__":
    main()

#!/usr/bin/env bash
set -euo pipefail

: "${FORGE_URL:?укажите адрес портала}"
: "${FORGE_API_KEY:?укажите личный ключ}"

AUTH="Authorization: Bearer ${FORGE_API_KEY}"

# Пауза перед повтором: число читается из текста сообщения об ошибке.
wait_seconds() {
  local msg="$1" n
  if n=$(printf '%s' "$msg" | grep -oE 'через [0-9]+ минут' | grep -oE '[0-9]+' | head -1); then
    echo $((n * 60))
  elif n=$(printf '%s' "$msg" | grep -oE 'через [0-9]+ мин' | grep -oE '[0-9]+' | head -1); then
    echo $((n * 60))
  elif n=$(printf '%s' "$msg" | grep -oE '[0-9]+ сек' | grep -oE '[0-9]+' | head -1); then
    echo "$n"
  else
    echo 60
  fi
}

# Запрос с повтором при 429 и таймаутом 60 секунд.
request() {
  local method="$1" url="$2" body="${3:-}"
  local attempt=0 response code body_text msg pause
  while :; do
    attempt=$((attempt + 1))
    if [ -n "$body" ]; then
      response=$(curl -sS --max-time 60 -w $'\n%{http_code}' -X "$method" \
        "$url" -H "$AUTH" -H "Content-Type: application/json" -d "$body")
    else
      response=$(curl -sS --max-time 60 -w $'\n%{http_code}' -X "$method" \
        "$url" -H "$AUTH")
    fi
    code="${response##*$'\n'}"
    body_text="${response%$'\n'*}"
    case "$code" in
      2*) printf '%s\n' "$body_text"; return 0 ;;
      429)
        msg=$(printf '%s' "$body_text" | jq -r '.error.message // empty')
        pause=$(wait_seconds "$msg")
        echo "429, повтор через ${pause} секунд (попытка ${attempt})" >&2
        sleep "$pause"
        ;;
      401|403)
        echo "код ${code}: $(printf '%s' "$body_text" | jq -r '.error.message')" >&2
        return 1
        ;;
      *)
        echo "код ${code}: ${body_text}" >&2
        return 1
        ;;
    esac
  done
}

# Пример: статус задачи с повтором при 429
request GET "${FORGE_URL}/api/tasks/7d4f2a91-3c58-4e6b-9a02-1f8e6c4d2b37" \
  | jq -r '.data.status'

import os
import re
import sys
import time

import requests

FORGE_URL = os.environ["FORGE_URL"].rstrip("/")
HEADERS = {"Authorization": f"Bearer {os.environ['FORGE_API_KEY']}"}

# Портал не шлёт заголовок Retry-After: пауза стоит в тексте сообщения.
MINUTES_RE = re.compile(r"через (\d+) мин(ут)?")
SECONDS_RE = re.compile(r"(\d+) сек")


def wait_seconds(message: str) -> int:
    match = MINUTES_RE.search(message)
    if match:
        return int(match.group(1)) * 60
    match = SECONDS_RE.search(message)
    if match:
        return int(match.group(1))
    return 60


def request(method: str, url: str, **kwargs) -> requests.Response:
    """Выполняет запрос с таймаутом и повторяет его при 429."""
    while True:
        response = requests.request(method, url, headers=HEADERS, timeout=60, **kwargs)
        if response.status_code != 429:
            return response
        message = ""
        try:
            message = response.json()["error"]["message"]
        except (ValueError, KeyError):
            pass
        pause = wait_seconds(message)
        print(f"429, повтор через {pause} секунд", file=sys.stderr)
        time.sleep(pause)


# Пример: статус задачи с повтором при 429
task = request("GET", f"{FORGE_URL}/api/tasks/7d4f2a91-3c58-4e6b-9a02-1f8e6c4d2b37").json()["data"]
print(task["status"])