소스 검색

refac: web search config

Timothy J. Baek 1 년 전
부모
커밋
a53796270f
2개의 변경된 파일35개의 추가작업 그리고 5개의 파일을 삭제
  1. 4 4
      backend/apps/rag/main.py
  2. 31 1
      src/lib/components/documents/Settings/WebParams.svelte

+ 4 - 4
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,

+ 31 - 1
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" />