|
@@ -392,6 +392,7 @@ async def get_rag_config(request: Request, user=Depends(get_admin_user)):
|
|
"exa_api_key": request.app.state.config.EXA_API_KEY,
|
|
"exa_api_key": request.app.state.config.EXA_API_KEY,
|
|
"result_count": request.app.state.config.RAG_WEB_SEARCH_RESULT_COUNT,
|
|
"result_count": request.app.state.config.RAG_WEB_SEARCH_RESULT_COUNT,
|
|
"concurrent_requests": request.app.state.config.RAG_WEB_SEARCH_CONCURRENT_REQUESTS,
|
|
"concurrent_requests": request.app.state.config.RAG_WEB_SEARCH_CONCURRENT_REQUESTS,
|
|
|
|
+ "domain_filter_list": request.app.state.config.RAG_WEB_SEARCH_DOMAIN_FILTER_LIST,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
}
|
|
@@ -441,6 +442,7 @@ class WebSearchConfig(BaseModel):
|
|
exa_api_key: Optional[str] = None
|
|
exa_api_key: Optional[str] = None
|
|
result_count: Optional[int] = None
|
|
result_count: Optional[int] = None
|
|
concurrent_requests: Optional[int] = None
|
|
concurrent_requests: Optional[int] = None
|
|
|
|
+ domain_filter_list: Optional[List[str]] = []
|
|
|
|
|
|
|
|
|
|
class WebConfig(BaseModel):
|
|
class WebConfig(BaseModel):
|
|
@@ -553,6 +555,9 @@ async def update_rag_config(
|
|
request.app.state.config.RAG_WEB_SEARCH_CONCURRENT_REQUESTS = (
|
|
request.app.state.config.RAG_WEB_SEARCH_CONCURRENT_REQUESTS = (
|
|
form_data.web.search.concurrent_requests
|
|
form_data.web.search.concurrent_requests
|
|
)
|
|
)
|
|
|
|
+ request.app.state.config.RAG_WEB_SEARCH_DOMAIN_FILTER_LIST = (
|
|
|
|
+ form_data.web.search.domain_filter_list
|
|
|
|
+ )
|
|
|
|
|
|
return {
|
|
return {
|
|
"status": True,
|
|
"status": True,
|
|
@@ -599,6 +604,7 @@ async def update_rag_config(
|
|
"exa_api_key": request.app.state.config.EXA_API_KEY,
|
|
"exa_api_key": request.app.state.config.EXA_API_KEY,
|
|
"result_count": request.app.state.config.RAG_WEB_SEARCH_RESULT_COUNT,
|
|
"result_count": request.app.state.config.RAG_WEB_SEARCH_RESULT_COUNT,
|
|
"concurrent_requests": request.app.state.config.RAG_WEB_SEARCH_CONCURRENT_REQUESTS,
|
|
"concurrent_requests": request.app.state.config.RAG_WEB_SEARCH_CONCURRENT_REQUESTS,
|
|
|
|
+ "domain_filter_list": request.app.state.config.RAG_WEB_SEARCH_DOMAIN_FILTER_LIST,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
}
|