@cortexkit/pi-anthropic-auth 2.1.1 → 2.2.0

package/dist/convert.d.ts

@@ -33,6 +33,9 @@ export declare function buildAnthropicRequest(modelId: string, context: Context,
 }, fastModeEnabled?: boolean, identity?: ClaudeCodeIdentity, controls?: {
     effortTransitions?: readonly MidConversationEffortTransition[];
     thinkingPrefixMismatchBehavior?: ThinkingPrefixMismatchBehavior;
+    turnEndNote?: {
+        tools: string[];
+    };
 }): Promise<{
     body: AnthropicRequestBody;
     bodyText: string;

package/dist/convert.js

@@ -1,4 +1,4 @@
-import { applyClaudeCodeMetadata, applyMidConversationOutputConfig, applyThinkingBindingControls, assertNoMeaningfulTrailingAssistant, buildBillingHeaderValue, CLAUDE_CODE_ENTRYPOINT, CLAUDE_CODE_IDENTITY, CLAUDE_FABLE_MYTHOS_5_SUMMARIZED_THINKING, CLAUDE_HAIKU_5_5_ADAPTIVE_THINKING, CLAUDE_OPUS_5_5_ADAPTIVE_THINKING, CLAUDE_OPUS_5_ADAPTIVE_THINKING, CLAUDE_SONNET_5_5_ADAPTIVE_THINKING, CLAUDE_SONNET_5_ADAPTIVE_THINKING, ClaudeCodeFirstUserTextTracker, isClaudeFableOrMythos5Model, isClaudeHaiku55Model, isClaudeOpus5Model, isClaudeOpus55Model, isClaudeSonnet5Model, isClaudeSonnet55Model, isFastModeSupportedModel, isOpenAIReasoningSignature, orderClaudeCodeBody, signRequestBody, stripEmptyTrailingAssistantMessages, } from '@cortexkit/anthropic-auth-core';
+import { applyClaudeCodeMetadata, applyMidConversationOutputConfig, applyThinkingBindingControls, applyTurnEndNotes, assertNoMeaningfulTrailingAssistant, buildBillingHeaderValue, CLAUDE_CODE_ENTRYPOINT, CLAUDE_CODE_IDENTITY, CLAUDE_FABLE_MYTHOS_5_SUMMARIZED_THINKING, CLAUDE_HAIKU_5_5_ADAPTIVE_THINKING, CLAUDE_OPUS_5_5_ADAPTIVE_THINKING, CLAUDE_OPUS_5_ADAPTIVE_THINKING, CLAUDE_SONNET_5_5_ADAPTIVE_THINKING, CLAUDE_SONNET_5_ADAPTIVE_THINKING, ClaudeCodeFirstUserTextTracker, isClaudeFableOrMythos5Model, isClaudeHaiku55Model, isClaudeOpus5Model, isClaudeOpus55Model, isClaudeSonnet5Model, isClaudeSonnet55Model, isFastModeSupportedModel, isOpenAIReasoningSignature, orderClaudeCodeBody, signRequestBody, stripEmptyTrailingAssistantMessages, } from '@cortexkit/anthropic-auth-core';
 import { collapseSystemMessages, getCurrentSystemPrompt, getCurrentTools, normalizeContext, } from './transcript.js';
 // Anchor identifying Pi's documentation paragraph — the only part of the prompt
 // that Anthropic currently rejects in system[]. Unknown prompt shapes take the
@@ -111,7 +111,7 @@ function convertTextAndImages(content) {
     }
     return blocks;
 }
