b60439152b
- harness/es.py: append-only event store + проекции (образец — кооп-портал) - harness/levels.py: уровни 1с/1м/1ч/1д, контекст ∝ масштабу - harness/scheduler.py: уровневые потоки, идемпотентные тики, absorbed-стабы - harness/isolated_runner.py + _llm_worker.py: LLM-тик в сабпроцессе (глобальный asyncio-рантайм Strands не переживает 2+ create_harness) - cli.py: init/task/tasks/status/events + requeue (написан самим минутным агентом) - 9router: отдельный ключ sk-simbiot, модель auto (glm-5.3 первым)
130 lines
No EOL
4.5 KiB
Python
130 lines
No EOL
4.5 KiB
Python
"""Simbiot Harness — Strands runtime-адаптер.
|
||
|
||
create_harness() на тик: модель/effort из конфига уровня.
|
||
Hooks → трасса в event store (агрегат trace).
|
||
"""
|
||
import json
|
||
import os
|
||
import uuid
|
||
from datetime import datetime, timezone
|
||
|
||
_HARNESS_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
||
|
||
|
||
def _router_key():
|
||
env = os.environ.get("ROUTER_KEY", "")
|
||
if not env:
|
||
path = os.path.join(_HARNESS_DIR, ".env")
|
||
if os.path.exists(path):
|
||
for line in open(path):
|
||
if line.startswith("ROUTER_KEY="):
|
||
env = line.split("=", 1)[1].strip()
|
||
break
|
||
return env
|
||
|
||
|
||
def setup_provider():
|
||
"""9router как openai-совместимый провайдер strands."""
|
||
key = _router_key()
|
||
if not key:
|
||
raise RuntimeError("ROUTER_KEY не найден (.env)")
|
||
os.environ["OPENAI_API_KEY"] = key
|
||
os.environ["OPENAI_BASE_URL"] = os.environ.get("ROUTER_BASE_URL", "http://127.0.0.1:20128/v1")
|
||
|
||
|
||
def _wire_trace_hooks(agent, es, trace_id: str):
|
||
"""Каждая стадия агентного цикла → событие агрегата trace."""
|
||
from strands import Agent
|
||
from strands.hooks import BeforeInvocationEvent, BeforeToolCallEvent, AfterToolCallEvent
|
||
|
||
seq = {"n": 0}
|
||
|
||
def _trace(kind: str, payload: dict):
|
||
seq["n"] += 1
|
||
es.append_event("trace", f"{trace_id}:{seq['n']}", kind, payload)
|
||
|
||
def on_before_invocation(event):
|
||
_trace("InvocationStarted", {"trace_id": trace_id, "ts": _now()})
|
||
|
||
def on_before_tool(event):
|
||
tool = getattr(event, "tool_use", None) or {}
|
||
_trace("ToolCalled", {
|
||
"tool": tool.get("name", "?"),
|
||
"input": json.dumps(tool.get("input", {}), ensure_ascii=False, default=str)[:4000],
|
||
})
|
||
|
||
def on_after_tool(event):
|
||
try:
|
||
res = str(getattr(event, "result", ""))[:4000]
|
||
except Exception:
|
||
res = "<unreadable>"
|
||
_trace("ToolResult", {"tool": "?", "result": res})
|
||
|
||
agent.add_hook(on_before_invocation, BeforeInvocationEvent)
|
||
agent.add_hook(on_before_tool, BeforeToolCallEvent)
|
||
agent.add_hook(on_after_tool, AfterToolCallEvent)
|
||
return agent
|
||
|
||
|
||
def _now():
|
||
return datetime.now(timezone.utc).isoformat()
|
||
|
||
|
||
def build_agent(level_cfg, session_id=None, extra_instructions=""):
|
||
"""Собирает Strands-агента для тика уровня. None если уровень без LLM."""
|
||
if not level_cfg.uses_llm:
|
||
return None
|
||
setup_provider()
|
||
from strands_harness import create_harness
|
||
|
||
instructions = level_cfg.mandate + ("\n\n" + extra_instructions if extra_instructions else "")
|
||
agent = create_harness(
|
||
model=f"openai/{level_cfg.model}",
|
||
effort=level_cfg.effort,
|
||
instructions=instructions,
|
||
session={"id": session_id} if session_id else None,
|
||
builtin_tools=("read", "write", "edit", "shell") if level_cfg.can_shell else [],
|
||
memory=False,
|
||
skills=False,
|
||
)
|
||
return agent
|
||
|
||
|
||
def run_cycle(level_cfg, es, prompt: str, session_id=None, trace_prefix=None):
|
||
"""Прогон тика уровня с LLM и трассой. Возвращает (text, usage_dict, trace_id)."""
|
||
setup_provider()
|
||
from strands_harness import create_harness
|
||
|
||
trace_id = trace_prefix or uuid.uuid4().hex[:12]
|
||
instructions = level_cfg.mandate
|
||
agent = create_harness(
|
||
model=f"openai/{level_cfg.model}",
|
||
effort=level_cfg.effort,
|
||
instructions=instructions,
|
||
session={"id": session_id} if session_id else None,
|
||
builtin_tools=("read", "write", "edit", "shell") if level_cfg.can_shell else [],
|
||
memory=False,
|
||
skills=False,
|
||
)
|
||
agent = _wire_trace_hooks(agent, es, trace_id)
|
||
result = agent(prompt)
|
||
# AgentResult: текст из message, usage из metrics.accumulated_usage
|
||
# (старый разбор agent.messages молчал → BudgetSpent всегда был пуст).
|
||
text = str(result)
|
||
usage = _extract_usage(result)
|
||
return text, usage, trace_id
|
||
|
||
|
||
def _extract_usage(result):
|
||
"""Usage из AgentResult.metrics.accumulated_usage (input/output/total)."""
|
||
try:
|
||
u = getattr(getattr(result, "metrics", None), "accumulated_usage", None)
|
||
if u:
|
||
return {
|
||
"prompt_tokens": u.get("inputTokens"),
|
||
"completion_tokens": u.get("outputTokens"),
|
||
"total_tokens": u.get("totalTokens"),
|
||
}
|
||
except Exception:
|
||
pass
|
||
return {} |