@junghanacs/entwurf 0.30.1 → 0.32.0

package/pi/pi-durable/carrier/subagent.js

@@ -0,0 +1,55 @@
+                                                     
+import {                        Type } from "@earendil-works/pi-ai";
+import {
+	AssistantEntry,
+	configure,
+	defineExtension,
+	defineTool,
+	             
+	               
+	                      
+} from "@earendil-works/pi-durable";
+
+async function answerText(api                  , answer         , context         )                  {
+	const entry = await api.commit((tx) => tx.entry(AssistantEntry, answer), context);
+	const message = entry?.model?.[0]                                ;
+	return message?.content.flatMap((content) => (content.type === "text" ? [content.text] : [])).join("") ?? "";
+}
+
+/**
+ * A foreground subagent: each call creates a child conversation owned by the call's task, runs the task there, and
+ * returns the child's answer. Aborting the call aborts the child. The child outlives the call, so the user can switch
+ * to it and keep talking.
+ */
+export const Subagent            = defineExtension({
+	name: "subagent",
+	tools: [
+		defineTool({
+			name: "subagent",
+			description:
+				"Delegate a self-contained task to a subagent with the same tools and get its answer back. Give it everything it needs to know; it does not see this conversation.",
+			parameters: Type.Object({ task: Type.String({ description: "What the subagent should do" }) }),
+			// A rerun after a crash finds the child it created and the submission it made.
+			replay: "safe",
+			execute: async (args, api, context) => {
+				const child = await api.commit(async (tx) => {
+					const existing = (await tx.scanConversations({ ownerTaskId: api.taskId }, 1)).items[0];
+					if (existing !== undefined) return existing.id;
+					// Starts as a copy of this conversation's agent; without this extension it cannot delegate further.
+					const created = await tx.createConversation({ ownership: { kind: "task", taskId: api.taskId } });
+					await configure(tx, created.id, { extensions: { remove: [Subagent] } });
+					return created.id;
+				}, context);
+				await api.details({ conversationId: child }, context);
+				const handle = (await api.conversation(child, context)) ;
+				const request = { type: "input", content: args.task, requestId: `subagent:${api.taskId}` }         ;
+				const settled = await (await handle.submit(request, context)).wait(context);
+				if (settled.status !== "done" || settled.type !== "input") {
+					throw new Error(`Subagent ${child} failed: ${settled.status}`);
+				}
+				const text = await answerText(api, settled.answer, context);
+				return { content: [{ type: "text", text }], details: { conversationId: child } };
+			},
+		}),
+	],
+});