@dyyz1993/pi-coding-agent 0.69.9 → 0.69.10

package/dist/core/agent-session.js

@@ -14,7 +14,8 @@
  */
 import { existsSync, mkdirSync, readFileSync, writeFileSync } from "node:fs";
 import { basename, dirname, join, resolve } from "node:path";
-import { isContextOverflow, modelsAreEqual, resetApiProviders, supportsXhigh } from "@dyyz1993/pi-ai";
+import { Agent, } from "@dyyz1993/pi-agent-core";
+import { complete, isContextOverflow, modelsAreEqual, resetApiProviders, supportsXhigh } from "@dyyz1993/pi-ai";
 import { getDocsPath } from "../config.js";
 import { theme } from "../modes/interactive/theme/theme.js";
 import { stripFrontmatter } from "../utils/frontmatter.js";
@@ -31,7 +32,7 @@ import { CURRENT_SESSION_VERSION, getLatestCompactionEntry } from "./session-man
 import { createSyntheticSourceInfo } from "./source-info.js";
 import { buildSystemPrompt } from "./system-prompt.js";
 import { createLocalBashOperations } from "./tools/bash.js";
-import { createAllToolDefinitions } from "./tools/index.js";
+import { createAllToolDefinitions, createTool } from "./tools/index.js";
 import { createToolDefinitionFromAgentTool } from "./tools/tool-definition-wrapper.js";
 /**
  * Parse a skill block from message text.
@@ -1622,6 +1623,7 @@ export class AgentSession {
         if (bindings.registerChannel !== undefined) {
             this._registerChannel = bindings.registerChannel;
             this._extensionRunner.flushPendingChannels(bindings.registerChannel);
+            this._extensionRunner.updateRegisterChannel(bindings.registerChannel);
         }
         this._applyExtensionBindings(this._extensionRunner);
         await this._extensionRunner.emit(this._sessionStartEvent);
@@ -1728,7 +1730,8 @@ export class AgentSession {
                 });
             },
             appendEntry: (customType, data) => {
-                this.sessionManager.appendCustomEntry(customType, data);
+                const id = this.sessionManager.appendCustomEntry(customType, data);
+                this._emit({ type: "custom_entry", customType, data, id });
             },
             setSessionName: (name) => {
                 this.sessionManager.appendSessionInfo(name);
@@ -1756,6 +1759,7 @@ export class AgentSession {
                 ((name) => {
                     throw new Error(`registerChannel("${name}") is only available in RPC mode`);
                 }),
+            callLLM: (options) => this.callLLM(options),
         }, {
             getModel: () => this.model,
             isIdle: () => !this.isStreaming,
@@ -1790,6 +1794,110 @@ export class AgentSession {
             },
         });
     }
+    async callLLM(options) {
+        const model = this.model;
+        if (!model)
+            throw new Error("No model selected");
+        const auth = await this._modelRegistry.getApiKeyAndHeaders(model);
+        if (!auth?.ok) {
+            throw new Error(auth?.error ?? `No API key configured for ${model.provider}`);
+        }
+        if (!auth.apiKey) {
+            throw new Error(`No API key configured for ${model.provider}`);
+        }
+        if (options.signal?.aborted) {
+            throw new Error("Aborted");
+        }
+        const messages = options.messages.map((m) => ({
+            role: m.role,
+            content: [{ type: "text", text: m.content }],
+            timestamp: Date.now(),
+        }));
+        if (!options.tools || options.tools.length === 0) {
+            const context = {
+                systemPrompt: options.systemPrompt,
+                messages,
+            };
+            const response = await complete(model, context, {
+                apiKey: auth.apiKey,
+                headers: auth.headers,
+                maxTokens: options.maxTokens,
+                signal: options.signal,
+            });
+            return response.content
+                .filter((c) => c.type === "text")
+                .map((c) => c.text)
+                .join("\n");
+        }
+        const toolInstances = options.tools
+            .map((name) => {
+            try {
+                const registered = this._toolRegistry.get(name);
+                if (registered)
+                    return registered;
+                return createTool(name, this._cwd);
+            }
+            catch {
+                return undefined;
+            }
+        })
+            .filter((t) => t !== undefined);
+        if (toolInstances.length === 0) {
+            const context = {
+                systemPrompt: options.systemPrompt,
+                messages,
+            };
+            const response = await complete(model, context, {
+                apiKey: auth.apiKey,
+                headers: auth.headers,
+                maxTokens: options.maxTokens,
+                signal: options.signal,
+            });
+            return response.content
+                .filter((c) => c.type === "text")
+                .map((c) => c.text)
+                .join("\n");
+        }
+        const agent = new Agent({
+            getApiKey: () => auth.apiKey,
+            initialState: {
+                systemPrompt: options.systemPrompt ?? "",
+                model,
+                thinkingLevel: "off",
+                tools: toolInstances,
+                messages: [],
+            },
+        });
+        if (options.signal?.aborted) {
+            throw new Error("Aborted");
+        }
+        let resultText = "";
+        const unsub = agent.subscribe((event) => {
+            if (event.type === "message_end" && "message" in event) {
+                const msg = event.message;
+                if (msg.role === "assistant") {
+                    const content = msg.content;
+                    if (Array.isArray(content)) {
+                        resultText = content
+                            .filter((c) => c.type === "text")
+                            .map((c) => c.text)
+                            .join("\n");
+                    }
+                }
+            }
+        });
+        try {
+            await agent.prompt({
+                role: "user",
+                content: [{ type: "text", text: options.messages[0]?.content ?? "" }],
+                timestamp: Date.now(),
+            });
+        }
+        finally {
+            unsub();
+        }
+        return resultText;
+    }
     _refreshToolRegistry(options) {
         const previousRegistryNames = new Set(this._toolRegistry.keys());
         const previousActiveToolNames = this.getActiveToolNames();