@ryanfw/prompt-orchestration-pipeline 0.13.5 → 0.14.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ryanfw/prompt-orchestration-pipeline",
-  "version": "0.13.5",
+  "version": "0.14.1",
   "description": "A Prompt-orchestration pipeline (POP) is a framework for building, running, and experimenting with complex chains of LLM tasks.",
   "type": "module",
   "main": "src/ui/server.js",

package/src/components/ui/RestartJobModal.jsx

@@ -130,25 +130,35 @@ export function RestartJobModal({
                 Cancel
               </Button>
 
-              {taskId && (
+              {taskId ? (
+                <>
+                  <Button
+                    variant="outline"
+                    onClick={() => onConfirm({ singleTask: false })}
+                    disabled={isSubmitting}
+                    className="min-w-[120px]"
+                  >
+                    {isSubmitting ? "Restarting..." : "Restart entire pipeline"}
+                  </Button>
+                  <Button
+                    variant="default"
+                    onClick={() => onConfirm({ singleTask: true })}
+                    disabled={isSubmitting}
+                    className="min-w-[120px]"
+                  >
+                    {isSubmitting ? "Running..." : "Re-run this task"}
+                  </Button>
+                </>
+              ) : (
                 <Button
-                  variant="outline"
-                  onClick={() => onConfirm({ singleTask: true })}
+                  variant="destructive"
+                  onClick={() => onConfirm({ singleTask: false })}
                   disabled={isSubmitting}
-                  className="min-w-[120px]"
+                  className="min-w-[80px]"
                 >
-                  {isSubmitting ? "Running..." : "Just this task"}
+                  {isSubmitting ? "Restarting..." : "Restart"}
                 </Button>
               )}
-
-              <Button
-                variant="destructive"
-                onClick={() => onConfirm({ singleTask: false })}
-                disabled={isSubmitting}
-                className="min-w-[80px]"
-              >
-                {isSubmitting ? "Restarting..." : "Restart"}
-              </Button>
             </Flex>
           </div>
         </div>

package/src/core/pipeline-runner.js

@@ -18,6 +18,8 @@ import { createJobLogger } from "./logger.js";
 import { LogEvent, LogFileExtension } from "../config/log-events.js";
 import { decideTransition } from "./lifecycle-policy.js";
 
+const getTaskName = (t) => (typeof t === "string" ? t : t.name);
+
 const ROOT = process.env.PO_ROOT || process.cwd();
 const DATA_DIR = path.join(ROOT, process.env.PO_DATA_DIR || "pipeline-data");
 const CURRENT_DIR =
@@ -104,6 +106,8 @@ const pipeline = JSON.parse(await fs.readFile(PIPELINE_DEF_PATH, "utf8"));
 // Validate pipeline format early with a friendly error message
 validatePipelineOrThrow(pipeline, PIPELINE_DEF_PATH);
 
+const taskNames = pipeline.tasks.map(getTaskName);
+
 const tasks = (await loadFreshModule(TASK_REGISTRY)).default;
 
 const status = JSON.parse(await fs.readFile(tasksStatusPath, "utf8"));
@@ -123,21 +127,21 @@ logger.group("Pipeline execution", {
 
 // Helper function to check if all upstream dependencies are completed
 function areDependenciesReady(taskName) {
-  const taskIndex = pipeline.tasks.indexOf(taskName);
+  const taskIndex = taskNames.indexOf(taskName);
   if (taskIndex === -1) return false;
 
-  const upstreamTasks = pipeline.tasks.slice(0, taskIndex);
+  const upstreamTasks = taskNames.slice(0, taskIndex);
   return upstreamTasks.every(
     (upstreamTask) => status.tasks[upstreamTask]?.state === TaskState.DONE
   );
 }
 
 try {
-  for (const taskName of pipeline.tasks) {
+  for (const taskName of taskNames) {
     // Skip tasks before startFromTask when targeting a specific restart point
     if (
       startFromTask &&
-      pipeline.tasks.indexOf(taskName) < pipeline.tasks.indexOf(startFromTask)
+      taskNames.indexOf(taskName) < taskNames.indexOf(startFromTask)
     ) {
       logger.log("Skipping task before restart point", {
         taskName,
@@ -158,30 +162,32 @@ try {
       continue;
     }
 
-    // Check lifecycle policy before starting task
-    const currentTaskState = status.tasks[taskName]?.state || "pending";
-    const dependenciesReady = areDependenciesReady(taskName);
-
-    const lifecycleDecision = decideTransition({
-      op: "start",
-      taskState: currentTaskState,
-      dependenciesReady,
-    });
+    // Check lifecycle policy before starting task (skip when startFromTask is set - user explicitly requested this task)
+    if (!startFromTask) {
+      const currentTaskState = status.tasks[taskName]?.state || "pending";
+      const dependenciesReady = areDependenciesReady(taskName);
 
-    if (!lifecycleDecision.ok) {
-      logger.warn("lifecycle_block", {
-        jobId,
-        taskId: taskName,
+      const lifecycleDecision = decideTransition({
         op: "start",
-        reason: lifecycleDecision.reason,
+        taskState: currentTaskState,
+        dependenciesReady,
       });
 
-      // Create typed error for endpoints to handle
-      const lifecycleError = new Error(lifecycleDecision.reason);
-      lifecycleError.httpStatus = 409;
-      lifecycleError.error = "unsupported_lifecycle";
-      lifecycleError.reason = lifecycleDecision.reason;
-      throw lifecycleError;
+      if (!lifecycleDecision.ok) {
+        logger.warn("lifecycle_block", {
+          jobId,
+          taskId: taskName,
+          op: "start",
+          reason: lifecycleDecision.reason,
+        });
+
+        // Create typed error for endpoints to handle
+        const lifecycleError = new Error(lifecycleDecision.reason);
+        lifecycleError.httpStatus = 409;
+        lifecycleError.error = "unsupported_lifecycle";
+        lifecycleError.reason = lifecycleDecision.reason;
+        throw lifecycleError;
+      }
     }
 
     logger.log("Starting task", { taskName });

package/src/llm/index.js

@@ -89,6 +89,18 @@ export function calculateCost(provider, model, usage) {
   return promptCost + completionCost;
 }
 
+// Helper function to detect if messages indicate JSON response is needed
+function shouldInferJsonFormat(messages) {
+  // Check first two messages for JSON keyword (case-insensitive)
+  const messagesToCheck = messages.slice(0, 2);
+  for (const msg of messagesToCheck) {
+    if (typeof msg?.content === "string" && /json/i.test(msg.content)) {
+      return true;
+    }
+  }
+  return false;
+}
+
 // Core chat function - no metrics handling needed!
 export async function chat(options) {
   console.log("[llm] chat() called with options:", {
@@ -199,6 +211,17 @@ export async function chat(options) {
       };
     } else if (provider === "openai") {
       console.log("[llm] Using OpenAI provider");
+
+      // Infer JSON format if not explicitly provided
+      const effectiveResponseFormat =
+        responseFormat === undefined ||
+        responseFormat === null ||
+        responseFormat === ""
+          ? shouldInferJsonFormat(messages)
+            ? "json_object"
+            : undefined
+          : responseFormat;
+
       const openaiArgs = {
         messages,
         model: model || "gpt-5-chat-latest",
@@ -211,8 +234,8 @@ export async function chat(options) {
         hasMessages: !!openaiArgs.messages,
         messageCount: openaiArgs.messages?.length,
       });
-      if (responseFormat !== undefined) {
-        openaiArgs.responseFormat = responseFormat;
+      if (effectiveResponseFormat !== undefined) {
+        openaiArgs.responseFormat = effectiveResponseFormat;
       }
       if (topP !== undefined) openaiArgs.topP = topP;
       if (frequencyPenalty !== undefined)
@@ -255,6 +278,17 @@ export async function chat(options) {
       }
     } else if (provider === "deepseek") {
       console.log("[llm] Using DeepSeek provider");
+
+      // Infer JSON format if not explicitly provided
+      const effectiveResponseFormat =
+        responseFormat === undefined ||
+        responseFormat === null ||
+        responseFormat === ""
+          ? shouldInferJsonFormat(messages)
+            ? "json_object"
+            : undefined
+          : responseFormat;
+
       const deepseekArgs = {
         messages,
         model: model || MODEL_CONFIG[DEFAULT_MODEL_BY_PROVIDER.deepseek].model,
@@ -274,8 +308,8 @@ export async function chat(options) {
       if (presencePenalty !== undefined)
         deepseekArgs.presencePenalty = presencePenalty;
       if (stop !== undefined) deepseekArgs.stop = stop;
-      if (responseFormat !== undefined) {
-        deepseekArgs.responseFormat = responseFormat;
+      if (effectiveResponseFormat !== undefined) {
+        deepseekArgs.responseFormat = effectiveResponseFormat;
       }
 
       console.log("[llm] Calling deepseekChat()...");
@@ -373,6 +407,17 @@ export async function chat(options) {
       }
     } else if (provider === "gemini") {
       console.log("[llm] Using Gemini provider");
+
+      // Infer JSON format if not explicitly provided
+      const effectiveResponseFormat =
+        responseFormat === undefined ||
+        responseFormat === null ||
+        responseFormat === ""
+          ? shouldInferJsonFormat(messages)
+            ? "json_object"
+            : undefined
+          : responseFormat;
+
       const geminiArgs = {
         messages,
         model: model || "gemini-2.5-flash",
@@ -387,8 +432,8 @@ export async function chat(options) {
       });
       if (topP !== undefined) geminiArgs.topP = topP;
       if (stop !== undefined) geminiArgs.stop = stop;
-      if (responseFormat !== undefined) {
-        geminiArgs.responseFormat = responseFormat;
+      if (effectiveResponseFormat !== undefined) {
+        geminiArgs.responseFormat = effectiveResponseFormat;
       }
 
       console.log("[llm] Calling geminiChat()...");

package/src/providers/anthropic.js

@@ -2,6 +2,7 @@ import {
   extractMessages,
   isRetryableError,
   sleep,
+  stripMarkdownFences,
   tryParseJSON,
   ensureJsonResponseFormat,
   ProviderJsonParseError,
@@ -77,10 +78,12 @@ export async function anthropicChat({
 
       // Extract text from response.content blocks
       const blocks = Array.isArray(data?.content) ? data.content : [];
-      const text = blocks
+      const rawText = blocks
         .filter((b) => b?.type === "text" && typeof b.text === "string")
         .map((b) => b.text)
         .join("");
+      // Always strip markdown fences first to prevent parse failures
+      const text = stripMarkdownFences(rawText);
       console.log("[Anthropic] Response text length:", text.length);
 
       // Parse JSON - this is required for all calls

package/src/providers/base.js

@@ -33,6 +33,25 @@ export async function sleep(ms) {
   return new Promise((r) => setTimeout(r, ms));
 }
 
+/**
+ * Strip markdown code fences from text unconditionally.
+ * Handles ```json, ```JSON, and plain ``` with or without newlines.
+ * @param {string} text - The text to strip fences from
+ * @returns {string} The cleaned text, or original if not a string
+ */
+export function stripMarkdownFences(text) {
+  if (typeof text !== "string") return text;
+  const trimmed = text.trim();
+  if (trimmed.startsWith("```")) {
+    // Remove opening fence (```json, ```JSON, or just ```)
+    let cleaned = trimmed.replace(/^```(?:json|JSON)?\s*\n?/, "");
+    // Remove closing fence
+    cleaned = cleaned.replace(/\n?```\s*$/, "");
+    return cleaned.trim();
+  }
+  return text;
+}
+
 export function tryParseJSON(text) {
   try {
     return JSON.parse(text);
@@ -85,7 +104,12 @@ export class ProviderJsonModeError extends Error {
  * Error thrown when JSON parsing fails and should not be retried
  */
 export class ProviderJsonParseError extends Error {
-  constructor(provider, model, sample, message = "Failed to parse JSON response") {
+  constructor(
+    provider,
+    model,
+    sample,
+    message = "Failed to parse JSON response"
+  ) {
     super(message);
     this.name = "ProviderJsonParseError";
     this.provider = provider;
@@ -109,8 +133,9 @@ export function ensureJsonResponseFormat(responseFormat, providerName) {
   }
 
   // Check for valid JSON format types
-  const isValidJsonFormat = 
+  const isValidJsonFormat =
     responseFormat === "json" ||
+    responseFormat === "json_object" ||
     responseFormat?.type === "json_object" ||
     responseFormat?.type === "json_schema";
 

package/src/providers/deepseek.js

@@ -2,6 +2,7 @@ import {
   extractMessages,
   isRetryableError,
   sleep,
+  stripMarkdownFences,
   tryParseJSON,
   ensureJsonResponseFormat,
   ProviderJsonParseError,
@@ -24,11 +25,12 @@ export async function deepseekChat({
     throw new Error("DeepSeek API key not configured");
   }
 
-  // Determine if JSON mode is requested
+  // Determine if JSON mode is requested (handle both object and string formats)
   const isJsonMode =
     responseFormat?.type === "json_object" ||
     responseFormat?.type === "json_schema" ||
-    responseFormat === "json";
+    responseFormat === "json" ||
+    responseFormat === "json_object";
 
   const { systemMsg, userMsg } = extractMessages(messages);
 
@@ -84,7 +86,10 @@ export async function deepseekChat({
       }
 
       const data = await response.json();
-      const content = data.choices[0].message.content;
+      const rawContent = data.choices[0].message.content;
+
+      // Always strip markdown fences first to prevent parse failures
+      const content = stripMarkdownFences(rawContent);
 
       // Parse JSON only in JSON mode; return raw string for text mode
       if (isJsonMode) {

package/src/providers/gemini.js

@@ -2,6 +2,7 @@ import {
   extractMessages,
   isRetryableError,
   sleep,
+  stripMarkdownFences,
   tryParseJSON,
   ensureJsonResponseFormat,
   ProviderJsonParseError,
@@ -32,7 +33,7 @@ export async function geminiChat(options) {
     frequencyPenalty,
     presencePenalty,
     stop,
-    maxRetries = 3
+    maxRetries = 3,
   } = options;
 
   // Validate response format (Gemini only supports JSON mode)
@@ -171,7 +172,9 @@ export async function geminiChat(options) {
         throw new Error("No content returned from Gemini API");
       }
 
-      const text = candidate.content.parts[0].text;
+      const rawText = candidate.content.parts[0].text;
+      // Always strip markdown fences first to prevent parse failures
+      const text = stripMarkdownFences(rawText);
       console.log(`[Gemini] Text length: ${text.length}`);
 
       // Parse JSON if required

package/src/providers/openai.js

@@ -3,6 +3,7 @@ import {
   extractMessages,
   isRetryableError,
   sleep,
+  stripMarkdownFences,
   tryParseJSON,
   ensureJsonResponseFormat,
   ProviderJsonParseError,
@@ -53,11 +54,12 @@ export async function openaiChat({
   console.log("[OpenAI] System message length:", systemMsg.length);
   console.log("[OpenAI] User message length:", userMsg.length);
 
-  // Determine if JSON mode is requested
+  // Determine if JSON mode is requested (handle both object and string formats)
   const isJsonMode =
     responseFormat?.json_schema ||
     responseFormat?.type === "json_object" ||
-    responseFormat === "json";
+    responseFormat === "json" ||
+    responseFormat === "json_object";
 
   let lastError;
   for (let attempt = 0; attempt <= maxRetries; attempt++) {
@@ -100,7 +102,9 @@ export async function openaiChat({
 
         console.log("[OpenAI] Calling responses.create...");
         const resp = await openai.responses.create(responsesReq);
-        const text = resp.output_text ?? "";
+        const rawText = resp.output_text ?? "";
+        // Always strip markdown fences first to prevent parse failures
+        const text = stripMarkdownFences(rawText);
         console.log("[OpenAI] Response received, text length:", text.length);
 
         // Approximate usage (tests don't assert exact values)
@@ -161,7 +165,9 @@ export async function openaiChat({
 
       console.log("[OpenAI] Calling chat.completions.create...");
       const classicRes = await openai.chat.completions.create(classicReq);
-      const classicText = classicRes?.choices?.[0]?.message?.content ?? "";
+      const rawClassicText = classicRes?.choices?.[0]?.message?.content ?? "";
+      // Always strip markdown fences first to prevent parse failures
+      const classicText = stripMarkdownFences(rawClassicText);
       console.log(
         "[OpenAI] Response received, text length:",
         classicText.length
@@ -228,7 +234,9 @@ export async function openaiChat({
         }
 
         const classicRes = await openai.chat.completions.create(classicReq);
-        const text = classicRes?.choices?.[0]?.message?.content ?? "";
+        const rawText = classicRes?.choices?.[0]?.message?.content ?? "";
+        // Always strip markdown fences first to prevent parse failures
+        const text = stripMarkdownFences(rawText);
 
         // Parse JSON only in JSON mode; return raw string for text mode
         if (isJsonMode) {

package/src/providers/zhipu.js

@@ -2,6 +2,7 @@ import {
   extractMessages,
   isRetryableError,
   sleep,
+  stripMarkdownFences,
   tryParseJSON,
   ensureJsonResponseFormat,
   ProviderJsonParseError,
@@ -104,7 +105,9 @@ export async function zhipuChat({
       console.log("[Zhipu] Response received from Zhipu API");
 
       // Extract text from response
-      const text = data?.choices?.[0]?.message?.content || "";
+      const rawText = data?.choices?.[0]?.message?.content || "";
+      // Always strip markdown fences first to prevent parse failures
+      const text = stripMarkdownFences(rawText);
       console.log("[Zhipu] Response text length:", text.length);
 
       // Parse JSON - this is required for all calls

package/src/ui/dist/assets/{index-cjHV9mYW.js → index-B5HMRkR9.js}

@@ -24711,17 +24711,28 @@ function RestartJobModal({
                     children: "Cancel"
                   }
                 ),
-                taskId && /* @__PURE__ */ jsxRuntimeExports.jsx(
-                  Button,
-                  {
-                    variant: "outline",
-                    onClick: () => onConfirm({ singleTask: true }),
-                    disabled: isSubmitting,
-                    className: "min-w-[120px]",
-                    children: isSubmitting ? "Running..." : "Just this task"
-                  }
-                ),
-                /* @__PURE__ */ jsxRuntimeExports.jsx(
+                taskId ? /* @__PURE__ */ jsxRuntimeExports.jsxs(jsxRuntimeExports.Fragment, { children: [
+                  /* @__PURE__ */ jsxRuntimeExports.jsx(
+                    Button,
+                    {
+                      variant: "outline",
+                      onClick: () => onConfirm({ singleTask: false }),
+                      disabled: isSubmitting,
+                      className: "min-w-[120px]",
+                      children: isSubmitting ? "Restarting..." : "Restart entire pipeline"
+                    }
+                  ),
+                  /* @__PURE__ */ jsxRuntimeExports.jsx(
+                    Button,
+                    {
+                      variant: "default",
+                      onClick: () => onConfirm({ singleTask: true }),
+                      disabled: isSubmitting,
+                      className: "min-w-[120px]",
+                      children: isSubmitting ? "Running..." : "Re-run this task"
+                    }
+                  )
+                ] }) : /* @__PURE__ */ jsxRuntimeExports.jsx(
                   Button,
                   {
                     variant: "destructive",