Agents: update pi-coding-agent API usage

This commit is contained in:
Mario Zechner 2026-01-31 07:35:49 +01:00
parent 51e72d41c2
commit cbc405c9e3
2 changed files with 31 additions and 9 deletions

View file

@ -3,6 +3,7 @@ import os from "node:os";
import { import {
createAgentSession, createAgentSession,
DefaultResourceLoader,
estimateTokens, estimateTokens,
SessionManager, SessionManager,
SettingsManager, SettingsManager,
@ -383,6 +384,17 @@ export async function compactEmbeddedPiSessionDirect(
sandboxEnabled: !!sandbox?.enabled, sandboxEnabled: !!sandbox?.enabled,
}); });
const resourceLoader = new DefaultResourceLoader({
cwd: resolvedWorkspace,
agentDir,
settingsManager,
additionalExtensionPaths,
noSkills: true,
systemPromptOverride: systemPrompt,
agentsFilesOverride: () => ({ agentsFiles: [] }),
});
await resourceLoader.reload();
const { session } = await createAgentSession({ const { session } = await createAgentSession({
cwd: resolvedWorkspace, cwd: resolvedWorkspace,
agentDir, agentDir,
@ -394,10 +406,7 @@ export async function compactEmbeddedPiSessionDirect(
customTools, customTools,
sessionManager, sessionManager,
settingsManager, settingsManager,
additionalExtensionPaths, resourceLoader,
skills: [],
contextFiles: [],
systemPrompt,
}); });
try { try {

View file

@ -4,7 +4,12 @@ import os from "node:os";
import type { AgentMessage } from "@mariozechner/pi-agent-core"; import type { AgentMessage } from "@mariozechner/pi-agent-core";
import type { AssistantMessage, ImageContent } from "@mariozechner/pi-ai"; import type { AssistantMessage, ImageContent } from "@mariozechner/pi-ai";
import { streamSimple } from "@mariozechner/pi-ai"; import { streamSimple } from "@mariozechner/pi-ai";
import { createAgentSession, SessionManager, SettingsManager } from "@mariozechner/pi-coding-agent"; import {
createAgentSession,
DefaultResourceLoader,
SessionManager,
SettingsManager,
} from "@mariozechner/pi-coding-agent";
import { resolveHeartbeatPrompt } from "../../../auto-reply/heartbeat.js"; import { resolveHeartbeatPrompt } from "../../../auto-reply/heartbeat.js";
import { import {
@ -447,6 +452,17 @@ export async function runEmbeddedAttempt(
const allCustomTools = [...customTools, ...clientToolDefs]; const allCustomTools = [...customTools, ...clientToolDefs];
const resourceLoader = new DefaultResourceLoader({
cwd: resolvedWorkspace,
agentDir,
settingsManager,
additionalExtensionPaths,
noSkills: true,
systemPromptOverride: systemPrompt,
agentsFilesOverride: () => ({ agentsFiles: [] }),
});
await resourceLoader.reload();
({ session } = await createAgentSession({ ({ session } = await createAgentSession({
cwd: resolvedWorkspace, cwd: resolvedWorkspace,
agentDir, agentDir,
@ -458,10 +474,7 @@ export async function runEmbeddedAttempt(
customTools: allCustomTools, customTools: allCustomTools,
sessionManager, sessionManager,
settingsManager, settingsManager,
additionalExtensionPaths, resourceLoader,
skills: [],
contextFiles: [],
systemPrompt,
})); }));
if (!session) { if (!session) {
throw new Error("Embedded agent session missing"); throw new Error("Embedded agent session missing");