UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-16 05:24:02 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refac: web search config

Browse Source
This commit is contained in:
Timothy J. Baek 2024-06-01 20:08:08 -07:00
parent fbdfb7e4fa
commit a53796270f
2 changed files with 35 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
backend/apps/rag/main.py
View File

@ -370,7 +370,7 @@ async def get_rag_config(user=Depends(get_admin_user)):
"web": { "web": {
"ssl_verification": app.state.config.ENABLE_RAG_WEB_LOADER_SSL_VERIFICATION, "ssl_verification": app.state.config.ENABLE_RAG_WEB_LOADER_SSL_VERIFICATION,
"search": { "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, "engine": app.state.config.RAG_WEB_SEARCH_ENGINE,
"searxng_query_url": app.state.config.SEARXNG_QUERY_URL, "searxng_query_url": app.state.config.SEARXNG_QUERY_URL,
"google_pse_api_key": app.state.config.GOOGLE_PSE_API_KEY, "google_pse_api_key": app.state.config.GOOGLE_PSE_API_KEY,
@ -397,7 +397,7 @@ class YoutubeLoaderConfig(BaseModel):
class WebSearchConfig(BaseModel): class WebSearchConfig(BaseModel):
enable: bool enabled: bool
engine: Optional[str] = None engine: Optional[str] = None
searxng_query_url: Optional[str] = None searxng_query_url: Optional[str] = None
google_pse_api_key: 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 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.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.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 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": { "web": {
"ssl_verification": app.state.config.ENABLE_RAG_WEB_LOADER_SSL_VERIFICATION, "ssl_verification": app.state.config.ENABLE_RAG_WEB_LOADER_SSL_VERIFICATION,
"search": { "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, "engine": app.state.config.RAG_WEB_SEARCH_ENGINE,
"searxng_query_url": app.state.config.SEARXNG_QUERY_URL, "searxng_query_url": app.state.config.SEARXNG_QUERY_URL,
"google_pse_api_key": app.state.config.GOOGLE_PSE_API_KEY, "google_pse_api_key": app.state.config.GOOGLE_PSE_API_KEY,

32
src/lib/components/documents/Settings/WebParams.svelte
View File

@ -58,7 +58,7 @@
{$i18n.t('Enable Web Search')} {$i18n.t('Enable Web Search')}
</div> </div>
<Switch bind:state={webConfig.search.enable} /> <Switch bind:state={webConfig.search.enabled} />
</div> </div>
</div> </div>
@ -191,6 +191,36 @@
{/if} {/if}
</div> </div>
{/if} {/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> </div>
<hr class=" dark:border-gray-850 my-2" /> <hr class=" dark:border-gray-850 my-2" />