-function convertMessages(messages, targetModelId) {
+function convertMessages(messages, targetModelId, visibleReplies) {
     const result = [];
     for (let index = 0; index < messages.length; index++) {
         const message = messages[index];
@@ -142,9 +142,12 @@ function convertMessages(messages, targetModelId) {
                 continue;
             }
             const blocks = [];
+            let visibleText = '';
             for (const block of message.content) {
                 if (block.type === 'text' && block.text.trim()) {
-                    blocks.push({ type: 'text', text: sanitize(block.text) });
+                    const text = sanitize(block.text);
+                    blocks.push({ type: 'text', text });
+                    visibleText += text;
                 }
                 else if (block.type === 'thinking') {
                     const thinking = block;
@@ -213,8 +216,15 @@ function convertMessages(messages, targetModelId) {
                     });
                 }
             }
-            if (blocks.length)
-                result.push({ role: 'assistant', content: blocks });
+            if (blocks.length) {
+                const converted = { role: 'assistant', content: blocks };
+                result.push(converted);
+                // Converted thinking can become text for replay, but it was never a
+                // visible reply. Count only actual text blocks, using the exact outgoing
+                // strings collected above, so the note decision for this message stays
+                // the same on every later request.
+                visibleReplies.set(converted, visibleText);
+            }
             continue;
         }
         if (message.role === 'toolResult') {
@@ -453,7 +463,8 @@ export async function buildAnthropicRequest(modelId, context, options, cache, fa
     // first so those losses cannot hide a meaningful final assistant turn. Check
     // the Anthropic messages again after conversion removes empty user messages.
     assertNoMeaningfulTrailingAssistant(context.messages);
-    const messages = convertMessages(context.messages, modelId);
+    const visibleReplies = new WeakMap();
+    const messages = convertMessages(context.messages, modelId, visibleReplies);
     stripEmptyTrailingAssistantMessages(messages);
     const system = [
         {
@@ -566,6 +577,13 @@ export async function buildAnthropicRequest(modelId, context, options, cache, fa
         }
     }
     applyMidConversationOutputConfig(body, controls.effortTransitions ?? []);
+    if (identity) {
+        applyTurnEndNotes(body, {
+            tools: controls.turnEndNote?.tools,
+            mapToolName: toClaudeCodeToolName,
+            visibleText: (message) => visibleReplies.get(message) ?? '',
+        });
+    }
     addEphemeralCacheControl(body);
     applyCacheMode(body, cache.enabled, cache.mode);
     if (identity) {

package/dist/native.d.ts

@@ -1,4 +1,4 @@
-import { type NativeAccountRuntime, type NativeAccountSnapshot, type NativeCustodyClient, type NativeCustodyReceipt, type NativeCustodyReporterSource, type NativeRefreshResult } from '@cortexkit/anthropic-auth-core';
+import { type NativeAccountRuntime, type NativeAccountRuntimeOptions, type NativeAccountSnapshot, type NativeCustodyClient, type NativeCustodyReceipt, type NativeCustodyReporterSource, type NativeRefreshResult } from '@cortexkit/anthropic-auth-core';
 export type PiNativeAttempt = {
     source: 'local';
     accountIdentity: string;
@@ -38,8 +38,14 @@ export declare class PiNativeRuntime {
         publish(write: () => Promise<unknown>): void;
         drain(): Promise<void>;
     };
+    private readonly routingPublications;
+    ownRoutingPublications(router: {
+        drain(): Promise<void>;
+    }): void;
+    drainRoutingPublications(): Promise<void>;
     constructor(storagePath: string, options?: {
         connect?: () => Promise<NativeCustodyClient>;
+        local?: NativeAccountRuntimeOptions['local'];
         vaultPoll?: PiVaultPollOptions;
     });
     service(): Promise<NativeAccountRuntime>;
@@ -68,6 +74,7 @@ export declare class PiNativeRuntime {
 }
 export declare function getPiNativeRuntime(storagePath?: string, options?: {
     connect?: () => Promise<NativeCustodyClient>;
+    local?: NativeAccountRuntimeOptions['local'];
     vaultPoll?: PiVaultPollOptions;
 }): PiNativeRuntime;
 export declare function closePiNativeRuntime(storagePath?: string): Promise<void>;

package/dist/native.js

@@ -66,6 +66,16 @@ export class PiNativeRuntime {
     closed = false;
     closing;
     usagePublications = createNativeUsagePublisher('native-auth');
+    routingPublications = new Set();
+    ownRoutingPublications(router) {
+        this.routingPublications.add(router);
+    }
+    async drainRoutingPublications() {
+        await Promise.all([
+            ...[...this.routingPublications].map((router) => router.drain()),
+            this.initialization.then((runtime) => runtime.drainRoutingPublications(), () => { }),
+        ]);
+    }
     constructor(storagePath, options = {}) {
         const env = { ...process.env };
         this.env = env;
@@ -76,6 +86,7 @@ export class PiNativeRuntime {
             const runtime = acquireNativeAccountRuntime({
                 paths,
                 host: 'pi',
+                local: options.local,
                 vault: {
                     // Core checks Pi enrollment and reads a fresh scoped receipt on every send.
                     // Keep the injected connector's identity rather than wrapping it per instance.
@@ -269,8 +280,11 @@ export class PiNativeRuntime {
         this.vaultWatch?.close();
         this.closing ??= (async () => {
             await this.usagePublications.drain();
+            await this.drainRoutingPublications();
+            this.routingPublications.clear();
             await this.initialization.then(async (runtime) => {
                 await runtime.drainQuotaPublications();
+                await runtime.drainRoutingPublications();
                 runtime.close();
             }, () => { });
         })();
@@ -282,11 +296,15 @@ export function getPiNativeRuntime(storagePath = getPiAccountStoragePath(), opti
     const key = JSON.stringify([
         contextKey,
         nativeRuntimeOptionIdentity(options.connect),
+        nativeRuntimeOptionIdentity(options.local?.onLockEvent),
+        nativeRuntimeOptionIdentity(options.local?.refreshToken),
+        nativeRuntimeOptionIdentity(options.local?.resolveIdentity),
+        nativeRuntimeOptionIdentity(options.local?.now),
     ]);
     const existing = runtimes.get(key);
     if (existing)
         return existing.runtime;
-    if (!options.connect) {
+    if (!options.connect && !options.local) {
         // A supplied connector installs the context's test transport. Ordinary host
         // lookups reuse it, but an explicitly different connector owns another key.
         const configured = [...runtimes.values()].find((entry) => entry.contextKey === contextKey);

package/dist/stream.js

@@ -109,10 +109,16 @@ async function getPiStickyRouter(storagePath) {
     const { path, vaultSeed } = await getPiNativeRuntime(storagePath).routingStorage();
     let router = stickyRouters.get(path);
     if (!router) {
-        router = new StickySessionRouter({ path, vaultSeed });
+        const runtime = await getPiNativeRuntime(storagePath).service();
+        router = new StickySessionRouter({
+            path,
+            vaultSeed,
+            onLockEvent: runtime.onLockEvent,
+        });
         setBoundedService(stickyRouters, path, router);
     }
     router.setVaultSeed(vaultSeed);
+    getPiNativeRuntime(storagePath).ownRoutingPublications(router);
     return router;
 }
 export async function clearPiStickyRoutingSession(storagePath, sessionId) {
@@ -357,6 +363,7 @@ async function sendAnthropicRequest(options) {
         mode: getCache1hPersistentMode(storage),
     }, isFastModePersistentlyEnabled(storage), identity, {
         effortTransitions: options.apiAccount ? [] : options.effortTransitions,
+        turnEndNote: storage.turnEndNote,
         thinkingPrefixMismatchBehavior: options.apiAccount
             ? 'account-default'
             : getThinkingPrefixMismatchBehavior(storage),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cortexkit/pi-anthropic-auth",
-  "version": "2.1.1",
+  "version": "2.2.0",
   "type": "module",
   "repository": {
     "type": "git",
@@ -36,7 +36,7 @@
     "prepublishOnly": "bun run build"
   },
   "dependencies": {
-    "@cortexkit/anthropic-auth-core": "2.1.1"
+    "@cortexkit/anthropic-auth-core": "2.2.0"
   },
   "peerDependencies": {
     "@earendil-works/pi-ai": ">=0.86.1 <2",