fix(auth): preserve auto-pin preference

Co-authored-by: Mykyta Bozhenko <21245729+cheeeee@users.noreply.github.com>

src/auto-reply/reply/agent-runner-execution.ts

@@ -200,6 +200,10 @@ export async function runAgentTurnWithFallback(params: {
                 throw err;
               });
           }
+          const authProfileId =
+            provider === params.followupRun.run.provider
+              ? params.followupRun.run.authProfileId
+              : undefined;
           return runEmbeddedPiAgent({
             sessionId: params.followupRun.run.sessionId,
             sessionKey: params.sessionKey,
@@ -222,7 +226,10 @@ export async function runAgentTurnWithFallback(params: {
             enforceFinalTag: resolveEnforceFinalTag(params.followupRun.run, provider),
             provider,
             model,
-            authProfileId: params.followupRun.run.authProfileId,
+            authProfileId,
+            authProfileIdSource: authProfileId
+              ? params.followupRun.run.authProfileIdSource
+              : undefined,
             thinkLevel: params.followupRun.run.thinkLevel,
             verboseLevel: params.followupRun.run.verboseLevel,
             reasoningLevel: params.followupRun.run.reasoningLevel,

src/auto-reply/reply/agent-runner-memory.ts

@@ -96,8 +96,12 @@ export async function runMemoryFlushIfNeeded(params: {
         params.followupRun.run.config,
         resolveAgentIdFromSessionKey(params.followupRun.run.sessionKey),
       ),
-      run: (provider, model) =>
-        runEmbeddedPiAgent({
+      run: (provider, model) => {
+        const authProfileId =
+          provider === params.followupRun.run.provider
+            ? params.followupRun.run.authProfileId
+            : undefined;
+        return runEmbeddedPiAgent({
           sessionId: params.followupRun.run.sessionId,
           sessionKey: params.sessionKey,
           messageProvider: params.sessionCtx.Provider?.trim().toLowerCase() || undefined,
@@ -119,7 +123,10 @@ export async function runMemoryFlushIfNeeded(params: {
           enforceFinalTag: resolveEnforceFinalTag(params.followupRun.run, provider),
           provider,
           model,
-          authProfileId: params.followupRun.run.authProfileId,
+          authProfileId,
+          authProfileIdSource: authProfileId
+            ? params.followupRun.run.authProfileIdSource
+            : undefined,
           thinkLevel: params.followupRun.run.thinkLevel,
           verboseLevel: params.followupRun.run.verboseLevel,
           reasoningLevel: params.followupRun.run.reasoningLevel,
@@ -136,7 +143,8 @@ export async function runMemoryFlushIfNeeded(params: {
               }
             }
           },
-        }),
+        });
+      },
     });
     let memoryFlushCompactionCount =
       activeSessionEntry?.compactionCount ??

src/auto-reply/reply/followup-runner.ts

@@ -138,8 +138,10 @@ export function createFollowupRunner(params: {
             queued.run.config,
             resolveAgentIdFromSessionKey(queued.run.sessionKey),
           ),
-          run: (provider, model) =>
-            runEmbeddedPiAgent({
+          run: (provider, model) => {
+            const authProfileId =
+              provider === queued.run.provider ? queued.run.authProfileId : undefined;
+            return runEmbeddedPiAgent({
               sessionId: queued.run.sessionId,
               sessionKey: queued.run.sessionKey,
               messageProvider: queued.run.messageProvider,
@@ -154,7 +156,8 @@ export function createFollowupRunner(params: {
               enforceFinalTag: queued.run.enforceFinalTag,
               provider,
               model,
-              authProfileId: queued.run.authProfileId,
+              authProfileId,
+              authProfileIdSource: authProfileId ? queued.run.authProfileIdSource : undefined,
               thinkLevel: queued.run.thinkLevel,
               verboseLevel: queued.run.verboseLevel,
               reasoningLevel: queued.run.reasoningLevel,
@@ -171,7 +174,8 @@ export function createFollowupRunner(params: {
                   autoCompactionCompleted = true;
                 }
               },
-            }),
+            });
+          },
         });
         runResult = fallbackResult.result;
         fallbackProvider = fallbackResult.provider;

src/auto-reply/reply/get-reply-run.ts

@@ -152,7 +152,13 @@ async function resolveSessionAuthProfileOverride(params: {
   let current = sessionEntry.authProfileOverride?.trim();
   if (current && !order.includes(current)) current = undefined;
 
-  const source = sessionEntry.authProfileOverrideSource ?? (current ? "user" : undefined);
+  const source =
+    sessionEntry.authProfileOverrideSource ??
+    (typeof sessionEntry.authProfileOverrideCompactionCount === "number"
+      ? "auto"
+      : current
+        ? "user"
+        : undefined);
   if (source === "user" && current && !isNewSession) {
     return current;
   }
@@ -406,6 +412,7 @@ export async function runPreparedReply(
     storePath,
     isNewSession,
   });
+  const authProfileIdSource = sessionEntry?.authProfileOverrideSource;
   const followupRun = {
     prompt: queuedBody,
     messageId: sessionCtx.MessageSid,
@@ -430,6 +437,7 @@ export async function runPreparedReply(
       provider,
       model,
       authProfileId,
+      authProfileIdSource,
       thinkLevel: resolvedThinkLevel,
       verboseLevel: resolvedVerboseLevel,
       reasoningLevel: resolvedReasoningLevel,

src/auto-reply/reply/queue/types.ts

@@ -53,6 +53,7 @@ export type FollowupRun = {
     provider: string;
     model: string;
     authProfileId?: string;
+    authProfileIdSource?: "auto" | "user";
     thinkLevel?: ThinkLevel;
     verboseLevel?: VerboseLevel;
     reasoningLevel?: ReasoningLevel;