@bastani/atomic 0.5.0-3 → 0.5.0-5

package/.atomic/workflows/hello/claude/index.ts

@@ -14,31 +14,28 @@ export default defineWorkflow({
   name: "hello",
   description: "Two-session Claude demo: describe → summarize",
 })
-  .session({
-    name: "describe",
-    description: "Ask Claude to describe the project",
-    run: async (ctx) => {
-      await createClaudeSession({ paneId: ctx.paneId });
-      await claudeQuery({
-        paneId: ctx.paneId,
-        prompt: ctx.userPrompt,
-      });
-      // Save transcript via Claude Agent SDK (reads from ~/.claude session files)
-      ctx.save(ctx.sessionId);
-    },
-  })
-  .session({
-    name: "summarize",
-    description: "Summarize the previous session's output",
-    run: async (ctx) => {
-      await createClaudeSession({ paneId: ctx.paneId });
-      const research = await ctx.transcript("describe");
+  .run(async (ctx) => {
+    const describe = await ctx.session(
+      { name: "describe", description: "Ask Claude to describe the project" },
+      async (s) => {
+        await createClaudeSession({ paneId: s.paneId });
+        await claudeQuery({ paneId: s.paneId, prompt: s.userPrompt });
+        s.save(s.sessionId);
+      },
+    );
+
+    await ctx.session(
+      { name: "summarize", description: "Summarize the previous session's output" },
+      async (s) => {
+        await createClaudeSession({ paneId: s.paneId });
+        const research = await s.transcript(describe);
 
-      await claudeQuery({
-        paneId: ctx.paneId,
-        prompt: `Read ${research.path} and summarize it in 2-3 bullet points.`,
-      });
-      ctx.save(ctx.sessionId);
-    },
+        await claudeQuery({
+          paneId: s.paneId,
+          prompt: `Read ${research.path} and summarize it in 2-3 bullet points.`,
+        });
+        s.save(s.sessionId);
+      },
+    );
   })
   .compile();

package/.atomic/workflows/hello/copilot/index.ts

@@ -10,49 +10,59 @@
 import { defineWorkflow } from "@bastani/atomic/workflows";
 import { CopilotClient, approveAll } from "@github/copilot-sdk";
 
+/**
+ * `CopilotSession.sendAndWait` defaults to a 60s timeout and THROWS on
+ * expiry, which crashes the workflow mid-stage. Override with a generous
+ * 30-minute budget so legitimate long-running agent work completes.
+ */
+const SEND_TIMEOUT_MS = 30 * 60 * 1000;
+
 export default defineWorkflow({
   name: "hello",
   description: "Two-session Copilot demo: describe → summarize",
 })
-  .session({
-    name: "describe",
-    description: "Ask the agent to describe the project",
-    run: async (ctx) => {
-      const client = new CopilotClient({ cliUrl: ctx.serverUrl });
-      await client.start();
+  .run(async (ctx) => {
+    const describe = await ctx.session(
+      { name: "describe", description: "Ask the agent to describe the project" },
+      async (s) => {
+        const client = new CopilotClient({ cliUrl: s.serverUrl });
+        await client.start();
 
-      const session = await client.createSession({ onPermissionRequest: approveAll });
-      await client.setForegroundSessionId(session.sessionId);
+        const session = await client.createSession({ onPermissionRequest: approveAll });
+        await client.setForegroundSessionId(session.sessionId);
 
-      await session.sendAndWait({ prompt: ctx.userPrompt });
+        await session.sendAndWait({ prompt: s.userPrompt }, SEND_TIMEOUT_MS);
 
-      // Save Copilot messages for the next session
-      ctx.save(await session.getMessages() );
+        s.save(await session.getMessages());
 
-      await session.disconnect();
-      await client.stop();
-    },
-  })
-  .session({
-    name: "summarize",
-    description: "Summarize the previous session's output",
-    run: async (ctx) => {
-      const research = await ctx.transcript("describe");
+        await session.disconnect();
+        await client.stop();
+      },
+    );
+
+    await ctx.session(
+      { name: "summarize", description: "Summarize the previous session's output" },
+      async (s) => {
+        const research = await s.transcript(describe);
 
-      const client = new CopilotClient({ cliUrl: ctx.serverUrl });
-      await client.start();
+        const client = new CopilotClient({ cliUrl: s.serverUrl });
+        await client.start();
 
-      const session = await client.createSession({ onPermissionRequest: approveAll });
-      await client.setForegroundSessionId(session.sessionId);
+        const session = await client.createSession({ onPermissionRequest: approveAll });
+        await client.setForegroundSessionId(session.sessionId);
 
-      await session.sendAndWait({
-        prompt: `Summarize the following in 2-3 bullet points:\n\n${research.content}`,
-      });
+        await session.sendAndWait(
+          {
+            prompt: `Summarize the following in 2-3 bullet points:\n\n${research.content}`,
+          },
+          SEND_TIMEOUT_MS,
+        );
 
-      ctx.save(await session.getMessages());
+        s.save(await session.getMessages());
 
-      await session.disconnect();
-      await client.stop();
-    },
+        await session.disconnect();
+        await client.stop();
+      },
+    );
   })
   .compile();

package/.atomic/workflows/hello/opencode/index.ts

@@ -14,45 +14,45 @@ export default defineWorkflow({
   name: "hello",
   description: "Two-session OpenCode demo: describe → summarize",
 })
-  .session({
-    name: "describe",
-    description: "Ask the agent to describe the project",
-    run: async (ctx) => {
-      const client = createOpencodeClient({ baseUrl: ctx.serverUrl });
-
-      const session = await client.session.create({ title: "describe" });
-      await client.tui.selectSession({ sessionID: session.data!.id });
-
-      const result = await client.session.prompt({
-        sessionID: session.data!.id,
-        parts: [{ type: "text", text: ctx.userPrompt }],
-      });
-
-      // Save OpenCode response parts for the next session
-      ctx.save(result.data!);
-    },
-  })
-  .session({
-    name: "summarize",
-    description: "Summarize the previous session's output",
-    run: async (ctx) => {
-      const research = await ctx.transcript("describe");
-      const client = createOpencodeClient({ baseUrl: ctx.serverUrl });
-
-      const session = await client.session.create({ title: "summarize" });
-      await client.tui.selectSession({ sessionID: session.data!.id });
-
-      const result = await client.session.prompt({
-        sessionID: session.data!.id,
-        parts: [
-          {
-            type: "text",
-            text: `Summarize the following in 2-3 bullet points:\n\n${research.content}`,
-          },
-        ],
-      });
-
-      ctx.save(result.data!);
-    },
+  .run(async (ctx) => {
+    const describe = await ctx.session(
+      { name: "describe", description: "Ask the agent to describe the project" },
+      async (s) => {
+        const client = createOpencodeClient({ baseUrl: s.serverUrl });
+
+        const session = await client.session.create({ title: "describe" });
+        await client.tui.selectSession({ sessionID: session.data!.id });
+
+        const result = await client.session.prompt({
+          sessionID: session.data!.id,
+          parts: [{ type: "text", text: s.userPrompt }],
+        });
+
+        s.save(result.data!);
+      },
+    );
+
+    await ctx.session(
+      { name: "summarize", description: "Summarize the previous session's output" },
+      async (s) => {
+        const research = await s.transcript(describe);
+        const client = createOpencodeClient({ baseUrl: s.serverUrl });
+
+        const session = await client.session.create({ title: "summarize" });
+        await client.tui.selectSession({ sessionID: session.data!.id });
+
+        const result = await client.session.prompt({
+          sessionID: session.data!.id,
+          parts: [
+            {
+              type: "text",
+              text: `Summarize the following in 2-3 bullet points:\n\n${research.content}`,
+            },
+          ],
+        });
+
+        s.save(result.data!);
+      },
+    );
   })
   .compile();

package/.atomic/workflows/hello-parallel/claude/index.ts

@@ -14,63 +14,63 @@ export default defineWorkflow({
   name: "hello-parallel",
   description: "Parallel Claude demo: describe → [summarize-a, summarize-b] → merge",
 })
-  .session({
-    name: "describe",
-    description: "Ask Claude to describe the project",
-    run: async (ctx) => {
-      await createClaudeSession({ paneId: ctx.paneId });
-      await claudeQuery({ paneId: ctx.paneId, prompt: ctx.userPrompt });
-      ctx.save(ctx.sessionId);
-    },
-  })
-  .session([
-    {
-      name: "summarize-a",
-      description: "Summarize the description as bullet points",
-      run: async (ctx) => {
-        const research = await ctx.transcript("describe");
-        await createClaudeSession({ paneId: ctx.paneId });
-        await claudeQuery({
-          paneId: ctx.paneId,
-          prompt: `Read ${research.path} and summarize it in 2-3 bullet points.`,
-        });
-        ctx.save(ctx.sessionId);
+  .run(async (ctx) => {
+    const describe = await ctx.session(
+      { name: "describe", description: "Ask Claude to describe the project" },
+      async (s) => {
+        await createClaudeSession({ paneId: s.paneId });
+        await claudeQuery({ paneId: s.paneId, prompt: s.userPrompt });
+        s.save(s.sessionId);
       },
-    },
-    {
-      name: "summarize-b",
-      description: "Summarize the description as a one-liner",
-      run: async (ctx) => {
-        const research = await ctx.transcript("describe");
-        await createClaudeSession({ paneId: ctx.paneId });
+    );
+
+    const [summarizeA, summarizeB] = await Promise.all([
+      ctx.session(
+        { name: "summarize-a", description: "Summarize the description as bullet points" },
+        async (s) => {
+          const research = await s.transcript(describe);
+          await createClaudeSession({ paneId: s.paneId });
+          await claudeQuery({
+            paneId: s.paneId,
+            prompt: `Read ${research.path} and summarize it in 2-3 bullet points.`,
+          });
+          s.save(s.sessionId);
+        },
+      ),
+      ctx.session(
+        { name: "summarize-b", description: "Summarize the description as a one-liner" },
+        async (s) => {
+          const research = await s.transcript(describe);
+          await createClaudeSession({ paneId: s.paneId });
+          await claudeQuery({
+            paneId: s.paneId,
+            prompt: `Read ${research.path} and summarize it in a single sentence.`,
+          });
+          s.save(s.sessionId);
+        },
+      ),
+    ]);
+
+    await ctx.session(
+      { name: "merge", description: "Merge both summaries into a final output" },
+      async (s) => {
+        const bullets = await s.transcript(summarizeA);
+        const oneliner = await s.transcript(summarizeB);
+        await createClaudeSession({ paneId: s.paneId });
         await claudeQuery({
-          paneId: ctx.paneId,
-          prompt: `Read ${research.path} and summarize it in a single sentence.`,
+          paneId: s.paneId,
+          prompt: [
+            "Combine the following two summaries into one concise paragraph:",
+            "",
+            "## Bullet points",
+            bullets.content,
+            "",
+            "## One-liner",
+            oneliner.content,
+          ].join("\n"),
         });
-        ctx.save(ctx.sessionId);
+        s.save(s.sessionId);
       },
-    },
-  ])
-  .session({
-    name: "merge",
-    description: "Merge both summaries into a final output",
-    run: async (ctx) => {
-      const bullets = await ctx.transcript("summarize-a");
-      const oneliner = await ctx.transcript("summarize-b");
-      await createClaudeSession({ paneId: ctx.paneId });
-      await claudeQuery({
-        paneId: ctx.paneId,
-        prompt: [
-          "Combine the following two summaries into one concise paragraph:",
-          "",
-          "## Bullet points",
-          bullets.content,
-          "",
-          "## One-liner",
-          oneliner.content,
-        ].join("\n"),
-      });
-      ctx.save(ctx.sessionId);
-    },
+    );
   })
   .compile();

package/.atomic/workflows/hello-parallel/copilot/index.ts

@@ -11,95 +11,114 @@
 import { defineWorkflow } from "@bastani/atomic/workflows";
 import { CopilotClient, approveAll } from "@github/copilot-sdk";
 
+/**
+ * `CopilotSession.sendAndWait` defaults to a 60s timeout and THROWS on
+ * expiry, which crashes the workflow mid-stage. Override with a generous
+ * 30-minute budget so legitimate long-running agent work completes.
+ */
+const SEND_TIMEOUT_MS = 30 * 60 * 1000;
+
 export default defineWorkflow({
   name: "hello-parallel",
   description: "Parallel Copilot demo: describe → [summarize-a, summarize-b] → merge",
 })
-  .session({
-    name: "describe",
-    description: "Ask the agent to describe the project",
-    run: async (ctx) => {
-      const client = new CopilotClient({ cliUrl: ctx.serverUrl });
-      await client.start();
-
-      const session = await client.createSession({ onPermissionRequest: approveAll });
-      await client.setForegroundSessionId(session.sessionId);
-      await session.sendAndWait({ prompt: ctx.userPrompt });
-
-      ctx.save(await session.getMessages());
-      await session.disconnect();
-      await client.stop();
-    },
-  })
-  .session([
-    {
-      name: "summarize-a",
-      description: "Summarize the description as bullet points",
-      run: async (ctx) => {
-        const research = await ctx.transcript("describe");
-
-        const client = new CopilotClient({ cliUrl: ctx.serverUrl });
+  .run(async (ctx) => {
+    // Sequential: describe
+    const describe = await ctx.session(
+      { name: "describe", description: "Ask the agent to describe the project" },
+      async (s) => {
+        const client = new CopilotClient({ cliUrl: s.serverUrl });
         await client.start();
 
         const session = await client.createSession({ onPermissionRequest: approveAll });
         await client.setForegroundSessionId(session.sessionId);
-        await session.sendAndWait({
-          prompt: `Summarize the following in 2-3 bullet points:\n\n${research.content}`,
-        });
+        await session.sendAndWait({ prompt: s.userPrompt }, SEND_TIMEOUT_MS);
 
-        ctx.save(await session.getMessages());
+        s.save(await session.getMessages());
         await session.disconnect();
         await client.stop();
       },
-    },
-    {
-      name: "summarize-b",
-      description: "Summarize the description as a one-liner",
-      run: async (ctx) => {
-        const research = await ctx.transcript("describe");
-
-        const client = new CopilotClient({ cliUrl: ctx.serverUrl });
+    );
+
+    // Parallel: summarize-a + summarize-b
+    const [summarizeA, summarizeB] = await Promise.all([
+      ctx.session(
+        { name: "summarize-a", description: "Summarize the description as bullet points" },
+        async (s) => {
+          const research = await s.transcript(describe);
+
+          const client = new CopilotClient({ cliUrl: s.serverUrl });
+          await client.start();
+
+          const session = await client.createSession({ onPermissionRequest: approveAll });
+          await client.setForegroundSessionId(session.sessionId);
+          await session.sendAndWait(
+            {
+              prompt: `Summarize the following in 2-3 bullet points:\n\n${research.content}`,
+            },
+            SEND_TIMEOUT_MS,
+          );
+
+          s.save(await session.getMessages());
+          await session.disconnect();
+          await client.stop();
+        },
+      ),
+      ctx.session(
+        { name: "summarize-b", description: "Summarize the description as a one-liner" },
+        async (s) => {
+          const research = await s.transcript(describe);
+
+          const client = new CopilotClient({ cliUrl: s.serverUrl });
+          await client.start();
+
+          const session = await client.createSession({ onPermissionRequest: approveAll });
+          await client.setForegroundSessionId(session.sessionId);
+          await session.sendAndWait(
+            {
+              prompt: `Summarize the following in a single sentence:\n\n${research.content}`,
+            },
+            SEND_TIMEOUT_MS,
+          );
+
+          s.save(await session.getMessages());
+          await session.disconnect();
+          await client.stop();
+        },
+      ),
+    ]);
+
+    // Sequential: merge
+    await ctx.session(
+      { name: "merge", description: "Merge both summaries into a final output" },
+      async (s) => {
+        const bullets = await s.transcript(summarizeA);
+        const oneliner = await s.transcript(summarizeB);
+
+        const client = new CopilotClient({ cliUrl: s.serverUrl });
         await client.start();
 
         const session = await client.createSession({ onPermissionRequest: approveAll });
         await client.setForegroundSessionId(session.sessionId);
-        await session.sendAndWait({
-          prompt: `Summarize the following in a single sentence:\n\n${research.content}`,
-        });
+        await session.sendAndWait(
+          {
+            prompt: [
+              "Combine the following two summaries into one concise paragraph:",
+              "",
+              "## Bullet points",
+              bullets.content,
+              "",
+              "## One-liner",
+              oneliner.content,
+            ].join("\n"),
+          },
+          SEND_TIMEOUT_MS,
+        );
 
-        ctx.save(await session.getMessages());
+        s.save(await session.getMessages());
         await session.disconnect();
         await client.stop();
       },
-    },
-  ])
-  .session({
-    name: "merge",
-    description: "Merge both summaries into a final output",
-    run: async (ctx) => {
-      const bullets = await ctx.transcript("summarize-a");
-      const oneliner = await ctx.transcript("summarize-b");
-
-      const client = new CopilotClient({ cliUrl: ctx.serverUrl });
-      await client.start();
-
-      const session = await client.createSession({ onPermissionRequest: approveAll });
-      await client.setForegroundSessionId(session.sessionId);
-      await session.sendAndWait({
-        prompt: [
-          "Combine the following two summaries into one concise paragraph:",
-          "",
-          "## Bullet points",
-          bullets.content,
-          "",
-          "## One-liner",
-          oneliner.content,
-        ].join("\n"),
-      });
-
-      ctx.save(await session.getMessages());
-      await session.disconnect();
-      await client.stop();
-    },
+    );
   })
   .compile();