pi-codebuddy-sdk 0.1.17 → 0.1.19

package/README.md

@@ -1,5 +1,7 @@
 # pi-codebuddy-sdk
 
+Forked from [pi-claude-bridge](https://github.com/elidickinson/pi-claude-bridge). Replaces Claude Agent SDK with CodeBuddy Agent SDK.
+
 回退到 `@tencent-ai/agent-sdk` 的 Pi provider 扩展。无反向代理，直接使用本地 CodeBuddy CLI，合规安全。
 
 ## 安装
@@ -7,8 +9,8 @@
 ### 方式 1：全局安装（日常使用）
 
 ```bash
-ln -s ~/Desktop/pi-codebuddy/src ~/.pi/agent/extensions/pi-codebuddy
-cd ~/Desktop/pi-codebuddy && npm install
+ln -s ~/Desktop/usaslahser/pi-codebuddy-sdk/src ~/.pi/agent/extensions/pi-codebuddy-sdk
+cd ~/Desktop/usaslahser/pi-codebuddy-sdk && npm install
 ```
 
 重启 Pi 即可自动发现。
@@ -16,7 +18,7 @@ cd ~/Desktop/pi-codebuddy && npm install
 ### 方式 2：临时加载（测试）
 
 ```bash
-pi -e ~/Desktop/pi-codebuddy/src/index.ts --provider codebuddy --model <model>
+pi -e ~/Desktop/usaslahser/pi-codebuddy-sdk/src/index.ts --provider codebuddy --model <model>
 ```
 
 ## 认证
@@ -71,32 +73,27 @@ pi --provider codebuddy --model claude-sonnet-4.6
 | 模型 | 说明 |
 |------|------|
 | `claude-sonnet-4.6` | Claude Sonnet 4.6，支持 thinking |
-| `claude-opus-4.8` | Claude Opus 4.8，支持 thinking + image |
-| `gemini-3.1-pro` | Gemini 3.1 Pro，支持 thinking + image |
-| `gpt-5.5` | GPT-5.5，支持 thinking + image |
 | `hy3-preview-agent-ioa` | 混元 3 Preview，iOA 内免费无限额度 |
+| `deepseek-v4-pro-ioa` | DeepSeek V4 Pro |
+| `glm-5.2-ioa` | GLM 5.2 |
 
 ## 特性
 
 | 特性 | 说明 |
 |------|------|
-| **动态模型发现** | 启动时从 CodeBuddy SDK 拉取 29 个模型，自动标注 thinking/image 能力 |
-| **Session 池复用** | 多轮对话复用同一 CodeBuddy CLI 进程，首轮后每轮省 ~6s 启动 |
+| **动态模型发现** | 启动时从 CodeBuddy SDK 拉取模型，自动标注 thinking/image 能力 |
 | **Thinking 控制** | Pi 的 thinking level 映射到 CodeBuddy 的 thinking/effort 参数 |
 | **Image 输入** | 原生支持 Claude/Gemini/GPT 的多模态图片输入 |
 | **Abort 处理** | Esc 取消 → CodeBuddy `interrupt()`，干净终止 |
 | **Token 统计** | 包含 cache_read / cache_write / total_cost_usd |
 | **YOLO 模式** | `bypassPermissions`，所有工具自动批准（与 pi-cursor-sdk 一致） |
+| **Stateless** | 每次调用独立 query()，无 session 复用，简单可靠 |
 
 ## FAQ
 
-**Q: 和 pi-cursor-sdk 有什么区别？**
-
-pi-cursor-sdk 需要自建 MCP bridge 桥接 Cursor SDK 工具，pi-codebuddy SDK 原生处理工具执行，代码量减少 60%+。
-
-**Q: 怎么查看当前用了多少 tokens？**
+**Q: 和 pi-claude-bridge 有什么区别？**
 
-每次回复后 Pi 会显示 usage，包含 input/output/cache/total 及费用。
+pi-claude-bridge 使用 Anthropic Claude Agent SDK + MCP bridge，需要 session 管理和复杂的工具桥接。pi-codebuddy-sdk 使用腾讯 CodeBuddy Agent SDK，原生工具执行，代码量减少 60%+。
 
 **Q: maxTurns 限制？**
 
@@ -104,4 +101,4 @@ pi-cursor-sdk 需要自建 MCP bridge 桥接 Cursor SDK 工具，pi-codebuddy SD
 
 **Q: 支持哪些 CodeBuddy 版本？**
 
-CodeBuddy CLI ≥ 2.x（`@tencent-ai/agent-sdk` ≥ 0.3.0）。
+CodeBuddy CLI >= 2.x（`@tencent-ai/agent-sdk` >= 0.3.0）。

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "pi-codebuddy-sdk",
-  "version": "0.1.17",
-  "description": "CodeBuddy Agent SDK provider for Pi — native tool execution, 29+ models, thinking controls, session pool",
+  "version": "0.1.19",
+  "description": "CodeBuddy Agent SDK provider for Pi — native tool execution, thinking controls, stateless one-shot queries",
   "author": "slahser",
   "license": "MIT",
   "type": "module",
@@ -44,6 +44,7 @@
   },
   "devDependencies": {
     "@earendil-works/pi-ai": "^0.80.2",
-    "@earendil-works/pi-coding-agent": "^0.80.2"
+    "@earendil-works/pi-coding-agent": "^0.80.2",
+    "@earendil-works/pi-tui": "^0.80.2"
   }
 }

