@treeseed/agent 0.8.5

package/dist/api/agent-routes.js

@@ -0,0 +1,327 @@
+import { listRegisteredAgentHandlers as listCoreRegisteredAgentHandlers } from "../agents/registry.js";
+import { buildTaskContext, enqueueTaskFromSdk } from "../services/common.js";
+import { jsonError, requireScope } from "./http.js";
+async function listRegisteredHandlers() {
+  return listCoreRegisteredAgentHandlers();
+}
+async function safeListRegisteredHandlers() {
+  try {
+    return {
+      handlers: await listRegisteredHandlers(),
+      error: null
+    };
+  } catch (error) {
+    return {
+      handlers: [],
+      error: error instanceof Error ? error.message : String(error)
+    };
+  }
+}
+function withPrefix(prefix, path) {
+  return `${prefix}${path}`.replace(/\/{2,}/g, "/");
+}
+function actor(body, fallback) {
+  return String(body.actor ?? fallback);
+}
+function authorizeRequest(c, options) {
+  const routeUnauthorized = options.authorize?.(c);
+  if (routeUnauthorized) {
+    return routeUnauthorized;
+  }
+  if (options.scope) {
+    return requireScope(c, options.scope);
+  }
+  return null;
+}
+function registerAgentRoutes(app, options) {
+  const prefix = options.prefix ?? "/agent";
+  const defaultActor = options.defaultActor ?? "api";
+  app.get(withPrefix(prefix, "/healthz"), async (c) => {
+    const registration = await safeListRegisteredHandlers();
+    return c.json({
+      ok: true,
+      service: "treeseed-agent-api",
+      handlerCount: registration.handlers.length,
+      registrationError: registration.error
+    });
+  });
+  app.get(withPrefix(prefix, "/specs"), async (c) => {
+    const unauthorized = authorizeRequest(c, options);
+    if (unauthorized) return unauthorized;
+    const payload = await options.sdk.listAgentSpecs({ enabled: true });
+    return c.json({
+      ok: true,
+      payload,
+      handlers: await listRegisteredHandlers()
+    });
+  });
+  app.post(withPrefix(prefix, "/workdays/start"), async (c) => {
+    const unauthorized = authorizeRequest(c, options);
+    if (unauthorized) return unauthorized;
+    return jsonError(c, 410, "Starting workdays through /agent/workdays/start is deprecated. Use project workday policy and workday requests instead.");
+  });
+  app.post(withPrefix(prefix, "/workdays/:id/close"), async (c) => {
+    const unauthorized = authorizeRequest(c, options);
+    if (unauthorized) return unauthorized;
+    const body = await c.req.json().catch(() => ({}));
+    const result = await options.sdk.closeWorkDay({
+      id: c.req.param("id"),
+      state: body.state,
+      summary: body.summary ?? null,
+      actor: actor(body, defaultActor)
+    });
+    return result.payload ? c.json(result) : jsonError(c, 404, "Unknown work day.");
+  });
+  app.post(withPrefix(prefix, "/tasks"), async (c) => {
+    const unauthorized = authorizeRequest(c, options);
+    if (unauthorized) return unauthorized;
+    const body = await c.req.json().catch(() => ({}));
+    const result = await options.sdk.createTask({
+      id: typeof body.id === "string" ? body.id : void 0,
+      workDayId: String(body.workDayId ?? ""),
+      agentId: String(body.agentId ?? ""),
+      type: String(body.type ?? ""),
+      state: typeof body.state === "string" ? body.state : "pending",
+      priority: body.priority === void 0 ? void 0 : Number(body.priority),
+      idempotencyKey: String(body.idempotencyKey ?? ""),
+      payload: body.payload ?? {},
+      payloadHash: typeof body.payloadHash === "string" ? body.payloadHash : null,
+      maxAttempts: body.maxAttempts === void 0 ? void 0 : Number(body.maxAttempts),
+      availableAt: typeof body.availableAt === "string" ? body.availableAt : void 0,
+      graphVersion: typeof body.graphVersion === "string" ? body.graphVersion : null,
+      parentTaskId: typeof body.parentTaskId === "string" ? body.parentTaskId : null,
+      actor: actor(body, defaultActor)
+    });
+    return c.json(result);
+  });
+  app.post(withPrefix(prefix, "/tasks/search"), async (c) => {
+    const unauthorized = authorizeRequest(c, options);
+    if (unauthorized) return unauthorized;
+    const body = await c.req.json().catch(() => ({}));
+    const result = await options.sdk.searchTasks({
+      workDayId: typeof body.workDayId === "string" ? body.workDayId : void 0,
+      agentId: typeof body.agentId === "string" ? body.agentId : void 0,
+      state: Array.isArray(body.state) || typeof body.state === "string" ? body.state : void 0,
+      limit: body.limit === void 0 ? void 0 : Number(body.limit)
+    });
+    return c.json(result);
+  });
+  app.post(withPrefix(prefix, "/tasks/:id/claim"), async (c) => {
+    const unauthorized = authorizeRequest(c, options);
+    if (unauthorized) return unauthorized;
+    const body = await c.req.json().catch(() => ({}));
+    const result = await options.sdk.claimTask({
+      id: c.req.param("id"),
+      workerId: String(body.workerId ?? "worker"),
+      leaseSeconds: Number(body.leaseSeconds ?? 120),
+      actor: actor(body, defaultActor)
+    });
+    return result.payload ? c.json(result) : jsonError(c, 404, "Unknown task.");
+  });
+  app.post(withPrefix(prefix, "/tasks/:id/progress"), async (c) => {
+    const unauthorized = authorizeRequest(c, options);
+    if (unauthorized) return unauthorized;
+    const body = await c.req.json().catch(() => ({}));
+    const result = await options.sdk.recordTaskProgress({
+      id: c.req.param("id"),
+      workerId: typeof body.workerId === "string" ? body.workerId : null,
+      state: typeof body.state === "string" ? body.state : void 0,
+      appendEvent: body.appendEvent,
+      patch: body.patch,
+      actor: actor(body, defaultActor)
+    });
+    return result.payload ? c.json(result) : jsonError(c, 404, "Unknown task.");
+  });
+  app.post(withPrefix(prefix, "/tasks/:id/complete"), async (c) => {
+    const unauthorized = authorizeRequest(c, options);
+    if (unauthorized) return unauthorized;
+    const body = await c.req.json().catch(() => ({}));
+    const result = await options.sdk.completeTask({
+      id: c.req.param("id"),
+      output: body.output ?? null,
+      outputRef: typeof body.outputRef === "string" ? body.outputRef : null,
+      summary: body.summary ?? null,
+      actor: actor(body, defaultActor)
+    });
+    return result.payload ? c.json(result) : jsonError(c, 404, "Unknown task.");
+  });
+  app.post(withPrefix(prefix, "/tasks/:id/fail"), async (c) => {
+    const unauthorized = authorizeRequest(c, options);
+    if (unauthorized) return unauthorized;
+    const body = await c.req.json().catch(() => ({}));
+    const result = await options.sdk.failTask({
+      id: c.req.param("id"),
+      errorCode: typeof body.errorCode === "string" ? body.errorCode : null,
+      errorMessage: String(body.errorMessage ?? "Task failed"),
+      retryable: Boolean(body.retryable),
+      nextVisibleAt: typeof body.nextVisibleAt === "string" ? body.nextVisibleAt : null,
+      actor: actor(body, defaultActor)
+    });
+    return result.payload ? c.json(result) : jsonError(c, 404, "Unknown task.");
+  });
+  app.post(withPrefix(prefix, "/tasks/:id/requeue"), async (c) => {
+    const unauthorized = authorizeRequest(c, options);
+    if (unauthorized) return unauthorized;
+    const body = await c.req.json().catch(() => ({}));
+    try {
+      return c.json(await enqueueTaskFromSdk(options.sdk, {
+        taskId: c.req.param("id"),
+        queueName: typeof body.queueName === "string" ? body.queueName : void 0,
+        deliveryDelaySeconds: body.delaySeconds === void 0 ? void 0 : Number(body.delaySeconds),
+        actor: actor(body, defaultActor)
+      }));
+    } catch (error) {
+      const message = error instanceof Error ? error.message : String(error);
+      return jsonError(c, /Unknown task/.test(message) ? 404 : /Queue producer/.test(message) ? 501 : 500, message);
+    }
+  });
+  app.post(withPrefix(prefix, "/tasks/:id/followups"), async (c) => {
+    const unauthorized = authorizeRequest(c, options);
+    if (unauthorized) return unauthorized;
+    const body = await c.req.json().catch(() => ({}));
+    const current = await options.sdk.get({ model: "task", id: c.req.param("id") });
+    if (!current.payload) {
+      return jsonError(c, 404, "Unknown task.");
+    }
+    const currentTask = current.payload;
+    const followups = Array.isArray(body.followups) ? body.followups : [];
+    const created = [];
+    for (const followup of followups) {
+      created.push(await options.sdk.createTask({
+        workDayId: String(followup.workDayId ?? currentTask.workDayId ?? ""),
+        agentId: String(followup.agentId ?? currentTask.agentId ?? ""),
+        type: String(followup.type ?? "followup"),
+        priority: followup.priority === void 0 ? void 0 : Number(followup.priority),
+        idempotencyKey: String(followup.idempotencyKey ?? `${c.req.param("id")}:${created.length}`),
+        payload: followup.payload ?? {},
+        graphVersion: typeof followup.graphVersion === "string" ? followup.graphVersion : null,
+        parentTaskId: c.req.param("id"),
+        actor: actor(followup, defaultActor)
+      }));
+    }
+    return c.json({ ok: true, payload: created.map((entry) => entry.payload) });
+  });
+  app.post(withPrefix(prefix, "/queue/enqueue"), async (c) => {
+    const unauthorized = authorizeRequest(c, options);
+    if (unauthorized) return unauthorized;
+    const body = await c.req.json().catch(() => ({}));
+    try {
+      return c.json(await enqueueTaskFromSdk(options.sdk, {
+        taskId: String(body.taskId ?? ""),
+        queueName: typeof body.queueName === "string" ? body.queueName : void 0,
+        deliveryDelaySeconds: body.deliveryDelaySeconds === void 0 ? void 0 : Number(body.deliveryDelaySeconds),
+        actor: actor(body, defaultActor)
+      }));
+    } catch (error) {
+      const message = error instanceof Error ? error.message : String(error);
+      return jsonError(c, /Unknown task/.test(message) ? 404 : /Queue push client/.test(message) ? 501 : 500, message);
+    }
+  });
+  app.post(withPrefix(prefix, "/reports"), async (c) => {
+    const unauthorized = authorizeRequest(c, options);
+    if (unauthorized) return unauthorized;
+    const body = await c.req.json().catch(() => ({}));
+    const result = await options.sdk.createReport({
+      id: typeof body.id === "string" ? body.id : void 0,
+      workDayId: String(body.workDayId ?? ""),
+      kind: String(body.kind ?? "workday_summary"),
+      body: body.body ?? {},
+      renderedRef: typeof body.renderedRef === "string" ? body.renderedRef : null,
+      sentAt: typeof body.sentAt === "string" ? body.sentAt : null,
+      actor: actor(body, defaultActor)
+    });
+    return c.json(result);
+  });
+  app.post(withPrefix(prefix, "/context/resolve-task"), async (c) => {
+    const unauthorized = authorizeRequest(c, options);
+    if (unauthorized) return unauthorized;
+    const body = await c.req.json().catch(() => ({}));
+    return c.json({
+      ok: true,
+      payload: await buildTaskContext(options.sdk, String(body.taskId ?? ""))
+    });
+  });
+  app.post(withPrefix(prefix, "/graph/search"), async (c) => {
+    const unauthorized = authorizeRequest(c, options);
+    if (unauthorized) return unauthorized;
+    const body = await c.req.json().catch(() => ({}));
+    const query = String(body.query ?? "");
+    const scope = String(body.scope ?? "sections");
+    const payload = scope === "files" ? await options.sdk.searchFiles(query, body.options) : scope === "entities" ? await options.sdk.searchEntities(query, body.options) : await options.sdk.searchSections(query, body.options);
+    return c.json({ ok: true, payload });
+  });
+  app.post(withPrefix(prefix, "/graph/subgraph"), async (c) => {
+    const unauthorized = authorizeRequest(c, options);
+    if (unauthorized) return unauthorized;
+    const body = await c.req.json().catch(() => ({}));
+    const payload = await options.sdk.getSubgraph(
+      Array.isArray(body.seedIds) ? body.seedIds.map(String) : [],
+      body.options
+    );
+    return c.json({ ok: true, payload });
+  });
+  app.post(withPrefix(prefix, "/graph/query"), async (c) => {
+    const unauthorized = authorizeRequest(c, options);
+    if (unauthorized) return unauthorized;
+    const body = await c.req.json().catch(() => ({}));
+    const payload = await options.sdk.queryGraph(body);
+    if (typeof body.workDayId === "string" && body.workDayId) {
+      await options.sdk.create({
+        model: "graph_run",
+        data: {
+          workDayId: body.workDayId,
+          corpusHash: String(body.corpusHash ?? "query-graph"),
+          graphVersion: String(body.graphVersion ?? ""),
+          queryJson: JSON.stringify(body),
+          seedIdsJson: JSON.stringify(payload.seedIds),
+          selectedNodeIdsJson: JSON.stringify(payload.nodes.map((entry) => entry.node.id)),
+          statsJson: JSON.stringify({ nodeCount: payload.nodes.length, edgeCount: payload.edges.length })
+        },
+        actor: actor(body, defaultActor)
+      });
+    }
+    return c.json({ ok: true, payload });
+  });
+  app.post(withPrefix(prefix, "/graph/context-pack"), async (c) => {
+    const unauthorized = authorizeRequest(c, options);
+    if (unauthorized) return unauthorized;
+    const body = await c.req.json().catch(() => ({}));
+    const payload = await options.sdk.buildContextPack(body);
+    if (typeof body.workDayId === "string" && body.workDayId) {
+      await options.sdk.create({
+        model: "graph_run",
+        data: {
+          workDayId: body.workDayId,
+          corpusHash: String(body.corpusHash ?? "context-pack"),
+          graphVersion: String(body.graphVersion ?? ""),
+          queryJson: JSON.stringify(body),
+          seedIdsJson: JSON.stringify(payload.seedIds),
+          selectedNodeIdsJson: JSON.stringify(payload.includedNodeIds),
+          statsJson: JSON.stringify({
+            nodeCount: payload.nodes.length,
+            edgeCount: payload.edges.length,
+            totalTokenEstimate: payload.totalTokenEstimate
+          })
+        },
+        actor: actor(body, defaultActor)
+      });
+    }
+    return c.json({ ok: true, payload });
+  });
+  app.post(withPrefix(prefix, "/graph/parse-dsl"), async (c) => {
+    const unauthorized = authorizeRequest(c, options);
+    if (unauthorized) return unauthorized;
+    const body = await c.req.json().catch(() => ({}));
+    const payload = await options.sdk.parseGraphDsl(String(body.source ?? body.query ?? ""));
+    return c.json({ ok: true, payload });
+  });
+  app.get(withPrefix(prefix, "/graph/node/:id"), async (c) => {
+    const unauthorized = authorizeRequest(c, options);
+    if (unauthorized) return unauthorized;
+    const payload = await options.sdk.getGraphNode(c.req.param("id"));
+    return payload ? c.json({ ok: true, payload }) : jsonError(c, 404, "Unknown graph node.");
+  });
+}
+export {
+  registerAgentRoutes
+};

package/dist/api/app.d.ts

@@ -0,0 +1,8 @@
+import { Hono } from 'hono';
+import type { ApiServerOptions, AppVariables } from './types.ts';
+export declare function createTreeseedApiRouter(options?: ApiServerOptions): Hono<{
+    Variables: AppVariables;
+}, import("hono/types").BlankSchema, "/">;
+export declare function createTreeseedApiApp(options?: ApiServerOptions): Hono<{
+    Variables: AppVariables;
+}, import("hono/types").BlankSchema, "/">;