diff --git a/frontend/app/api/search/route.ts b/frontend/app/api/search/route.ts
index ac76a4a..7a852a2 100644
--- a/frontend/app/api/search/route.ts
+++ b/frontend/app/api/search/route.ts
@@ -19,16 +19,13 @@ const updateSource = function (model, source, messages, isSearch) {
if (model === O1_MIMI || model === O1_PREVIEW) {
return SearchCategory.O1;
}
+ const file = messages[0].attachments?.[0];
+ if (file && file.startsWith('local-')) {
+ return SearchCategory.KNOWLEDGE_BASE;
+ }
if (!isSearch) {
return SearchCategory.CHAT;
}
- const file = messages[0].attachments?.[0];
- if (file) {
- if (file.startsWith('local-')) {
- return SearchCategory.KNOWLEDGE_BASE;
- }
- return SearchCategory.ALL;
- }
const query = messages[messages.length - 1].content;
if (containsValidUrl(query)) {
return SearchCategory.ALL;
diff --git a/frontend/components/layout/user-account-nav.tsx b/frontend/components/layout/user-account-nav.tsx
index 96df5f6..d2d005d 100644
--- a/frontend/components/layout/user-account-nav.tsx
+++ b/frontend/components/layout/user-account-nav.tsx
@@ -1,7 +1,7 @@
'use client';
import Link from 'next/link';
-import { Gem, Images, LogOut, Settings } from 'lucide-react';
+import { Gem, Images, LogOut, Settings, SquareLibrary } from 'lucide-react';
import type { User } from 'next-auth';
import { signOut } from 'next-auth/react';
import { DropdownMenu, DropdownMenuContent, DropdownMenuItem, DropdownMenuSeparator, DropdownMenuTrigger } from '@/components/ui/dropdown-menu';
@@ -50,9 +50,9 @@ export function UserAccountNav({ user }: UserAccountNavProps) {
-
-
- Images
+
+
+ Upgrade Plan
@@ -62,9 +62,15 @@ export function UserAccountNav({ user }: UserAccountNavProps) {
-
-
- Upgrade Plan
+
+
+ Indexes
+
+
+
+
+
+ Images