package/src/agents-md.ts

@@ -0,0 +1,39 @@
+// AGENTS.md discovery for forwarding to CodeBuddy.
+
+import { existsSync, readFileSync } from "fs";
+import { homedir } from "os";
+import { dirname, join, resolve } from "path";
+
+const GLOBAL_AGENTS_PATH = join(homedir(), ".pi", "agent", "AGENTS.md");
+
+export function resolveAgentsMdPath(): string | undefined {
+  const fromCwd = findAgentsMdInParents(process.cwd());
+  if (fromCwd) return fromCwd;
+  if (existsSync(GLOBAL_AGENTS_PATH)) return GLOBAL_AGENTS_PATH;
+  return undefined;
+}
+
+export function findAgentsMdInParents(startDir: string): string | undefined {
+  let current = resolve(startDir);
+  while (true) {
+    const candidate = join(current, "AGENTS.md");
+    if (existsSync(candidate)) return candidate;
+    const parent = dirname(current);
+    if (parent === current) break;
+    current = parent;
+  }
+  return undefined;
+}
+
+export function extractAgentsAppend(): string | undefined {
+  const agentsPath = resolveAgentsMdPath();
+  if (!agentsPath) return undefined;
+  try {
+    const content = readFileSync(agentsPath, "utf-8").trim();
+    if (!content) return undefined;
+    const sanitized = content; // keep as-is for CodeBuddy
+    return sanitized.length > 0 ? `# CODEBUDDY.md\n\n${sanitized}` : undefined;
+  } catch {
+    return undefined;
+  }
+}

package/src/askcodebuddy-ui.ts

