Start using pi

Author: NonlinearFruit

README.md

@@ -63,6 +63,7 @@ The actual dotfiles for various tools
 | glide |
 | nvim |
 | ollama |
+| pi |
 | termux |
 | tmux |
 | tools |
@@ -104,6 +105,7 @@ Helpful automation for various tasks
 | nvim-plugins |  |
 | nvims |  |
 | passphrase |  |
+| print-pi-session |  |
 | profile-nvim | [demo](.tapes/profile-nvim.gif) |
 | reprint |  |
 | ssh |  |

pi/save-last-extension.ts

@@ -0,0 +1,39 @@
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
+import { writeFile } from "node:fs/promises";
+import { join } from "node:path";
+
+export default function (pi: ExtensionAPI) {
+  pi.registerCommand("save-last", {
+    description: "Export the most recent AI response to a txt file",
+    handler: async (args, ctx) => {
+      ctx.ui.notify(`Export started`, "success");
+      const entries = ctx.sessionManager.getBranch();
+
+      let lastAssistantText = null;
+      for (let i = entries.length - 1; i >= 0; i--) {
+        const entry = entries[i];
+        if (entry.type === "message" && entry.message.role !== "user") {
+          const textParts = entry.message.content
+            .filter(block => block.type === "text")
+            .map(block => block.text);
+          if (textParts.length > 0) {
+            lastAssistantText = textParts.join("\n");
+            break;
+          }
+        }
+      }
+
+      if (!lastAssistantText) {
+        ctx.ui.notify("No AI response found in this session.", "warn");
+        return;
+      }
+
+      const timestamp = new Date().toISOString().replace(/[:.]/g, "-");
+      const filename = args.trim() || `ai-response-${timestamp}.txt`;
+      const filepath = join(ctx.cwd, filename);
+
+      await writeFile(filepath, lastAssistantText, "utf8");
+      ctx.ui.notify(`Exported to ${filename}`, "success");
+    },
+  });
+}

scripts/print-pi-session

@@ -0,0 +1,62 @@
+#!/usr/bin/env nu
+
+# print-pi-session | bat --language md
+def main [] {
+  select-session
+  | get data
+  | where type == message
+  | get message
+  | structure-messages
+  | each { format-message }
+  | to text
+}
+
+def select-session [] {
+  ls ~/.pi/agent/sessions/*/*.jsonl
+  | get name
+  | each {|file|
+    open
+    | lines
+    | each { from json }
+    | {
+      file: $file
+      data: $in
+    }
+  }
+  | each {|thing|
+    {
+      cwd: ($thing.data | first | get cwd | str replace "/home/bolen/projects/" "")
+      age: ($thing.data | first | get timestamp | into datetime | date humanize)
+      first-prompt: ($thing.data | where type == message | get message | where role == user | first | get content.text.0)
+      file: $thing.file
+      data: $thing.data
+    }
+  }
+  | input list --display {|thing|
+    $"[($thing.age)] ($thing.cwd): ($thing.first-prompt)"
+  }
+}
+
+def structure-messages [] {
+  each {|msg|
+    {
+      role: $msg.role
+      content: ($msg | get content | flatten | where type == text | get text | to text)
+    }
+  }
+  | where content != ""
+}
+
+def format-message [] {
+  $"(to-color $in.role)($in.role)(ansi reset): ($in.content)"
+}
+
+def to-color [role] {
+  {
+    user: (ansi green)
+    assistant: (ansi yellow)
+    toolResult: (ansi blue)
+  }
+  | get -o $role
+  | default (ansi default)
+}