pi-goosedump 0.9.4 → 0.9.6

package/index.ts

@@ -373,6 +373,43 @@ function resolveTarget(
   return null;
 }
 
+const GOOSE_PROVIDERS = ['pi', 'claude', 'codex', 'crush', 'gemini', 'goose', 'opencode'] as const;
+
+type HistoryOperation = 'search' | 'read' | 'grep';
+
+const HISTORY_TARGET_GUIDELINE =
+  'Use the current Pi session by default; non-Pi providers require an explicit session ID.';
+
+function historyProviderParam(operation: HistoryOperation) {
+  return Type.Optional(
+    Type.Union(
+      GOOSE_PROVIDERS.map((provider) => Type.Literal(provider)),
+      {
+        default: 'pi',
+        description: `Source provider whose sessions to ${operation}. Defaults to pi. Non-pi providers have no current session, so they require an explicit session id.`,
+      },
+    ),
+  );
+}
+
+function historySessionParam() {
+  return Type.Optional(
+    Type.String({
+      description:
+        'Context/session ID. Defaults to the current Pi session for pi; required when provider is not pi.',
+    }),
+  );
+}
+
+function historyScopeParam() {
+  return Type.Optional(
+    Type.Union([Type.Literal('lineage'), Type.Literal('all')], {
+      default: 'lineage',
+      description: 'Scope: lineage (current branch) or all entries in the session',
+    }),
+  );
+}
+
 async function goosedumpCompact(
   target: string,
   options: { scope?: string; from?: string; until: string },
@@ -1129,46 +1166,18 @@ export function createGoosedumpIntegration(
           label: 'goose_search',
           description:
             'Search coding agent session history by ranked relevance. Returns compact entry overviews with entry IDs; use goose_get to expand specific IDs to full content. Defaults to the current Pi session.',
-          promptSnippet:
-            'goose_search({ query, provider?, session?, scope?, page? }) - rank messages by query relevance; defaults to current Pi session',
+          promptSnippet: 'Search session history by ranked relevance',
           promptGuidelines: [
             "Use goose_search to find relevant context in this or another coding agent's session history.",
             'Start with compact ranked results, then use goose_get to expand interesting entry IDs to full content.',
             'Default scope is "lineage" (current branch); use scope: "all" to include all entries in the session.',
-            'Omit session to search the current Pi session; set provider (e.g. "claude", "gemini") with a session id to browse another tool\'s sessions.',
+            HISTORY_TARGET_GUIDELINE,
           ],
           parameters: Type.Object({
             query: Type.String({ description: 'Search query to rank messages by relevance' }),
-            provider: Type.Optional(
-              Type.Union(
-                [
-                  Type.Literal('pi'),
-                  Type.Literal('claude'),
-                  Type.Literal('codex'),
-                  Type.Literal('crush'),
-                  Type.Literal('gemini'),
-                  Type.Literal('goose'),
-                  Type.Literal('opencode'),
-                ],
-                {
-                  default: 'pi',
-                  description:
-                    'Source provider whose sessions to search. Defaults to pi. Non-pi providers have no current session, so they require an explicit session id.',
-                },
-              ),
-            ),
-            session: Type.Optional(
-              Type.String({
-                description:
-                  'Context/session ID. Defaults to the current Pi session for pi; required when provider is not pi.',
-              }),
-            ),
-            scope: Type.Optional(
-              Type.Union([Type.Literal('lineage'), Type.Literal('all')], {
-                default: 'lineage',
-                description: 'Scope: lineage (current branch) or all entries in the session',
-              }),
-            ),
+            provider: historyProviderParam('search'),
+            session: historySessionParam(),
+            scope: historyScopeParam(),
             page: Type.Optional(
               Type.Integer({
                 minimum: 1,
@@ -1212,47 +1221,21 @@ export function createGoosedumpIntegration(
           label: 'goose_get',
           description:
             'Expand specific entry IDs (from goose_search or goose_grep) to their full content. Defaults to the current Pi session.',
-          promptSnippet:
-            'goose_get({ ids, provider?, session? }) - expand entry IDs to full content; defaults to current Pi session',
+          promptSnippet: 'Expand session entries by ID to full content',
           promptGuidelines: [
             'Use goose_get to expand entry IDs returned by goose_search or goose_grep into their full content.',
             'Pass only the specific IDs you need; full transcripts are expensive.',
+            'Default scope is "lineage" (current branch); use scope: "all" to retrieve entries from all session branches.',
+            HISTORY_TARGET_GUIDELINE,
           ],
           parameters: Type.Object({
             ids: Type.Array(Type.String(), {
               minItems: 1,
               description: 'Entry IDs to expand (show full content)',
             }),
-            provider: Type.Optional(
-              Type.Union(
-                [
-                  Type.Literal('pi'),
-                  Type.Literal('claude'),
-                  Type.Literal('codex'),
-                  Type.Literal('crush'),
-                  Type.Literal('gemini'),
-                  Type.Literal('goose'),
-                  Type.Literal('opencode'),
-                ],
-                {
-                  default: 'pi',
-                  description:
-                    'Source provider whose sessions to read. Defaults to pi. Non-pi providers have no current session, so they require an explicit session id.',
-                },
-              ),
-            ),
-            session: Type.Optional(
-              Type.String({
-                description:
-                  'Context/session ID. Defaults to the current Pi session for pi; required when provider is not pi.',
-              }),
-            ),
-            scope: Type.Optional(
-              Type.Union([Type.Literal('lineage'), Type.Literal('all')], {
-                default: 'lineage',
-                description: 'Scope: lineage (current branch) or all entries in the session',
-              }),
-            ),
+            provider: historyProviderParam('read'),
+            session: historySessionParam(),
+            scope: historyScopeParam(),
           }),
           async execute(_id, params, _signal, _onUpdate, ctx): Promise<AgentToolResult<void>> {
             if (!goosedumpAvailable) return goosedumpUnavailableResult();
@@ -1285,45 +1268,18 @@ export function createGoosedumpIntegration(
           label: 'goose_grep',
           description:
             'Filter session messages by glob pattern (e.g. *rand*). Returns compact entry overviews with entry IDs; use goose_get to expand. Defaults to the current Pi session.',
-          promptSnippet:
-            'goose_grep({ pattern, provider?, session?, scope? }) - filter messages by glob; defaults to current Pi session',
+          promptSnippet: 'Filter session history by glob pattern',
           promptGuidelines: [
             'Use goose_grep to filter session messages by glob pattern when you need exact substring/glob matching instead of ranked relevance.',
             'Expand interesting entry IDs with goose_get.',
             'Default scope is "lineage" (current branch); use scope: "all" to include all entries in the session.',
+            HISTORY_TARGET_GUIDELINE,
           ],
           parameters: Type.Object({
             pattern: Type.String({ description: 'Glob pattern to filter messages (e.g. *rand*)' }),
-            provider: Type.Optional(
-              Type.Union(
-                [
-                  Type.Literal('pi'),
-                  Type.Literal('claude'),
-                  Type.Literal('codex'),
-                  Type.Literal('crush'),
-                  Type.Literal('gemini'),
-                  Type.Literal('goose'),
-                  Type.Literal('opencode'),
-                ],
-                {
-                  default: 'pi',
-                  description:
-                    'Source provider whose sessions to grep. Defaults to pi. Non-pi providers have no current session, so they require an explicit session id.',
-                },
-              ),
-            ),
-            session: Type.Optional(
-              Type.String({
-                description:
-                  'Context/session ID. Defaults to the current Pi session for pi; required when provider is not pi.',
-              }),
-            ),
-            scope: Type.Optional(
-              Type.Union([Type.Literal('lineage'), Type.Literal('all')], {
-                default: 'lineage',
-                description: 'Scope: lineage (current branch) or all entries in the session',
-              }),
-            ),
+            provider: historyProviderParam('grep'),
+            session: historySessionParam(),
+            scope: historyScopeParam(),
           }),
           async execute(_id, params, _signal, _onUpdate, ctx): Promise<AgentToolResult<void>> {
             if (!goosedumpAvailable) return goosedumpUnavailableResult();
@@ -1356,8 +1312,7 @@ export function createGoosedumpIntegration(
           label: 'goose_compact',
           description:
             "Trigger a goosedump compaction of the current session now, optionally keeping the last N user turns verbatim while summarizing everything before. Compaction runs asynchronously via Pi's compaction flow.",
-          promptSnippet:
-            "goose_compact({ keep? }) - compact now; keep last N user turns (turn-based), omit to compact up to Pi's cut",
+          promptSnippet: 'Compact session history; optionally keep recent user turns',
           promptGuidelines: [
             'Use goose_compact to compact the session proactively with goosedump when context is large or nearing limits.',
             "Pass keep:N (N >= 1) to keep the last N user turns verbatim and summarize everything before them; omit to compact up to Pi's token-based cut.",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-goosedump",
-  "version": "0.9.4",
+  "version": "0.9.6",
   "description": "Pi extension for goosedump-backed session-history search, entry expansion, and compaction",
   "keywords": [
     "compaction",
@@ -21,17 +21,19 @@
   ],
   "type": "module",
   "scripts": {
-    "fmt": "oxfmt index.ts package.json tsconfig.json .oxfmtrc.json README.md",
-    "lint": "oxlint index.ts",
+    "fmt": "oxfmt index.ts tests package.json tsconfig.json .oxfmtrc.json README.md",
+    "lint": "oxlint index.ts tests",
     "check": "tsc --noEmit",
-    "all": "npm run fmt && npm run lint && npm run check",
-    "ci:fmt": "oxfmt --check index.ts package.json tsconfig.json .oxfmtrc.json README.md",
+    "test": "vitest run",
+    "all": "npm run fmt && npm run lint && npm run check && npm test",
+    "ci:fmt": "oxfmt --check index.ts tests package.json tsconfig.json .oxfmtrc.json README.md",
     "ci:lint": "npm run lint",
-    "ci:check": "npm run check"
+    "ci:check": "npm run check",
+    "ci:test": "npm test"
   },
   "dependencies": {
     "@earendil-works/pi-tui": "^0.78.0",
-    "@jarkkojs/goosedump": "^0.9.4",
+    "@jarkkojs/goosedump": "^0.9.6",
     "@sinclair/typebox": "^0.34.49"
   },
   "devDependencies": {
@@ -39,7 +41,8 @@
     "@types/node": "^24.0.0",
     "oxfmt": "^0.53.0",
     "oxlint": "^1.68.0",
-    "typescript": "^5.0.0"
+    "typescript": "^5.0.0",
+    "vitest": "^4.1.10"
   },
   "peerDependencies": {
     "@earendil-works/pi-coding-agent": "^0.78.0"