Agents: update pi-coding-agent API usage
This commit is contained in:
parent
51e72d41c2
commit
cbc405c9e3
2 changed files with 31 additions and 9 deletions
|
|
@ -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 {
|
||||||
|
|
|
||||||
|
|
@ -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");
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue