@agent-native/core 0.34.0 → 0.35.0

package/dist/agent/production-agent.js

@@ -21,6 +21,10 @@ import { isReasoningEffort, normalizeReasoningEffortForModel, } from "../shared/
 import { isAgentActionStopError } from "../action.js";
 import { preUploadImageAttachments } from "../file-upload/pre-upload-attachments.js";
 import { extensionIdFromPathname } from "../extensions/path.js";
+import { applyContextDirectives } from "./context-xray/apply-directives.js";
+import { loadContextDirectives } from "./context-xray/directives-store.js";
+import { buildManifest, writeContextManifest, } from "./context-xray/manifest.js";
+import { computeProtectedSegmentIds } from "./context-xray/segments.js";
 // Register built-in engines on first import
 registerBuiltinEngines();
 export { PROVIDER_TO_ENV };
@@ -1006,6 +1010,37 @@ export async function runAgentLoop(opts) {
         let bufferedAssistantText = "";
         let terminalStopReason;
         const toolCallErrors = new Map();
+        let contextMessages = messages;
+        if (opts.threadId) {
+            try {
+                const directives = await loadContextDirectives(opts.threadId, {
+                    ownerEmail: opts.ownerEmail ?? null,
+                });
+                const protectedSegmentIds = computeProtectedSegmentIds(messages);
+                const { messages: transformedMessages, appliedStatus } = applyContextDirectives(messages, directives, {
+                    protectedSegmentIds,
+                });
+                const manifest = await buildManifest({
+                    threadId: opts.threadId,
+                    ...(opts.turnId ? { turnId: opts.turnId } : {}),
+                    model,
+                    rawMessages: messages,
+                    sentMessages: transformedMessages,
+                    appliedStatus,
+                    directives,
+                    protectedSegmentIds,
+                    source: "structured",
+                    enforceable: true,
+                });
+                contextMessages = transformedMessages;
+                void writeContextManifest(opts.threadId, manifest).catch((err) => {
+                    console.warn("[context-xray] failed to write manifest:", err instanceof Error ? err.message : String(err));
+                });
+            }
+            catch (err) {
+                console.warn("[context-xray] context transform skipped:", err instanceof Error ? err.message : String(err));
+            }
+        }
         for (let retry = 0;; retry++) {
             assistantContent = undefined;
             bufferedAssistantText = "";
@@ -1015,7 +1050,7 @@ export async function runAgentLoop(opts) {
                 const streamOpts = {
                     model,
                     systemPrompt,
-                    messages,
+                    messages: contextMessages,
                     tools,
                     abortSignal: signal,
                     maxOutputTokens: resolveMaxOutputTokensForEngine(engine.name, opts.maxOutputTokens),
@@ -1907,6 +1942,10 @@ export function createProductionAgentHandler(options) {
         }
         // Start agent loop in background via run-manager
         const runId = generateRunId();
+        const effectiveThreadId = threadId ?? runId;
+        const effectiveTurnId = typeof requestTurnId === "string" && requestTurnId.trim()
+            ? requestTurnId.trim()
+            : runId;
         if (options.onRunPrepared && !internalContinuation) {
             const messageToPersist = typeof requestDisplayMessage === "string" &&
                 requestDisplayMessage.trim().length > 0
@@ -1919,7 +1958,7 @@ export function createProductionAgentHandler(options) {
                 attachments: requestAttachments,
             });
         }
-        startRun(runId, threadId ?? runId, async (send, signal) => {
+        startRun(runId, effectiveThreadId, async (send, signal) => {
             send({ type: "activity", label: "Starting agent" });
             // Notify listeners that a run has started (used by agent teams)
             if (options.onRunStart) {
@@ -2164,6 +2203,9 @@ export function createProductionAgentHandler(options) {
                 executionMode: requestMode,
                 maxIterations: loopSettings.maxIterations,
                 finalResponseGuard: options.finalResponseGuard,
+                ...(threadId
+                    ? { threadId: effectiveThreadId, turnId: effectiveTurnId }
+                    : {}),
             };
             send({ type: "activity", label: "Contacting model" });
             let loopUsage;
@@ -2226,9 +2268,7 @@ export function createProductionAgentHandler(options) {
             // Fold continuation runs of one logical turn onto a single durable
             // assistant message. Falls back to the runId (turn == run) when the
             // client doesn't supply a turnId.
-            turnId: typeof requestTurnId === "string" && requestTurnId.trim()
-                ? requestTurnId.trim()
-                : runId,
+            turnId: effectiveTurnId,
         });
         // Subscribe to the run and stream events to the client
         const stream = subscribeToRun(runId, 0);