clay-server 2.47.0-beta.1 → 2.47.0-beta.3

package/lib/yoke/adapters/codex.js

@@ -597,6 +597,27 @@ function flattenEvent(notification, state) {
   return events;
 }
 
+function createEventState(model) {
+  return {
+    blockCounter: 0,
+    threadId: null,
+    turnStarted: false,
+    lastUsage: null,
+    lastInputTokens: null,
+    done: false,
+    aborted: false,
+    loopStarted: false,
+    model: model || "gpt-5.6-terra",
+    textBlocks: {},
+    textLengths: {},
+    thinkingBlocks: {},
+    thinkingLengths: {},
+    toolBlocks: {},
+    commandInputs: {},
+    planTexts: {},
+  };
+}
+
 // --- QueryHandle ---
 
 function createCodexQueryHandle(appServer, queryOpts) {
@@ -611,25 +632,7 @@ function createCodexQueryHandle(appServer, queryOpts) {
     return state.aborted || (abortController && abortController.signal && abortController.signal.aborted);
   }
 
-  var state = {
-    blockCounter: 0,
-    threadId: null,
-    turnStarted: false,
-    lastUsage: null,
-    lastInputTokens: null, // from thread/tokenUsage/updated
-    done: false,
-    aborted: false,
-    loopStarted: false,
-    model: queryOpts.model || "gpt-5.6-terra",
-    // Track incremental text deltas
-    textBlocks: {},     // itemId -> true (text_start sent)
-    textLengths: {},    // itemId -> last sent length
-    thinkingBlocks: {}, // itemId -> blockId
-    thinkingLengths: {}, // itemId -> last sent length
-    toolBlocks: {},     // itemId -> blockId (for tool_start dedup)
-    commandInputs: {},  // itemId -> command captured from approval/start events
-    planTexts: {},      // itemId -> streamed plan text
-  };
+  var state = createEventState(queryOpts.model);
 
   // Internal event buffer for async iterator
   var eventBuffer = [];
@@ -867,7 +870,7 @@ function createCodexQueryHandle(appServer, queryOpts) {
       var threadParams = {
         model: queryOpts.model || "gpt-5.6-terra",
         sandbox: queryOpts.sandboxMode || "workspace-write",
-        approvalPolicy: queryOpts.approvalPolicy || "on-failure",
+        approvalPolicy: queryOpts.approvalPolicy || "on-request",
         cwd: queryOpts.cwd,
         skipGitRepoCheck: true,
       };
@@ -1139,6 +1142,11 @@ function createCodexAdapter(opts) {
       slashCommands: skillNames || [],
       fastModeState: null,
       capabilities: {
+        effort: true,
+        fork: true,
+        rollback: true,
+        sessionListing: false,
+        sessionRename: false,
         thinking: true,
         betas: false,
         rewind: false,
@@ -1463,7 +1471,7 @@ function createCodexAdapter(opts) {
       if (queryOpts.toolPolicy === "allow-all") {
         handleOpts.approvalPolicy = "never";
       } else {
-        handleOpts.approvalPolicy = codexOpts.approvalPolicy || "on-failure";
+        handleOpts.approvalPolicy = codexOpts.approvalPolicy || "on-request";
       }
 
       // Sandbox mode
@@ -1590,4 +1598,9 @@ function createCodexAdapter(opts) {
 
 module.exports = {
   createCodexAdapter: createCodexAdapter,
+  contractTestKit: {
+    createEventState: createEventState,
+    createQueryHandle: createCodexQueryHandle,
+    normalizeEvent: flattenEvent,
+  },
 };