🐛 Page display logic fix in speed mode and page jump logic fix in full mode

Author: Phinease

frontend/app/[locale]/chat/components/chatHeader.tsx

@@ -50,8 +50,8 @@ export function ChatHeader({ title, onRename }: ChatHeaderProps) {
   const inputRef = useRef<HTMLInputElement>(null);
   const { t } = useTranslation("common");
   const { currentLanguage, handleLanguageChange } = useLanguageSwitch();
-  const { user } = useAuth();
-  const isAdmin = user?.role === USER_ROLES.ADMIN;
+  const { user, isSpeedMode } = useAuth();
+  const isAdmin = isSpeedMode || user?.role === USER_ROLES.ADMIN;
 
   const goToModelSetup = () => {
     router.push(`/${currentLanguage}/setup/models`);

frontend/app/[locale]/chat/internal/chatInterface.tsx

@@ -61,7 +61,7 @@ const getI18nKeyByType = (type: string): string => {
 
 export function ChatInterface() {
   const router = useRouter();
-  const { user } = useAuth(); // Get user information
+  const { user, isSpeedMode } = useAuth(); // Get user information
   const [input, setInput] = useState("");
   // Replace the original messages state
   const [sessionMessages, setSessionMessages] = useState<{
@@ -1578,8 +1578,8 @@ export function ChatInterface() {
     // Both admin and regular users now use dropdown menus
   };
 
-  // Settings menu items based on user role
-  const settingsMenuItems = user?.role === "admin" ? [
+  // Settings menu items based on user role (speed mode is treated as admin)
+  const settingsMenuItems = (isSpeedMode || user?.role === "admin") ? [
     // Admin has three options
     {
       key: "models",

frontend/app/[locale]/chat/internal/memory/memoryManageModal.tsx

@@ -45,10 +45,10 @@ const MemoryManageModal: React.FC<MemoryManageModalProps> = ({
   userRole,
 }) => {
   // Get user role from authentication context
-  const { user } = useAuth();
+  const { user, isSpeedMode } = useAuth();
   const { message } = App.useApp();
   const role: (typeof USER_ROLES)[keyof typeof USER_ROLES] = (userRole ??
-    (user?.role === USER_ROLES.ADMIN ? USER_ROLES.ADMIN : USER_ROLES.USER)) as (typeof USER_ROLES)[keyof typeof USER_ROLES];
+    (isSpeedMode || user?.role === USER_ROLES.ADMIN ? USER_ROLES.ADMIN : USER_ROLES.USER)) as (typeof USER_ROLES)[keyof typeof USER_ROLES];
 
   // Get user role from other hooks / context
   const currentUserId = "user1";

frontend/app/[locale]/page.tsx

@@ -77,7 +77,7 @@ export default function Home() {
 
     // Handle operations that require login
     const handleAuthRequired = (e: React.MouseEvent) => {
-      if (!user) {
+      if (!isSpeedMode && !user) {
         e.preventDefault();
         setLoginPromptOpen(true);
       }
@@ -102,7 +102,7 @@ export default function Home() {
 
     // Handle operations that require admin privileges
     const handleAdminRequired = (e: React.MouseEvent) => {
-      if (user?.role !== "admin") {
+      if (!isSpeedMode && user?.role !== "admin") {
         e.preventDefault();
         setAdminRequiredPromptOpen(true);
       }
@@ -244,15 +244,15 @@ export default function Home() {
               transition={{ duration: 0.8, delay: 0.4 }}
               className="flex flex-col sm:flex-row gap-4"
             >
-              <Link href={user ? "/chat" : "#"} onClick={handleAuthRequired}>
+              <Link href={isSpeedMode || user ? "/chat" : "#"} onClick={handleAuthRequired}>
                 <Button className="bg-blue-600 hover:bg-blue-700 text-white px-8 py-6 rounded-full text-lg font-medium shadow-lg hover:shadow-xl transition-all duration-300 group">
                   <Bot className="mr-2 h-5 w-5 group-hover:animate-pulse" />
                   {t("page.startChat")}
                 </Button>
               </Link>
 
               <Link
-                href={user?.role === "admin" ? "/setup" : "#"}
+                href={isSpeedMode || user?.role === "admin" ? "/setup" : "#"}
                 onClick={handleAdminRequired}
               >
                 <Button className="bg-blue-600 hover:bg-blue-700 text-white px-8 py-6 rounded-full text-lg font-medium shadow-lg hover:shadow-xl transition-all duration-300 group">

frontend/app/[locale]/setup/agents/page.tsx

@@ -108,6 +108,11 @@ export default function AgentSetupPage() {
     duration: 0.4,
   };
 
+  // Prevent rendering if user doesn't have permission (full mode)
+  if (!isSpeedMode && !userLoading && (!user || user.role !== USER_ROLES.ADMIN)) {
+    return null;
+  }
+
   return (
     <SetupLayout
       connectionStatus={connectionStatus}

frontend/app/[locale]/setup/knowledges/page.tsx

@@ -123,7 +123,7 @@ export default function KnowledgeSetupPage() {
 
   // Handle back button click
   const handleBack = () => {
-    if (user?.role === USER_ROLES.ADMIN) {
+    if (isSpeedMode || user?.role === USER_ROLES.ADMIN) {
       router.push("/setup/models");
     } else {
       message.error(t("setup.page.error.adminOnly"));
@@ -152,8 +152,8 @@ export default function KnowledgeSetupPage() {
     duration: 0.4,
   };
 
-  // Determine which button to show based on user role
-  const isAdmin = user?.role === USER_ROLES.ADMIN;
+  // Determine which button to show based on user role (speed mode is treated as admin)
+  const isAdmin = isSpeedMode || user?.role === USER_ROLES.ADMIN;
 
   return (
     <SetupLayout

frontend/app/[locale]/setup/models/page.tsx

@@ -12,6 +12,7 @@ import modelEngineService from "@/services/modelEngineService";
 import {configStore} from "@/lib/config";
 import { configService } from "@/services/configService";
 import {
+  USER_ROLES,
   CONNECTION_STATUS,
   ConnectionStatus,
   MODEL_STATUS,
@@ -51,7 +52,13 @@ export default function ModelSetupPage() {
       openLoginModal();
       return;
     }
-  }, [isSpeedMode, user, userLoading, openLoginModal]);
+
+    // Only admin users can access this page (full mode)
+    if (!isSpeedMode && user && user.role !== USER_ROLES.ADMIN) {
+      router.push("/setup/knowledges");
+      return;
+    }
+  }, [isSpeedMode, user, userLoading, openLoginModal, router]);
 
   // Check the connection status when the page is initialized
   useEffect(() => {
@@ -206,6 +213,11 @@ export default function ModelSetupPage() {
     duration: 0.4,
   };
 
+  // Prevent rendering if user doesn't have permission (full mode)
+  if (!isSpeedMode && !userLoading && (!user || user.role !== USER_ROLES.ADMIN)) {
+    return null;
+  }
+
   return (
     <SetupLayout
       connectionStatus={connectionStatus}

frontend/app/[locale]/setup/page.tsx

@@ -7,19 +7,25 @@ import {USER_ROLES} from "@/const/modelConfig";
 
 export default function SetupPage() {
   const router = useRouter();
-  const { user, isLoading: userLoading } = useAuth();
+  const { user, isLoading: userLoading, isSpeedMode } = useAuth();
 
   useEffect(() => {
-    if (!userLoading && user) {
-      // Redirect based on user role
-      if (user.role === USER_ROLES.ADMIN) {
+    if (!userLoading) {
+      if (isSpeedMode) {
+        // In speed mode, go directly to models page
         router.push("/setup/models");
+      } else if (user) {
+        // Redirect based on user role
+        if (user.role === USER_ROLES.ADMIN) {
+          router.push("/setup/models");
+        } else {
+          router.push("/setup/knowledges");
+        }
       } else {
-        router.push("/setup/knowledges");
+        // Full mode without login, redirect to home
+        router.push("/");
       }
-    } else if (!userLoading && !user) {
-      router.push("/");
     }
-  }, [user, userLoading, router]);
+  }, [user, userLoading, isSpeedMode, router]);
   return null;
 }