tau-all-agent 0.1.0 → 0.1.1

package/README.md

@@ -43,3 +43,7 @@ Skills:
 - `sentry`
 - `update-changelog`
 - `web-design`
+
+Themes:
+
+- `tau-dark`

package/extensions/answer.ts

@@ -11,7 +11,7 @@
  */
 
 import { complete, type Model, type Api, type UserMessage } from "@mariozechner/pi-ai";
-import type { ExtensionAPI, ExtensionContext } from "@mariozechner/pi-coding-agent";
+import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
 import { BorderedLoader } from "@mariozechner/pi-coding-agent";
 import {
   type Component,
@@ -464,138 +464,131 @@ class QnAComponent implements Component {
 }
 
 export default function (pi: ExtensionAPI) {
-  const answerHandler = async (ctx: ExtensionContext) => {
-    if (!ctx.hasUI) {
-      ctx.ui.notify("answer requires interactive mode", "error");
-      return;
-    }
+  pi.registerCommand("answer", {
+    description: "Extract questions from last assistant message into interactive Q&A",
+    handler: async (_args, ctx) => {
+      if (!ctx.hasUI) {
+        ctx.ui.notify("answer requires interactive mode", "error");
+        return;
+      }
 
-    if (!ctx.model) {
-      ctx.ui.notify("No model selected", "error");
-      return;
-    }
+      if (!ctx.model) {
+        ctx.ui.notify("No model selected", "error");
+        return;
+      }
 
-    // Find the last assistant message on the current branch
-    const branch = ctx.sessionManager.getBranch();
-    let lastAssistantText: string | undefined;
-
-    for (let i = branch.length - 1; i >= 0; i--) {
-      const entry = branch[i];
-      if (entry.type === "message") {
-        const msg = entry.message;
-        if ("role" in msg && msg.role === "assistant") {
-          if (msg.stopReason !== "stop") {
-            ctx.ui.notify(`Last assistant message incomplete (${msg.stopReason})`, "error");
-            return;
-          }
-          const textParts = msg.content
-            .filter((c): c is { type: "text"; text: string } => c.type === "text")
-            .map((c) => c.text);
-          if (textParts.length > 0) {
-            lastAssistantText = textParts.join("\n");
-            break;
+      // Find the last assistant message on the current branch
+      const branch = ctx.sessionManager.getBranch();
+      let lastAssistantText: string | undefined;
+
+      for (let i = branch.length - 1; i >= 0; i--) {
+        const entry = branch[i];
+        if (entry.type === "message") {
+          const msg = entry.message;
+          if ("role" in msg && msg.role === "assistant") {
+            if (msg.stopReason !== "stop") {
+              ctx.ui.notify(`Last assistant message incomplete (${msg.stopReason})`, "error");
+              return;
+            }
+            const textParts = msg.content
+              .filter((c): c is { type: "text"; text: string } => c.type === "text")
+              .map((c) => c.text);
+            if (textParts.length > 0) {
+              lastAssistantText = textParts.join("\n");
+              break;
+            }
           }
         }
       }
-    }
 
-    if (!lastAssistantText) {
-      ctx.ui.notify("No assistant messages found", "error");
-      return;
-    }
-
-    // Select the best model for extraction (prefer a fast model in the current family).
-    const extractionSelection = await selectExtractionModel(ctx.model, ctx.modelRegistry);
-    if (!extractionSelection) {
-      ctx.ui.notify("No configured Pi model is available for question extraction", "error");
-      return;
-    }
+      if (!lastAssistantText) {
+        ctx.ui.notify("No assistant messages found", "error");
+        return;
+      }
 
-    // Run extraction with loader UI
-    const extractionResult = await ctx.ui.custom<ExtractionResult | null>(
-      (tui, theme, _kb, done) => {
-        const loader = new BorderedLoader(
-          tui,
-          theme,
-          `Extracting questions using ${extractionSelection.model.id}...`,
-        );
-        loader.onAbort = () => done(null);
-
-        const doExtract = async () => {
-          const userMessage: UserMessage = {
-            role: "user",
-            content: [{ type: "text", text: lastAssistantText! }],
-            timestamp: Date.now(),
-          };
+      // Select the best model for extraction (prefer a fast model in the current family).
+      const extractionSelection = await selectExtractionModel(ctx.model, ctx.modelRegistry);
+      if (!extractionSelection) {
+        ctx.ui.notify("No configured Pi model is available for question extraction", "error");
+        return;
+      }
 
-          const response = await complete(
-            extractionSelection.model,
-            { systemPrompt: SYSTEM_PROMPT, messages: [userMessage] },
-            {
-              apiKey: extractionSelection.apiKey,
-              headers: extractionSelection.headers,
-              signal: loader.signal,
-            },
+      // Run extraction with loader UI
+      const extractionResult = await ctx.ui.custom<ExtractionResult | null>(
+        (tui, theme, _kb, done) => {
+          const loader = new BorderedLoader(
+            tui,
+            theme,
+            `Extracting questions using ${extractionSelection.model.id}...`,
           );
+          loader.onAbort = () => done(null);
+
+          const doExtract = async () => {
+            const userMessage: UserMessage = {
+              role: "user",
+              content: [{ type: "text", text: lastAssistantText! }],
+              timestamp: Date.now(),
+            };
+
+            const response = await complete(
+              extractionSelection.model,
+              { systemPrompt: SYSTEM_PROMPT, messages: [userMessage] },
+              {
+                apiKey: extractionSelection.apiKey,
+                headers: extractionSelection.headers,
+                signal: loader.signal,
+              },
+            );
+
+            if (response.stopReason === "aborted") {
+              return null;
+            }
+
+            const responseText = response.content
+              .filter((c): c is { type: "text"; text: string } => c.type === "text")
+              .map((c) => c.text)
+              .join("\n");
+
+            return parseExtractionResult(responseText);
+          };
 
-          if (response.stopReason === "aborted") {
-            return null;
-          }
-
-          const responseText = response.content
-            .filter((c): c is { type: "text"; text: string } => c.type === "text")
-            .map((c) => c.text)
-            .join("\n");
-
-          return parseExtractionResult(responseText);
-        };
-
-        doExtract()
-          .then(done)
-          .catch(() => done(null));
-
-        return loader;
-      },
-    );
-
-    if (extractionResult === null) {
-      ctx.ui.notify("Cancelled", "info");
-      return;
-    }
+          doExtract()
+            .then(done)
+            .catch(() => done(null));
 
-    if (extractionResult.questions.length === 0) {
-      ctx.ui.notify("No questions found in the last message", "info");
-      return;
-    }
+          return loader;
+        },
+      );
 
-    // Show the Q&A component
-    const answersResult = await ctx.ui.custom<string | null>((tui, _theme, _kb, done) => {
-      return new QnAComponent(extractionResult.questions, tui, done);
-    });
+      if (extractionResult === null) {
+        ctx.ui.notify("Cancelled", "info");
+        return;
+      }
 
-    if (answersResult === null) {
-      ctx.ui.notify("Cancelled", "info");
-      return;
-    }
+      if (extractionResult.questions.length === 0) {
+        ctx.ui.notify("No questions found in the last message", "info");
+        return;
+      }
 
-    // Send the answers directly as a message and trigger a turn
-    pi.sendMessage(
-      {
-        customType: "answers",
-        content: "Here are my answers to your questions:\n\n" + answersResult,
-        display: true,
-      },
-      { triggerTurn: true },
-    );
-  };
+      // Show the Q&A component
+      const answersResult = await ctx.ui.custom<string | null>((tui, _theme, _kb, done) => {
+        return new QnAComponent(extractionResult.questions, tui, done);
+      });
 
-  pi.registerCommand("answer", {
-    description: "Extract questions from last assistant message into interactive Q&A",
-    handler: (_args, ctx) => answerHandler(ctx),
-  });
+      if (answersResult === null) {
+        ctx.ui.notify("Cancelled", "info");
+        return;
+      }
 
-  pi.registerShortcut("ctrl+.", {
-    description: "Extract and answer questions",
-    handler: answerHandler,
+      // Send the answers directly as a message and trigger a turn
+      pi.sendMessage(
+        {
+          customType: "answers",
+          content: "Here are my answers to your questions:\n\n" + answersResult,
+          display: true,
+        },
+        { triggerTurn: true },
+      );
+    },
   });
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tau-all-agent",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "Full Tau distribution for Pi",
   "keywords": [
     "pi-extension",
@@ -65,11 +65,15 @@
       "skills/web-design",
       "skills/homeassistant-ops",
       "skills/openscad"
+    ],
+    "themes": [
+      "themes/tau-dark.json"
     ]
   },
   "files": [
     "extensions",
     "skills",
+    "themes",
     "README.md",
     "LICENSE"
   ]

package/skills/oracle/scripts/oracle-bundle

@@ -130,10 +130,31 @@ done
 
 include_patterns=()
 exclude_patterns=()
+explicit_files=()
+
+add_pattern() {
+  local is_exclude="$1"
+  local pattern="$2"
+  if [[ "$is_exclude" -eq 1 ]]; then
+    exclude_patterns+=("$pattern")
+  else
+    include_patterns+=("$pattern")
+  fi
+}
+
+add_exact_file() {
+  local is_exclude="$1"
+  local file="$2"
+  if [[ "$is_exclude" -eq 1 ]]; then
+    exclude_patterns+=("$file")
+  else
+    explicit_files+=("$file")
+  fi
+}
 
 for sel in "${selectors[@]}"; do
   is_exclude=0
-  if [[ "$sel" == !* ]]; then
+  if [[ "${sel:0:1}" == "!" ]]; then
     is_exclude=1
     sel="${sel:1}"
   fi
@@ -148,69 +169,96 @@ for sel in "${selectors[@]}"; do
     if [[ "$sel" == "${ROOT_DIR}/"* ]]; then
       sel="${sel#"${ROOT_DIR}/"}"
     else
-      echo "error: absolute paths must be under repo root: ${sel}" >&2
-      exit 2
+      if [[ "$is_exclude" -eq 1 ]]; then
+        add_pattern "$is_exclude" "$sel"
+        continue
+      fi
+      if [[ ! -f "$sel" ]]; then
+        echo "error: absolute paths outside repo must name an existing file: ${sel}" >&2
+        exit 2
+      fi
+      add_exact_file "$is_exclude" "$sel"
+      continue
     fi
   fi
 
-  if [[ -e "$sel" && -d "$sel" ]] && ! has_glob_chars "$sel"; then
-    sel="${sel}/**"
+  sel="${sel#./}"
+
+  if [[ -f "$sel" ]]; then
+    add_exact_file "$is_exclude" "$sel"
+    continue
   fi
 
-  if [[ $is_exclude -eq 1 ]]; then
-    exclude_patterns+=("$sel")
-  else
-    include_patterns+=("$sel")
+  if [[ -d "$sel" ]] && ! has_glob_chars "$sel"; then
+    sel="${sel}/**"
   fi
+
+  add_pattern "$is_exclude" "$sel"
 done
 
-if [[ "${#include_patterns[@]}" -eq 0 ]]; then
+if [[ "${#include_patterns[@]}" -eq 0 && "${#explicit_files[@]}" -eq 0 ]]; then
   echo "error: no include patterns provided (need at least one non-! --file)" >&2
   exit 2
 fi
 
 all_files=()
-if git rev-parse --is-inside-work-tree >/dev/null 2>&1; then
-  while IFS= read -r f; do
-    [[ -n "$f" ]] && all_files+=("$f")
-  done < <(git ls-files -co --exclude-standard)
-else
-  while IFS= read -r f; do
-    f="${f#./}"
-    [[ -n "$f" ]] && all_files+=("$f")
-  done < <(find . -type f -print)
+if [[ "${#include_patterns[@]}" -gt 0 ]]; then
+  if git rev-parse --is-inside-work-tree >/dev/null 2>&1; then
+    while IFS= read -r f; do
+      [[ -n "$f" ]] && all_files+=("$f")
+    done < <(git ls-files -co --exclude-standard)
+  else
+    while IFS= read -r f; do
+      f="${f#./}"
+      [[ -n "$f" ]] && all_files+=("$f")
+    done < <(find . -type f -print)
+  fi
 fi
 
 selected=()
 skipped=()
 
-for rel in "${all_files[@]}"; do
-  [[ -z "$rel" ]] && continue
-  path_has_ignored_dir "$rel" && continue
+select_file() {
+  local file="$1"
+  local size=""
 
-  if ! matches_any_pattern "$rel" "${include_patterns[@]}"; then
-    continue
-  fi
+  [[ -z "$file" ]] && return 0
 
-  if [[ "${#exclude_patterns[@]}" -gt 0 ]] && matches_any_pattern "$rel" "${exclude_patterns[@]}"; then
-    continue
+  if [[ "${#exclude_patterns[@]}" -gt 0 ]] && matches_any_pattern "$file" "${exclude_patterns[@]}"; then
+    return 0
   fi
 
-  if [[ ! -f "$rel" ]]; then
-    continue
+  if [[ ! -f "$file" ]]; then
+    return 0
   fi
 
-  size="$(file_size_bytes "$rel")"
+  size="$(file_size_bytes "$file")"
   if [[ "$size" -gt "$MAX_BYTES" ]]; then
-    skipped+=("${rel} (too large: ${size} bytes)")
+    skipped+=("${file} (too large: ${size} bytes)")
+    return 0
+  fi
+
+  selected+=("$file")
+}
+
+for file in "${explicit_files[@]}"; do
+  select_file "$file"
+done
+
+for rel in "${all_files[@]}"; do
+  [[ -z "$rel" ]] && continue
+
+  path_has_ignored_dir "$rel" && continue
+
+  if ! matches_any_pattern "$rel" "${include_patterns[@]}"; then
     continue
   fi
 
-  selected+=("$rel")
+  select_file "$rel"
 done
 
 if [[ "$DRY_RUN" -eq 1 ]]; then
-  printf '%s\n' "${selected[@]}" | LC_ALL=C sort
+  printf '%s\n' "${selected[@]}" | LC_ALL=C sort -u
   if [[ "${#skipped[@]}" -gt 0 ]]; then
     printf '\nSkipped:\n' >&2
     printf '%s\n' "${skipped[@]}" >&2
@@ -219,7 +267,7 @@ if [[ "$DRY_RUN" -eq 1 ]]; then
 fi
 
 printf '# Task\n\n%s\n\n# Files\n' "$(trim "$PROMPT")"
-printf '%s\n' "${selected[@]}" | LC_ALL=C sort | while IFS= read -r rel; do
+printf '%s\n' "${selected[@]}" | LC_ALL=C sort -u | while IFS= read -r rel; do
   [[ -z "$rel" ]] && continue
   printf '\n## %s\n\n```\n' "$rel"
   cat "$rel"

package/themes/tau-dark.json

@@ -0,0 +1,85 @@
+{
+  "$schema": "https://raw.githubusercontent.com/badlogic/pi-mono/main/packages/coding-agent/src/modes/interactive/theme/theme-schema.json",
+  "name": "tau-dark",
+  "vars": {
+    "accent": "#8abeb7",
+    "blue": "#5f87af",
+    "green": "#b5bd68",
+    "red": "#cc6666",
+    "yellow": "#f0c674",
+    "gray": "#808080",
+    "dimGray": "#666666",
+    "darkGray": "#505050",
+    "heading": "#DCDCAA",
+    "selectedBg": "#3a3a4a",
+    "userMsgBg": "#343541",
+    "toolPendingBg": "#282832",
+    "toolSuccessBg": "#283228",
+    "toolErrorBg": "#3c2828",
+    "customMsgBg": "#2d2838"
+  },
+  "colors": {
+    "accent": "accent",
+    "border": "blue",
+    "borderAccent": "accent",
+    "borderMuted": "darkGray",
+    "success": "green",
+    "error": "red",
+    "warning": "yellow",
+    "muted": "gray",
+    "dim": "dimGray",
+    "text": "",
+    "thinkingText": "gray",
+
+    "selectedBg": "selectedBg",
+    "userMessageBg": "userMsgBg",
+    "userMessageText": "",
+    "customMessageBg": "customMsgBg",
+    "customMessageText": "",
+    "customMessageLabel": "#9575cd",
+    "toolPendingBg": "toolPendingBg",
+    "toolSuccessBg": "toolSuccessBg",
+    "toolErrorBg": "toolErrorBg",
+    "toolTitle": "",
+    "toolOutput": "gray",
+
+    "mdHeading": "heading",
+    "mdLink": "#81a2be",
+    "mdLinkUrl": "dimGray",
+    "mdCode": "accent",
+    "mdCodeBlock": "green",
+    "mdCodeBlockBorder": "gray",
+    "mdQuote": "gray",
+    "mdQuoteBorder": "gray",
+    "mdHr": "gray",
+    "mdListBullet": "accent",
+
+    "toolDiffAdded": "green",
+    "toolDiffRemoved": "red",
+    "toolDiffContext": "gray",
+
+    "syntaxComment": "#6A9955",
+    "syntaxKeyword": "#569CD6",
+    "syntaxFunction": "#DCDCAA",
+    "syntaxVariable": "#9CDCFE",
+    "syntaxString": "#CE9178",
+    "syntaxNumber": "#B5CEA8",
+    "syntaxType": "#4EC9B0",
+    "syntaxOperator": "#D4D4D4",
+    "syntaxPunctuation": "#D4D4D4",
+
+    "thinkingOff": "darkGray",
+    "thinkingMinimal": "dimGray",
+    "thinkingLow": "#657B83",
+    "thinkingMedium": "#6f9bb3",
+    "thinkingHigh": "#78a8a1",
+    "thinkingXhigh": "accent",
+
+    "bashMode": "green"
+  },
+  "export": {
+    "pageBg": "#18181e",
+    "cardBg": "#1e1e24",
+    "infoBg": "#3c3728"
+  }
+}