@anthonyhaussman/opencode-agy-auth 1.3.5-alpha.0 → 1.3.5-alpha.1

package/dist/index.js

@@ -2098,6 +2098,35 @@ function getLatestSignature(sessionId) {
   }
   return void 0;
 }
+function getSignatureForText(sessionId, text) {
+  if (!sessionId || !text) return void 0;
+  const textHash = hashText(text);
+  const sessionMemCache = signatureCache.get(sessionId);
+  if (sessionMemCache) {
+    const entry = sessionMemCache.get(textHash);
+    if (entry) {
+      if (Date.now() - entry.timestamp <= SIGNATURE_CACHE_TTL_MS) {
+        return entry.signature;
+      }
+      sessionMemCache.delete(textHash);
+    }
+  }
+  if (diskCache) {
+    const diskKey = makeDiskKey(sessionId, textHash);
+    const diskValue = diskCache.retrieve(diskKey);
+    if (diskValue) {
+      if (!sessionMemCache) {
+        const newMemCache = /* @__PURE__ */ new Map();
+        signatureCache.set(sessionId, newMemCache);
+        newMemCache.set(textHash, { signature: diskValue, timestamp: Date.now() });
+      } else {
+        sessionMemCache.set(textHash, { signature: diskValue, timestamp: Date.now() });
+      }
+      return diskValue;
+    }
+  }
+  return void 0;
+}
 
 // src/sdk/request/turn-state-tracker.ts
 import { existsSync as existsSync4, mkdirSync as mkdirSync3, readFileSync as readFileSync4, writeFileSync as writeFileSync4, renameSync as renameSync3, unlinkSync as unlinkSync3 } from "fs";
