Clean up all references to authenticatedData

- De facto, was being assumed everywhere if authenticatedData is null, that it's not logged in. This isn't true because the data can still be loading. Update the hook to send additional states.
- Bonus: Delete model picker code and a slew of unused imports.
This commit is contained in:
sabaimran
2024-12-21 08:45:43 -08:00
parent cc7fd1163f
commit e9dae4240e
15 changed files with 155 additions and 341 deletions

View File

@@ -144,7 +144,6 @@ function CreateAgentCard(props: CreateAgentCardProps) {
<DialogHeader>Create Agent</DialogHeader> <DialogHeader>Create Agent</DialogHeader>
{!props.userProfile && showLoginPrompt && ( {!props.userProfile && showLoginPrompt && (
<LoginPrompt <LoginPrompt
loginRedirectMessage="Sign in to start chatting with a specialized agent"
onOpenChange={setShowLoginPrompt} onOpenChange={setShowLoginPrompt}
isMobileWidth={props.isMobileWidth} isMobileWidth={props.isMobileWidth}
/> />
@@ -174,7 +173,11 @@ export default function Agents() {
const { data, error, mutate } = useSWR<AgentData[]>("agents", agentsFetcher, { const { data, error, mutate } = useSWR<AgentData[]>("agents", agentsFetcher, {
revalidateOnFocus: false, revalidateOnFocus: false,
}); });
const authenticatedData = useAuthenticatedData(); const {
data: authenticatedData,
error: authenticationError,
isLoading: authenticationLoading,
} = useAuthenticatedData();
const { userConfig } = useUserConfig(true); const { userConfig } = useUserConfig(true);
const [showLoginPrompt, setShowLoginPrompt] = useState(false); const [showLoginPrompt, setShowLoginPrompt] = useState(false);
const isMobileWidth = useIsMobileWidth(); const isMobileWidth = useIsMobileWidth();
@@ -297,38 +300,39 @@ export default function Agents() {
<div className={`pt-6 md:pt-8 flex justify-between`}> <div className={`pt-6 md:pt-8 flex justify-between`}>
<h1 className="text-3xl flex items-center">Agents</h1> <h1 className="text-3xl flex items-center">Agents</h1>
<div className="ml-auto float-right border p-2 pt-3 rounded-xl font-bold hover:bg-stone-100 dark:hover:bg-neutral-900"> <div className="ml-auto float-right border p-2 pt-3 rounded-xl font-bold hover:bg-stone-100 dark:hover:bg-neutral-900">
<CreateAgentCard {authenticatedData && (
data={{ <CreateAgentCard
slug: "", data={{
name: "", slug: "",
persona: "", name: "",
color: "", persona: "",
icon: "", color: "",
privacy_level: "private", icon: "",
managed_by_admin: false, privacy_level: "private",
chat_model: "", managed_by_admin: false,
input_tools: [], chat_model: "",
output_modes: [], input_tools: [],
}} output_modes: [],
userProfile={authenticatedData} }}
isMobileWidth={isMobileWidth} userProfile={authenticatedData}
filesOptions={filesData || []} isMobileWidth={isMobileWidth}
modelOptions={userConfig?.chat_model_options || []} filesOptions={filesData || []}
selectedChatModelOption={defaultModelOption?.name || ""} modelOptions={userConfig?.chat_model_options || []}
isSubscribed={isSubscribed} selectedChatModelOption={defaultModelOption?.name || ""}
setAgentChangeTriggered={setAgentChangeTriggered} isSubscribed={isSubscribed}
inputToolOptions={ setAgentChangeTriggered={setAgentChangeTriggered}
agentConfigurationOptions?.input_tools || {} inputToolOptions={
} agentConfigurationOptions?.input_tools || {}
outputModeOptions={ }
agentConfigurationOptions?.output_modes || {} outputModeOptions={
} agentConfigurationOptions?.output_modes || {}
/> }
/>
)}
</div> </div>
</div> </div>
{showLoginPrompt && ( {showLoginPrompt && (
<LoginPrompt <LoginPrompt
loginRedirectMessage="Sign in to start chatting with a specialized agent"
onOpenChange={setShowLoginPrompt} onOpenChange={setShowLoginPrompt}
isMobileWidth={isMobileWidth} isMobileWidth={isMobileWidth}
/> />
@@ -345,53 +349,59 @@ export default function Agents() {
</Alert> </Alert>
<div className="pt-6 md:pt-8"> <div className="pt-6 md:pt-8">
<div className={`${styles.agentList}`}> <div className={`${styles.agentList}`}>
{personalAgents.map((agent) => ( {authenticatedData &&
<AgentCard personalAgents.map((agent) => (
key={agent.slug} <AgentCard
data={agent} key={agent.slug}
userProfile={authenticatedData} data={agent}
isMobileWidth={isMobileWidth} userProfile={authenticatedData}
filesOptions={filesData ?? []} isMobileWidth={isMobileWidth}
selectedChatModelOption={defaultModelOption?.name || ""} filesOptions={filesData ?? []}
isSubscribed={isSubscribed} selectedChatModelOption={
setAgentChangeTriggered={setAgentChangeTriggered} defaultModelOption?.name || ""
modelOptions={userConfig?.chat_model_options || []} }
editCard={true} isSubscribed={isSubscribed}
agentSlug={agentSlug || ""} setAgentChangeTriggered={setAgentChangeTriggered}
inputToolOptions={ modelOptions={userConfig?.chat_model_options || []}
agentConfigurationOptions?.input_tools || {} editCard={true}
} agentSlug={agentSlug || ""}
outputModeOptions={ inputToolOptions={
agentConfigurationOptions?.output_modes || {} agentConfigurationOptions?.input_tools || {}
} }
/> outputModeOptions={
))} agentConfigurationOptions?.output_modes || {}
}
/>
))}
</div> </div>
</div> </div>
<div className="pt-6 md:pt-8"> <div className="pt-6 md:pt-8">
<h2 className="text-2xl">Explore</h2> <h2 className="text-2xl">Explore</h2>
<div className={`${styles.agentList}`}> <div className={`${styles.agentList}`}>
{publicAgents.map((agent) => ( {!authenticationLoading &&
<AgentCard publicAgents.map((agent) => (
key={agent.slug} <AgentCard
data={agent} key={agent.slug}
userProfile={authenticatedData} data={agent}
isMobileWidth={isMobileWidth} userProfile={authenticatedData || null}
editCard={false} isMobileWidth={isMobileWidth}
filesOptions={filesData ?? []} editCard={false}
selectedChatModelOption={defaultModelOption?.name || ""} filesOptions={filesData ?? []}
isSubscribed={isSubscribed} selectedChatModelOption={
setAgentChangeTriggered={setAgentChangeTriggered} defaultModelOption?.name || ""
modelOptions={userConfig?.chat_model_options || []} }
agentSlug={agentSlug || ""} isSubscribed={isSubscribed}
inputToolOptions={ setAgentChangeTriggered={setAgentChangeTriggered}
agentConfigurationOptions?.input_tools || {} modelOptions={userConfig?.chat_model_options || []}
} agentSlug={agentSlug || ""}
outputModeOptions={ inputToolOptions={
agentConfigurationOptions?.output_modes || {} agentConfigurationOptions?.input_tools || {}
} }
/> outputModeOptions={
))} agentConfigurationOptions?.output_modes || {}
}
/>
))}
</div> </div>
</div> </div>
</div> </div>

View File

@@ -980,7 +980,11 @@ function AutomationComponentWrapper(props: AutomationComponentWrapperProps) {
} }
export default function Automations() { export default function Automations() {
const authenticatedData = useAuthenticatedData(); const {
data: authenticatedData,
error: authenticationError,
isLoading: authenticationLoading,
} = useAuthenticatedData();
const { const {
data: personalAutomations, data: personalAutomations,
error, error,
@@ -1068,9 +1072,6 @@ export default function Automations() {
</div> </div>
{showLoginPrompt && ( {showLoginPrompt && (
<LoginPrompt <LoginPrompt
loginRedirectMessage={
"Create an account to make your own automation"
}
onOpenChange={setShowLoginPrompt} onOpenChange={setShowLoginPrompt}
isMobileWidth={isMobileWidth} isMobileWidth={isMobileWidth}
/> />
@@ -1114,7 +1115,7 @@ export default function Automations() {
<Suspense> <Suspense>
<SharedAutomationCard <SharedAutomationCard
isMobileWidth={isMobileWidth} isMobileWidth={isMobileWidth}
authenticatedData={authenticatedData} authenticatedData={authenticatedData || null}
locationData={locationData} locationData={locationData}
isLoggedIn={authenticatedData ? true : false} isLoggedIn={authenticatedData ? true : false}
setShowLoginPrompt={setShowLoginPrompt} setShowLoginPrompt={setShowLoginPrompt}
@@ -1123,7 +1124,8 @@ export default function Automations() {
</Suspense> </Suspense>
{isLoading && <InlineLoading message="booting up your automations" />} {isLoading && <InlineLoading message="booting up your automations" />}
<div className={`${styles.automationsLayout}`}> <div className={`${styles.automationsLayout}`}>
{personalAutomations && {authenticatedData &&
personalAutomations &&
personalAutomations.map((automation) => ( personalAutomations.map((automation) => (
<AutomationsCard <AutomationsCard
isMobileWidth={isMobileWidth} isMobileWidth={isMobileWidth}
@@ -1135,17 +1137,18 @@ export default function Automations() {
setShowLoginPrompt={setShowLoginPrompt} setShowLoginPrompt={setShowLoginPrompt}
/> />
))} ))}
{allNewAutomations.map((automation) => ( {authenticatedData &&
<AutomationsCard allNewAutomations.map((automation) => (
isMobileWidth={isMobileWidth} <AutomationsCard
key={automation.id} isMobileWidth={isMobileWidth}
authenticatedData={authenticatedData} key={automation.id}
automation={automation} authenticatedData={authenticatedData}
locationData={locationData} automation={automation}
isLoggedIn={authenticatedData ? true : false} locationData={locationData}
setShowLoginPrompt={setShowLoginPrompt} isLoggedIn={authenticatedData ? true : false}
/> setShowLoginPrompt={setShowLoginPrompt}
))} />
))}
</div> </div>
<h3 className="text-xl py-4">Explore</h3> <h3 className="text-xl py-4">Explore</h3>
<div className={`${styles.automationsLayout}`}> <div className={`${styles.automationsLayout}`}>
@@ -1154,7 +1157,7 @@ export default function Automations() {
isMobileWidth={isMobileWidth} isMobileWidth={isMobileWidth}
setNewAutomationData={setNewAutomationData} setNewAutomationData={setNewAutomationData}
key={automation.id} key={automation.id}
authenticatedData={authenticatedData} authenticatedData={authenticatedData || null}
automation={automation} automation={automation}
locationData={locationData} locationData={locationData}
isLoggedIn={authenticatedData ? true : false} isLoggedIn={authenticatedData ? true : false}

View File

@@ -193,7 +193,11 @@ export default function Chat() {
timezone: Intl.DateTimeFormat().resolvedOptions().timeZone, timezone: Intl.DateTimeFormat().resolvedOptions().timeZone,
}, },
}; };
const authenticatedData = useAuthenticatedData(); const {
data: authenticatedData,
error: authenticationError,
isLoading: authenticationLoading,
} = useAuthenticatedData();
const isMobileWidth = useIsMobileWidth(); const isMobileWidth = useIsMobileWidth();
useEffect(() => { useEffect(() => {
@@ -425,7 +429,7 @@ export default function Chat() {
<div className={styles.chatBoxBody}> <div className={styles.chatBoxBody}>
<Suspense fallback={<Loading />}> <Suspense fallback={<Loading />}>
<ChatBodyData <ChatBodyData
isLoggedIn={authenticatedData !== null} isLoggedIn={authenticatedData ? true : false}
streamedMessages={messages} streamedMessages={messages}
setStreamedMessages={setMessages} setStreamedMessages={setMessages}
chatOptionsData={chatOptionsData} chatOptionsData={chatOptionsData}

View File

@@ -19,13 +19,11 @@ const fetcher = (url: string) =>
.catch((err) => console.warn(err)); .catch((err) => console.warn(err));
export function useAuthenticatedData() { export function useAuthenticatedData() {
const { data, error } = useSWR<UserProfile>("/api/v1/user", fetcher, { const { data, error, isLoading } = useSWR<UserProfile>("/api/v1/user", fetcher, {
revalidateOnFocus: false, revalidateOnFocus: false,
}); });
if (error || !data || data.detail === "Forbidden") return null; return { data, error, isLoading };
return data;
} }
export interface ModelOptions { export interface ModelOptions {

View File

@@ -326,7 +326,6 @@ export function AgentCard(props: AgentCardProps) {
<Card className={`shadow-md rounded-xl hover:shadow-lg dark:bg-muted`}> <Card className={`shadow-md rounded-xl hover:shadow-lg dark:bg-muted`}>
{showLoginPrompt && ( {showLoginPrompt && (
<LoginPrompt <LoginPrompt
loginRedirectMessage={`Sign in to start chatting with ${props.data.name}`}
onOpenChange={setShowLoginPrompt} onOpenChange={setShowLoginPrompt}
isMobileWidth={props.isMobileWidth} isMobileWidth={props.isMobileWidth}
/> />

View File

@@ -3,7 +3,6 @@
import styles from "./sidePanel.module.css"; import styles from "./sidePanel.module.css";
import { useEffect, useMemo, useState } from "react"; import { useEffect, useMemo, useState } from "react";
import { useRef } from "react";
import { mutate } from "swr"; import { mutate } from "swr";
@@ -102,14 +101,10 @@ import {
} from "@/components/ui/alert-dialog"; } from "@/components/ui/alert-dialog";
import { modifyFileFilterForConversation } from "@/app/common/chatFunctions"; import { modifyFileFilterForConversation } from "@/app/common/chatFunctions";
import { ScrollAreaScrollbar } from "@radix-ui/react-scroll-area"; import { ScrollAreaScrollbar } from "@radix-ui/react-scroll-area";
import { KhojLogoType } from "@/app/components/logo/khojLogo";
import NavMenu from "@/app/components/navMenu/navMenu";
import { getIconFromIconName } from "@/app/common/iconUtils"; import { getIconFromIconName } from "@/app/common/iconUtils";
import LoginPrompt from "../loginPrompt/loginPrompt";
import { import {
SidebarGroup, SidebarGroup,
SidebarGroupLabel, SidebarGroupLabel,
SidebarMenu,
SidebarMenuAction, SidebarMenuAction,
SidebarMenuButton, SidebarMenuButton,
SidebarMenuItem, SidebarMenuItem,
@@ -908,9 +903,12 @@ export default function AllConversations(props: SidePanelProps) {
const [data, setData] = useState<ChatHistory[] | null>(null); const [data, setData] = useState<ChatHistory[] | null>(null);
const [organizedData, setOrganizedData] = useState<GroupedChatHistory | null>(null); const [organizedData, setOrganizedData] = useState<GroupedChatHistory | null>(null);
const [subsetOrganizedData, setSubsetOrganizedData] = useState<GroupedChatHistory | null>(null); const [subsetOrganizedData, setSubsetOrganizedData] = useState<GroupedChatHistory | null>(null);
const [showLoginPrompt, setShowLoginPrompt] = useState(false);
const authenticatedData = useAuthenticatedData(); const {
data: authenticatedData,
error: authenticationError,
isLoading: authenticationLoading,
} = useAuthenticatedData();
const { data: chatSessions, isLoading } = useChatSessionsFetchRequest( const { data: chatSessions, isLoading } = useChatSessionsFetchRequest(
authenticatedData ? `/api/chat/sessions` : "", authenticatedData ? `/api/chat/sessions` : "",
); );

View File

@@ -1,18 +1,14 @@
import { Calendar, Home, Inbox, Search, Settings } from "lucide-react";
import { import {
Sidebar, Sidebar,
SidebarContent, SidebarContent,
SidebarFooter, SidebarFooter,
SidebarGroup, SidebarGroup,
SidebarGroupContent, SidebarGroupContent,
SidebarGroupLabel,
SidebarHeader, SidebarHeader,
SidebarMenu, SidebarMenu,
SidebarMenuButton, SidebarMenuButton,
SidebarMenuItem, SidebarMenuItem,
} from "@/components/ui/sidebar"; } from "@/components/ui/sidebar";
import Link from "next/link";
import { import {
KhojAgentLogo, KhojAgentLogo,
KhojAutomationLogo, KhojAutomationLogo,
@@ -21,12 +17,12 @@ import {
KhojSearchLogo, KhojSearchLogo,
} from "../logo/khojLogo"; } from "../logo/khojLogo";
import { Gear } from "@phosphor-icons/react/dist/ssr"; import { Gear } from "@phosphor-icons/react/dist/ssr";
import { House, Plus } from "@phosphor-icons/react"; import { Plus } from "@phosphor-icons/react";
import { useEffect, useState } from "react"; import { useEffect, useState } from "react";
import { useAuthenticatedData } from "@/app/common/auth";
import AllConversations from "../allConversations/allConversations"; import AllConversations from "../allConversations/allConversations";
import NavMenu from "../navMenu/navMenu"; import NavMenu from "../navMenu/navMenu";
import { useSidebar } from "@/components/ui/sidebar"; import { useSidebar } from "@/components/ui/sidebar";
import { useIsMobileWidth } from "@/app/common/utils";
// Menu items. // Menu items.
const items = [ const items = [
@@ -66,21 +62,11 @@ interface AppSidebarProps {
} }
export function AppSidebar(props: AppSidebarProps) { export function AppSidebar(props: AppSidebarProps) {
const [isMobileWidth, setIsMobileWidth] = useState(false); const isMobileWidth = useIsMobileWidth();
const { state, open, setOpen, openMobile, setOpenMobile, isMobile, toggleSidebar } = const { state, open, setOpen, openMobile, setOpenMobile, isMobile, toggleSidebar } =
useSidebar(); useSidebar();
useEffect(() => {
const handleResize = () => {
setIsMobileWidth(window.innerWidth < 768);
};
handleResize();
window.addEventListener("resize", handleResize);
return () => window.removeEventListener("resize", handleResize);
}, []);
return ( return (
<Sidebar collapsible={"icon"} variant="sidebar"> <Sidebar collapsible={"icon"} variant="sidebar">
<SidebarHeader> <SidebarHeader>

View File

@@ -409,7 +409,6 @@ export const ChatInputArea = forwardRef<HTMLTextAreaElement, ChatInputProps>((pr
<LoginPrompt <LoginPrompt
onOpenChange={setShowLoginPrompt} onOpenChange={setShowLoginPrompt}
isMobileWidth={props.isMobileWidth} isMobileWidth={props.isMobileWidth}
loginRedirectMessage={loginRedirectMessage}
/> />
)} )}
{uploading && ( {uploading && (

View File

@@ -29,7 +29,6 @@ import { Card, CardContent } from "@/components/ui/card";
import { InputOTP, InputOTPGroup, InputOTPSlot } from "@/components/ui/input-otp"; import { InputOTP, InputOTPGroup, InputOTPSlot } from "@/components/ui/input-otp";
export interface LoginPromptProps { export interface LoginPromptProps {
loginRedirectMessage: string;
onOpenChange: (open: boolean) => void; onOpenChange: (open: boolean) => void;
isMobileWidth?: boolean; isMobileWidth?: boolean;
} }

View File

@@ -1,23 +0,0 @@
select.modelPicker {
font-size: small;
padding-top: 0.5rem;
padding-bottom: 0.5rem;
padding-left: 0.75rem;
padding-right: 0.75rem;
border: none;
border-width: 1px;
display: flex;
align-items: center;
height: 2.5rem;
justify-content: space-between;
border-radius: calc(0.5rem - 2px);
}
select.modelPicker:after {
grid-area: select;
justify-self: end;
}
div.modelPicker {
margin-top: 8px;
}

View File

@@ -1,166 +0,0 @@
import { useAuthenticatedData } from "@/app/common/auth";
import React, { useEffect } from "react";
import useSWR from "swr";
import {
AlertDialog,
AlertDialogAction,
AlertDialogCancel,
AlertDialogContent,
AlertDialogDescription,
AlertDialogFooter,
AlertDialogHeader,
AlertDialogTitle,
} from "@/components/ui/alert-dialog";
import styles from "./modelPicker.module.css";
export interface Model {
id: number;
chat_model: string;
}
// Custom fetcher function to fetch options
const fetchOptionsRequest = async (url: string) => {
const response = await fetch(url, {
method: "GET",
headers: {
"Content-Type": "application/json",
},
});
return response.json();
};
export const useOptionsRequest = (url: string) => {
const { data, error } = useSWR<Model[]>(url, fetchOptionsRequest);
return {
data,
isLoading: !error && !data,
isError: error,
};
};
const fetchSelectedModel = async (url: string) => {
const response = await fetch(url, {
method: "GET",
headers: {
"Content-Type": "application/json",
},
});
return response.json();
};
export const useSelectedModel = (url: string) => {
const { data, error } = useSWR<Model>(url, fetchSelectedModel);
return {
data,
isLoading: !error && !data,
isError: error,
};
};
interface ModelPickerProps {
disabled?: boolean;
setModelUsed?: (model: Model) => void;
initialModel?: Model;
}
export const ModelPicker: React.FC<any> = (props: ModelPickerProps) => {
const { data: models } = useOptionsRequest("/api/model/chat/options");
const { data: selectedModel } = useSelectedModel("/api/model/chat");
const [openLoginDialog, setOpenLoginDialog] = React.useState(false);
let userData = useAuthenticatedData();
const setModelUsed = props.setModelUsed;
useEffect(() => {
if (setModelUsed && selectedModel) {
setModelUsed(selectedModel);
}
}, [selectedModel, setModelUsed]);
if (!models) {
return <div>Loading...</div>;
}
function onSelect(model: Model) {
if (!userData) {
setOpenLoginDialog(true);
return;
}
if (props.setModelUsed) {
props.setModelUsed(model);
}
fetch("/api/model/chat" + "?id=" + String(model.id), {
method: "POST",
body: JSON.stringify(model),
})
.then((response) => {
if (!response.ok) {
throw new Error("Failed to select model");
}
})
.catch((error) => {
console.error("Failed to select model", error);
});
}
function isSelected(model: Model) {
if (props.initialModel) {
return model.id === props.initialModel.id;
}
return selectedModel?.id === model.id;
}
return (
<div className={styles.modelPicker}>
<select
className={styles.modelPicker}
onChange={(e) => {
const selectedModelId = Number(e.target.value);
const selectedModel = models.find((model) => model.id === selectedModelId);
if (selectedModel) {
onSelect(selectedModel);
} else {
console.error("Selected model not found", e.target.value);
}
}}
disabled={props.disabled}
>
{models?.map((model) => (
<option key={model.id} value={model.id} selected={isSelected(model)}>
{model.chat_model}
</option>
))}
</select>
<AlertDialog open={openLoginDialog} onOpenChange={setOpenLoginDialog}>
<AlertDialogContent>
<AlertDialogHeader>
<AlertDialogTitle>
You must be logged in to configure your model.
</AlertDialogTitle>
<AlertDialogDescription>
Once you create an account with Khoj, you can configure your model and
use a whole suite of other features. Check out our{" "}
<a href="https://docs.khoj.dev/">documentation</a> to learn more.
</AlertDialogDescription>
</AlertDialogHeader>
<AlertDialogFooter>
<AlertDialogCancel>Cancel</AlertDialogCancel>
<AlertDialogAction
onClick={() => {
window.location.href = window.location.origin + "/login";
}}
>
Sign in
</AlertDialogAction>
</AlertDialogFooter>
</AlertDialogContent>
</AlertDialog>
</div>
);
};

View File

@@ -1,20 +1,10 @@
"use client"; "use client";
import styles from "./navMenu.module.css";
import Link from "next/link"; import Link from "next/link";
import { useAuthenticatedData } from "@/app/common/auth"; import { useAuthenticatedData } from "@/app/common/auth";
import { useState, useEffect } from "react"; import { useState, useEffect } from "react";
import { Avatar, AvatarImage, AvatarFallback } from "@/components/ui/avatar"; import { Avatar, AvatarImage, AvatarFallback } from "@/components/ui/avatar";
import {
Menubar,
MenubarContent,
MenubarItem,
MenubarMenu,
MenubarSeparator,
MenubarTrigger,
} from "@/components/ui/menubar";
import { import {
DropdownMenu, DropdownMenu,
DropdownMenuContent, DropdownMenuContent,
@@ -22,8 +12,7 @@ import {
DropdownMenuSeparator, DropdownMenuSeparator,
DropdownMenuTrigger, DropdownMenuTrigger,
} from "@/components/ui/dropdown-menu"; } from "@/components/ui/dropdown-menu";
import { Moon, Sun, UserCircle, Question, GearFine, ArrowRight, Code } from "@phosphor-icons/react"; import { Moon, Sun, UserCircle, Question, ArrowRight, Code } from "@phosphor-icons/react";
import { KhojAgentLogo, KhojAutomationLogo, KhojSearchLogo } from "../logo/khojLogo";
import { useIsMobileWidth } from "@/app/common/utils"; import { useIsMobileWidth } from "@/app/common/utils";
import LoginPrompt from "../loginPrompt/loginPrompt"; import LoginPrompt from "../loginPrompt/loginPrompt";
import { Button } from "@/components/ui/button"; import { Button } from "@/components/ui/button";
@@ -55,10 +44,13 @@ interface NavMenuProps {
} }
export default function NavMenu({ sideBarIsOpen }: NavMenuProps) { export default function NavMenu({ sideBarIsOpen }: NavMenuProps) {
const userData = useAuthenticatedData(); const {
data: userData,
error: authenticationError,
isLoading: authenticationLoading,
} = useAuthenticatedData();
const [darkMode, setDarkMode] = useState(false); const [darkMode, setDarkMode] = useState(false);
const [initialLoadDone, setInitialLoadDone] = useState(false); const [initialLoadDone, setInitialLoadDone] = useState(false);
const isMobileWidth = useIsMobileWidth();
const [showLoginPrompt, setShowLoginPrompt] = useState(false); const [showLoginPrompt, setShowLoginPrompt] = useState(false);
useEffect(() => { useEffect(() => {
@@ -97,6 +89,12 @@ export default function NavMenu({ sideBarIsOpen }: NavMenuProps) {
return ( return (
<SidebarMenu className="border-none p-0 m-0"> <SidebarMenu className="border-none p-0 m-0">
<SidebarMenuItem className="p-0 m-0"> <SidebarMenuItem className="p-0 m-0">
{showLoginPrompt && (
<LoginPrompt
onOpenChange={(isOpen) => setShowLoginPrompt(isOpen)}
isMobileWidth={useIsMobileWidth()}
/>
)}
<DropdownMenu> <DropdownMenu>
<DropdownMenuTrigger asChild> <DropdownMenuTrigger asChild>
<SidebarMenuButton className="p-0 m-0 rounded-lg" asChild> <SidebarMenuButton className="p-0 m-0 rounded-lg" asChild>

View File

@@ -222,7 +222,6 @@ function ChatBodyData(props: ChatBodyDataProps) {
<LoginPrompt <LoginPrompt
onOpenChange={setShowLoginPrompt} onOpenChange={setShowLoginPrompt}
isMobileWidth={props.isMobileWidth} isMobileWidth={props.isMobileWidth}
loginRedirectMessage={"Login to your second brain"}
/> />
)} )}
{!props.isLoggedIn && ( {!props.isLoggedIn && (
@@ -425,7 +424,11 @@ export default function Home() {
const { userConfig: initialUserConfig, isLoadingUserConfig } = useUserConfig(true); const { userConfig: initialUserConfig, isLoadingUserConfig } = useUserConfig(true);
const [userConfig, setUserConfig] = useState<UserConfig | null>(null); const [userConfig, setUserConfig] = useState<UserConfig | null>(null);
const authenticatedData = useAuthenticatedData(); const {
data: authenticatedData,
error: authenticationError,
isLoading: authenticationLoading,
} = useAuthenticatedData();
const handleConversationIdChange = (newConversationId: string) => { const handleConversationIdChange = (newConversationId: string) => {
setConversationID(newConversationId); setConversationID(newConversationId);
@@ -477,15 +480,17 @@ export default function Home() {
<title>Khoj AI - Your Second Brain</title> <title>Khoj AI - Your Second Brain</title>
<div className={`${styles.chatBox}`}> <div className={`${styles.chatBox}`}>
<div className={`${styles.chatBoxBody}`}> <div className={`${styles.chatBoxBody}`}>
<ChatBodyData {!authenticationLoading && (
isLoggedIn={authenticatedData !== null} <ChatBodyData
chatOptionsData={chatOptionsData} isLoggedIn={authenticatedData ? true : false}
setUploadedFiles={setUploadedFiles} chatOptionsData={chatOptionsData}
isMobileWidth={isMobileWidth} setUploadedFiles={setUploadedFiles}
onConversationIdChange={handleConversationIdChange} isMobileWidth={isMobileWidth}
userConfig={userConfig} onConversationIdChange={handleConversationIdChange}
isLoadingUserConfig={isLoadingUserConfig} userConfig={userConfig}
/> isLoadingUserConfig={isLoadingUserConfig}
/>
)}
</div> </div>
</div> </div>
</div> </div>

View File

@@ -38,7 +38,6 @@ import {
Key, Key,
Palette, Palette,
UserCircle, UserCircle,
FileMagnifyingGlass,
Trash, Trash,
Copy, Copy,
CreditCard, CreditCard,
@@ -499,7 +498,6 @@ enum PhoneNumberValidationState {
} }
export default function SettingsView() { export default function SettingsView() {
const [title, setTitle] = useState("Settings");
const { apiKeys, generateAPIKey, copyAPIKey, deleteAPIKey } = useApiKeys(); const { apiKeys, generateAPIKey, copyAPIKey, deleteAPIKey } = useApiKeys();
const { userConfig: initialUserConfig } = useUserConfig(true); const { userConfig: initialUserConfig } = useUserConfig(true);
const [userConfig, setUserConfig] = useState<UserConfig | null>(null); const [userConfig, setUserConfig] = useState<UserConfig | null>(null);
@@ -514,6 +512,8 @@ export default function SettingsView() {
const { toast } = useToast(); const { toast } = useToast();
const isMobileWidth = useIsMobileWidth(); const isMobileWidth = useIsMobileWidth();
const title = "Settings";
const cardClassName = const cardClassName =
"w-full lg:w-1/3 grid grid-flow-column border border-gray-300 shadow-md rounded-lg border dark:border-none dark:bg-muted border-opacity-50"; "w-full lg:w-1/3 grid grid-flow-column border border-gray-300 shadow-md rounded-lg border dark:border-none dark:bg-muted border-opacity-50";

View File

@@ -123,7 +123,11 @@ export default function SharedChat() {
const [paramSlug, setParamSlug] = useState<string | undefined>(undefined); const [paramSlug, setParamSlug] = useState<string | undefined>(undefined);
const [images, setImages] = useState<string[]>([]); const [images, setImages] = useState<string[]>([]);
const authenticatedData = useAuthenticatedData(); const {
data: authenticatedData,
error: authenticationError,
isLoading: authenticationLoading,
} = useAuthenticatedData();
const isMobileWidth = useIsMobileWidth(); const isMobileWidth = useIsMobileWidth();
useEffect(() => { useEffect(() => {
@@ -222,7 +226,7 @@ export default function SharedChat() {
conversationId={conversationId} conversationId={conversationId}
streamedMessages={messages} streamedMessages={messages}
setQueryToProcess={setQueryToProcess} setQueryToProcess={setQueryToProcess}
isLoggedIn={authenticatedData !== null} isLoggedIn={authenticatedData ? true : false}
publicConversationSlug={paramSlug} publicConversationSlug={paramSlug}
chatOptionsData={chatOptionsData} chatOptionsData={chatOptionsData}
setTitle={setTitle} setTitle={setTitle}