feat: include entityContext in interpretWriteTurn prompt payload

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>

Author: MaxLinCode

packages/integrations/src/index.test.ts

@@ -4,6 +4,7 @@ import {
 } from "@atlas/core";
 import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
 
+import { interpretWriteTurnSystemPrompt } from "./prompts/interpret-write-turn";
 import {
   buildGoogleCalendarOAuthUrl,
   createGoogleCalendarAdapter,
@@ -1153,4 +1154,67 @@ describe("integrations", () => {
       "Telegram editMessageText failed with status 400: Bad Request: message can't be edited.",
     );
   });
+
+  it("includes entityContext in the write interpretation prompt payload and prompt instructions", async () => {
+    const parse = vi.fn(async () => ({
+      output_parsed: {
+        operationKind: "edit",
+        actionDomain: "task",
+        targetRef: {
+          entityId: "task-1",
+          description: null,
+          entityKind: null,
+        },
+        taskName: null,
+        fields: {
+          scheduleFields: null,
+          taskFields: null,
+        },
+        confidence: {},
+        unresolvedFields: [],
+      },
+    }));
+
+    await interpretWriteTurnWithResponses(
+      {
+        currentTurnText: "Move gym to 5",
+        turnType: "edit_request",
+        entityContext:
+          'Known entities:\n- "Gym" (task, scheduled) [id: task-1]',
+      },
+      {
+        responses: {
+          parse,
+        },
+      },
+    );
+
+    expect(parse).toHaveBeenCalledWith(
+      expect.objectContaining({
+        input: expect.arrayContaining([
+          expect.objectContaining({
+            role: "system",
+            content: expect.arrayContaining([
+              expect.objectContaining({
+                text: expect.stringContaining(
+                  "Resolve entity references only against the provided entityContext.",
+                ),
+              }),
+            ]),
+          }),
+          expect.objectContaining({
+            role: "user",
+            content: expect.arrayContaining([
+              expect.objectContaining({
+                text: expect.stringContaining("Entity context:\nKnown entities:"),
+              }),
+            ]),
+          }),
+        ]),
+      }),
+    );
+    expect(interpretWriteTurnSystemPrompt).toContain(
+      "Never invent an entity ID",
+    );
+  });
 });

packages/integrations/src/openai.ts

@@ -330,7 +330,7 @@ export async function interpretWriteTurnWithResponses(
         content: [
           {
             type: "input_text",
-            text: JSON.stringify(buildWriteInterpretationPromptContext(context)),
+            text: buildWriteInterpretationPromptContext(context),
           },
         ],
       },
@@ -413,12 +413,24 @@ function buildSlotExtractorPromptContext(context: SlotExtractorInput) {
 function buildWriteInterpretationPromptContext(
   context: WriteInterpretationInput,
 ) {
-  return {
-    currentTurnText: context.currentTurnText,
-    turnType: context.turnType,
-    priorPendingWriteOperation: context.priorPendingWriteOperation ?? null,
-    conversationContext: context.conversationContext ?? null,
-  };
+  return [
+    "Current turn text:",
+    context.currentTurnText,
+    "",
+    "Turn type:",
+    context.turnType,
+    "",
+    "Prior pending write operation:",
+    context.priorPendingWriteOperation
+      ? JSON.stringify(context.priorPendingWriteOperation, null, 2)
+      : "None",
+    "",
+    "Conversation context:",
+    context.conversationContext ?? "None",
+    "",
+    "Entity context:",
+    context.entityContext ?? "None",
+  ].join("\n");
 }
 
 function buildTurnRoutingPromptContext(context: TurnRoutingInput) {