From 190153fbc25d89ef685a0f56ba3f9dedb8d00ca3 Mon Sep 17 00:00:00 2001
From: Willie Zutz <boarder2@bit-shift.com>
Date: Tue, 24 Jun 2025 00:10:25 -0600
Subject: [PATCH] feat: Update web search agent to return top 12 results and
 enhance message box interactivity

---
 src/app/api/chat/route.ts                          |  1 +
 src/components/MessageBox.tsx                      |  2 +-
 .../MessageInputActions/Optimization.tsx           | 14 +++++++-------
 src/lib/agents/webSearchAgent.ts                   |  6 +++---
 4 files changed, 12 insertions(+), 11 deletions(-)

diff --git a/src/app/api/chat/route.ts b/src/app/api/chat/route.ts
index 68f346b..3a17bcf 100644
--- a/src/app/api/chat/route.ts
+++ b/src/app/api/chat/route.ts
@@ -425,6 +425,7 @@ export const POST = async (req: Request) => {
           }),
         ),
       );
+      writer.close();
       cleanupCancelToken(message.messageId);
     });
 
diff --git a/src/components/MessageBox.tsx b/src/components/MessageBox.tsx
index 793ff8f..c28686e 100644
--- a/src/components/MessageBox.tsx
+++ b/src/components/MessageBox.tsx
@@ -93,7 +93,7 @@ const MessageBox = ({
           ) : (
             <>
               <div className="flex items-center">
-                <h2 className="text-black dark:text-white font-medium text-3xl">
+                <h2 className="text-black dark:text-white font-medium text-3xl" onClick={startEditMessage}>
                   {message.content}
                 </h2>
                 <button
diff --git a/src/components/MessageInputActions/Optimization.tsx b/src/components/MessageInputActions/Optimization.tsx
index 5ac823f..23869da 100644
--- a/src/components/MessageInputActions/Optimization.tsx
+++ b/src/components/MessageInputActions/Optimization.tsx
@@ -15,13 +15,13 @@ const OptimizationModes = [
       'Prioritize speed and get the quickest possible answer. Minimum effort retrieving web content.',
     icon: <Zap size={20} className="text-[#FF9800]" />,
   },
-  {
-    key: 'balanced',
-    title: 'Balanced',
-    description:
-      'Find the right balance between speed and accuracy. Medium effort retrieving web content.',
-    icon: <Sliders size={20} className="text-[#4CAF50]" />,
-  },
+  // {
+  //   key: 'balanced',
+  //   title: 'Balanced',
+  //   description:
+  //     'Find the right balance between speed and accuracy. Medium effort retrieving web content.',
+  //   icon: <Sliders size={20} className="text-[#4CAF50]" />,
+  // },
   // {
   //   key: 'quality',
   //   title: 'Quality',
diff --git a/src/lib/agents/webSearchAgent.ts b/src/lib/agents/webSearchAgent.ts
index cf2a7ce..de973b9 100644
--- a/src/lib/agents/webSearchAgent.ts
+++ b/src/lib/agents/webSearchAgent.ts
@@ -175,10 +175,10 @@ export class WebSearchAgent {
         }),
       );
 
-      // Sort by relevance score and take top 8 results
+      // Sort by relevance score and take top 12 results
       const previewContents: PreviewContent[] = resultsWithSimilarity
         .sort((a, b) => b.similarity - a.similarity)
-        .slice(0, 8)
+        .slice(0, 12)
         .map(({ result }) => ({
           title: result.title || 'Untitled',
           snippet: result.content || '',
@@ -301,7 +301,7 @@ export class WebSearchAgent {
 
         // Summarize the top 2 search results
         for (const result of resultsWithSimilarity
-          .slice(0, 8)
+          .slice(0, 12)
           .map((r) => r.result)) {
           if (this.signal.aborted) {
             console.warn('Search operation aborted by signal');