Update files

packages/app/src/app/agent/[serverId]/[agentId].tsx

@@ -562,7 +562,7 @@ function AgentScreenContent({
 
   if (!agent) {
     return (
-      <View style={styles.container}>
+      <View style={styles.container} testID="agent-not-found">
         <MenuHeader title="Agent" />
         <View style={styles.errorContainer}>
           <Text style={styles.errorText}>Agent not found</Text>

packages/app/src/components/git-diff-pane.tsx

@@ -1,4 +1,5 @@
 import { useState, useCallback, useEffect, useId, useMemo, useRef, memo, type ReactElement } from "react";
+import { useRouter } from "expo-router";
 import {
   View,
   Text,
@@ -13,7 +14,6 @@ import { ScrollView, type ScrollView as ScrollViewType } from "react-native-gest
 import { StyleSheet, useUnistyles } from "react-native-unistyles";
 import { ChevronRight, GitBranch } from "lucide-react-native";
 import { useMutation, useQueryClient } from "@tanstack/react-query";
-import { useRouter } from "expo-router";
 import { useSessionStore } from "@/stores/session-store";
 import {
   useCheckoutDiffQuery,
@@ -557,7 +557,6 @@ export function GitDiffPane({ serverId, agentId }: GitDiffPaneProps) {
         predicate: (query) =>
           Array.isArray(query.queryKey) && query.queryKey[0] === "paseoWorktreeList",
       });
-      // Archiving a worktree also archives all its agents, so navigate to new agent screen
       router.replace("/");
     },
     onError: (err) => {