@@ -0,0 +1,90 @@
+// Status-line rendering helpers for the AskClaude tool.
+//
+// While Claude Code runs inside an AskClaude call, the pi TUI can't surface
+// each tool_use individually — there's only one status row for the whole
+// delegation. These helpers shape a tool_use record into a short, path-aware
+// label (e.g. "Read(src/foo.ts)", "Bash(git log --oneline…)") and collapse
+// runs of the same tool so the line doesn't flicker. Used only by
+// promptAndWait; the provider path exposes tools directly through pi's TUI
+// and doesn't need this.
+
+export interface ToolCallState {
+	name: string;
+	status: string;
+	rawInput?: unknown;
+}
+
+export function extractPath(rawInput: unknown): string | undefined {
+	if (!rawInput || typeof rawInput !== "object") return undefined;
+	const input = rawInput as Record<string, unknown>;
+	if (typeof input.file_path === "string") return input.file_path;
+	if (typeof input.path === "string") return input.path;
+	if (typeof input.command === "string") return input.command.substring(0, 80);
+	return undefined;
+}
+
+export function shortPath(p: string): string {
+	const cwd = process.cwd();
+	if (p.startsWith(cwd + "/")) return p.slice(cwd.length + 1);
+	if (p.startsWith("/")) {
+		const parts = p.split("/");
+		if (parts.length > 3) return parts.slice(-2).join("/");
+	}
+	return p;
+}
+
+export function formatToolAction(tc: ToolCallState): string | undefined {
+	const path = extractPath(tc.rawInput);
+	const verb = tc.name.toLowerCase().split(/\s/)[0];
+	if (verb === "read" || verb === "readfile") {
+		return path ? `Read(${shortPath(path)})` : "Read";
+	} else if (verb === "glob") {
+		const input = tc.rawInput as Record<string, unknown> | undefined;
+		const pat = typeof input?.pattern === "string" ? input.pattern.slice(0, 40) : "";
+		return pat ? `Glob(${pat})` : "Glob";
+	} else if (verb === "edit" || verb === "write" || verb === "writefile" || verb === "multiedit") {
+		return path ? `Edit(${shortPath(path)})` : "Edit";
+	} else if (verb === "bashoutput") {
+		return undefined; // redundant with preceding Bash call
+	} else if (verb === "bash" || verb === "terminal") {
+		return path ? `Bash(${path})` : "Bash";
+	} else if (verb === "agent") {
+		const input = tc.rawInput as Record<string, unknown> | undefined;
+		return `Agent(${String(input?.description ?? "").slice(0, 40)})`;
+	} else if (verb === "grep") {
+		const input = tc.rawInput as Record<string, unknown> | undefined;
+		const pat = typeof input?.pattern === "string" ? input.pattern.slice(0, 40) : "";
+		return pat ? `Grep(${pat})` : "Grep";
+	} else if (verb === "skill") {
+		const input = tc.rawInput as Record<string, unknown> | undefined;
+		const name = typeof input?.skill === "string" ? input.skill.slice(0, 40) : "";
+		return name ? `Skill(${name})` : "Skill";
+	} else if (verb === "todowrite" || verb === "taskcreate" || verb === "taskupdate") {
+		const todos = Array.isArray((tc.rawInput as any)?.todos) ? (tc.rawInput as any).todos : [];
+		const current = todos.find((t: any) => t.status === "in_progress") ?? todos.find((t: any) => t.status === "pending");
+		const label = current ? String(current.content ?? "").slice(0, 40) : "";
+		return label || undefined;
+	} else if (verb === "askclaude") {
+		// Recursive — don't show AskClaude in its own action summary
+		return undefined;
+	}
+	return tc.name;
+}
+
+export function buildActionSummary(calls: Map<string, ToolCallState>): string {
+	const parts: string[] = [];
+	let prevVerb = "";
+	for (const [, tc] of calls) {
+		const action = formatToolAction(tc);
+		if (!action) continue;
+		const verb = tc.name.toLowerCase().split(/\s/)[0];
+		// Collapse consecutive calls to the same tool — keep only the latest
+		if (verb === prevVerb) {
+			parts[parts.length - 1] = action;
+		} else {
+			parts.push(action);
+		}
+		prevVerb = verb;
+	}
+	return parts.join("; ");
+}

package/src/config.ts

@@ -0,0 +1,46 @@
+// User-facing extension config. Loaded once at extension registration from
+// ~/.pi/agent/codebuddy-sdk.json and the project Pi config directory, project
+// overriding global. Missing or unparseable files are ignored.
+
+import { CONFIG_DIR_NAME } from "@earendil-works/pi-coding-agent";
+import { existsSync, readFileSync } from "fs";
+import { homedir } from "os";
+import { join } from "path";
+
+export interface Config {
+  askCodebuddy?: {
+    enabled?: boolean;
+    name?: string;
+    label?: string;
+    description?: string;
+    defaultMode?: "full" | "read" | "none";
+    defaultIsolated?: boolean;
+    allowFullMode?: boolean;
+    appendSkills?: boolean;
+  };
+  provider?: {
+    appendSystemPrompt?: boolean;
+    pathToCodebuddyExecutable?: string;
+    plan?: "pro" | "max";
+    longContextExtraUsage?: boolean;
+  };
+}
+
+export function tryParseJson(path: string): Partial<Config> {
+  if (!existsSync(path)) return {};
+  try {
+    return JSON.parse(readFileSync(path, "utf-8"));
+  } catch (e) {
+    console.error(`codebuddy-sdk: failed to parse ${path}: ${e}`);
+    return {};
+  }
+}
+
+export function loadConfig(cwd: string): Config {
+  const global = tryParseJson(join(homedir(), ".pi", "agent", "codebuddy-sdk.json"));
+  const project = tryParseJson(join(cwd, CONFIG_DIR_NAME, "codebuddy-sdk.json"));
+  return {
+    askCodebuddy: { ...global.askCodebuddy, ...project.askCodebuddy },
+    provider: { ...global.provider, ...project.provider },
+  };
+}

package/src/convert.ts

