pi-codebuddy-sdk 0.1.15 → 0.1.16

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-codebuddy-sdk",
-  "version": "0.1.15",
+  "version": "0.1.16",
   "description": "CodeBuddy Agent SDK provider for Pi — native tool execution, 29+ models, thinking controls, session pool",
   "author": "slahser",
   "license": "MIT",

package/src/index.ts

@@ -10,7 +10,7 @@
  * Two files: index.ts (registration + model discovery), provider.ts (streaming).
  */
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
-import type { ProviderApi, SimpleStreamOptions } from "@earendil-works/pi-ai/compat";
+import type { ProviderApi } from "@earendil-works/pi-ai/compat";
 import { streamCodebuddy } from "./provider.js";
 
 // ── provider registration ──
@@ -25,7 +25,6 @@ export default function activate(api: ExtensionAPI) {
     streamSimple: streamCodebuddy,
   });
 
-  // Lazy model discovery at startup
   queueMicrotask(async () => {
     try {
       const models = await discoverModels();
@@ -38,8 +37,7 @@ export default function activate(api: ExtensionAPI) {
         streamSimple: streamCodebuddy,
       });
     } catch {
-      // Model discovery failed — provider registered with empty models,
-      // user can still try manually.
+      // Model discovery failed — provider stays with empty models, user can try manually.
     }
   });
 }
@@ -52,13 +50,13 @@ async function getSdk() {
   return _sdk;
 }
 
-export async function discoverModels(): Promise<ProviderApi["models"]> {
+async function discoverModels(): Promise<ProviderApi["models"]> {
   const sdk = await getSdk();
   const q = sdk.query({ prompt: " ", options: { maxTurns: 0 } });
   const supported = await q.supportedModels();
   await q.interrupt();
 
-  const models = supported.map((m) => {
+  return supported.map((m) => {
     const caps = detectCapabilities(m.id);
     return {
       id: m.id,
@@ -72,11 +70,9 @@ export async function discoverModels(): Promise<ProviderApi["models"]> {
       maxTokens: caps.maxTokens,
     };
   });
-
-  return models;
 }
 
-// ── capability detection (pattern-based, like pi-claude-bridge) ──
+// ── capability detection ──
 
 const THINKING_PATTERNS = [/claude/i, /gemini/i, /gpt-5/i];
 const IMAGE_PATTERNS = [/claude/i, /gemini/i, /gpt/i];

package/src/provider.ts

@@ -2,7 +2,7 @@
  * Provider entry point — pi-claude-bridge pattern.
  *
  * For each Pi turn:
- * 1. Extract the latest user message (strip context-mode injection)
+ * 1. Extract the latest user message
  * 2. Call sdk.query() — stateless, CodeBuddy uses its own system prompt
  * 3. Stream events back to Pi (text, thinking, toolcalls)
  */
@@ -54,12 +54,6 @@ function makeErrorPartial(model: Model<any>, error: unknown): AssistantMessage {
 
 // ── extract user message ──
 
-/**
- * Extract the latest user message from Pi Context.
- * Strips context-mode session state injection.
- *
- * Like pi-claude-bridge's extractUserPrompt() but with context-mode defense.
- */
 function extractUserMessage(context: Context): string {
   const msgs = context.messages;
   for (let i = msgs.length - 1; i >= 0; i--) {
@@ -162,7 +156,7 @@ export function streamCodebuddy(
         if (options.signal.aborted) { await onAbort(); return; }
       }
 
-      // ── per-block tracking (for stream_event delta accumulation) ──
+      // ── per-block tracking ──
       type BlockKind = "text" | "thinking" | "tool_use" | "redacted_thinking";
       const blockKind = new Map<number, BlockKind>();
       const blockAccum = new Map<number, string>();
@@ -238,7 +232,6 @@ export function streamCodebuddy(
           continue;
         }
 
-        // Token-level streaming via stream_event (preferred)
         if (msg.type === "stream_event") {
           hasStreamEvents = true;
           const evt = msg.event;
@@ -260,7 +253,7 @@ export function streamCodebuddy(
           continue;
         }
 
-        // Fallback: full assistant messages
+        // Fallback: full assistant messages (when model doesn't support streaming)
         if (msg.type === "assistant" && !hasStreamEvents) {
           let ci = 0;
           for (const block of msg.message.content) {

package/src/provider-debug.patch

@@ -1 +0,0 @@
-# Temporary: add debug logging to extractUserMessage