@voidwire/llm-summarize 3.3.0 → 3.5.0

package/index.ts

@@ -18,8 +18,20 @@ import { join } from "path";
 // Types
 // ============================================================================
 
+export interface Extraction {
+  term: string;
+  type: "project" | "topic" | "tool" | "person";
+  confidence: "high" | "medium";
+}
+
 export interface SessionInsights {
   summary: string;
+  // Quick mode extraction fields
+  should_search?: boolean;
+  extractions?: Extraction[];
+  // Insights mode fields
+  current_focus?: string;
+  next_steps?: string[];
   decisions?: string[];
   patterns_used?: string[];
   preferences_expressed?: string[];
@@ -29,6 +41,7 @@ export interface SessionInsights {
 
 export interface SummarizeResult {
   insights?: SessionInsights;
+  rawText?: string;
   error?: string;
   model?: string;
   tokens_used?: number;
@@ -48,6 +61,8 @@ export interface SummarizeOptions {
   mode?: "quick" | "insights";
   /** User name to include in summary (e.g., "Rudy") */
   userName?: string;
+  /** Override the system prompt (bypasses mode-based prompt selection) */
+  systemPrompt?: string;
 }
 
 export type ProviderType = "anthropic" | "openai" | "ollama";
@@ -59,13 +74,71 @@ export type SummarizeMode = "quick" | "insights";
 
 /**
  * Build quick mode prompt with optional user name
+ * Now includes context extraction for knowledge retrieval
  */
 function buildQuickPrompt(userName?: string): string {
-  const nameInstruction = userName ? `Start with "${userName}".` : "";
+  const name = userName || "User";
+
+  return `You are a context classifier for knowledge retrieval. Analyze conversation context to determine what prior knowledge would be valuable.
+
+Input format:
+Project: <project name>
+Previous Assistant: <last assistant message>
+User Prompt: <current user message>
+
+Produce JSON with:
+1. summary: Brief description (1-2 sentences) of what the user is doing/asking. Start with "${name}".
+2. should_search: Whether to search the knowledge base
+3. extractions: Terms worth searching for
+
+should_search = true when:
+- References past work, decisions, discussions
+- Mentions project, tool, or person by name
+- Asks "what was...", "how did we...", "remember when..."
+- Technical domain benefits from prior learnings
+
+should_search = false when:
+- Greetings, acknowledgments ("ready", "thanks", "ok")
+- Simple commands ("run tests", "commit this")
+- Continuation signals ("yes", "do it", "go ahead")
+
+Extraction types:
+- project: Named codebase, repo, system (sable, lore, momentum)
+- topic: Domain, concept, technical area (hooks, authentication, Tier 2)
+- tool: Library, CLI, framework (llm-summarize, SQLite, Bun)
+- person: Named individual
+
+Confidence:
+- high: Explicitly stated
+- medium: Strongly implied
+
+Skip generic words. Only extract terms that yield useful knowledge results.
+
+<example>
+Project: sable
+Previous Assistant: I'll update the UserPromptSubmit hook to call llm-summarize.
+User Prompt: What does Lore return for project queries?
+
+{"summary": "${name} is asking about Lore's return format for project queries", "should_search": true, "extractions": [{"term": "Lore", "type": "project", "confidence": "high"}, {"term": "project queries", "type": "topic", "confidence": "high"}]}
+</example>
+
+<example>
+Project: sable
+Previous Assistant: The extraction prompt is ready. Should I add it?
+User Prompt: yes do it
+
+{"summary": "${name} is confirming to proceed with the extraction prompt", "should_search": false, "extractions": []}
+</example>
+
+<example>
+Project: sable
+Previous Assistant: Starting new session.
+User Prompt: What was the issue we hit with the stop hook last time?
+
+{"summary": "${name} is asking about a previous issue with the stop hook", "should_search": true, "extractions": [{"term": "stop hook", "type": "topic", "confidence": "high"}, {"term": "sable", "type": "project", "confidence": "medium"}]}
+</example>
 
-  return `Summarize what the user is asking or doing in one sentence.
-${nameInstruction}
-Output JSON only: {"summary": "One sentence summary"}`;
+Output valid JSON only. No markdown, no explanation.`;
 }
 
 /**
@@ -84,6 +157,8 @@ Your job: extract what's worth remembering for future sessions.
 <output_schema>
 {
   "summary": "One sentence capturing what was accomplished and how",
+  "current_focus": "What work is actively in progress — the specific task, feature, or thread (omit if session was exploratory with no clear focus)",
+  "next_steps": ["What should happen when work resumes — concrete actions, not vague intentions"],
   "decisions": ["Decision made with reasoning and trade-offs considered"],
   "patterns_used": ["Development pattern or approach, with context on why it was chosen"],
   "preferences_expressed": ["Preference revealed through direction or feedback"],
@@ -353,12 +428,14 @@ async function callAnthropic(
 
     if (!insights) {
       return {
+        rawText: content,
         error: `Failed to parse response as JSON: ${content.slice(0, 200)}`,
       };
     }
 
     return {
       insights,
+      rawText: content,
       model,
       tokens_used: result.usage?.output_tokens,
     };
@@ -419,12 +496,14 @@ async function callOpenAI(
 
     if (!insights) {
       return {
+        rawText: content,
         error: `Failed to parse response as JSON: ${content.slice(0, 200)}`,
       };
     }
 
     return {
       insights,
+      rawText: content,
       model,
       tokens_used: result.usage?.completion_tokens,
     };
@@ -486,12 +565,14 @@ async function callOllama(
 
     if (!insights) {
       return {
+        rawText: content,
         error: `Failed to parse response as JSON: ${content.slice(0, 200)}`,
       };
     }
 
     return {
       insights,
+      rawText: content,
       model,
       tokens_used: result.eval_count,
     };
@@ -529,7 +610,7 @@ export async function summarize(
   const apiKey = config.apiKey;
   const mode: SummarizeMode = options?.mode || "insights";
   const userName = options?.userName;
-  const systemPrompt = getPromptForMode(mode, userName);
+  const systemPrompt = options?.systemPrompt || getPromptForMode(mode, userName);
 
   // Validate config
   if (!provider) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@voidwire/llm-summarize",
-  "version": "3.3.0",
+  "version": "3.5.0",
   "description": "Structured session insight extraction for knowledge systems",
   "type": "module",
   "main": "./index.ts",