32 lines
1.5 KiB
Python
32 lines
1.5 KiB
Python
|
from typing import Optional, Dict, Any
|
||
|
|
from fastapi import APIRouter, Depends, HTTPException, status
|
||
|
|
from pydantic import BaseModel, Field
|
||
|
|
|
||
|
|
from app.api.llm import get_current_user, get_admin_user, CurrentUser
|
||
|
|
from app.services.web_search_config_store import get_web_search_config, save_web_search_config
|
||
|
|
|
||
|
|
router = APIRouter()
|
||
|
|
|
||
|
|
class WebSearchConfigModel(BaseModel):
|
||
|
|
provider: str = Field(default="duckduckgo", description="Web search provider (brave, tavily, duckduckgo, searxng, jina)")
|
||
|
|
api_key: Optional[str] = Field(default="", description="API Key for the provider")
|
||
|
|
base_url: Optional[str] = Field(default="", description="Base URL for SearXNG")
|
||
|
|
max_results: int = Field(default=5, description="Maximum number of search results")
|
||
|
|
|
||
|
|
def _sanitize_config(config: Dict[str, Any], is_admin: bool) -> Dict[str, Any]:
|
||
|
|
sanitized = config.copy()
|
||
|
|
if not is_admin:
|
||
|
|
sanitized["api_key"] = None
|
||
|
|
return sanitized
|
||
|
|
|
||
|
|
@router.get("/web-search/config", response_model=WebSearchConfigModel)
|
||
|
|
def get_config(current_user: CurrentUser = Depends(get_current_user)):
|
||
|
|
config = get_web_search_config()
|
||
|
|
return WebSearchConfigModel(**_sanitize_config(config, current_user.is_admin))
|
||
|
|
|
||
|
|
@router.put("/web-search/config", response_model=WebSearchConfigModel)
|
||
|
|
def update_config(config: WebSearchConfigModel, _: CurrentUser = Depends(get_admin_user)):
|
||
|
|
config_dict = config.dict()
|
||
|
|
save_web_search_config(config_dict)
|
||
|
|
return WebSearchConfigModel(**config_dict)
|