@@ -2136,7 +2165,7 @@ function hashString(str) {
 }
 function isThinkingPart(part) {
   if (!part || typeof part !== "object") return false;
-  return part.thought === true || part.type === "thinking" || part.type === "redacted_thinking";
+  return part.thought === true || part.type === "thinking" || part.type === "redacted_thinking" || part.type === "reasoning";
 }
 function isFunctionResponsePart(part) {
   return part && typeof part === "object" && "functionResponse" in part;
@@ -2156,7 +2185,7 @@ function messageHasThinking(msg) {
   }
   if (Array.isArray(msg.content)) {
     return msg.content.some(
-      (block) => block?.type === "thinking" || block?.type === "redacted_thinking"
+      (block) => block?.type === "thinking" || block?.type === "redacted_thinking" || block?.type === "reasoning"
     );
   }
   return false;
@@ -2263,7 +2292,7 @@ function deduplicateThinkingText(response, sentBuffer, displayedThinkingHashes)
       if (!Array.isArray(content.parts)) return candidate;
       const newParts = content.parts.map((part) => {
         const p = part;
-        if (p.thought === true || p.type === "thinking") {
+        if (p.thought === true || p.type === "thinking" || p.type === "reasoning") {
           const fullText = p.text || p.thinking || "";
           if (displayedThinkingHashes) {
             const hash2 = hashString(fullText);
@@ -2299,7 +2328,7 @@ function deduplicateThinkingText(response, sentBuffer, displayedThinkingHashes)
     let thinkingIndex = 0;
     const newContent = resp.content.map((block) => {
       const b = block;
-      if (b?.type === "thinking") {
+      if (b?.type === "thinking" || b?.type === "reasoning") {
         const fullText = b.thinking || b.text || "";
         if (displayedThinkingHashes) {
           const hash2 = hashString(fullText);
@@ -2345,17 +2374,18 @@ function cacheThinkingSignaturesFromResponse(response, signatureSessionKey, sign
       if (!Array.isArray(content.parts)) return;
       content.parts.forEach((part) => {
         const p = part;
-        if (p.thought === true || p.type === "thinking") {
+        if (p.thought === true || p.type === "thinking" || p.type === "reasoning") {
           const text = p.text || p.thinking || "";
           if (text) {
             const current = thoughtBuffer.get(index) ?? "";
             thoughtBuffer.set(index, current + text);
           }
         }
-        if (p.thoughtSignature) {
+        const partSignature = p.thoughtSignature || p.state?.thoughtSignature || p.signature;
+        if (partSignature) {
           const fullText = thoughtBuffer.get(index) ?? "";
           if (fullText) {
-            const signature = p.thoughtSignature;
+            const signature = partSignature;
             onCacheSignature?.(signatureSessionKey, fullText, signature);
             signatureStore.set(signatureSessionKey, { text: fullText, signature });
           }
@@ -2367,17 +2397,18 @@ function cacheThinkingSignaturesFromResponse(response, signatureSessionKey, sign
     const CLAUDE_BUFFER_KEY = 0;
     resp.content.forEach((block) => {
       const b = block;
-      if (b?.type === "thinking") {
+      if (b?.type === "thinking" || b?.type === "reasoning") {
         const text = b.thinking || b.text || "";
         if (text) {
           const current = thoughtBuffer.get(CLAUDE_BUFFER_KEY) ?? "";
           thoughtBuffer.set(CLAUDE_BUFFER_KEY, current + text);
         }
       }
-      if (b?.signature) {
+      const blockSignature = b?.signature || b?.thoughtSignature || b?.state?.thoughtSignature;
+      if (blockSignature) {
         const fullText = thoughtBuffer.get(CLAUDE_BUFFER_KEY) ?? "";
         if (fullText) {
-          const signature = b.signature;
+          const signature = blockSignature;
           onCacheSignature?.(signatureSessionKey, fullText, signature);
           signatureStore.set(signatureSessionKey, { text: fullText, signature });
         }
@@ -2453,7 +2484,7 @@ function createStreamingTransformer(signatureStore, callbacks, options = {}) {
           hasSeenUsageMetadata = true;
         }
         if (!streamHasThinking) {
-          streamHasThinking = event.includes('"thought":true') || event.includes('"type":"thinking"');
+          streamHasThinking = event.includes('"thought":true') || event.includes('"type":"thinking"') || event.includes('"type":"reasoning"');
         }
         if (!streamHasToolCalls) {
           streamHasToolCalls = event.includes('"functionCall"');
@@ -2477,7 +2508,7 @@ function createStreamingTransformer(signatureStore, callbacks, options = {}) {
           hasSeenUsageMetadata = true;
         }
         if (!streamHasThinking) {
-          streamHasThinking = buffer.includes('"thought":true') || buffer.includes('"type":"thinking"');
+          streamHasThinking = buffer.includes('"thought":true') || buffer.includes('"type":"thinking"') || buffer.includes('"type":"reasoning"');
         }
         if (!streamHasToolCalls) {
           streamHasToolCalls = buffer.includes('"functionCall"');
@@ -18903,7 +18934,7 @@ function getModelEnum(modelName) {
 function isClaudeModel(model) {
   return model.toLowerCase().includes("claude");
 }
-function sanitizeClaudeThinkingParts(contents, model) {
+function sanitizeClaudeThinkingParts(contents, model, sessionId) {
   if (!isClaudeModel(model) || !Array.isArray(contents)) {
     return;
   }
@@ -18913,12 +18944,49 @@ function sanitizeClaudeThinkingParts(contents, model) {
     if (!Array.isArray(turn.parts)) continue;
     const filteredParts = turn.parts.filter((part) => {
       if (!part || typeof part !== "object") return true;
-      const isThinking = part.thought === true || part.type === "thinking" || typeof part.thinking === "string";
+      const isThinking = part.thought === true || part.type === "thinking" || part.type === "reasoning" || typeof part.thinking === "string";
       if (!isThinking) {
         return true;
       }
-      const sig = typeof part.signature === "string" ? part.signature : typeof part.thoughtSignature === "string" ? part.thoughtSignature : void 0;
+      let sig = typeof part.signature === "string" ? part.signature : typeof part.thoughtSignature === "string" ? part.thoughtSignature : typeof part.state?.thoughtSignature === "string" ? part.state.thoughtSignature : void 0;
+      let recovered = false;
+      if ((!sig || sig === "skip_thought_signature_validator") && sessionId) {
+        const text = typeof part.text === "string" ? part.text : typeof part.thinking === "string" ? part.thinking : void 0;
+        if (text) {
+          const recoveredSig = getSignatureForText(sessionId, text);
+          if (recoveredSig) {
+            sig = recoveredSig;
+            recovered = true;
+          }
+        }
+        if (!sig) {
+          const latest = getLatestSignature(sessionId);
+          if (latest) {
+            sig = latest;
+            recovered = true;
+          }
+        }
+      }
       const isValidSig = typeof sig === "string" && sig.length > 0 && sig !== "skip_thought_signature_validator";
+      if (isValidSig) {
+        if (part.type === "reasoning") {
+          part.thoughtSignature = sig;
+          part.signature = sig;
+          if (part.state) {
+            part.state.thoughtSignature = sig;
+          }
+        } else if (recovered) {
+          if (typeof part.thoughtSignature === "string" || part.thought) {
+            part.thoughtSignature = sig;
+          }
+          if (part.type === "thinking" || typeof part.thinking === "string") {
+            part.signature = sig;
+          }
+          if (!part.thoughtSignature && !part.signature) {
+            part.signature = sig;
+          }
+        }
+      }
       return isValidSig;
     });
     if (filteredParts.length === 0) {
@@ -18989,7 +19057,7 @@ function transformRequestBody(body, projectId, effectiveModel, requestedModel, t
         contents2 = ensureTrailingUserTurn(contents2);
         const latestSig = getLatestSignature(sessionId2);
         applyLatestSignature(contents2, latestSig);
-        sanitizeClaudeThinkingParts(contents2, effectiveModel || requestedModel);
+        sanitizeClaudeThinkingParts(contents2, effectiveModel || requestedModel, sessionId2);
         requestPayloadInside.contents = contents2;
       }
       return { body: JSON.stringify(wrappedBody2), userPromptId: userPromptId2, sessionId: sessionId2 };
@@ -19031,7 +19099,7 @@ function transformRequestBody(body, projectId, effectiveModel, requestedModel, t
       contents = ensureTrailingUserTurn(contents);
       const latestSig = getLatestSignature(sessionId);
       applyLatestSignature(contents, latestSig);
-      sanitizeClaudeThinkingParts(contents, effectiveModel || requestedModel);
+      sanitizeClaudeThinkingParts(contents, effectiveModel || requestedModel, sessionId);
       requestPayload.contents = contents;
     }
     if ("model" in requestPayload) {
@@ -20469,6 +20537,26 @@ function createV2HttpRequestHook(state = {}) {
         }
       } catch {
       }
+      let bodyPayload = event.body;
+      if (bodyPayload === void 0 && event.request) {
+        if (typeof event.request.clone === "function") {
+          try {
+            bodyPayload = await event.request.clone().text();
+          } catch {
+          }
+        } else if (event.request.body !== void 0) {
+          bodyPayload = event.request.body;
+        }
+      }
+      if (bodyPayload) {
+        try {
+          const parsed2 = typeof bodyPayload === "string" ? JSON.parse(bodyPayload) : bodyPayload;
+          if (parsed2 && typeof parsed2 === "object" && typeof parsed2.sessionId === "string") {
+            state.lastSessionId = parsed2.sessionId;
+          }
+        } catch {
+        }
+      }
       if (authRecord && authRecord.access) {
         let projectContext;
         try {
@@ -20495,17 +20583,6 @@ function createV2HttpRequestHook(state = {}) {
             modifiedUrl = modifiedUrl.replace(`models/${originalBase}`, `models/${resolvedBase}`);
           }
         }
-        let bodyPayload = event.body;
-        if (bodyPayload === void 0 && event.request) {
-          if (typeof event.request.clone === "function") {
-            try {
-              bodyPayload = await event.request.clone().text();
-            } catch {
-            }
-          } else if (event.request.body !== void 0) {
-            bodyPayload = event.request.body;
-          }
-        }
         let preExistingUserAgent;
         if (typeof event.headers?.get === "function") {
           preExistingUserAgent = event.headers.get("user-agent") || event.headers.get("User-Agent") || void 0;
@@ -20529,6 +20606,9 @@ function createV2HttpRequestHook(state = {}) {
           authRecord.access,
           projectContext.effectiveProjectId
         );
+        if (transformed.sessionId) {
+          state.lastSessionId = transformed.sessionId;
+        }
         const targetUrl = toUrlString(transformed.request);
         if (typeof event.url === "string") {
           event.url = targetUrl;
@@ -20645,7 +20725,7 @@ function createV2HttpResponseHook(state = {}) {
         !!isStreaming,
         null,
         parsedModel,
-        void 0,
+        state.lastSessionId,
         void 0
       );
       event.response = transformed;
@@ -20687,6 +20767,26 @@ function registerV2SessionHooks(ctx) {
 
 // src/plugin/v2.ts
 async function setupOpenCodeV2(ctx) {
+  try {
+    initDiskSignatureCache({
+      enabled: true,
+      memory_ttl_seconds: 3600,
+      disk_ttl_seconds: 86400,
+      write_interval_seconds: 30
+    });
+  } catch (e) {
+    console.warn(`[Agy Auth] initDiskSignatureCache failed, running without signature disk cache: ${e instanceof Error ? e.message : e}`);
+  }
+  try {
+    initTurnStateTracker();
+  } catch (e) {
+    console.warn(`[Agy Auth] initTurnStateTracker failed, running without turn-state tracking: ${e instanceof Error ? e.message : e}`);
+  }
+  try {
+    initCooldownPersistence();
+  } catch (e) {
+    console.warn(`[Agy Auth] initCooldownPersistence failed, running without cooldown disk persistence: ${e instanceof Error ? e.message : e}`);
+  }
   const resolveVariants = (modelId) => {
     const mapping = TIER_MAPPING[modelId];
     if (!mapping) return void 0;