2dd0d1f2cf
Use autoescape for Jinja rendering to satisfy bandit checks and\nremove deprecated Clerk SignInButton props to restore type checks.\nAlso ignore tsbuildinfo artifacts and tidy boot instructions.\n\nCo-Authored-By: Claude <noreply@anthropic.com>
123 lines
3.7 KiB
Python
123 lines
3.7 KiB
Python
from __future__ import annotations
|
|
|
|
import re
|
|
from pathlib import Path
|
|
from uuid import uuid4
|
|
|
|
from jinja2 import Environment, FileSystemLoader, StrictUndefined, select_autoescape
|
|
|
|
from app.core.config import settings
|
|
from app.integrations.openclaw_gateway import ensure_session, send_message
|
|
from app.models.agents import Agent
|
|
|
|
TEMPLATE_FILES = [
|
|
"AGENTS.md",
|
|
"BOOT.md",
|
|
"BOOTSTRAP.md",
|
|
"HEARTBEAT.md",
|
|
"IDENTITY.md",
|
|
"SOUL.md",
|
|
"TOOLS.md",
|
|
"USER.md",
|
|
]
|
|
|
|
|
|
def _repo_root() -> Path:
|
|
return Path(__file__).resolve().parents[3]
|
|
|
|
|
|
def _templates_root() -> Path:
|
|
return _repo_root() / "templates"
|
|
|
|
|
|
def _slugify(value: str) -> str:
|
|
slug = re.sub(r"[^a-z0-9]+", "-", value.lower()).strip("-")
|
|
return slug or uuid4().hex
|
|
|
|
|
|
def _template_env() -> Environment:
|
|
return Environment(
|
|
loader=FileSystemLoader(_templates_root()),
|
|
autoescape=select_autoescape(default=True),
|
|
undefined=StrictUndefined,
|
|
keep_trailing_newline=True,
|
|
)
|
|
|
|
|
|
def _read_templates(context: dict[str, str]) -> dict[str, str]:
|
|
env = _template_env()
|
|
templates: dict[str, str] = {}
|
|
for name in TEMPLATE_FILES:
|
|
path = _templates_root() / name
|
|
if not path.exists():
|
|
templates[name] = ""
|
|
continue
|
|
template = env.get_template(name)
|
|
templates[name] = template.render(**context).strip()
|
|
return templates
|
|
|
|
|
|
def _render_file_block(name: str, content: str) -> str:
|
|
body = content if content else f"# {name}\n\nTODO: add content\n"
|
|
return f"\n{name}\n```md\n{body}\n```\n"
|
|
|
|
|
|
def _workspace_path(agent_name: str) -> str:
|
|
root = settings.openclaw_workspace_root or "~/.openclaw/workspaces"
|
|
root = root.rstrip("/")
|
|
return f"{root}/{_slugify(agent_name)}"
|
|
|
|
|
|
def build_provisioning_message(agent: Agent, auth_token: str) -> str:
|
|
agent_id = str(agent.id)
|
|
workspace_path = _workspace_path(agent.name)
|
|
session_key = agent.openclaw_session_id or ""
|
|
base_url = settings.base_url or "REPLACE_WITH_BASE_URL"
|
|
|
|
context = {
|
|
"agent_name": agent.name,
|
|
"agent_id": agent_id,
|
|
"session_key": session_key,
|
|
"workspace_path": workspace_path,
|
|
"base_url": base_url,
|
|
"auth_token": auth_token,
|
|
"main_session_key": settings.openclaw_main_session_key or "agent:main:main",
|
|
"workspace_root": settings.openclaw_workspace_root or "~/.openclaw/workspaces",
|
|
"user_name": "Unset",
|
|
"user_preferred_name": "Unset",
|
|
"user_timezone": "Unset",
|
|
"user_notes": "Fill in user context.",
|
|
}
|
|
|
|
templates = _read_templates(context)
|
|
|
|
file_blocks = "".join(
|
|
_render_file_block(name, templates.get(name, "")) for name in TEMPLATE_FILES
|
|
)
|
|
|
|
return (
|
|
"Provision a new OpenClaw agent workspace.\n\n"
|
|
f"Agent name: {agent.name}\n"
|
|
f"Agent id: {agent_id}\n"
|
|
f"Session key: {session_key}\n"
|
|
f"Workspace path: {workspace_path}\n\n"
|
|
f"Base URL: {base_url}\n"
|
|
f"Auth token: {auth_token}\n\n"
|
|
"Steps:\n"
|
|
"1) Create the workspace directory.\n"
|
|
"2) Write the files below with the exact contents.\n"
|
|
"3) Update TOOLS.md if BASE_URL/AUTH_TOKEN must change.\n"
|
|
"4) Leave BOOTSTRAP.md in place; the agent should run it on first start and delete it.\n"
|
|
"5) Register agent id in OpenClaw so it uses this workspace path.\n\n"
|
|
"Files:" + file_blocks
|
|
)
|
|
|
|
|
|
async def send_provisioning_message(agent: Agent, auth_token: str) -> None:
|
|
main_session = settings.openclaw_main_session_key
|
|
if not main_session:
|
|
return
|
|
await ensure_session(main_session, label="Main Agent")
|
|
message = build_provisioning_message(agent, auth_token)
|
|
await send_message(message, session_key=main_session, deliver=False)
|