@cortexkit/pi-anthropic-auth 2.1.0 → 2.1.2

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) {

package/package.json

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