From a53796270f3cf7c1291bf901ee18bee03bb19a3b Mon Sep 17 00:00:00 2001
From: "Timothy J. Baek" <timothyjrbeck@gmail.com>
Date: Sat, 1 Jun 2024 20:08:08 -0700
Subject: [PATCH] refac: web search config

---
 backend/apps/rag/main.py                      |  8 ++---
 .../documents/Settings/WebParams.svelte       | 32 ++++++++++++++++++-
 2 files changed, 35 insertions(+), 5 deletions(-)

diff --git a/backend/apps/rag/main.py b/backend/apps/rag/main.py
index 8293acfbb..058c05015 100644
--- a/backend/apps/rag/main.py
+++ b/backend/apps/rag/main.py
@@ -370,7 +370,7 @@ async def get_rag_config(user=Depends(get_admin_user)):
         "web": {
             "ssl_verification": app.state.config.ENABLE_RAG_WEB_LOADER_SSL_VERIFICATION,
             "search": {
-                "enable": app.state.config.ENABLE_RAG_WEB_SEARCH,
+                "enabled": app.state.config.ENABLE_RAG_WEB_SEARCH,
                 "engine": app.state.config.RAG_WEB_SEARCH_ENGINE,
                 "searxng_query_url": app.state.config.SEARXNG_QUERY_URL,
                 "google_pse_api_key": app.state.config.GOOGLE_PSE_API_KEY,
@@ -397,7 +397,7 @@ class YoutubeLoaderConfig(BaseModel):
 
 
 class WebSearchConfig(BaseModel):
-    enable: bool
+    enabled: bool
     engine: Optional[str] = None
     searxng_query_url: Optional[str] = None
     google_pse_api_key: Optional[str] = None
@@ -443,7 +443,7 @@ async def update_rag_config(form_data: ConfigUpdateForm, user=Depends(get_admin_
             form_data.web.web_loader_ssl_verification
         )
 
-        app.state.config.ENABLE_RAG_WEB_SEARCH = form_data.web.search.enable
+        app.state.config.ENABLE_RAG_WEB_SEARCH = form_data.web.search.enabled
         app.state.config.RAG_WEB_SEARCH_ENGINE = form_data.web.search.engine
         app.state.config.SEARXNG_QUERY_URL = form_data.web.search.searxng_query_url
         app.state.config.GOOGLE_PSE_API_KEY = form_data.web.search.google_pse_api_key
@@ -475,7 +475,7 @@ async def update_rag_config(form_data: ConfigUpdateForm, user=Depends(get_admin_
         "web": {
             "ssl_verification": app.state.config.ENABLE_RAG_WEB_LOADER_SSL_VERIFICATION,
             "search": {
-                "enable": app.state.config.ENABLE_RAG_WEB_SEARCH,
+                "enabled": app.state.config.ENABLE_RAG_WEB_SEARCH,
                 "engine": app.state.config.RAG_WEB_SEARCH_ENGINE,
                 "searxng_query_url": app.state.config.SEARXNG_QUERY_URL,
                 "google_pse_api_key": app.state.config.GOOGLE_PSE_API_KEY,
diff --git a/src/lib/components/documents/Settings/WebParams.svelte b/src/lib/components/documents/Settings/WebParams.svelte
index 5b15e3e41..89d4442b0 100644
--- a/src/lib/components/documents/Settings/WebParams.svelte
+++ b/src/lib/components/documents/Settings/WebParams.svelte
@@ -58,7 +58,7 @@
 							{$i18n.t('Enable Web Search')}
 						</div>
 
-						<Switch bind:state={webConfig.search.enable} />
+						<Switch bind:state={webConfig.search.enabled} />
 					</div>
 				</div>
 
@@ -191,6 +191,36 @@
 						{/if}
 					</div>
 				{/if}
+
+				{#if webConfig.search.enabled}
+					<div class="mt-2 flex gap-2 mb-1">
+						<div class="w-full">
+							<div class=" self-center text-xs font-medium mb-1">
+								{$i18n.t('Search Result Count')}
+							</div>
+
+							<input
+								class="w-full rounded-lg py-2 px-4 text-sm dark:text-gray-300 dark:bg-gray-850 outline-none"
+								placeholder={$i18n.t('Search Result Count')}
+								bind:value={webConfig.search.result_count}
+								required
+							/>
+						</div>
+
+						<div class="w-full">
+							<div class=" self-center text-xs font-medium mb-1">
+								{$i18n.t('Concurrent Requests')}
+							</div>
+
+							<input
+								class="w-full rounded-lg py-2 px-4 text-sm dark:text-gray-300 dark:bg-gray-850 outline-none"
+								placeholder={$i18n.t('Concurrent Requests')}
+								bind:value={webConfig.search.concurrent_requests}
+								required
+							/>
+						</div>
+					</div>
+				{/if}
 			</div>
 
 			<hr class=" dark:border-gray-850 my-2" />