refactor: Instead of custom validation of question, rely on the built in FastAPI validator, so generated API docs also reflects this expectation correctly

This commit is contained in:
Aravind Karnam
2025-03-14 09:40:50 +05:30
parent a3954dd4c6
commit c190ba816d

View File

@@ -101,13 +101,9 @@ async def get_markdown(
async def llm_endpoint( async def llm_endpoint(
request: Request, request: Request,
url: str = Path(...), url: str = Path(...),
q: Optional[str] = Query(None), q: str = Query(...),
token_data: Optional[Dict] = Depends(token_dependency) token_data: Optional[Dict] = Depends(token_dependency)
): ):
if not q:
raise HTTPException(status_code=400, detail="Query parameter 'q' is required")
if not url.startswith(('http://', 'https://')):
url = 'https://' + url
try: try:
answer = await handle_llm_qa(url, q, config) answer = await handle_llm_qa(url, q, config)
return JSONResponse({"answer": answer}) return JSONResponse({"answer": answer})
@@ -136,7 +132,6 @@ async def crawl(
): ):
if not crawl_request.urls: if not crawl_request.urls:
raise HTTPException(status_code=400, detail="At least one URL required") raise HTTPException(status_code=400, detail="At least one URL required")
results = await handle_crawl_request( results = await handle_crawl_request(
urls=crawl_request.urls, urls=crawl_request.urls,
browser_config=crawl_request.browser_config, browser_config=crawl_request.browser_config,