@liy/agent-runner 0.2.2 → 0.2.4

package/dist/index.js

@@ -15637,7 +15637,6 @@ var driverOwnedPiFlags = /* @__PURE__ */ new Set([
   "--provider",
   "--model",
   "--thinking",
-  "--no-extensions",
   "--no-prompt-templates",
   "--no-themes"
 ]);
@@ -15683,7 +15682,6 @@ function buildPiRpcArgs(harness) {
     "--thinking",
     thinking,
     "--no-session",
-    "--no-extensions",
     "--no-prompt-templates",
     "--no-themes",
     ...extraArgs
@@ -15846,6 +15844,9 @@ function validateExtraPiArgs(args) {
     if (driverOwnedPiFlags.has(flag)) {
       throw new Error(`pi-rpc harness args cannot include driver-owned Pi flag: ${flag}`);
     }
+    if (flag === "--no-extensions") {
+      throw new Error("pi-rpc harness args cannot disable Pi extensions");
+    }
     if (blockedPiFlags.has(flag)) {
       throw new Error(`pi-rpc harness args cannot include unsupported Pi flag: ${flag}`);
     }
@@ -15868,6 +15869,52 @@ function formatPiRpcError(parsed, redactText) {
   return "unknown error";
 }
 
+// src/prompt.ts
+import { existsSync, readFileSync } from "node:fs";
+var harnessPromptTemplateFileName = "execute-task.md";
+function buildHarnessPrompt(spec) {
+  return renderHarnessPromptTemplate({
+    template: loadDefaultHarnessPromptTemplate(),
+    values: {
+      workspaceDir: spec.workspaceDir,
+      completionFile: `${spec.workspaceDir}/state/completion.json`,
+      artifactRoot: `${spec.workspaceDir}/artifacts`,
+      taskTitle: spec.task.title,
+      taskContext: spec.task.context,
+      intentJson: JSON.stringify(spec.intent, null, 2),
+      userQueryJson: JSON.stringify(spec.userQuery, null, 2),
+      approvedToolsJson: JSON.stringify(spec.toolProvisioning.tools, null, 2)
+    }
+  });
+}
+function resolveAgentRunnerPromptDir(importUrl = import.meta.url) {
+  const candidates = [
+    new URL("../prompts/", importUrl),
+    new URL("./prompts/", importUrl)
+  ];
+  return candidates.find(hasHarnessPromptTemplate) ?? candidates[0];
+}
+function hasHarnessPromptTemplate(promptDir) {
+  return existsSync(new URL(harnessPromptTemplateFileName, promptDir));
+}
+function loadDefaultHarnessPromptTemplate() {
+  return readFileSync(new URL(harnessPromptTemplateFileName, resolveAgentRunnerPromptDir()), "utf8");
+}
+function renderHarnessPromptTemplate(input) {
+  let rendered = input.template;
+  const placeholders = new Set(
+    Array.from(input.template.matchAll(/{{([a-zA-Z0-9_]+)}}/g), (match) => match[1])
+  );
+  for (const placeholder of placeholders) {
+    const value = input.values[placeholder];
+    if (value === void 0) {
+      throw new Error(`missing harness prompt template value: ${placeholder}`);
+    }
+    rendered = rendered.replaceAll(`{{${placeholder}}}`, value);
+  }
+  return rendered.trimEnd();
+}
+
 // src/runner.ts
 var forbiddenSecretKeyPattern = /(API[_-]?KEY|ACCESS[_-]?KEY|SECRET|TOKEN|PASSWORD|POSTGRES|DATABASE_URL|SPRITES|OSS|ALIYUN)/i;
 var allowedHarnessEnvKeys = /* @__PURE__ */ new Set([
@@ -15975,36 +16022,6 @@ function createHarnessEnvironment(input) {
   assertNoForbiddenHarnessEnv(harnessEnv, allowedForbiddenKeys);
   return harnessEnv;
 }
-function buildHarnessPrompt(spec) {
-  return [
-    "You are the Mote execute_task Task Agent.",
-    "",
-    "Use only the configured model and tool endpoints. A direct provider token may be present temporarily; never log it, persist it, print it, or write it to artifacts.",
-    `Workspace: ${spec.workspaceDir}`,
-    `State file: ${spec.workspaceDir}/state/completion.json`,
-    `Artifact root: ${spec.workspaceDir}/artifacts`,
-    "",
-    "Task",
-    `Title: ${spec.task.title}`,
-    spec.task.context,
-    "",
-    "Intent",
-    JSON.stringify(spec.intent, null, 2),
-    "",
-    "User Query",
-    JSON.stringify(spec.userQuery, null, 2),
-    "",
-    "Approved Tools",
-    JSON.stringify(spec.toolProvisioning.tools, null, 2),
-    "",
-    "Artifact contract",
-    "- Allocate a UUID directory under artifacts for each artifact.",
-    "- Write artifact.json in that directory using the TaskArtifactDraft contract.",
-    "- Dataset artifacts must put schema and preview rows in artifact.json and the full CSV in data.csv.",
-    "- Write terminal status to state/completion.json with status, summary, and artifactIds.",
-    "- Never include secrets, tokens, provider keys, DB credentials, or OSS credentials in artifacts, logs, or completion."
-  ].join("\n");
-}
 async function readTaskRunnerSpec(path) {
   return JSON.parse(await readFile3(path, "utf8"));
 }