From 9d1b9f3820fa272e27606bec1ea0938aebbc8aca Mon Sep 17 00:00:00 2001
From: miloschwartz <mschwartz10612@gmail.com>
Date: Fri, 25 Sep 2026 19:59:13 -0400
Subject: [PATCH] add mobile ai icon

---
 components/header-search.tsx | 21 +++++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/components/header-search.tsx b/components/header-search.tsx
index f5774b4..991d732 100644
--- a/components/header-search.tsx
+++ b/components/header-search.tsx
@@ -7,6 +7,7 @@ import {
   type SearchTriggerProps,
 } from 'fumadocs-ui/layouts/shared/slots/search-trigger';
 import { cn } from '@/lib/cn';
+import { buttonVariants } from '@/components/ui/button';
 import { useAISearchContext } from './ai/search';
 
 /**
@@ -33,8 +34,24 @@ function HeaderSearchFull({ className, ...props }: FullSearchTriggerProps) {
   );
 }
 
-function HeaderSearchSm(props: SearchTriggerProps) {
-  return <SearchTrigger {...props} />;
+function HeaderSearchSm({ className, ...props }: SearchTriggerProps) {
+  const { open, setOpen } = useAISearchContext();
+
+  return (
+    <>
+      <SearchTrigger {...props} className={className} />
+      <button
+        type="button"
+        className={cn(buttonVariants({ variant: 'ghost', size: 'icon-sm' }), 'p-2')}
+        aria-label="Ask AI"
+        aria-pressed={open}
+        data-state={open ? 'open' : 'closed'}
+        onClick={() => setOpen(!open)}
+      >
+        <Sparkles />
+      </button>
+    </>
+  );
 }
 
 export const headerSearchSlot = { full: HeaderSearchFull, sm: HeaderSearchSm };