Update main.py

main.py

@@ -73,6 +73,11 @@ class DeepResearchRequest(BaseModel):
     query: str
     search_time: int = 300  # Default to 5 minutes
 
+class SearchRequest(BaseModel):
+    query: str
+    search_time: int = 60  # Default: 1 minute for search-only
+    max_results: int = 20  # Number of results to return
+
 app = FastAPI(
     title="AI Deep Research API",
     description="Provides comprehensive research reports from real web searches within 5 minutes.",
@@ -908,6 +913,44 @@ async def deep_research_endpoint(request: DeepResearchRequest):
         headers={"Cache-Control": "no-cache", "Connection": "keep-alive"}
     )
 
+@app.post("/v1/search")
+async def search_only_endpoint(request: SearchRequest):
+    """Search-only endpoint that returns JSON (no streaming)."""
+    if not request.query or len(request.query.strip()) < 3:
+        raise HTTPException(status_code=400, detail="Query must be at least 3 characters long")
+
+    # Clamp durations and limits
+    search_time = min(max(int(request.search_time), 5), 300)
+    max_results = min(max(int(request.max_results), 1), MAX_SOURCES_TO_PROCESS * 2)
+
+    aggregated: List[Dict[str, str]] = []
+    async for update in continuous_search(request.query.strip(), search_time):
+        # We ignore status/warning events; only keep final results
+        if update.get("event") == "final_search_results":
+            aggregated = update.get("data", [])
+
+    # Deduplicate by normalized link
+    dedup: List[Dict[str, str]] = []
+    seen: set = set()
+    for r in aggregated:
+        link = clean_url(r.get("link", ""))
+        title = r.get("title", "")
+        snippet = r.get("snippet", "")
+        if not link:
+            continue
+        if link in seen:
+            continue
+        seen.add(link)
+        dedup.append({"title": title, "link": link, "snippet": snippet})
+        if len(dedup) >= max_results:
+            break
+
+    return {
+        "query": request.query.strip(),
+        "count": len(dedup),
+        "results": dedup,
+    }
+
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=8000)