realtime-avatar 0.3.0

package/dist/tools.js

@@ -0,0 +1,121 @@
+// ../tools/src/index.ts
+var MAX_TOOLS = 32;
+var MAX_MANIFEST_BYTES = 12e3;
+var MAX_RESULT_CHARS = 8192;
+var TOOL_DEADLINE_MS = 2500;
+var REGISTER_METHOD = "rta.tools.register";
+var INVOKE_METHOD = "rta.tool.invoke";
+var NAME_RE = /^[a-zA-Z0-9_-]{1,64}$/;
+function candidateIdentities(room) {
+  const all = Array.from(room.remoteParticipants.values(), (p) => p.identity).filter(Boolean);
+  const agents = all.filter((id) => /^agent[-_]/i.test(id));
+  return [...agents, ...all.filter((id) => !agents.includes(id))];
+}
+function methodMissing(error) {
+  const code = error?.code;
+  if (code === "UNSUPPORTED_METHOD" || code === "RECIPIENT_NOT_FOUND" || code === 1401 || code === 1402) {
+    return true;
+  }
+  const message = error instanceof Error ? error.message : String(error ?? "");
+  return /not supported|unsupported_method|recipient_not_found|recipient not found/i.test(message);
+}
+var sleep = (ms) => new Promise((resolve) => setTimeout(resolve, ms));
+function buildManifest(tools) {
+  const descriptors = [];
+  const rejected = [];
+  for (const [name, tool] of Object.entries(tools)) {
+    if (!NAME_RE.test(name)) {
+      rejected.push({ name, reason: "name must match [a-zA-Z0-9_-]{1,64}" });
+      continue;
+    }
+    if (!tool?.description) {
+      rejected.push({ name, reason: "description is required \u2014 it is how she knows when to call it" });
+      continue;
+    }
+    if (typeof tool.execute !== "function") {
+      rejected.push({ name, reason: "execute must be a function" });
+      continue;
+    }
+    if (descriptors.length >= MAX_TOOLS) {
+      rejected.push({ name, reason: `over the ${MAX_TOOLS}-tool limit for one session` });
+      continue;
+    }
+    descriptors.push({ name, description: tool.description, parameters: tool.parameters ?? {} });
+  }
+  return { descriptors, rejected };
+}
+async function attachAvatarTools(room, tools, options = {}) {
+  const { descriptors, rejected } = buildManifest(tools);
+  const payload = JSON.stringify({ tools: descriptors });
+  if (payload.length > MAX_MANIFEST_BYTES) {
+    throw new Error(
+      `tool manifest is ${payload.length} bytes, over the ${MAX_MANIFEST_BYTES} limit \u2014 shorten descriptions or register fewer tools`
+    );
+  }
+  room.localParticipant.registerRpcMethod(INVOKE_METHOD, async ({ payload: payload2 }) => {
+    let name = "", callId = "";
+    try {
+      const req = JSON.parse(payload2);
+      name = req.name;
+      callId = req.call_id;
+      const tool = tools[name];
+      if (!tool) return JSON.stringify({ ok: false, call_id: callId, error: `no tool named ${name}` });
+      const args = req.args ? JSON.parse(req.args) : {};
+      options.onInvoke?.(name, args);
+      const run = tool.execute;
+      const controller = new AbortController();
+      const timer = setTimeout(() => controller.abort(), TOOL_DEADLINE_MS);
+      let result;
+      try {
+        result = await run(args, { signal: controller.signal, callId });
+      } finally {
+        clearTimeout(timer);
+      }
+      const encoded = typeof result === "string" ? result : JSON.stringify(result ?? null);
+      if (encoded.length > MAX_RESULT_CHARS) {
+        return JSON.stringify({
+          ok: false,
+          call_id: callId,
+          error: `result was ${encoded.length} chars, over the ${MAX_RESULT_CHARS} limit`
+        });
+      }
+      return JSON.stringify({ ok: true, call_id: callId, result: encoded });
+    } catch (error) {
+      return JSON.stringify({
+        ok: false,
+        call_id: callId,
+        error: error instanceof Error ? error.message : "tool failed"
+      });
+    }
+  });
+  const deadline = Date.now() + Math.max(0, options.timeoutMs ?? 8e3);
+  let lastError = null;
+  let sawCandidate = false;
+  for (; ; ) {
+    for (const identity of candidateIdentities(room)) {
+      sawCandidate = true;
+      try {
+        const raw = await room.localParticipant.performRpc({
+          destinationIdentity: identity,
+          method: REGISTER_METHOD,
+          payload
+        });
+        const result = JSON.parse(raw);
+        return {
+          accepted: result.accepted ?? [],
+          rejected: [...rejected, ...result.rejected ?? []]
+        };
+      } catch (error) {
+        lastError = error;
+        if (!methodMissing(error)) throw error;
+      }
+    }
+    if (Date.now() >= deadline) break;
+    await sleep(200);
+  }
+  throw new Error(
+    sawCandidate ? `no participant accepted the tool manifest \u2014 either this session was not minted with the \`client_tools\` capability (a server-side grant), or the agent never armed registration within ${options.timeoutMs ?? 8e3}ms` + (lastError instanceof Error ? ` (last error: ${lastError.message})` : "") : "no remote participant joined the room before the timeout \u2014 nothing to register with"
+  );
+}
+
+export { MAX_MANIFEST_BYTES, MAX_RESULT_CHARS, MAX_TOOLS, TOOL_DEADLINE_MS, attachAvatarTools };