feat: Telegram-инбокс — сообщения Алекса как события
- harness/tg_inbox.py: long-polling поллер (@juno42bot) + send_reply - события MessageReceived/ReplySent/MessageIgnored (агрегат inbox) - проекция inbox в read model (кап 100) - контекст минутного тика: входящие ждущие ответа + мандат триажа - cli.py tg-reply: Telegram-отправка + ReplySent закрывает входящее - systemd simbiot-harness-tg.service - E2E: «йоу» от Алекса → минутный тик ответил в TG + дайджест в сторе
This commit is contained in:
parent
c941ec7ec8
commit
94cc921ccd
11 changed files with 175 additions and 3 deletions
1
.gitignore
vendored
1
.gitignore
vendored
|
|
@ -6,3 +6,4 @@ venv/
|
||||||
.agent/
|
.agent/
|
||||||
__pycache__/
|
__pycache__/
|
||||||
/tmp/
|
/tmp/
|
||||||
|
.env
|
||||||
|
|
|
||||||
|
|
@ -43,6 +43,7 @@ Strands Harness SDK как исполнительное ядро, модели
|
||||||
./venv/bin/python cli.py events -n 50
|
./venv/bin/python cli.py events -n 50
|
||||||
./venv/bin/python cli.py requeue <task-id> -r "причина"
|
./venv/bin/python cli.py requeue <task-id> -r "причина"
|
||||||
./venv/bin/python cli.py promote <task-id> -r "вне класса уровня" # S→M→L
|
./venv/bin/python cli.py promote <task-id> -r "вне класса уровня" # S→M→L
|
||||||
|
./venv/bin/python cli.py tg-reply <msg-id> "текст" # ответ на входящее TG
|
||||||
```
|
```
|
||||||
|
|
||||||
## Запуск
|
## Запуск
|
||||||
|
|
@ -52,6 +53,13 @@ systemctl enable --now simbiot-harness
|
||||||
journalctl -u simbiot-harness -f
|
journalctl -u simbiot-harness -f
|
||||||
```
|
```
|
||||||
|
|
||||||
|
## Telegram-инбокс
|
||||||
|
|
||||||
|
Бот @juno42bot → long polling (`simbiot-harness-tg.service`) →
|
||||||
|
`MessageReceived` в стор → минутный уровень видит входящие в контексте →
|
||||||
|
отвечает `cli.py tg-reply` (Telegram-отправка + `ReplySent` закрывает входящее).
|
||||||
|
Белый список чатов в `.env` (`SIMBIOT_TG_ALLOWED`), чужие → `MessageIgnored`.
|
||||||
|
|
||||||
## Модели
|
## Модели
|
||||||
|
|
||||||
9router (router.cil.by), отдельный ключ `sk-simbiot` в `.env`, комбо `auto`:
|
9router (router.cil.by), отдельный ключ `sk-simbiot` в `.env`, комбо `auto`:
|
||||||
|
|
|
||||||
Binary file not shown.
9
cli.py
9
cli.py
|
|
@ -31,6 +31,10 @@ def main():
|
||||||
p_pr.add_argument("task_id")
|
p_pr.add_argument("task_id")
|
||||||
p_pr.add_argument("-r", "--reason", default="manual promote")
|
p_pr.add_argument("-r", "--reason", default="manual promote")
|
||||||
|
|
||||||
|
p_tgr = sub.add_parser("tg-reply", help="ответить на входящее TG-сообщение")
|
||||||
|
p_tgr.add_argument("msg_id", help="id входящего (m-NNN)")
|
||||||
|
p_tgr.add_argument("text")
|
||||||
|
|
||||||
sub.add_parser("status", help="состояние харнеса")
|
sub.add_parser("status", help="состояние харнеса")
|
||||||
sub.add_parser("tasks", help="доска задач")
|
sub.add_parser("tasks", help="доска задач")
|
||||||
|
|
||||||
|
|
@ -61,6 +65,11 @@ def main():
|
||||||
es.init_db()
|
es.init_db()
|
||||||
tid, new_size = commands.promote_task(args.task_id, args.reason)
|
tid, new_size = commands.promote_task(args.task_id, args.reason)
|
||||||
print(f"OK: задача {tid} повышена до [{new_size}]")
|
print(f"OK: задача {tid} повышена до [{new_size}]")
|
||||||
|
elif args.cmd == "tg-reply":
|
||||||
|
es.init_db()
|
||||||
|
from harness import tg_inbox
|
||||||
|
mid = tg_inbox.send_reply(args.msg_id, args.text)
|
||||||
|
print(f"OK: ответ на {mid} отправлен")
|
||||||
elif args.cmd == "status":
|
elif args.cmd == "status":
|
||||||
es.init_db()
|
es.init_db()
|
||||||
print(json.dumps(commands.status(), ensure_ascii=False, indent=1))
|
print(json.dumps(commands.status(), ensure_ascii=False, indent=1))
|
||||||
|
|
|
||||||
Binary file not shown.
Binary file not shown.
Binary file not shown.
|
|
@ -194,6 +194,26 @@ def _apply_event(model: dict, ev: dict):
|
||||||
spend.append(data)
|
spend.append(data)
|
||||||
if len(spend) > 500: # кап: не растить модель линейно
|
if len(spend) > 500: # кап: не растить модель линейно
|
||||||
model["spend"] = spend[-500:]
|
model["spend"] = spend[-500:]
|
||||||
|
elif etype == "MessageReceived":
|
||||||
|
model.setdefault("inbox", {})[aid] = {
|
||||||
|
**data, "replied": False, "ignored": False,
|
||||||
|
"ts": ev["timestamp"],
|
||||||
|
}
|
||||||
|
inbox = model["inbox"]
|
||||||
|
if len(inbox) > 100: # кап инбокса: последние 100 входящих
|
||||||
|
keep = sorted(inbox.items(), key=lambda kv: kv[1].get("ts", ""))[-100:]
|
||||||
|
model["inbox"] = dict(keep)
|
||||||
|
elif etype == "ReplySent":
|
||||||
|
m = model.setdefault("inbox", {}).get(aid)
|
||||||
|
if m:
|
||||||
|
m["replied"] = True
|
||||||
|
m["reply_text"] = data.get("text", "")
|
||||||
|
elif etype == "MessageIgnored":
|
||||||
|
m = model.setdefault("inbox", {}).get(aid)
|
||||||
|
if m:
|
||||||
|
m["ignored"] = True
|
||||||
|
else:
|
||||||
|
model["inbox"][aid] = {"ignored": True, "reason": data.get("reason"), "ts": ev["timestamp"]}
|
||||||
# trace-события (InvocationStarted/ToolCalled/ToolResult) не материализуем — они для дебага
|
# trace-события (InvocationStarted/ToolCalled/ToolResult) не материализуем — они для дебага
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -43,9 +43,12 @@ LEVELS = {
|
||||||
"min": CycleLevel(
|
"min": CycleLevel(
|
||||||
name="min", label="Минута · пульс", every=MIN, uses_llm=True,
|
name="min", label="Минута · пульс", every=MIN, uses_llm=True,
|
||||||
model="auto", effort="low", ctx_tokens=4_000, task_classes=("S",),
|
model="auto", effort="low", ctx_tokens=4_000, task_classes=("S",),
|
||||||
mandate=("Ты — минутный пульс харнеса Simbiot. Триаж входящих, быстрые ответы, "
|
mandate=("Ты — минутный пульс харнеса Simbiot. Триаж входящих TG-сообщений: "
|
||||||
"мелкие фиксы (S-задачи до 5 минут). Отвечай кратко, не углубляйся. "
|
"если во ВХОДЯЩИХ есть сообщение без ответа — ответь на него командой "
|
||||||
"Если задача больше твоего класса — откажись и предложи промоушен."),
|
"cli.py tg-reply <id> \"текст\" (тон: кратко, по-русски, тёпло; можно "
|
||||||
|
"выполнить просьбу инструментами и ответить результатом). Быстрые "
|
||||||
|
"мелкие фиксы (S-задачи до 5 минут) тоже твои. Не растекайся. "
|
||||||
|
"Если просьба большая — подай её как задачу (cli.py task) и скажи это."),
|
||||||
can_shell=True,
|
can_shell=True,
|
||||||
),
|
),
|
||||||
"hour": CycleLevel(
|
"hour": CycleLevel(
|
||||||
|
|
|
||||||
|
|
@ -89,6 +89,17 @@ def build_context(level_cfg, model: dict, task=None):
|
||||||
for d in digests[:6]:
|
for d in digests[:6]:
|
||||||
parts.append(f"- [{d['level']}] {d['text'][:400]}")
|
parts.append(f"- [{d['level']}] {d['text'][:400]}")
|
||||||
|
|
||||||
|
# инбокс: входящие TG-сообщения, ждущие ответа (в приоритете)
|
||||||
|
inbox = [m for m in (model.get("inbox") or {}).values()
|
||||||
|
if not m.get("replied") and not m.get("ignored") and m.get("text")]
|
||||||
|
if inbox:
|
||||||
|
lines = []
|
||||||
|
for m in inbox[:10]:
|
||||||
|
m_id = "m-" + str(m.get("update_id"))
|
||||||
|
who = m.get("username") or m.get("first_name") or m.get("user_id")
|
||||||
|
lines.append(f"- [{m_id}] от {who}: {m['text'][:300]}")
|
||||||
|
parts.append("ВХОДЯЩИЕ СООБЩЕНИЯ (ответь через: ./venv/bin/python cli.py tg-reply <id> \"текст\"):\n" + "\n".join(lines))
|
||||||
|
|
||||||
# доска задач
|
# доска задач
|
||||||
tasks = [t for t in model["tasks"].values() if t["status"] in ("ready", "running")]
|
tasks = [t for t in model["tasks"].values() if t["status"] in ("ready", "running")]
|
||||||
if tasks:
|
if tasks:
|
||||||
|
|
|
||||||
120
harness/tg_inbox.py
Normal file
120
harness/tg_inbox.py
Normal file
|
|
@ -0,0 +1,120 @@
|
||||||
|
"""Simbiot Harness — Telegram-инбокс.
|
||||||
|
|
||||||
|
Входящие сообщения Алекса → события MessageReceived в стор →
|
||||||
|
минутный уровень видит их в контексте и может ответить через
|
||||||
|
`cli.py tg-reply <msg_id> <text>` (событие ReplySent закрывает входящее).
|
||||||
|
|
||||||
|
Поллер — отдельный процесс (systemd simbiot-harness-tg), long polling
|
||||||
|
getUpdates offset=last+1. Идемпотентность: update_id → aggregate_id.
|
||||||
|
"""
|
||||||
|
import json
|
||||||
|
import os
|
||||||
|
import time
|
||||||
|
import urllib.request
|
||||||
|
import urllib.parse
|
||||||
|
|
||||||
|
from . import es
|
||||||
|
|
||||||
|
_HARNESS_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||||||
|
_ENV_PATH = os.path.join(_HARNESS_DIR, ".env")
|
||||||
|
|
||||||
|
TG_TOKEN = None
|
||||||
|
ALLOWED = set()
|
||||||
|
|
||||||
|
|
||||||
|
def _load_env():
|
||||||
|
global TG_TOKEN, ALLOWED
|
||||||
|
if os.path.exists(_ENV_PATH):
|
||||||
|
with open(_ENV_PATH) as f:
|
||||||
|
for line in f:
|
||||||
|
line = line.strip()
|
||||||
|
if line.startswith("SIMBIOT_TG_TOKEN="):
|
||||||
|
TG_TOKEN = line.split("=", 1)[1].strip()
|
||||||
|
elif line.startswith("SIMBIOT_TG_ALLOWED="):
|
||||||
|
ALLOWED = {x.strip() for x in line.split("=", 1)[1].split(",") if x.strip()}
|
||||||
|
|
||||||
|
|
||||||
|
def _api(method: str, params: dict, timeout: int = 35):
|
||||||
|
url = f"https://api.telegram.org/bot{TG_TOKEN}/{method}"
|
||||||
|
data = urllib.parse.urlencode(params).encode()
|
||||||
|
req = urllib.request.Request(url, data=data)
|
||||||
|
with urllib.request.urlopen(req, timeout=timeout) as r:
|
||||||
|
return json.loads(r.read().decode())
|
||||||
|
|
||||||
|
|
||||||
|
def send_message(chat_id, text: str, reply_to=None):
|
||||||
|
"""Отправка сообщения (используется tg-reply и алертами)."""
|
||||||
|
_load_env()
|
||||||
|
params = {"chat_id": chat_id, "text": text, "parse_mode": "HTML"}
|
||||||
|
if reply_to:
|
||||||
|
params["reply_to_message_id"] = reply_to
|
||||||
|
return _api("sendMessage", params)
|
||||||
|
|
||||||
|
|
||||||
|
def receive_message(update: dict):
|
||||||
|
"""MessageReceived в стор. Идемпотентно по update_id."""
|
||||||
|
msg = update.get("message") or update.get("edited_message") or {}
|
||||||
|
chat = msg.get("chat") or {}
|
||||||
|
user = msg.get("from") or {}
|
||||||
|
data = {
|
||||||
|
"update_id": update["update_id"],
|
||||||
|
"chat_id": chat.get("id"),
|
||||||
|
"user_id": user.get("id"),
|
||||||
|
"username": user.get("username"),
|
||||||
|
"first_name": user.get("first_name"),
|
||||||
|
"text": (msg.get("text") or "")[:4000],
|
||||||
|
"date": msg.get("date"),
|
||||||
|
}
|
||||||
|
es.append_event("inbox", f"m-{update['update_id']}", "MessageReceived", data)
|
||||||
|
return f"m-{update['update_id']}"
|
||||||
|
|
||||||
|
|
||||||
|
def send_reply(msg_id: str, text: str):
|
||||||
|
"""Ответ на входящее: Telegram-отправка + событие ReplySent (закрывает входящее)."""
|
||||||
|
model = es.run_projections()
|
||||||
|
m = (model.get("inbox") or {}).get(msg_id)
|
||||||
|
if m is None:
|
||||||
|
raise ValueError(f"входящее {msg_id} не найдено")
|
||||||
|
if m.get("replied"):
|
||||||
|
raise ValueError(f"на {msg_id} уже отвечено")
|
||||||
|
r = send_message(m["chat_id"], text, reply_to=None)
|
||||||
|
es.append_event("inbox", msg_id, "ReplySent", {
|
||||||
|
"to_chat": m["chat_id"], "text": text[:2000], "tg_ok": bool(r.get("ok")),
|
||||||
|
})
|
||||||
|
return msg_id
|
||||||
|
|
||||||
|
|
||||||
|
def poll_loop():
|
||||||
|
"""Long polling getUpdates → MessageReceived. Отдельный процесс."""
|
||||||
|
_load_env()
|
||||||
|
if not TG_TOKEN:
|
||||||
|
raise SystemExit("SIMBIOT_TG_TOKEN не задан в .env")
|
||||||
|
es.init_db()
|
||||||
|
offset = 0
|
||||||
|
print("[tg] инбокс-поллер запущен", flush=True)
|
||||||
|
while True:
|
||||||
|
try:
|
||||||
|
updates = _api("getUpdates", {"offset": offset, "timeout": 30, "limit": 16})
|
||||||
|
for u in updates.get("result", []):
|
||||||
|
offset = max(offset, u["update_id"] + 1)
|
||||||
|
msg = u.get("message") or u.get("edited_message")
|
||||||
|
if not msg:
|
||||||
|
continue
|
||||||
|
user = (msg.get("from") or {}).get("id")
|
||||||
|
if ALLOWED and str(user) not in ALLOWED:
|
||||||
|
# чужой: пишем событие, но не отвечаем (потенциально спам)
|
||||||
|
es.append_event("inbox", f"m-{u['update_id']}", "MessageIgnored", {
|
||||||
|
"user_id": user, "username": (msg.get("from") or {}).get("username"),
|
||||||
|
"reason": "not_allowed", "text": (msg.get("text") or "")[:500],
|
||||||
|
})
|
||||||
|
continue
|
||||||
|
mid = receive_message(u)
|
||||||
|
print(f"[tg] {mid} от {(msg.get('from') or {}).get('username') or user}: "
|
||||||
|
f"{(msg.get('text') or '')[:60]}", flush=True)
|
||||||
|
except Exception as e:
|
||||||
|
print(f"[tg] ошибка поллинга: {e}", flush=True)
|
||||||
|
time.sleep(5)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
poll_loop()
|
||||||
Loading…
Reference in a new issue