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