|
@@ -391,6 +391,8 @@ async def get_rag_config(request: Request, user=Depends(get_admin_user)):
|
|
|
"BYPASS_WEB_SEARCH_EMBEDDING_AND_RETRIEVAL": request.app.state.config.BYPASS_WEB_SEARCH_EMBEDDING_AND_RETRIEVAL,
|
|
|
"SEARXNG_QUERY_URL": request.app.state.config.SEARXNG_QUERY_URL,
|
|
|
"YACY_QUERY_URL": request.app.state.config.YACY_QUERY_URL,
|
|
|
+ "YACY_USERNAME": request.app.state.config.YACY_USERNAME,
|
|
|
+ "YACY_PASSWORD": request.app.state.config.YACY_PASSWORD,
|
|
|
"GOOGLE_PSE_API_KEY": request.app.state.config.GOOGLE_PSE_API_KEY,
|
|
|
"GOOGLE_PSE_ENGINE_ID": request.app.state.config.GOOGLE_PSE_ENGINE_ID,
|
|
|
"BRAVE_SEARCH_API_KEY": request.app.state.config.BRAVE_SEARCH_API_KEY,
|
|
@@ -437,6 +439,8 @@ class WebConfig(BaseModel):
|
|
|
BYPASS_WEB_SEARCH_EMBEDDING_AND_RETRIEVAL: Optional[bool] = None
|
|
|
SEARXNG_QUERY_URL: Optional[str] = None
|
|
|
YACY_QUERY_URL: Optional[str] = None
|
|
|
+ YACY_USERNAME: Optional[str] = None
|
|
|
+ YACY_PASSWORD: Optional[str] = None
|
|
|
GOOGLE_PSE_API_KEY: Optional[str] = None
|
|
|
GOOGLE_PSE_ENGINE_ID: Optional[str] = None
|
|
|
BRAVE_SEARCH_API_KEY: Optional[str] = None
|
|
@@ -655,6 +659,8 @@ async def update_rag_config(
|
|
|
)
|
|
|
request.app.state.config.SEARXNG_QUERY_URL = form_data.web.SEARXNG_QUERY_URL
|
|
|
request.app.state.config.YACY_QUERY_URL = form_data.web.YACY_QUERY_URL
|
|
|
+ request.app.state.config.YACY_USERNAME = form_data.web.YACY_USERNAME
|
|
|
+ request.app.state.config.YACY_PASSWORD = form_data.web.YACY_PASSWORD
|
|
|
request.app.state.config.GOOGLE_PSE_API_KEY = form_data.web.GOOGLE_PSE_API_KEY
|
|
|
request.app.state.config.GOOGLE_PSE_ENGINE_ID = (
|
|
|
form_data.web.GOOGLE_PSE_ENGINE_ID
|
|
@@ -754,6 +760,8 @@ async def update_rag_config(
|
|
|
"BYPASS_WEB_SEARCH_EMBEDDING_AND_RETRIEVAL": request.app.state.config.BYPASS_WEB_SEARCH_EMBEDDING_AND_RETRIEVAL,
|
|
|
"SEARXNG_QUERY_URL": request.app.state.config.SEARXNG_QUERY_URL,
|
|
|
"YACY_QUERY_URL": request.app.state.config.YACY_QUERY_URL,
|
|
|
+ "YACY_USERNAME": request.app.state.config.YACY_USERNAME,
|
|
|
+ "YACY_PASSWORD": request.app.state.config.YACY_PASSWORD,
|
|
|
"GOOGLE_PSE_API_KEY": request.app.state.config.GOOGLE_PSE_API_KEY,
|
|
|
"GOOGLE_PSE_ENGINE_ID": request.app.state.config.GOOGLE_PSE_ENGINE_ID,
|
|
|
"BRAVE_SEARCH_API_KEY": request.app.state.config.BRAVE_SEARCH_API_KEY,
|
|
@@ -1271,7 +1279,7 @@ def search_web(request: Request, engine: str, query: str) -> list[SearchResult]:
|
|
|
"""Search the web using a search engine and return the results as a list of SearchResult objects.
|
|
|
Will look for a search engine API key in environment variables in the following order:
|
|
|
- SEARXNG_QUERY_URL
|
|
|
- - YACY_QUERY_URL
|
|
|
+ - YACY_QUERY_URL + YACY_USERNAME + YACY_PASSWORD
|
|
|
- GOOGLE_PSE_API_KEY + GOOGLE_PSE_ENGINE_ID
|
|
|
- BRAVE_SEARCH_API_KEY
|
|
|
- KAGI_SEARCH_API_KEY
|
|
@@ -1305,6 +1313,8 @@ def search_web(request: Request, engine: str, query: str) -> list[SearchResult]:
|
|
|
if request.app.state.config.YACY_QUERY_URL:
|
|
|
return search_yacy(
|
|
|
request.app.state.config.YACY_QUERY_URL,
|
|
|
+ request.app.state.config.YACY_USERNAME,
|
|
|
+ request.app.state.config.YACY_PASSWORD,
|
|
|
query,
|
|
|
request.app.state.config.WEB_SEARCH_RESULT_COUNT,
|
|
|
request.app.state.config.WEB_SEARCH_DOMAIN_FILTER_LIST,
|