From d98051862d1c45a01f1dce116f0ac17a3db0f26c Mon Sep 17 00:00:00 2001
From: Jun Siang Cheah <me@jscheah.me>
Date: Sun, 12 May 2024 16:05:00 +0800
Subject: [PATCH] feat: mark websearch docs differently from standard docs

---
 .../chat/Messages/UserMessage.svelte          | 32 +++++++++++++++++++
 src/routes/(app)/+page.svelte                 |  6 ++--
 src/routes/(app)/c/[id]/+page.svelte          | 17 +++++-----
 3 files changed, 44 insertions(+), 11 deletions(-)

diff --git a/src/lib/components/chat/Messages/UserMessage.svelte b/src/lib/components/chat/Messages/UserMessage.svelte
index f8746e1f2..d46043189 100644
--- a/src/lib/components/chat/Messages/UserMessage.svelte
+++ b/src/lib/components/chat/Messages/UserMessage.svelte
@@ -159,6 +159,38 @@
 										<div class=" text-gray-500 text-sm">{$i18n.t('Collection')}</div>
 									</div>
 								</button>
+							{:else if file.type === 'websearch'}
+								<button
+									class="h-16 w-[15rem] flex items-center space-x-3 px-2.5 dark:bg-gray-600 rounded-xl border border-gray-200 dark:border-none text-left"
+									type="button"
+								>
+									<div class="p-2.5 bg-red-400 text-white rounded-lg">
+										<svg
+											xmlns="http://www.w3.org/2000/svg"
+											viewBox="0 0 24 24"
+											fill="currentColor"
+											class="w-6 h-6"
+										>
+											<path d="M11.625 16.5a1.875 1.875 0 1 0 0-3.75 1.875 1.875 0 0 0 0 3.75Z" />
+											<path
+												fill-rule="evenodd"
+												d="M5.625 1.5H9a3.75 3.75 0 0 1 3.75 3.75v1.875c0 1.036.84 1.875 1.875 1.875H16.5a3.75 3.75 0 0 1 3.75 3.75v7.875c0 1.035-.84 1.875-1.875 1.875H5.625a1.875 1.875 0 0 1-1.875-1.875V3.375c0-1.036.84-1.875 1.875-1.875Zm6 16.5c.66 0 1.277-.19 1.797-.518l1.048 1.048a.75.75 0 0 0 1.06-1.06l-1.047-1.048A3.375 3.375 0 1 0 11.625 18Z"
+												clip-rule="evenodd"
+											/>
+											<path
+												d="M14.25 5.25a5.23 5.23 0 0 0-1.279-3.434 9.768 9.768 0 0 1 6.963 6.963A5.23 5.23 0 0 0 16.5 7.5h-1.875a.375.375 0 0 1-.375-.375V5.25Z"
+											/>
+										</svg>
+									</div>
+
+									<div class="flex flex-col justify-center -space-y-0.5">
+										<div class=" dark:text-gray-100 text-sm font-medium line-clamp-1">
+											{file.name}
+										</div>
+
+										<div class=" text-gray-500 text-sm">{$i18n.t('Search Results')}</div>
+									</div>
+								</button>
 							{/if}
 						</div>
 					{/each}
diff --git a/src/routes/(app)/+page.svelte b/src/routes/(app)/+page.svelte
index ee06b97b9..e85b6ee80 100644
--- a/src/routes/(app)/+page.svelte
+++ b/src/routes/(app)/+page.svelte
@@ -334,7 +334,7 @@
 		parentMessage.files.push({
 			collection_name: searchDocument!.collection_name,
 			name: searchQuery,
-			type: 'doc',
+			type: 'websearch',
 			upload_status: true,
 			error: ''
 		});
@@ -401,7 +401,7 @@
 		const docs = messages
 			.filter((message) => message?.files ?? null)
 			.map((message) =>
-				message.files.filter((item) => item.type === 'doc' || item.type === 'collection')
+				message.files.filter((item) => ['doc', 'collection', 'websearch'].includes(item.type))
 			)
 			.flat(1);
 
@@ -594,7 +594,7 @@
 		const docs = messages
 			.filter((message) => message?.files ?? null)
 			.map((message) =>
-				message.files.filter((item) => item.type === 'doc' || item.type === 'collection')
+				message.files.filter((item) => ['doc', 'collection', 'websearch'].includes(item.type))
 			)
 			.flat(1);
 
diff --git a/src/routes/(app)/c/[id]/+page.svelte b/src/routes/(app)/c/[id]/+page.svelte
index 17a830741..c88b627a08 100644
--- a/src/routes/(app)/c/[id]/+page.svelte
+++ b/src/routes/(app)/c/[id]/+page.svelte
@@ -30,7 +30,11 @@
 		getTagsById,
 		updateChatById
 	} from '$lib/apis/chats';
-	import { generateOpenAIChatCompletion, generateSearchQuery, generateTitle } from '$lib/apis/openai';
+	import {
+		generateOpenAIChatCompletion,
+		generateSearchQuery,
+		generateTitle
+	} from '$lib/apis/openai';
 
 	import MessageInput from '$lib/components/chat/MessageInput.svelte';
 	import Messages from '$lib/components/chat/Messages.svelte';
@@ -321,10 +325,7 @@
 		}
 		responseMessage.progress = $i18n.t("Searching the web for '{{searchQuery}}'", { searchQuery });
 		messages = messages;
-		const searchDocument = await runWebSearch(
-			localStorage.token,
-			searchQuery,
-		);
+		const searchDocument = await runWebSearch(localStorage.token, searchQuery);
 		if (!searchDocument) {
 			toast.warning($i18n.t('No search results found'));
 			responseMessage.progress = undefined;
@@ -338,7 +339,7 @@
 		parentMessage.files.push({
 			collection_name: searchDocument!.collection_name,
 			name: searchQuery,
-			type: 'doc',
+			type: 'websearch',
 			upload_status: true,
 			error: ''
 		});
@@ -405,7 +406,7 @@
 		const docs = messages
 			.filter((message) => message?.files ?? null)
 			.map((message) =>
-				message.files.filter((item) => item.type === 'doc' || item.type === 'collection')
+				message.files.filter((item) => ['doc', 'collection', 'websearch'].includes(item.type))
 			)
 			.flat(1);
 
@@ -598,7 +599,7 @@
 		const docs = messages
 			.filter((message) => message?.files ?? null)
 			.map((message) =>
-				message.files.filter((item) => item.type === 'doc' || item.type === 'collection')
+				message.files.filter((item) => ['doc', 'collection', 'websearch'].includes(item.type))
 			)
 			.flat(1);