@sema-agent/core 1.326.0 → 1.328.0

package/dist/core/runner/prepare-task.js

@@ -73,6 +73,10 @@ const ungatedWarnedShapes = new WeakMap();
 const advisedPolicyNames = new WeakMap();
 const ADVISED_KEYS_CAP = 64;
 export const DEFAULT_IRREVERSIBLE_SCOPE = "irreversible";
+export function isFableFamilyModelId(id) {
+    const tail = id.toLowerCase().split("/").pop() ?? "";
+    return /^claude-fable-\d/.test(tail) || /^claude-mythos-5(?!\d)/.test(tail);
+}
 function batchContextAt(messages, currentId) {
     let batch = [];
     for (const m of messages) {
@@ -141,6 +145,7 @@ export async function prepareTask(spec, deps, sessions, resume, _memorySelector,
         exclude: spec.excludeTools ? Object.freeze([...spec.excludeTools]) : undefined,
         defer: spec.deferTools ? Object.freeze([...spec.deferTools]) : undefined,
     };
+    const promptProfile = spec.promptProfile ?? internals?.promptProfile ?? "simple";
     if (spec.resumeAt !== undefined) {
         if (resume) {
             const e = new Error("resumeAt cannot be combined with a durable resume");
@@ -182,6 +187,7 @@ export async function prepareTask(spec, deps, sessions, resume, _memorySelector,
     }
     const resolvedRole = resolveTaskModel(spec, deps);
     const model = resolvedRole.model;
+    const fableMitigations = isFableFamilyModelId(model.id);
     const thinking = spec.thinking ?? resolvedRole.thinking ?? model.defaultThinking;
     const compModel = spec.compactionModel
         ? resolveModel(spec.compactionModel, deps.models)
@@ -670,6 +676,7 @@ export async function prepareTask(spec, deps, sessions, resume, _memorySelector,
                 clientContext: spec.clientContext,
                 excludeTools: toolFaceSnapshot.exclude,
                 deferTools: toolFaceSnapshot.defer,
+                promptProfile,
                 getApiKeyAndHeaders: spec.getApiKeyAndHeaders,
                 parentCwd: taskRootPath,
                 activeSkillScope: () => skillScope.active(),
@@ -766,6 +773,7 @@ export async function prepareTask(spec, deps, sessions, resume, _memorySelector,
                 governanceBaseline: deps.workflowGovernanceBaseline,
                 parentExcludeTools: toolFaceSnapshot.exclude,
                 parentDeferTools: toolFaceSnapshot.defer,
+                parentPromptProfile: promptProfile,
                 models: deps.models,
                 agents: deps.agents,
                 builtinAgents: deps.builtinAgents,
@@ -1500,6 +1508,13 @@ export async function prepareTask(spec, deps, sessions, resume, _memorySelector,
                 if (excluded.has(tools[i].name))
                     tools.splice(i, 1);
         }
+        if (promptProfile === "classic") {
+            for (let i = 0; i < tools.length; i++) {
+                const t = tools[i];
+                if (t.descriptionClassic !== undefined)
+                    tools[i] = { ...t, description: t.descriptionClassic };
+            }
+        }
         const assembled = assemblePrompt({
             provider,
             isDefaultProvider: provider === defaultPromptProvider,
@@ -1512,6 +1527,8 @@ export async function prepareTask(spec, deps, sessions, resume, _memorySelector,
                 ...featureFlags,
                 orchestrationEnabled: featureFlags.orchestrationEnabled && tools.some((t) => t.name === "Workflow"),
                 orchestrationDeferred: featureFlags.orchestrationDeferred && tools.some((t) => t.name === "Workflow"),
+                promptProfile,
+                fableMitigations,
             },
             mcpInstructionsBlock,
             memoryBlock,