@@ -0,0 +1,105 @@
+// Pure pi -> SDK message conversion helpers.
+
+import type { Message as PiMessage } from "@earendil-works/pi-ai";
+
+export const PROVIDER_ID = "codebuddy";
+
+export const PI_TO_SDK_TOOL_NAME: Record<string, string> = {
+  read: "Read", write: "Write", edit: "Edit", bash: "Bash",
+};
+
+export function sanitizeToolId(id: string, cache: Map<string, string>): string {
+  const existing = cache.get(id);
+  if (existing) return existing;
+  const clean = id.replace(/[^a-zA-Z0-9_-]/g, "_");
+  cache.set(id, clean);
+  return clean;
+}
+
+export function mapPiToolNameToSdk(name: string, customToolNameToSdk?: Map<string, string>): string {
+  if (!name) return "";
+  const normalized = name.toLowerCase();
+  if (customToolNameToSdk) {
+    const mapped = customToolNameToSdk.get(name) ?? customToolNameToSdk.get(normalized);
+    if (mapped) return mapped;
+  }
+  if (PI_TO_SDK_TOOL_NAME[normalized]) return PI_TO_SDK_TOOL_NAME[normalized];
+  // ponytail: simple PascalCase fallback instead of change-case dep
+  return name.charAt(0).toUpperCase() + name.slice(1);
+}
+
+export function messageContentToText(
+  content: string | Array<{ type: string; text?: string; data?: string; mimeType?: string }>,
+): string {
+  if (typeof content === "string") return content;
+  if (!Array.isArray(content)) return "";
+  const parts = [];
+  let hasText = false;
+  for (const block of content) {
+    if (block.type === "text" && block.text) { parts.push(block.text); hasText = true; }
+    else if (block.type !== "text" && block.type !== "image") { parts.push(`[${block.type}]`); }
+  }
+  return hasText ? parts.join("\n") : "";
+}
+
+// SessionMessage is an internal type used by the provider's session persistence.
+// It mirrors the Anthropic API message format.
+export interface SessionMessage {
+  role: "user" | "assistant";
+  content: string | Array<Record<string, unknown>>;
+}
+
+/** Convert pi message array to SDK-compatible format. */
+export function convertPiMessages(
+  messages: PiMessage[],
+  customToolNameToSdk?: Map<string, string>,
+): { anthropicMessages: SessionMessage[]; sanitizedIds: Map<string, string> } {
+  const anthropicMessages: SessionMessage[] = [];
+  const sanitizedIds = new Map<string, string>();
+
+  for (const msg of messages) {
+    if (msg.role === "user") {
+      if (typeof msg.content === "string") {
+        anthropicMessages.push({ role: "user", content: msg.content || "[empty]" });
+      } else if (Array.isArray(msg.content)) {
+        const parts: Array<Record<string, unknown>> = [];
+        for (const block of msg.content) {
+          if (block.type === "text" && block.text) parts.push({ type: "text", text: block.text });
+          else if (block.type === "image" && (block as any).data && (block as any).mimeType) {
+            parts.push({ type: "image", source: { type: "base64", media_type: (block as any).mimeType, data: (block as any).data } });
+          }
+        }
+        anthropicMessages.push({ role: "user", content: parts.length ? parts : "[image]" });
+      } else {
+        anthropicMessages.push({ role: "user", content: "[empty]" });
+      }
+    } else if (msg.role === "assistant") {
+      const content = Array.isArray(msg.content) ? msg.content : [];
+      const blocks: Array<Record<string, unknown>> = [];
+      for (const block of content) {
+        if (block.type === "text" && block.text) {
+          blocks.push({ type: "text", text: block.text });
+        } else if (block.type === "thinking") {
+          const sig = (block as any).thinkingSignature;
+          const isCodebuddyProvider = msg.provider === PROVIDER_ID || msg.api === "codebuddy-sdk";
+          if (isCodebuddyProvider && sig) {
+            blocks.push({ type: "thinking", thinking: block.thinking ?? "", signature: sig });
+          }
+        } else if (block.type === "toolCall") {
+          const toolName = mapPiToolNameToSdk((block as any).name, customToolNameToSdk);
+          blocks.push({ type: "tool_use", id: sanitizeToolId((block as any).id, sanitizedIds), name: toolName, input: (block as any).arguments ?? {} });
+        }
+      }
+      if (!blocks.length) blocks.push({ type: "text", text: "[incompatible content omitted]" });
+      anthropicMessages.push({ role: "assistant", content: blocks });
+    } else if (msg.role === "toolResult") {
+      const text = typeof msg.content === "string" ? msg.content : messageContentToText(msg.content);
+      anthropicMessages.push({
+        role: "user",
+        content: [{ type: "tool_result", tool_use_id: sanitizeToolId((msg as any).toolCallId, sanitizedIds), content: text || "", is_error: (msg as any).isError }],
+      });
+    }
+  }
+
+  return { anthropicMessages, sanitizedIds };
+}